import { decrypt, encrypt } from 'crypto-js/aes';
|
import Base64 from 'crypto-js/enc-base64';
|
import Hex from 'crypto-js/enc-hex';
|
import UTF8, { parse } from 'crypto-js/enc-utf8';
|
import md5 from 'crypto-js/md5';
|
import ECB from 'crypto-js/mode-ecb';
|
import pkcs7 from 'crypto-js/pad-pkcs7';
|
|
export interface EncryptionParams {
|
iv: string;
|
key: string;
|
useHex: boolean;
|
}
|
|
const defaultKey = encryptByMd5((import.meta as any).env.VITE_CIPHER_KEY);
|
|
export class AesEncryption {
|
private iv;
|
private key;
|
private useHex = true;
|
|
constructor(opt: Partial<EncryptionParams> = {}) {
|
const { iv, key = defaultKey, useHex = true } = opt;
|
if (key) this.key = parse(key);
|
if (iv) this.iv = parse(iv);
|
this.useHex = !!useHex;
|
}
|
|
decryptByAES(cipherText: string) {
|
const realCipherText = this.useHex
|
? Base64.stringify(Hex.parse(cipherText))
|
: cipherText;
|
return decrypt(
|
realCipherText,
|
this.key as CryptoJS.lib.WordArray,
|
this.getOptions(),
|
).toString(UTF8);
|
}
|
|
encryptByAES(cipherText: string) {
|
const result = encrypt(
|
cipherText,
|
this.key as CryptoJS.lib.WordArray,
|
this.getOptions(),
|
).toString();
|
return this.useHex ? Hex.stringify(Base64.parse(result)) : result;
|
}
|
|
private getOptions() {
|
return {
|
iv: this.iv,
|
mode: ECB,
|
padding: pkcs7,
|
};
|
}
|
}
|
|
export function encryptByBase64(cipherText: string) {
|
return UTF8.parse(cipherText).toString(Base64);
|
}
|
|
export function decodeByBase64(cipherText: string) {
|
return Base64.parse(cipherText).toString(UTF8);
|
}
|
|
export function encryptByMd5(password: string) {
|
return md5(password).toString();
|
}
|