.select for .find and .findOne

Author: doguhanokumus

lib/mongoose-field-encryption.js

@@ -172,6 +172,19 @@ const fieldEncryption = function (schema, options) {
     }
   }
 
+  function findHook(_next) {
+    const next = getCompatitibleNextFunc(_next);
+    if (this.$useProjection) {
+      Object.keys(this._fields).forEach(key => {
+        if (fieldsToEncrypt.indexOf(key) === -1) return;
+        if (!this._fields[key] || (this._fields[key] !== 1 && this._fields[key] !== true)) return;
+        this._fields[`__enc_${key}`] = 1;
+        this._fields[`__enc_${key}_d`] = 1;
+      })
+    }
+    next()
+  }
+
   function updateHook(_next) {
     const next = getCompatitibleNextFunc(_next);
     for (const field of fieldsToEncrypt) {
@@ -249,6 +262,7 @@ const fieldEncryption = function (schema, options) {
     }
   });
 
+  schema.pre(['find', 'findOne'], findHook);
   schema.pre("findOneAndUpdate", updateHook);
 
   schema.pre("update", updateHook);