111 lines
3.9 KiB
JavaScript
111 lines
3.9 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.TaskWrapper = void 0;
|
|
const through = require("through");
|
|
const clearline_regex_constants_1 = require("../constants/clearline-regex.constants");
|
|
const state_constants_1 = require("../constants/state.constants");
|
|
const listr_error_interface_1 = require("../interfaces/listr-error.interface");
|
|
const listr_1 = require("../listr");
|
|
const prompt_1 = require("../utils/prompt");
|
|
/**
|
|
* Extend the task to have more functionality while accesing from the outside.
|
|
*/
|
|
class TaskWrapper {
|
|
constructor(task, errors, options) {
|
|
this.task = task;
|
|
this.errors = errors;
|
|
this.options = options;
|
|
}
|
|
/** Change the title of the current task. */
|
|
set title(data) {
|
|
this.task.title$ = data;
|
|
}
|
|
/** Get the title of the current task. */
|
|
get title() {
|
|
return this.task.title;
|
|
}
|
|
/** Send a output to the output channel. */
|
|
set output(data) {
|
|
this.task.output$ = data;
|
|
}
|
|
/** Get the output from the output channel. */
|
|
get output() {
|
|
return this.task.output;
|
|
}
|
|
/** Create a new subtask with given renderer selection from the parent task. */
|
|
newListr(task, options) {
|
|
let tasks;
|
|
if (typeof task === 'function') {
|
|
tasks = task(this);
|
|
}
|
|
else {
|
|
tasks = task;
|
|
}
|
|
return new listr_1.Listr(tasks, options);
|
|
}
|
|
/** Report a error in process for error collection. */
|
|
report(error) {
|
|
var _a, _b;
|
|
/* istanbul ignore if */
|
|
if (error instanceof listr_error_interface_1.ListrError) {
|
|
for (const err of error.errors) {
|
|
this.errors.push(err);
|
|
this.task.message$ = { error: err.message || ((_a = this.task) === null || _a === void 0 ? void 0 : _a.title) || 'Task with no title.' };
|
|
}
|
|
}
|
|
else {
|
|
this.errors.push(error);
|
|
this.task.message$ = { error: error.message || ((_b = this.task) === null || _b === void 0 ? void 0 : _b.title) || 'Task with no title.' };
|
|
}
|
|
}
|
|
/** Skip current task. */
|
|
skip(message) {
|
|
var _a;
|
|
this.task.state$ = state_constants_1.ListrTaskState.SKIPPED;
|
|
if (message) {
|
|
this.task.message$ = { skip: message || ((_a = this.task) === null || _a === void 0 ? void 0 : _a.title) || 'Task with no title.' };
|
|
}
|
|
}
|
|
/** Get the number of retrying, else returns false */
|
|
isRetrying() {
|
|
return this.task.isRetrying() ? this.task.retry : { count: 0 };
|
|
}
|
|
/**
|
|
* Create a new Enquirer prompt using prompt options.
|
|
*
|
|
* Since process.stdout is controlled by Listr, this will passthrough all Enquirer data through internal stdout.
|
|
*/
|
|
async prompt(options) {
|
|
var _a;
|
|
return prompt_1.createPrompt.bind(this)(options, { ...(_a = this.options) === null || _a === void 0 ? void 0 : _a.injectWrapper });
|
|
}
|
|
/** Cancels the current prompt attach to this task. */
|
|
cancelPrompt(throwError = false) {
|
|
return prompt_1.destroyPrompt.bind(this)(throwError);
|
|
}
|
|
/**
|
|
* Pass stream of data to internal stdout.
|
|
*
|
|
* Since Listr2 takes control of process.stdout utilizing the default renderer, any data outputted to process.stdout
|
|
* will corupt its looks.
|
|
*
|
|
* This returns a fake stream to pass any stream inside Listr as task data.
|
|
*/
|
|
stdout() {
|
|
return through((chunk) => {
|
|
const pattern = new RegExp(clearline_regex_constants_1.CLEAR_LINE_REGEX, 'gmi');
|
|
chunk = chunk.toString();
|
|
chunk = chunk.replace(pattern, '');
|
|
chunk = chunk.replace(new RegExp(clearline_regex_constants_1.BELL_REGEX, 'gmi'), '');
|
|
if (chunk !== '') {
|
|
this.output = chunk;
|
|
}
|
|
});
|
|
}
|
|
/** Run this task. */
|
|
run(ctx) {
|
|
return this.task.run(ctx, this);
|
|
}
|
|
}
|
|
exports.TaskWrapper = TaskWrapper;
|