Merge pull request #43 from hapinessjs/enhance-prepare-object

prepareUpdateObject should be recursive

Author: juneil

package-lock.json

@@ -1,6 +1,6 @@
 {
     "name": "@hapiness/mongo",
-    "version": "2.0.0",
+    "version": "2.0.1",
     "description": "Hapiness Module for MongoDB usage",
     "main": "commonjs/index.js",
     "types": "index.d.ts",

package.json

@@ -10,30 +10,13 @@ export class MongoUtils {
         }
     }
 
-    public static prepareUpdateObject(dto: any): any {
-        if (!dto || !Object.keys(dto).length) {
-            return {};
-        }
-
-        const preparedObject: any = {};
-
-        Object
-            .keys(dto)
-            .forEach(rootKey => {
-                if (!!dto[rootKey] && typeof dto[rootKey] === 'object') {
-                    Object
-                        .keys(dto[rootKey])
-                        .forEach(
-                            childKey => {
-                                preparedObject[`${rootKey}.${childKey}`] = dto[rootKey][childKey];
-                            }
-                        );
-                } else {
-                    preparedObject[rootKey] = dto[rootKey];
-                }
-            });
-
-        return preparedObject;
+    public static prepareUpdateObject(dto: any, prefix?: string): any {
+        return Object.entries(dto || {}).reduce((acc, [key, value]) => {
+            const _key = [prefix, key].filter(item => item).join('.');
+            return !Array.isArray(value) && value && typeof value === 'object' && Object.keys(value).length ?
+                Object.assign(MongoUtils.prepareUpdateObject(value, _key), acc) :
+                Object.assign({ [_key]: value }, acc);
+        }, {});
     }
 
     static filterFindCondition(condition: any): any {

src/module/services/mongo-utils.service.ts

@@ -129,6 +129,54 @@ export class MongoUtilTest {
             });
     }
 
+    /**
+     * `MongoUtils` function `prepareUpdateObject` should return an unfold object (3)
+     */
+    @test('- `MongoUtils` function `prepareUpdateObject` should return an unfold object (3)')
+    testMongoUtilPrepareUpdateObjecUnfoldObject3() {
+        const dateObject = new Date();
+        unit
+            .object(
+                MongoUtils
+                    .prepareUpdateObject(
+                        {
+                            test1: 'value',
+                            test2: {
+                                key1: 'value1'
+                            },
+                            test3: {
+                                key1: {
+                                    key2: {
+                                        key3: {
+                                            key4: {
+                                                key5: 'value2'
+                                            }
+                                        },
+                                        key6: 'value3'
+                                    }
+                                },
+                                key7: {
+                                    key8: 'value4'
+                                },
+                                key8: 'value5',
+                                key9: dateObject,
+                                key10: [1, 2, 3, 4, 5, 6]
+                            }
+                        }
+                    )
+            )
+            .is({
+               test1: 'value',
+               'test2.key1': 'value1',
+               'test3.key1.key2.key3.key4.key5': 'value2',
+               'test3.key1.key2.key6': 'value3',
+               'test3.key7.key8': 'value4',
+               'test3.key8': 'value5',
+               'test3.key9': dateObject,
+               'test3.key10': [1, 2, 3, 4, 5, 6]
+              });
+    }
+
     /**
      * `MongoUtils` function `filterFindCondition` without id or _id should return the object
      */