加密解密修改

aotuzuche · Jun 10, 2021 · ddf9867 · ddf9867
1 parent 3fbd7b6
commit ddf9867
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 28 deletions.
diff --git a/README.md b/README.md
@@ -554,9 +554,9 @@ Desensitize.email('[email protected]'); // l*******@*******
 ```js
 import { Decrypt, Encrypt } from 'auto-libs';
 
-// 加密 data
-Encrypt();
+// AES加密
+Encrypt.AES();
 
-// 解密 data
-Decrypt();
+// AES解密
+Decrypt.AES();
 ```
diff --git a/lib/crypto.ts b/lib/crypto.ts
@@ -1,30 +1,34 @@
 import CryptoJS from 'crypto-js';
 
-// 解密方法(ECB)
-const Decrypt = (data: any, key: string) => {
-  if (isEmpty(data)) {
-    return '';
-  }
-  const decrypt = CryptoJS.AES.decrypt(data, key, {
-    mode: CryptoJS.mode.ECB,
-    padding: CryptoJS.pad.Pkcs7,
-  });
-  const decryptedStr = decrypt.toString(CryptoJS.enc.Utf8);
-  return decryptedStr;
+// 加密方法
+const Encrypt = {
+  AES(data: any, key: string): string {
+    if (isEmpty(data)) {
+      return '';
+    }
+    const cip = CryptoJS.enc.Utf8.parse(typeof data === 'object' ? JSON.stringify(data) : data);
+    const encrypted = CryptoJS.AES.encrypt(cip, key, {
+      mode: CryptoJS.mode.ECB,
+      padding: CryptoJS.pad.Pkcs7,
+    });
+    const encryptedStr = encrypted.toString();
+    return encryptedStr;
+  },
 };
 
-// 加密方法(ECB)
-const Encrypt = (data: any, key: string) => {
-  if (isEmpty(data)) {
-    return '';
-  }
-  const cip = CryptoJS.enc.Utf8.parse(typeof data === 'object' ? JSON.stringify(data) : data);
-  const encrypted = CryptoJS.AES.encrypt(cip, key, {
-    mode: CryptoJS.mode.ECB,
-    padding: CryptoJS.pad.Pkcs7,
-  });
-  const encryptedStr = encrypted.toString();
-  return encryptedStr;
+// 解密方法
+const Decrypt = {
+  AES(data: any, key: string): string {
+    if (isEmpty(data)) {
+      return '';
+    }
+    const decrypt = CryptoJS.AES.decrypt(data, key, {
+      mode: CryptoJS.mode.ECB,
+      padding: CryptoJS.pad.Pkcs7,
+    });
+    const decryptedStr = decrypt.toString(CryptoJS.enc.Utf8);
+    return decryptedStr;
+  },
 };
 
 // 判空
@@ -33,4 +37,4 @@ const isEmpty = (data: any) => {
   return isEmpty;
 };
 
-export { Decrypt, Encrypt };
+export { Encrypt, Decrypt };