feat: add ragflow web project & add pnpm workspace file
This commit is contained in:
73
ragflow_web/src/components/theme-provider.tsx
Normal file
73
ragflow_web/src/components/theme-provider.tsx
Normal file
@@ -0,0 +1,73 @@
|
||||
import { ThemeEnum } from '@/constants/common';
|
||||
import React, { createContext, useContext, useEffect, useState } from 'react';
|
||||
|
||||
type ThemeProviderProps = {
|
||||
children: React.ReactNode;
|
||||
defaultTheme?: ThemeEnum;
|
||||
storageKey?: string;
|
||||
};
|
||||
|
||||
type ThemeProviderState = {
|
||||
theme: ThemeEnum;
|
||||
setTheme: (theme: ThemeEnum) => void;
|
||||
};
|
||||
|
||||
const initialState: ThemeProviderState = {
|
||||
theme: ThemeEnum.Light,
|
||||
setTheme: () => null,
|
||||
};
|
||||
|
||||
const ThemeProviderContext = createContext<ThemeProviderState>(initialState);
|
||||
|
||||
export function ThemeProvider({
|
||||
children,
|
||||
defaultTheme = ThemeEnum.Dark,
|
||||
storageKey = 'vite-ui-theme',
|
||||
...props
|
||||
}: ThemeProviderProps) {
|
||||
const [theme, setTheme] = useState<ThemeEnum>(
|
||||
() => (localStorage.getItem(storageKey) as ThemeEnum) || defaultTheme,
|
||||
);
|
||||
|
||||
useEffect(() => {
|
||||
const root = window.document.documentElement;
|
||||
root.classList.remove(ThemeEnum.Light, ThemeEnum.Dark);
|
||||
localStorage.setItem(storageKey, theme);
|
||||
root.classList.add(theme);
|
||||
}, [storageKey, theme]);
|
||||
|
||||
return (
|
||||
<ThemeProviderContext.Provider
|
||||
{...props}
|
||||
value={{
|
||||
theme,
|
||||
setTheme,
|
||||
}}
|
||||
>
|
||||
{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;
|
||||
};
|
||||
|
||||
export const useIsDarkTheme = () => {
|
||||
const { theme } = useTheme();
|
||||
|
||||
return theme === ThemeEnum.Dark;
|
||||
};
|
||||
|
||||
export function useSwitchToDarkThemeOnMount() {
|
||||
const { setTheme } = useTheme();
|
||||
|
||||
useEffect(() => {
|
||||
setTheme(ThemeEnum.Dark);
|
||||
}, [setTheme]);
|
||||
}
|
||||
Reference in New Issue
Block a user