Order code

Author: luwes

publish-size.js

@@ -1,30 +1,36 @@
-const { repo, sha, event, branch, pull_request_number, ci } = require('ci-env');
+const { repo, sha, event, branch, ci } = require('ci-env');
 const axios = require('axios');
-const SIZE_STORE_ENDPOINT = process.env.SIZE_STORE_ENDPOINT || 'https://size-plugin-store.now.sh' ;
+const SIZE_STORE_ENDPOINT =
+  process.env.SIZE_STORE_ENDPOINT || 'https://size-plugin-store.now.sh';
 
 // TODO: add option to turn off publishing of sizes.
 
-async function publishDiff(diff,filename) {
-	if (process.env.NODE_ENV !=='test' &&  ci && event == 'pull_request') {
-		try {
-			const params = { ci,repo, branch, sha, diff,filename };
-			await axios.post(`${SIZE_STORE_ENDPOINT}/diff`, params);
-		}
-		catch (error) {
-			console.error('error: while publishing diff', error);
-		}
-	}
+async function publishDiff(diff, filename) {
+  if (process.env.NODE_ENV !== 'test' && ci && event == 'pull_request') {
+    try {
+      const params = { ci, repo, branch, sha, diff, filename };
+      await axios.post(`${SIZE_STORE_ENDPOINT}/diff`, params);
+    } catch (error) {
+      console.error('error: while publishing diff', error);
+    }
+  }
 }
-async function publishSizes(size,filename) {
-	// TODO: read allowed branch from configuration
-	if (process.env.NODE_ENV !=='test' &&  ci && event == 'push' && branch==='master') {
-		try {
-			const params = { ci,repo, branch, sha, size,filename };
-			await axios.post(`${SIZE_STORE_ENDPOINT}/size`, params);
-		}
-		catch (error) {
-			console.error('error: while publishing sizes', error);
-		}
-	}
+
+async function publishSizes(size, filename) {
+  // TODO: read allowed branch from configuration
+  if (
+    process.env.NODE_ENV !== 'test' &&
+    ci &&
+    event == 'push' &&
+    branch === 'master'
+  ) {
+    try {
+      const params = { ci, repo, branch, sha, size, filename };
+      await axios.post(`${SIZE_STORE_ENDPOINT}/size`, params);
+    } catch (error) {
+      console.error('error: while publishing sizes', error);
+    }
+  }
 }
-module.exports = { publishSizes,publishDiff };
+
+module.exports = { publishSizes, publishDiff };

rollup-plugin-size.js

@@ -43,8 +43,8 @@ const defaults = {
   brotli: false,
   pattern: '**/*.{mjs,js,jsx,css,html}',
   exclude: undefined,
-  writeFile:true,
-  publish:false,
+  writeFile: true,
+  publish: false,
   columnWidth: 20
 };
 /**
@@ -71,14 +71,19 @@ function bundleSize(_options) {
     initialSizes = await load(path.resolve(outputOptions.dir));
     outputSizes(bundle).catch(console.error);
   }
-  function filterFiles(files) {
-    const isMatched = minimatch.filter(pattern);
-    const isExcluded = exclude ? minimatch.filter(exclude) : () => false;
-    return files.filter(file => isMatched(file) && !isExcluded(file));
+
+  async function load(outputPath) {
+    const data = await readFromDisk(filename);
+    if (data.length) {
+      const [{ files }] = data;
+      return toFileMap(files);
+    }
+    return getSizes(outputPath);
   }
+
   async function readFromDisk(filename) {
     try {
-      if(options.writeFile){
+      if (options.writeFile) {
         return [];
       }
       await fs.ensureFile(filename);
@@ -88,20 +93,40 @@ function bundleSize(_options) {
       return [];
     }
   }
+
+  async function getSizes(cwd) {
+    const files = await glob(pattern, { cwd, ignore: exclude });
+
+    const sizes = await Promise.all(
+      filterFiles(files).map(file =>
+        compressionSize.file(path.join(cwd, file)).catch(() => null)
+      )
+    );
+
+    return toMap(files, sizes);
+  }
+
+  function filterFiles(files) {
+    const isMatched = minimatch.filter(pattern);
+    const isExcluded = exclude ? minimatch.filter(exclude) : () => false;
+    return files.filter(file => isMatched(file) && !isExcluded(file));
+  }
+
   async function writeToDisk(filename, stats) {
     if (
       process.env.NODE_ENV === 'production' &&
       stats.files.some(file => file.diff !== 0)
     ) {
       const data = await readFromDisk(filename);
       data.unshift(stats);
-      if(options.writeFile){
+      if (options.writeFile) {
         await fs.ensureFile(filename);
         await fs.writeJSON(filename, data);
       }
-      options.publish && await publishSizes(data, options.filename);
+      options.publish && (await publishSizes(data, options.filename));
     }
   }
+
   async function save(files) {
     const stats = {
       timestamp: Date.now(),
@@ -112,19 +137,11 @@ function bundleSize(_options) {
         diff: file.size - file.sizeBefore
       }))
     };
-    options.publish && await publishDiff(stats, options.filename);
+    options.publish && (await publishDiff(stats, options.filename));
     options.save && (await options.save(stats));
     await writeToDisk(filename, stats);
   }
-  async function load(outputPath) {
 
-    const data = await readFromDisk(filename);
-    if (data.length) {
-      const [{ files }] = data;
-      return toFileMap(files);
-    }
-    return getSizes(outputPath);
-  }
   async function outputSizes(assets) {
     const sizesBefore = await Promise.resolve(initialSizes);
     const assetNames = filterFiles(Object.keys(assets));
@@ -190,18 +207,6 @@ function bundleSize(_options) {
     output && console.log('\n' + output);
   }
 
-  async function getSizes(cwd) {
-    const files = await glob(pattern, { cwd, ignore: exclude });
-
-    const sizes = await Promise.all(
-      filterFiles(files).map(file =>
-        compressionSize.file(path.join(cwd, file)).catch(() => null)
-      )
-    );
-
-    return toMap(files, sizes);
-  }
-
   return {
     name: 'rollup-plugin-size',
     generateBundle