refactor: reuse createLangDictionary

src/node/config.ts

@@ -4,7 +4,12 @@ import chalk from 'chalk'
 import globby from 'globby'
 import { AliasOptions, UserConfig as ViteConfig } from 'vite'
 import { Options as VuePluginOptions } from '@vitejs/plugin-vue'
-import { SiteData, HeadConfig, LocaleConfig } from './shared'
+import {
+  SiteData,
+  HeadConfig,
+  LocaleConfig,
+  createLangDictionary
+} from './shared'
 import { resolveAliases, APP_PATH, DEFAULT_THEME_PATH } from './alias'
 import { MarkdownOptions } from './markdown/markdown'
 
@@ -142,6 +147,9 @@ export async function resolveSiteData(
     head: userConfig.head || [],
     themeConfig: userConfig.themeConfig || {},
     locales: userConfig.locales || {},
+    langs: createLangDictionary(
+      userConfig.themeConfig && userConfig.themeConfig.locales
+    ),
     customData: userConfig.customData || {}
   }
 }

src/shared/shared.ts

@@ -36,6 +36,15 @@ function resolveLocales<T>(
   return localeRoot ? locales[localeRoot] : undefined
 }
 
+export function createLangDictionary(locales: any | undefined) {
+  return locales
+    ? Object.keys(locales).reduce((langs, path) => {
+        langs[path] = locales![path].label
+        return langs
+      }, {} as Record<string, string>)
+    : {}
+}
+
 // this merges the locales data to the main data by the route
 export function resolveSiteDataByRoute(
   siteData: SiteData,
@@ -61,12 +70,7 @@ export function resolveSiteDataByRoute(
     lang: (localeData || siteData).lang,
     // clean the locales to reduce the bundle size
     locales: {},
-    langs: siteData.themeConfig.locales
+    langs: createLangDictionary(siteData.themeConfig.locales)
-      ? Object.keys(siteData.themeConfig.locales).reduce((locales, path) => {
-          locales[path] = siteData.themeConfig.locales![path].label
-          return locales
-        }, {} as Record<string, string>)
-      : {}
   }
 }