初始化夜间模式

lib/theme.js

@@ -2,31 +2,33 @@ import cookie from 'react-cookies'
 
 /**
  * 初始化主题
- * @param theme 用户默认主题state
+ * @param
+ * @param isDarkMode
  * @param updateDarkMode 更改主题ChangeState函数
  * @description 读取cookie中存的用户主题
  */
-export const initTheme = (isDarkMode, updateDarkMode) => {
-  // 若未指定主题，则从时间和浏览器偏好中决定初始主题
+export const initDarkMode = (isDarkMode, updateDarkMode) => {
   if (!isDarkMode) {
-    const date = new Date()
-    const prefersDarkMode = window.matchMedia('(prefers-color-scheme: dark)').matches
-    const useDark = prefersDarkMode || (date.getHours() >= 18 || date.getHours() < 6)
-    if (useDark) {
-      isDarkMode = true
-    } else {
-      isDarkMode = false
-    }
+    isDarkMode = isPreferDark()
   }
   if (typeof window !== 'undefined') {
-    const htmlElement = document.getElementsByTagName('html')
-    htmlElement.className = ''
     updateDarkMode(isDarkMode)
     saveDarkModeToCookies(isDarkMode)
-    htmlElement.classList?.add(isDarkMode)
+    document.getElementsByTagName('html')[0].setAttribute('class', isDarkMode ? 'dark' : 'light')
   }
 }
 
+/**
+ * 是否优先深色模式
+ * @returns {*}
+ */
+export function isPreferDark () {
+  // 系统深色模式或时间是夜间时，强行置为夜间模式
+  const date = new Date()
+  const prefersDarkMode = window.matchMedia('(prefers-color-scheme: dark)').matches
+  return prefersDarkMode || (date.getHours() >= 18 || date.getHours() < 6)
+}
+
 /**
  * 读取默认主题
  * @returns {*}