@dynamatix/gb-schemas 0.18.2 → 0.18.3

package/applications/solicitor.model.js

@@ -1,4 +1,5 @@
 import mongoose from "mongoose";
+import mongooseEncryption from "../utils/encryption.middleware";
 
 const Schema = mongoose.Schema;
 
@@ -31,5 +32,6 @@ const solicitorSchema = new Schema({
     telephone: { type: String,},
 });
 
+solicitorSchema.plugin(mongooseEncryption);
 const SolicitorModel = mongoose.model("Solicitor", solicitorSchema);
 export default SolicitorModel;

package/package.json

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

package/utils/encryption.middleware.js

@@ -0,0 +1,109 @@
+import dotenv from 'dotenv';
+import { decryptObject, encryptObject } from '@dynamatix/gb-schemas/utils';
+
+dotenv.config();
+
+// Read excluded collections from .env
+const EXCLUDED_COLLECTIONS = process.env.EXCLUDED_COLLECTIONS?.split(",") || [];
+
+export default function mongooseEncryption(schema, options) {
+    console.log("========MONGOOSE ENCRYPTION IS IN PROGRESS");
+
+    const collectionName = options?.collection || schema.options.collection;
+
+    // Skip encryption for excluded collections
+    if (EXCLUDED_COLLECTIONS.includes(collectionName)) {
+        return;
+    }
+
+    // Encrypt before saving (e.g., insert, update)
+    schema.pre("save", function (next) {
+        // Encrypt the document before saving to the database
+        this.set(encryptObject(this.toObject(), collectionName)); // Encrypt the document
+        next();
+    });
+
+    // Encrypt before creating a document
+    schema.pre("create", function (next) {
+        // Encrypt only the fields before saving to the database
+        this.set(encryptObject(this.toObject(), collectionName)); // Encrypt the document
+        next();
+    });
+
+    // Decrypt after creating a document
+    schema.post("create", function (docs) {
+        for (let doc of docs) {
+            doc.set(decryptObject(doc.toObject(), collectionName)); // Decrypt document fields
+        }
+    });
+
+    // Decrypt after update operations (single or multiple documents)
+    schema.post("update", function (result) {
+        if (Array.isArray(result)) {
+            result.forEach(doc => doc.set(decryptObject(doc.toObject(), collectionName)));
+        } else {
+            result.set(decryptObject(result.toObject(), collectionName)); // Decrypt single document
+        }
+    });
+
+    // Decrypt after updateOne operations
+    schema.post("updateOne", function (result) {
+        if (result && result.toObject) {
+            result.set(decryptObject(result.toObject(), collectionName)); // Decrypt single document
+        }
+    });
+
+    // Decrypt after find operations (e.g., find, findMany)
+    schema.post("find", function (docs) {
+        console.log("find", docs);
+        docs.forEach(doc => {
+            doc.set(decryptObject(doc.toObject(), collectionName)); // Decrypt document fields
+        });
+    });
+
+    // Decrypt after findById operations
+    schema.post("findById", function (doc) {
+        console.log("findById", doc);
+        if (doc) doc.set(decryptObject(doc.toObject(), collectionName)); // Decrypt single document
+    });
+
+    // Decrypt after findOne operations
+    schema.post("findOne", function (doc) {
+        console.log("findOne", doc);
+        if (doc) doc.set(decryptObject(doc.toObject(), collectionName)); // Decrypt single document
+    });
+
+    // Decrypt after findOneAndUpdate operations
+    schema.post("findOneAndUpdate", function (doc) {
+        console.log("findOneAndUpdate", doc);
+        if (doc) {
+            doc.set(decryptObject(doc.toObject(), collectionName)); // Decrypt single document
+        }
+    });
+
+    // Decrypt after findOneAndDelete operations
+    schema.post("findOneAndDelete", function (doc) {
+        console.log("findOneAndDelete", doc);
+        if (doc) doc.set(decryptObject(doc.toObject(), collectionName)); // Decrypt single document
+    });
+
+    // Decrypt after deleteMany operations
+    schema.post("deleteMany", function (result) {
+        console.log("deleteMany", result);
+        if (Array.isArray(result)) {
+            result.forEach(doc => doc.set(decryptObject(doc.toObject(), collectionName))); // Decrypt document fields
+        }
+    });
+
+    // Decrypt after deleteOne operations
+    schema.post("deleteOne", function (doc) {
+        console.log("deleteOne", doc);
+        if (doc) doc.set(decryptObject(doc.toObject(), collectionName)); // Decrypt single document
+    });
+
+    // Decrypt after remove operations
+    schema.post("remove", function (doc) {
+        console.log("remove", doc);
+        if (doc) doc.set(decryptObject(doc.toObject(), collectionName)); // Decrypt single document
+    });
+}