UNPKG

@vela-ui/react

Version:

Vela UI React components

48 lines (46 loc) 1.33 kB
// src/components/provider/theme-provider.tsx import { createContext, useContext, useEffect, useState } from "react"; import { jsx } from "react/jsx-runtime"; var initialState = { theme: "system", setTheme: () => null }; var ThemeProviderContext = createContext(initialState); function ThemeProvider({ children, defaultTheme = "system", storageKey = "vela-ui-theme", ...props }) { const [theme, setTheme] = useState( () => localStorage.getItem(storageKey) || defaultTheme ); useEffect(() => { const root = window.document.documentElement; root.classList.remove("light", "dark"); if (theme === "system") { const systemTheme = window.matchMedia("(prefers-color-scheme: dark)").matches ? "dark" : "light"; root.classList.add(systemTheme); return; } root.classList.add(theme); }, [theme]); const value = { theme, setTheme: (theme2) => { localStorage.setItem(storageKey, theme2); setTheme(theme2); } }; return /* @__PURE__ */ jsx(ThemeProviderContext.Provider, { ...props, value, children }); } var useTheme = () => { const context = useContext(ThemeProviderContext); if (context === void 0) throw new Error("useTheme must be used within a ThemeProvider"); return context; }; export { ThemeProviderContext, ThemeProvider, useTheme };