@shanepadgett/tau-agent 0.33.0 → 0.34.0

package/extensions/explore/ast/languages/odin.ts

@@ -1,407 +0,0 @@
-import type { Node, Tree } from "web-tree-sitter";
-import type { ExtractResult, GrammarAdapter, LanguageCapabilities } from "../adapter.ts";
-import { resolveOdinFileDep } from "./odin-file-deps.ts";
-import type { CallSite, Decl, ImportRef, Visibility } from "../ir.ts";
-import {
-	applyDoc,
-	callSite,
-	docSpanBefore,
-	endLine,
-	finishDecl,
-	qualify,
-	startLine,
-	unquote,
-	walkBodyNodes,
-	type DocSpan,
-} from "./tree.ts";
-
-const CAPABILITIES: LanguageCapabilities = {
-	shape: true,
-	search: true,
-	fileDeps: true,
-	callEdges: true,
-	packageSurface: false,
-};
-
-const IMPORT_NOISE = new Set([
-	"package",
-	"import",
-	"foreign",
-	"proc",
-	"where",
-	"when",
-	"if",
-	"else",
-	"for",
-	"in",
-	"switch",
-	"case",
-	"break",
-	"continue",
-	"fallthrough",
-	"return",
-	"defer",
-	"using",
-	"struct",
-	"union",
-	"enum",
-	"bit_set",
-	"map",
-	"dynamic",
-	"cast",
-	"transmute",
-	"auto_cast",
-	"true",
-	"false",
-	"nil",
-	"context",
-]);
-
-const EXTENSIONS = [".odin"] as const;
-
-/**
- * Node kinds from tree-sitter-odin v1.3.0 node-types + locals.scm definitions:
- *   procedure_declaration, overloaded_procedure_declaration → function
- *   struct_declaration / union_declaration / bit_field_declaration → struct (union/bit_field closest kind)
- *   enum_declaration → enum
- *   const_declaration / const_type_declaration → constant
- *   variable_declaration (`:=`) → variable, or function when value is `procedure`
- *   var_declaration (`:`) → variable
- *   field → struct field
- *   foreign_block → walk nested package-level decls
- * Package-level only (plan). Nested procs inside function bodies stay out.
- */
-const PROCEDURE_DECLARATION = "procedure_declaration";
-const OVERLOADED_PROCEDURE_DECLARATION = "overloaded_procedure_declaration";
-const STRUCT_DECLARATION = "struct_declaration";
-const ENUM_DECLARATION = "enum_declaration";
-const UNION_DECLARATION = "union_declaration";
-const BIT_FIELD_DECLARATION = "bit_field_declaration";
-const CONST_DECLARATION = "const_declaration";
-const CONST_TYPE_DECLARATION = "const_type_declaration";
-const VAR_DECLARATION = "var_declaration";
-const VARIABLE_DECLARATION = "variable_declaration";
-const FOREIGN_BLOCK = "foreign_block";
-const IMPORT_DECLARATION = "import_declaration";
-const ATTRIBUTES = "attributes";
-const ATTRIBUTE = "attribute";
-const IDENTIFIER = "identifier";
-const PROCEDURE = "procedure";
-const FIELD = "field";
-const BLOCK = "block";
-
-const NESTED_SCOPE = new Set([
-	"procedure_declaration",
-	"overloaded_procedure_declaration",
-	"struct_declaration",
-	"enum_declaration",
-]);
-
-function extractCalls(body: Node | null): CallSite[] {
-	if (body === null) return [];
-	const out: CallSite[] = [];
-	walkBodyNodes(
-		body,
-		(n) => NESTED_SCOPE.has(n.type),
-		(node) => {
-			if (node.type !== "call_expression" && node.type !== "selector_call_expression") return;
-			const fn = node.childForFieldName("function") ?? node.namedChildren[0];
-			if (fn === null || fn === undefined) return;
-			let leafNode = fn;
-			let receiver = "";
-			if (fn.type === "member_expression" || fn.type === "selector_expression") {
-				const fieldNode = fn.childForFieldName("field") ?? fn.namedChildren[fn.namedChildren.length - 1];
-				const obj = fn.childForFieldName("argument") ?? fn.childForFieldName("operand") ?? fn.namedChildren[0];
-				if (fieldNode !== null && fieldNode !== undefined) leafNode = fieldNode;
-				if (obj !== null && obj !== undefined) receiver = obj.text;
-			}
-			const site = callSite(
-				leafNode.text,
-				startLine(leafNode),
-				leafNode.startIndex,
-				leafNode.endIndex,
-				"call",
-				receiver,
-			);
-			if (site !== undefined) out.push(site);
-		},
-	);
-	return out;
-}
-
-function odinDoc(node: Node, source: string): DocSpan | undefined {
-	return docSpanBefore(node, source);
-}
-
-function hasPrivateAttribute(node: Node): boolean {
-	const attrs = node.namedChildren.find((child) => child.type === ATTRIBUTES);
-	if (attrs === undefined) return false;
-	for (const attr of attrs.namedChildren) {
-		if (attr.type !== ATTRIBUTE) continue;
-		for (const child of attr.namedChildren) {
-			if (child.type === IDENTIFIER && child.text === "private") return true;
-		}
-	}
-	return false;
-}
-
-function visibilityOf(node: Node): Visibility {
-	return hasPrivateAttribute(node) ? "private" : "public";
-}
-
-/** First non-attribute identifier child — name position for all odin decls in this grammar. */
-function firstIdentifier(node: Node): string {
-	for (const child of node.namedChildren) {
-		if (child.type === ATTRIBUTES) continue;
-		if (child.type === IDENTIFIER) return child.text;
-	}
-	return "";
-}
-
-function procedureBody(node: Node): Node | null {
-	const proc = node.namedChildren.find((child) => child.type === PROCEDURE);
-	if (proc === undefined) return null;
-	// `---` foreign stubs use uninitialized, not block.
-	return proc.namedChildren.find((child) => child.type === BLOCK) ?? null;
-}
-
-function leaf(
-	node: Node,
-	owner: string,
-	kind: Decl["kind"],
-	name: string,
-	visibility: Visibility,
-	body: Node | null,
-	children: Decl[],
-	doc: DocSpan | undefined,
-): Decl | undefined {
-	if (name.length === 0) return undefined;
-	const callable = kind === "function" || kind === "method" || kind === "constructor";
-	const decl = finishDecl(
-		{
-			kind,
-			name,
-			qualifiedName: qualify(owner, name),
-			startLine: startLine(node),
-			endLine: endLine(node),
-			startOffset: node.startIndex,
-			endOffset: node.endIndex,
-			visibility,
-			exported: visibility === "public",
-			children,
-			calls: callable ? extractCalls(body) : [],
-			bases: [],
-		},
-		body,
-	);
-	if (body === null) decl.signatureEndOffset = node.endIndex;
-	return applyDoc(decl, doc);
-}
-
-/** struct_declaration children of type `field` (not `struct_field` — that is composite-literal elements). */
-function structFields(node: Node, owner: string): Decl[] {
-	const out: Decl[] = [];
-	for (const child of node.namedChildren) {
-		if (child.type !== FIELD) continue;
-		const names: string[] = [];
-		for (const part of child.namedChildren) {
-			if (part.type === IDENTIFIER) names.push(part.text);
-			if (part.type === "type") break;
-		}
-		for (const name of names) {
-			// `using _: T` embedding — skip blank identifier.
-			if (name === "_") continue;
-			const decl = leaf(child, owner, "field", name, "public", null, [], undefined);
-			if (decl !== undefined) out.push(decl);
-		}
-	}
-	return out;
-}
-
-function enumMembers(node: Node, owner: string): Decl[] {
-	const out: Decl[] = [];
-	let seenName = false;
-	for (const child of node.namedChildren) {
-		if (child.type !== IDENTIFIER) continue;
-		if (!seenName) {
-			seenName = true;
-			continue;
-		}
-		const decl = leaf(child, owner, "enumMember", child.text, "public", null, [], undefined);
-		if (decl !== undefined) out.push(decl);
-	}
-	return out;
-}
-
-/**
- * bit_field_declaration flattens members as identifier/type/number triples after the backing type.
- * Shape: Name :: bit_field Backing { field: T | bits, ... }
- */
-function bitFieldMembers(node: Node, owner: string): Decl[] {
-	const out: Decl[] = [];
-	let seenName = false;
-	let seenBackingType = false;
-	for (const child of node.namedChildren) {
-		if (child.type === ATTRIBUTES) continue;
-		if (!seenName && child.type === IDENTIFIER) {
-			seenName = true;
-			continue;
-		}
-		if (seenName && !seenBackingType && child.type === "type") {
-			seenBackingType = true;
-			continue;
-		}
-		if (!seenBackingType) continue;
-		if (child.type === IDENTIFIER) {
-			const decl = leaf(child, owner, "field", child.text, "public", null, [], undefined);
-			if (decl !== undefined) out.push(decl);
-		}
-	}
-	return out;
-}
-
-function cutSignatureBefore(node: Node, firstChild: Node | undefined, decl: Decl): void {
-	if (firstChild === undefined) return;
-	decl.signatureEndOffset = firstChild.startIndex;
-	decl.bodyStartOffset = firstChild.startIndex;
-	decl.bodyEndOffset = node.endIndex;
-}
-
-function declsFromNode(node: Node, owner: string, source: string): Decl[] {
-	if (node.isError || node.type === "ERROR") {
-		const out: Decl[] = [];
-		for (const child of node.children) {
-			if (child.isNamed || child.isError) out.push(...declsFromNode(child, owner, source));
-		}
-		return out;
-	}
-
-	const doc = odinDoc(node, source);
-	const visibility = visibilityOf(node);
-
-	switch (node.type) {
-		case PROCEDURE_DECLARATION:
-		case OVERLOADED_PROCEDURE_DECLARATION: {
-			const name = firstIdentifier(node);
-			const body = node.type === PROCEDURE_DECLARATION ? procedureBody(node) : null;
-			const decl = leaf(node, owner, "function", name, visibility, body, [], doc);
-			return decl === undefined ? [] : [decl];
-		}
-		case STRUCT_DECLARATION: {
-			const name = firstIdentifier(node);
-			const qn = qualify(owner, name);
-			const children = structFields(node, qn);
-			const firstField = node.namedChildren.find((child) => child.type === FIELD);
-			const decl = leaf(node, owner, "struct", name, visibility, null, children, doc);
-			if (decl !== undefined) cutSignatureBefore(node, firstField, decl);
-			return decl === undefined ? [] : [decl];
-		}
-		case UNION_DECLARATION: {
-			// locals.scm: union → type; shared IR uses struct as closest kind.
-			const name = firstIdentifier(node);
-			const decl = leaf(node, owner, "struct", name, visibility, null, [], doc);
-			return decl === undefined ? [] : [decl];
-		}
-		case BIT_FIELD_DECLARATION: {
-			// locals.scm: bit_field → type; shared IR uses struct.
-			const name = firstIdentifier(node);
-			const qn = qualify(owner, name);
-			const children = bitFieldMembers(node, qn);
-			const decl = leaf(node, owner, "struct", name, visibility, null, children, doc);
-			// Signature through backing type: first member identifier after backing `type`.
-			const firstMember = children[0];
-			if (decl !== undefined && firstMember !== undefined) {
-				decl.signatureEndOffset = firstMember.startOffset;
-				decl.bodyStartOffset = firstMember.startOffset;
-				decl.bodyEndOffset = node.endIndex;
-			}
-			return decl === undefined ? [] : [decl];
-		}
-		case ENUM_DECLARATION: {
-			const name = firstIdentifier(node);
-			const qn = qualify(owner, name);
-			const children = enumMembers(node, qn);
-			const decl = leaf(node, owner, "enum", name, visibility, null, children, doc);
-			return decl === undefined ? [] : [decl];
-		}
-		case CONST_DECLARATION:
-		case CONST_TYPE_DECLARATION: {
-			// locals.scm: const_declaration → constant; const_type_declaration → type (value const with explicit type).
-			const name = firstIdentifier(node);
-			// `_ :: runtime` package aliases are not outline decls.
-			if (name === "_") return [];
-			const decl = leaf(node, owner, "constant", name, visibility, null, [], doc);
-			return decl === undefined ? [] : [decl];
-		}
-		case VAR_DECLARATION: {
-			// `name: Type` / `name: Type = value`
-			const name = firstIdentifier(node);
-			if (name === "_") return [];
-			const decl = leaf(node, owner, "variable", name, visibility, null, [], doc);
-			return decl === undefined ? [] : [decl];
-		}
-		case VARIABLE_DECLARATION: {
-			// `name := value` — function when value is a procedure literal (locals.scm still marks var).
-			const name = firstIdentifier(node);
-			if (name === "_") return [];
-			const proc = node.namedChildren.find((child) => child.type === PROCEDURE);
-			if (proc !== undefined) {
-				const body = proc.namedChildren.find((child) => child.type === BLOCK) ?? null;
-				const decl = leaf(node, owner, "function", name, visibility, body, [], doc);
-				return decl === undefined ? [] : [decl];
-			}
-			const decl = leaf(node, owner, "variable", name, visibility, null, [], doc);
-			return decl === undefined ? [] : [decl];
-		}
-		case FOREIGN_BLOCK: {
-			// Package-level foreign { ... } — extract nested decls at same owner.
-			const block = node.namedChildren.find((child) => child.type === BLOCK);
-			if (block === undefined) return [];
-			const out: Decl[] = [];
-			for (const child of block.namedChildren) {
-				out.push(...declsFromNode(child, owner, source));
-			}
-			return out;
-		}
-		default:
-			return [];
-	}
-}
-
-function collectImports(root: Node): ImportRef[] {
-	const imports: ImportRef[] = [];
-	for (const node of root.namedChildren) {
-		if (node.type !== IMPORT_DECLARATION) continue;
-		const str = node.namedChildren.find((child) => child.type === "string");
-		const specifier = str === undefined ? node.text : unquote(str.text);
-		imports.push({
-			specifier,
-			startLine: startLine(node),
-			startOffset: node.startIndex,
-			endOffset: node.endIndex,
-			bindings: [],
-		});
-	}
-	return imports;
-}
-
-function extractOdin(tree: Tree, source: string): ExtractResult {
-	const root = tree.rootNode;
-	const decls: Decl[] = [];
-	for (const node of root.children) {
-		if (node.isNamed || node.isError) decls.push(...declsFromNode(node, "", source));
-	}
-	return { decls, imports: collectImports(root) };
-}
-
-export function odinAdapter(): GrammarAdapter {
-	return {
-		mode: "grammar",
-		id: "odin",
-		extensions: EXTENSIONS,
-		capabilities: CAPABILITIES,
-		importNoiseIdentifiers: IMPORT_NOISE,
-		extract: extractOdin,
-		resolveFileDep: resolveOdinFileDep,
-	};
-}

package/extensions/explore/ast/queries/impact.ts

@@ -1,250 +0,0 @@
-import type { ExploreEngine } from "../engine.ts";
-import type { Candidate } from "../identity.ts";
-import type { ExploreFileGraph, FileGraphEdge } from "../graph/file-graph.ts";
-import {
-	queryRelationships,
-	type RelationshipCertainty,
-	type RelationshipSite,
-	type RelationshipSiteKind,
-} from "../graph/relationships.ts";
-import { resolveCompositeTarget, type CompositeTarget } from "./composite-target.ts";
-
-const IMPACT_SECTION_LIMIT = 50;
-export const IMPACT_DEPTH_DEFAULT = 2;
-export const IMPACT_DEPTH_MAX = 5;
-
-export type ImpactMode = "all" | "deps" | "dependents";
-
-export type ImpactSectionId = "callees" | "fileImports" | "callers" | "fileImporters" | "transitiveDependents";
-
-export type ImpactSiteRow = {
-	kind: "site";
-	path: string;
-	line: number;
-	siteKind: RelationshipSiteKind;
-	name: string;
-	preview: string;
-	certainty: RelationshipCertainty;
-};
-
-export type ImpactFileRow = {
-	kind: "file";
-	path: string | undefined;
-	externalId: string | undefined;
-	depth: number;
-};
-
-export type ImpactRow = ImpactSiteRow | ImpactFileRow;
-
-export type ImpactSection = {
-	id: ImpactSectionId;
-	label: string;
-	rows: ImpactRow[];
-	limit: number;
-	omitted: boolean;
-};
-
-export type ImpactTarget = CompositeTarget;
-
-export type ImpactResult =
-	| {
-			kind: "resolved";
-			target: ImpactTarget;
-			sections: ImpactSection[];
-			parseDegraded: boolean;
-			notes: string[];
-	  }
-	| { kind: "candidates"; candidates: Candidate[] }
-	| { kind: "notFound" }
-	| { kind: "error"; message: string };
-
-export type ImpactQueryArgs = {
-	engine: ExploreEngine;
-	graph: ExploreFileGraph;
-	scopePath: string;
-	targetPath: string | undefined;
-	name: string;
-	line: number | undefined;
-	depth: number;
-	mode: ImpactMode;
-	signal: AbortSignal;
-};
-
-function siteRows(sites: readonly RelationshipSite[]): ImpactSiteRow[] {
-	return sites.slice(0, IMPACT_SECTION_LIMIT).map((site) => ({
-		kind: "site" as const,
-		path: site.path,
-		line: site.line,
-		siteKind: site.kind,
-		name: site.name,
-		preview: site.preview,
-		certainty: site.certainty,
-	}));
-}
-
-function fileRowsFromEdges(edges: readonly FileGraphEdge[]): ImpactFileRow[] {
-	const rows: ImpactFileRow[] = [];
-	for (const edge of edges) {
-		if (rows.length >= IMPACT_SECTION_LIMIT) break;
-		if (edge.kind === "internal") {
-			rows.push({ kind: "file", path: edge.to, externalId: undefined, depth: 1 });
-		} else {
-			rows.push({ kind: "file", path: undefined, externalId: edge.id, depth: 1 });
-		}
-	}
-	return rows;
-}
-
-async function relationshipSection(
-	args: ImpactQueryArgs,
-	op: "callees" | "callers",
-	id: ImpactSectionId,
-	label: string,
-	target: ImpactTarget,
-): Promise<{ section: ImpactSection | undefined; parseDegraded: boolean; note: string | undefined }> {
-	const result = await queryRelationships({
-		engine: args.engine,
-		graph: args.graph,
-		scopePath: args.scopePath,
-		op,
-		targetPath: target.path,
-		name: target.name,
-		line: target.startLine,
-		resultLimit: IMPACT_SECTION_LIMIT,
-		signal: args.signal,
-	});
-	if (result.kind === "resolved") {
-		const rows = siteRows(result.sites);
-		if (rows.length === 0) return { section: undefined, parseDegraded: result.parseDegraded, note: undefined };
-		return {
-			section: {
-				id,
-				label,
-				rows,
-				limit: IMPACT_SECTION_LIMIT,
-				omitted: result.resultLimitReached || result.sites.length > IMPACT_SECTION_LIMIT,
-			},
-			parseDegraded: result.parseDegraded,
-			note: undefined,
-		};
-	}
-	if (result.kind === "error") {
-		return { section: undefined, parseDegraded: false, note: result.message };
-	}
-	return { section: undefined, parseDegraded: false, note: undefined };
-}
-
-/**
- * Blast radius composite: symbol callees/callers + file imports/importers + transitive file dependents.
- * Symbol hops are depth 1 only; `depth` applies to reverse file BFS only.
- */
-export async function queryImpact(args: ImpactQueryArgs): Promise<ImpactResult> {
-	const { graph, signal } = args;
-	if (args.depth < 1) {
-		return { kind: "error", message: "depth must be >= 1" };
-	}
-	const depth = Math.min(args.depth, IMPACT_DEPTH_MAX);
-
-	const resolved = await resolveCompositeTarget(
-		args.engine,
-		args.scopePath,
-		args.targetPath,
-		args.name,
-		args.line,
-		"impact",
-		signal,
-	);
-	if (resolved.kind !== "resolved") return resolved;
-
-	const target = resolved.value.target;
-	const mode = args.mode;
-	const wantDeps = mode === "all" || mode === "deps";
-	const wantDependents = mode === "all" || mode === "dependents";
-	const sections: ImpactSection[] = [];
-	const notes: string[] = [];
-	let parseDegraded = resolved.value.parseDegraded;
-
-	if (wantDeps) {
-		const callees = await relationshipSection(args, "callees", "callees", "callees", target);
-		if (callees.section !== undefined) sections.push(callees.section);
-		if (callees.parseDegraded) parseDegraded = true;
-		if (callees.note !== undefined) notes.push(callees.note);
-
-		try {
-			const edges = await graph.forwardEdges(target.path, signal);
-			const rows = fileRowsFromEdges(edges);
-			if (rows.length > 0) {
-				sections.push({
-					id: "fileImports",
-					label: "file imports",
-					rows,
-					limit: IMPACT_SECTION_LIMIT,
-					omitted: edges.length > IMPACT_SECTION_LIMIT,
-				});
-			}
-		} catch (error) {
-			notes.push(error instanceof Error ? error.message : String(error));
-		}
-	}
-
-	if (wantDependents) {
-		const callers = await relationshipSection(args, "callers", "callers", "callers", target);
-		if (callers.section !== undefined) sections.push(callers.section);
-		if (callers.parseDegraded) parseDegraded = true;
-		if (callers.note !== undefined) notes.push(callers.note);
-
-		try {
-			const reverseLimit = IMPACT_SECTION_LIMIT * Math.max(depth, 1);
-			const reverse = await graph.reverseDeps(target.path, depth, reverseLimit, signal);
-			const importerHits = reverse.hits.filter((hit) => hit.depth === 1);
-			const importers = importerHits.slice(0, IMPACT_SECTION_LIMIT).map(
-				(hit): ImpactFileRow => ({
-					kind: "file",
-					path: hit.path,
-					externalId: hit.externalId,
-					depth: hit.depth,
-				}),
-			);
-			if (importers.length > 0) {
-				sections.push({
-					id: "fileImporters",
-					label: "file importers",
-					rows: importers,
-					limit: IMPACT_SECTION_LIMIT,
-					omitted: importerHits.length > IMPACT_SECTION_LIMIT,
-				});
-			}
-
-			if (depth >= 2) {
-				const transitiveHits = reverse.hits.filter((hit) => hit.depth >= 2 && hit.depth <= depth);
-				const transitive = transitiveHits.slice(0, IMPACT_SECTION_LIMIT).map(
-					(hit): ImpactFileRow => ({
-						kind: "file",
-						path: hit.path,
-						externalId: hit.externalId,
-						depth: hit.depth,
-					}),
-				);
-				if (transitive.length > 0) {
-					sections.push({
-						id: "transitiveDependents",
-						label: "transitive dependents",
-						rows: transitive,
-						limit: IMPACT_SECTION_LIMIT,
-						omitted: transitiveHits.length > IMPACT_SECTION_LIMIT || reverse.resultLimitReached,
-					});
-				}
-			}
-		} catch (error) {
-			notes.push(error instanceof Error ? error.message : String(error));
-		}
-	}
-
-	return {
-		kind: "resolved",
-		target,
-		sections,
-		parseDegraded,
-		notes: [...new Set(notes)],
-	};
-}