can1357
1bde755933
- Added process-wide singleton instances for InternalUrlRouter, AsyncJobManager, and MCPManager. - Changed internal URL protocols to resolve through registered sessions and scan all active roots/datasets for matches. - Refactored agent, artifact, memory, rule, skill, jobs, and mcp handlers to use shared manager and rule/skill state. - Removed per-session protocol/tool wiring and switched tests to initialize and reset global singleton state.
487 lines
18 KiB
TypeScript
487 lines
18 KiB
TypeScript
import * as path from "node:path";
|
|
import type { AgentTool, AgentToolContext, AgentToolResult, AgentToolUpdateCallback } from "@oh-my-pi/pi-agent-core";
|
|
import { type AstFindMatch, astGrep } from "@oh-my-pi/pi-natives";
|
|
import type { Component } from "@oh-my-pi/pi-tui";
|
|
import { Text } from "@oh-my-pi/pi-tui";
|
|
import { prompt, untilAborted } from "@oh-my-pi/pi-utils";
|
|
import { type Static, Type } from "@sinclair/typebox";
|
|
import type { RenderResultOptions } from "../extensibility/custom-tools/types";
|
|
import { InternalUrlRouter } from "../internal-urls";
|
|
import type { Theme } from "../modes/theme/theme";
|
|
import astGrepDescription from "../prompts/tools/ast-grep.md" with { type: "text" };
|
|
import { Ellipsis, Hasher, type RenderCache, renderStatusLine, renderTreeList, truncateToWidth } from "../tui";
|
|
import { resolveFileDisplayMode } from "../utils/file-display-mode";
|
|
import type { ToolSession } from ".";
|
|
import { createFileRecorder, formatResultPath } from "./file-recorder";
|
|
import { formatGroupedFiles } from "./grouped-file-output";
|
|
import { formatMatchLine } from "./match-line-format";
|
|
import type { OutputMeta } from "./output-meta";
|
|
import {
|
|
formatPathRelativeToCwd,
|
|
hasGlobPathChars,
|
|
normalizePathLikeInput,
|
|
parseSearchPath,
|
|
partitionExistingPaths,
|
|
resolveExplicitSearchPaths,
|
|
resolveToCwd,
|
|
} from "./path-utils";
|
|
import {
|
|
dedupeParseErrors,
|
|
formatCodeFrameLine,
|
|
formatCount,
|
|
formatEmptyMessage,
|
|
formatErrorMessage,
|
|
formatParseErrors,
|
|
PARSE_ERRORS_LIMIT,
|
|
PREVIEW_LIMITS,
|
|
} from "./render-utils";
|
|
import { ToolError } from "./tool-errors";
|
|
import { toolResult } from "./tool-result";
|
|
|
|
const astGrepSchema = Type.Object({
|
|
pat: Type.String({ description: "ast pattern", examples: ["console.log($$$)"] }),
|
|
paths: Type.Array(Type.String({ description: "file, directory, glob, or internal URL to search" }), {
|
|
minItems: 1,
|
|
description: "files, directories, globs, or internal URLs to search",
|
|
examples: [["src/"], ["src/foo.ts"], ["src/**/*.ts"], ["src/", "packages/"]],
|
|
}),
|
|
skip: Type.Optional(Type.Number({ description: "matches to skip", default: 0 })),
|
|
});
|
|
|
|
async function runMultiTargetAstGrep(
|
|
targets: Array<{ basePath: string; glob?: string }>,
|
|
options: { patterns: string[]; commonBasePath: string; skip: number; limit: number; signal?: AbortSignal },
|
|
): Promise<{
|
|
matches: AstFindMatch[];
|
|
totalMatches: number;
|
|
filesWithMatches: number;
|
|
filesSearched: number;
|
|
limitReached: boolean;
|
|
parseErrors?: string[];
|
|
}> {
|
|
const aggregatedMatches: AstFindMatch[] = [];
|
|
const parseErrors: string[] = [];
|
|
let totalMatches = 0;
|
|
let filesSearched = 0;
|
|
let limitReached = false;
|
|
for (const target of targets) {
|
|
const targetResult = await astGrep({
|
|
patterns: options.patterns,
|
|
path: target.basePath,
|
|
glob: target.glob,
|
|
offset: 0,
|
|
limit: options.skip + options.limit + 1,
|
|
includeMeta: true,
|
|
signal: options.signal,
|
|
});
|
|
totalMatches += targetResult.totalMatches;
|
|
filesSearched += targetResult.filesSearched;
|
|
limitReached = limitReached || targetResult.limitReached;
|
|
if (targetResult.parseErrors) parseErrors.push(...targetResult.parseErrors);
|
|
for (const match of targetResult.matches) {
|
|
const absolute = path.resolve(target.basePath, match.path);
|
|
const rebased = path.relative(options.commonBasePath, absolute).replace(/\\/g, "/");
|
|
aggregatedMatches.push({ ...match, path: rebased });
|
|
}
|
|
}
|
|
aggregatedMatches.sort((left, right) => {
|
|
const pathCmp = left.path.localeCompare(right.path);
|
|
if (pathCmp !== 0) return pathCmp;
|
|
if (left.startLine !== right.startLine) return left.startLine - right.startLine;
|
|
if (left.startColumn !== right.startColumn) return left.startColumn - right.startColumn;
|
|
if (left.byteStart !== right.byteStart) return left.byteStart - right.byteStart;
|
|
return left.byteEnd - right.byteEnd;
|
|
});
|
|
const visible = aggregatedMatches.slice(options.skip);
|
|
const paged = visible.slice(0, options.limit);
|
|
const filesWithMatches = new Set(aggregatedMatches.map(match => match.path)).size;
|
|
return {
|
|
matches: paged,
|
|
totalMatches,
|
|
filesWithMatches,
|
|
filesSearched,
|
|
limitReached: limitReached || visible.length > options.limit,
|
|
parseErrors: parseErrors.length > 0 ? parseErrors : undefined,
|
|
};
|
|
}
|
|
|
|
export interface AstGrepToolDetails {
|
|
matchCount: number;
|
|
fileCount: number;
|
|
filesSearched: number;
|
|
limitReached: boolean;
|
|
parseErrors?: string[];
|
|
scopePath?: string;
|
|
files?: string[];
|
|
fileMatches?: Array<{ path: string; count: number }>;
|
|
meta?: OutputMeta;
|
|
/** Pre-formatted text for the user-visible TUI render. Mirrors `result.text` lines but uses
|
|
* a `│` gutter and `*` to mark match lines. The TUI uses this directly so it never parses model-facing text. */
|
|
displayContent?: string;
|
|
}
|
|
|
|
export class AstGrepTool implements AgentTool<typeof astGrepSchema, AstGrepToolDetails> {
|
|
readonly name = "ast_grep";
|
|
readonly label = "AST Grep";
|
|
readonly summary = "Search code with AST patterns (structural grep)";
|
|
readonly description: string;
|
|
readonly parameters = astGrepSchema;
|
|
readonly strict = true;
|
|
readonly loadMode = "discoverable";
|
|
|
|
constructor(private readonly session: ToolSession) {
|
|
this.description = prompt.render(astGrepDescription);
|
|
}
|
|
|
|
async execute(
|
|
_toolCallId: string,
|
|
params: Static<typeof astGrepSchema>,
|
|
signal?: AbortSignal,
|
|
_onUpdate?: AgentToolUpdateCallback<AstGrepToolDetails>,
|
|
_context?: AgentToolContext,
|
|
): Promise<AgentToolResult<AstGrepToolDetails>> {
|
|
return untilAborted(signal, async () => {
|
|
const pattern = params.pat.trim();
|
|
if (pattern.length === 0) {
|
|
throw new ToolError("`pat` must be a non-empty pattern");
|
|
}
|
|
const patterns = [pattern];
|
|
const skip = params.skip === undefined ? 0 : Math.floor(params.skip);
|
|
if (!Number.isFinite(skip) || skip < 0) {
|
|
throw new ToolError("skip must be a non-negative number");
|
|
}
|
|
const formatScopePath = (targetPath: string): string => formatPathRelativeToCwd(targetPath, this.session.cwd);
|
|
let searchPath: string;
|
|
let scopePath: string;
|
|
let globFilter: string | undefined;
|
|
let multiTargets: Array<{ basePath: string; glob?: string }> | undefined;
|
|
const rawPaths = params.paths.map(normalizePathLikeInput);
|
|
if (rawPaths.some(rawPath => rawPath.length === 0)) {
|
|
throw new ToolError("`paths` must contain non-empty paths or globs");
|
|
}
|
|
const internalRouter = InternalUrlRouter.instance();
|
|
const resolvedPathInputs: string[] = [];
|
|
for (const rawPath of rawPaths) {
|
|
if (!internalRouter.canHandle(rawPath)) {
|
|
resolvedPathInputs.push(rawPath);
|
|
continue;
|
|
}
|
|
if (hasGlobPathChars(rawPath)) {
|
|
throw new ToolError(`Glob patterns are not supported for internal URLs: ${rawPath}`);
|
|
}
|
|
const resource = await internalRouter.resolve(rawPath);
|
|
if (!resource.sourcePath) {
|
|
throw new ToolError(`Cannot search internal URL without backing file: ${rawPath}`);
|
|
}
|
|
resolvedPathInputs.push(resource.sourcePath);
|
|
}
|
|
let effectivePathInputs = resolvedPathInputs;
|
|
if (resolvedPathInputs.length > 1) {
|
|
const partition = await partitionExistingPaths(resolvedPathInputs, this.session.cwd, parseSearchPath);
|
|
if (partition.valid.length === 0) {
|
|
throw new ToolError(`Path not found: ${partition.missing.join(", ")}`);
|
|
}
|
|
effectivePathInputs = partition.valid;
|
|
}
|
|
if (effectivePathInputs.length === 1) {
|
|
const parsedPath = parseSearchPath(effectivePathInputs[0] ?? ".");
|
|
searchPath = resolveToCwd(parsedPath.basePath, this.session.cwd);
|
|
globFilter = parsedPath.glob;
|
|
scopePath = formatScopePath(searchPath);
|
|
} else {
|
|
const multiSearchPath = await resolveExplicitSearchPaths(effectivePathInputs, this.session.cwd, globFilter);
|
|
if (!multiSearchPath) {
|
|
throw new ToolError("`paths` must contain at least one path or glob");
|
|
}
|
|
searchPath = multiSearchPath.basePath;
|
|
globFilter = multiSearchPath.targets ? undefined : multiSearchPath.glob;
|
|
multiTargets = multiSearchPath.targets;
|
|
scopePath = multiSearchPath.scopePath;
|
|
}
|
|
|
|
const resolvedSearchPath = searchPath;
|
|
scopePath = scopePath ?? formatScopePath(resolvedSearchPath);
|
|
let isDirectory: boolean;
|
|
try {
|
|
const stat = await Bun.file(resolvedSearchPath).stat();
|
|
isDirectory = stat.isDirectory();
|
|
} catch {
|
|
throw new ToolError(`Path not found: ${scopePath}`);
|
|
}
|
|
|
|
const DEFAULT_AST_LIMIT = 50;
|
|
const result = multiTargets
|
|
? await runMultiTargetAstGrep(multiTargets, {
|
|
patterns,
|
|
commonBasePath: resolvedSearchPath,
|
|
skip,
|
|
limit: DEFAULT_AST_LIMIT,
|
|
signal,
|
|
})
|
|
: await astGrep({
|
|
patterns,
|
|
path: resolvedSearchPath,
|
|
glob: globFilter,
|
|
offset: skip,
|
|
includeMeta: true,
|
|
signal,
|
|
});
|
|
|
|
const normalizedParseErrors = (result.parseErrors ?? []).map(error => {
|
|
const parseError = error.match(/^.+: (.+: parse error \(syntax tree contains error nodes\))$/);
|
|
return parseError?.[1] ?? error;
|
|
});
|
|
const dedupedParseErrors = dedupeParseErrors(normalizedParseErrors);
|
|
const formatPath = (filePath: string): string =>
|
|
formatResultPath(filePath, isDirectory, resolvedSearchPath, this.session.cwd);
|
|
|
|
const { record: recordFile, list: fileList } = createFileRecorder();
|
|
const fileMatchCounts = new Map<string, number>();
|
|
const matchesByFile = new Map<string, AstFindMatch[]>();
|
|
for (const match of result.matches) {
|
|
const relativePath = formatPath(match.path);
|
|
recordFile(relativePath);
|
|
if (!matchesByFile.has(relativePath)) {
|
|
matchesByFile.set(relativePath, []);
|
|
}
|
|
matchesByFile.get(relativePath)!.push(match);
|
|
}
|
|
|
|
const baseDetails: AstGrepToolDetails = {
|
|
matchCount: result.totalMatches,
|
|
fileCount: result.filesWithMatches,
|
|
filesSearched: result.filesSearched,
|
|
limitReached: result.limitReached,
|
|
...(dedupedParseErrors.length > 0 ? { parseErrors: dedupedParseErrors } : {}),
|
|
scopePath,
|
|
files: fileList,
|
|
fileMatches: [],
|
|
};
|
|
|
|
if (result.matches.length === 0) {
|
|
const noMatchMessage = dedupedParseErrors.length
|
|
? "No matches found. Parse issues mean the query may be mis-scoped; narrow `paths` before concluding absence."
|
|
: "No matches found";
|
|
const parseMessage = dedupedParseErrors.length
|
|
? `\n${formatParseErrors(dedupedParseErrors).join("\n")}`
|
|
: "";
|
|
return toolResult(baseDetails).text(`${noMatchMessage}${parseMessage}`).done();
|
|
}
|
|
|
|
const useHashLines = resolveFileDisplayMode(this.session).hashLines;
|
|
const outputLines: string[] = [];
|
|
const displayLines: string[] = [];
|
|
const renderMatchesForFile = (relativePath: string): { model: string[]; display: string[] } => {
|
|
const modelOut: string[] = [];
|
|
const displayOut: string[] = [];
|
|
const fileMatches = matchesByFile.get(relativePath) ?? [];
|
|
const lineNumberWidth = fileMatches.reduce((width, match) => {
|
|
const lineCount = match.text.split("\n").length;
|
|
const endLine = match.startLine + lineCount - 1;
|
|
return Math.max(width, String(match.startLine).length, String(endLine).length);
|
|
}, 0);
|
|
for (const match of fileMatches) {
|
|
const matchLines = match.text.split("\n");
|
|
for (let index = 0; index < matchLines.length; index++) {
|
|
const lineNumber = match.startLine + index;
|
|
const isMatch = index === 0;
|
|
const line = matchLines[index] ?? "";
|
|
modelOut.push(formatMatchLine(lineNumber, line, isMatch, { useHashLines }));
|
|
displayOut.push(formatCodeFrameLine(isMatch ? "*" : " ", lineNumber, line, lineNumberWidth));
|
|
}
|
|
if (match.metaVariables && Object.keys(match.metaVariables).length > 0) {
|
|
const serializedMeta = Object.entries(match.metaVariables)
|
|
.sort(([left], [right]) => left.localeCompare(right))
|
|
.map(([key, value]) => `${key}=${value}`)
|
|
.join(", ");
|
|
modelOut.push(` meta: ${serializedMeta}`);
|
|
displayOut.push(` meta: ${serializedMeta}`);
|
|
}
|
|
fileMatchCounts.set(relativePath, (fileMatchCounts.get(relativePath) ?? 0) + 1);
|
|
}
|
|
return { model: modelOut, display: displayOut };
|
|
};
|
|
|
|
if (isDirectory) {
|
|
const grouped = formatGroupedFiles(fileList, relativePath => {
|
|
const rendered = renderMatchesForFile(relativePath);
|
|
return { modelLines: rendered.model, displayLines: rendered.display };
|
|
});
|
|
outputLines.push(...grouped.model);
|
|
displayLines.push(...grouped.display);
|
|
} else {
|
|
for (const relativePath of fileList) {
|
|
const rendered = renderMatchesForFile(relativePath);
|
|
outputLines.push(...rendered.model);
|
|
displayLines.push(...rendered.display);
|
|
}
|
|
}
|
|
|
|
const details: AstGrepToolDetails = {
|
|
...baseDetails,
|
|
fileMatches: fileList.map(filePath => ({
|
|
path: filePath,
|
|
count: fileMatchCounts.get(filePath) ?? 0,
|
|
})),
|
|
displayContent: displayLines.join("\n"),
|
|
};
|
|
if (result.limitReached) {
|
|
outputLines.push("", "Result limit reached; narrow paths or increase limit.");
|
|
}
|
|
if (dedupedParseErrors.length) {
|
|
outputLines.push("", ...formatParseErrors(dedupedParseErrors));
|
|
}
|
|
|
|
return toolResult(details).text(outputLines.join("\n")).done();
|
|
});
|
|
}
|
|
}
|
|
|
|
// =============================================================================
|
|
// TUI Renderer
|
|
// =============================================================================
|
|
|
|
interface AstGrepRenderArgs {
|
|
pat?: string;
|
|
paths?: string[];
|
|
skip?: number;
|
|
}
|
|
|
|
const COLLAPSED_MATCH_LIMIT = PREVIEW_LIMITS.COLLAPSED_LINES * 2;
|
|
|
|
export const astGrepToolRenderer = {
|
|
inline: true,
|
|
renderCall(args: AstGrepRenderArgs, _options: RenderResultOptions, uiTheme: Theme): Component {
|
|
const meta: string[] = [];
|
|
if (args.paths?.length) meta.push(`in ${args.paths.join(", ")}`);
|
|
if (args.skip !== undefined && args.skip > 0) meta.push(`skip:${args.skip}`);
|
|
|
|
const description = args.pat ?? "?";
|
|
const text = renderStatusLine({ icon: "pending", title: "AST Grep", description, meta }, uiTheme);
|
|
return new Text(text, 0, 0);
|
|
},
|
|
|
|
renderResult(
|
|
result: { content: Array<{ type: string; text?: string }>; details?: AstGrepToolDetails; isError?: boolean },
|
|
options: RenderResultOptions,
|
|
uiTheme: Theme,
|
|
args?: AstGrepRenderArgs,
|
|
): Component {
|
|
const details = result.details;
|
|
|
|
if (result.isError) {
|
|
const errorText = result.content?.find(c => c.type === "text")?.text || "Unknown error";
|
|
return new Text(formatErrorMessage(errorText, uiTheme), 0, 0);
|
|
}
|
|
|
|
const matchCount = details?.matchCount ?? 0;
|
|
const fileCount = details?.fileCount ?? 0;
|
|
const filesSearched = details?.filesSearched ?? 0;
|
|
const limitReached = details?.limitReached ?? false;
|
|
|
|
if (matchCount === 0) {
|
|
const description = args?.pat;
|
|
const meta = ["0 matches"];
|
|
if (details?.scopePath) meta.push(`in ${details.scopePath}`);
|
|
if (filesSearched > 0) meta.push(`searched ${filesSearched}`);
|
|
const header = renderStatusLine({ icon: "warning", title: "AST Grep", description, meta }, uiTheme);
|
|
const lines = [header, formatEmptyMessage("No matches found", uiTheme)];
|
|
if (details?.parseErrors?.length) {
|
|
lines.push(uiTheme.fg("warning", "Query may be mis-scoped; narrow `paths` before concluding absence"));
|
|
const capped = details.parseErrors.slice(0, PARSE_ERRORS_LIMIT);
|
|
for (const err of capped) {
|
|
lines.push(uiTheme.fg("warning", ` - ${err}`));
|
|
}
|
|
if (details.parseErrors.length > PARSE_ERRORS_LIMIT) {
|
|
lines.push(uiTheme.fg("dim", ` … ${details.parseErrors.length - PARSE_ERRORS_LIMIT} more`));
|
|
}
|
|
}
|
|
return new Text(lines.join("\n"), 0, 0);
|
|
}
|
|
|
|
const summaryParts = [formatCount("match", matchCount), formatCount("file", fileCount)];
|
|
const meta = [...summaryParts];
|
|
if (details?.scopePath) meta.push(`in ${details.scopePath}`);
|
|
meta.push(`searched ${filesSearched}`);
|
|
if (limitReached) meta.push(uiTheme.fg("warning", "limit reached"));
|
|
const description = args?.pat;
|
|
const header = renderStatusLine(
|
|
{ icon: limitReached ? "warning" : "success", title: "AST Grep", description, meta },
|
|
uiTheme,
|
|
);
|
|
|
|
const textContent = result.details?.displayContent ?? result.content?.find(c => c.type === "text")?.text ?? "";
|
|
const rawLines = textContent.split("\n");
|
|
const hasSeparators = rawLines.some(line => line.trim().length === 0);
|
|
const allGroups: string[][] = [];
|
|
if (hasSeparators) {
|
|
let current: string[] = [];
|
|
for (const line of rawLines) {
|
|
if (line.trim().length === 0) {
|
|
if (current.length > 0) {
|
|
allGroups.push(current);
|
|
current = [];
|
|
}
|
|
continue;
|
|
}
|
|
current.push(line);
|
|
}
|
|
if (current.length > 0) allGroups.push(current);
|
|
} else {
|
|
const nonEmpty = rawLines.filter(line => line.trim().length > 0);
|
|
if (nonEmpty.length > 0) {
|
|
allGroups.push(nonEmpty);
|
|
}
|
|
}
|
|
const matchGroups = allGroups.filter(
|
|
group => !group[0]?.startsWith("Result limit reached") && !group[0]?.startsWith("Parse issues:"),
|
|
);
|
|
|
|
const extraLines: string[] = [];
|
|
if (limitReached) {
|
|
extraLines.push(uiTheme.fg("warning", "limit reached; narrow paths or increase limit"));
|
|
}
|
|
if (details?.parseErrors?.length) {
|
|
const total = details.parseErrors.length;
|
|
const label =
|
|
total > PARSE_ERRORS_LIMIT
|
|
? `${PARSE_ERRORS_LIMIT} / ${total} parse issues`
|
|
: `${total} parse issue${total !== 1 ? "s" : ""}`;
|
|
extraLines.push(uiTheme.fg("warning", label));
|
|
}
|
|
|
|
let cached: RenderCache | undefined;
|
|
return {
|
|
render(width: number): string[] {
|
|
const { expanded } = options;
|
|
const key = new Hasher().bool(expanded).u32(width).digest();
|
|
if (cached?.key === key) return cached.lines;
|
|
const matchLines = renderTreeList(
|
|
{
|
|
items: matchGroups,
|
|
expanded,
|
|
maxCollapsed: matchGroups.length,
|
|
maxCollapsedLines: COLLAPSED_MATCH_LIMIT,
|
|
itemType: "match",
|
|
renderItem: group =>
|
|
group.map(line => {
|
|
if (line.startsWith("## ")) return uiTheme.fg("dim", line);
|
|
if (line.startsWith("# ")) return uiTheme.fg("accent", line);
|
|
if (line.startsWith(" meta:")) return uiTheme.fg("dim", line);
|
|
return uiTheme.fg("toolOutput", line);
|
|
}),
|
|
},
|
|
uiTheme,
|
|
);
|
|
const rendered = [header, ...matchLines, ...extraLines].map(l => truncateToWidth(l, width, Ellipsis.Omit));
|
|
cached = { key, lines: rendered };
|
|
return rendered;
|
|
},
|
|
invalidate() {
|
|
cached = undefined;
|
|
},
|
|
};
|
|
},
|
|
mergeCallAndResult: true,
|
|
};
|