Novo layout

geradoresfe/src/context/ThemeContext.tsx

@@ -1,58 +1,54 @@
-"use client";
-
-import type React from "react";
-import { createContext, useState, useContext, useEffect } from "react";
+import { createContext, useContext, useState, useEffect, ReactNode } from "react";
 
 type Theme = "light" | "dark";
 
 type ThemeContextType = {
-  theme: Theme;
-  toggleTheme: () => void;
+    theme: Theme;
+    toggleTheme: () => void;
 };
 
 const ThemeContext = createContext<ThemeContextType | undefined>(undefined);
 
-export const ThemeProvider: React.FC<{ children: React.ReactNode }> = ({
-  children,
-}) => {
-  const [theme, setTheme] = useState<Theme>("light");
-  const [isInitialized, setIsInitialized] = useState(false);
+export const ThemeProvider = ({ children }: { children: ReactNode }) => {
+    const [theme, setTheme] = useState<Theme | null>(null); // null = tema ainda não carregado
 
-  useEffect(() => {
-    // This code will only run on the client side
-    const savedTheme = localStorage.getItem("theme") as Theme | null;
-    const initialTheme = savedTheme || "light"; // Default to light theme
+    useEffect(() => {
+        const storedTheme = localStorage.getItem("theme") as Theme | null;
+        const prefersDark = window.matchMedia("(prefers-color-scheme: dark)").matches;
+        const initialTheme = storedTheme ?? (prefersDark ? "dark" : "light");
+        setTheme(initialTheme);
+    }, []);
 
-    setTheme(initialTheme);
-    setIsInitialized(true);
-  }, []);
+    useEffect(() => {
+        if (theme === null) return; // ainda a carregar
 
-  useEffect(() => {
-    if (isInitialized) {
-      localStorage.setItem("theme", theme);
-      if (theme === "dark") {
-        document.documentElement.classList.add("dark");
-      } else {
-        document.documentElement.classList.remove("dark");
-      }
-    }
-  }, [theme, isInitialized]);
+        const root = document.documentElement;
+        root.classList.remove("light", "dark");
+        root.classList.add(theme);
+        localStorage.setItem("theme", theme);
+    }, [theme]);
 
-  const toggleTheme = () => {
-    setTheme((prevTheme) => (prevTheme === "light" ? "dark" : "light"));
-  };
+    const toggleTheme = () => {
+        setTheme((prev) => {
+            const nextTheme = prev === "dark" ? "light" : "dark";
+            console.log("ToggleTheme:", prev, "->", nextTheme);
+            return nextTheme;
+        });
+        //setTheme((prev) => (prev === "dark" ? "light" : "dark"));
+    };
 
-  return (
-    <ThemeContext.Provider value={{ theme, toggleTheme }}>
-      {children}
-    </ThemeContext.Provider>
-  );
+    // Enquanto tema não carregou, não renderiza nada
+    if (theme === null) return null;
+
+    return (
+        <ThemeContext.Provider value={{ theme, toggleTheme }}>
+            {children}
+        </ThemeContext.Provider>
+    );
 };
 
-export const useTheme = () => {
-  const context = useContext(ThemeContext);
-  if (context === undefined) {
-    throw new Error("useTheme must be used within a ThemeProvider");
-  }
-  return context;
+export const useTheme = (): ThemeContextType => {
+    const context = useContext(ThemeContext);
+    if (!context) throw new Error("useTheme must be used within ThemeProvider");
+    return context;
 };