@vela-ui/react
Version:
Vela UI React components
48 lines (46 loc) • 1.33 kB
JavaScript
// 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
};