Merge pull request #185 from Develonaut/patch-1

Persist current store value
hipstersmoothie · Sep 14, 2022 · 3a2d770 · 3a2d770
2 parents 3b61d46 + 41a6972
commit 3a2d770
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/Tool.tsx b/src/Tool.tsx
@@ -179,8 +179,13 @@ export const DarkMode = ({ api }: DarkModeProps) => {
   /** When storybook params change update the stored themes */
   React.useEffect(() => {
     const currentStore = store();
-
-    updateStore({ ...currentStore, ...darkModeParams });
+    // Ensure we use the stores `current` value first to persist
+    // themeing between page loads and story changes.
+    updateStore({
+      ...currentStore,
+      ...darkModeParams,
+      current: currentStore.current || darkModeParams.current,
+    });
     renderTheme()
   }, [darkModeParams, renderTheme])