Fix flatten issue with toJSON

lib/JSON2CSVBase.js

@@ -4,7 +4,6 @@ const os = require('os');
 const lodashGet = require('lodash.get');
 const lodashSet = require('lodash.set');
 const lodashCloneDeep = require('lodash.clonedeep');
-const flatten = require('flat');
 
 class JSON2CSVBase {
   constructor(opts) {
@@ -64,7 +63,7 @@ class JSON2CSVBase {
       ? this.unwindData(row, this.opts.unwind)
       : [row];
     if (this.opts.flatten) {
-      return processedRow.map(flatten);
+      return processedRow.map(this.flatten);
     }
 
     return processedRow;
@@ -192,10 +191,42 @@ class JSON2CSVBase {
     return stringifiedValue;
   }
 
+  /**
+   * Performs the flattening of a data row recursively
+   *
+   * @param {Object} dataRow Original JSON object
+   * @returns {Object} Flattened object
+   */
+  flatten(dataRow) {
+    function step (obj, flatDataRow, currentPath) {
+      Object.keys(obj).forEach((key) => {
+        const value = obj[key];
+
+        const newPath = currentPath
+          ? `${currentPath}.${key}`
+          : key;
+
+        if (typeof value !== 'object'
+          || Array.isArray(value)
+          || Object.prototype.toString.call(value.toJSON) === '[object Function]'
+          || !Object.keys(value).length) {
+          flatDataRow[newPath] = value;
+          return;
+        }
+
+        step(value, flatDataRow, newPath);
+      });
+
+      return flatDataRow;
+    }
+
+    return step(dataRow, {});
+  }
+
   /**
    * Performs the unwind recursively in specified sequence
    *
-   * @param {Array} dataRow Original JSON object
+   * @param {Object} dataRow Original JSON object
    * @param {String[]} unwindPaths The paths as strings to be used to deconstruct the array
    * @returns {Array} Array of objects containing all rows after unwind of chosen paths
    */

package.json

@@ -39,10 +39,8 @@
   "dependencies": {
     "cli-table2": "^0.2.0",
     "commander": "^2.8.1",
-    "flat": "^4.0.0",
     "jsonparse": "^1.3.1",
     "lodash.clonedeep": "^4.5.0",
-    "lodash.flatten": "^4.4.0",
     "lodash.get": "^4.4.0",
     "lodash.set": "^4.3.0"
   },

test/JSON2CSVParser.js

@@ -275,6 +275,18 @@ module.exports = (testRunner, jsonFixtures, csvFixtures) => {
     t.end();
   });
 
+  testRunner.add('should support flattenning JSON with toJSON', (t) => {
+    const opts = {
+      flatten: true
+    };
+
+    const parser = new Json2csvParser(opts);
+    const csv = parser.parse(jsonFixtures.flattenToJSON);
+
+    t.equal(csv, csvFixtures.flattenToJSON);
+    t.end();
+  });
+
   testRunner.add('should unwind and flatten an object in the right order', (t) => {
     const opts = {
       unwind: ['items'],

test/fixtures/csv/flattenToJSON.csv

@@ -0,0 +1,2 @@
+"hello.world","lorem.ipsum.dolor"
+"good afternoon","good evening"

test/fixtures/json/flattenToJSON.js

@@ -0,0 +1,13 @@
+module.exports = {
+  hello: {
+    world: {
+      again: 'good morning',
+      toJSON: () =>'good afternoon'
+    }
+  },
+  lorem: {
+    ipsum: {
+      dolor: 'good evening'
+    }
+  }
+}