Method execute

src/codegraph/tools.ts:266–326

(args: z.infer<typeof FindReferencesArgs>, ctx: ToolContext)

  argsSchema = FindReferencesArgs;

  async execute(args: z.infer<typeof FindReferencesArgs>, ctx: ToolContext): Promise<ToolResult> {
    const db = requireDB();
    if ('error' in db) return { content: db.error, isError: true };

    const defs = db.findSymbolsByName(args.name);
    const defKeys = args.include_definitions
      ? new Set<string>()
      : new Set(defs.map(d => `${d.file_path}:${d.start_line}`));

    const escaped = args.name.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
    const limit = args.limit ?? 200;

    const rgArgs = ['-n', '--no-heading', '--word-regexp', '--max-count', String(limit + defs.length), '-e', escaped];
    if (args.language) rgArgs.push('--type', args.language);
    rgArgs.push(ctx.cwd);

    const search = await runLineSearch(ctx.cwd, {
      rgArgs,
      regex: new RegExp(`\\b${escaped}\\b`),
      language: args.language,
      maxCount: limit + defs.length,
      signal: ctx.signal,
    });
    if ('error' in search) return { content: search.error, isError: true };

    const lines = search.stdout.split('\n').filter(Boolean);
    const refs: Array<{ file: string; line: number; preview: string }> = [];
    for (const raw of lines) {
      const firstColon = raw.indexOf(':');
      const secondColon = raw.indexOf(':', firstColon + 1);
      if (firstColon === -1 || secondColon === -1) continue;
      const file = raw.slice(0, firstColon);
      const lineNum = parseInt(raw.slice(firstColon + 1, secondColon), 10);
      if (!Number.isFinite(lineNum)) continue;
      if (defKeys.has(`${file}:${lineNum}`)) continue;
      refs.push({ file, line: lineNum, preview: raw.slice(secondColon + 1).trim().slice(0, 200) });
      if (refs.length >= limit) break;
    }
    if (refs.length === 0) {
      const note = args.include_definitions
        ? 'No references found at all'
        : 'No references found (other than definitions)';
      return { content: `[NO_REFERENCES] ${note} for "${args.name}". The project may not be indexed (try /index).` };
    }
    const byFile = new Map<string, typeof refs>();
    for (const r of refs) {
      const rel = path.relative(ctx.cwd, r.file);
      if (!byFile.has(rel)) byFile.set(rel, []);
      byFile.get(rel)!.push(r);
    }
    const out: string[] = [`Found ${refs.length} reference${refs.length > 1 ? 's' : ''} to "${args.name}" across ${byFile.size} file${byFile.size > 1 ? 's' : ''}:`];
    for (const [rel, items] of byFile) {
      out.push(`\n  ${rel}  (${items.length})`);
      for (const it of items.slice(0, 30)) {
        out.push(`    :${it.line}  ${it.preview}`);
      }
      if (items.length > 30) out.push(`    ... [+${items.length - 30} more in this file]`);
    }