import type { BasicKeys, BasicStore } from '@/utils/cache/persistent'
import { Persistent } from '@/utils/cache/persistent'
import { CacheTypeEnum, EDGE_JWT_TOKEN_KEY, JWT_TOKEN_KEY, REFRESH_TOKEN_KEY, SHARE_JWT_TOKEN_KEY, SHARE_REFRESH_TOKEN_KEY } from '@/enums/cacheEnum'
import { projectSetting } from '@/settings/projectSetting'
import type { DrawioConfig } from '@/core/Library/types'
import { isFromEdge } from '@/core/LoadData/module/useParseParams'

const { permissionCacheType } = projectSetting
const isLocal = permissionCacheType === CacheTypeEnum.LOCAL

export function getAuthCache<T>(key: BasicKeys) {
  const fn = isLocal ? Persistent.getLocal : Persistent.getSession
  return fn(key) as T
}

export function setAuthCache<T extends BasicKeys>(key: T, value: BasicStore[T]) {
  const fn = isLocal ? Persistent.setLocal : Persistent.setSession
  return fn(key, value, true)
}

export function clearAuthCache(immediate = true) {
  const fn = isLocal ? Persistent.clearLocal : Persistent.clearSession
  return fn(immediate)
}

export function getJwtToken(): string {
  return getAuthCache(isFromEdge() ? EDGE_JWT_TOKEN_KEY : JWT_TOKEN_KEY)
}
export function getRefreshToken(): string {
  return getAuthCache(REFRESH_TOKEN_KEY)
}

export function getShareJwtToken(): string {
  return getAuthCache(SHARE_JWT_TOKEN_KEY)
}

export function getShareRefreshToken(): string {
  return getAuthCache(SHARE_REFRESH_TOKEN_KEY)
}

export function getLocale() {
  const { language } = (JSON.parse(localStorage.getItem('.drawio-config') || '{}') || {}) as DrawioConfig

  return language
}