feat: add theme property checkup

traefik · Jan 30, 2024 · ca565c4 · ca565c4
1 parent adb6b34
commit ca565c4
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/App.tsx b/src/App.tsx
@@ -1,11 +1,19 @@
 import HubButton from 'components/HubButton'
 import { darkTheme, lightTheme } from 'components/themes'
+import { useMemo } from 'react'
 import { ThemeProvider } from 'styled-components'
 
 export const App = ({ theme = 'light' }: { theme?: 'light' | 'dark' }) => {
+  const safeTheme = useMemo(() => {
+    if (!['light', 'dark'].includes(theme)) {
+      return 'light'
+    }
+    return theme
+  }, [theme])
+
   return (
-    <ThemeProvider theme={theme === 'dark' ? darkTheme : lightTheme}>
-      <HubButton theme={theme} />
+    <ThemeProvider theme={safeTheme === 'dark' ? darkTheme : lightTheme}>
+      <HubButton theme={safeTheme} />
     </ThemeProvider>
   )
 }