bonsai-analyzer/src/stats/getModulesByChunk.js

Trim your dependency tree.

/*
 * @flow
 */

import type {ChunkID, Module, RawStats} from '../types/Stats';

export default function getModulesByChunk(
  stats: RawStats,
  chunkWhitelist: Array<ChunkID>,
): {
  [key: ChunkID]: {
    id: ChunkID,
    length: number,
    modules: Array<Module>,
  },
} {
  const map = {};

  stats.chunks.forEach((chunk) => {
    if (chunkWhitelist.includes(chunk.id)) {
      map[chunk.id] = {
        id: chunk.id,
        length: 0,
        modules: [],
      };
    }
  });

  stats.modules.forEach((module) => {
    module.chunks.forEach((chunk) => {
      if (map[chunk]) {
        map[chunk].modules = map[chunk].modules.concat(module);
      }
    });
  });

  Object.keys(map).forEach((key) => {
    map[key].length = map[key].modules.length;

    if (map[key].length === 0) {
      delete map[key];
    }
  });

  return map;
}