import { z } from 'zod';
import { ToolDefinition, ToolContext } from './types.js';
/**
* 跟踪对象分配工具定义
*/
export const trackAllocationsTool: ToolDefinition = {
name: 'track_allocations',
description: '跟踪对象分配',
inputSchema: z.object({
url: z.string().optional().describe('页面 URL（可选）'),
duration: z.number().optional().default(5000).describe('跟踪时长（毫秒），默认 5000'),
topN: z.number().int().positive().max(200).optional().describe('Top N（调用栈）数量，默认 20'),
collectGarbage: z.boolean().optional().describe('采集前是否触发 GC，默认 false'),
maxSnapshotBytes: z.number().int().positive().optional().describe('raw snapshot 采集最大字节数，默认 200MB'),
maxParseBytes: z.number().int().positive().optional().describe('JSON.parse 解析最大字节数，默认 50MB'),
export: z.object({
mode: z.enum(['none', 'file', 'inline', 'both']).optional().describe(
[
'raw profile（带 trace 的 heap snapshot）导出方式：',
'- none：不导出（只返回摘要 summary）',
'- file：导出为文件（推荐，避免响应过大）',
'- inline：把快照片段放到返回里（会截断，可能很大）',
'- both：同时 file + inline',
].join('\n')
),
filePath: z.string().optional().describe(
[
'file/both 模式的输出文件路径。',
'推荐使用相对路径（相对 MCP Server 进程工作目录）：例如 ./.heapsnapshot/alloc.heapsnapshot',
'不填时默认写入当前目录下的 `./.heapsnapshot/` 目录，并自动创建目录。',
].join('\n')
),
maxInlineBytes: z.number().int().positive().optional().describe(
'inline/both 模式 inline 输出最大字节数（超出截断）。'
),
}).optional().describe('raw profile 导出选项'),
}),
handler: async (
args: {
url?: string;
duration?: number;
topN?: number;
collectGarbage?: boolean;
maxSnapshotBytes?: number;
maxParseBytes?: number;
export?: { mode?: 'none' | 'file' | 'inline' | 'both'; filePath?: string; maxInlineBytes?: number };
},
context: ToolContext
) => {
const tracking = await context.heapHandler.trackAllocations({
url: args.url,
duration: args.duration,
topN: args.topN,
collectGarbage: args.collectGarbage,
maxSnapshotBytes: args.maxSnapshotBytes,
maxParseBytes: args.maxParseBytes,
export: args.export,
});
return {
content: [
{
type: 'text',
text: JSON.stringify(tracking, null, 2),
},
],
};
},
};
