feat: encryption predicate option

generalpiston · Apr 17, 2024 · 4d27298 · 4d27298
1 parent 8f963a9
commit 4d27298
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/entity.ts b/src/entity.ts
@@ -14,6 +14,9 @@ export function encrypt<T extends ObjectLiteral>(entity: any): any {
     let { propertyName, mode, target } = columnMetadata;
     let options: ExtendedColumnOptions = columnMetadata.options;
     let encrypt = options.encrypt;
+    if (encrypt?.encryptionPredicate && !encrypt?.encryptionPredicate(entity)) {
+      continue;
+    }
     if (
       encrypt &&
       mode === 'regular' &&
@@ -42,6 +45,9 @@ export function decrypt<T extends ObjectLiteral>(entity: any): any {
     let { propertyName, mode, target } = columnMetadata;
     let options: ExtendedColumnOptions = columnMetadata.options;
     let encrypt = options.encrypt;
+    if (encrypt?.encryptionPredicate && !encrypt?.encryptionPredicate(entity)) {
+      continue;
+    }
     if (
       encrypt &&
       mode === 'regular' &&

diff --git a/src/options/EncryptionOptions.ts b/src/options/EncryptionOptions.ts
@@ -5,4 +5,5 @@ export interface EncryptionOptions {
   iv?: string;    //// For testing mainly.
   authTagLength?: number;
   looseMatching?: boolean;
+  encryptionPredicate?: (entity: any) => boolean;
 }