@azteam/express 1.2.127 → 1.2.128

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@azteam/express",
-    "version": "1.2.127",
+    "version": "1.2.128",
     "description": "",
     "main": "src/index.js",
     "repository": {

package/src/ApiServer.js

@@ -12,16 +12,32 @@ import _ from 'lodash';
 import 'express-async-errors';
 import {errorCatch, ErrorException, NOT_FOUND, UNKNOWN} from '@azteam/error';
 
-function omitItem(item, guard) {
+
+const RES_TYPE = {
+    ARRAY: 'ARRAY',
+    OBJECT: 'OBJECT',
+    DOCS: 'DOCS'
+};
+
+function omitItem(item, guard, allows) {
+
+    if (_.isArray(guard)) {
+        guard = _.difference(guard, allows);
+    }
+
     if (item.toJSON) {
         item = item.toJSON();
     }
     if (_.isObject(item)) {
+        if (guard === '*') {
+            return _.pick(item, allows);
+        }
         return _.omit(item, guard);
     }
     return item;
 }
 
+
 class ApiServer {
     constructor(currentDir = '', options = {}) {
         this.options = options;
@@ -160,46 +176,54 @@ class ApiServer {
 
                 res.success = function(data = {}, guard = [], allows = []) {
                     if (data) {
-                        guard = [
-                            ...guard,
-                            '__v',
-                            '_id',
-                            'deleted_at',
-                            'updated_at',
-                            'created_id',
-                            'modified_id'
-                        ];
-
-                        if (_.isArray(data) || data.docs) {
+                        let resType = RES_TYPE.ARRAY;
+                        if (_.isObject(data)) {
+                            resType = RES_TYPE.OBJECT;
+                            if (data.docs) {
+                                resType = RES_TYPE.DOCS;
+                            }
+                        }
+
+                        if (_.isArray(guard)) {
                             guard = [
                                 ...guard,
-                                'metadata_disable',
-                                'metadata_keywords',
-                                'metadata_description',
-                                'metadata_image_url'
+                                '__v',
+                                '_id',
+                                'deleted_at',
+                                'updated_at',
+                                'created_id',
+                                'modified_id'
                             ];
+                            if (resType === RES_TYPE.ARRAY || resType === RES_TYPE.DOCS) {
+                                guard = [
+                                    ...guard,
+                                    'metadata_disable',
+                                    'metadata_keywords',
+                                    'metadata_description',
+                                    'metadata_image_url'
+                                ];
+                            }
                         }
 
-                        guard = _.difference(guard, allows);
-
-                        if (_.isArray(data)) {
-                            data = _.map(data, item => {
-                                return omitItem(item, guard);
+                        let guardData = data;
+                        if (resType === RES_TYPE.DOCS) {
+                            guardData.docs = _.map(data.docs, item => {
+                                return omitItem(item, guard, allows);
                             });
-                        } else if (_.isObject(data)) {
-                            if (data.docs) {
-                                data.docs = _.map(data.docs, item => {
-                                    return omitItem(item, guard);
-                                });
-                            } else {
-                                data = omitItem(data, guard);
-                            }
+                        } else if (resType === RES_TYPE.ARRAY) {
+                            guardData = _.map(data, item => {
+                                return omitItem(item, guard, allows);
+                            });
+                        } else if (resType === RES_TYPE.OBJECT) {
+                            guardData = omitItem(data, guard, allows);
                         }
+
+
                     }
 
                     return res.json({
                         success: true,
-                        data,
+                        data: guardData,
                         options: req.resOptions
                     });
                 };