prompt: add y/N or Y/n if it misses from prompt

magic · Apr 25, 2020 · 715eec1 · 715eec1
1 parent fee7832
commit 715eec1
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/prompt/index.mjs b/src/prompt/index.mjs
@@ -7,6 +7,18 @@ export const prompt = (msg = '', options = {}) =>
   new Promise((resolve, reject) => {
     const { yesNo = false, pass = false, std = process, yesDefault = false } = options
 
+    if (yesNo) {
+      if (yesDefault) {
+        if (!msg.includes('Y/n')) {
+          msg += '(Y/n): '
+        }
+      } else {
+        if (!msg.includes('y/N')) {
+          msg += '(y/N): '
+        }
+      }
+    }
+
     const rl = readline.createInterface({
       input: std.stdin,
       output: std.stdout,