120 lines
2.5 KiB
JavaScript
120 lines
2.5 KiB
JavaScript
|
'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;
|