Blame view

src/utils/cache/storageCache.ts 2.87 KB
1
import { cacheCipher } from '/@/settings/encryptionSetting';
Vben authored
2
3
import type { EncryptionParams } from '/@/utils/cipher';
import { AesEncryption } from '/@/utils/cipher';
4
5
import { isNullOrUnDef } from '/@/utils/is';
6
export interface CreateStorageParams extends EncryptionParams {
7
  prefixKey: string;
8
9
  storage: Storage;
  hasEncrypt: boolean;
10
  timeout?: Nullable<number>;
11
12
13
14
15
16
}
export const createStorage = ({
  prefixKey = '',
  storage = sessionStorage,
  key = cacheCipher.key,
  iv = cacheCipher.iv,
17
  timeout = null,
18
  hasEncrypt = true,
19
}: Partial<CreateStorageParams> = {}) => {
20
21
22
23
  if (hasEncrypt && [key.length, iv.length].some((item) => item !== 16)) {
    throw new Error('When hasEncrypt is true, the key or iv must be 16 bits!');
  }
Vben authored
24
  const encryption = new AesEncryption({ key, iv });
25
26

  /**
27
28
   * Cache class
   * Construction parameters can be passed into sessionStorage, localStorage,
29
30
31
32
33
34
   * @class Cache
   * @example
   */
  const WebStorage = class WebStorage {
    private storage: Storage;
    private prefixKey?: string;
Vben authored
35
    private encryption: AesEncryption;
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
    private hasEncrypt: boolean;
    /**
     *
     * @param {*} storage
     */
    constructor() {
      this.storage = storage;
      this.prefixKey = prefixKey;
      this.encryption = encryption;
      this.hasEncrypt = hasEncrypt;
    }

    private getKey(key: string) {
      return `${this.prefixKey}${key}`.toUpperCase();
    }

    /**
53
     * Set cache
54
55
     * @param {string} key
     * @param {*} value
56
     * @param {*} expire Expiration time in seconds
57
58
     * @memberof Cache
     */
59
    set(key: string, value: any, expire: number | null = timeout) {
60
61
      const stringData = JSON.stringify({
        value,
62
63
        time: Date.now(),
        expire: !isNullOrUnDef(expire) ? new Date().getTime() + expire * 1000 : null,
64
65
66
67
68
69
70
71
      });
      const stringifyValue = this.hasEncrypt
        ? this.encryption.encryptByAES(stringData)
        : stringData;
      this.storage.setItem(this.getKey(key), stringifyValue);
    }

    /**
72
     * Read cache
73
     * @param {string} key
74
     * @param {*} def
75
76
77
     * @memberof Cache
     */
    get(key: string, def: any = null): any {
78
79
80
81
82
83
84
85
86
      const val = this.storage.getItem(this.getKey(key));
      if (!val) return def;

      try {
        const decVal = this.hasEncrypt ? this.encryption.decryptByAES(val) : val;
        const data = JSON.parse(decVal);
        const { value, expire } = data;
        if (isNullOrUnDef(expire) || expire >= new Date().getTime()) {
          return value;
87
        }
88
89
90
        this.remove(key);
      } catch (e) {
        return def;
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
      }
    }

    /**
     * Delete cache based on key
     * @param {string} key
     * @memberof Cache
     */
    remove(key: string) {
      this.storage.removeItem(this.getKey(key));
    }

    /**
     * Delete all caches of this instance
     */
    clear(): void {
      this.storage.clear();
    }
  };
  return new WebStorage();
};