@dynamatix/gb-schemas 0.21.2 → 0.21.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dynamatix/gb-schemas",
-  "version": "0.21.2",
+  "version": "0.21.4",
   "description": "All the schemas for gatehouse bank back-end.",
   "main": "index.js",
   "scripts": {

package/users/permission.model.js

@@ -1,6 +1,7 @@
 import mongoose from "mongoose";
+import mongooseEncryption from "../utils/encryption.middleware";
 
-const PermissionSchema = new mongoose.Schema({
+const permissionSchema = new mongoose.Schema({
     name: { type: String, required: true },
     controllerName: { type: String, required: true },
     methodName: { type: String, required: true }
@@ -8,6 +9,7 @@ const PermissionSchema = new mongoose.Schema({
     timestamps: true
 });
 
-const PermissionModel = mongoose.model('Permission', PermissionSchema);
+permissionSchema.plugin(mongooseEncryption);
+const PermissionModel = mongoose.model('Permission', permissionSchema);
 
 export default PermissionModel;

package/users/role-group.model.js

@@ -1,4 +1,5 @@
 import mongoose from "mongoose";
+import mongooseEncryption from "../utils/encryption.middleware";
 
 const RoleGroupSchema = new mongoose.Schema({
     groupId: { type: Number, required: true },
@@ -12,5 +13,6 @@ const RoleGroupSchema = new mongoose.Schema({
     timestamps: true
 });
 
+RoleGroupSchema.plugin(mongooseEncryption);
 const RoleGroupModel = mongoose.model('RoleGroup', RoleGroupSchema);
 export default RoleGroupModel;

package/users/role.model.js

@@ -1,4 +1,5 @@
 import mongoose from "mongoose";
+import mongooseEncryption from "../utils/encryption.middleware";
 
 const RoleSchema = new mongoose.Schema({
     roleId: { type: Number, required: true },
@@ -11,7 +12,7 @@ const RoleSchema = new mongoose.Schema({
     timestamps: true
 });
 
-
+RoleSchema.plugin(mongooseEncryption);
 const RoleModel = mongoose.model('Role', RoleSchema);
 
 export default RoleModel;

package/utils/encryption.middleware.js

@@ -23,6 +23,30 @@ export default function mongooseEncryption(schema, options) {
         next();
     });
 
+    // Encrypt before bulkWrite operations
+    schema.pre("bulkWrite", function (operations, next) {
+        operations.forEach(operation => {
+            if (operation.insertOne && operation.insertOne.document) {
+                operation.insertOne.document = encryptObject(operation.insertOne.document, collectionName);
+            } else if (operation.updateOne && operation.updateOne.update) {
+                operation.updateOne.update = encryptObject(operation.updateOne.update, collectionName);
+            } else if (operation.updateMany && operation.updateMany.update) {
+                operation.updateMany.update = encryptObject(operation.updateMany.update, collectionName);
+            }
+        });
+        next();
+    });
+
+    // Decrypt after bulkWrite operations
+    schema.post("bulkWrite", function (result) {
+        result.result.insertedDocs.forEach(doc => {
+            doc.set(decryptObject(doc.toObject(), collectionName)); // Decrypt document fields
+        });
+        result.result.modifiedDocs.forEach(doc => {
+            doc.set(decryptObject(doc.toObject(), collectionName)); // Decrypt document fields
+        });
+    });
+
     // Encrypt before creating a document
     schema.pre("create", function (next) {
         // Encrypt only the fields before saving to the database