42 lines
1.1 KiB
TypeScript
42 lines
1.1 KiB
TypeScript
'use client';
|
|
|
|
import { createContext, useContext, useEffect, useState } from 'react';
|
|
|
|
type Theme = 'light' | 'dark';
|
|
|
|
interface ThemeContextValue {
|
|
theme: Theme;
|
|
toggleTheme: () => void;
|
|
}
|
|
|
|
const ThemeContext = createContext<ThemeContextValue>({
|
|
theme: 'dark',
|
|
toggleTheme: () => {},
|
|
});
|
|
|
|
export const useTheme = () => useContext(ThemeContext);
|
|
|
|
export const ThemeProvider = ({ children }: { children: React.ReactNode }) => {
|
|
const [theme, setTheme] = useState<Theme>('dark');
|
|
const [mounted, setMounted] = useState(false);
|
|
|
|
useEffect(() => {
|
|
const stored = localStorage.getItem('theme') as Theme | null;
|
|
const initial = stored || 'dark';
|
|
setTheme(initial);
|
|
document.documentElement.classList.toggle('dark', initial === 'dark');
|
|
setMounted(true);
|
|
}, []);
|
|
|
|
const toggleTheme = () => {
|
|
const next = theme === 'dark' ? 'light' : 'dark';
|
|
setTheme(next);
|
|
localStorage.setItem('theme', next);
|
|
document.documentElement.classList.toggle('dark', next === 'dark');
|
|
};
|
|
|
|
if (!mounted) return null;
|
|
|
|
return <ThemeContext.Provider value={{ theme, toggleTheme }}>{children}</ThemeContext.Provider>;
|
|
};
|