agent-loft 0.2.31 → 0.2.32

package/src/core/tasks-bridge.ts

@@ -19,6 +19,7 @@ import { describeSchedule, formatDateTime, nextRun, normalizeSchedule, parseLoca
 import { TASK_TOOL_TIMEOUT_SEC, tasksSocketPath, verifyTaskCaller } from "./tasks-mcp.js";
 import { createTask, deleteTask, getTask, listTasks, TASKS_PER_USER, updateTask, type Task, type TaskInput } from "./tasks.js";
 import { getUser } from "./users.js";
+import { SKILL_TOOL_NAMES, skillToolCall } from "./skills-bridge.js";
 
 type Reply = { text: string; isError?: boolean };
 const fail = (text: string): Reply => ({ text, isError: true });
@@ -85,6 +86,8 @@ async function call(userId: string, convKey: string, tool: string, args: Record<
   if (!user || user.disabled) return fail("账号已停用。");
   const conv = getConversation(convKey);
   if (!conv || memoryOwnerOf(conv) !== user.id) return fail("这条对话不属于当前用户，不能管理定时任务。");
+  // 技能工具走的也是这条通道（见 tasks-mcp.ts 的 skillsMcpSpawn）
+  if (SKILL_TOOL_NAMES.includes(tool)) return skillToolCall(user.id, convKey, tool, args);
   const now = Date.now();
   const inTaskTurn = !!hub.currentTaskTurn(convKey);
 
@@ -211,7 +214,8 @@ export function startTasksBridge(): void {
       buf += chunk;
       const n = buf.indexOf("\n");
       if (n < 0) {
-        if (buf.length > 64 * 1024) sock.destroy();
+        // 技能工具会带着整份 SKILL.md 和附带文件过来，放宽到 8MB
+        if (buf.length > 8 * 1024 * 1024) sock.destroy();
         return;
       }
       const line = buf.slice(0, n);

package/src/core/tasks-mcp.ts

@@ -36,6 +36,17 @@ export function verifyTaskCaller(userId: string, convKey: string, sig: string):
   return got.length === want.length && crypto.timingSafeEqual(got, want);
 }
 
+/**
+ * 技能工具（让 agent 帮用户建、改技能）也走这条通道：同一个服务脚本，AGENT_LOFT_TOOLSET=skills 时
+ * 只报技能那几个工具，主进程按工具名分给 skills-bridge.ts。单独挂成一个 MCP 服务，工具名才叫得顺。
+ */
+export const SKILLS_MCP_NAME = "agent_loft_skills";
+
+export function skillsMcpSpawn(userId: string, convKey: string, via: string): StdioSpawn {
+  const base = tasksMcpSpawn(userId, convKey, via);
+  return { ...base, env: { ...base.env, AGENT_LOFT_TOOLSET: "skills" } };
+}
+
 export function tasksMcpSpawn(userId: string, convKey: string, via: string): StdioSpawn {
   return {
     command: process.execPath,

package/src/core/types.ts

@@ -188,8 +188,27 @@ export type SessionOptions = {
   };
   /** 模式换过、这个 agent 会话还不知道：换不了系统提示的 agent（Codex）要在第一条消息里说一声 */
   announceMode?: boolean;
-  /** 给 agent 定时任务工具（core/tasks-mcp.ts）：替谁、在哪条对话里管理。只有有主人的对话才给 */
+  /**
+   * 给 agent 定时任务工具（core/tasks-mcp.ts）：替谁、在哪条对话里管理。只有有主人的对话才给。
+   * 技能工具（让 agent 帮用户建、改技能）也跟着它一起给。
+   */
   tasks?: { userId: string; convKey: string };
+  /** 这条对话主人的技能（core/skill-runtime.ts）。没主人的对话不给：机器上的技能照旧，没有 agent-loft 的技能 */
+  skills?: SessionSkills;
+};
+
+export type SessionSkills = {
+  userId: string;
+  /** 管理员：机器上原有的技能照常能用；成员：只有 agent-loft 的技能和 agent 自带的 */
+  admin: boolean;
+  /** Claude 插件目录（里面是这个人启用的技能），一个都没有就不给 */
+  pluginDir?: string;
+  /** 启用的技能：名字、说明、SKILL.md 的路径 */
+  list: { name: string; description: string; path: string }[];
+  /** 要注入的环境变量（技能声明过、这个人填了值的） */
+  env: Record<string, string>;
+  /** 变量配没配好的说明，拼进系统提示 / 背景 */
+  prompt?: string;
 };
 
 export type ConversationMode = "chat" | "work";

package/src/core/zip.ts

@@ -0,0 +1,98 @@
+// 最小的 zip 解包：上传技能包（.zip / .skill）用。只认最常见的两种：不压缩（stored）和 deflate，
+// 加密的、zip64 的直接报错。不落盘，解到内存里交给调用方校验路径、大小后再写。
+import zlib from "node:zlib";
+
+export type ZipEntry = { path: string; data: Buffer };
+
+const EOCD_SIG = 0x06054b50;
+const CENTRAL_SIG = 0x02014b50;
+const LOCAL_SIG = 0x04034b50;
+
+/** maxTotal：解压后的总大小上限（防 zip 炸弹），超了就报错 */
+export function readZip(buf: Buffer, maxTotal: number, maxEntries = 1000): ZipEntry[] {
+  // 目录尾记录在最后 22 字节 + 最长 64KB 的注释里
+  let eocd = -1;
+  for (let i = buf.length - 22; i >= Math.max(0, buf.length - 22 - 0xffff); i--) {
+    if (buf.readUInt32LE(i) === EOCD_SIG) {
+      eocd = i;
+      break;
+    }
+  }
+  if (eocd < 0) throw new Error("不是 zip 文件");
+  const count = buf.readUInt16LE(eocd + 10);
+  let offset = buf.readUInt32LE(eocd + 16);
+  if (count === 0xffff || offset === 0xffffffff) throw new Error("不支持 zip64 格式的压缩包");
+  if (count > maxEntries) throw new Error(`压缩包里的文件太多（最多 ${maxEntries} 个）`);
+
+  const out: ZipEntry[] = [];
+  let total = 0;
+  for (let n = 0; n < count; n++) {
+    if (offset + 46 > buf.length || buf.readUInt32LE(offset) !== CENTRAL_SIG) throw new Error("压缩包损坏");
+    const flags = buf.readUInt16LE(offset + 8);
+    const method = buf.readUInt16LE(offset + 10);
+    const compressed = buf.readUInt32LE(offset + 20);
+    const size = buf.readUInt32LE(offset + 24);
+    const nameLen = buf.readUInt16LE(offset + 28);
+    const extraLen = buf.readUInt16LE(offset + 30);
+    const commentLen = buf.readUInt16LE(offset + 32);
+    const local = buf.readUInt32LE(offset + 42);
+    // 第 11 位：文件名是 UTF-8。没标的多半也是（macOS / 新版 Windows），按 UTF-8 读
+    const name = buf.subarray(offset + 46, offset + 46 + nameLen).toString("utf8");
+    offset += 46 + nameLen + extraLen + commentLen;
+    if (name.endsWith("/")) continue; // 目录
+    if (flags & 1) throw new Error("不支持加密的压缩包");
+    if (size === 0xffffffff || compressed === 0xffffffff) throw new Error("不支持 zip64 格式的压缩包");
+    total += size;
+    if (total > maxTotal) throw new Error(`解压后太大（上限 ${Math.round(maxTotal / 1024 / 1024)}MB）`);
+    if (local + 30 > buf.length || buf.readUInt32LE(local) !== LOCAL_SIG) throw new Error("压缩包损坏");
+    const start = local + 30 + buf.readUInt16LE(local + 26) + buf.readUInt16LE(local + 28);
+    const raw = buf.subarray(start, start + compressed);
+    let data: Buffer;
+    if (method === 0) data = Buffer.from(raw);
+    else if (method === 8) data = zlib.inflateRawSync(raw, { maxOutputLength: Math.max(1, size) });
+    else throw new Error(`不支持的压缩方式（${method}）`);
+    if (data.length !== size) throw new Error("压缩包损坏");
+    out.push({ path: name, data });
+  }
+  return out;
+}
+
+/** 测试和导出用：打一个不压缩的 zip */
+export function writeZip(entries: ZipEntry[]): Buffer {
+  const locals: Buffer[] = [];
+  const centrals: Buffer[] = [];
+  let offset = 0;
+  for (const e of entries) {
+    const name = Buffer.from(e.path, "utf8");
+    const crc = zlib.crc32(e.data);
+    const local = Buffer.alloc(30);
+    local.writeUInt32LE(LOCAL_SIG, 0);
+    local.writeUInt16LE(20, 4);
+    local.writeUInt16LE(0x800, 6);
+    local.writeUInt32LE(crc, 14);
+    local.writeUInt32LE(e.data.length, 18);
+    local.writeUInt32LE(e.data.length, 22);
+    local.writeUInt16LE(name.length, 26);
+    locals.push(local, name, e.data);
+    const central = Buffer.alloc(46);
+    central.writeUInt32LE(CENTRAL_SIG, 0);
+    central.writeUInt16LE(20, 4);
+    central.writeUInt16LE(20, 6);
+    central.writeUInt16LE(0x800, 8);
+    central.writeUInt32LE(crc, 16);
+    central.writeUInt32LE(e.data.length, 20);
+    central.writeUInt32LE(e.data.length, 24);
+    central.writeUInt16LE(name.length, 28);
+    central.writeUInt32LE(offset, 42);
+    centrals.push(central, name);
+    offset += 30 + name.length + e.data.length;
+  }
+  const dir = Buffer.concat(centrals);
+  const end = Buffer.alloc(22);
+  end.writeUInt32LE(EOCD_SIG, 0);
+  end.writeUInt16LE(entries.length, 8);
+  end.writeUInt16LE(entries.length, 10);
+  end.writeUInt32LE(dir.length, 12);
+  end.writeUInt32LE(offset, 16);
+  return Buffer.concat([...locals, dir, end]);
+}

package/src/main.ts

@@ -59,6 +59,10 @@ async function main(): Promise<void> {
   startTaskRunner();
   // 先查一遍 agent 装没装、登没登录，网页打开时新对话里就只列出能用的
   void agentHealth().catch(() => {});
+  // 读一次 Claude 的命令表：成员会话的技能白名单要带上 Claude Code 自带的那些（见 claude/session.ts 的 claudeBuiltins）
+  if (config().agents.claude.enabled) {
+    setTimeout(() => void hub.driver("claude")?.commands?.(config().workspaceRoot).catch(() => {}), 5000).unref();
+  }
 
   console.log(`[启动] workspace=${config().workspaceRoot}`);
   if (config().workspaceRoot === process.cwd()) {

package/src/mcp/tasks-server.ts

@@ -1,4 +1,5 @@
 // 定时任务工具的 MCP 服务（stdio）：给 agent 用的 list / create / update / delete。
+// 环境变量 AGENT_LOFT_TOOLSET=skills 时换成技能工具（list / get / create / update_skill），走同一条通道。
 //
 // 这里只是个传话的：每次调用原样转给主进程（core/tasks-bridge.ts，经 unix socket），带上拉起时
 // 定死的用户、对话和签名（见 core/tasks-mcp.ts）。校验、授权卡、真正的增删改都在主进程里做。
@@ -103,6 +104,126 @@ const TOOLS = [
   },
 ];
 
+// ---------- 技能工具（AGENT_LOFT_TOOLSET=skills 时只报这几个） ----------
+
+const VARS = {
+  type: "array",
+  description:
+    "技能要用的环境变量（API 地址、密钥之类）。值由用户自己在网页上填，运行时注入到他的会话环境里，SKILL.md 里用 $NAME 引用。" +
+    "变量名用大写字母、数字和 _，最好带上技能相关的前缀（比如 WEATHER_API_KEY）。",
+  items: {
+    type: "object",
+    properties: {
+      name: { type: "string" },
+      description: { type: "string", description: "给用户看的说明：这是什么、去哪里拿" },
+      secret: { type: "boolean", description: "密钥、令牌之类：网页上不回显" },
+      required: { type: "boolean" },
+      hosts: {
+        type: "array",
+        items: { type: "string" },
+        description:
+          "secret 时建议给：这个密钥只允许发往的地址（域名或 IP，可带端口，比如 api.example.com）。给了就是受保护的密钥：不放进环境变量，agent 看不到值，" +
+          "只能用 skill_request 工具在请求里写 {{NAME}} 由 agent-loft 代发。这时 body 里要写成用 skill_request 调用，而不是 curl $NAME。",
+      },
+    },
+    required: ["name"],
+    additionalProperties: false,
+  },
+};
+
+const FILES = {
+  type: "array",
+  description: "技能附带的文件（脚本、模板、参考资料），路径相对技能目录，比如 scripts/fetch.sh。只支持文本。",
+  items: {
+    type: "object",
+    properties: { path: { type: "string" }, content: { type: "string" } },
+    required: ["path", "content"],
+    additionalProperties: false,
+  },
+};
+
+const SKILL_TOOLS = [
+  {
+    name: "list_skills",
+    description: "列出用户在 agent-loft 里启用着的技能（名字、说明、谁建的、变量配没配）。",
+    inputSchema: { type: "object", properties: {}, additionalProperties: false },
+    annotations: { title: "查看技能", readOnlyHint: true, openWorldHint: false },
+  },
+  {
+    name: "get_skill",
+    description: "读一个技能的 SKILL.md 和附带文件清单；给 file 时读那个附带文件。改技能之前先读。",
+    inputSchema: {
+      type: "object",
+      properties: { name: { type: "string", description: "技能名" }, file: { type: "string", description: "附带文件的相对路径（可选）" } },
+      required: ["name"],
+      additionalProperties: false,
+    },
+    annotations: { title: "读技能", readOnlyHint: true, openWorldHint: false },
+  },
+  {
+    name: "create_skill",
+    description:
+      "把一套做事的方法存成用户的技能（agent-loft 管理，Claude Code 和 Codex 都能用；格式同 Claude Code 的 SKILL.md）。" +
+      "用户说「把这个做成技能 / 以后都这么做 / 存成 skill」时用。写法：description 一句话说清什么时候该用它（agent 靠它判断）；" +
+      "body 是给 agent 的操作说明（Markdown），步骤具体、可以直接照做；要调接口的，把地址、密钥做成 vars，body 里用 $NAME 引用，不要把真实的密钥写进去。" +
+      "调用后会弹授权卡给用户确认。建好后只有用户自己看得到，下一轮起可用。不要写到 ~/.claude/skills 之类的目录里，那是机器上的，只有管理员能用。",
+    inputSchema: {
+      type: "object",
+      properties: {
+        name: { type: "string", description: "技能名：小写字母、数字和 -，最长 48 个字符，全站唯一" },
+        description: { type: "string", description: "什么时候用这个技能（一两句话）" },
+        body: { type: "string", description: "SKILL.md 的正文（不含 frontmatter）" },
+        files: FILES,
+        vars: VARS,
+      },
+      required: ["name", "description", "body"],
+      additionalProperties: false,
+    },
+    annotations: { title: "新建技能", readOnlyHint: false, destructiveHint: false, idempotentHint: false, openWorldHint: false },
+  },
+  {
+    name: "skill_request",
+    description:
+      "替你发一个 HTTP 请求，url、headers、body 里的 {{变量名}} 由 agent-loft 换成用户在技能里填的值 —— 你看不到密钥本身。" +
+      "技能里的受保护密钥（list_skills 里标了「受保护」的）只能这样用，而且只能发往用户确认过的地址；普通变量也可以这样用。" +
+      "不跟随跳转；响应里出现的变量值会被抹掉；最多返回 256KB 文本。",
+    inputSchema: {
+      type: "object",
+      properties: {
+        method: { type: "string", enum: ["GET", "POST", "PUT", "PATCH", "DELETE", "HEAD"], description: "默认 GET" },
+        url: { type: "string", description: "完整地址，比如 https://api.example.com/v1/weather?city=北京&key={{WEATHER_API_KEY}}" },
+        headers: { type: "object", additionalProperties: { type: "string" }, description: "比如 { \"Authorization\": \"Bearer {{MY_TOKEN}}\" }" },
+        body: { type: "string", description: "请求体（JSON 记得自己序列化，并带上 content-type）" },
+      },
+      required: ["url"],
+      additionalProperties: false,
+    },
+    // openWorldHint 不标：Codex 见到它会先弹 MCP 审批（agent-loft 不接，等于拒绝）。能发往哪里由 agent-loft 自己卡着
+    annotations: { title: "代发请求", readOnlyHint: false, destructiveHint: false, idempotentHint: false, openWorldHint: false },
+  },
+  {
+    name: "update_skill",
+    description: "修改用户自己建的一个技能：说明、正文、附带文件、变量，只给要改的。会弹授权卡让用户确认。分享出去的技能改了之后，装了的人也会用上新的。",
+    inputSchema: {
+      type: "object",
+      properties: {
+        name: { type: "string", description: "技能名（不能改名）" },
+        description: { type: "string" },
+        body: { type: "string", description: "新的 SKILL.md 正文（整段替换）" },
+        files: FILES,
+        delete_files: { type: "array", items: { type: "string" }, description: "要删掉的附带文件" },
+        vars: VARS,
+      },
+      required: ["name"],
+      additionalProperties: false,
+    },
+    annotations: { title: "修改技能", readOnlyHint: false, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+  },
+];
+
+const skillsOnly = process.env.AGENT_LOFT_TOOLSET === "skills";
+const ACTIVE = skillsOnly ? SKILL_TOOLS : TOOLS;
+
 function text(t: string, isError = false) {
   return { content: [{ type: "text", text: t }], ...(isError ? { isError: true } : {}) };
 }
@@ -111,7 +232,7 @@ function text(t: string, isError = false) {
 function forward(tool: string, args: Record<string, unknown>): Promise<{ text: string; isError?: boolean }> {
   return new Promise((resolve) => {
     if (!socketPath || !caller.user || !caller.conv) {
-      resolve({ text: "定时任务工具没有正确配置（缺少用户或对话）。", isError: true });
+      resolve({ text: `${skillsOnly ? "技能" : "定时任务"}工具没有正确配置（缺少用户或对话）。`, isError: true });
       return;
     }
     const sock = net.createConnection(socketPath);
@@ -147,16 +268,16 @@ async function handle(req: Request): Promise<unknown> {
       return {
         protocolVersion: typeof req.params?.protocolVersion === "string" ? req.params.protocolVersion : "2025-06-18",
         capabilities: { tools: {} },
-        serverInfo: { name: "agent-loft-tasks", version: "1.0.0" },
+        serverInfo: { name: skillsOnly ? "agent-loft-skills" : "agent-loft-tasks", version: "1.0.0" },
       };
     case "ping":
       return {};
     case "tools/list":
-      return { tools: TOOLS };
+      return { tools: ACTIVE };
     case "tools/call": {
       const params = req.params ?? {};
       const name = String(params.name ?? "");
-      if (!TOOLS.some((t) => t.name === name)) return text(`没有工具 ${name}`, true);
+      if (!ACTIVE.some((t) => t.name === name)) return text(`没有工具 ${name}`, true);
       const res = await forward(name, (params.arguments as Record<string, unknown> | undefined) ?? {});
       return text(res.text, !!res.isError);
     }