import { defineStore } from 'pinia' import { type Ref } from 'vue' import type { User } from '@/api/type/user' import UserApi from '@/api/user/user' import LoginApi from '@/api/user/login' import { cloneDeep } from 'lodash' import ThemeApi from '@/api/systemSettings/theme' // import { defaultPlatformSetting } from '@/utils/theme' import { useLocalStorage } from '@vueuse/core' import { localeConfigKey, getBrowserLang } from '@/locales/index' import useThemeStore from './theme' import { useElementPlusTheme } from 'use-element-plus-theme' import { defaultPlatformSetting } from '@/utils/theme.ts' export interface userStateTypes { userType: number // 1 系统操作者 2 对话用户 userInfo: User | null version?: string license_is_valid: boolean edition: 'CE' | 'PE' | 'EE' themeInfo: any token: any } const useLoginStore = defineStore('user', { state: (): userStateTypes => ({ userType: 1, // 1 系统操作者 2 对话用户 userInfo: null, version: '', license_is_valid: false, edition: 'CE', themeInfo: null, token: '', }), actions: { getLanguage() { return this.userType === 1 ? localStorage.getItem('MaxKB-locale') || getBrowserLang() : sessionStorage.getItem('language') || getBrowserLang() }, isDefaultTheme() { return !this.themeInfo?.theme || this.themeInfo?.theme === '#3370FF' }, setTheme(data: any) { const { changeTheme } = useElementPlusTheme(this.themeInfo?.theme) changeTheme(data?.['theme']) this.themeInfo = cloneDeep(data) }, async profile(loading?: Ref) { return UserApi.getUserProfile(loading).then((ok) => { this.userInfo = ok.data useLocalStorage(localeConfigKey, 'en-US').value = ok?.data?.language || this.getLanguage() const theme = useThemeStore() theme.setTheme() return this.asyncGetProfile() }) }, async asyncGetProfile() { return new Promise((resolve, reject) => { UserApi.getProfile() .then(async (ok) => { // this.version = ok.data?.version || '-' console.log(ok) this.license_is_valid = ok.data.license_is_valid this.edition = ok.data.edition if (this.isEE() || this.isPE()) { // await this.theme() } else { this.themeInfo = { ...defaultPlatformSetting, } } resolve(ok) }) .catch((error) => { reject(error) }) }) }, getPermissions() { if (this.userInfo) { return this.isXPack && this.XPACK_LICENSE_IS_VALID ? [...this.userInfo?.permissions, 'x-pack'] : this.userInfo?.permissions } else { return this.userInfo?.permissions } }, getRole() { if (this.userInfo) { return this.userInfo?.role } else { return '' } }, async theme(loading?: Ref) { return await ThemeApi.getThemeInfo(loading).then((ok) => { this.setTheme(ok.data) // window.document.title = this.themeInfo['title'] || 'MaxKB' // const link = document.querySelector('link[rel="icon"]') as any // if (link) { // link['href'] = this.themeInfo['icon'] || '/favicon.ico' // } }) }, showXpack() { return this.edition != 'CE' }, isEnterprise() { return this.edition != 'CE' && !this.license_is_valid }, isExpire() { return this.edition != 'CE' && !this.license_is_valid }, isCE() { return this.edition == 'CE' && this.license_is_valid }, isPE() { return this.edition == 'PE' && this.license_is_valid }, isEE() { return this.edition == 'EE' && this.license_is_valid }, // changeUserType(num: number, token?: string) { // this.userType = num // this.userAccessToken = token // }, async dingCallback(code: string) { return LoginApi.getDingCallback(code).then((ok) => { this.token = ok.data localStorage.setItem('token', ok.data) return this.profile() }) }, async dingOauth2Callback(code: string) { return LoginApi.getDingOauth2Callback(code).then((ok) => { this.token = ok.data localStorage.setItem('token', ok.data) return this.profile() }) }, async wecomCallback(code: string) { return LoginApi.getWecomCallback(code).then((ok) => { this.token = ok.data localStorage.setItem('token', ok.data) return this.profile() }) }, async larkCallback(code: string) { return LoginApi.getLarkCallback(code).then((ok) => { this.token = ok.data localStorage.setItem('token', ok.data) return this.profile() }) }, async logout() { return LoginApi.logout().then(() => { localStorage.removeItem('token') return true }) }, async getAuthType() { return LoginApi.getAuthType().then((ok) => { return ok.data }) }, async getQrType() { return LoginApi.getQrType().then((ok) => { return ok.data }) }, async getQrSource() { return LoginApi.getQrSource().then((ok) => { return ok.data }) }, async postUserLanguage(lang: string, loading?: Ref) { return new Promise((resolve, reject) => { LoginApi.postLanguage({ language: lang }, loading) .then(async (ok) => { useLocalStorage(localeConfigKey, 'en-US').value = lang window.location.reload() resolve(ok) }) .catch((error) => { reject(error) }) }) }, }, }) export default useLoginStore