cli: refactor out cli.promptForInput (#300)

Author: joyeecheung

lib/cli.js

@@ -27,13 +27,9 @@ class CLI {
     this.SPINNER_STATUS = SPINNER_STATUS;
   }
 
-  prompt(question, defaultAnswer = true) {
-    const option =
-      `[${(defaultAnswer ? 'Y' : 'y')}/${(defaultAnswer ? 'n' : 'N')}]`;
-    this.separator();
-    const promptText = `${chalk.bold.cyan('?')} ${question} ${option} `;
+  promptForInput(question) {
     return new Promise((resolve, reject) => {
-      read({prompt: promptText}, (err, answer) => {
+      read({prompt: question}, (err, answer) => {
         if (err) {
           this.log(`\nCanceled: ${err.message}`);
           reject(new Error(IGNORE));
@@ -43,19 +39,27 @@ class CLI {
           reject(new Error(IGNORE));
           return;
         }
-        const trimmed = answer.toLowerCase().trim();
-        if (!trimmed) {
-          resolve(defaultAnswer);
-          return;
-        } else if (trimmed === 'y') {
-          resolve(true);
-          return;
-        }
-        resolve(false);
+        resolve(answer);
       });
     });
   }
 
+  async prompt(question, defaultAnswer = true) {
+    const option =
+      `[${(defaultAnswer ? 'Y' : 'y')}/${(defaultAnswer ? 'n' : 'N')}]`;
+    this.separator();
+    const promptText = `${chalk.bold.cyan('?')} ${question} ${option} `;
+
+    const answer = await this.promptForInput(promptText);
+    const trimmed = answer.toLowerCase().trim();
+    if (!trimmed) {
+      return defaultAnswer;
+    } else if (trimmed === 'y') {
+      return true;
+    }
+    return false;
+  }
+
   startSpinner(text) {
     this.spinner.text = text;
     this.spinner.start();