Interactively choose script when none is specified

Author: RyanZim

README.md

@@ -14,6 +14,7 @@ npm install edit-script
 $ edit-script --help
 Edit npm scripts from the command line without worrying about json escaping.
 
+    edit-script
     edit-script <script>
 ```
 

cli.js

@@ -7,10 +7,11 @@ var findPkg = require('pkg-up');
 var script = process.argv[2];
 
 // help message:
-if (script === '--help') {
+if (process.argv[2] === '--help') {
   console.log(
     `Edit npm scripts from the command line without worrying about json escaping.
 
+    edit-script
     edit-script <script>`
   );
   process.exit();
@@ -20,37 +21,24 @@ var pkgPath;
 var pkg = {}
 var scripts = {};
 
-if (!script) {
-  console.error('Error: Must pass a script name');
-  process.exit(1);
-}
+const NEW_SCRIPT_SYMBOL = Symbol('Create new script');
+const EXIT_SYMBOL = Symbol('Exit');
 
+// Find package.json path:
 findPkg()
 .then(function (p) {
   if (!p) throw new Error('No package.json file found!');
   pkgPath = p;
+  // Load package.json:
   return fs.readJson(pkgPath);
 })
 .then(function (data) {
+  // Assign global variables:
   pkg = data;
   scripts = pkg.scripts;
-
-  return inquirer.prompt([
-    {
-      type: 'confirm',
-      name: 'create',
-      message: `The script "${script}" does not exist. Create it?`,
-      when: !scripts[script],
-    },
-  ]);
-})
-.then(function (answers) {
-  if (!scripts[script] && !answers.create) {
-    console.log('Aborting');
-    process.exit();
-  }
 })
-.then(function () {
+.then(getScriptName)
+.then(function editScript() {
   return inquirer.prompt([
     {
       type: 'editor',
@@ -59,18 +47,100 @@ findPkg()
       default: scripts[script],
     },
   ])
-})
-.then(function (answers) {
-  var val = answers.script.trim();
-  if (!val) {
-    console.log('Deleting script.');
-    delete scripts[script];
+  .then(function (answers) {
+    var val = answers.script.trim();
+    if (!val) {
+      console.log('Deleting script.');
+      delete scripts[script];
+    } else {
+      scripts[script] = val;
+    }
+    return fs.writeJson(pkgPath, pkg)
+  })
+  .catch(function (err) {
+    console.error(err);
+    process.exit(1);
+  });
+});
+
+function getScriptName() {
+  if (script) {
+    // Verify creation if the script does not exist:
+    return inquirer.prompt([
+      {
+        type: 'confirm',
+        name: 'create',
+        message: `The script "${script}" does not exist. Create it?`,
+        when: !scripts[script],
+      },
+    ])
+    .then(function (answers) {
+      if (!script && !scripts[script] && !answers.create) {
+        console.log('Aborting');
+        process.exit();
+      }
+    });
   } else {
-    scripts[script] = val;
+    // Get choices:
+    var choices = Object.keys(scripts).map(function (key) {
+      return {
+        name: pad(key, scripts[key]),
+        value: key,
+        short: key,
+      }
+    });
+    // Add aditional choices:
+    choices.push(new inquirer.Separator());
+    choices.push({
+      name: 'Create a new script',
+      value: NEW_SCRIPT_SYMBOL,
+    });
+    choices.push({
+      name: 'Exit edit-script',
+      value: EXIT_SYMBOL,
+    })
+    // Prompt for script name:
+    return inquirer.prompt([
+      {
+        type: 'list',
+        name: 'script',
+        message: 'Select a script to edit:',
+        choices: choices,
+      },
+    ])
+    .then(function (answers) {
+      switch (answers.script) {
+        case NEW_SCRIPT_SYMBOL:
+          // Get script name:
+          return inquirer.prompt([{
+            type: 'input',
+            name: 'name',
+            message: 'Enter the script name:',
+            validate: function (val) {
+              if (!val) return 'Script name must not be empty';
+              else return true;
+            },
+          }])
+          .then(function (answers) {
+            // Set it:
+            script = answers.name;
+          })
+          break;
+        case EXIT_SYMBOL:
+          process.exit();
+        default:
+          script = answers.script;
+      }
+    })
   }
-  return fs.writeJson(pkgPath, pkg)
-})
-.catch(function (err) {
-  console.error(err);
-  process.exit(1);
-});
+}
+
+function pad(str1, str2) {
+  var padLen = 60 - (str1.length + str2.length);
+  // Ensure at least one space:
+  var pad = ' ';
+  for (var i = 1; i < padLen; i++) {
+    pad += ' ';
+  }
+  return str1 + pad + str2;
+}