gemcap-be-common 1.0.7 → 1.0.9

package/src/plugins/id.plugin.js

@@ -0,0 +1,53 @@
+module.exports = function mongooseLeanId(schema) {
+    schema.post('find', attachId);
+    schema.post('findOne', attachId);
+    schema.post('findOneAndUpdate', attachId);
+};
+function attachId(res) {
+    if (res === null) {
+        return;
+    }
+    function replaceId(res) {
+        if (Array.isArray(res)) {
+            res.forEach(function (v) {
+                if (isObjectId(v)) {
+                    return;
+                }
+                if (v._id) {
+                    v.id = v._id.toString();
+                }
+                Object.keys(v).map(function (k) {
+                    if (Array.isArray(v[k])) {
+                        replaceId(v[k]);
+                    }
+                });
+            });
+        }
+        else {
+            if (isObjectId(res)) {
+                return res;
+            }
+            if (res._id) {
+                res.id = res._id.toString();
+            }
+            Object.keys(res).map(function (k) {
+                if (Array.isArray(res[k])) {
+                    replaceId(res[k]);
+                }
+            });
+        }
+    }
+    if (this._mongooseOptions.lean) {
+        replaceId(res);
+    }
+}
+function isObjectId(v) {
+    if (v == null) {
+        return false;
+    }
+    var proto = Object.getPrototypeOf(v);
+    if (proto == null || proto.constructor == null || proto.constructor.name !== 'ObjectID') {
+        return false;
+    }
+    return v._bsontype === 'ObjectID';
+}