feat: Root page refers language registered from cookie

Refs: #1
atsphinx · May 14, 2024 · 93bf728 · 93bf728
1 parent 688b975
commit 93bf728
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/atsphinx/mini18n/templates/mini18n/index.html b/src/atsphinx/mini18n/templates/mini18n/index.html
@@ -10,12 +10,18 @@
 <body>
   <script>
     const url = new URL(window.location.href);
+    const default_language = '{{ config.mini18n_default_language }}';
     const targets = [
       '{{ config.mini18n_basepath }}',
       '{{ config.mini18n_basepath }}/index.html',
     ];
     if (targets.includes(url.pathname)) {
-      url.pathname = '{{ config.mini18n_basepath }}{{ config.mini18n_default_language }}/';
+      const fromCookie = document.cookie.split("; ").find(kv => kv.startsWith("lang="));
+      if (fromCookie) {
+        url.pathname = `{{ config.mini18n_basepath }}${fromCookie.split("=")[1]}/`;
+      } else {
+        url.pathname = '{{ config.mini18n_basepath }}{{ config.mini18n_default_language }}/';
+      }
       location.href = url.href;
     }
   </script>

diff --git a/tests/test_javascript.py b/tests/test_javascript.py
@@ -78,3 +78,6 @@ def test__select_language(app: SphinxTestApp, page: Page):
         cookies = page.context.cookies()
         assert cookies[0]["name"] == "lang"
         assert cookies[0]["value"] == "ja"
+        page.goto(url)
+        time.sleep(0.5)
+        assert page.url.endswith("/ja/")