naive-ui/demo/store/index.js
2021-04-05 17:59:04 +08:00

175 lines
4.4 KiB
JavaScript

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, useIsMobile } 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))
const isMobileRef = useIsMobile()
return {
themeEditorStyle: computed(() => {
return isMobileRef.value ? 'right: 18px; bottom: 24px;' : undefined
}),
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')
}