Blame view

node_modules/inquirer/lib/ui/prompt.js 3.21 KB
ce4c83ff   wxy   初始提交
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
  'use strict';
  var _ = require('lodash');
  var rx = require('rx-lite-aggregates');
  var util = require('util');
  var runAsync = require('run-async');
  var utils = require('../utils/utils');
  var Base = require('./baseUI');
  
  /**
   * Base interface class other can inherits from
   */
  
  var PromptUI = module.exports = function (prompts, opt) {
    Base.call(this, opt);
    this.prompts = prompts;
  };
  util.inherits(PromptUI, Base);
  
  PromptUI.prototype.run = function (questions) {
    // Keep global reference to the answers
    this.answers = {};
  
    // Make sure questions is an array.
    if (_.isPlainObject(questions)) {
      questions = [questions];
    }
  
    // Create an observable, unless we received one as parameter.
    // Note: As this is a public interface, we cannot do an instanceof check as we won't
    // be using the exact same object in memory.
    var obs = _.isArray(questions) ? rx.Observable.from(questions) : questions;
  
    this.process = obs
      .concatMap(this.processQuestion.bind(this))
      // `publish` creates a hot Observable. It prevents duplicating prompts.
      .publish();
  
    this.process.connect();
  
    return this.process
      .reduce(function (answers, answer) {
        _.set(this.answers, answer.name, answer.answer);
        return this.answers;
      }.bind(this), {})
      .toPromise(Promise)
      .then(this.onCompletion.bind(this));
  };
  
  /**
   * Once all prompt are over
   */
  
  PromptUI.prototype.onCompletion = function (answers) {
    this.close();
  
    return answers;
  };
  
  PromptUI.prototype.processQuestion = function (question) {
    question = _.clone(question);
    return rx.Observable.defer(function () {
      var obs = rx.Observable.of(question);
  
      return obs
        .concatMap(this.setDefaultType.bind(this))
        .concatMap(this.filterIfRunnable.bind(this))
        .concatMap(utils.fetchAsyncQuestionProperty.bind(null, question, 'message', this.answers))
        .concatMap(utils.fetchAsyncQuestionProperty.bind(null, question, 'default', this.answers))
        .concatMap(utils.fetchAsyncQuestionProperty.bind(null, question, 'choices', this.answers))
        .concatMap(this.fetchAnswer.bind(this));
    }.bind(this));
  };
  
  PromptUI.prototype.fetchAnswer = function (question) {
    var Prompt = this.prompts[question.type];
    this.activePrompt = new Prompt(question, this.rl, this.answers);
    return rx.Observable.defer(function () {
      return rx.Observable.fromPromise(this.activePrompt.run().then(function (answer) {
        return {name: question.name, answer: answer};
      }));
    }.bind(this));
  };
  
  PromptUI.prototype.setDefaultType = function (question) {
    // Default type to input
    if (!this.prompts[question.type]) {
      question.type = 'input';
    }
    return rx.Observable.defer(function () {
      return rx.Observable.return(question);
    });
  };
  
  PromptUI.prototype.filterIfRunnable = function (question) {
    if (question.when === false) {
      return rx.Observable.empty();
    }
  
    if (!_.isFunction(question.when)) {
      return rx.Observable.return(question);
    }
  
    var answers = this.answers;
    return rx.Observable.defer(function () {
      return rx.Observable.fromPromise(
        runAsync(question.when)(answers).then(function (shouldRun) {
          if (shouldRun) {
            return question;
          }
        })
      ).filter(function (val) {
        return val != null;
      });
    });
  };