�Add validation to ensure the validity of themeName from localStorage

BusHanyang · Dec 23, 2024 · 33f0002 · 33f0002
1 parent 651dba4
commit 33f0002
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/app/context/ThemeContext.tsx b/src/app/context/ThemeContext.tsx
@@ -1,6 +1,6 @@
 import React from 'react'
 
-export const enum THEME {
+export enum THEME {
   LIGHT = 'light',
   DARK = 'dark',
   CHRISTMAS = 'christmas',
@@ -22,7 +22,11 @@ export const useDarkmodeContext = () => {
 export const DarkmodeContextProvider = ({
   children,
 }: React.PropsWithChildren) => {
-  const themeName = window.localStorage.getItem('theme') as THEME || THEME.LIGHT
+  const themeName = 
+    Object
+      .values(THEME)
+      .includes(window.localStorage.getItem('theme') as THEME) 
+    ? window.localStorage.getItem('theme') as THEME : THEME.LIGHT
   const [theme, setTheme] = React.useState<THEME>(themeName)
 
   return (