"fix" theme preview

2 files changed, 13 insertions, 3 deletions

diff --git a/src/services/style_setter/style_setter.js b/src/services/style_setter/style_setter.js
index 9e068a1e..78e7428d 100644
--- a/src/services/style_setter/style_setter.js
+++ b/src/services/style_setter/style_setter.js
@@ -61,7 +61,11 @@ export const generateTheme = async (input, callbacks, debug) => {
   }
 
   // Assuming that "worst case scenario background" is panel background since it's the most likely one
-  const themes3 = init(extraRules, extraRules[0].directives['--bg'].split('|')[1].trim(), debug)
+  const themes3 = init({
+    extraRules,
+    ultimateBackgroundColor: extraRules[0].directives['--bg'].split('|')[1].trim(),
+    debug
+  })
 
   console.log('DEBUG 2 IS', debug)
 
diff --git a/src/services/theme_data/theme_data_3.service.js b/src/services/theme_data/theme_data_3.service.js
index d1043c52..e98b19a7 100644
--- a/src/services/theme_data/theme_data_3.service.js
+++ b/src/services/theme_data/theme_data_3.service.js
@@ -149,7 +149,13 @@ const ruleToSelector = genericRuleToSelector(components)
 
 export const getEngineChecksum = () => engineChecksum
 
-export const init = (extraRuleset, ultimateBackgroundColor, debug) => {
+export const init = ({
+  extraRuleset,
+  ultimateBackgroundColor,
+  debug = false,
+  liteMode = false,
+  rootComponentName = 'Root'
+}) => {
   const staticVars = {}
   const stacked = {}
   const computed = {}
@@ -451,7 +457,7 @@ export const init = (extraRuleset, ultimateBackgroundColor, debug) => {
   }
 
   const t0 = performance.now()
-  const combinations = processInnerComponent(components.Root)
+  const combinations = processInnerComponent(components[rootComponentName] ?? components.Root)
   const t1 = performance.now()
   console.debug('Tree traveral took ' + (t1 - t0) + ' ms')