refactor(agent): tighten harness environment and resources

2026-05-06 19:41:17 +02:00
parent e1ca501da8
commit 617d8b317d
21 changed files with 1305 additions and 389 deletions
@@ -1,4 +1,100 @@
-import type { PromptTemplate } from "./types.js";
+import { parse } from "yaml";
+import type { ExecutionEnv, FileInfo, PromptTemplate } from "./types.js";
+
+interface PromptTemplateFrontmatter {
+	description?: string;
+	"argument-hint"?: string;
+	[key: string]: unknown;
+}
+
+export async function loadPromptTemplates(env: ExecutionEnv, paths: string | string[]): Promise<PromptTemplate[]> {
+	const templates: PromptTemplate[] = [];
+	for (const path of Array.isArray(paths) ? paths : [paths]) {
+		const info = await safeFileInfo(env, path);
+		if (!info) continue;
+		const kind = await resolveKind(env, info);
+		if (kind === "directory") {
+			templates.push(...(await loadTemplatesFromDir(env, info.path)));
+		} else if (kind === "file" && info.name.endsWith(".md")) {
+			const template = await loadTemplateFromFile(env, info.path);
+			if (template) templates.push(template);
+		}
+	}
+	return templates;
+}
+
+async function loadTemplatesFromDir(env: ExecutionEnv, dir: string): Promise<PromptTemplate[]> {
+	const templates: PromptTemplate[] = [];
+	let entries: FileInfo[];
+	try {
+		entries = await env.listDir(dir);
+	} catch {
+		return templates;
+	}
+
+	for (const entry of entries.sort((a, b) => a.name.localeCompare(b.name))) {
+		const kind = await resolveKind(env, entry);
+		if (kind !== "file" || !entry.name.endsWith(".md")) continue;
+		const template = await loadTemplateFromFile(env, entry.path);
+		if (template) templates.push(template);
+	}
+	return templates;
+}
+
+async function loadTemplateFromFile(env: ExecutionEnv, filePath: string): Promise<PromptTemplate | null> {
+	try {
+		const rawContent = await env.readTextFile(filePath);
+		const { frontmatter, body } = parseFrontmatter<PromptTemplateFrontmatter>(rawContent);
+		const firstLine = body.split("\n").find((line) => line.trim());
+		let description = typeof frontmatter.description === "string" ? frontmatter.description : "";
+		if (!description && firstLine) {
+			description = firstLine.slice(0, 60);
+			if (firstLine.length > 60) description += "...";
+		}
+		return {
+			name: basenameEnvPath(filePath).replace(/\.md$/i, ""),
+			description,
+			content: body,
+		};
+	} catch {
+		return null;
+	}
+}
+
+async function safeFileInfo(env: ExecutionEnv, path: string): Promise<FileInfo | undefined> {
+	try {
+		return await env.fileInfo(path);
+	} catch {
+		return undefined;
+	}
+}
+
+async function resolveKind(env: ExecutionEnv, info: FileInfo): Promise<"file" | "directory" | undefined> {
+	if (info.kind === "file" || info.kind === "directory") return info.kind;
+	try {
+		const realPath = await env.realPath(info.path);
+		const target = await env.fileInfo(realPath);
+		return target.kind === "file" || target.kind === "directory" ? target.kind : undefined;
+	} catch {
+		return undefined;
+	}
+}
+
+function parseFrontmatter<T extends Record<string, unknown>>(content: string): { frontmatter: T; body: string } {
+	const normalized = content.replace(/\r\n/g, "\n").replace(/\r/g, "\n");
+	if (!normalized.startsWith("---")) return { frontmatter: {} as T, body: normalized };
+	const endIndex = normalized.indexOf("\n---", 3);
+	if (endIndex === -1) return { frontmatter: {} as T, body: normalized };
+	const yamlString = normalized.slice(4, endIndex);
+	const body = normalized.slice(endIndex + 4).trim();
+	return { frontmatter: (parse(yamlString) ?? {}) as T, body };
+}
+
+function basenameEnvPath(path: string): string {
+	const normalized = path.replace(/\/+$/, "");
+	const slashIndex = normalized.lastIndexOf("/");
+	return slashIndex === -1 ? normalized : normalized.slice(slashIndex + 1);
+}

 export function parseCommandArgs(argsString: string): string[] {
 	const args: string[] = [];