Fix edge cases

src/analyzer.js

@@ -27,7 +27,10 @@ function getViewerData(bundleStats, bundleDir, opts) {
   const isAssetIncluded = createAssetsFilter(excludeAssets);
 
   // Sometimes all the information is located in `children` array (e.g. problem in #10)
-  if (bundleStats.assets.length === 0 && bundleStats.children.length > 0) {
+  if (
+    (bundleStats.assets == null || bundleStats.assets.length === 0)
+    && bundleStats.children.length > 0
+  ) {
     const {children} = bundleStats;
     bundleStats = bundleStats.children[0];
     // Sometimes if there are additional child chunks produced add them as child assets,
@@ -164,7 +167,7 @@ function getViewerData(bundleStats, bundleDir, opts) {
     statSize: asset.tree.size || asset.size,
     parsedSize: asset.parsedSize,
     gzipSize: asset.gzipSize,
-    groups: asset.tree.children.map(i => i.toChartData()),
+    groups: _.invokeMap(asset.tree.children, 'toChartData'),
     isInitialByEntrypoint: chunkToInitialByEntrypoint[filename] ?? {}
   }));
 }

src/parseUtils.js

@@ -1,6 +1,7 @@
 const fs = require('fs');
 const acorn = require('acorn');
 const walk = require('acorn-walk');
+const _ = require('lodash');
 
 module.exports = {
   parseBundle
@@ -139,12 +140,12 @@ function parseBundle(bundlePath) {
     }
   );
 
-  let modules;
+  const modules = {};
 
   if (walkState.locations) {
-    modules = walkState.locations.map(loc => content.slice(loc.start, loc.end));
-  } else {
-    modules = {};
+    Object.entries(walkState.locations).forEach(([id, loc]) => {
+      modules[id] = content.slice(loc.start, loc.end);
+    });
   }
 
   return {

src/tree/BaseFolder.js

@@ -1,3 +1,5 @@
+import _ from 'lodash';
+
 import Node from './Node';
 
 export default class BaseFolder extends Node {
@@ -109,7 +111,7 @@ export default class BaseFolder extends Node {
       label: this.name,
       path: this.path,
       statSize: this.size,
-      groups: this.children.map(i => i.toChartData())
+      groups: _.invokeMap(this.children, 'toChartData')
     };
   }
 

src/tree/ConcatenatedModule.js

@@ -1,3 +1,4 @@
+import _ from 'lodash';
 import Module from './Module';
 import ContentModule from './ContentModule';
 import ContentFolder from './ContentFolder';
@@ -73,14 +74,14 @@ export default class ConcatenatedModule extends Module {
   }
 
   mergeNestedFolders() {
-    this.children.forEach(child => child.mergeNestedFolders());
+    _.invokeMap(this.children, 'mergeNestedFolders');
   }
 
   toChartData() {
     return {
       ...super.toChartData(),
       concatenated: true,
-      groups: this.children.map(i => i.toChartData())
+      groups: _.invokeMap(this.children, 'toChartData')
     };
   }