diff options
Diffstat (limited to 'docs/src/components/ThemeToggle.tsx')
-rw-r--r-- | docs/src/components/ThemeToggle.tsx | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/docs/src/components/ThemeToggle.tsx b/docs/src/components/ThemeToggle.tsx index 31ab5ea74..95de4fba1 100644 --- a/docs/src/components/ThemeToggle.tsx +++ b/docs/src/components/ThemeToggle.tsx @@ -43,13 +43,13 @@ const icons = [ ]; const ThemeToggle: FunctionalComponent = () => { - const [theme, setTheme] = useState(themes[0]); + const [theme, setTheme] = useState(() => { + if (typeof localStorage === 'undefined') { + return themes[0]; + } - useEffect(() => { - const user = localStorage.getItem('theme'); - if (!user) return; - setTheme(user); - }, []); + return localStorage.getItem('theme') || themes[0]; + }); useEffect(() => { const root = document.documentElement; |