add branches feature

swyxio · Mar 11, 2020 · 0c1f9be · 0c1f9be
1 parent b54b792
commit 0c1f9be
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -1,11 +1,24 @@
 const pluginDecrypt = require('./pluginDecrypt');
+const chalk = require('chalk');
 
 module.exports = function netlify404nomore(conf) {
   return {
     name: 'netlify-plugin-encrypted-files',
-    onInit() {
+    onInit({ pluginConfig: { branches } }) {
       console.log('decrypting files');
-      pluginDecrypt({});
+      if (branches && branches.includes(process.env.BRANCH)) {
+        pluginDecrypt({});
+      } else if (typeof branches === 'undefined') {
+        pluginDecrypt({});
+      } else {
+        console.log(
+          'branches specified',
+          chalk.yellow(branches.join(' ')),
+          `but our current branch is ${chalk.blue(
+            process.env.BRANCH
+          )}, skipping decryption`
+        );
+      }
     }
   };
 };