fix: useTheme hook when rainbow prop does not exist in context

nexxtway · Sep 17, 2020 · 8966169 · 8966169
1 parent ea1ddfc
commit 8966169
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/src/libs/hooks/useTheme.js b/src/libs/hooks/useTheme.js
@@ -5,14 +5,20 @@ import defaultTheme from '../../styles/defaultTheme';
 
 export default function useTheme() {
     const themeContext = useContext(ThemeContext);
-    return useMemo(
-        () =>
+    return useMemo(() => {
+        if (themeContext && !themeContext.rainbow) {
+            themeContext.rainbow = {
+                ...defaultTheme,
+                ...normalizeThemeColors(defaultTheme),
+            };
+        }
+        return (
             themeContext || {
                 rainbow: {
                     ...defaultTheme,
                     ...normalizeThemeColors(defaultTheme),
                 },
-            },
-        [themeContext],
-    );
+            }
+        );
+    }, [themeContext]);
 }