naive-ui/demo/store/index.js

import { computed, ref, provide, reactive, toRef, inject } from 'vue'
import { useRoute, useRouter } from 'vue-router'
import { useMemo } from 'vooks'
import {
  NConfigProvider,
  darkTheme,
  enUS,
  zhCN,
  dateEnUS,
  dateZhCN,
  useOsTheme
} from 'naive-ui'
import { TsConfigProvider } from '../../themes/tusimple/src'
import { i18n } from '../utils/composables'
import {
  createDocumentationMenuOptions,
  createComponentMenuOptions
} from './menu-options'
import hljs from './hljs'

const storeKey = 'site-store'

export function siteSetup () {
  const route = useRoute()
  const router = useRouter()
  // display mode
  const _displayModeRef = ref(window.localStorage.getItem('mode') ?? 'debug')
  const displayModeRef = computed({
    get () {
      return _displayModeRef.value
    },
    set (value) {
      _displayModeRef.value = value
      window.localStorage.setItem('mode', value)
    }
  })
  // locale
  const localeNameRef = useMemo({
    get () {
      return route.path.startsWith('/zh-CN') ? 'zh-CN' : 'en-US'
    },
    set (locale) {
      router.push(changeLangInPath(route.fullPath, locale))
    }
  })
  const localeRef = computed(() => {
    return localeNameRef.value === 'zh-CN' ? zhCN : enUS
  })
  // useMemo
  const dateLocaleRef = useMemo(() => {
    return route.params.lang === 'zh-CN' ? dateZhCN : dateEnUS
  })
  // theme
  const osThemeRef = useOsTheme()
  const themeNameRef = useMemo({
    get () {
      switch (route.params.theme) {
        case 'os-theme':
          return osThemeRef.value
        case 'dark':
          return 'dark'
        default:
          return 'light'
      }
    },
    set (theme) {
      router.push(changeThemeInPath(route.fullPath, theme))
    }
  })
  const themeRef = computed(() => {
    const { value } = themeNameRef
    return value === 'dark' ? darkTheme : null
  })
  // config provider
  const configProviderNameRef = ref(
    process.env.TUSIMPLE ? 'tusimple' : 'default'
  )
  const configProviderRef = computed(() => {
    return configProviderNameRef.value === 'tusimple'
      ? TsConfigProvider
      : NConfigProvider
  })
  // options
  const docOptionsRef = computed(() =>
    createDocumentationMenuOptions({
      theme: themeNameRef.value,
      lang: localeNameRef.value,
      mode: displayModeRef.value
    })
  )
  const componentOptionsRef = computed(() =>
    createComponentMenuOptions({
      theme: themeNameRef.value,
      lang: localeNameRef.value,
      mode: displayModeRef.value
    })
  )
  const flattenedDocOptionsRef = computed(() => {
    const flattenedItems = []
    const traverse = (items) => {
      if (!items) return
      items.forEach((item) => {
        if (item.children) traverse(item.children)
        else flattenedItems.push(item)
      })
    }
    traverse(docOptionsRef.value)
    traverse(componentOptionsRef.value)
    return flattenedItems
  })
  provide(
    storeKey,
    reactive({
      themeName: themeNameRef,
      configProviderName: configProviderNameRef,
      localeName: localeNameRef,
      displayMode: displayModeRef,
      docOptions: docOptionsRef,
      componentOptions: componentOptionsRef,
      flattenedDocOptions: flattenedDocOptionsRef
    })
  )
  i18n.provide(computed(() => localeNameRef.value))
  return {
    configProvider: configProviderRef,
    hljs,
    themeName: themeNameRef,
    theme: themeRef,
    locale: localeRef,
    dateLocale: dateLocaleRef
  }
}

function changeLangInPath (path, lang) {
  const langReg = /^\/(zh-CN|en-US)\//
  return path.replace(langReg, `/${lang}/`)
}

function changeThemeInPath (path, theme) {
  const themeReg = /(^\/[^/]+\/)([^/]+)/
  return path.replace(themeReg, '$1' + theme)
}

export function useDisplayMode () {
  return toRef(inject(storeKey), 'displayMode')
}

export function useLocaleName () {
  return toRef(inject(storeKey), 'localeName')
}

export function useThemeName () {
  return toRef(inject(storeKey), 'themeName')
}

export function useDocOptions () {
  return toRef(inject(storeKey), 'docOptions')
}

export function useComponentOptions () {
  return toRef(inject(storeKey), 'componentOptions')
}

export function useFlattenedDocOptions () {
  return toRef(inject(storeKey), 'flattenedDocOptions')
}

export function useConfigProviderName () {
  return toRef(inject(storeKey), 'configProviderName')
}