113 lines
2.8 KiB
TypeScript
113 lines
2.8 KiB
TypeScript
import { cacheCipher } from '/@/settings/encryptionSetting';
|
|
|
|
import type { EncryptionParams } from '/@/utils/cipher';
|
|
|
|
import { AesEncryption } from '/@/utils/cipher';
|
|
|
|
import { isNullOrUnDef } from '/@/utils/is';
|
|
|
|
export interface CreateStorageParams extends EncryptionParams {
|
|
prefixKey: string;
|
|
storage: Storage;
|
|
hasEncrypt: boolean;
|
|
timeout?: Nullable<number>;
|
|
}
|
|
export const createStorage = ({
|
|
prefixKey = '',
|
|
storage = sessionStorage,
|
|
key = cacheCipher.key,
|
|
iv = cacheCipher.iv,
|
|
timeout = null,
|
|
hasEncrypt = true,
|
|
}: Partial<CreateStorageParams> = {}) => {
|
|
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!');
|
|
}
|
|
|
|
const encryption = new AesEncryption({ key, iv });
|
|
|
|
/**
|
|
*Cache class
|
|
*Construction parameters can be passed into sessionStorage, localStorage,
|
|
* @class Cache
|
|
* @example
|
|
*/
|
|
const WebStorage = class WebStorage {
|
|
private storage: Storage;
|
|
private prefixKey?: string;
|
|
private encryption: AesEncryption;
|
|
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();
|
|
}
|
|
|
|
/**
|
|
*
|
|
* Set cache
|
|
* @param {string} key
|
|
* @param {*} value
|
|
* @expire Expiration time in seconds
|
|
* @memberof Cache
|
|
*/
|
|
set(key: string, value: any, expire: number | null = timeout) {
|
|
const stringData = JSON.stringify({
|
|
value,
|
|
time: Date.now(),
|
|
expire: !isNullOrUnDef(expire) ? new Date().getTime() + expire * 1000 : null,
|
|
});
|
|
const stringifyValue = this.hasEncrypt ? this.encryption.encryptByAES(stringData) : stringData;
|
|
this.storage.setItem(this.getKey(key), stringifyValue);
|
|
}
|
|
|
|
/**
|
|
*Read cache
|
|
* @param {string} key
|
|
* @memberof Cache
|
|
*/
|
|
get(key: string, def: any = null): any {
|
|
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;
|
|
}
|
|
this.remove(key);
|
|
} catch (e) {
|
|
return def;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 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();
|
|
};
|