Merge pull request #4 from kuldeepkeshwar/master

move common logic to size-plugin-core & add support for size-plugin bot

Author: luwes

README.md

@@ -15,6 +15,8 @@
 
 > 🙋 Using Webpack? Check out the original [size-plugin](https://github.com/GoogleChromeLabs/size-plugin).
 
+> 🙋‍♂ Using CI ? Check out [size-plugin-bot](https://github.com/kuldeepkeshwar/size-plugin-bot) 🤖 to automate intergation with CI
+
 ## Installation
 
 Install `rollup-plugin-size` as a development dependency using npm:

package.json

@@ -10,12 +10,7 @@
   "author": "Wesley Luyten <me@wesleyluyten.com> (https://wesleyluyten.com)",
   "license": "Apache-2.0",
   "dependencies": {
-    "brotli-size": "^0.0.3",
-    "chalk": "^2.4.2",
-    "glob": "^7.1.3",
-    "gzip-size": "^5.0.0",
-    "minimatch": "^3.0.4",
-    "pretty-bytes": "^5.1.0"
+    "size-plugin-core": "^0.0.6"
   },
   "devDependencies": {
     "eslint": "^5.14.0",

rollup-plugin-size.js

@@ -1,145 +1,51 @@
-/**
- * Copyright 2018 Google LLC
- *
- * Licensed under the Apache License, Version 2.0 (the "License"); you may not
- * use this file except in compliance with the License. You may obtain a copy of
- * the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
- * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
- * License for the specific language governing permissions and limitations under
- * the License.
- */
-
 const path = require('path');
-const chalk = require('chalk');
-const { promisify } = require('util');
-const globPromise = require('glob');
-const minimatch = require('minimatch');
-const gzipSize = require('gzip-size');
-const brotliSize = require('brotli-size');
-const prettyBytes = require('pretty-bytes');
-const { toMap, dedupe } = require('./utils.js');
-const glob = promisify(globPromise);
-
-brotliSize.file = (path, options) => {
-  return new Promise((resolve, reject) => {
-    const stream = fs.createReadStream(path);
-    stream.on('error', reject);
-
-    const brotliStream = stream.pipe(brotliSize.stream(options));
-    brotliStream.on('error', reject);
-    brotliStream.on('brotli-size', resolve);
-  });
-};
-
+const SizePluginCore = require('size-plugin-core');
 
 const defaults = {
   gzip: true,
   brotli: false,
-  pattern: '**/*.{mjs,js,css,html}',
+  pattern: '**/*.{mjs,js,jsx,css,html}',
   exclude: undefined,
-  columnWidth: 20
+  writeFile: true,
+  publish: false
 };
-
+/**
+ * Size Plugin for Rollup
+ * @param {Object} options
+ * @param {string} [options.gzip] use gzip compression for files
+ * @param {string} [options.brotli] use brotli compression for files
+ * @param {string} [options.pattern] minimatch pattern of files to track
+ * @param {string} [options.exclude] minimatch pattern of files NOT to track
+ * @param {string} [options.filename] file name to save filesizes to disk
+ * @param {boolean} [options.publish] option to publish filesizes to size-plugin-store
+ * @param {boolean} [options.writeFile] option to save filesizes to disk
+ */
 function bundleSize(options) {
-  const { pattern, exclude, columnWidth, brotli } = Object.assign(defaults, options);
-
-  let max = '';
-  let firstTime = true;
-  let initialSizes;
-
-  const compressionSize = brotli ? brotliSize : gzipSize;
-
-  function buildStart() {
-    max = '';
-  }
-
-  function renderChunk(code) {
-    max = max += code;
-    return null;
-  }
+  const coreOptions = Object.assign(defaults, options);
+  coreOptions.compression = coreOptions.brotli ? 'brotli' : 'gzip';
 
+  const core = new SizePluginCore(coreOptions);
   async function generateBundle(outputOptions, bundle) {
-    if (firstTime) {
-      firstTime = false;
-      initialSizes = await getSizes(path.dirname(outputOptions.file));
-    }
-    outputSizes(bundle).catch(console.error);
-  }
-
-  async function outputSizes(assets) {
-    // map of filenames to their previous size
-    const sizesBefore = await Promise.resolve(initialSizes);
-    const isMatched = minimatch.filter(pattern);
-    const isExcluded = exclude ? minimatch.filter(exclude) : () => false;
-    const assetNames = Object.keys(assets).filter(
-      file => isMatched(file) && !isExcluded(file)
-    );
-    const sizes = await Promise.all(
-      assetNames.map(name => compressionSize(assets[name].code))
-    );
-
-    // map of de-hashed filenames to their final size
-    initialSizes = toMap(assetNames, sizes);
-
-    // get a list of unique filenames
-    const files = Object.keys(initialSizes).filter(dedupe);
-
-    const width = Math.max(...files.map(file => file.length));
-    let output = '';
-    for (const name of files) {
-      const size = initialSizes[name] || 0;
-      const delta = size - (sizesBefore[name] || 0);
-      const msg =
-        new Array(
-          (width !== name.length ? width : columnWidth) - name.length + 2
-        ).join(' ') +
-        name +
-        ' ⏤  ';
-      const color =
-        size > 100 * 1024
-          ? 'red'
-          : size > 40 * 1024
-          ? 'yellow'
-          : size > 20 * 1024
-          ? 'cyan'
-          : 'green';
-      let sizeText = chalk[color](prettyBytes(size));
-      if (delta) {
-        let deltaText = (delta > 0 ? '+' : '') + prettyBytes(delta);
-        if (delta > 1024) {
-          sizeText = chalk.bold(sizeText);
-          deltaText = chalk.red(deltaText);
-        } else if (delta < -10) {
-          deltaText = chalk.green(deltaText);
-        }
-        sizeText += ` (${deltaText})`;
-      }
-      output += msg + sizeText;
-    }
-    if (output) {
-      console.log(output);
+    try {
+      const assets = Object.keys(bundle).reduce((agg, key) => {
+        agg[key] = {
+          source: bundle[key].code
+        };
+        return agg;
+      }, {});
+      const outputPath = outputOptions.dir
+        ? path.resolve(outputOptions.dir)
+        : path.dirname(outputOptions.file);
+      outputOptions.file &&
+        (core.options.pattern = `${Object.keys(assets).pop()}`);
+      const output = await core.execute(assets, outputPath);
+      output && console.log(output);
+    } catch (error) {
+      console.error(error);
     }
   }
-
-  async function getSizes(cwd) {
-    const files = await glob(pattern, { cwd, ignore: exclude });
-
-    const sizes = await Promise.all(
-      files.map(file => compressionSize.file(path.join(cwd, file)).catch(() => null))
-    );
-
-    return toMap(files, sizes);
-  }
-
   return {
     name: 'rollup-plugin-size',
-    buildStart,
-    renderChunk,
     generateBundle
   };
 }