BundleAnalyzerPlugin.js 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159
  1. "use strict";
  2. function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
  3. function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
  4. function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i] != null ? arguments[i] : {}; var ownKeys = Object.keys(source); if (typeof Object.getOwnPropertySymbols === 'function') { ownKeys = ownKeys.concat(Object.getOwnPropertySymbols(source).filter(function (sym) { return Object.getOwnPropertyDescriptor(source, sym).enumerable; })); } ownKeys.forEach(function (key) { _defineProperty(target, key, source[key]); }); } return target; }
  5. function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
  6. const bfj = require('bfj');
  7. const path = require('path');
  8. const mkdir = require('mkdirp');
  9. const {
  10. bold
  11. } = require('chalk');
  12. const Logger = require('./Logger');
  13. const viewer = require('./viewer');
  14. class BundleAnalyzerPlugin {
  15. constructor(opts) {
  16. this.opts = _objectSpread({
  17. analyzerMode: 'server',
  18. analyzerHost: '127.0.0.1',
  19. analyzerPort: 8888,
  20. reportFilename: 'report.html',
  21. defaultSizes: 'parsed',
  22. openAnalyzer: true,
  23. generateStatsFile: false,
  24. statsFilename: 'stats.json',
  25. statsOptions: null,
  26. excludeAssets: null,
  27. logLevel: 'info',
  28. // deprecated
  29. startAnalyzer: true
  30. }, opts);
  31. this.server = null;
  32. this.logger = new Logger(this.opts.logLevel);
  33. }
  34. apply(compiler) {
  35. this.compiler = compiler;
  36. const done = (stats, callback) => {
  37. callback = callback || (() => {});
  38. stats = stats.toJson(this.opts.statsOptions);
  39. const actions = [];
  40. if (this.opts.generateStatsFile) {
  41. actions.push(() => this.generateStatsFile(stats));
  42. } // Handling deprecated `startAnalyzer` flag
  43. if (this.opts.analyzerMode === 'server' && !this.opts.startAnalyzer) {
  44. this.opts.analyzerMode = 'disabled';
  45. }
  46. if (this.opts.analyzerMode === 'server') {
  47. actions.push(() => this.startAnalyzerServer(stats));
  48. } else if (this.opts.analyzerMode === 'static') {
  49. actions.push(() => this.generateStaticReport(stats));
  50. }
  51. if (actions.length) {
  52. // Making analyzer logs to be after all webpack logs in the console
  53. setImmediate(
  54. /*#__PURE__*/
  55. _asyncToGenerator(function* () {
  56. try {
  57. yield Promise.all(actions.map(action => action()));
  58. callback();
  59. } catch (e) {
  60. callback(e);
  61. }
  62. }));
  63. } else {
  64. callback();
  65. }
  66. };
  67. if (compiler.hooks) {
  68. compiler.hooks.done.tapAsync('webpack-bundle-analyzer', done);
  69. } else {
  70. compiler.plugin('done', done);
  71. }
  72. }
  73. generateStatsFile(stats) {
  74. var _this = this;
  75. return _asyncToGenerator(function* () {
  76. const statsFilepath = path.resolve(_this.compiler.outputPath, _this.opts.statsFilename);
  77. mkdir.sync(path.dirname(statsFilepath));
  78. try {
  79. yield bfj.write(statsFilepath, stats, {
  80. space: 2,
  81. promises: 'ignore',
  82. buffers: 'ignore',
  83. maps: 'ignore',
  84. iterables: 'ignore',
  85. circular: 'ignore'
  86. });
  87. _this.logger.info(`${bold('Webpack Bundle Analyzer')} saved stats file to ${bold(statsFilepath)}`);
  88. } catch (error) {
  89. _this.logger.error(`${bold('Webpack Bundle Analyzer')} error saving stats file to ${bold(statsFilepath)}: ${error}`);
  90. }
  91. })();
  92. }
  93. startAnalyzerServer(stats) {
  94. var _this2 = this;
  95. return _asyncToGenerator(function* () {
  96. if (_this2.server) {
  97. (yield _this2.server).updateChartData(stats);
  98. } else {
  99. _this2.server = viewer.startServer(stats, {
  100. openBrowser: _this2.opts.openAnalyzer,
  101. host: _this2.opts.analyzerHost,
  102. port: _this2.opts.analyzerPort,
  103. bundleDir: _this2.getBundleDirFromCompiler(),
  104. logger: _this2.logger,
  105. defaultSizes: _this2.opts.defaultSizes,
  106. excludeAssets: _this2.opts.excludeAssets
  107. });
  108. }
  109. })();
  110. }
  111. generateStaticReport(stats) {
  112. var _this3 = this;
  113. return _asyncToGenerator(function* () {
  114. yield viewer.generateReport(stats, {
  115. openBrowser: _this3.opts.openAnalyzer,
  116. reportFilename: path.resolve(_this3.compiler.outputPath, _this3.opts.reportFilename),
  117. bundleDir: _this3.getBundleDirFromCompiler(),
  118. logger: _this3.logger,
  119. defaultSizes: _this3.opts.defaultSizes,
  120. excludeAssets: _this3.opts.excludeAssets
  121. });
  122. })();
  123. }
  124. getBundleDirFromCompiler() {
  125. return this.compiler.outputFileSystem.constructor.name === 'MemoryFileSystem' ? null : this.compiler.outputPath;
  126. }
  127. }
  128. module.exports = BundleAnalyzerPlugin;