promisify prompt.get

flatiron · Mar 3, 2020 · b92a9a9 · b92a9a9
1 parent 0ff93b6
commit b92a9a9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/lib/prompt.js b/lib/prompt.js
@@ -192,6 +192,16 @@ 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(function (resolve, reject) {
+    prompt._get(schema, function (err, result) {
+      return 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.