import { streamText } from "ai";import { createBashTool } from "bash-tool";import { loadAllSkillsToMemory } from "@/lib/skill/loadToMemory";import { loadUserWorkspace, saveUserWorkspace } from "@/lib/skill/workspace";
export async function POST(req: Request) {  const { userId, userChatId } = await authenticate(req);
  // 从 S3/磁盘加载 skills  const skills = await prisma.agentSkill.findMany({ where: { userId } });  const skillFiles = await loadAllSkillsToMemory(skills);
  // 加载持久化的 workspace  const workspaceFiles = await loadUserWorkspace(userId);
  // 给 skills 加前缀实现隔离  const skillFilesWithPrefix = Object.fromEntries(    Object.entries(skillFiles).map(([path, content]) => [`skills/${path}`, content])  );
  // 创建沙箱  const { tools: bashTools, sandbox } = await createBashTool({    files: {      ...workspaceFiles,         // 用户的工作文件（根目录）      ...skillFilesWithPrefix,   // Skills（skills/ 子目录）    },  });
  // 合并所有工具  const tools = {    ...baseTools,              // webSearch, reasoningThinking 等    bash: bashTools.bash,    readFile: bashTools.readFile,    writeFile: bashTools.writeFile,    exportFolder: exportFolderTool({ sandbox, userId }),  };
  // 流式响应 + 持久化  const result = streamText({    model: llm("claude-sonnet-4-5"),    system: buildSystemPrompt({ skills, locale }),    messages: await loadMessages(userChatId, { tools }),    tools,
    onStepFinish: async (step) => {      await saveStepToDB(step);      await trackTokens(step);    },
    onFinish: async () => {      // 持久化 workspace 变更      await saveUserWorkspace(userId, sandbox);    },  });
  return result.toUIMessageStreamResponse();}

import type { Sandbox } from "bash-tool";
export async function loadUserWorkspace(userId: number): Promise<Record<string, string>> {  const workspacePath = getWorkspacePath(userId); // .next/cache/sandbox/user/{id}/workspace  const files: Record<string, string> = {};
  await loadDirectoryRecursively(workspacePath, "", files);  return files;}
export async function saveUserWorkspace(userId: number, sandbox: Sandbox): Promise<void> {  const workspacePath = getWorkspacePath(userId);
  // 获取所有非 skills/ 的文件  const findResult = await sandbox.executeCommand(    `find . -type f ! -path "./skills/*" 2>/dev/null || echo ""`  );
  const filePaths = findResult.stdout.split("\n").filter(Boolean);
  // 清空 workspace 并保存最新状态  await fs.rm(workspacePath, { recursive: true, force: true });  await fs.mkdir(workspacePath, { recursive: true });
  for (const filePath of filePaths) {    const content = await sandbox.readFile(filePath);    const fullPath = path.join(workspacePath, filePath.replace(/^\.\//, ""));    await fs.mkdir(path.dirname(fullPath), { recursive: true });    await fs.writeFile(fullPath, content);  }}

export async function buildUniversalSystemPrompt({ userId, locale, skills }) {  const skillsXml = skills.map(s => `<skill>  <name>${s.name}</name>  <description>${s.description}</description>  <location>skills/${s.name}/SKILL.md</location></skill>  `).join('\n');
  return `你是一个拥有专业技能的 Universal Agent。
## 可用技能
${skillsXml}
## Workspace 结构
\`\`\`sandbox/├── skills/                 # 只读技能（来自 S3）│   ├── technical-writer/│   └── market-researcher/└── my-project/            # 持久化工作空间    └── README.md\`\`\`
## 如何使用技能
1. **加载技能**：\`cat skills/technical-writer/SKILL.md\`2. **扮演角色**：完全遵循技能的指令3. **使用参考资料**：\`cat skills/technical-writer/references/examples.md\`
## 文件操作
- **创建**：\`writeFile({ path: "project/index.js", content: "..." })\`- **读取**：\`cat project/index.js\` 或 \`readFile({ path: "project/index.js" })\`- **导出**：\`exportFolder({ folderPath: "project" })\` 供用户下载
根目录的所有文件都会在对话间持久化。
## 准则
- 用户请求专业工作时加载相应技能- 精确遵循技能指令——这是你的专业知识- 在根目录创建文件（不在 skills/ 下）- 使用 bash 命令探索（ls、find、grep 等）`;}

// ❌ 增量方案：跟踪并只保存变更的文件await saveChangedFiles(changedFiles);
// ✅ 完整同步：保存完整 workspace 状态await saveEntireWorkspace(allFiles);

webpack: (config, { isServer, webpack }) => {  if (isServer) {    // 只外部化原生二进制文件    config.externals.push("@mongodb-js/zstd", "node-liblzma");
    // 忽略仅浏览器的 worker.js    config.plugins.push(      new webpack.IgnorePlugin({        resourceRegExp: /^\.\/worker\.js$/,        contextRegExp: /just-bash/,      })    );  }  return config;}

// 定期清理旧的导出文件export async function cleanupOldExports() {  const exportsDir = path.join(process.cwd(), ".next/cache/sandbox");  const cutoff = Date.now() - 24 * 60 * 60 * 1000; // 24 小时
  for (const userId of await fs.readdir(path.join(exportsDir, "user"))) {    const exportsPath = getExportsPath(Number(userId));    for (const file of await fs.readdir(exportsPath)) {      const stat = await fs.stat(path.join(exportsPath, file));      if (stat.mtimeMs < cutoff) {        await fs.unlink(path.join(exportsPath, file));      }    }  }}