pi-agent-flow 1.8.1 → 1.8.3

package/src/agents.ts

@@ -1,299 +0,0 @@
-/**
- * Flow discovery and configuration.
- *
- * Flows are Markdown files with YAML frontmatter that define name, description,
- * optional model/tools, and a system prompt body.
- *
- * Lookup locations:
- *   - User flows:    ~/.pi/agent/agents/*.md by default, or
- *                    $PI_CODING_AGENT_DIR/agents/*.md when the env var is set
- *   - Project flows: .pi/agents/*.md  (walks up from cwd)
- */
-
-import { parseFrontmatter } from "@mariozechner/pi-coding-agent";
-import * as fs from "node:fs";
-import * as os from "node:os";
-import * as path from "node:path";
-
-export type FlowScope = "user" | "project" | "both" | "bundled" | "all";
-
-export type FlowTier = "lite" | "flash" | "full";
-
-export interface FlowConfig {
-	name: string;
-	description: string;
-	tools?: string[];
-	model?: string;
-	thinking?: string;
-	maxDepth?: number;
-	inheritContext?: boolean;
-	tier?: FlowTier;
-	systemPrompt: string;
-	source: "user" | "project" | "bundled";
-	filePath: string;
-}
-
-export interface FlowDiscoveryResult {
-	flows: FlowConfig[];
-	projectFlowsDir: string | null;
-}
-
-/** Determine the model tier for a given flow name. */
-export function getFlowTier(flowName: string): FlowTier {
-	const normalized = flowName.toLowerCase().trim();
-	switch (normalized) {
-		case "scout":
-		case "debug":
-			return "lite";
-		case "build":
-		case "audit":
-			return "flash";
-		case "ideas":
-		case "craft":
-			return "full";
-		default:
-			return "flash";
-	}
-}
-
-/** Return the bundled flow names associated with a given tier. */
-export function getTierFlowNames(tier: FlowTier): string[] {
-	switch (tier) {
-		case "lite":
-			return ["scout", "debug"];
-		case "flash":
-			return ["build", "audit"];
-		case "full":
-			return ["ideas", "craft"];
-	}
-}
-
-// ---------------------------------------------------------------------------
-// Internal helpers
-// ---------------------------------------------------------------------------
-
-function isDirectory(p: string): boolean {
-	try { return fs.statSync(p).isDirectory(); } catch { return false; }
-}
-
-function getUserFlowsDir(): string {
-	const configDir = process.env["PI_CODING_AGENT_DIR"]?.trim() || path.join(os.homedir(), ".pi", "agent");
-	return path.join(configDir, "agents");
-}
-
-/** Get the bundled flows directory from the plugin's location. */
-function getBundledFlowsDir(): string {
-	// Method 1: import.meta.url (ESM)
-	// When source lives in src/, agents/ is one level up at the package root.
-	try {
-		if (import.meta.url) {
-			const pluginDir = path.dirname(new URL(import.meta.url).pathname);
-			// Check same directory first, then parent (for src/ layout)
-			for (const base of [pluginDir, path.dirname(pluginDir)]) {
-				const dir = path.join(base, "agents");
-				if (fs.existsSync(dir)) return dir;
-			}
-		}
-	} catch {}
-
-	// Method 2: __dirname (CommonJS / jiti)
-	try {
-		if (typeof __dirname !== "undefined") {
-			for (const base of [__dirname, path.dirname(__dirname)]) {
-				const dir = path.join(base, "agents");
-				if (fs.existsSync(dir)) return dir;
-			}
-		}
-	} catch {}
-
-	// Method 3: Find from require.resolve
-	try {
-		const resolved = require.resolve("pi-agent-flow/package.json");
-		const dir = path.join(path.dirname(resolved), "agents");
-		if (fs.existsSync(dir)) return dir;
-	} catch {}
-
-	// Fallback: cwd
-	return path.join(process.cwd(), "agents");
-}
-
-/** Walk up from `cwd` looking for a `.pi/agents` directory. */
-function findNearestProjectFlowsDir(cwd: string): string | null {
-	let dir = cwd;
-	while (true) {
-		const candidate = path.join(dir, ".pi", "agents");
-		if (isDirectory(candidate)) return candidate;
-		const parent = path.dirname(dir);
-		if (parent === dir) return null;
-		dir = parent;
-	}
-}
-
-/** Parse a single flow markdown file into a FlowConfig. Returns null on skip. */
-function parseFlowFile(filePath: string, source: "user" | "project" | "bundled"): FlowConfig | null {
-	let content: string;
-	try { content = fs.readFileSync(filePath, "utf-8"); } catch { return null; }
-
-	let parsed: { frontmatter: Record<string, unknown>; body: string };
-	try {
-		parsed = parseFrontmatter<Record<string, unknown>>(content);
-	} catch (err) {
-		const message = err instanceof Error ? err.message : String(err);
-		console.warn(`[pi-agent-flow] Skipping invalid flow file "${filePath}": ${message}`);
-		return null;
-	}
-
-	const frontmatter = parsed.frontmatter ?? {};
-	const body = parsed.body ?? "";
-
-	const name = typeof frontmatter.name === "string" ? frontmatter.name.trim().toLowerCase() : "";
-	const description = typeof frontmatter.description === "string" ? frontmatter.description.trim() : "";
-	if (!name || !description) {
-		if (!name) console.warn(`[pi-agent-flow] Skipping flow file "${filePath}": missing or empty 'name' field.`);
-		if (!description) console.warn(`[pi-agent-flow] Skipping flow file "${filePath}": missing or empty 'description' field.`);
-		return null;
-	}
-
-	// Warn about unknown frontmatter keys
-	const knownKeys = new Set([
-		"name", "description", "tools", "model", "thinking",
-		"maxDepth", "inheritContext", "tier",
-	]);
-	for (const key of Object.keys(frontmatter)) {
-		if (!knownKeys.has(key)) {
-			console.warn(`[pi-agent-flow] Unknown frontmatter key "${key}" in "${filePath}". This field will be ignored.`);
-		}
-	}
-
-	let tools: string[] | undefined;
-	if (typeof frontmatter.tools === "string") {
-		const parsedTools = frontmatter.tools
-			.split(",")
-			.map((t) => t.trim())
-			.filter(Boolean);
-		if (parsedTools.length > 0) tools = parsedTools;
-	} else if (Array.isArray(frontmatter.tools)) {
-		const parsedTools = frontmatter.tools
-			.filter((t): t is string => typeof t === "string")
-			.map((t) => t.trim())
-			.filter(Boolean);
-		if (parsedTools.length > 0) tools = parsedTools;
-	} else if (frontmatter.tools !== undefined) {
-		console.warn(
-			`[pi-agent-flow] Ignoring invalid tools field in "${filePath}". Expected a comma-separated string or string array.`,
-		);
-	}
-
-	let maxDepth: number | undefined;
-	if (typeof frontmatter.maxDepth === "number") {
-		maxDepth = frontmatter.maxDepth;
-	} else if (typeof frontmatter.maxDepth === "string") {
-		const parsed = Number(frontmatter.maxDepth);
-		if (Number.isFinite(parsed) && parsed >= 0) maxDepth = parsed;
-	}
-
-	let inheritContext: boolean | undefined;
-	if (typeof frontmatter.inheritContext === "boolean") {
-		inheritContext = frontmatter.inheritContext;
-	} else if (typeof frontmatter.inheritContext === "string") {
-		const normalized = frontmatter.inheritContext.trim().toLowerCase();
-		if (normalized === "true" || normalized === "yes" || normalized === "1") {
-			inheritContext = true;
-		} else if (normalized === "false" || normalized === "no" || normalized === "0") {
-			inheritContext = false;
-		} else {
-			console.warn(
-				`[pi-agent-flow] Ignoring invalid inheritContext value "${frontmatter.inheritContext}" in "${filePath}". Expected true/false.`,
-			);
-		}
-	} else if (frontmatter.inheritContext !== undefined) {
-		console.warn(
-			`[pi-agent-flow] Ignoring invalid inheritContext field in "${filePath}". Expected boolean or string.`,
-		);
-	}
-
-	// Tier: prefer explicit frontmatter, fall back to name-based inference
-	let tier: FlowTier | undefined;
-	if (typeof frontmatter.tier === "string") {
-		const normalized = frontmatter.tier.trim().toLowerCase();
-		if (normalized === "lite" || normalized === "flash" || normalized === "full") {
-			tier = normalized;
-		} else {
-			console.warn(`[pi-agent-flow] Ignoring invalid tier "${frontmatter.tier}" in "${filePath}". Expected lite, flash, or full.`);
-		}
-	}
-
-	return {
-		name,
-		description,
-		tools,
-		model: typeof frontmatter.model === "string" ? frontmatter.model : undefined,
-		thinking: typeof frontmatter.thinking === "string" ? frontmatter.thinking : undefined,
-		maxDepth,
-		inheritContext,
-		tier: tier ?? getFlowTier(name),
-		systemPrompt: body,
-		source,
-		filePath,
-	};
-}
-
-/** Load all flow definitions from a directory. */
-function loadFlowsFromDir(dir: string, source: "user" | "project" | "bundled"): FlowConfig[] {
-	if (!fs.existsSync(dir)) return [];
-
-	let entries: fs.Dirent[];
-	try { entries = fs.readdirSync(dir, { withFileTypes: true }); } catch { return []; }
-	entries.sort((a, b) => a.name.localeCompare(b.name));
-
-	const flows: FlowConfig[] = [];
-	for (const entry of entries) {
-		if (!entry.name.endsWith(".md")) continue;
-		if (!entry.isFile() && !entry.isSymbolicLink()) continue;
-
-		const flow = parseFlowFile(path.join(dir, entry.name), source);
-		if (flow) flows.push(flow);
-	}
-	return flows;
-}
-
-function mergeFlows(...groups: FlowConfig[][]): FlowConfig[] {
-	const flowMap = new Map<string, FlowConfig>();
-	for (const group of groups) {
-		for (const flow of group) flowMap.set(flow.name.toLowerCase(), flow);
-	}
-	return Array.from(flowMap.values());
-}
-
-// ---------------------------------------------------------------------------
-// Public API
-// ---------------------------------------------------------------------------
-
-/**
- * Discover all available flows according to the requested scope.
- *
- * Precedence is: bundled < user < project.
- */
-export function discoverFlows(cwd: string, scope: FlowScope): FlowDiscoveryResult {
-	const bundledFlowsDir = getBundledFlowsDir();
-	const userFlowsDir = getUserFlowsDir();
-	const projectFlowsDir = findNearestProjectFlowsDir(cwd);
-
-	const bundledFlows = scope === "user" || scope === "project" ? [] : loadFlowsFromDir(bundledFlowsDir, "bundled");
-	const userFlows = scope === "project" || scope === "bundled" ? [] : loadFlowsFromDir(userFlowsDir, "user");
-	const projectFlows = scope === "user" || scope === "bundled" || !projectFlowsDir ? [] : loadFlowsFromDir(projectFlowsDir, "project");
-
-	if (scope === "bundled") {
-		return { flows: bundledFlows, projectFlowsDir };
-	}
-	if (scope === "user") {
-		return { flows: mergeFlows(bundledFlows, userFlows), projectFlowsDir };
-	}
-	if (scope === "project") {
-		return { flows: mergeFlows(projectFlows), projectFlowsDir };
-	}
-	return {
-		flows: mergeFlows(bundledFlows, userFlows, projectFlows),
-		projectFlowsDir,
-	};
-}

package/src/ambient.d.ts

@@ -1,107 +0,0 @@
-/** Minimal ambient declarations for peer dependencies */
-
-declare module "@mariozechner/pi-coding-agent" {
-	export interface ExtensionAPI {
-		registerFlag(name: string, config: { description: string; type: string }): void;
-		getFlag(name: string): unknown;
-		getActiveTools(): string[];
-		on(event: string, callback: (...args: any[]) => any): void;
-		emit(event: string, ...args: any[]): void;
-		registerTool(tool: {
-			name: string;
-			label: string;
-			description: string;
-			parameters: any;
-			execute: (...args: any[]) => Promise<any>;
-			renderCall?: (...args: any[]) => any;
-			renderResult?: (...args: any[]) => any;
-		}): void;
-		setActiveTools(tools: string[]): void;
-	}
-	export interface ExtensionContext {
-		cwd: string;
-		hasUI: boolean;
-		ui: { confirm: (title: string, body: string) => Promise<boolean> };
-		sessionManager: { getSessionDir(): string; getHeader(): unknown; getBranch(): unknown[] };
-	}
-	export function parseFrontmatter<T extends Record<string, unknown>>(content: string): { frontmatter: T; body: string };
-	export function getMarkdownTheme(): any;
-	export const DEFAULT_MAX_BYTES: number;
-	export const DEFAULT_MAX_LINES: number;
-	export function truncateHead(text: string, options: { maxBytes?: number; maxLines?: number }): { content: string };
-	export function createBashToolDefinition(
-		cwd: string,
-		options?: {
-			shellPath?: string;
-			commandPrefix?: string;
-			operations?: any;
-			spawnHook?: any;
-		},
-	): {
-		name: string;
-		label: string;
-		description: string;
-		parameters: any;
-		execute: (...args: any[]) => Promise<any>;
-		renderCall?: (...args: any[]) => any;
-		renderResult?: (...args: any[]) => any;
-	};
-	/** Test-only exports provided by tests/__mocks__/pi-coding-agent.ts. */
-	export const bashToolExecuteCalls: any[][];
-	export function __setBashToolExecuteImpl(fn: (...args: any[]) => Promise<any>): void;
-	export function __resetBashToolMock(): void;
-}
-
-declare module "@mariozechner/pi-tui" {
-	export class Text {
-		constructor(text: string, width: number, height: number);
-		toString(): string;
-	}
-	export class TruncatedText {
-		constructor(text: string, paddingX?: number, paddingY?: number);
-		toString(): string;
-	}
-	export class Container {
-		children: any[];
-		addChild(child: any): void;
-	}
-	export class Markdown {
-		constructor(text: string, width: number, height: number, theme?: any);
-	}
-	export class Spacer {
-		constructor(height: number);
-	}
-}
-
-declare module "@mariozechner/pi-agent-core" {
-	export interface AgentToolResult<T = any> {
-		content: any[];
-		details?: T;
-		isError?: boolean;
-	}
-}
-
-declare module "@mariozechner/pi-ai" {
-	export interface Message {
-		role: string;
-		content: string | any[];
-		usage?: any;
-		model?: string;
-		stopReason?: string;
-		errorMessage?: string;
-	}
-}
-
-declare module "@sinclair/typebox" {
-	export const Type: {
-		Object: (properties: Record<string, any>, options?: any) => any;
-		String: (options?: any) => any;
-		Number: (options?: any) => any;
-		Array: (items: any, options?: any) => any;
-		Optional: (schema: any) => any;
-		Boolean: (options?: any) => any;
-		Union: (variants: any[], options?: any) => any;
-		Literal: (value: string) => any;
-	};
-	export type Static<T> = any;
-}