analyzer.js 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156
  1. 'use strict';
  2. var fs = require('fs');
  3. var path = require('path');
  4. var _ = require('lodash');
  5. var gzipSize = require('gzip-size');
  6. var Logger = require('./Logger');
  7. var Folder = require('./tree/Folder').default;
  8. var _require = require('./parseUtils'),
  9. parseBundle = _require.parseBundle;
  10. var FILENAME_QUERY_REGEXP = /\?.*$/;
  11. module.exports = {
  12. getViewerData,
  13. readStatsFromFile
  14. };
  15. function getViewerData(bundleStats, bundleDir, opts) {
  16. var _ref = opts || {},
  17. _ref$logger = _ref.logger,
  18. logger = _ref$logger === undefined ? new Logger() : _ref$logger;
  19. // Sometimes all the information is located in `children` array (e.g. problem in #10)
  20. if (_.isEmpty(bundleStats.assets) && !_.isEmpty(bundleStats.children)) {
  21. bundleStats = bundleStats.children[0];
  22. }
  23. // Picking only `*.js` assets from bundle that has non-empty `chunks` array
  24. bundleStats.assets = _.filter(bundleStats.assets, function (asset) {
  25. // Removing query part from filename (yes, somebody uses it for some reason and Webpack supports it)
  26. // See #22
  27. asset.name = asset.name.replace(FILENAME_QUERY_REGEXP, '');
  28. return _.endsWith(asset.name, '.js') && !_.isEmpty(asset.chunks);
  29. });
  30. // Trying to parse bundle assets and get real module sizes if `bundleDir` is provided
  31. var bundlesSources = null;
  32. var parsedModules = null;
  33. if (bundleDir) {
  34. bundlesSources = {};
  35. parsedModules = {};
  36. var _iteratorNormalCompletion = true;
  37. var _didIteratorError = false;
  38. var _iteratorError = undefined;
  39. try {
  40. for (var _iterator = bundleStats.assets[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = true) {
  41. var statAsset = _step.value;
  42. var assetFile = path.join(bundleDir, statAsset.name);
  43. var bundleInfo = void 0;
  44. try {
  45. bundleInfo = parseBundle(assetFile);
  46. } catch (err) {
  47. bundleInfo = null;
  48. }
  49. if (!bundleInfo) {
  50. logger.warn(`\nCouldn't parse bundle asset "${assetFile}".\n` + 'Analyzer will use module sizes from stats file.\n');
  51. parsedModules = null;
  52. bundlesSources = null;
  53. break;
  54. }
  55. bundlesSources[statAsset.name] = bundleInfo.src;
  56. _.assign(parsedModules, bundleInfo.modules);
  57. }
  58. } catch (err) {
  59. _didIteratorError = true;
  60. _iteratorError = err;
  61. } finally {
  62. try {
  63. if (!_iteratorNormalCompletion && _iterator.return) {
  64. _iterator.return();
  65. }
  66. } finally {
  67. if (_didIteratorError) {
  68. throw _iteratorError;
  69. }
  70. }
  71. }
  72. }
  73. var modules = getBundleModules(bundleStats);
  74. var assets = _.transform(bundleStats.assets, function (result, statAsset) {
  75. var asset = result[statAsset.name] = _.pick(statAsset, 'size');
  76. if (bundlesSources) {
  77. asset.parsedSize = bundlesSources[statAsset.name].length;
  78. asset.gzipSize = gzipSize.sync(bundlesSources[statAsset.name]);
  79. }
  80. // Picking modules from current bundle script
  81. asset.modules = _(modules).filter(function (statModule) {
  82. return assetHasModule(statAsset, statModule, parsedModules);
  83. }).each(function (statModule) {
  84. if (parsedModules) {
  85. statModule.parsedSrc = parsedModules[statModule.id];
  86. }
  87. });
  88. asset.tree = createModulesTree(asset.modules);
  89. }, {});
  90. return _.transform(assets, function (result, asset, filename) {
  91. result.push({
  92. label: filename,
  93. // Not using `asset.size` here provided by Webpack because it can be very confusing when `UglifyJsPlugin` is used.
  94. // In this case all module sizes from stats file will represent unminified module sizes, but `asset.size` will
  95. // be the size of minified bundle.
  96. statSize: asset.tree.size,
  97. parsedSize: asset.parsedSize,
  98. gzipSize: asset.gzipSize,
  99. groups: _.invokeMap(asset.tree.children, 'toChartData')
  100. });
  101. }, []);
  102. }
  103. function readStatsFromFile(filename) {
  104. return JSON.parse(fs.readFileSync(filename, 'utf8'));
  105. }
  106. function getBundleModules(bundleStats) {
  107. return _(bundleStats.chunks).map('modules').concat(bundleStats.modules).compact().flatten().uniqBy('id').value();
  108. }
  109. function assetHasModule(statAsset, statModule, parsedModules) {
  110. // Checking if this module is the part of asset chunks
  111. var moduleIsInsideAsset = _.some(statModule.chunks, function (moduleChunk) {
  112. return _.includes(statAsset.chunks, moduleChunk);
  113. });
  114. return moduleIsInsideAsset && (
  115. // ...and that we have found it during parsing
  116. !parsedModules || _.has(parsedModules, statModule.id));
  117. }
  118. function createModulesTree(modules) {
  119. var root = new Folder('.');
  120. _.each(modules, function (module) {
  121. return root.addModule(module);
  122. });
  123. return root;
  124. }