|
|
@@ -17,39 +17,90 @@ along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
For commercial licensing, please contact support@quantumnous.com
|
|
|
*/
|
|
|
|
|
|
-import { createContext, useCallback, useContext, useState } from 'react';
|
|
|
+import { createContext, useCallback, useContext, useState, useEffect } from 'react';
|
|
|
|
|
|
const ThemeContext = createContext(null);
|
|
|
export const useTheme = () => useContext(ThemeContext);
|
|
|
|
|
|
+const ActualThemeContext = createContext(null);
|
|
|
+export const useActualTheme = () => useContext(ActualThemeContext);
|
|
|
+
|
|
|
const SetThemeContext = createContext(null);
|
|
|
export const useSetTheme = () => useContext(SetThemeContext);
|
|
|
|
|
|
+// 检测系统主题偏好
|
|
|
+const getSystemTheme = () => {
|
|
|
+ if (typeof window !== 'undefined' && window.matchMedia) {
|
|
|
+ return window.matchMedia('(prefers-color-scheme: dark)').matches ? 'dark' : 'light';
|
|
|
+ }
|
|
|
+ return 'light';
|
|
|
+};
|
|
|
+
|
|
|
export const ThemeProvider = ({ children }) => {
|
|
|
const [theme, _setTheme] = useState(() => {
|
|
|
try {
|
|
|
- return localStorage.getItem('theme-mode') || null;
|
|
|
+ return localStorage.getItem('theme-mode') || 'auto';
|
|
|
} catch {
|
|
|
- return null;
|
|
|
+ return 'auto';
|
|
|
}
|
|
|
});
|
|
|
|
|
|
- const setTheme = useCallback((input) => {
|
|
|
- _setTheme(input ? 'dark' : 'light');
|
|
|
+ const [systemTheme, setSystemTheme] = useState(getSystemTheme());
|
|
|
+
|
|
|
+ // 计算实际应用的主题
|
|
|
+ const actualTheme = theme === 'auto' ? systemTheme : theme;
|
|
|
+
|
|
|
+ // 监听系统主题变化
|
|
|
+ useEffect(() => {
|
|
|
+ if (typeof window !== 'undefined' && window.matchMedia) {
|
|
|
+ const mediaQuery = window.matchMedia('(prefers-color-scheme: dark)');
|
|
|
|
|
|
+ const handleSystemThemeChange = (e) => {
|
|
|
+ setSystemTheme(e.matches ? 'dark' : 'light');
|
|
|
+ };
|
|
|
+
|
|
|
+ mediaQuery.addEventListener('change', handleSystemThemeChange);
|
|
|
+
|
|
|
+ return () => {
|
|
|
+ mediaQuery.removeEventListener('change', handleSystemThemeChange);
|
|
|
+ };
|
|
|
+ }
|
|
|
+ }, []);
|
|
|
+
|
|
|
+ // 应用主题到DOM
|
|
|
+ useEffect(() => {
|
|
|
const body = document.body;
|
|
|
- if (!input) {
|
|
|
+ if (actualTheme === 'dark') {
|
|
|
+ body.setAttribute('theme-mode', 'dark');
|
|
|
+ document.documentElement.classList.add('dark');
|
|
|
+ } else {
|
|
|
body.removeAttribute('theme-mode');
|
|
|
- localStorage.setItem('theme-mode', 'light');
|
|
|
+ document.documentElement.classList.remove('dark');
|
|
|
+ }
|
|
|
+ }, [actualTheme]);
|
|
|
+
|
|
|
+ const setTheme = useCallback((newTheme) => {
|
|
|
+ let themeValue;
|
|
|
+
|
|
|
+ if (typeof newTheme === 'boolean') {
|
|
|
+ // 向后兼容原有的 boolean 参数
|
|
|
+ themeValue = newTheme ? 'dark' : 'light';
|
|
|
+ } else if (typeof newTheme === 'string') {
|
|
|
+ // 新的字符串参数支持 'light', 'dark', 'auto'
|
|
|
+ themeValue = newTheme;
|
|
|
} else {
|
|
|
- body.setAttribute('theme-mode', 'dark');
|
|
|
- localStorage.setItem('theme-mode', 'dark');
|
|
|
+ themeValue = 'auto';
|
|
|
}
|
|
|
+
|
|
|
+ _setTheme(themeValue);
|
|
|
+ localStorage.setItem('theme-mode', themeValue);
|
|
|
}, []);
|
|
|
|
|
|
return (
|
|
|
<SetThemeContext.Provider value={setTheme}>
|
|
|
- <ThemeContext.Provider value={theme}>{children}</ThemeContext.Provider>
|
|
|
+ <ActualThemeContext.Provider value={actualTheme}>
|
|
|
+ <ThemeContext.Provider value={theme}>{children}</ThemeContext.Provider>
|
|
|
+ </ActualThemeContext.Provider>
|
|
|
</SetThemeContext.Provider>
|
|
|
);
|
|
|
};
|