|
| 1 | +import { useCallback, useEffect, useState } from "react"; |
| 2 | +import { IThemeContextValue, Theme } from "./ThemeContext"; |
| 3 | +import { darkTheme, lightTheme } from "./themes"; |
| 4 | + |
| 5 | +const LOCAL_STORAGE_THEME_KEY = "get-started-with-agents-app-theme-preference"; |
| 6 | + |
| 7 | +export function useMediaQuery(query: string): boolean { |
| 8 | + const [matches, setMatches] = useState(() => { |
| 9 | + if (typeof window !== "undefined") { |
| 10 | + return window.matchMedia(query).matches; |
| 11 | + } |
| 12 | + return false; |
| 13 | + }); |
| 14 | + |
| 15 | + useEffect(() => { |
| 16 | + if (typeof window !== "undefined") { |
| 17 | + const mediaQuery = window.matchMedia(query); |
| 18 | + const updateMatches = () => { |
| 19 | + setMatches(mediaQuery.matches); |
| 20 | + }; |
| 21 | + |
| 22 | + mediaQuery.addEventListener("change", updateMatches); |
| 23 | + |
| 24 | + updateMatches(); |
| 25 | + return () => { |
| 26 | + mediaQuery.removeEventListener("change", updateMatches); |
| 27 | + }; |
| 28 | + } |
| 29 | + return () => { |
| 30 | + // Cleanup if needed |
| 31 | + }; |
| 32 | + }, [query]); |
| 33 | + |
| 34 | + return matches; |
| 35 | +} |
| 36 | + |
| 37 | +export const useThemeProvider = (): IThemeContextValue => { |
| 38 | + const prefersDark = useMediaQuery("(prefers-color-scheme: dark)"); |
| 39 | + |
| 40 | + const [savedTheme, setSavedTheme] = useState<Theme>(() => { |
| 41 | + if (typeof localStorage !== "undefined") { |
| 42 | + const storedTheme = localStorage.getItem( |
| 43 | + LOCAL_STORAGE_THEME_KEY |
| 44 | + ) as Theme; |
| 45 | + if (storedTheme && ["Light", "Dark", "System"].includes(storedTheme)) { |
| 46 | + return storedTheme; |
| 47 | + } |
| 48 | + } |
| 49 | + return "System"; |
| 50 | + }); |
| 51 | + |
| 52 | + const isDarkMode = |
| 53 | + savedTheme === "System" ? prefersDark : savedTheme === "Dark"; |
| 54 | + |
| 55 | + const currentTheme = isDarkMode ? "Dark" : "Light"; |
| 56 | + |
| 57 | + const themeStyles = isDarkMode ? darkTheme : lightTheme; |
| 58 | + |
| 59 | + const setTheme = useCallback((newTheme: Theme) => { |
| 60 | + setSavedTheme(newTheme); |
| 61 | + if (typeof localStorage !== "undefined") { |
| 62 | + localStorage.setItem(LOCAL_STORAGE_THEME_KEY, newTheme); |
| 63 | + } |
| 64 | + }, []); |
| 65 | + |
| 66 | + useEffect(() => { |
| 67 | + // Update document color scheme for browser UI |
| 68 | + if (typeof document !== "undefined") { |
| 69 | + document.documentElement.style.colorScheme = isDarkMode |
| 70 | + ? "dark" |
| 71 | + : "light"; |
| 72 | + } |
| 73 | + }, [isDarkMode]); |
| 74 | + |
| 75 | + return { |
| 76 | + theme: savedTheme, |
| 77 | + savedTheme, |
| 78 | + currentTheme, |
| 79 | + themeStyles, |
| 80 | + setTheme, |
| 81 | + isDarkMode, |
| 82 | + }; |
| 83 | +}; |
0 commit comments