chatccc 0.2.190 → 0.2.192

package/src/__tests__/agent-delegate-task-rpc.test.ts

@@ -1,165 +1,165 @@
-import { Readable } from "node:stream";
-import { describe, expect, it, vi, beforeEach } from "vitest";
-
-import type { PlatformAdapter } from "../platform-adapter.ts";
-
-const mockDelegateAgentTask = vi.hoisted(() => vi.fn());
-
-vi.mock("../agent-delegate-task.ts", () => ({
-  delegateAgentTask: mockDelegateAgentTask,
-}));
-
-import {
-  AGENT_DELEGATE_TASK_PATH,
-  buildAgentDelegateTaskCapabilityPrompt,
-  handleAgentDelegateTaskRequest,
-} from "../agent-delegate-task-rpc.ts";
-import { ABD_APPEND_PROMPT } from "../shared-prefix.ts";
-
-function request(body: unknown, path = AGENT_DELEGATE_TASK_PATH, method = "POST"): Readable & {
-  url?: string;
-  method?: string;
-  headers: Record<string, string>;
-} {
-  const req = Readable.from([Buffer.from(JSON.stringify(body), "utf8")]) as Readable & {
-    url?: string;
-    method?: string;
-    headers: Record<string, string>;
-  };
-  req.url = path;
-  req.method = method;
-  req.headers = { "content-type": "application/json; charset=utf-8" };
-  return req;
-}
-
-function response() {
-  const res = {
-    statusCode: 0,
-    headers: {} as Record<string, string>,
-    body: "",
-    writeHead(status: number, headers: Record<string, string>) {
-      this.statusCode = status;
-      this.headers = headers;
-      return this;
-    },
-    end(chunk?: string) {
-      this.body += chunk ?? "";
-      return this;
-    },
-  };
-  return res;
-}
-
-function platform(kind: "feishu" | "wechat" = "feishu"): PlatformAdapter {
-  return {
-    kind,
-    sendText: vi.fn(async () => true),
-    sendCard: vi.fn(async () => true),
-    sendRawCard: vi.fn(async () => true),
-    createGroup: vi.fn(async () => "chat-id"),
-    updateChatInfo: vi.fn(async () => {}),
-    getChatInfo: vi.fn(async () => ({ name: "chat", description: "" })),
-    disbandChat: vi.fn(async () => {}),
-    setChatAvatar: vi.fn(async () => {}),
-    extractSessionInfo: vi.fn(() => null),
-    cardCreate: vi.fn(async () => "card-id"),
-    cardSend: vi.fn(async () => "message-id"),
-    cardUpdate: vi.fn(async () => {}),
-  };
-}
-
-describe("agent delegate task RPC", () => {
-  beforeEach(() => {
-    mockDelegateAgentTask.mockReset();
-    mockDelegateAgentTask.mockResolvedValue({
-      chatId: "chat-new",
-      sessionId: "sid-new",
-      tool: "codex",
-      cwd: "F:\\repo",
-    });
-  });
-
-  it("delegates a task through the local API", async () => {
-    const req = request({
-      tool: "codex",
-      cwd: "F:\\repo",
-      open_id: "ou-user",
-      prompt: "帮我分析日志",
-    });
-    const res = response();
-
-    await handleAgentDelegateTaskRequest(req as never, res as never, platform());
-
-    expect(res.statusCode).toBe(200);
-    expect(JSON.parse(res.body)).toMatchObject({
-      ok: true,
-      chat_id: "chat-new",
-      session_id: "sid-new",
-      tool: "codex",
-    });
-    expect(mockDelegateAgentTask).toHaveBeenCalledWith(expect.objectContaining({
-      tool: "codex",
-      cwd: expect.stringContaining("repo"),
-      promptText: "帮我分析日志",
-      openIds: ["ou-user"],
-      chatNamePrefix: "帮我分析日志",
-    }));
-  });
-
-  it("applies the shared ABD prefix to API prompts", async () => {
-    const req = request({
-      tool: "claude",
-      cwd: "F:\\repo",
-      openIds: ["ou-user"],
-      prompt: "/abd帮我分析",
-    });
-    const res = response();
-
-    await handleAgentDelegateTaskRequest(req as never, res as never, platform());
-
-    expect(res.statusCode).toBe(200);
-    expect(mockDelegateAgentTask).toHaveBeenCalledWith(expect.objectContaining({
-      promptText: `帮我分析\n\n---\n${ABD_APPEND_PROMPT}`,
-      chatNamePrefix: "帮我分析",
-    }));
-  });
-
-  it("rejects non-Feishu platforms", async () => {
-    const req = request({
-      tool: "codex",
-      cwd: "F:\\repo",
-      open_id: "ou-user",
-      prompt: "task",
-    });
-    const res = response();
-
-    await handleAgentDelegateTaskRequest(req as never, res as never, platform("wechat"));
-
-    expect(res.statusCode).toBe(409);
-    expect(mockDelegateAgentTask).not.toHaveBeenCalled();
-  });
-
-  it("rejects missing required parameters", async () => {
-    const req = request({ tool: "codex", cwd: "F:\\repo", prompt: "task" });
-    const res = response();
-
-    await handleAgentDelegateTaskRequest(req as never, res as never, platform());
-
-    expect(res.statusCode).toBe(400);
-    expect(JSON.parse(res.body).error).toContain("open_id");
-    expect(mockDelegateAgentTask).not.toHaveBeenCalled();
-  });
-
-  it("builds prompt instructions for the delegate endpoint", () => {
-    const prompt = buildAgentDelegateTaskCapabilityPrompt({
-      url: `http://127.0.0.1:18080${AGENT_DELEGATE_TASK_PATH}`,
-      cwd: "F:/repo",
-    });
-
-    expect(prompt).toContain("POST http://127.0.0.1:18080/api/agent/delegate-task");
-    expect(prompt).toContain('"tool":"codex|claude|cursor"');
-    expect(prompt).toContain('"cwd":"absolute working directory"');
-    expect(prompt).toContain("project prompt injection and IM skills still apply");
-    expect(prompt).toContain("Current working directory: F:/repo");
-  });
-});
+import { Readable } from "node:stream";
+import { describe, expect, it, vi, beforeEach } from "vitest";
+
+import type { PlatformAdapter } from "../platform-adapter.ts";
+
+const mockDelegateAgentTask = vi.hoisted(() => vi.fn());
+
+vi.mock("../agent-delegate-task.ts", () => ({
+  delegateAgentTask: mockDelegateAgentTask,
+}));
+
+import {
+  AGENT_DELEGATE_TASK_PATH,
+  buildAgentDelegateTaskCapabilityPrompt,
+  handleAgentDelegateTaskRequest,
+} from "../agent-delegate-task-rpc.ts";
+import { ABD_APPEND_PROMPT } from "../shared-prefix.ts";
+
+function request(body: unknown, path = AGENT_DELEGATE_TASK_PATH, method = "POST"): Readable & {
+  url?: string;
+  method?: string;
+  headers: Record<string, string>;
+} {
+  const req = Readable.from([Buffer.from(JSON.stringify(body), "utf8")]) as Readable & {
+    url?: string;
+    method?: string;
+    headers: Record<string, string>;
+  };
+  req.url = path;
+  req.method = method;
+  req.headers = { "content-type": "application/json; charset=utf-8" };
+  return req;
+}
+
+function response() {
+  const res = {
+    statusCode: 0,
+    headers: {} as Record<string, string>,
+    body: "",
+    writeHead(status: number, headers: Record<string, string>) {
+      this.statusCode = status;
+      this.headers = headers;
+      return this;
+    },
+    end(chunk?: string) {
+      this.body += chunk ?? "";
+      return this;
+    },
+  };
+  return res;
+}
+
+function platform(kind: "feishu" | "wechat" = "feishu"): PlatformAdapter {
+  return {
+    kind,
+    sendText: vi.fn(async () => true),
+    sendCard: vi.fn(async () => true),
+    sendRawCard: vi.fn(async () => true),
+    createGroup: vi.fn(async () => "chat-id"),
+    updateChatInfo: vi.fn(async () => {}),
+    getChatInfo: vi.fn(async () => ({ name: "chat", description: "" })),
+    disbandChat: vi.fn(async () => {}),
+    setChatAvatar: vi.fn(async () => {}),
+    extractSessionInfo: vi.fn(() => null),
+    cardCreate: vi.fn(async () => "card-id"),
+    cardSend: vi.fn(async () => "message-id"),
+    cardUpdate: vi.fn(async () => {}),
+  };
+}
+
+describe("agent delegate task RPC", () => {
+  beforeEach(() => {
+    mockDelegateAgentTask.mockReset();
+    mockDelegateAgentTask.mockResolvedValue({
+      chatId: "chat-new",
+      sessionId: "sid-new",
+      tool: "codex",
+      cwd: "F:\\repo",
+    });
+  });
+
+  it("delegates a task through the local API", async () => {
+    const req = request({
+      tool: "codex",
+      cwd: "F:\\repo",
+      open_id: "ou-user",
+      prompt: "帮我分析日志",
+    });
+    const res = response();
+
+    await handleAgentDelegateTaskRequest(req as never, res as never, platform());
+
+    expect(res.statusCode).toBe(200);
+    expect(JSON.parse(res.body)).toMatchObject({
+      ok: true,
+      chat_id: "chat-new",
+      session_id: "sid-new",
+      tool: "codex",
+    });
+    expect(mockDelegateAgentTask).toHaveBeenCalledWith(expect.objectContaining({
+      tool: "codex",
+      cwd: expect.stringContaining("repo"),
+      promptText: "帮我分析日志",
+      openIds: ["ou-user"],
+      chatNamePrefix: "帮我分析日志",
+    }));
+  });
+
+  it("applies the shared ABD prefix to API prompts", async () => {
+    const req = request({
+      tool: "claude",
+      cwd: "F:\\repo",
+      openIds: ["ou-user"],
+      prompt: "/abd帮我分析",
+    });
+    const res = response();
+
+    await handleAgentDelegateTaskRequest(req as never, res as never, platform());
+
+    expect(res.statusCode).toBe(200);
+    expect(mockDelegateAgentTask).toHaveBeenCalledWith(expect.objectContaining({
+      promptText: `帮我分析\n\n---\n${ABD_APPEND_PROMPT}`,
+      chatNamePrefix: "帮我分析",
+    }));
+  });
+
+  it("rejects non-Feishu platforms", async () => {
+    const req = request({
+      tool: "codex",
+      cwd: "F:\\repo",
+      open_id: "ou-user",
+      prompt: "task",
+    });
+    const res = response();
+
+    await handleAgentDelegateTaskRequest(req as never, res as never, platform("wechat"));
+
+    expect(res.statusCode).toBe(409);
+    expect(mockDelegateAgentTask).not.toHaveBeenCalled();
+  });
+
+  it("rejects missing required parameters", async () => {
+    const req = request({ tool: "codex", cwd: "F:\\repo", prompt: "task" });
+    const res = response();
+
+    await handleAgentDelegateTaskRequest(req as never, res as never, platform());
+
+    expect(res.statusCode).toBe(400);
+    expect(JSON.parse(res.body).error).toContain("open_id");
+    expect(mockDelegateAgentTask).not.toHaveBeenCalled();
+  });
+
+  it("builds prompt instructions for the delegate endpoint", () => {
+    const prompt = buildAgentDelegateTaskCapabilityPrompt({
+      url: `http://127.0.0.1:18080${AGENT_DELEGATE_TASK_PATH}`,
+      cwd: "F:/repo",
+    });
+
+    expect(prompt).toContain("POST http://127.0.0.1:18080/api/agent/delegate-task");
+    expect(prompt).toContain('"tool":"codex|claude|cursor"');
+    expect(prompt).toContain('"cwd":"absolute working directory"');
+    expect(prompt).toContain("project prompt injection and IM skills still apply");
+    expect(prompt).toContain("Current working directory: F:/repo");
+  });
+});

package/src/__tests__/builtin-chat-session.test.ts

@@ -0,0 +1,76 @@
+import { mkdtemp } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+
+import { afterEach, describe, expect, it, vi } from "vitest";
+
+const streamTextMock = vi.fn();
+const generateTextMock = vi.fn();
+
+vi.mock("@ai-sdk/openai-compatible", () => ({
+  createOpenAICompatible: vi.fn(() => (modelId: string) => ({ modelId })),
+}));
+
+vi.mock("ai", () => ({
+  streamText: streamTextMock,
+  generateText: generateTextMock,
+}));
+
+async function collect(iterable: AsyncIterable<unknown>): Promise<unknown[]> {
+  const events: unknown[] = [];
+  for await (const event of iterable) events.push(event);
+  return events;
+}
+
+async function* textStream(...chunks: string[]): AsyncIterable<string> {
+  for (const chunk of chunks) yield chunk;
+}
+
+afterEach(() => {
+  streamTextMock.mockReset();
+  generateTextMock.mockReset();
+});
+
+describe("ChatSession context management", () => {
+  it("loads persisted context, compacts older messages, and persists the new assistant reply", async () => {
+    const { ChatSession } = await import("../builtin/index.ts");
+    const dir = await mkdtemp(join(tmpdir(), "chatccc-session-context-"));
+
+    const seed = new ChatSession(
+      { apiKey: "sk-test" },
+      {
+        persist: true,
+        contextDir: dir,
+        sessionId: "integration",
+        compactAtTokens: 10_000,
+      },
+    );
+    streamTextMock.mockReturnValueOnce({ textStream: textStream("old answer") });
+    await collect(seed.chat("old question"));
+
+    generateTextMock.mockResolvedValueOnce({ text: "## 当前任务状态\n- 旧问题已总结" });
+    streamTextMock.mockReturnValueOnce({ textStream: textStream("new answer") });
+
+    const restored = new ChatSession(
+      { apiKey: "sk-test" },
+      {
+        persist: true,
+        contextDir: dir,
+        sessionId: "integration",
+        compactAtTokens: 1,
+        keepRecentMessages: 1,
+      },
+    );
+    const events = await collect(restored.chat("new question"));
+
+    expect(generateTextMock).toHaveBeenCalledOnce();
+    expect(streamTextMock).toHaveBeenLastCalledWith(expect.objectContaining({
+      messages: expect.arrayContaining([
+        expect.objectContaining({ content: expect.stringContaining("旧问题已总结") }),
+        expect.objectContaining({ role: "user", content: "new question" }),
+      ]),
+    }));
+    expect(events).toContainEqual({ type: "compact", compactedMessages: 2 });
+    expect(restored.history.map((m) => m.content).join("\n")).toContain("new answer");
+  });
+});

package/src/__tests__/builtin-config.test.ts

@@ -1,33 +1,33 @@
-import { afterEach, describe, expect, it } from "vitest";
-
-import { ChatSession } from "../builtin/index.ts";
-import { config } from "../config.ts";
-
-const originalDeepSeekApiKey = process.env.DEEPSEEK_API_KEY;
-const originalCccConfig = structuredClone(config.ccc);
-
-afterEach(() => {
-  if (originalDeepSeekApiKey === undefined) {
-    delete process.env.DEEPSEEK_API_KEY;
-  } else {
-    process.env.DEEPSEEK_API_KEY = originalDeepSeekApiKey;
-  }
-  config.ccc = structuredClone(originalCccConfig);
-});
-
-describe("builtin ChatSession config", () => {
-  it("does not fall back to DEEPSEEK_API_KEY environment variable", () => {
-    config.ccc = {
-      DEEPSEEK_API_KEY: "",
-      DEEPSEEK_BASE_URL: "https://api.deepseek.com/v1",
-      model: "deepseek-v4-pro",
-    };
-    process.env.DEEPSEEK_API_KEY = "sk-env-should-not-be-used";
-
-    expect(() => new ChatSession()).toThrow("ccc.DEEPSEEK_API_KEY 未设置");
-  });
-
-  it("allows constructor parameters to override config defaults", () => {
-    expect(() => new ChatSession({ apiKey: "sk-test" })).not.toThrow();
-  });
-});
+import { afterEach, describe, expect, it } from "vitest";
+
+import { ChatSession } from "../builtin/index.ts";
+import { config } from "../config.ts";
+
+const originalDeepSeekApiKey = process.env.DEEPSEEK_API_KEY;
+const originalCccConfig = structuredClone(config.ccc);
+
+afterEach(() => {
+  if (originalDeepSeekApiKey === undefined) {
+    delete process.env.DEEPSEEK_API_KEY;
+  } else {
+    process.env.DEEPSEEK_API_KEY = originalDeepSeekApiKey;
+  }
+  config.ccc = structuredClone(originalCccConfig);
+});
+
+describe("builtin ChatSession config", () => {
+  it("does not fall back to DEEPSEEK_API_KEY environment variable", () => {
+    config.ccc = {
+      DEEPSEEK_API_KEY: "",
+      DEEPSEEK_BASE_URL: "https://api.deepseek.com/v1",
+      model: "deepseek-v4-pro",
+    };
+    process.env.DEEPSEEK_API_KEY = "sk-env-should-not-be-used";
+
+    expect(() => new ChatSession()).toThrow("ccc.DEEPSEEK_API_KEY 未设置");
+  });
+
+  it("allows constructor parameters to override config defaults", () => {
+    expect(() => new ChatSession({ apiKey: "sk-test" })).not.toThrow();
+  });
+});

package/src/__tests__/builtin-context.test.ts

@@ -0,0 +1,126 @@
+import { mkdtemp, readFile } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+
+import { describe, expect, it } from "vitest";
+
+import {
+  BuiltinContextManager,
+  estimateBuiltinContextTokens,
+  serializeMessagesForSummary,
+} from "../builtin/context.ts";
+
+describe("BuiltinContextManager", () => {
+  it("persists and restores summary, messages, and total message count", async () => {
+    const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-"));
+
+    const first = new BuiltinContextManager({
+      persist: true,
+      contextDir: dir,
+      sessionId: "persisted",
+    });
+    first.setSummary("## 用户目标\n- 保留这个摘要");
+    first.appendMessage({ role: "user", content: "第一轮" });
+    first.appendMessage({ role: "assistant", content: "第一轮回复" });
+    first.save();
+
+    const restored = new BuiltinContextManager({
+      persist: true,
+      contextDir: dir,
+      sessionId: "persisted",
+    });
+
+    expect(restored.summary).toContain("保留这个摘要");
+    expect(restored.messages).toEqual([
+      { role: "user", content: "第一轮" },
+      { role: "assistant", content: "第一轮回复" },
+    ]);
+    expect(restored.totalMessages).toBe(2);
+  });
+
+  it("selects only older messages for compaction and keeps recent messages raw", () => {
+    const context = new BuiltinContextManager({
+      compactAtTokens: 1,
+      keepRecentMessages: 2,
+      persist: false,
+    });
+    context.appendMessage({ role: "user", content: "旧用户消息" });
+    context.appendMessage({ role: "assistant", content: "旧助手回复" });
+    context.appendMessage({ role: "user", content: "近期用户消息" });
+    context.appendMessage({ role: "assistant", content: "近期助手回复" });
+
+    const plan = context.planCompaction();
+
+    expect(plan).not.toBeNull();
+    expect(plan?.oldMessages).toEqual([
+      { role: "user", content: "旧用户消息" },
+      { role: "assistant", content: "旧助手回复" },
+    ]);
+    expect(plan?.recentMessages).toEqual([
+      { role: "user", content: "近期用户消息" },
+      { role: "assistant", content: "近期助手回复" },
+    ]);
+  });
+
+  it("applies a compacted summary and builds model messages with summary plus recent raw turns", () => {
+    const context = new BuiltinContextManager({
+      compactAtTokens: 1,
+      keepRecentMessages: 1,
+      persist: false,
+    });
+    context.appendMessage({ role: "user", content: "old" });
+    context.appendMessage({ role: "assistant", content: "recent" });
+
+    const plan = context.planCompaction();
+    expect(plan).not.toBeNull();
+    context.applyCompaction("## 当前任务状态\n- 已压缩旧上下文", plan!);
+
+    expect(context.summary).toContain("已压缩旧上下文");
+    expect(context.messages).toEqual([{ role: "assistant", content: "recent" }]);
+    expect(context.buildModelMessages()).toEqual([
+      {
+        role: "user",
+        content: expect.stringContaining("较早对话摘要"),
+      },
+      { role: "assistant", content: "recent" },
+    ]);
+  });
+
+  it("reset clears memory and the persisted context file", async () => {
+    const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-reset-"));
+    const context = new BuiltinContextManager({
+      persist: true,
+      contextDir: dir,
+      sessionId: "reset",
+    });
+    context.setSummary("summary");
+    context.appendMessage({ role: "user", content: "hello" });
+    context.save();
+
+    context.reset();
+
+    expect(context.summary).toBe("");
+    expect(context.messages).toEqual([]);
+    expect(context.totalMessages).toBe(0);
+
+    const raw = await readFile(join(dir, "reset", "context.json"), "utf8");
+    const persisted = JSON.parse(raw) as { summary: string; messages: unknown[]; totalMessages: number };
+    expect(persisted.summary).toBe("");
+    expect(persisted.messages).toEqual([]);
+    expect(persisted.totalMessages).toBe(0);
+  });
+});
+
+describe("builtin context helpers", () => {
+  it("estimates tokens from summary and messages", () => {
+    expect(estimateBuiltinContextTokens("abc", [{ role: "user", content: "abcdef" }]))
+      .toBeGreaterThanOrEqual(3);
+  });
+
+  it("serializes messages for summarization with stable role labels", () => {
+    expect(serializeMessagesForSummary([
+      { role: "user", content: "你好" },
+      { role: "assistant", content: "收到" },
+    ])).toContain("user");
+  });
+});

package/src/__tests__/builtin-sigint.test.ts

@@ -0,0 +1,56 @@
+import { describe, expect, it } from "vitest";
+
+import { createCtrlCState } from "../builtin/sigint.ts";
+
+describe("builtin CLI Ctrl+C state", () => {
+  it("requires two presses to interrupt an active generation", () => {
+    let now = 1_000;
+    const state = createCtrlCState({ windowMs: 2_000, now: () => now });
+
+    expect(state.press(true)).toBe("arm-interrupt");
+    expect(state.press(true)).toBe("interrupt");
+  });
+
+  it("requires two presses to exit while idle", () => {
+    let now = 1_000;
+    const state = createCtrlCState({ windowMs: 2_000, now: () => now });
+
+    expect(state.press(false)).toBe("arm-exit");
+    expect(state.press(false)).toBe("exit");
+  });
+
+  it("expires the pending confirmation after the window", () => {
+    let now = 1_000;
+    const state = createCtrlCState({ windowMs: 2_000, now: () => now });
+
+    expect(state.press(true)).toBe("arm-interrupt");
+
+    now += 2_001;
+
+    expect(state.press(true)).toBe("arm-interrupt");
+    expect(state.press(true)).toBe("interrupt");
+  });
+
+  it("does not convert a pending interrupt into an idle exit after reset", () => {
+    let now = 1_000;
+    const state = createCtrlCState({ windowMs: 2_000, now: () => now });
+
+    expect(state.press(true)).toBe("arm-interrupt");
+
+    state.reset();
+    now += 1_000;
+
+    expect(state.press(false)).toBe("arm-exit");
+  });
+
+  it("does not confirm a different action with the second press", () => {
+    let now = 1_000;
+    const state = createCtrlCState({ windowMs: 2_000, now: () => now });
+
+    expect(state.press(true)).toBe("arm-interrupt");
+    now += 500;
+
+    expect(state.press(false)).toBe("arm-exit");
+    expect(state.press(false)).toBe("exit");
+  });
+});

package/src/__tests__/card-plain-text.test.ts

@@ -16,11 +16,11 @@ describe("cardJsonToPlainText", () => {
     expect(text).toContain("/new");
     expect(text).toContain("/new cursor");
     expect(text).toContain("/new codex");
-    expect(text).toContain("/restart");
-    expect(text).toContain("/update");
-    expect(text).toContain("/cd");
-    expect(text).toContain("/abd");
-  });
+    expect(text).toContain("/restart");
+    expect(text).toContain("/update");
+    expect(text).toContain("/cd");
+    expect(text).toContain("/abd");
+  });
 
   it("converts status cards from v1 card format", () => {
     const text = cardJsonToPlainText(buildStatusCard("status body", "green"));