From b3f722d8fff1c328594329909da650d9d494d735 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=B4=94=E5=BA=86=E6=89=8D=E4=B8=A8=E9=9D=99=E8=A7=85?=
 <cqc@cuiqingcai.com>
Date: Sat, 11 May 2024 17:39:35 +0800
Subject: [PATCH] align locale theme across sites (#38)

---
 ...-949fa604-3730-4605-a6cc-506cd925fda6.json |  7 ++
 src/components/common/DarkSelector.vue        | 19 +++---
 src/components/common/LocaleSelector.vue      | 14 +++-
 src/i18n/index.ts                             |  4 +-
 src/main.ts                                   | 64 ++++++++++---------
 src/router/index.ts                           |  4 +-
 src/store/common/actions.ts                   | 10 ---
 src/store/common/getters.ts                   | 10 ---
 src/store/common/models.ts                    |  2 -
 src/store/common/mutations.ts                 | 10 ---
 src/store/common/state.ts                     |  7 +-
 src/utils/initializer.ts                      | 60 ++++++++++++++---
 12 files changed, 122 insertions(+), 89 deletions(-)
 create mode 100644 change/@acedatacloud-hub-949fa604-3730-4605-a6cc-506cd925fda6.json

diff --git a/change/@acedatacloud-hub-949fa604-3730-4605-a6cc-506cd925fda6.json b/change/@acedatacloud-hub-949fa604-3730-4605-a6cc-506cd925fda6.json
new file mode 100644
index 00000000..a5ca9ce3
--- /dev/null
+++ b/change/@acedatacloud-hub-949fa604-3730-4605-a6cc-506cd925fda6.json
@@ -0,0 +1,7 @@
+{
+  "type": "patch",
+  "comment": "align across site for theme and locale",
+  "packageName": "@acedatacloud/hub",
+  "email": "cqc@cuiqingcai.com",
+  "dependentChangeType": "patch"
+}
diff --git a/src/components/common/DarkSelector.vue b/src/components/common/DarkSelector.vue
index a7a5b10b..25195595 100644
--- a/src/components/common/DarkSelector.vue
+++ b/src/components/common/DarkSelector.vue
@@ -10,6 +10,8 @@ import { defineComponent } from 'vue';
 import { toggleDark } from 'vue-dark-switch';
 import { FontAwesomeIcon } from '@fortawesome/vue-fontawesome';
 import { ElButton } from 'element-plus';
+import { getCookie, setCookie } from 'typescript-cookie';
+import { getDomain } from '@/utils/initializer';
 
 export default defineComponent({
   components: {
@@ -19,20 +21,13 @@ export default defineComponent({
   emits: ['update:dark'],
   computed: {
     dark() {
-      return this.$store.getters.dark;
+      return getCookie('THEME') === 'dark';
     }
-    // switchValue() {
-    //   return isDark.value;
-    // }
   },
   watch: {
     dark(val) {
       this.setDark(val);
     }
-    // switchValue(val) {
-    //   console.log('switchValue', val);
-    //   // this.$store.dispatch('setDark', val);
-    // }
   },
   mounted() {
     console.log('mounted', this.dark);
@@ -41,12 +36,18 @@ export default defineComponent({
   methods: {
     setDark(flag: boolean) {
       toggleDark(flag);
-      this.$store.dispatch('setDark', flag);
+      this.setCookie(flag);
       if (flag === true) {
         document.documentElement.classList.add('dark');
       } else if (flag === false) {
         document.documentElement.classList.remove('dark');
       }
+    },
+    setCookie(isDark: boolean) {
+      setCookie('THEME', isDark ? 'dark' : 'light', {
+        path: '/',
+        domain: getDomain()
+      });
     }
   }
 });
diff --git a/src/components/common/LocaleSelector.vue b/src/components/common/LocaleSelector.vue
index e6e44e95..501e58d6 100644
--- a/src/components/common/LocaleSelector.vue
+++ b/src/components/common/LocaleSelector.vue
@@ -22,8 +22,10 @@
 
 <script lang="ts">
 import { defineComponent } from 'vue';
-import { ElDropdown, ElDropdownItem, ElDropdownMenu, ElButton } from 'element-plus';
+import { ElDropdown, ElDropdownItem, ElDropdownMenu } from 'element-plus';
 import { SUPPORTED_LOCALES, setI18nLanguage } from '@/i18n';
+import { setCookie } from 'typescript-cookie';
+import { getDomain } from '@/utils/initializer';
 
 export default defineComponent({
   name: 'LocaleSelector',
@@ -39,9 +41,17 @@ export default defineComponent({
   },
   methods: {
     async onSelectLocale(locale: string) {
+      // change router
+      this.$router.push({ query: { ...this.$route.query, locale: undefined } });
       await setI18nLanguage(locale);
-      this.$store.dispatch('setLocale', locale);
+      this.setCookie(locale);
       window.location.reload();
+    },
+    setCookie(locale: string) {
+      setCookie('LOCALE', locale, {
+        path: '/',
+        domain: getDomain()
+      });
     }
   }
 });
diff --git a/src/i18n/index.ts b/src/i18n/index.ts
index 87320004..3d69fba7 100644
--- a/src/i18n/index.ts
+++ b/src/i18n/index.ts
@@ -47,8 +47,8 @@ export const loadLocalResource = async (name: string, locale: string) => {
   }
 };
 
-export const getLocale = (): string => {
-  const canonicalLang = Intl.getCanonicalLocales(navigator.language)?.[0];
+export const getLocale = (lang?: string): string => {
+  const canonicalLang = Intl.getCanonicalLocales(lang || navigator.language)?.[0];
   const supportedLocales = SUPPORTED_LOCALES.map((locale) => locale.value);
   // if the canonical language is supported, use it
   if (canonicalLang && supportedLocales.includes(canonicalLang)) {
diff --git a/src/main.ts b/src/main.ts
index fad63e9f..8a518e94 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -15,39 +15,43 @@ import copyToClipboard from 'copy-to-clipboard';
 import { initializeCookies, initializeFavicon, initializeTitle } from './utils/initializer';
 import config from './plugins/config';
 
-initializeCookies();
-initializeTitle();
-initializeFavicon();
+const main = async () => {
+  await initializeCookies();
+  await initializeTitle();
+  await initializeFavicon();
 
-const app = createApp(App);
+  const app = createApp(App);
 
-app.use(router);
-app.use(store);
-app.use(i18n);
-app.use(dayjs, {
-  formatString: 'YYYY-MM-DD HH:mm:ss'
-});
-app.use(config);
-app.directive('loading', vLoading);
-app.mount('#app');
-console.debug('app mounted');
+  app.use(router);
+  app.use(store);
+  app.use(i18n);
+  app.use(dayjs, {
+    formatString: 'YYYY-MM-DD HH:mm:ss'
+  });
+  app.use(config);
+  app.directive('loading', vLoading);
+  app.mount('#app');
+  console.debug('app mounted');
 
-app.directive('highlight', async (el) => {
-  const blocks = el.querySelectorAll('pre code');
-  blocks.forEach((block: HTMLPreElement) => {
-    // create the copy button
-    const copy = document.createElement('button');
-    copy.innerHTML = i18n.global.t('common.button.copy').toString();
-    // add the event listener to each click
-    copy.addEventListener('click', () => {
-      copyToClipboard(block.innerText);
+  app.directive('highlight', async (el) => {
+    const blocks = el.querySelectorAll('pre code');
+    blocks.forEach((block: HTMLPreElement) => {
+      // create the copy button
+      const copy = document.createElement('button');
+      copy.innerHTML = i18n.global.t('common.button.copy').toString();
+      // add the event listener to each click
+      copy.addEventListener('click', () => {
+        copyToClipboard(block.innerText);
+      });
+      // append the copy button to each code block
+      block.parentElement?.prepend(copy);
+      hl.highlightBlock(block);
     });
-    // append the copy button to each code block
-    block.parentElement?.prepend(copy);
-    hl.highlightBlock(block);
   });
-});
 
-// make app available globally
-// @ts-ignore
-window.app = app;
+  // make app available globally
+  // @ts-ignore
+  window.app = app;
+};
+
+main();
diff --git a/src/router/index.ts b/src/router/index.ts
index 421addec..f0ffd26a 100644
--- a/src/router/index.ts
+++ b/src/router/index.ts
@@ -8,8 +8,8 @@ import chatdoc from './chatdoc';
 import qrart from './qrart';
 
 import { ROUTE_CHAT_CONVERSATION_NEW, ROUTE_INDEX } from './constants';
-import store from '@/store';
 import { DEFAULT_LOCALE, setI18nLanguage } from '@/i18n';
+import { getCookie } from 'typescript-cookie';
 
 const routes = [
   {
@@ -34,7 +34,7 @@ const router = createRouter({
 });
 
 router.beforeEach(async (to, from, next) => {
-  const locale = store.state.locale || DEFAULT_LOCALE;
+  const locale = getCookie('LOCALE') || DEFAULT_LOCALE;
   await setI18nLanguage(locale);
   return next();
 });
diff --git a/src/store/common/actions.ts b/src/store/common/actions.ts
index 24ea37d7..9cddf4b1 100644
--- a/src/store/common/actions.ts
+++ b/src/store/common/actions.ts
@@ -25,14 +25,6 @@ export const setToken = ({ commit }: ActionContext<IRootState, IRootState>, payl
   commit('setToken', payload);
 };
 
-export const setDark = ({ commit }: ActionContext<IRootState, IRootState>, payload: boolean) => {
-  commit('setDark', payload);
-};
-
-export const setLocale = ({ commit }: ActionContext<IRootState, IRootState>, payload: string) => {
-  commit('setLocale', payload);
-};
-
 export const setUser = ({ commit }: ActionContext<IRootState, IRootState>, payload: IUser) => {
   commit('setUser', payload);
 };
@@ -61,13 +53,11 @@ export const getToken = async ({ commit }: ActionContext<IRootState, IRootState>
 };
 
 export default {
-  setLocale,
   resetToken,
   resetAll,
   resetUser,
   resetSetting,
   setToken,
-  setDark,
   setUser,
   getToken,
   getUser
diff --git a/src/store/common/getters.ts b/src/store/common/getters.ts
index a3159410..6c8c5fc5 100644
--- a/src/store/common/getters.ts
+++ b/src/store/common/getters.ts
@@ -8,14 +8,6 @@ export const user = (state: IRootState): any => {
   return state.user;
 };
 
-export const dark = (state: IRootState): boolean => {
-  return state.dark;
-};
-
-export const locale = (state: IRootState): string => {
-  return state.locale;
-};
-
 export const token = (state: IRootState): any => {
   return state.token;
 };
@@ -25,10 +17,8 @@ export const setting = (state: IRootState): any => {
 };
 
 export default {
-  locale,
   authenticated,
   user,
   token,
-  dark,
   setting
 };
diff --git a/src/store/common/models.ts b/src/store/common/models.ts
index 32cea1be..fc41d50a 100644
--- a/src/store/common/models.ts
+++ b/src/store/common/models.ts
@@ -12,8 +12,6 @@ export interface ICommonState {
   token: IToken;
   user?: IUser;
   setting?: ISetting;
-  locale: string;
-  dark: boolean;
 }
 
 export interface IRootState extends ICommonState {
diff --git a/src/store/common/mutations.ts b/src/store/common/mutations.ts
index ff496248..7fef07a6 100644
--- a/src/store/common/mutations.ts
+++ b/src/store/common/mutations.ts
@@ -8,14 +8,6 @@ export const setUser = (state: IRootState, payload: IUser): void => {
   };
 };
 
-export const setLocale = (state: IRootState, payload: string): void => {
-  state.locale = payload;
-};
-
-export const setDark = (state: IRootState, payload: boolean): void => {
-  state.dark = payload;
-};
-
 export const setToken = (state: IRootState, payload: any): void => {
   state.token = {
     ...state.token,
@@ -43,11 +35,9 @@ export const setSetting = (state: IRootState, payload: any): void => {
 };
 
 export default {
-  setLocale,
   setUser,
   resetUser,
   setToken,
-  setDark,
   resetToken,
   setSetting,
   resetSetting
diff --git a/src/store/common/state.ts b/src/store/common/state.ts
index a82dd89f..d06b2677 100644
--- a/src/store/common/state.ts
+++ b/src/store/common/state.ts
@@ -2,13 +2,11 @@ import { IRootState } from './models';
 import chatState from '../chat/state';
 import midjourneyState from '../midjourney/state';
 import chatdocState from '../chatdoc/state';
-import { getLocale } from '@/i18n';
+import qrartState from '../qrart/state';
 
 export default (): IRootState => {
   return {
     user: {},
-    locale: getLocale(),
-    dark: window.matchMedia('(prefers-color-scheme: dark)').matches,
     token: {
       access: undefined,
       refresh: undefined,
@@ -20,6 +18,7 @@ export default (): IRootState => {
     },
     chatdoc: chatdocState(),
     chat: chatState(),
-    midjourney: midjourneyState()
+    midjourney: midjourneyState(),
+    qrart: qrartState()
   };
 };
diff --git a/src/utils/initializer.ts b/src/utils/initializer.ts
index 00725570..9e75457b 100644
--- a/src/utils/initializer.ts
+++ b/src/utils/initializer.ts
@@ -1,13 +1,23 @@
-import { setCookie } from 'typescript-cookie';
+import { getCookie, setCookie } from 'typescript-cookie';
 import config from '@/config';
 import favicon from '@/assets/images/favicon.ico';
+import { getLocale } from '@/i18n';
 
-/**
- * Initialize cookies.
- */
-export const initializeCookies = () => {
+export const getDomain = () => {
+  const host = window.location.hostname;
+  // process test env and prod env, for example:
+  // hub.acedata.cloud -> .acedata.cloud
+  // hub.test.acedata.cloud -> .acedata.cloud
+  const domain = host.replace(/^\S+?\.(test\.|local\.)?/, '.');
+  console.log('cookies domain', domain);
+  return domain;
+};
+
+export const initializeCookies = async () => {
   // parse the query string and set to cookies
   const query = new URLSearchParams(window.location.search);
+
+  // set the inviter id to cookies
   const inviterId = query.get('inviter_id');
   if (inviterId) {
     // set the cookie to expire in 7 days
@@ -16,7 +26,41 @@ export const initializeCookies = () => {
     console.log('set INVITER_ID to cookies', inviterId);
     setCookie('INVITER_ID', inviterId, {
       expires: expiration,
-      path: '/'
+      path: '/',
+      domain: getDomain()
+    });
+  }
+
+  // set the theme to cookies
+  const theme = query.get('theme');
+  if (theme) {
+    console.log('set THEME to cookies', theme);
+    setCookie('THEME', theme, {
+      path: '/',
+      domain: getDomain()
+    });
+  } else if (!getCookie('THEME')) {
+    const isDark = window.matchMedia('(prefers-color-scheme: dark)').matches;
+    console.log('set THEME to cookies', isDark ? 'dark' : 'light');
+    setCookie('THEME', isDark ? 'dark' : 'light', {
+      path: '/',
+      domain: getDomain()
+    });
+  }
+
+  // set the locale to cookies
+  const lang = query.get('lang');
+  let locale = undefined;
+  if (lang) {
+    locale = getLocale(lang);
+  } else if (!getCookie('LOCALE')) {
+    locale = getLocale();
+  }
+  if (locale) {
+    console.log('set LOCALE to cookies', locale);
+    setCookie('LOCALE', locale, {
+      path: '/',
+      domain: getDomain()
     });
   }
 };
@@ -24,7 +68,7 @@ export const initializeCookies = () => {
 /**
  * Initialize title.
  */
-export const initializeTitle = () => {
+export const initializeTitle = async () => {
   // set the title from config.global.title
   const title = config.global.title;
   // find the title element or insert a new one
@@ -39,7 +83,7 @@ export const initializeTitle = () => {
 /**
  * Initialize favicon.
  */
-export const initializeFavicon = () => {
+export const initializeFavicon = async () => {
   // by default use favicon which imported
   // if faviconUrl is set in config, use it instead
   const favIconUrl = config.global.faviconUrl;