@huyooo/ai-chat-core 0.2.19 → 0.2.21

package/src/test-utils/mock-sse.ts

@@ -0,0 +1,32 @@
+/**
+ * 测试用：根据 SSE 数据行创建 Response，用于 mock fetch
+ */
+export function createSSEResponse(sseDataLines: string[], options?: { ok?: boolean; status?: number }): Response {
+  const ok = options?.ok ?? true
+  const status = options?.status ?? 200
+  const body = new ReadableStream<Uint8Array>({
+    start(controller) {
+      const encoder = new TextEncoder()
+      for (const line of sseDataLines) {
+        controller.enqueue(encoder.encode(`data: ${line}\n`))
+      }
+      controller.close()
+    },
+  })
+  return new Response(body, {
+    status,
+    ok,
+    headers: { 'Content-Type': 'text/event-stream' },
+  })
+}
+
+/**
+ * 收集异步生成器的所有值
+ */
+export async function collectAsync<T>(gen: AsyncGenerator<T>): Promise<T[]> {
+  const out: T[] = []
+  for await (const x of gen) {
+    out.push(x)
+  }
+  return out
+}

package/src/tools.ts

@@ -0,0 +1,162 @@
+/**
+ * 工具执行器
+ */
+
+import type { ToolExecutor } from './types';
+
+/** 危险命令黑名单 */
+const DANGEROUS_COMMANDS = [
+  'rm -rf /', 'rm -rf ~', 'format', 'mkfs', 'dd if=',
+  'shutdown', 'reboot', 'sudo rm', 'sudo format'
+];
+
+function isDangerousCommand(command: string): boolean {
+  const lowerCommand = command.toLowerCase().trim();
+  return DANGEROUS_COMMANDS.some(dangerous => 
+    lowerCommand.includes(dangerous.toLowerCase())
+  );
+}
+
+/**
+ * 创建默认工具执行器（Node.js 环境）
+ * 
+ * 用于执行 shell 命令，包含基本安全检查
+ * 支持通过 AbortSignal 取消命令执行
+ */
+export function createDefaultToolExecutor(defaultCwd: string = process.cwd()): ToolExecutor {
+  return {
+    async executeCommand(command: string, cwd?: string, signal?: AbortSignal, hooks?: { onStdout?: (chunk: string) => void; onStderr?: (chunk: string) => void }) {
+      // 检查是否已取消
+      if (signal?.aborted) {
+        return { success: false, error: '操作已取消' };
+      }
+      
+      if (isDangerousCommand(command)) {
+        return { success: false, error: '该命令被安全策略阻止' };
+      }
+
+      try {
+        const { spawn } = await import('child_process');
+        
+        return new Promise((resolve) => {
+          let stdout = '';
+          let stderr = '';
+          let killed = false;
+          let killTimer: NodeJS.Timeout | null = null;
+          
+          const child = spawn('sh', ['-c', command], {
+            cwd: cwd || defaultCwd,
+            env: process.env,
+            // 关键：让 sh 及其子进程处于同一进程组，便于取消/超时时“一锅端”
+            // macOS/Linux: 可用负 PID kill 进程组；Windows 会忽略 detached 的语义
+            detached: process.platform !== 'win32',
+          });
+
+          const killProcessTree = (signalName: NodeJS.Signals) => {
+            if (killed) return;
+            killed = true;
+            try {
+              if (process.platform !== 'win32' && typeof child.pid === 'number') {
+                // 负 PID => 发送到该进程组（避免 sh 死了但子进程继续跑）
+                process.kill(-child.pid, signalName);
+              } else {
+                child.kill(signalName);
+              }
+            } catch {
+              try {
+                child.kill(signalName);
+              } catch {
+                // ignore
+              }
+            }
+
+            // 兜底：若 SIGTERM 不生效，短延迟后再 SIGKILL
+            if (killTimer) clearTimeout(killTimer);
+            killTimer = setTimeout(() => {
+              try {
+                if (process.platform !== 'win32' && typeof child.pid === 'number') {
+                  process.kill(-child.pid, 'SIGKILL');
+                } else {
+                  child.kill('SIGKILL');
+                }
+              } catch {
+                // ignore
+              }
+            }, 1500);
+          };
+          
+          // 设置超时
+          const timeout = setTimeout(() => {
+            if (killed) return;
+            killProcessTree('SIGTERM');
+            resolve({ success: false, error: '命令执行超时（30秒）' });
+          }, 30000);
+          
+          // 监听取消信号
+          const abortHandler = () => {
+            if (killed) return;
+            killProcessTree('SIGTERM');
+            clearTimeout(timeout);
+            resolve({ success: false, error: '操作已取消' });
+          };
+          signal?.addEventListener('abort', abortHandler);
+          
+          child.stdout?.on('data', (data) => {
+            const chunk = data.toString();
+            stdout += chunk;
+            hooks?.onStdout?.(chunk);
+          });
+          
+          child.stderr?.on('data', (data) => {
+            const chunk = data.toString();
+            stderr += chunk;
+            hooks?.onStderr?.(chunk);
+          });
+          
+          child.on('close', (code) => {
+            clearTimeout(timeout);
+            signal?.removeEventListener('abort', abortHandler);
+            if (killTimer) {
+              clearTimeout(killTimer);
+              killTimer = null;
+            }
+            
+            if (killed) return;
+            
+            const output = stdout.trim();
+            const errOutput = stderr.trim();
+            
+            if (code === 0) {
+              resolve({ success: true, output: output || errOutput || '执行成功' });
+            } else {
+              // 非零退出码但有输出时，也算成功（某些命令会这样）
+              if (output) {
+                const warning = errOutput ? `\n[警告: ${errOutput}]` : '';
+                resolve({ success: true, output: output + warning });
+              } else if (errOutput) {
+                resolve({ success: false, error: errOutput });
+              } else {
+                resolve({ success: false, error: `命令退出码: ${code}` });
+              }
+            }
+          });
+          
+          child.on('error', (err) => {
+            clearTimeout(timeout);
+            signal?.removeEventListener('abort', abortHandler);
+            if (killTimer) {
+              clearTimeout(killTimer);
+              killTimer = null;
+            }
+            
+            if (!killed) {
+              resolve({ success: false, error: err.message });
+            }
+          });
+        });
+      } catch (error) {
+        return { success: false, error: error instanceof Error ? error.message : String(error) };
+      }
+    }
+  };
+}