canrd-outside / order-erp-front · Files

import type { LockInfo, UserInfo } from '/#/store';
import type { ProjectConfig } from '/#/config';

import type { RouteLocationNormalized } from 'vue-router';

import { createLocalStorage, createSessionStorage } from '/@/utils/cache';

import { Memory } from './memory';
import {
  TOKEN_KEY,
  USER_INFO_KEY,
  ROLES_KEY,
  LOCK_INFO_KEY,
  PROJ_CFG_KEY,
  APP_LOCAL_CACHE_KEY,
  APP_SESSION_CACHE_KEY,

  MULTIPLE_TABS_KEY,

} from '/@/enums/cacheEnum';
import { DEFAULT_CACHE_TIME } from '/@/settings/encryptionSetting';
import { toRaw } from 'vue';

import { pick, omit } from 'lodash-es';

interface BasicStore {
  [TOKEN_KEY]: string | number | null | undefined;

  [USER_INFO_KEY]: UserInfo;
  [ROLES_KEY]: string[];
  [LOCK_INFO_KEY]: LockInfo;
  [PROJ_CFG_KEY]: ProjectConfig;

  [MULTIPLE_TABS_KEY]: RouteLocationNormalized[];

}

type LocalStore = BasicStore;

type SessionStore = BasicStore;

export type BasicKeys = keyof BasicStore;
type LocalKeys = keyof LocalStore;
type SessionKeys = keyof SessionStore;

const ls = createLocalStorage();
const ss = createSessionStorage();

const localMemory = new Memory(DEFAULT_CACHE_TIME);
const sessionMemory = new Memory(DEFAULT_CACHE_TIME);

function initPersistentMemory() {

  const localCache = ls.get(APP_LOCAL_CACHE_KEY);

  const sessionCache = ss.get(APP_SESSION_CACHE_KEY);

  localCache && localMemory.resetCache(localCache);
  sessionCache && sessionMemory.resetCache(sessionCache);

}

export class Persistent {
  static getLocal<T>(key: LocalKeys) {
    return localMemory.get(key)?.value as Nullable<T>;

  }

  static setLocal(key: LocalKeys, value: LocalStore[LocalKeys], immediate = false): void {
    localMemory.set(key, toRaw(value));
    immediate && ls.set(APP_LOCAL_CACHE_KEY, localMemory.getCache);
  }

  static removeLocal(key: LocalKeys, immediate = false): void {

    localMemory.remove(key);

    immediate && ls.set(APP_LOCAL_CACHE_KEY, localMemory.getCache);

  }

  static clearLocal(immediate = false): void {

    localMemory.clear();

    immediate && ls.clear();

  }

  static getSession<T>(key: SessionKeys) {
    return sessionMemory.get(key)?.value as Nullable<T>;

  }

  static setSession(key: SessionKeys, value: SessionStore[SessionKeys], immediate = false): void {
    sessionMemory.set(key, toRaw(value));

    immediate && ss.set(APP_SESSION_CACHE_KEY, sessionMemory.getCache);

  }

  static removeSession(key: SessionKeys, immediate = false): void {

    sessionMemory.remove(key);

    immediate && ss.set(APP_SESSION_CACHE_KEY, sessionMemory.getCache);

  }

  static clearSession(immediate = false): void {

    sessionMemory.clear();

    immediate && ss.clear();

  }

  static clearAll(immediate = false) {

    sessionMemory.clear();
    localMemory.clear();

    if (immediate) {
      ls.clear();
      ss.clear();
    }

  }

}

window.addEventListener('beforeunload', function () {

  // TOKEN_KEY 在登录或注销时已经写入到storage了，此处为了解决同时打开多个窗口时token不同步的问题

  // LOCK_INFO_KEY 在锁屏和解锁时写入，此处也不应修改

  ls.set(APP_LOCAL_CACHE_KEY, {

    ...omit(localMemory.getCache, LOCK_INFO_KEY),
    ...pick(ls.get(APP_LOCAL_CACHE_KEY), [TOKEN_KEY, USER_INFO_KEY, LOCK_INFO_KEY]),

  });
  ss.set(APP_SESSION_CACHE_KEY, {

    ...omit(sessionMemory.getCache, LOCK_INFO_KEY),
    ...pick(ss.get(APP_SESSION_CACHE_KEY), [TOKEN_KEY, USER_INFO_KEY, LOCK_INFO_KEY]),

  });

});

function storageChange(e: any) {
  const { key, newValue, oldValue } = e;

  if (!key) {
    Persistent.clearAll();
    return;
  }

  if (!!newValue && !!oldValue) {
    if (APP_LOCAL_CACHE_KEY === key) {
      Persistent.clearLocal();

    }

    if (APP_SESSION_CACHE_KEY === key) {
      Persistent.clearSession();

    }
  }

}

window.addEventListener('storage', storageChange);

initPersistentMemory();