fix: :bug: ThemeToggle should initialize from localstorage immediately (#748)

Co-authored-by: Tony Sullivan <tony.f.sullivan@gmail.com>

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;