fix(ui): useDetectTheme not updating if it was not auto

itsjavi · Aug 30, 2023 · a2c38c5 · a2c38c5
1 parent 247c2b5
commit a2c38c5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/storylite/src/app/hooks/useDetectTheme.ts b/packages/storylite/src/app/hooks/useDetectTheme.ts
@@ -19,7 +19,9 @@ export function useDetectTheme(): string {
       const isAuto = params.theme.value === 'auto' || !params.theme.value
 
       if (!isAuto) {
-        document.body.setAttribute(themeAttrName, String(params.theme.value ?? ''))
+        const value = String(params.theme.value ?? '')
+        document.body.setAttribute(themeAttrName, value)
+        setDetectedTheme(value as SLColorScheme)
 
         return
       }
@@ -35,7 +37,7 @@ export function useDetectTheme(): string {
     return () => {
       mediaQuery.removeEventListener('change', handleThemeChange)
     }
-  }, [params])
+  }, [params.theme.value, userConfig.themeAttribute])
 
   return detectedTheme
 }