94 lines
2.5 KiB
TypeScript
94 lines
2.5 KiB
TypeScript
import storage from '@/services/local-storage.service'
|
|
import { TTheme, TThemeSetting } from '@/types'
|
|
import { createContext, useContext, useEffect, useState } from 'react'
|
|
|
|
type ThemeProviderProps = {
|
|
children: React.ReactNode
|
|
defaultTheme?: TTheme
|
|
}
|
|
|
|
type ThemeProviderState = {
|
|
themeSetting: TThemeSetting
|
|
theme: TTheme
|
|
setThemeSetting: (themeSetting: TThemeSetting) => Promise<void>
|
|
}
|
|
|
|
function getSystemTheme() {
|
|
return window.matchMedia('(prefers-color-scheme: dark)').matches ? 'dark' : 'light'
|
|
}
|
|
|
|
const ThemeProviderContext = createContext<ThemeProviderState | undefined>(undefined)
|
|
|
|
export function ThemeProvider({ children, ...props }: ThemeProviderProps) {
|
|
const [themeSetting, setThemeSetting] = useState<TThemeSetting>(
|
|
(localStorage.getItem('themeSetting') as TThemeSetting | null) ?? 'system'
|
|
)
|
|
const [theme, setTheme] = useState<TTheme>('light')
|
|
|
|
useEffect(() => {
|
|
const init = async () => {
|
|
const themeSetting = storage.getThemeSetting()
|
|
if (themeSetting === 'system') {
|
|
setTheme(getSystemTheme())
|
|
return
|
|
}
|
|
setTheme(themeSetting)
|
|
}
|
|
|
|
init()
|
|
}, [])
|
|
|
|
useEffect(() => {
|
|
if (themeSetting !== 'system') return
|
|
|
|
const mediaQuery = window.matchMedia('(prefers-color-scheme: dark)')
|
|
const handleChange = (e: MediaQueryListEvent) => {
|
|
setTheme(e.matches ? 'dark' : 'light')
|
|
}
|
|
mediaQuery.addEventListener('change', handleChange)
|
|
setTheme(mediaQuery.matches ? 'dark' : 'light')
|
|
|
|
return () => {
|
|
mediaQuery.removeEventListener('change', handleChange)
|
|
}
|
|
}, [themeSetting])
|
|
|
|
useEffect(() => {
|
|
const updateTheme = async () => {
|
|
const root = window.document.documentElement
|
|
root.classList.remove('light', 'dark')
|
|
root.classList.add(theme)
|
|
localStorage.setItem('theme', theme)
|
|
}
|
|
updateTheme()
|
|
}, [theme])
|
|
|
|
return (
|
|
<ThemeProviderContext.Provider
|
|
{...props}
|
|
value={{
|
|
themeSetting: themeSetting,
|
|
theme: theme,
|
|
setThemeSetting: async (themeSetting: TThemeSetting) => {
|
|
storage.setThemeSetting(themeSetting)
|
|
setThemeSetting(themeSetting)
|
|
if (themeSetting === 'system') {
|
|
setTheme(getSystemTheme())
|
|
return
|
|
}
|
|
setTheme(themeSetting)
|
|
}
|
|
}}
|
|
>
|
|
{children}
|
|
</ThemeProviderContext.Provider>
|
|
)
|
|
}
|
|
|
|
export const useTheme = () => {
|
|
const context = useContext(ThemeProviderContext)
|
|
|
|
if (context === undefined) throw new Error('useTheme must be used within a ThemeProvider')
|
|
|
|
return context
|
|
}
|