From 6271e513766519db1325364ac26a7fbd0761a0da Mon Sep 17 00:00:00 2001
From: Alex Hua <sealexh@gmail.com>
Date: Sat, 8 Apr 2023 21:11:22 +0800
Subject: [PATCH] Make auto lang more robust

---
 layouts/partials/head/autoLang.html | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/layouts/partials/head/autoLang.html b/layouts/partials/head/autoLang.html
index 1bbbbb1..15805fb 100644
--- a/layouts/partials/head/autoLang.html
+++ b/layouts/partials/head/autoLang.html
@@ -1,27 +1,25 @@
 {{ if .Site.IsMultiLingual }}
 <script>
     (function () {
-        const BASE_URL = '{{ .Site.BaseURL }}';
-        const DEFAULT_LANG = '{{ .Site.Language.Lang }}';
-
-        if (!window.location.href.endsWith(BASE_URL)) return;
-
+        if (window.location.pathname !== '/') return;        
+        
+        const CURRENT_LANG = '{{ .Site.Language.Lang }}';
+        
         let userLangList = [...navigator.languages];
         if (localStorage.userLang)
             userLangList.unshift(localStorage.userLang);
-        else
-            localStorage.userLang = userLangList[0].toLowerCase();
-
+        
         let supportLangs = new Set();
         {{ range .Site.Languages }}
         supportLangs.add('{{ .Lang }}');
         {{ end }}
         for (let userLang of userLangList) {
             userLang = userLang.toLowerCase();
-            if (DEFAULT_LANG == userLang) {
+            if (CURRENT_LANG == userLang) {
                 return;
             }
             if (supportLangs.has(userLang)) {
+                localStorage.userLang = userLang;
                 window.location.href = window.location.href + userLang;
                 return;
             }