From f5f604094e1fc8aebf23ab47d02e67228942bb00 Mon Sep 17 00:00:00 2001
From: tangly1024 <tlyong1992@hotmail.com>
Date: Wed, 16 Mar 2022 12:19:51 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=BB=E9=A2=98=E5=88=9D=E5=A7=8B=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 lib/global.js | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/lib/global.js b/lib/global.js
index 0432b5e0..c9564cce 100644
--- a/lib/global.js
+++ b/lib/global.js
@@ -17,9 +17,7 @@ export function GlobalContextProvider ({ children }) {
   const [locale, updateLocale] = useState(generateLocaleDict('en-US'))
   const [isDarkMode, updateDarkMode] = useState(false)
   const [onLoading, changeLoadingState] = useState(false)
-  // 会优先读取浏览器中的主题参数
-  const [theme, setTheme] = useState(Router?.router?.query?.theme || BLOG.THEME)
-
+  const [theme, setTheme] = useState(BLOG.THEME)
   Router.events.on('routeChangeStart', (...args) => {
     changeLoadingState(true)
   })
@@ -30,6 +28,11 @@ export function GlobalContextProvider ({ children }) {
 
   useEffect(() => {
     if (!hasInit) {
+      const userTheme = getQueryVariable('theme')
+      console.log('用户主题', userTheme)
+      if (userTheme && ALL_THEME.indexOf(userTheme) > -1) {
+        setTheme(userTheme)
+      }
       initLocale(locale, updateLocale)
       initDarkMode(isDarkMode, updateDarkMode)
       hasInit = true
@@ -43,6 +46,16 @@ export function GlobalContextProvider ({ children }) {
   )
 }
 
+function getQueryVariable (variable) {
+  const query = window.location.search.substring(1)
+  const vars = query.split('&')
+  for (let i = 0; i < vars.length; i++) {
+    const pair = vars[i].split('=')
+    if (pair[0] == variable) { return pair[1] }
+  }
+  return (false)
+}
+
 /**
  * 深度合并两个对象
  * @param target