From d0d0955c0e811ce951640cc7ec0e82ef64ff6276 Mon Sep 17 00:00:00 2001
From: JimmFly <yangjinfei001@gmail.com>
Date: Tue, 10 Jan 2023 18:47:13 +0800
Subject: [PATCH] feat: add language local storage

---
 packages/i18n/src/index.ts | 49 +++++++++++++++++++++-----------------
 1 file changed, 27 insertions(+), 22 deletions(-)

diff --git a/packages/i18n/src/index.ts b/packages/i18n/src/index.ts
index 4095c417ad..714cc40f31 100644
--- a/packages/i18n/src/index.ts
+++ b/packages/i18n/src/index.ts
@@ -1,6 +1,5 @@
 import i18next, { Resource } from 'i18next';
 import { Trans, initReactI18next, useTranslation } from 'react-i18next';
-import detector from 'i18next-browser-languagedetector';
 import { LOCALES } from './resources/index.js';
 import type en_US from './resources/en.json';
 
@@ -22,7 +21,7 @@ declare module 'react-i18next' {
   }
 }
 
-// const STORAGE_KEY = 'i18n_lng';
+const STORAGE_KEY = 'i18n_lng';
 
 export { Trans, i18n, useTranslation, LOCALES };
 
@@ -34,33 +33,39 @@ const resources = LOCALES.reduce<Resource>(
 const fallbackLng = LOCALES[0].tag;
 const standardizeLocale = (language: string) => {
   if (LOCALES.find(locale => locale.tag === language)) return language;
+  if (language === 'zh-CN' || language === 'zh') {
+    return 'zh-Hans';
+  }
+  if (language.slice(0, 2).toLowerCase() === 'zh') {
+    return 'zh-Hant';
+  }
   if (LOCALES.find(locale => locale.tag === language.slice(0, 2).toLowerCase()))
     return language;
   return fallbackLng;
 };
+let language = 'en';
 
-const language = standardizeLocale(
-  //   localStorage.getItem(STORAGE_KEY) ??
-  //     (typeof navigator !== 'undefined' ? navigator.language : 'en')
-  'en'
-);
-
+if (typeof window !== 'undefined') {
+  const localStorageLanguage = localStorage.getItem(STORAGE_KEY);
+  if (localStorageLanguage) {
+    language = standardizeLocale(localStorageLanguage);
+  } else {
+    language = standardizeLocale(navigator.language);
+  }
+}
 const i18n = i18next.createInstance();
-i18n
-  .use(detector)
-  .use(initReactI18next)
-  .init({
-    lng: language,
-    fallbackLng,
-    debug: false,
-    resources,
-    interpolation: {
-      escapeValue: false, // not needed for react as it escapes by default
-    },
-  });
+i18n.use(initReactI18next).init({
+  lng: language,
+  fallbackLng,
+  debug: false,
+  resources,
+  interpolation: {
+    escapeValue: false, // not needed for react as it escapes by default
+  },
+});
 
-i18n.on('languageChanged', () => {
-  //   localStorage.setItem(STORAGE_KEY, lng);
+i18n.on('languageChanged', lng => {
+  localStorage.setItem(STORAGE_KEY, lng);
 });
 
 // const I18nProvider = I18nextProvider;