@undefineds.co/xpod 0.1.0-local.202602081751

package/dist/supervisor/Supervisor.js

@@ -0,0 +1,174 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Supervisor = void 0;
+const node_child_process_1 = require("node:child_process");
+const tree_kill_1 = __importDefault(require("tree-kill"));
+const MAX_RESTARTS = 5;
+class Supervisor {
+    constructor() {
+        this.processes = new Map();
+        this.states = new Map();
+        this.configs = new Map();
+        this.isShuttingDown = false;
+        // 确保父进程退出时清理所有子进程
+        process.on('exit', () => this.killAll());
+        process.on('SIGINT', () => this.shutdown('SIGINT'));
+        process.on('SIGTERM', () => this.shutdown('SIGTERM'));
+    }
+    async shutdown(signal) {
+        if (this.isShuttingDown)
+            return;
+        this.isShuttingDown = true;
+        console.log(`[Supervisor] Received ${signal}, stopping all services...`);
+        await this.stopAll();
+        process.exit(0);
+    }
+    killAll() {
+        // 同步杀掉所有子进程（用于 process.on('exit')）
+        for (const [name, child] of this.processes) {
+            if (child.pid) {
+                try {
+                    process.kill(child.pid, 'SIGKILL');
+                }
+                catch {
+                    // 进程可能已经退出
+                }
+            }
+        }
+    }
+    setStatusChangeHandler(handler) {
+        this.onStatusChange = handler;
+    }
+    register(config) {
+        this.configs.set(config.name, config);
+        this.states.set(config.name, {
+            name: config.name,
+            status: 'stopped',
+            restartCount: 0,
+        });
+    }
+    async startAll() {
+        for (const name of this.configs.keys()) {
+            this.start(name);
+        }
+    }
+    async stopAll() {
+        this.isShuttingDown = true;
+        const promises = [];
+        for (const name of this.processes.keys()) {
+            promises.push(this.stop(name));
+        }
+        await Promise.all(promises);
+    }
+    start(name) {
+        const config = this.configs.get(name);
+        const state = this.states.get(name);
+        if (!config || !state)
+            return;
+        if (state.status === 'running' || state.status === 'starting')
+            return;
+        console.log(`[Supervisor] Starting ${name}...`);
+        this.updateState(name, { status: 'starting', startTime: Date.now() });
+        const env = { ...process.env, ...config.env };
+        const child = (0, node_child_process_1.spawn)(config.command, config.args, {
+            stdio: ['ignore', 'pipe', 'pipe'],
+            env,
+            cwd: config.cwd || process.cwd(),
+            detached: false, // 确保子进程不脱离父进程
+        });
+        this.processes.set(name, child);
+        this.updateState(name, { status: 'running', pid: child.pid });
+        const prefixLog = (name, data, isError = false) => {
+            const output = data.toString();
+            const lines = output.split('\n');
+            for (const line of lines) {
+                const trimmed = line.trim();
+                if (trimmed) {
+                    if (isError) {
+                        console.error(`[${name}] ${trimmed}`);
+                    }
+                    else {
+                        console.log(`[${name}] ${trimmed}`);
+                    }
+                }
+            }
+        };
+        child.stdout?.on('data', (data) => {
+            prefixLog(name, data, false);
+        });
+        child.stderr?.on('data', (data) => {
+            prefixLog(name, data, true);
+        });
+        child.on('error', (err) => {
+            console.error(`[Supervisor] Error spawning ${name}:`, err);
+            this.updateState(name, { status: 'crashed' });
+        });
+        child.on('exit', (code, signal) => {
+            console.log(`[Supervisor] ${name} exited with code ${code} signal ${signal}`);
+            const currentState = this.states.get(name);
+            const wasManualStop = currentState?.status === 'stopped';
+            this.updateState(name, {
+                status: 'stopped',
+                lastExitCode: code ?? undefined,
+                pid: undefined,
+            });
+            this.processes.delete(name);
+            // Auto-restart on crash (not on manual stop or shutdown)
+            if (code !== 0 && !wasManualStop && !this.isShuttingDown) {
+                const newState = this.states.get(name);
+                const restartCount = (newState?.restartCount || 0) + 1;
+                if (restartCount <= MAX_RESTARTS) {
+                    this.updateState(name, { restartCount });
+                    console.log(`[Supervisor] Restarting ${name} in 2s... (attempt ${restartCount}/${MAX_RESTARTS})`);
+                    setTimeout(() => this.start(name), 2000);
+                }
+                else {
+                    console.error(`[Supervisor] ${name} exceeded max restarts (${MAX_RESTARTS}), giving up`);
+                }
+            }
+        });
+    }
+    stop(name) {
+        return new Promise((resolve) => {
+            const child = this.processes.get(name);
+            if (!child || !child.pid) {
+                resolve();
+                return;
+            }
+            // Mark as stopped first to prevent auto-restart
+            this.updateState(name, { status: 'stopped' });
+            (0, tree_kill_1.default)(child.pid, 'SIGTERM', (err) => {
+                if (err)
+                    console.error(`[Supervisor] Failed to kill ${name}:`, err);
+                resolve();
+            });
+        });
+    }
+    getStatus(name) {
+        const state = this.states.get(name);
+        if (state && state.status === 'running' && state.startTime) {
+            return { ...state, uptime: Date.now() - state.startTime };
+        }
+        return state;
+    }
+    getAllStatus() {
+        return Array.from(this.states.values()).map((s) => {
+            if (s.status === 'running' && s.startTime) {
+                return { ...s, uptime: Date.now() - s.startTime };
+            }
+            return s;
+        });
+    }
+    updateState(name, update) {
+        const state = this.states.get(name);
+        if (state) {
+            Object.assign(state, update);
+            this.onStatusChange?.(name, { ...state });
+        }
+    }
+}
+exports.Supervisor = Supervisor;
+//# sourceMappingURL=Supervisor.js.map

package/dist/supervisor/Supervisor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Supervisor.js","sourceRoot":"","sources":["../../src/supervisor/Supervisor.ts"],"names":[],"mappings":";;;;;;AAAA,2DAA8D;AAC9D,0DAA6B;AAG7B,MAAM,YAAY,GAAG,CAAC,CAAC;AAEvB,MAAa,UAAU;IAOrB;QANQ,cAAS,GAA8B,IAAI,GAAG,EAAE,CAAC;QACjD,WAAM,GAA8B,IAAI,GAAG,EAAE,CAAC;QAC9C,YAAO,GAA+B,IAAI,GAAG,EAAE,CAAC;QAEhD,mBAAc,GAAG,KAAK,CAAC;QAG7B,kBAAkB;QAClB,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QACzC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QACpD,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IACxD,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,MAAc;QACnC,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO;QAChC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,OAAO,CAAC,GAAG,CAAC,yBAAyB,MAAM,4BAA4B,CAAC,CAAC;QACzE,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAEO,OAAO;QACb,mCAAmC;QACnC,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;gBACd,IAAI,CAAC;oBACH,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBACrC,CAAC;gBAAC,MAAM,CAAC;oBACP,WAAW;gBACb,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAEM,sBAAsB,CAAC,OAA4B;QACxD,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;IAChC,CAAC;IAEM,QAAQ,CAAC,MAAqB;QACnC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACtC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE;YAC3B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,MAAM,EAAE,SAAS;YACjB,YAAY,EAAE,CAAC;SAChB,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,QAAQ;QACnB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;YACvC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,OAAO;QAClB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,MAAM,QAAQ,GAAoB,EAAE,CAAC;QACrC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,CAAC;YACzC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACjC,CAAC;QACD,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC;IAEM,KAAK,CAAC,IAAY;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK;YAAE,OAAO;QAE9B,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,KAAK,UAAU;YAAE,OAAO;QAEtE,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,KAAK,CAAC,CAAC;QAChD,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAEtE,MAAM,GAAG,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,GAAG,EAAE,CAAC;QAE9C,MAAM,KAAK,GAAG,IAAA,0BAAK,EAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,EAAE;YAC/C,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;YACjC,GAAG;YACH,GAAG,EAAE,MAAM,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,EAAE;YAChC,QAAQ,EAAE,KAAK,EAAE,cAAc;SAChC,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;QAE9D,MAAM,SAAS,GAAG,CAAC,IAAY,EAAE,IAAY,EAAE,OAAO,GAAG,KAAK,EAAE,EAAE;YAChE,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC/B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC5B,IAAI,OAAO,EAAE,CAAC;oBACZ,IAAI,OAAO,EAAE,CAAC;wBACZ,OAAO,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC,CAAC;oBACxC,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC,CAAC;oBACtC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAChC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAChC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACxB,OAAO,CAAC,KAAK,CAAC,+BAA+B,IAAI,GAAG,EAAE,GAAG,CAAC,CAAC;YAC3D,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YAChC,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,qBAAqB,IAAI,WAAW,MAAM,EAAE,CAAC,CAAC;YAC9E,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,aAAa,GAAG,YAAY,EAAE,MAAM,KAAK,SAAS,CAAC;YAEzD,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE;gBACrB,MAAM,EAAE,SAAS;gBACjB,YAAY,EAAE,IAAI,IAAI,SAAS;gBAC/B,GAAG,EAAE,SAAS;aACf,CAAC,CAAC;YACH,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAE5B,yDAAyD;YACzD,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;gBACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBACvC,MAAM,YAAY,GAAG,CAAC,QAAQ,EAAE,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;gBAEvD,IAAI,YAAY,IAAI,YAAY,EAAE,CAAC;oBACjC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;oBACzC,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,sBAAsB,YAAY,IAAI,YAAY,GAAG,CAAC,CAAC;oBAClG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;gBAC3C,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,KAAK,CAAC,gBAAgB,IAAI,2BAA2B,YAAY,cAAc,CAAC,CAAC;gBAC3F,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,IAAI,CAAC,IAAY;QACtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACvC,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;gBACzB,OAAO,EAAE,CAAC;gBACV,OAAO;YACT,CAAC;YAED,gDAAgD;YAChD,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;YAE9C,IAAA,mBAAI,EAAC,KAAK,CAAC,GAAG,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE;gBACjC,IAAI,GAAG;oBAAE,OAAO,CAAC,KAAK,CAAC,+BAA+B,IAAI,GAAG,EAAE,GAAG,CAAC,CAAC;gBACpE,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,SAAS,CAAC,IAAY;QAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YAC3D,OAAO,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;QAC5D,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEM,YAAY;QACjB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YAChD,IAAI,CAAC,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,CAAC,SAAS,EAAE,CAAC;gBAC1C,OAAO,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,SAAS,EAAE,CAAC;YACpD,CAAC;YACD,OAAO,CAAC,CAAC;QACX,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,WAAW,CAAC,IAAY,EAAE,MAA6B;QAC7D,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC7B,IAAI,CAAC,cAAc,EAAE,CAAC,IAAI,EAAE,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;CACF;AAvLD,gCAuLC","sourcesContent":["import { spawn, type ChildProcess } from 'node:child_process';\nimport kill from 'tree-kill';\nimport type { ServiceConfig, ServiceState, StatusChangeHandler } from './types';\n\nconst MAX_RESTARTS = 5;\n\nexport class Supervisor {\n  private processes: Map<string, ChildProcess> = new Map();\n  private states: Map<string, ServiceState> = new Map();\n  private configs: Map<string, ServiceConfig> = new Map();\n  private onStatusChange?: StatusChangeHandler;\n  private isShuttingDown = false;\n\n  constructor() {\n    // 确保父进程退出时清理所有子进程\n    process.on('exit', () => this.killAll());\n    process.on('SIGINT', () => this.shutdown('SIGINT'));\n    process.on('SIGTERM', () => this.shutdown('SIGTERM'));\n  }\n\n  private async shutdown(signal: string): Promise<void> {\n    if (this.isShuttingDown) return;\n    this.isShuttingDown = true;\n    console.log(`[Supervisor] Received ${signal}, stopping all services...`);\n    await this.stopAll();\n    process.exit(0);\n  }\n\n  private killAll(): void {\n    // 同步杀掉所有子进程（用于 process.on('exit')）\n    for (const [name, child] of this.processes) {\n      if (child.pid) {\n        try {\n          process.kill(child.pid, 'SIGKILL');\n        } catch {\n          // 进程可能已经退出\n        }\n      }\n    }\n  }\n\n  public setStatusChangeHandler(handler: StatusChangeHandler): void {\n    this.onStatusChange = handler;\n  }\n\n  public register(config: ServiceConfig): void {\n    this.configs.set(config.name, config);\n    this.states.set(config.name, {\n      name: config.name,\n      status: 'stopped',\n      restartCount: 0,\n    });\n  }\n\n  public async startAll(): Promise<void> {\n    for (const name of this.configs.keys()) {\n      this.start(name);\n    }\n  }\n\n  public async stopAll(): Promise<void> {\n    this.isShuttingDown = true;\n    const promises: Promise<void>[] = [];\n    for (const name of this.processes.keys()) {\n      promises.push(this.stop(name));\n    }\n    await Promise.all(promises);\n  }\n\n  public start(name: string): void {\n    const config = this.configs.get(name);\n    const state = this.states.get(name);\n    if (!config || !state) return;\n\n    if (state.status === 'running' || state.status === 'starting') return;\n\n    console.log(`[Supervisor] Starting ${name}...`);\n    this.updateState(name, { status: 'starting', startTime: Date.now() });\n\n    const env = { ...process.env, ...config.env };\n\n    const child = spawn(config.command, config.args, {\n      stdio: ['ignore', 'pipe', 'pipe'],\n      env,\n      cwd: config.cwd || process.cwd(),\n      detached: false, // 确保子进程不脱离父进程\n    });\n\n    this.processes.set(name, child);\n    this.updateState(name, { status: 'running', pid: child.pid });\n\n    const prefixLog = (name: string, data: Buffer, isError = false) => {\n      const output = data.toString();\n      const lines = output.split('\\n');\n      for (const line of lines) {\n        const trimmed = line.trim();\n        if (trimmed) {\n          if (isError) {\n            console.error(`[${name}] ${trimmed}`);\n          } else {\n            console.log(`[${name}] ${trimmed}`);\n          }\n        }\n      }\n    };\n\n    child.stdout?.on('data', (data) => {\n      prefixLog(name, data, false);\n    });\n\n    child.stderr?.on('data', (data) => {\n      prefixLog(name, data, true);\n    });\n\n    child.on('error', (err) => {\n      console.error(`[Supervisor] Error spawning ${name}:`, err);\n      this.updateState(name, { status: 'crashed' });\n    });\n\n    child.on('exit', (code, signal) => {\n      console.log(`[Supervisor] ${name} exited with code ${code} signal ${signal}`);\n      const currentState = this.states.get(name);\n      const wasManualStop = currentState?.status === 'stopped';\n\n      this.updateState(name, {\n        status: 'stopped',\n        lastExitCode: code ?? undefined,\n        pid: undefined,\n      });\n      this.processes.delete(name);\n\n      // Auto-restart on crash (not on manual stop or shutdown)\n      if (code !== 0 && !wasManualStop && !this.isShuttingDown) {\n        const newState = this.states.get(name);\n        const restartCount = (newState?.restartCount || 0) + 1;\n\n        if (restartCount <= MAX_RESTARTS) {\n          this.updateState(name, { restartCount });\n          console.log(`[Supervisor] Restarting ${name} in 2s... (attempt ${restartCount}/${MAX_RESTARTS})`);\n          setTimeout(() => this.start(name), 2000);\n        } else {\n          console.error(`[Supervisor] ${name} exceeded max restarts (${MAX_RESTARTS}), giving up`);\n        }\n      }\n    });\n  }\n\n  public stop(name: string): Promise<void> {\n    return new Promise((resolve) => {\n      const child = this.processes.get(name);\n      if (!child || !child.pid) {\n        resolve();\n        return;\n      }\n\n      // Mark as stopped first to prevent auto-restart\n      this.updateState(name, { status: 'stopped' });\n\n      kill(child.pid, 'SIGTERM', (err) => {\n        if (err) console.error(`[Supervisor] Failed to kill ${name}:`, err);\n        resolve();\n      });\n    });\n  }\n\n  public getStatus(name: string): ServiceState | undefined {\n    const state = this.states.get(name);\n    if (state && state.status === 'running' && state.startTime) {\n      return { ...state, uptime: Date.now() - state.startTime };\n    }\n    return state;\n  }\n\n  public getAllStatus(): ServiceState[] {\n    return Array.from(this.states.values()).map((s) => {\n      if (s.status === 'running' && s.startTime) {\n        return { ...s, uptime: Date.now() - s.startTime };\n      }\n      return s;\n    });\n  }\n\n  private updateState(name: string, update: Partial<ServiceState>): void {\n    const state = this.states.get(name);\n    if (state) {\n      Object.assign(state, update);\n      this.onStatusChange?.(name, { ...state });\n    }\n  }\n}\n"]}

package/dist/supervisor/index.d.ts

@@ -0,0 +1,2 @@
+export { Supervisor } from './Supervisor';
+export type { ServiceConfig, ServiceState, ServiceStatus, StatusChangeHandler } from './types';

package/dist/supervisor/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Supervisor = void 0;
+var Supervisor_1 = require("./Supervisor");
+Object.defineProperty(exports, "Supervisor", { enumerable: true, get: function () { return Supervisor_1.Supervisor; } });
+//# sourceMappingURL=index.js.map

package/dist/supervisor/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/supervisor/index.ts"],"names":[],"mappings":";;;AAAA,2CAA0C;AAAjC,wGAAA,UAAU,OAAA","sourcesContent":["export { Supervisor } from './Supervisor';\nexport type { ServiceConfig, ServiceState, ServiceStatus, StatusChangeHandler } from './types';\n"]}

package/dist/supervisor/types.d.ts

@@ -0,0 +1,19 @@
+export interface ServiceConfig {
+    name: string;
+    command: string;
+    args: string[];
+    env?: Record<string, string>;
+    cwd?: string;
+    readyUrl?: string;
+}
+export type ServiceStatus = 'stopped' | 'starting' | 'running' | 'crashed';
+export interface ServiceState {
+    name: string;
+    status: ServiceStatus;
+    pid?: number;
+    startTime?: number;
+    uptime?: number;
+    lastExitCode?: number;
+    restartCount: number;
+}
+export type StatusChangeHandler = (name: string, state: ServiceState) => void;

package/dist/supervisor/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/supervisor/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/supervisor/types.ts"],"names":[],"mappings":"","sourcesContent":["export interface ServiceConfig {\n  name: string;\n  command: string;\n  args: string[];\n  env?: Record<string, string>;\n  cwd?: string;\n  readyUrl?: string;\n}\n\nexport type ServiceStatus = 'stopped' | 'starting' | 'running' | 'crashed';\n\nexport interface ServiceState {\n  name: string;\n  status: ServiceStatus;\n  pid?: number;\n  startTime?: number;\n  uptime?: number;\n  lastExitCode?: number;\n  restartCount: number;\n}\n\nexport type StatusChangeHandler = (name: string, state: ServiceState) => void;\n"]}

package/dist/task/DrizzleTaskQueue.d.ts

@@ -0,0 +1,60 @@
+/**
+ * DrizzleTaskQueue - 基于 drizzle-solid 的任务队列
+ *
+ * 简化模型：消息 + Agent
+ *
+ * 任务存储在 Pod 的 /tasks/{YYYY-MM-DD}.ttl 中，按天分片
+ */
+import type { SolidDatabase } from '@undefineds.co/drizzle-solid';
+import { Task as taskTable } from './schema';
+import type { Task, TaskStatus, CreateTaskInput, TaskQueue, TaskQueueStats, ListTasksOptions } from './types';
+export interface DrizzleTaskQueueOptions {
+    /**
+     * Pod base URL
+     */
+    podBaseUrl: string;
+    /**
+     * drizzle-solid 数据库实例
+     */
+    db: SolidDatabase<{
+        task: typeof taskTable;
+    }>;
+}
+/**
+ * 基于 drizzle-solid 的任务队列实现
+ */
+export declare class DrizzleTaskQueue implements TaskQueue {
+    protected readonly logger: import("global-logger-factory").Logger<unknown>;
+    private readonly podBaseUrl;
+    private readonly db;
+    constructor(options: DrizzleTaskQueueOptions);
+    /**
+     * 创建任务
+     */
+    createTask(input: CreateTaskInput): Promise<Task>;
+    /**
+     * 获取任务
+     */
+    getTask(taskId: string): Promise<Task | null>;
+    /**
+     * 获取任务列表
+     */
+    listTasks(options?: ListTasksOptions): Promise<Task[]>;
+    /**
+     * 更新任务状态
+     */
+    updateTaskStatus(taskId: string, status: TaskStatus, updates?: Partial<Task>): Promise<void>;
+    /**
+     * 删除任务
+     */
+    deleteTask(taskId: string): Promise<void>;
+    /**
+     * 获取队列统计
+     */
+    getStats(): Promise<TaskQueueStats>;
+    private generateTaskId;
+    /**
+     * 将数据库记录转换为 Task 接口
+     */
+    private dbTaskToTask;
+}

package/dist/task/DrizzleTaskQueue.js

@@ -0,0 +1,171 @@
+"use strict";
+/**
+ * DrizzleTaskQueue - 基于 drizzle-solid 的任务队列
+ *
+ * 简化模型：消息 + Agent
+ *
+ * 任务存储在 Pod 的 /tasks/{YYYY-MM-DD}.ttl 中，按天分片
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DrizzleTaskQueue = void 0;
+const global_logger_factory_1 = require("global-logger-factory");
+const crypto_1 = require("crypto");
+const drizzle_solid_1 = require("@undefineds.co/drizzle-solid");
+const schema_1 = require("./schema");
+/**
+ * 基于 drizzle-solid 的任务队列实现
+ */
+class DrizzleTaskQueue {
+    constructor(options) {
+        this.logger = (0, global_logger_factory_1.getLoggerFor)(this);
+        this.podBaseUrl = options.podBaseUrl.endsWith('/') ? options.podBaseUrl : `${options.podBaseUrl}/`;
+        this.db = options.db;
+    }
+    /**
+     * 创建任务
+     */
+    async createTask(input) {
+        const id = this.generateTaskId();
+        const now = new Date();
+        const taskData = {
+            id,
+            agent: input.agent,
+            message: input.message,
+            status: schema_1.TaskStatus.PENDING,
+            createdAt: now,
+        };
+        await this.db.insert(schema_1.Task).values(taskData);
+        this.logger.info(`Created task ${id} for agent "${input.agent}"`);
+        return {
+            id,
+            agent: input.agent,
+            message: input.message,
+            status: 'pending',
+            createdAt: now,
+        };
+    }
+    /**
+     * 获取任务
+     */
+    async getTask(taskId) {
+        const tasks = await this.db.select().from(schema_1.Task).where((0, drizzle_solid_1.eq)(schema_1.Task.id, taskId));
+        if (tasks.length === 0) {
+            return null;
+        }
+        return this.dbTaskToTask(tasks[0]);
+    }
+    /**
+     * 获取任务列表
+     */
+    async listTasks(options) {
+        let query = this.db.select().from(schema_1.Task);
+        // 状态过滤
+        if (options?.status) {
+            query = query.where((0, drizzle_solid_1.eq)(schema_1.Task.status, options.status));
+        }
+        const tasks = await query;
+        // 转换为 Task 接口
+        let result = tasks.map((t) => this.dbTaskToTask(t));
+        // Agent 过滤
+        if (options?.agent) {
+            result = result.filter((t) => t.agent === options.agent);
+        }
+        // 排序
+        if (options?.orderBy) {
+            const order = options.order === 'asc' ? 1 : -1;
+            result.sort((a, b) => {
+                const aVal = a[options.orderBy];
+                const bVal = b[options.orderBy];
+                if (aVal instanceof Date && bVal instanceof Date) {
+                    return (aVal.getTime() - bVal.getTime()) * order;
+                }
+                return 0;
+            });
+        }
+        // 分页
+        const offset = options?.offset ?? 0;
+        const limit = options?.limit ?? result.length;
+        return result.slice(offset, offset + limit);
+    }
+    /**
+     * 更新任务状态
+     */
+    async updateTaskStatus(taskId, status, updates) {
+        const now = new Date();
+        const updateData = {
+            status,
+        };
+        // 根据状态设置时间字段
+        if (status === 'running') {
+            updateData.startedAt = now;
+        }
+        else if (status === 'completed' || status === 'failed') {
+            updateData.completedAt = now;
+        }
+        // 合并其他更新
+        if (updates) {
+            if (updates.result !== undefined)
+                updateData.result = updates.result;
+            if (updates.error !== undefined)
+                updateData.error = updates.error;
+        }
+        await this.db.update(schema_1.Task).set(updateData).where((0, drizzle_solid_1.eq)(schema_1.Task.id, taskId));
+        this.logger.info(`Task ${taskId} status changed to ${status}`);
+    }
+    /**
+     * 删除任务
+     */
+    async deleteTask(taskId) {
+        await this.db.delete(schema_1.Task).where((0, drizzle_solid_1.eq)(schema_1.Task.id, taskId));
+        this.logger.info(`Deleted task ${taskId}`);
+    }
+    /**
+     * 获取队列统计
+     */
+    async getStats() {
+        const tasks = await this.db.select().from(schema_1.Task);
+        const stats = {
+            pending: 0,
+            running: 0,
+            completed: 0,
+            failed: 0,
+            total: tasks.length,
+            byAgent: {},
+        };
+        for (const task of tasks) {
+            const taskRecord = task;
+            const status = taskRecord.status ?? 'pending';
+            stats[status]++;
+            // 按 Agent 统计
+            const agent = taskRecord.agent ?? 'unknown';
+            stats.byAgent[agent] = (stats.byAgent[agent] ?? 0) + 1;
+        }
+        return stats;
+    }
+    // ============================================
+    // Private Methods
+    // ============================================
+    generateTaskId() {
+        const timestamp = Date.now().toString(36);
+        const random = (0, crypto_1.randomBytes)(4).toString('hex');
+        return `task-${timestamp}-${random}`;
+    }
+    /**
+     * 将数据库记录转换为 Task 接口
+     */
+    dbTaskToTask(dbTask) {
+        return {
+            id: dbTask.id,
+            agent: dbTask.agent,
+            message: dbTask.message,
+            status: dbTask.status ?? 'pending',
+            createdAt: dbTask.createdAt,
+            startedAt: dbTask.startedAt,
+            completedAt: dbTask.completedAt,
+            result: dbTask.result,
+            error: dbTask.error,
+        };
+    }
+}
+exports.DrizzleTaskQueue = DrizzleTaskQueue;
+//# sourceMappingURL=DrizzleTaskQueue.js.map

package/dist/task/DrizzleTaskQueue.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"DrizzleTaskQueue.js","sourceRoot":"","sources":["../../src/task/DrizzleTaskQueue.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;AAEH,iEAAqD;AACrD,mCAAqC;AACrC,gEAAkD;AAGlD,qCAA4E;AAqB5E;;GAEG;AACH,MAAa,gBAAgB;IAK3B,YAAmB,OAAgC;QAJhC,WAAM,GAAG,IAAA,oCAAY,EAAC,IAAI,CAAC,CAAC;QAK7C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,UAAU,GAAG,CAAC;QACnG,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,UAAU,CAAC,KAAsB;QAC5C,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QACjC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QAEvB,MAAM,QAAQ,GAAG;YACf,EAAE;YACF,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAE,mBAAe,CAAC,OAAO;YAC/B,SAAS,EAAE,GAAG;SACf,CAAC;QAEF,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAS,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAEjD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,eAAe,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;QAElE,OAAO;YACL,EAAE;YACF,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,GAAG;SACf,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,OAAO,CAAC,MAAc;QACjC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAS,CAAC,CAAC,KAAK,CAAC,IAAA,kBAAE,EAAC,aAAS,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAErF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,SAAS,CAAC,OAA0B;QAC/C,IAAI,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAS,CAAC,CAAC;QAE7C,OAAO;QACP,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;YACpB,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAA,kBAAE,EAAC,aAAS,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC;QAE1B,cAAc;QACd,IAAI,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAA0B,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAE7E,WAAW;QACX,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;YACnB,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAO,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK,CAAC,CAAC;QACjE,CAAC;QAED,KAAK;QACL,IAAI,OAAO,EAAE,OAAO,EAAE,CAAC;YACrB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAO,EAAE,CAAO,EAAE,EAAE;gBAC/B,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,OAAQ,CAAC,CAAC;gBACjC,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,OAAQ,CAAC,CAAC;gBACjC,IAAI,IAAI,YAAY,IAAI,IAAI,IAAI,YAAY,IAAI,EAAE,CAAC;oBACjD,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,KAAK,CAAC;gBACnD,CAAC;gBACD,OAAO,CAAC,CAAC;YACX,CAAC,CAAC,CAAC;QACL,CAAC;QAED,KAAK;QACL,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC;QACpC,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC;QAC9C,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,KAAK,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,gBAAgB,CAC3B,MAAc,EACd,MAAkB,EAClB,OAAuB;QAEvB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QAEvB,MAAM,UAAU,GAA4B;YAC1C,MAAM;SACP,CAAC;QAEF,aAAa;QACb,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,UAAU,CAAC,SAAS,GAAG,GAAG,CAAC;QAC7B,CAAC;aAAM,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACzD,UAAU,CAAC,WAAW,GAAG,GAAG,CAAC;QAC/B,CAAC;QAED,SAAS;QACT,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS;gBAAE,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YACrE,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS;gBAAE,UAAU,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QACpE,CAAC;QAED,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAS,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,IAAA,kBAAE,EAAC,aAAS,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAEhF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,MAAM,sBAAsB,MAAM,EAAE,CAAC,CAAC;IACjE,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,UAAU,CAAC,MAAc;QACpC,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAS,CAAC,CAAC,KAAK,CAAC,IAAA,kBAAE,EAAC,aAAS,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,MAAM,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,QAAQ;QACnB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAS,CAAC,CAAC;QAErD,MAAM,KAAK,GAAmB;YAC5B,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;YACV,SAAS,EAAE,CAAC;YACZ,MAAM,EAAE,CAAC;YACT,KAAK,EAAE,KAAK,CAAC,MAAM;YACnB,OAAO,EAAE,EAAE;SACZ,CAAC;QAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,UAAU,GAAG,IAA+B,CAAC;YACnD,MAAM,MAAM,GAAI,UAAU,CAAC,MAAqB,IAAI,SAAS,CAAC;YAC9D,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;YAEhB,aAAa;YACb,MAAM,KAAK,GAAG,UAAU,CAAC,KAAe,IAAI,SAAS,CAAC;YACtD,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACzD,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,+CAA+C;IAC/C,kBAAkB;IAClB,+CAA+C;IAEvC,cAAc;QACpB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAA,oBAAW,EAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC9C,OAAO,QAAQ,SAAS,IAAI,MAAM,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,MAA+B;QAClD,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,EAAY;YACvB,KAAK,EAAE,MAAM,CAAC,KAAe;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAiB;YACjC,MAAM,EAAG,MAAM,CAAC,MAAqB,IAAI,SAAS;YAClD,SAAS,EAAE,MAAM,CAAC,SAAiB;YACnC,SAAS,EAAE,MAAM,CAAC,SAA6B;YAC/C,WAAW,EAAE,MAAM,CAAC,WAA+B;YACnD,MAAM,EAAE,MAAM,CAAC,MAAiB;YAChC,KAAK,EAAE,MAAM,CAAC,KAA2B;SAC1C,CAAC;IACJ,CAAC;CACF;AAzLD,4CAyLC","sourcesContent":["/**\n * DrizzleTaskQueue - 基于 drizzle-solid 的任务队列\n *\n * 简化模型：消息 + Agent\n *\n * 任务存储在 Pod 的 /tasks/{YYYY-MM-DD}.ttl 中，按天分片\n */\n\nimport { getLoggerFor } from 'global-logger-factory';\nimport { randomBytes } from 'crypto';\nimport { eq } from '@undefineds.co/drizzle-solid';\nimport type { SolidDatabase } from '@undefineds.co/drizzle-solid';\n\nimport { Task as taskTable, TaskStatus as TaskStatusConst } from './schema';\nimport type {\n  Task,\n  TaskStatus,\n  CreateTaskInput,\n  TaskQueue,\n  TaskQueueStats,\n  ListTasksOptions,\n} from './types';\n\nexport interface DrizzleTaskQueueOptions {\n  /**\n   * Pod base URL\n   */\n  podBaseUrl: string;\n  /**\n   * drizzle-solid 数据库实例\n   */\n  db: SolidDatabase<{ task: typeof taskTable }>;\n}\n\n/**\n * 基于 drizzle-solid 的任务队列实现\n */\nexport class DrizzleTaskQueue implements TaskQueue {\n  protected readonly logger = getLoggerFor(this);\n  private readonly podBaseUrl: string;\n  private readonly db: SolidDatabase<{ task: typeof taskTable }>;\n\n  public constructor(options: DrizzleTaskQueueOptions) {\n    this.podBaseUrl = options.podBaseUrl.endsWith('/') ? options.podBaseUrl : `${options.podBaseUrl}/`;\n    this.db = options.db;\n  }\n\n  /**\n   * 创建任务\n   */\n  public async createTask(input: CreateTaskInput): Promise<Task> {\n    const id = this.generateTaskId();\n    const now = new Date();\n\n    const taskData = {\n      id,\n      agent: input.agent,\n      message: input.message,\n      status: TaskStatusConst.PENDING,\n      createdAt: now,\n    };\n\n    await this.db.insert(taskTable).values(taskData);\n\n    this.logger.info(`Created task ${id} for agent \"${input.agent}\"`);\n\n    return {\n      id,\n      agent: input.agent,\n      message: input.message,\n      status: 'pending',\n      createdAt: now,\n    };\n  }\n\n  /**\n   * 获取任务\n   */\n  public async getTask(taskId: string): Promise<Task | null> {\n    const tasks = await this.db.select().from(taskTable).where(eq(taskTable.id, taskId));\n\n    if (tasks.length === 0) {\n      return null;\n    }\n\n    return this.dbTaskToTask(tasks[0]);\n  }\n\n  /**\n   * 获取任务列表\n   */\n  public async listTasks(options?: ListTasksOptions): Promise<Task[]> {\n    let query = this.db.select().from(taskTable);\n\n    // 状态过滤\n    if (options?.status) {\n      query = query.where(eq(taskTable.status, options.status));\n    }\n\n    const tasks = await query;\n\n    // 转换为 Task 接口\n    let result = tasks.map((t: Record<string, unknown>) => this.dbTaskToTask(t));\n\n    // Agent 过滤\n    if (options?.agent) {\n      result = result.filter((t: Task) => t.agent === options.agent);\n    }\n\n    // 排序\n    if (options?.orderBy) {\n      const order = options.order === 'asc' ? 1 : -1;\n      result.sort((a: Task, b: Task) => {\n        const aVal = a[options.orderBy!];\n        const bVal = b[options.orderBy!];\n        if (aVal instanceof Date && bVal instanceof Date) {\n          return (aVal.getTime() - bVal.getTime()) * order;\n        }\n        return 0;\n      });\n    }\n\n    // 分页\n    const offset = options?.offset ?? 0;\n    const limit = options?.limit ?? result.length;\n    return result.slice(offset, offset + limit);\n  }\n\n  /**\n   * 更新任务状态\n   */\n  public async updateTaskStatus(\n    taskId: string,\n    status: TaskStatus,\n    updates?: Partial<Task>,\n  ): Promise<void> {\n    const now = new Date();\n\n    const updateData: Record<string, unknown> = {\n      status,\n    };\n\n    // 根据状态设置时间字段\n    if (status === 'running') {\n      updateData.startedAt = now;\n    } else if (status === 'completed' || status === 'failed') {\n      updateData.completedAt = now;\n    }\n\n    // 合并其他更新\n    if (updates) {\n      if (updates.result !== undefined) updateData.result = updates.result;\n      if (updates.error !== undefined) updateData.error = updates.error;\n    }\n\n    await this.db.update(taskTable).set(updateData).where(eq(taskTable.id, taskId));\n\n    this.logger.info(`Task ${taskId} status changed to ${status}`);\n  }\n\n  /**\n   * 删除任务\n   */\n  public async deleteTask(taskId: string): Promise<void> {\n    await this.db.delete(taskTable).where(eq(taskTable.id, taskId));\n    this.logger.info(`Deleted task ${taskId}`);\n  }\n\n  /**\n   * 获取队列统计\n   */\n  public async getStats(): Promise<TaskQueueStats> {\n    const tasks = await this.db.select().from(taskTable);\n\n    const stats: TaskQueueStats = {\n      pending: 0,\n      running: 0,\n      completed: 0,\n      failed: 0,\n      total: tasks.length,\n      byAgent: {},\n    };\n\n    for (const task of tasks) {\n      const taskRecord = task as Record<string, unknown>;\n      const status = (taskRecord.status as TaskStatus) ?? 'pending';\n      stats[status]++;\n\n      // 按 Agent 统计\n      const agent = taskRecord.agent as string ?? 'unknown';\n      stats.byAgent[agent] = (stats.byAgent[agent] ?? 0) + 1;\n    }\n\n    return stats;\n  }\n\n  // ============================================\n  // Private Methods\n  // ============================================\n\n  private generateTaskId(): string {\n    const timestamp = Date.now().toString(36);\n    const random = randomBytes(4).toString('hex');\n    return `task-${timestamp}-${random}`;\n  }\n\n  /**\n   * 将数据库记录转换为 Task 接口\n   */\n  private dbTaskToTask(dbTask: Record<string, unknown>): Task {\n    return {\n      id: dbTask.id as string,\n      agent: dbTask.agent as string,\n      message: dbTask.message as string,\n      status: (dbTask.status as TaskStatus) ?? 'pending',\n      createdAt: dbTask.createdAt as Date,\n      startedAt: dbTask.startedAt as Date | undefined,\n      completedAt: dbTask.completedAt as Date | undefined,\n      result: dbTask.result as unknown,\n      error: dbTask.error as string | undefined,\n    };\n  }\n}\n"]}

package/dist/task/TaskExecutor.d.ts

@@ -0,0 +1,82 @@
+/**
+ * TaskExecutor - 任务执行器
+ *
+ * 职责：
+ * - 从队列获取待执行任务
+ * - 路由到对应的 Agent
+ * - 调用 Agent 执行
+ * - 更新任务状态
+ */
+import type { Agent, TaskQueue, TaskExecutor as ITaskExecutor } from './types';
+export interface TaskExecutorOptions {
+    /** 任务队列 */
+    taskQueue: TaskQueue;
+    /** Pod base URL */
+    podBaseUrl: string;
+    /** 获取认证 fetch */
+    getAuthenticatedFetch: () => Promise<typeof fetch>;
+    /** 轮询间隔（毫秒） */
+    pollInterval?: number;
+}
+/**
+ * 任务执行器实现
+ */
+export declare class TaskExecutor implements ITaskExecutor {
+    private readonly logger;
+    private readonly taskQueue;
+    private readonly podBaseUrl;
+    private readonly getAuthenticatedFetch;
+    private readonly pollInterval;
+    private readonly agents;
+    private running;
+    private pollTimer;
+    constructor(options: TaskExecutorOptions);
+    /**
+     * 注册 Agent
+     */
+    registerAgent(agent: Agent): void;
+    /**
+     * 启动执行器
+     *
+     * 会先检查 Agent SDK 鉴权状态，如果未鉴权则抛出错误。
+     * 这避免了在服务端环境中 SDK 尝试弹出浏览器进行交互式登录。
+     */
+    start(): Promise<void>;
+    /**
+     * 检查 Agent SDK 鉴权状态
+     *
+     * 在服务端环境中，如果 SDK 未鉴权，会尝试弹出浏览器进行交互式登录，
+     * 这在无界面环境中会失败。因此需要在启动时检查鉴权状态。
+     *
+     * @throws CodeBuddyAuthError 如果未鉴权
+     */
+    private checkAgentAuthentication;
+    /**
+     * 停止执行器
+     */
+    stop(): Promise<void>;
+    /**
+     * 手动触发执行特定任务
+     */
+    trigger(taskId: string): Promise<void>;
+    /**
+     * 获取执行器状态
+     */
+    isRunning(): boolean;
+    /**
+     * 调度下一次轮询
+     */
+    private schedulePoll;
+    /**
+     * 轮询待执行任务
+     */
+    private poll;
+    /**
+     * 执行任务
+     */
+    private executeTask;
+    /**
+     * 创建 Agent 执行上下文
+     */
+    private createContext;
+}