promisify prompt.get

flatiron · Mar 2, 2020 · 8e1ac94 · 8e1ac94
1 parent 0ff93b6
commit 8e1ac94
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/prompt.js b/lib/prompt.js
@@ -192,6 +192,14 @@ prompt.history = function (search) {
 // Gets input from the user via stdin for the specified message(s) `msg`.
 //
 prompt.get = function (schema, callback) {
+  if (typeof callback === 'function') return prompt._get(schema, callback);
+
+  return new Promise((resolve, reject) => {
+    prompt._get(schema, (err, result) => err ? reject(err) : resolve(result));
+  })
+};
+
+prompt._get = function (schema, callback) {
   //
   // Transforms a full JSON-schema into an array describing path and sub-schemas.
   // Used for iteration purposes.