Update Theme.Togglr.jsx

Akshatchaube01 · Jul 21, 2024 · d7fadb3 · d7fadb3
1 parent 1ad4b53
commit d7fadb3
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/components/ThemeToggle.jsx b/src/components/ThemeToggle.jsx
@@ -6,15 +6,19 @@ const ThemeToggle = () => {
     useEffect(() => {
         const savedTheme = localStorage.getItem('theme') || 'light';
         setTheme(savedTheme);
-        document.documentElement.classList.add(`theme-${savedTheme}`);
+        applyTheme(savedTheme);
     }, []);
 
+    const applyTheme = (selectedTheme) => {
+        document.documentElement.classList.remove(`theme-${theme}`);
+        document.documentElement.classList.add(`theme-${selectedTheme}`);
+        localStorage.setItem('theme', selectedTheme);
+    };
+
     const toggleTheme = () => {
         const newTheme = theme === 'light' ? 'dark' : 'light';
         setTheme(newTheme);
-        document.documentElement.classList.remove(`theme-${theme}`);
-        document.documentElement.classList.add(`theme-${newTheme}`);
-        localStorage.setItem('theme', newTheme);
+        applyTheme(newTheme);
     };
 
     return (
@@ -34,4 +38,3 @@ const ThemeToggle = () => {
 };
 
 export default ThemeToggle;
-