agent-tower 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/app.d.ts +2 -0
- package/dist/app.d.ts.map +1 -0
- package/dist/app.js +62 -0
- package/dist/app.js.map +1 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +131 -0
- package/dist/cli.js.map +1 -0
- package/dist/core/container.d.ts +11 -0
- package/dist/core/container.d.ts.map +1 -0
- package/dist/core/container.js +66 -0
- package/dist/core/container.js.map +1 -0
- package/dist/core/event-bus.d.ts +58 -0
- package/dist/core/event-bus.d.ts.map +1 -0
- package/dist/core/event-bus.js +29 -0
- package/dist/core/event-bus.js.map +1 -0
- package/dist/errors.d.ts +22 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +36 -0
- package/dist/errors.js.map +1 -0
- package/dist/executors/base.executor.d.ts +160 -0
- package/dist/executors/base.executor.d.ts.map +1 -0
- package/dist/executors/base.executor.js +122 -0
- package/dist/executors/base.executor.js.map +1 -0
- package/dist/executors/claude-code.executor.d.ts +78 -0
- package/dist/executors/claude-code.executor.d.ts.map +1 -0
- package/dist/executors/claude-code.executor.js +160 -0
- package/dist/executors/claude-code.executor.js.map +1 -0
- package/dist/executors/command-builder.d.ts +64 -0
- package/dist/executors/command-builder.d.ts.map +1 -0
- package/dist/executors/command-builder.js +144 -0
- package/dist/executors/command-builder.js.map +1 -0
- package/dist/executors/cursor-agent.executor.d.ts +57 -0
- package/dist/executors/cursor-agent.executor.d.ts.map +1 -0
- package/dist/executors/cursor-agent.executor.js +109 -0
- package/dist/executors/cursor-agent.executor.js.map +1 -0
- package/dist/executors/default-profiles.json +29 -0
- package/dist/executors/execution-env.d.ts +64 -0
- package/dist/executors/execution-env.d.ts.map +1 -0
- package/dist/executors/execution-env.js +97 -0
- package/dist/executors/execution-env.js.map +1 -0
- package/dist/executors/gemini-cli.executor.d.ts +55 -0
- package/dist/executors/gemini-cli.executor.d.ts.map +1 -0
- package/dist/executors/gemini-cli.executor.js +117 -0
- package/dist/executors/gemini-cli.executor.js.map +1 -0
- package/dist/executors/index.d.ts +49 -0
- package/dist/executors/index.d.ts.map +1 -0
- package/dist/executors/index.js +95 -0
- package/dist/executors/index.js.map +1 -0
- package/dist/executors/profiles.d.ts +61 -0
- package/dist/executors/profiles.d.ts.map +1 -0
- package/dist/executors/profiles.js +192 -0
- package/dist/executors/profiles.js.map +1 -0
- package/dist/git/git-cli.d.ts +59 -0
- package/dist/git/git-cli.d.ts.map +1 -0
- package/dist/git/git-cli.js +156 -0
- package/dist/git/git-cli.js.map +1 -0
- package/dist/git/worktree.manager.d.ts +166 -0
- package/dist/git/worktree.manager.d.ts.map +1 -0
- package/dist/git/worktree.manager.js +481 -0
- package/dist/git/worktree.manager.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +37 -0
- package/dist/index.js.map +1 -0
- package/dist/mcp/context.d.ts +15 -0
- package/dist/mcp/context.d.ts.map +1 -0
- package/dist/mcp/context.js +19 -0
- package/dist/mcp/context.js.map +1 -0
- package/dist/mcp/http-client.d.ts +46 -0
- package/dist/mcp/http-client.d.ts.map +1 -0
- package/dist/mcp/http-client.js +100 -0
- package/dist/mcp/http-client.js.map +1 -0
- package/dist/mcp/index.d.ts +3 -0
- package/dist/mcp/index.d.ts.map +1 -0
- package/dist/mcp/index.js +36 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/mcp/server.d.ts +6 -0
- package/dist/mcp/server.d.ts.map +1 -0
- package/dist/mcp/server.js +32 -0
- package/dist/mcp/server.js.map +1 -0
- package/dist/mcp/tools/projects.d.ts +7 -0
- package/dist/mcp/tools/projects.d.ts.map +1 -0
- package/dist/mcp/tools/projects.js +12 -0
- package/dist/mcp/tools/projects.js.map +1 -0
- package/dist/mcp/tools/sessions.d.ts +7 -0
- package/dist/mcp/tools/sessions.d.ts.map +1 -0
- package/dist/mcp/tools/sessions.js +22 -0
- package/dist/mcp/tools/sessions.js.map +1 -0
- package/dist/mcp/tools/tasks.d.ts +7 -0
- package/dist/mcp/tools/tasks.d.ts.map +1 -0
- package/dist/mcp/tools/tasks.js +68 -0
- package/dist/mcp/tools/tasks.js.map +1 -0
- package/dist/mcp/tools/workspaces.d.ts +7 -0
- package/dist/mcp/tools/workspaces.d.ts.map +1 -0
- package/dist/mcp/tools/workspaces.js +45 -0
- package/dist/mcp/tools/workspaces.js.map +1 -0
- package/dist/mcp/types.d.ts +109 -0
- package/dist/mcp/types.d.ts.map +1 -0
- package/dist/mcp/types.js +55 -0
- package/dist/mcp/types.js.map +1 -0
- package/dist/middleware/tunnel-auth.d.ts +13 -0
- package/dist/middleware/tunnel-auth.d.ts.map +1 -0
- package/dist/middleware/tunnel-auth.js +39 -0
- package/dist/middleware/tunnel-auth.js.map +1 -0
- package/dist/output/__tests__/claude-code-parser-token.test.d.ts +2 -0
- package/dist/output/__tests__/claude-code-parser-token.test.d.ts.map +1 -0
- package/dist/output/__tests__/claude-code-parser-token.test.js +233 -0
- package/dist/output/__tests__/claude-code-parser-token.test.js.map +1 -0
- package/dist/output/__tests__/cursor-agent-parser-token.test.d.ts +2 -0
- package/dist/output/__tests__/cursor-agent-parser-token.test.d.ts.map +1 -0
- package/dist/output/__tests__/cursor-agent-parser-token.test.js +95 -0
- package/dist/output/__tests__/cursor-agent-parser-token.test.js.map +1 -0
- package/dist/output/__tests__/msg-store-token.test.d.ts +2 -0
- package/dist/output/__tests__/msg-store-token.test.d.ts.map +1 -0
- package/dist/output/__tests__/msg-store-token.test.js +38 -0
- package/dist/output/__tests__/msg-store-token.test.js.map +1 -0
- package/dist/output/claude-code-parser.d.ts +96 -0
- package/dist/output/claude-code-parser.d.ts.map +1 -0
- package/dist/output/claude-code-parser.js +498 -0
- package/dist/output/claude-code-parser.js.map +1 -0
- package/dist/output/cursor-agent-parser.d.ts +87 -0
- package/dist/output/cursor-agent-parser.d.ts.map +1 -0
- package/dist/output/cursor-agent-parser.js +476 -0
- package/dist/output/cursor-agent-parser.js.map +1 -0
- package/dist/output/index.d.ts +9 -0
- package/dist/output/index.d.ts.map +1 -0
- package/dist/output/index.js +13 -0
- package/dist/output/index.js.map +1 -0
- package/dist/output/msg-store.d.ts +110 -0
- package/dist/output/msg-store.d.ts.map +1 -0
- package/dist/output/msg-store.js +274 -0
- package/dist/output/msg-store.js.map +1 -0
- package/dist/output/types.d.ts +90 -0
- package/dist/output/types.d.ts.map +1 -0
- package/dist/output/types.js +93 -0
- package/dist/output/types.js.map +1 -0
- package/dist/output/utils/ansi.d.ts +8 -0
- package/dist/output/utils/ansi.d.ts.map +1 -0
- package/dist/output/utils/ansi.js +17 -0
- package/dist/output/utils/ansi.js.map +1 -0
- package/dist/output/utils/patch.d.ts +57 -0
- package/dist/output/utils/patch.d.ts.map +1 -0
- package/dist/output/utils/patch.js +113 -0
- package/dist/output/utils/patch.js.map +1 -0
- package/dist/pipeline/agent-pipeline.d.ts +28 -0
- package/dist/pipeline/agent-pipeline.d.ts.map +1 -0
- package/dist/pipeline/agent-pipeline.js +76 -0
- package/dist/pipeline/agent-pipeline.js.map +1 -0
- package/dist/routes/attachments.d.ts +3 -0
- package/dist/routes/attachments.d.ts.map +1 -0
- package/dist/routes/attachments.js +141 -0
- package/dist/routes/attachments.js.map +1 -0
- package/dist/routes/demo.d.ts +3 -0
- package/dist/routes/demo.d.ts.map +1 -0
- package/dist/routes/demo.js +152 -0
- package/dist/routes/demo.js.map +1 -0
- package/dist/routes/files.d.ts +3 -0
- package/dist/routes/files.d.ts.map +1 -0
- package/dist/routes/files.js +313 -0
- package/dist/routes/files.js.map +1 -0
- package/dist/routes/filesystem.d.ts +3 -0
- package/dist/routes/filesystem.d.ts.map +1 -0
- package/dist/routes/filesystem.js +126 -0
- package/dist/routes/filesystem.js.map +1 -0
- package/dist/routes/git.d.ts +3 -0
- package/dist/routes/git.d.ts.map +1 -0
- package/dist/routes/git.js +170 -0
- package/dist/routes/git.js.map +1 -0
- package/dist/routes/index.d.ts +3 -0
- package/dist/routes/index.d.ts.map +1 -0
- package/dist/routes/index.js +45 -0
- package/dist/routes/index.js.map +1 -0
- package/dist/routes/notifications.d.ts +10 -0
- package/dist/routes/notifications.d.ts.map +1 -0
- package/dist/routes/notifications.js +54 -0
- package/dist/routes/notifications.js.map +1 -0
- package/dist/routes/profiles.d.ts +12 -0
- package/dist/routes/profiles.d.ts.map +1 -0
- package/dist/routes/profiles.js +76 -0
- package/dist/routes/profiles.js.map +1 -0
- package/dist/routes/projects.d.ts +3 -0
- package/dist/routes/projects.d.ts.map +1 -0
- package/dist/routes/projects.js +94 -0
- package/dist/routes/projects.js.map +1 -0
- package/dist/routes/sessions.d.ts +12 -0
- package/dist/routes/sessions.d.ts.map +1 -0
- package/dist/routes/sessions.js +109 -0
- package/dist/routes/sessions.js.map +1 -0
- package/dist/routes/system.d.ts +3 -0
- package/dist/routes/system.d.ts.map +1 -0
- package/dist/routes/system.js +42 -0
- package/dist/routes/system.js.map +1 -0
- package/dist/routes/tasks.d.ts +3 -0
- package/dist/routes/tasks.d.ts.map +1 -0
- package/dist/routes/tasks.js +136 -0
- package/dist/routes/tasks.js.map +1 -0
- package/dist/routes/terminals.d.ts +3 -0
- package/dist/routes/terminals.d.ts.map +1 -0
- package/dist/routes/terminals.js +54 -0
- package/dist/routes/terminals.js.map +1 -0
- package/dist/routes/tunnel.d.ts +3 -0
- package/dist/routes/tunnel.d.ts.map +1 -0
- package/dist/routes/tunnel.js +36 -0
- package/dist/routes/tunnel.js.map +1 -0
- package/dist/routes/workspaces.d.ts +3 -0
- package/dist/routes/workspaces.d.ts.map +1 -0
- package/dist/routes/workspaces.js +171 -0
- package/dist/routes/workspaces.js.map +1 -0
- package/dist/services/commit-message.service.d.ts +32 -0
- package/dist/services/commit-message.service.d.ts.map +1 -0
- package/dist/services/commit-message.service.js +202 -0
- package/dist/services/commit-message.service.js.map +1 -0
- package/dist/services/notifications/feishu-channel.d.ts +9 -0
- package/dist/services/notifications/feishu-channel.d.ts.map +1 -0
- package/dist/services/notifications/feishu-channel.js +51 -0
- package/dist/services/notifications/feishu-channel.js.map +1 -0
- package/dist/services/notifications/index.d.ts +3 -0
- package/dist/services/notifications/index.d.ts.map +1 -0
- package/dist/services/notifications/index.js +2 -0
- package/dist/services/notifications/index.js.map +1 -0
- package/dist/services/notifications/notification.service.d.ts +10 -0
- package/dist/services/notifications/notification.service.d.ts.map +1 -0
- package/dist/services/notifications/notification.service.js +85 -0
- package/dist/services/notifications/notification.service.js.map +1 -0
- package/dist/services/notifications/os-channel.d.ts +8 -0
- package/dist/services/notifications/os-channel.d.ts.map +1 -0
- package/dist/services/notifications/os-channel.js +38 -0
- package/dist/services/notifications/os-channel.js.map +1 -0
- package/dist/services/notifications/types.d.ts +11 -0
- package/dist/services/notifications/types.d.ts.map +1 -0
- package/dist/services/notifications/types.js +2 -0
- package/dist/services/notifications/types.js.map +1 -0
- package/dist/services/project.service.d.ts +91 -0
- package/dist/services/project.service.d.ts.map +1 -0
- package/dist/services/project.service.js +138 -0
- package/dist/services/project.service.js.map +1 -0
- package/dist/services/session-manager.d.ts +164 -0
- package/dist/services/session-manager.d.ts.map +1 -0
- package/dist/services/session-manager.js +569 -0
- package/dist/services/session-manager.js.map +1 -0
- package/dist/services/task.service.d.ts +179 -0
- package/dist/services/task.service.d.ts.map +1 -0
- package/dist/services/task.service.js +281 -0
- package/dist/services/task.service.js.map +1 -0
- package/dist/services/terminal-manager.d.ts +64 -0
- package/dist/services/terminal-manager.d.ts.map +1 -0
- package/dist/services/terminal-manager.js +217 -0
- package/dist/services/terminal-manager.js.map +1 -0
- package/dist/services/tunnel.service.d.ts +16 -0
- package/dist/services/tunnel.service.d.ts.map +1 -0
- package/dist/services/tunnel.service.js +77 -0
- package/dist/services/tunnel.service.js.map +1 -0
- package/dist/services/workspace.service.d.ts +219 -0
- package/dist/services/workspace.service.d.ts.map +1 -0
- package/dist/services/workspace.service.js +351 -0
- package/dist/services/workspace.service.js.map +1 -0
- package/dist/socket/events.d.ts +2 -0
- package/dist/socket/events.d.ts.map +1 -0
- package/dist/socket/events.js +3 -0
- package/dist/socket/events.js.map +1 -0
- package/dist/socket/index.d.ts +20 -0
- package/dist/socket/index.d.ts.map +1 -0
- package/dist/socket/index.js +77 -0
- package/dist/socket/index.js.map +1 -0
- package/dist/socket/middleware/auth.d.ts +13 -0
- package/dist/socket/middleware/auth.d.ts.map +1 -0
- package/dist/socket/middleware/auth.js +29 -0
- package/dist/socket/middleware/auth.js.map +1 -0
- package/dist/socket/middleware/error-handler.d.ts +11 -0
- package/dist/socket/middleware/error-handler.d.ts.map +1 -0
- package/dist/socket/middleware/error-handler.js +39 -0
- package/dist/socket/middleware/error-handler.js.map +1 -0
- package/dist/socket/middleware/index.d.ts +3 -0
- package/dist/socket/middleware/index.d.ts.map +1 -0
- package/dist/socket/middleware/index.js +3 -0
- package/dist/socket/middleware/index.js.map +1 -0
- package/dist/socket/rooms.d.ts +30 -0
- package/dist/socket/rooms.d.ts.map +1 -0
- package/dist/socket/rooms.js +39 -0
- package/dist/socket/rooms.js.map +1 -0
- package/dist/socket/socket-gateway.d.ts +21 -0
- package/dist/socket/socket-gateway.d.ts.map +1 -0
- package/dist/socket/socket-gateway.js +173 -0
- package/dist/socket/socket-gateway.js.map +1 -0
- package/dist/types/index.d.ts +41 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +41 -0
- package/dist/types/index.js.map +1 -0
- package/dist/utils/index.d.ts +12 -0
- package/dist/utils/index.d.ts.map +1 -0
- package/dist/utils/index.js +25 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/web/assets/AgentDemoPage-WdWEQtyU.js +1 -0
- package/dist/web/assets/DemoPage-C4Z_ftHI.js +4 -0
- package/dist/web/assets/NotificationSettingsPage-FrLfgt9y.js +1 -0
- package/dist/web/assets/ProfileSettingsPage-Cs93rTRr.js +3 -0
- package/dist/web/assets/ProjectKanbanPage-BmlRVfzo.js +75 -0
- package/dist/web/assets/ProjectKanbanPage-G5OQeJvU.css +1 -0
- package/dist/web/assets/api-client-CEBkg5lt.js +1 -0
- package/dist/web/assets/circle-alert-BEozlsMd.js +1 -0
- package/dist/web/assets/code-block-OCS4YCEC-CGgYYS4A.js +2 -0
- package/dist/web/assets/folder-picker-BP3x3phB.js +1 -0
- package/dist/web/assets/index-BuAijQvi.js +13 -0
- package/dist/web/assets/index-CxV866Yg.css +1 -0
- package/dist/web/assets/loader-circle-yEIqsdJ1.js +1 -0
- package/dist/web/assets/mermaid-NOHMQCX5-CIqtI2jF.js +123 -0
- package/dist/web/assets/mermaid-NOHMQCX5-lI9o3AWT.css +1 -0
- package/dist/web/assets/modal-BoX2RXaV.js +1 -0
- package/dist/web/assets/query-keys-BD_s_Etj.js +1 -0
- package/dist/web/assets/use-profiles-Rh8rBsAt.js +1 -0
- package/dist/web/index.html +14 -0
- package/dist/web/vite.svg +1 -0
- package/node_modules/@agent-tower/shared/dist/dev-port.d.ts +12 -0
- package/node_modules/@agent-tower/shared/dist/dev-port.d.ts.map +1 -0
- package/node_modules/@agent-tower/shared/dist/dev-port.js +31 -0
- package/node_modules/@agent-tower/shared/dist/dev-port.js.map +1 -0
- package/node_modules/@agent-tower/shared/dist/index.d.ts +5 -0
- package/node_modules/@agent-tower/shared/dist/index.d.ts.map +1 -0
- package/node_modules/@agent-tower/shared/dist/index.js +5 -0
- package/node_modules/@agent-tower/shared/dist/index.js.map +1 -0
- package/node_modules/@agent-tower/shared/dist/log-adapter.d.ts +72 -0
- package/node_modules/@agent-tower/shared/dist/log-adapter.d.ts.map +1 -0
- package/node_modules/@agent-tower/shared/dist/log-adapter.js +144 -0
- package/node_modules/@agent-tower/shared/dist/log-adapter.js.map +1 -0
- package/node_modules/@agent-tower/shared/dist/socket/events.d.ts +133 -0
- package/node_modules/@agent-tower/shared/dist/socket/events.d.ts.map +1 -0
- package/node_modules/@agent-tower/shared/dist/socket/events.js +32 -0
- package/node_modules/@agent-tower/shared/dist/socket/events.js.map +1 -0
- package/node_modules/@agent-tower/shared/dist/socket/index.d.ts +2 -0
- package/node_modules/@agent-tower/shared/dist/socket/index.d.ts.map +1 -0
- package/node_modules/@agent-tower/shared/dist/socket/index.js +2 -0
- package/node_modules/@agent-tower/shared/dist/socket/index.js.map +1 -0
- package/node_modules/@agent-tower/shared/dist/types.d.ts +168 -0
- package/node_modules/@agent-tower/shared/dist/types.d.ts.map +1 -0
- package/node_modules/@agent-tower/shared/dist/types.js +53 -0
- package/node_modules/@agent-tower/shared/dist/types.js.map +1 -0
- package/node_modules/@agent-tower/shared/package.json +38 -0
- package/package.json +46 -0
- package/prisma/schema.prisma +130 -0
- package/scripts/postinstall.js +120 -0
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
import { TunnelService } from '../services/tunnel.service.js';
|
|
2
|
+
/**
|
|
3
|
+
* 判断请求是否来自 Cloudflare 隧道
|
|
4
|
+
* cloudflared 代理请求时,Cloudflare 边缘会注入 CF-Connecting-IP / CF-Ray 等头
|
|
5
|
+
*/
|
|
6
|
+
function isTunnelRequest(request) {
|
|
7
|
+
return !!(request.headers['cf-connecting-ip'] || request.headers['cf-ray']);
|
|
8
|
+
}
|
|
9
|
+
/**
|
|
10
|
+
* 从请求中提取 token
|
|
11
|
+
* 支持 Authorization: Bearer <token> 和 ?token=<token> 两种方式
|
|
12
|
+
*/
|
|
13
|
+
function extractToken(request) {
|
|
14
|
+
const authHeader = request.headers.authorization;
|
|
15
|
+
if (authHeader?.startsWith('Bearer ')) {
|
|
16
|
+
return authHeader.slice(7);
|
|
17
|
+
}
|
|
18
|
+
const query = request.query;
|
|
19
|
+
return query?.token ?? null;
|
|
20
|
+
}
|
|
21
|
+
/**
|
|
22
|
+
* Fastify onRequest 钩子:隧道请求必须携带有效 token
|
|
23
|
+
* 本地直连请求(无 CF 头)自动放行
|
|
24
|
+
*/
|
|
25
|
+
export async function tunnelAuthHook(request, reply) {
|
|
26
|
+
if (!TunnelService.isRunning())
|
|
27
|
+
return;
|
|
28
|
+
if (!isTunnelRequest(request))
|
|
29
|
+
return;
|
|
30
|
+
const token = extractToken(request);
|
|
31
|
+
if (!token || !TunnelService.validateToken(token)) {
|
|
32
|
+
reply.code(401).send({
|
|
33
|
+
error: 'Unauthorized',
|
|
34
|
+
message: 'Valid tunnel token required',
|
|
35
|
+
});
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
export { isTunnelRequest };
|
|
39
|
+
//# sourceMappingURL=tunnel-auth.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tunnel-auth.js","sourceRoot":"","sources":["../../src/middleware/tunnel-auth.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAE9D;;;GAGG;AACH,SAAS,eAAe,CAAC,OAAuB;IAC9C,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;AAC9E,CAAC;AAED;;;GAGG;AACH,SAAS,YAAY,CAAC,OAAuB;IAC3C,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC;IACjD,IAAI,UAAU,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QACtC,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC7B,CAAC;IAED,MAAM,KAAK,GAAG,OAAO,CAAC,KAA+B,CAAC;IACtD,OAAO,KAAK,EAAE,KAAK,IAAI,IAAI,CAAC;AAC9B,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAuB,EACvB,KAAmB;IAEnB,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE;QAAE,OAAO;IACvC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC;QAAE,OAAO;IAEtC,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;IACpC,IAAI,CAAC,KAAK,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;QAClD,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;YACnB,KAAK,EAAE,cAAc;YACrB,OAAO,EAAE,6BAA6B;SACvC,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,OAAO,EAAE,eAAe,EAAE,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"claude-code-parser-token.test.d.ts","sourceRoot":"","sources":["../../../src/output/__tests__/claude-code-parser-token.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,233 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ClaudeCodeParser Token 提取测试
|
|
3
|
+
*/
|
|
4
|
+
import { describe, it, expect } from 'vitest';
|
|
5
|
+
import { MsgStore } from '../msg-store.js';
|
|
6
|
+
import { ClaudeCodeParser } from '../claude-code-parser.js';
|
|
7
|
+
function feedLine(parser, obj) {
|
|
8
|
+
parser.processData(JSON.stringify(obj) + '\n');
|
|
9
|
+
}
|
|
10
|
+
function getTokenEntries(store) {
|
|
11
|
+
const snap = store.getSnapshot();
|
|
12
|
+
return snap.entries.filter((e) => e.entryType === 'token_usage_info');
|
|
13
|
+
}
|
|
14
|
+
describe('Claude Code Token - 使用 assistant 消息的 per-turn usage', () => {
|
|
15
|
+
it('should use last assistant message usage (not cumulative result usage)', () => {
|
|
16
|
+
const store = new MsgStore();
|
|
17
|
+
const parser = new ClaudeCodeParser(store);
|
|
18
|
+
// 模拟多轮对话:assistant 消息携带 per-turn usage
|
|
19
|
+
feedLine(parser, {
|
|
20
|
+
type: 'assistant',
|
|
21
|
+
message: {
|
|
22
|
+
id: 'msg-1',
|
|
23
|
+
role: 'assistant',
|
|
24
|
+
content: [{ type: 'text', text: 'hello' }],
|
|
25
|
+
usage: {
|
|
26
|
+
input_tokens: 35000,
|
|
27
|
+
output_tokens: 500,
|
|
28
|
+
cache_creation_input_tokens: 0,
|
|
29
|
+
cache_read_input_tokens: 0,
|
|
30
|
+
},
|
|
31
|
+
},
|
|
32
|
+
});
|
|
33
|
+
// 第二轮 assistant(上下文增长)
|
|
34
|
+
feedLine(parser, {
|
|
35
|
+
type: 'assistant',
|
|
36
|
+
message: {
|
|
37
|
+
id: 'msg-2',
|
|
38
|
+
role: 'assistant',
|
|
39
|
+
content: [{ type: 'text', text: 'world' }],
|
|
40
|
+
usage: {
|
|
41
|
+
input_tokens: 2000,
|
|
42
|
+
output_tokens: 300,
|
|
43
|
+
cache_creation_input_tokens: 5000,
|
|
44
|
+
cache_read_input_tokens: 40000,
|
|
45
|
+
},
|
|
46
|
+
},
|
|
47
|
+
});
|
|
48
|
+
// result 消息(usage 是累计值,不应该使用)
|
|
49
|
+
feedLine(parser, {
|
|
50
|
+
type: 'result',
|
|
51
|
+
subtype: 'success',
|
|
52
|
+
usage: {
|
|
53
|
+
input_tokens: 999999, // 累计值,很大
|
|
54
|
+
output_tokens: 999999,
|
|
55
|
+
},
|
|
56
|
+
modelUsage: {
|
|
57
|
+
'claude-opus-4.6': {
|
|
58
|
+
inputTokens: 999999,
|
|
59
|
+
outputTokens: 999999,
|
|
60
|
+
contextWindow: 200000,
|
|
61
|
+
},
|
|
62
|
+
},
|
|
63
|
+
});
|
|
64
|
+
const entries = getTokenEntries(store);
|
|
65
|
+
expect(entries).toHaveLength(1);
|
|
66
|
+
const tu = entries[0].metadata.tokenUsage;
|
|
67
|
+
// 应该使用最后一条 assistant 消息的 per-turn usage:
|
|
68
|
+
// input_tokens(2000) + cache_creation(5000) + cache_read(40000) = 47000
|
|
69
|
+
expect(tu.totalTokens).toBe(47000);
|
|
70
|
+
expect(tu.modelContextWindow).toBe(200000);
|
|
71
|
+
});
|
|
72
|
+
it('should compute totalTokens as input + cache_creation + cache_read (matching Claude Code statusline)', () => {
|
|
73
|
+
const store = new MsgStore();
|
|
74
|
+
const parser = new ClaudeCodeParser(store);
|
|
75
|
+
feedLine(parser, {
|
|
76
|
+
type: 'assistant',
|
|
77
|
+
message: {
|
|
78
|
+
id: 'msg-1',
|
|
79
|
+
role: 'assistant',
|
|
80
|
+
content: [{ type: 'text', text: 'hi' }],
|
|
81
|
+
usage: {
|
|
82
|
+
input_tokens: 5000,
|
|
83
|
+
output_tokens: 200,
|
|
84
|
+
cache_creation_input_tokens: 10000,
|
|
85
|
+
cache_read_input_tokens: 30000,
|
|
86
|
+
},
|
|
87
|
+
},
|
|
88
|
+
});
|
|
89
|
+
feedLine(parser, {
|
|
90
|
+
type: 'result',
|
|
91
|
+
subtype: 'success',
|
|
92
|
+
modelUsage: {
|
|
93
|
+
'claude-opus-4.6': { contextWindow: 200000 },
|
|
94
|
+
},
|
|
95
|
+
});
|
|
96
|
+
const entries = getTokenEntries(store);
|
|
97
|
+
expect(entries).toHaveLength(1);
|
|
98
|
+
const tu = entries[0].metadata.tokenUsage;
|
|
99
|
+
// 5000 + 10000 + 30000 = 45000 (不含 output_tokens)
|
|
100
|
+
expect(tu.totalTokens).toBe(45000);
|
|
101
|
+
});
|
|
102
|
+
});
|
|
103
|
+
describe('Claude Code Token - contextWindow 提取', () => {
|
|
104
|
+
it('should extract contextWindow from modelUsage (camelCase)', () => {
|
|
105
|
+
const store = new MsgStore();
|
|
106
|
+
const parser = new ClaudeCodeParser(store);
|
|
107
|
+
feedLine(parser, {
|
|
108
|
+
type: 'assistant',
|
|
109
|
+
message: {
|
|
110
|
+
id: 'msg-1',
|
|
111
|
+
role: 'assistant',
|
|
112
|
+
content: [{ type: 'text', text: 'hi' }],
|
|
113
|
+
usage: { input_tokens: 100, output_tokens: 10 },
|
|
114
|
+
},
|
|
115
|
+
});
|
|
116
|
+
feedLine(parser, {
|
|
117
|
+
type: 'result',
|
|
118
|
+
subtype: 'success',
|
|
119
|
+
modelUsage: {
|
|
120
|
+
'claude-opus-4.6': { contextWindow: 200000 },
|
|
121
|
+
},
|
|
122
|
+
});
|
|
123
|
+
const entries = getTokenEntries(store);
|
|
124
|
+
expect(entries).toHaveLength(1);
|
|
125
|
+
expect(entries[0].metadata.tokenUsage.modelContextWindow).toBe(200000);
|
|
126
|
+
});
|
|
127
|
+
it('should fallback to model_usage (snake_case) for contextWindow', () => {
|
|
128
|
+
const store = new MsgStore();
|
|
129
|
+
const parser = new ClaudeCodeParser(store);
|
|
130
|
+
feedLine(parser, {
|
|
131
|
+
type: 'assistant',
|
|
132
|
+
message: {
|
|
133
|
+
id: 'msg-1',
|
|
134
|
+
role: 'assistant',
|
|
135
|
+
content: [{ type: 'text', text: 'hi' }],
|
|
136
|
+
usage: { input_tokens: 100, output_tokens: 50 },
|
|
137
|
+
},
|
|
138
|
+
});
|
|
139
|
+
feedLine(parser, {
|
|
140
|
+
type: 'result',
|
|
141
|
+
subtype: 'success',
|
|
142
|
+
model_usage: {
|
|
143
|
+
'claude-sonnet-4-20250514': { context_window: 200000 },
|
|
144
|
+
},
|
|
145
|
+
});
|
|
146
|
+
const entries = getTokenEntries(store);
|
|
147
|
+
expect(entries).toHaveLength(1);
|
|
148
|
+
expect(entries[0].metadata.tokenUsage.modelContextWindow).toBe(200000);
|
|
149
|
+
});
|
|
150
|
+
});
|
|
151
|
+
describe('Claude Code Token - 回退和边界用例', () => {
|
|
152
|
+
it('should fallback to result.usage when no assistant message has usage', () => {
|
|
153
|
+
const store = new MsgStore();
|
|
154
|
+
const parser = new ClaudeCodeParser(store);
|
|
155
|
+
feedLine(parser, {
|
|
156
|
+
type: 'result',
|
|
157
|
+
subtype: 'success',
|
|
158
|
+
usage: {
|
|
159
|
+
input_tokens: 100,
|
|
160
|
+
output_tokens: 50,
|
|
161
|
+
cache_creation_input_tokens: 20,
|
|
162
|
+
cache_read_input_tokens: 30,
|
|
163
|
+
},
|
|
164
|
+
});
|
|
165
|
+
const entries = getTokenEntries(store);
|
|
166
|
+
expect(entries).toHaveLength(1);
|
|
167
|
+
const tu = entries[0].metadata.tokenUsage;
|
|
168
|
+
expect(tu.totalTokens).toBe(200);
|
|
169
|
+
});
|
|
170
|
+
it('should not generate token_usage_info when result has no data', () => {
|
|
171
|
+
const store = new MsgStore();
|
|
172
|
+
const parser = new ClaudeCodeParser(store);
|
|
173
|
+
feedLine(parser, {
|
|
174
|
+
type: 'result',
|
|
175
|
+
subtype: 'success',
|
|
176
|
+
});
|
|
177
|
+
const entries = getTokenEntries(store);
|
|
178
|
+
expect(entries).toHaveLength(0);
|
|
179
|
+
});
|
|
180
|
+
it('should not extract token usage from tool_result subtype', () => {
|
|
181
|
+
const store = new MsgStore();
|
|
182
|
+
const parser = new ClaudeCodeParser(store);
|
|
183
|
+
feedLine(parser, {
|
|
184
|
+
type: 'result',
|
|
185
|
+
subtype: 'tool_result',
|
|
186
|
+
tool_use_id: 'tool-1',
|
|
187
|
+
usage: { input_tokens: 100, output_tokens: 50 },
|
|
188
|
+
});
|
|
189
|
+
const entries = getTokenEntries(store);
|
|
190
|
+
expect(entries).toHaveLength(0);
|
|
191
|
+
});
|
|
192
|
+
it('should not throw on malformed usage fields', () => {
|
|
193
|
+
const store = new MsgStore();
|
|
194
|
+
const parser = new ClaudeCodeParser(store);
|
|
195
|
+
feedLine(parser, {
|
|
196
|
+
type: 'result',
|
|
197
|
+
subtype: 'success',
|
|
198
|
+
usage: 'not-an-object',
|
|
199
|
+
});
|
|
200
|
+
// Parser should still work
|
|
201
|
+
feedLine(parser, {
|
|
202
|
+
type: 'assistant',
|
|
203
|
+
message: {
|
|
204
|
+
id: 'msg-1',
|
|
205
|
+
role: 'assistant',
|
|
206
|
+
content: [{ type: 'text', text: 'hello' }],
|
|
207
|
+
},
|
|
208
|
+
});
|
|
209
|
+
});
|
|
210
|
+
it('should handle assistant message without usage field', () => {
|
|
211
|
+
const store = new MsgStore();
|
|
212
|
+
const parser = new ClaudeCodeParser(store);
|
|
213
|
+
feedLine(parser, {
|
|
214
|
+
type: 'assistant',
|
|
215
|
+
message: {
|
|
216
|
+
id: 'msg-1',
|
|
217
|
+
role: 'assistant',
|
|
218
|
+
content: [{ type: 'text', text: 'hello' }],
|
|
219
|
+
// no usage field
|
|
220
|
+
},
|
|
221
|
+
});
|
|
222
|
+
feedLine(parser, {
|
|
223
|
+
type: 'result',
|
|
224
|
+
subtype: 'success',
|
|
225
|
+
usage: { input_tokens: 100, output_tokens: 50 },
|
|
226
|
+
});
|
|
227
|
+
const entries = getTokenEntries(store);
|
|
228
|
+
expect(entries).toHaveLength(1);
|
|
229
|
+
// Falls back to result.usage
|
|
230
|
+
expect(entries[0].metadata.tokenUsage.totalTokens).toBe(150);
|
|
231
|
+
});
|
|
232
|
+
});
|
|
233
|
+
//# sourceMappingURL=claude-code-parser-token.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"claude-code-parser-token.test.js","sourceRoot":"","sources":["../../../src/output/__tests__/claude-code-parser-token.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAA;AAC1C,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAA;AAE3D,SAAS,QAAQ,CAAC,MAAwB,EAAE,GAA4B;IACtE,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAA;AAChD,CAAC;AAED,SAAS,eAAe,CAAC,KAAe;IACtC,MAAM,IAAI,GAAG,KAAK,CAAC,WAAW,EAAE,CAAA;IAChC,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,kBAAkB,CAAC,CAAA;AACvE,CAAC;AAED,QAAQ,CAAC,qDAAqD,EAAE,GAAG,EAAE;IACnE,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAE1C,uCAAuC;QACvC,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,KAAK,EAAE;oBACL,YAAY,EAAE,KAAK;oBACnB,aAAa,EAAE,GAAG;oBAClB,2BAA2B,EAAE,CAAC;oBAC9B,uBAAuB,EAAE,CAAC;iBAC3B;aACF;SACF,CAAC,CAAA;QAEF,uBAAuB;QACvB,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,KAAK,EAAE;oBACL,YAAY,EAAE,IAAI;oBAClB,aAAa,EAAE,GAAG;oBAClB,2BAA2B,EAAE,IAAI;oBACjC,uBAAuB,EAAE,KAAK;iBAC/B;aACF;SACF,CAAC,CAAA;QAEF,8BAA8B;QAC9B,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,SAAS;YAClB,KAAK,EAAE;gBACL,YAAY,EAAE,MAAM,EAAG,SAAS;gBAChC,aAAa,EAAE,MAAM;aACtB;YACD,UAAU,EAAE;gBACV,iBAAiB,EAAE;oBACjB,WAAW,EAAE,MAAM;oBACnB,YAAY,EAAE,MAAM;oBACpB,aAAa,EAAE,MAAM;iBACtB;aACF;SACF,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAE/B,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,QAAS,CAAC,UAAW,CAAA;QAC3C,yCAAyC;QACzC,wEAAwE;QACxE,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAClC,MAAM,CAAC,EAAE,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qGAAqG,EAAE,GAAG,EAAE;QAC7G,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAE1C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBACvC,KAAK,EAAE;oBACL,YAAY,EAAE,IAAI;oBAClB,aAAa,EAAE,GAAG;oBAClB,2BAA2B,EAAE,KAAK;oBAClC,uBAAuB,EAAE,KAAK;iBAC/B;aACF;SACF,CAAC,CAAA;QAEF,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,SAAS;YAClB,UAAU,EAAE;gBACV,iBAAiB,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;aAC7C;SACF,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,QAAS,CAAC,UAAW,CAAA;QAC3C,kDAAkD;QAClD,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACpC,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAE1C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBACvC,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,aAAa,EAAE,EAAE,EAAE;aAChD;SACF,CAAC,CAAA;QAEF,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,SAAS;YAClB,UAAU,EAAE;gBACV,iBAAiB,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;aAC7C;SACF,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAS,CAAC,UAAW,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IAC1E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAE1C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBACvC,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,aAAa,EAAE,EAAE,EAAE;aAChD;SACF,CAAC,CAAA;QAEF,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,SAAS;YAClB,WAAW,EAAE;gBACX,0BAA0B,EAAE,EAAE,cAAc,EAAE,MAAM,EAAE;aACvD;SACF,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAS,CAAC,UAAW,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IAC1E,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAE1C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,SAAS;YAClB,KAAK,EAAE;gBACL,YAAY,EAAE,GAAG;gBACjB,aAAa,EAAE,EAAE;gBACjB,2BAA2B,EAAE,EAAE;gBAC/B,uBAAuB,EAAE,EAAE;aAC5B;SACF,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,QAAS,CAAC,UAAW,CAAA;QAC3C,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IAClC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAE1C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,SAAS;SACnB,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAE1C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,aAAa;YACtB,WAAW,EAAE,QAAQ;YACrB,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,aAAa,EAAE,EAAE,EAAE;SAChD,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAE1C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,SAAS;YAClB,KAAK,EAAE,eAA0B;SAClC,CAAC,CAAA;QAEF,2BAA2B;QAC3B,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC3C;SACF,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAE1C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE;gBACP,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,iBAAiB;aAClB;SACF,CAAC,CAAA;QAEF,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,SAAS;YAClB,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,aAAa,EAAE,EAAE,EAAE;SAChD,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,6BAA6B;QAC7B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAS,CAAC,UAAW,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IAChE,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cursor-agent-parser-token.test.d.ts","sourceRoot":"","sources":["../../../src/output/__tests__/cursor-agent-parser-token.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* CursorAgentParser Token 提取测试
|
|
3
|
+
* Property 2: Cursor Agent Token 提取正确性
|
|
4
|
+
*/
|
|
5
|
+
import { describe, it, expect } from 'vitest';
|
|
6
|
+
import fc from 'fast-check';
|
|
7
|
+
import { MsgStore } from '../msg-store.js';
|
|
8
|
+
import { CursorAgentParser } from '../cursor-agent-parser.js';
|
|
9
|
+
function feedLine(parser, obj) {
|
|
10
|
+
parser.processData(JSON.stringify(obj) + '\n');
|
|
11
|
+
}
|
|
12
|
+
function getTokenEntries(store) {
|
|
13
|
+
const snap = store.getSnapshot();
|
|
14
|
+
return snap.entries.filter((e) => e.entryType === 'token_usage_info');
|
|
15
|
+
}
|
|
16
|
+
describe('Feature: token-usage-display, Property 2: Cursor Agent Token 提取正确性', () => {
|
|
17
|
+
it('should compute totalTokens as sum of all usage fields', () => {
|
|
18
|
+
fc.assert(fc.property(fc.record({
|
|
19
|
+
input_tokens: fc.nat(),
|
|
20
|
+
output_tokens: fc.nat(),
|
|
21
|
+
cache_creation_input_tokens: fc.nat(),
|
|
22
|
+
cache_read_input_tokens: fc.nat(),
|
|
23
|
+
}), (usage) => {
|
|
24
|
+
const store = new MsgStore();
|
|
25
|
+
const parser = new CursorAgentParser(store);
|
|
26
|
+
feedLine(parser, {
|
|
27
|
+
type: 'result',
|
|
28
|
+
result: { usage },
|
|
29
|
+
});
|
|
30
|
+
const entries = getTokenEntries(store);
|
|
31
|
+
expect(entries).toHaveLength(1);
|
|
32
|
+
const tu = entries[0].metadata.tokenUsage;
|
|
33
|
+
const expectedTotal = usage.input_tokens +
|
|
34
|
+
usage.output_tokens +
|
|
35
|
+
usage.cache_creation_input_tokens +
|
|
36
|
+
usage.cache_read_input_tokens;
|
|
37
|
+
expect(tu.totalTokens).toBe(expectedTotal);
|
|
38
|
+
}), { numRuns: 100 });
|
|
39
|
+
});
|
|
40
|
+
it('should support camelCase token field names (inputTokens/outputTokens)', () => {
|
|
41
|
+
const store = new MsgStore();
|
|
42
|
+
const parser = new CursorAgentParser(store);
|
|
43
|
+
feedLine(parser, {
|
|
44
|
+
type: 'result',
|
|
45
|
+
result: {
|
|
46
|
+
usage: {
|
|
47
|
+
inputTokens: 500,
|
|
48
|
+
outputTokens: 200,
|
|
49
|
+
},
|
|
50
|
+
},
|
|
51
|
+
});
|
|
52
|
+
const entries = getTokenEntries(store);
|
|
53
|
+
expect(entries).toHaveLength(1);
|
|
54
|
+
const tu = entries[0].metadata.tokenUsage;
|
|
55
|
+
expect(tu.totalTokens).toBe(700);
|
|
56
|
+
});
|
|
57
|
+
});
|
|
58
|
+
describe('CursorAgentParser Token 边界用例', () => {
|
|
59
|
+
it('should not generate token_usage_info when result has no usage data (Req 2.2)', () => {
|
|
60
|
+
const store = new MsgStore();
|
|
61
|
+
const parser = new CursorAgentParser(store);
|
|
62
|
+
feedLine(parser, {
|
|
63
|
+
type: 'result',
|
|
64
|
+
result: { duration_ms: 1234 },
|
|
65
|
+
});
|
|
66
|
+
const entries = getTokenEntries(store);
|
|
67
|
+
expect(entries).toHaveLength(0);
|
|
68
|
+
});
|
|
69
|
+
it('should not generate token_usage_info when result is null', () => {
|
|
70
|
+
const store = new MsgStore();
|
|
71
|
+
const parser = new CursorAgentParser(store);
|
|
72
|
+
feedLine(parser, {
|
|
73
|
+
type: 'result',
|
|
74
|
+
result: null,
|
|
75
|
+
});
|
|
76
|
+
const entries = getTokenEntries(store);
|
|
77
|
+
expect(entries).toHaveLength(0);
|
|
78
|
+
});
|
|
79
|
+
it('should not throw on malformed result', () => {
|
|
80
|
+
const store = new MsgStore();
|
|
81
|
+
const parser = new CursorAgentParser(store);
|
|
82
|
+
// Should not throw
|
|
83
|
+
feedLine(parser, {
|
|
84
|
+
type: 'result',
|
|
85
|
+
result: 'not an object',
|
|
86
|
+
});
|
|
87
|
+
feedLine(parser, {
|
|
88
|
+
type: 'result',
|
|
89
|
+
result: { usage: 'not an object' },
|
|
90
|
+
});
|
|
91
|
+
const entries = getTokenEntries(store);
|
|
92
|
+
expect(entries).toHaveLength(0);
|
|
93
|
+
});
|
|
94
|
+
});
|
|
95
|
+
//# sourceMappingURL=cursor-agent-parser-token.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cursor-agent-parser-token.test.js","sourceRoot":"","sources":["../../../src/output/__tests__/cursor-agent-parser-token.test.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,MAAM,YAAY,CAAA;AAC3B,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAA;AAC1C,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAA;AAE7D,SAAS,QAAQ,CAAC,MAAyB,EAAE,GAA4B;IACvE,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAA;AAChD,CAAC;AAED,SAAS,eAAe,CAAC,KAAe;IACtC,MAAM,IAAI,GAAG,KAAK,CAAC,WAAW,EAAE,CAAA;IAChC,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,kBAAkB,CAAC,CAAA;AACvE,CAAC;AAED,QAAQ,CAAC,oEAAoE,EAAE,GAAG,EAAE;IAClF,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,EAAE,CAAC,MAAM,CACP,EAAE,CAAC,QAAQ,CACT,EAAE,CAAC,MAAM,CAAC;YACR,YAAY,EAAE,EAAE,CAAC,GAAG,EAAE;YACtB,aAAa,EAAE,EAAE,CAAC,GAAG,EAAE;YACvB,2BAA2B,EAAE,EAAE,CAAC,GAAG,EAAE;YACrC,uBAAuB,EAAE,EAAE,CAAC,GAAG,EAAE;SAClC,CAAC,EACF,CAAC,KAAK,EAAE,EAAE;YACR,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;YAC5B,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,KAAK,CAAC,CAAA;YAE3C,QAAQ,CAAC,MAAM,EAAE;gBACf,IAAI,EAAE,QAAQ;gBACd,MAAM,EAAE,EAAE,KAAK,EAAE;aAClB,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;YACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAE/B,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,QAAS,CAAC,UAAW,CAAA;YAC3C,MAAM,aAAa,GACjB,KAAK,CAAC,YAAY;gBAClB,KAAK,CAAC,aAAa;gBACnB,KAAK,CAAC,2BAA2B;gBACjC,KAAK,CAAC,uBAAuB,CAAA;YAC/B,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;QAC5C,CAAC,CACF,EACD,EAAE,OAAO,EAAE,GAAG,EAAE,CACjB,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,KAAK,CAAC,CAAA;QAE3C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE;gBACN,KAAK,EAAE;oBACL,WAAW,EAAE,GAAG;oBAChB,YAAY,EAAE,GAAG;iBAClB;aACF;SACF,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,QAAS,CAAC,UAAW,CAAA;QAC3C,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IAClC,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC5C,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;QACtF,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,KAAK,CAAC,CAAA;QAE3C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;SAC9B,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,KAAK,CAAC,CAAA;QAE3C,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,IAAI;SACb,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,KAAK,CAAC,CAAA;QAE3C,mBAAmB;QACnB,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,eAAe;SACxB,CAAC,CAAA;QAEF,QAAQ,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE;SACnC,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"msg-store-token.test.d.ts","sourceRoot":"","sources":["../../../src/output/__tests__/msg-store-token.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* MsgStore Token 往返一致性测试
|
|
3
|
+
* Property 3: Token 用量数据 MsgStore 往返一致性
|
|
4
|
+
*/
|
|
5
|
+
import { describe, it, expect } from 'vitest';
|
|
6
|
+
import fc from 'fast-check';
|
|
7
|
+
import { MsgStore } from '../msg-store.js';
|
|
8
|
+
import { createTokenUsageInfo } from '../types.js';
|
|
9
|
+
import { addNormalizedEntry } from '../utils/patch.js';
|
|
10
|
+
describe('Feature: token-usage-display, Property 3: Token 用量数据 MsgStore 往返一致性', () => {
|
|
11
|
+
it('should preserve token_usage_info through MsgStore round-trip', () => {
|
|
12
|
+
fc.assert(fc.property(fc.nat(), fc.option(fc.nat(), { nil: undefined }), (totalTokens, modelContextWindow) => {
|
|
13
|
+
const store = new MsgStore();
|
|
14
|
+
const entry = createTokenUsageInfo(totalTokens, modelContextWindow);
|
|
15
|
+
const index = store.entryIndex.next();
|
|
16
|
+
const patch = addNormalizedEntry(index, entry);
|
|
17
|
+
store.pushPatch(patch);
|
|
18
|
+
const snapshot = store.getSnapshot();
|
|
19
|
+
const recovered = snapshot.entries[index];
|
|
20
|
+
expect(recovered.entryType).toBe('token_usage_info');
|
|
21
|
+
expect(recovered.metadata.tokenUsage.totalTokens).toBe(totalTokens);
|
|
22
|
+
expect(recovered.metadata.tokenUsage.modelContextWindow).toBe(modelContextWindow);
|
|
23
|
+
}), { numRuns: 100 });
|
|
24
|
+
});
|
|
25
|
+
it('should preserve zero values in tokenUsage (Req 6.2)', () => {
|
|
26
|
+
const store = new MsgStore();
|
|
27
|
+
const entry = createTokenUsageInfo(0, 200000);
|
|
28
|
+
const index = store.entryIndex.next();
|
|
29
|
+
const patch = addNormalizedEntry(index, entry);
|
|
30
|
+
store.pushPatch(patch);
|
|
31
|
+
const snapshot = store.getSnapshot();
|
|
32
|
+
const recovered = snapshot.entries[index];
|
|
33
|
+
const tu = recovered.metadata.tokenUsage;
|
|
34
|
+
expect(tu.totalTokens).toBe(0);
|
|
35
|
+
expect(tu.modelContextWindow).toBe(200000);
|
|
36
|
+
});
|
|
37
|
+
});
|
|
38
|
+
//# sourceMappingURL=msg-store-token.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"msg-store-token.test.js","sourceRoot":"","sources":["../../../src/output/__tests__/msg-store-token.test.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,MAAM,YAAY,CAAA;AAC3B,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAA;AAC1C,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAA;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAA;AAEtD,QAAQ,CAAC,qEAAqE,EAAE,GAAG,EAAE;IACnF,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,EAAE,CAAC,MAAM,CACP,EAAE,CAAC,QAAQ,CACT,EAAE,CAAC,GAAG,EAAE,EACR,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EACvC,CAAC,WAAW,EAAE,kBAAkB,EAAE,EAAE;YAClC,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;YAC5B,MAAM,KAAK,GAAG,oBAAoB,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAA;YACnE,MAAM,KAAK,GAAG,KAAK,CAAC,UAAU,CAAC,IAAI,EAAE,CAAA;YACrC,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAA;YAC9C,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAA;YAEtB,MAAM,QAAQ,GAAG,KAAK,CAAC,WAAW,EAAE,CAAA;YACpC,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;YAEzC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;YACpD,MAAM,CAAC,SAAS,CAAC,QAAS,CAAC,UAAW,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;YACrE,MAAM,CAAC,SAAS,CAAC,QAAS,CAAC,UAAW,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;QACrF,CAAC,CACF,EACD,EAAE,OAAO,EAAE,GAAG,EAAE,CACjB,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,KAAK,GAAG,IAAI,QAAQ,EAAE,CAAA;QAC5B,MAAM,KAAK,GAAG,oBAAoB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAA;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,UAAU,CAAC,IAAI,EAAE,CAAA;QACrC,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAA;QAC9C,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAA;QAEtB,MAAM,QAAQ,GAAG,KAAK,CAAC,WAAW,EAAE,CAAA;QACpC,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QACzC,MAAM,EAAE,GAAG,SAAS,CAAC,QAAS,CAAC,UAAW,CAAA;QAE1C,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC9B,MAAM,CAAC,EAAE,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1,96 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Claude Code stream-json 输出解析器
|
|
3
|
+
* 将 Claude Code 的 JSON 流输出转换为标准化日志
|
|
4
|
+
*/
|
|
5
|
+
import type { MsgStore } from './msg-store.js';
|
|
6
|
+
/**
|
|
7
|
+
* Claude Code 解析器
|
|
8
|
+
*/
|
|
9
|
+
export declare class ClaudeCodeParser {
|
|
10
|
+
private msgStore;
|
|
11
|
+
private buffer;
|
|
12
|
+
private toolEntryMap;
|
|
13
|
+
private currentAssistantIndex;
|
|
14
|
+
private currentAssistantContent;
|
|
15
|
+
private streamingBlocks;
|
|
16
|
+
private streamingRole;
|
|
17
|
+
private streamingMessages;
|
|
18
|
+
private streamingMessageId;
|
|
19
|
+
private lastTurnUsage;
|
|
20
|
+
constructor(msgStore: MsgStore);
|
|
21
|
+
/** 使用 MsgStore 共享的索引提供器 */
|
|
22
|
+
private get indexProvider();
|
|
23
|
+
/**
|
|
24
|
+
* 处理原始输出数据
|
|
25
|
+
*/
|
|
26
|
+
processData(data: string): void;
|
|
27
|
+
/**
|
|
28
|
+
* 解析单行 JSON
|
|
29
|
+
*/
|
|
30
|
+
private parseLine;
|
|
31
|
+
/**
|
|
32
|
+
* 处理消息
|
|
33
|
+
*/
|
|
34
|
+
private handleMessage;
|
|
35
|
+
/**
|
|
36
|
+
* 处理系统消息
|
|
37
|
+
*/
|
|
38
|
+
private handleSystemMessage;
|
|
39
|
+
/**
|
|
40
|
+
* 处理助手消息
|
|
41
|
+
* 通过 message.id 查找 stream_event 阶段创建的 entry,用完整内容 replace
|
|
42
|
+
* 对于 tool_use(不走 stream_event),直接新建 entry
|
|
43
|
+
*/
|
|
44
|
+
private handleAssistantMessage;
|
|
45
|
+
/**
|
|
46
|
+
* 追加助手文本
|
|
47
|
+
*/
|
|
48
|
+
private appendAssistantText;
|
|
49
|
+
/**
|
|
50
|
+
* 刷新助手文本
|
|
51
|
+
*/
|
|
52
|
+
private flushAssistantText;
|
|
53
|
+
/**
|
|
54
|
+
* 处理工具使用
|
|
55
|
+
*/
|
|
56
|
+
private handleToolUse;
|
|
57
|
+
/**
|
|
58
|
+
* 格式化工具内容
|
|
59
|
+
*/
|
|
60
|
+
private formatToolContent;
|
|
61
|
+
/**
|
|
62
|
+
* 从 TodoWrite 工具调用中提取 todo 数据
|
|
63
|
+
*/
|
|
64
|
+
private extractTodoExtras;
|
|
65
|
+
/**
|
|
66
|
+
* 添加思考内容
|
|
67
|
+
*/
|
|
68
|
+
private addThinking;
|
|
69
|
+
/**
|
|
70
|
+
* 处理用户消息
|
|
71
|
+
* 跳过 — 用户消息已由 session.service.ts 在收到请求时主动推送,
|
|
72
|
+
* 这里如果再处理 Claude Code 回显的 user message 会导致重复
|
|
73
|
+
*/
|
|
74
|
+
private handleUserMessage;
|
|
75
|
+
/**
|
|
76
|
+
* 处理结果消息
|
|
77
|
+
*/
|
|
78
|
+
private handleResultMessage;
|
|
79
|
+
/**
|
|
80
|
+
* 处理错误消息
|
|
81
|
+
*/
|
|
82
|
+
private handleErrorMessage;
|
|
83
|
+
/**
|
|
84
|
+
* 处理 stream_event(增量流式消息)
|
|
85
|
+
*/
|
|
86
|
+
private handleStreamEvent;
|
|
87
|
+
/**
|
|
88
|
+
* 完成解析
|
|
89
|
+
*/
|
|
90
|
+
finish(): void;
|
|
91
|
+
}
|
|
92
|
+
/**
|
|
93
|
+
* 创建解析器并连接到 PTY
|
|
94
|
+
*/
|
|
95
|
+
export declare function createClaudeCodeParser(msgStore: MsgStore): ClaudeCodeParser;
|
|
96
|
+
//# sourceMappingURL=claude-code-parser.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"claude-code-parser.d.ts","sourceRoot":"","sources":["../../src/output/claude-code-parser.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAA;AA6H9C;;GAEG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAU;IAC1B,OAAO,CAAC,MAAM,CAAa;IAC3B,OAAO,CAAC,YAAY,CAAiC;IACrD,OAAO,CAAC,qBAAqB,CAAsB;IACnD,OAAO,CAAC,uBAAuB,CAAa;IAE5C,OAAO,CAAC,eAAe,CAAgD;IACvE,OAAO,CAAC,aAAa,CAAsB;IAE3C,OAAO,CAAC,iBAAiB,CAAgD;IACzE,OAAO,CAAC,kBAAkB,CAAsB;IAEhD,OAAO,CAAC,aAAa,CAAqI;gBAE9I,QAAQ,EAAE,QAAQ;IAI9B,2BAA2B;IAC3B,OAAO,KAAK,aAAa,GAExB;IAED;;OAEG;IACH,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAqB/B;;OAEG;IACH,OAAO,CAAC,SAAS;IAiBjB;;OAEG;IACH,OAAO,CAAC,aAAa;IAuBrB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAQ3B;;;;OAIG;IACH,OAAO,CAAC,sBAAsB;IAoD9B;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAiB3B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAK1B;;OAEG;IACH,OAAO,CAAC,aAAa;IAyBrB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA+BzB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAezB;;OAEG;IACH,OAAO,CAAC,WAAW;IAOnB;;;;OAIG;IACH,OAAO,CAAC,iBAAiB;IAIzB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IA0D3B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAO1B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAmHzB;;OAEG;IACH,MAAM,IAAI,IAAI;CAOf;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,QAAQ,GAAG,gBAAgB,CAE3E"}
|