初始化

康凯

2022-05-20 aa4c235a8ca67ea8b731f90c951a465e92c0a865

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
116
117
118
119
'use strict';
const pMap = require('p-map');
const Task = require('./lib/task');
const TaskWrapper = require('./lib/task-wrapper');
const renderer = require('./lib/renderer');
const ListrError = require('./lib/listr-error');
 
const runTask = (task, context, errors) => {
    if (!task.isEnabled()) {
        return Promise.resolve();
    }
 
    return new TaskWrapper(task, errors).run(context);
};
 
class Listr {
    constructor(tasks, opts) {
        if (tasks && !Array.isArray(tasks) && typeof tasks === 'object') {
            if (typeof tasks.title === 'string' && typeof tasks.task === 'function') {
                throw new TypeError('Expected an array of tasks or an options object, got a task object');
            }
 
            opts = tasks;
            tasks = [];
        }
 
        if (tasks && !Array.isArray(tasks)) {
            throw new TypeError('Expected an array of tasks');
        }
 
        this._options = Object.assign({
            showSubtasks: true,
            concurrent: false,
            renderer: 'default',
            nonTTYRenderer: 'verbose'
        }, opts);
        this._tasks = [];
 
        this.concurrency = 1;
        if (this._options.concurrent === true) {
            this.concurrency = Infinity;
        } else if (typeof this._options.concurrent === 'number') {
            this.concurrency = this._options.concurrent;
        }
 
        this._RendererClass = renderer.getRenderer(this._options.renderer, this._options.nonTTYRenderer);
 
        this.exitOnError = this._options.exitOnError;
 
        this.add(tasks || []);
    }
 
    _checkAll(context) {
        for (const task of this._tasks) {
            task.check(context);
        }
    }
 
    get tasks() {
        return this._tasks;
    }
 
    setRenderer(value) {
        this._RendererClass = renderer.getRenderer(value);
    }
 
    add(task) {
        const tasks = Array.isArray(task) ? task : [task];
 
        for (const task of tasks) {
            this._tasks.push(new Task(this, task, this._options));
        }
 
        return this;
    }
 
    render() {
        if (!this._renderer) {
            this._renderer = new this._RendererClass(this._tasks, this._options);
        }
 
        return this._renderer.render();
    }
 
    run(context) {
        this.render();
 
        context = context || Object.create(null);
 
        const errors = [];
 
        this._checkAll(context);
 
        const tasks = pMap(this._tasks, task => {
            this._checkAll(context);
            return runTask(task, context, errors);
        }, {concurrency: this.concurrency});
 
        return tasks
            .then(() => {
                if (errors.length > 0) {
                    const err = new ListrError('Something went wrong');
                    err.errors = errors;
                    throw err;
                }
 
                this._renderer.end();
 
                return context;
            })
            .catch(error => {
                error.context = context;
                this._renderer.end(error);
                throw error;
            });
    }
}
 
module.exports = Listr;