143 lines
3.8 KiB
TypeScript
143 lines
3.8 KiB
TypeScript
import { execSync } from 'node:child_process';
|
|
|
|
export const TaskCommandType = ['npm-install'] as const;
|
|
export type TaskCommand = {
|
|
key?: string;
|
|
/**
|
|
* 任务描述
|
|
*/
|
|
description?: string;
|
|
/**
|
|
* 命令, 执行的任务
|
|
*/
|
|
command: string;
|
|
type?: (typeof TaskCommandType)[number] | string;
|
|
/**
|
|
* 任务执行完成后，执行判断的命令
|
|
*/
|
|
after?: string;
|
|
/**
|
|
* 任务执行前，执行判断的命令
|
|
*/
|
|
before?: string;
|
|
/**
|
|
* 任务执行完成后, 检测输出的文本内容，如果有这个文本，表示任务执行成功
|
|
* 如果没有这个文本，表示任务执行失败
|
|
*/
|
|
afterCheck?: string;
|
|
/**
|
|
* 任务执行前, 检测输出的文本内容，如果有这个文本，表示任务已经安装
|
|
* 如果没有这个文本，表示任务没有安装
|
|
*/
|
|
beforeCheck?: string;
|
|
};
|
|
export class TasksCommand {
|
|
tasks: Map<string, TaskCommand> = new Map();
|
|
constructor() {}
|
|
addTask(task: TaskCommand, run?: boolean) {
|
|
const key = task?.key || task?.description;
|
|
if (!key) {
|
|
throw new Error('当前的任务没有key');
|
|
}
|
|
this.tasks.set(key, task);
|
|
if (run) {
|
|
this.runTask(key);
|
|
}
|
|
return this;
|
|
}
|
|
getTask(name: string) {
|
|
return this.tasks.get(name);
|
|
}
|
|
runTask(name: string) {
|
|
const task = this.getTask(name);
|
|
const end = (data?: { code: number; [key: string]: any }) => {
|
|
return {
|
|
...data,
|
|
task,
|
|
};
|
|
};
|
|
if (!task) {
|
|
return {
|
|
code: 500,
|
|
message: `没有找到 ${name} 这个任务`,
|
|
};
|
|
}
|
|
let { command, before, after, afterCheck, beforeCheck, type } = task;
|
|
if (type === 'npm-install' && !afterCheck) {
|
|
afterCheck = 'added';
|
|
}
|
|
if (before) {
|
|
const res = this.runCommand(before);
|
|
console.log('before', res, beforeCheck, this.checkForContainue(res, beforeCheck));
|
|
if (!this.checkForContainue(res, beforeCheck)) {
|
|
return end({
|
|
code: 200,
|
|
message: `当前任务不需要执行, ${command}`,
|
|
});
|
|
}
|
|
}
|
|
const res = this.runCommand(command);
|
|
console.log('runCommand', res);
|
|
if (res.code !== 200) {
|
|
return end(res);
|
|
}
|
|
let checkText = res.data || '';
|
|
if (after) {
|
|
const res = this.runCommand(after);
|
|
if (res.code !== 200) {
|
|
return end(res);
|
|
}
|
|
checkText = res.data || '';
|
|
}
|
|
if (afterCheck) {
|
|
const isSuccess = checkText?.includes?.(afterCheck);
|
|
return end({
|
|
code: isSuccess ? 200 : 500,
|
|
output: res.data,
|
|
check: afterCheck,
|
|
message: isSuccess ? `当前任务执行成功, ${command}` : `当前任务执行失败, ${command}`,
|
|
});
|
|
}
|
|
return end(res);
|
|
}
|
|
/**
|
|
* 检测是否需要继续执行。
|
|
* 1. res.code === 500 代表执行失败, 需要继续执行
|
|
* 2. res.code === 200 代表执行成功，但是需要检测输出内容
|
|
* 2.1 如果有 check 的内容，代表不需要继续执行
|
|
* 2.2 如果没有 check 的内容，代表需要继续执行
|
|
* @param res
|
|
* @param check
|
|
* @returns
|
|
*/
|
|
private checkForContainue(res: { data: string; code: number; [key: string]: any }, check?: string, isBefore = true) {
|
|
if (res.code !== 200) {
|
|
return true;
|
|
}
|
|
if (!check) {
|
|
return true;
|
|
}
|
|
const hasIncludes = res.data?.includes?.(check);
|
|
if (isBefore) {
|
|
// 代表已经安装, 不需要继续执行
|
|
return !hasIncludes;
|
|
}
|
|
return hasIncludes;
|
|
}
|
|
runCommand(command: string) {
|
|
try {
|
|
const res = execSync(command, { encoding: 'utf-8' });
|
|
return {
|
|
code: 200,
|
|
data: res.toString(),
|
|
};
|
|
} catch (error) {
|
|
return {
|
|
code: 500,
|
|
data: '',
|
|
message: error.toString(),
|
|
};
|
|
}
|
|
}
|
|
}
|