close prompt on error

lib/adapter.js

@@ -37,6 +37,12 @@ class TerminalAdapter {
     return string.split('\n').map(line => this[`_colorDiff${name}`](line)).join('\n');
   }
 
+  close() {
+    if (this.promptUi) {
+      this.promptUi.close();
+    }
+  }
+
   /**
    * Prompt a user for one or more questions and pass
    * the answer(s) to the provided callback.
@@ -51,16 +57,14 @@ class TerminalAdapter {
    * @param {Function} [cb] Deprecated: callback for backward compatibility.
    * @return {Object} promise answers
    */
-  prompt(questions, answers, cb) {
-    if (typeof answers === 'function') {
-      cb = answers;
-      answers = undefined;
-    }
-    const promise = this.promptModule(questions, answers);
-    if (typeof cb === 'function') {
-      promise.then(answers => cb(answers));
-    }
-    return promise;
+  prompt(questions, answers, cb = () => {}) {
+    const promptPromise = this.promptModule(questions, typeof answers === 'function' ? undefined : answers);
+    this.promptUi = promptPromise.ui;
+    promptPromise.then(result => {
+      delete this.promptUi;
+      (typeof answers === 'function' ? answers : cb)(result);
+    });
+    return promptPromise;
   }
 
   /**

lib/environment.js

@@ -1058,6 +1058,7 @@ class Environment extends Base {
 
       this.runLoop.on('error', error => {
         this.emit('error', error);
+        this.adapter.close();
       });
 
       this.runLoop.on('paused', () => {