"feat: 增加助手应用配置管理功能与服务器守护进程支持"

assistant/task-command/mod.ts

@@ -0,0 +1,142 @@
+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(),
+      };
+    }
+  }
+}