@oh-my-pi/pi-coding-agent 0.1.0

package/src/discovery/vscode.ts

@@ -0,0 +1,99 @@
+/**
+ * VS Code Provider
+ *
+ * Loads config from `.vscode` directory (project-only).
+ * Supports MCP server discovery from `mcp.json` with nested `mcp.servers` structure.
+ */
+
+import { registerProvider } from "../capability/index";
+import { type MCPServer, mcpCapability } from "../capability/mcp";
+import type { LoadContext, LoadResult } from "../capability/types";
+import { createSourceMeta, expandEnvVarsDeep, getProjectPath, parseJSON } from "./helpers";
+
+const PROVIDER_ID = "vscode";
+const DISPLAY_NAME = "VS Code";
+const PRIORITY = 20;
+
+// =============================================================================
+// MCP Servers
+// =============================================================================
+
+registerProvider<MCPServer>(mcpCapability.id, {
+	id: PROVIDER_ID,
+	displayName: DISPLAY_NAME,
+	description: "Load MCP servers from .vscode/mcp.json",
+	priority: PRIORITY,
+	load(ctx: LoadContext): LoadResult<MCPServer> {
+		const items: MCPServer[] = [];
+		const warnings: string[] = [];
+
+		// Project-only (VS Code doesn't support user-level MCP config)
+		const projectPath = getProjectPath(ctx, "vscode", "mcp.json");
+		if (projectPath && ctx.fs.isFile(projectPath)) {
+			const result = loadMCPConfig(ctx, projectPath, "project");
+			items.push(...result.items);
+			if (result.warnings) warnings.push(...result.warnings);
+		}
+
+		return { items, warnings };
+	},
+});
+
+/**
+ * Load MCP servers from a mcp.json file.
+ * VS Code uses nested structure: { "mcp": { "servers": { ... } } }
+ */
+function loadMCPConfig(ctx: LoadContext, path: string, level: "user" | "project"): LoadResult<MCPServer> {
+	const items: MCPServer[] = [];
+	const warnings: string[] = [];
+
+	const content = ctx.fs.readFile(path);
+	if (!content) {
+		warnings.push(`Failed to read ${path}`);
+		return { items, warnings };
+	}
+
+	const parsed = parseJSON<{ mcp?: { servers?: Record<string, unknown> } }>(content);
+	if (!parsed) {
+		warnings.push(`Invalid JSON in ${path}`);
+		return { items, warnings };
+	}
+
+	// VS Code uses nested structure: mcp.servers
+	const servers = parsed.mcp?.servers;
+	if (!servers || typeof servers !== "object") {
+		return { items, warnings };
+	}
+
+	for (const [name, config] of Object.entries(servers)) {
+		if (!config || typeof config !== "object") {
+			warnings.push(`Invalid config for server "${name}" in ${path}`);
+			continue;
+		}
+
+		const raw = config as Record<string, unknown>;
+
+		// Expand environment variables
+		const expanded = expandEnvVarsDeep(raw);
+
+		const server: MCPServer = {
+			name,
+			command: typeof expanded.command === "string" ? expanded.command : undefined,
+			args: Array.isArray(expanded.args) ? (expanded.args as string[]) : undefined,
+			env: expanded.env && typeof expanded.env === "object" ? (expanded.env as Record<string, string>) : undefined,
+			url: typeof expanded.url === "string" ? expanded.url : undefined,
+			headers:
+				expanded.headers && typeof expanded.headers === "object"
+					? (expanded.headers as Record<string, string>)
+					: undefined,
+			transport: ["stdio", "sse", "http"].includes(expanded.transport as string)
+				? (expanded.transport as "stdio" | "sse" | "http")
+				: undefined,
+			_source: createSourceMeta(PROVIDER_ID, path, level),
+		};
+
+		items.push(server);
+	}
+
+	return { items, warnings };
+}

package/src/discovery/windsurf.ts

@@ -0,0 +1,219 @@
+/**
+ * Windsurf (Codeium) Provider
+ *
+ * Loads configuration from Windsurf's config locations:
+ * - User: ~/.codeium/windsurf
+ * - Project: .windsurf
+ *
+ * Supports:
+ * - MCP servers from mcp_config.json
+ * - Rules from .windsurf/rules/*.md and ~/.codeium/windsurf/memories/global_rules.md
+ * - Legacy .windsurfrules file
+ */
+
+import { registerProvider } from "../capability/index";
+import { type MCPServer, mcpCapability } from "../capability/mcp";
+import { type Rule, ruleCapability } from "../capability/rule";
+import type { LoadContext, LoadResult } from "../capability/types";
+import {
+	createSourceMeta,
+	expandEnvVarsDeep,
+	getProjectPath,
+	getUserPath,
+	loadFilesFromDir,
+	parseFrontmatter,
+	parseJSON,
+} from "./helpers";
+
+const PROVIDER_ID = "windsurf";
+const DISPLAY_NAME = "Windsurf";
+const PRIORITY = 50;
+
+// =============================================================================
+// MCP Servers
+// =============================================================================
+
+function loadMCPServers(ctx: LoadContext): LoadResult<MCPServer> {
+	const items: MCPServer[] = [];
+	const warnings: string[] = [];
+
+	// User-level: ~/.codeium/windsurf/mcp_config.json
+	const userPath = getUserPath(ctx, "windsurf", "mcp_config.json");
+	if (userPath && ctx.fs.isFile(userPath)) {
+		const content = ctx.fs.readFile(userPath);
+		if (content) {
+			const config = parseJSON<{ mcpServers?: Record<string, unknown> }>(content);
+			if (config?.mcpServers) {
+				for (const [name, serverConfig] of Object.entries(config.mcpServers)) {
+					if (typeof serverConfig !== "object" || serverConfig === null) {
+						warnings.push(`Invalid server config for "${name}" in ${userPath}`);
+						continue;
+					}
+
+					const server = expandEnvVarsDeep(serverConfig as Record<string, unknown>);
+					items.push({
+						name,
+						command: server.command as string | undefined,
+						args: server.args as string[] | undefined,
+						env: server.env as Record<string, string> | undefined,
+						url: server.url as string | undefined,
+						headers: server.headers as Record<string, string> | undefined,
+						transport: server.type as "stdio" | "sse" | "http" | undefined,
+						_source: createSourceMeta(PROVIDER_ID, userPath, "user"),
+					});
+				}
+			}
+		}
+	}
+
+	// Project-level: .windsurf/mcp_config.json
+	const projectPath = getProjectPath(ctx, "windsurf", "mcp_config.json");
+	if (projectPath && ctx.fs.isFile(projectPath)) {
+		const content = ctx.fs.readFile(projectPath);
+		if (content) {
+			const config = parseJSON<{ mcpServers?: Record<string, unknown> }>(content);
+			if (config?.mcpServers) {
+				for (const [name, serverConfig] of Object.entries(config.mcpServers)) {
+					if (typeof serverConfig !== "object" || serverConfig === null) {
+						warnings.push(`Invalid server config for "${name}" in ${projectPath}`);
+						continue;
+					}
+
+					const server = expandEnvVarsDeep(serverConfig as Record<string, unknown>);
+					items.push({
+						name,
+						command: server.command as string | undefined,
+						args: server.args as string[] | undefined,
+						env: server.env as Record<string, string> | undefined,
+						url: server.url as string | undefined,
+						headers: server.headers as Record<string, string> | undefined,
+						transport: server.type as "stdio" | "sse" | "http" | undefined,
+						_source: createSourceMeta(PROVIDER_ID, projectPath, "project"),
+					});
+				}
+			}
+		}
+	}
+
+	return { items, warnings };
+}
+
+// =============================================================================
+// Rules
+// =============================================================================
+
+function loadRules(ctx: LoadContext): LoadResult<Rule> {
+	const items: Rule[] = [];
+	const warnings: string[] = [];
+
+	// User-level: ~/.codeium/windsurf/memories/global_rules.md
+	const userPath = getUserPath(ctx, "windsurf", "memories/global_rules.md");
+	if (userPath && ctx.fs.isFile(userPath)) {
+		const content = ctx.fs.readFile(userPath);
+		if (content) {
+			const { frontmatter, body } = parseFrontmatter(content);
+
+			// Validate and normalize globs
+			let globs: string[] | undefined;
+			if (Array.isArray(frontmatter.globs)) {
+				globs = frontmatter.globs.filter((g): g is string => typeof g === "string");
+			} else if (typeof frontmatter.globs === "string") {
+				globs = [frontmatter.globs];
+			}
+
+			items.push({
+				name: "global_rules",
+				path: userPath,
+				content: body,
+				globs,
+				alwaysApply: frontmatter.alwaysApply as boolean | undefined,
+				description: frontmatter.description as string | undefined,
+				ttsrTrigger: typeof frontmatter.ttsr_trigger === "string" ? frontmatter.ttsr_trigger : undefined,
+				_source: createSourceMeta(PROVIDER_ID, userPath, "user"),
+			});
+		}
+	}
+
+	// Project-level: .windsurf/rules/*.md
+	const projectRulesDir = getProjectPath(ctx, "windsurf", "rules");
+	if (projectRulesDir) {
+		const result = loadFilesFromDir<Rule>(ctx, projectRulesDir, PROVIDER_ID, "project", {
+			extensions: ["md"],
+			transform: (name, content, path, source) => {
+				const { frontmatter, body } = parseFrontmatter(content);
+				const ruleName = name.replace(/\.md$/, "");
+
+				// Validate and normalize globs
+				let globs: string[] | undefined;
+				if (Array.isArray(frontmatter.globs)) {
+					globs = frontmatter.globs.filter((g): g is string => typeof g === "string");
+				} else if (typeof frontmatter.globs === "string") {
+					globs = [frontmatter.globs];
+				}
+
+				return {
+					name: ruleName,
+					path,
+					content: body,
+					globs,
+					alwaysApply: frontmatter.alwaysApply as boolean | undefined,
+					description: frontmatter.description as string | undefined,
+					ttsrTrigger: typeof frontmatter.ttsr_trigger === "string" ? frontmatter.ttsr_trigger : undefined,
+					_source: source,
+				};
+			},
+		});
+		items.push(...result.items);
+		if (result.warnings) warnings.push(...result.warnings);
+	}
+
+	// Legacy: .windsurfrules in project root
+	const legacyPath = ctx.fs.walkUp(".windsurfrules", { file: true });
+	if (legacyPath) {
+		const content = ctx.fs.readFile(legacyPath);
+		if (content) {
+			const { frontmatter, body } = parseFrontmatter(content);
+
+			// Validate and normalize globs
+			let globs: string[] | undefined;
+			if (Array.isArray(frontmatter.globs)) {
+				globs = frontmatter.globs.filter((g): g is string => typeof g === "string");
+			} else if (typeof frontmatter.globs === "string") {
+				globs = [frontmatter.globs];
+			}
+
+			items.push({
+				name: "windsurfrules",
+				path: legacyPath,
+				content: body,
+				globs,
+				alwaysApply: frontmatter.alwaysApply as boolean | undefined,
+				description: frontmatter.description as string | undefined,
+				ttsrTrigger: typeof frontmatter.ttsr_trigger === "string" ? frontmatter.ttsr_trigger : undefined,
+				_source: createSourceMeta(PROVIDER_ID, legacyPath, "project"),
+			});
+		}
+	}
+
+	return { items, warnings };
+}
+
+// =============================================================================
+// Provider Registration
+// =============================================================================
+
+registerProvider<MCPServer>(mcpCapability.id, {
+	id: PROVIDER_ID,
+	displayName: DISPLAY_NAME,
+	description: "Load MCP servers from Windsurf config (mcp_config.json)",
+	priority: PRIORITY,
+	load: loadMCPServers,
+});
+
+registerProvider<Rule>(ruleCapability.id, {
+	id: PROVIDER_ID,
+	displayName: DISPLAY_NAME,
+	description: "Load rules from Windsurf (.windsurf/rules/*.md, memories/global_rules.md, .windsurfrules)",
+	priority: PRIORITY,
+	load: loadRules,
+});

package/src/index.ts

@@ -0,0 +1,192 @@
+// Core session management
+
+// Re-export TUI components for custom tool rendering
+export { Container, Markdown, Spacer, Text } from "@oh-my-pi/pi-tui";
+export {
+	AgentSession,
+	type AgentSessionConfig,
+	type AgentSessionEvent,
+	type AgentSessionEventListener,
+	type ModelCycleResult,
+	type PromptOptions,
+	type SessionStats,
+} from "./core/agent-session";
+// Auth and model registry
+export { type ApiKeyCredential, type AuthCredential, AuthStorage, type OAuthCredential } from "./core/auth-storage";
+// Compaction
+export {
+	type BranchPreparation,
+	type BranchSummaryResult,
+	type CollectEntriesResult,
+	type CompactionResult,
+	type CutPointResult,
+	calculateContextTokens,
+	collectEntriesForBranchSummary,
+	compact,
+	DEFAULT_COMPACTION_SETTINGS,
+	estimateTokens,
+	type FileOperations,
+	findCutPoint,
+	findTurnStartIndex,
+	type GenerateBranchSummaryOptions,
+	generateBranchSummary,
+	generateSummary,
+	getLastAssistantUsage,
+	prepareBranchEntries,
+	serializeConversation,
+	shouldCompact,
+} from "./core/compaction/index";
+// Custom commands
+export type {
+	CustomCommand,
+	CustomCommandAPI,
+	CustomCommandFactory,
+	CustomCommandSource,
+	CustomCommandsLoadResult,
+	LoadedCustomCommand,
+} from "./core/custom-commands/types";
+// Custom tools
+export type {
+	AgentToolUpdateCallback,
+	CustomTool,
+	CustomToolAPI,
+	CustomToolContext,
+	CustomToolFactory,
+	CustomToolSessionEvent,
+	CustomToolsLoadResult,
+	CustomToolUIContext,
+	ExecResult,
+	LoadedCustomTool,
+	RenderResultOptions,
+} from "./core/custom-tools/index";
+export { discoverAndLoadCustomTools, loadCustomTools } from "./core/custom-tools/index";
+export type * from "./core/hooks/index";
+// Hook system types and type guards
+export {
+	isBashToolResult,
+	isEditToolResult,
+	isFindToolResult,
+	isGrepToolResult,
+	isLsToolResult,
+	isReadToolResult,
+	isWriteToolResult,
+} from "./core/hooks/index";
+// Logging
+export { type Logger, logger } from "./core/logger";
+export { convertToLlm } from "./core/messages";
+export { ModelRegistry } from "./core/model-registry";
+// SDK for programmatic usage
+export {
+	type BuildSystemPromptOptions,
+	buildSystemPrompt,
+	type CreateAgentSessionOptions,
+	type CreateAgentSessionResult,
+	// Factory
+	createAgentSession,
+	createBashTool,
+	// Tool factories (for custom cwd)
+	createCodingTools,
+	createEditTool,
+	createFindTool,
+	createGrepTool,
+	createLsTool,
+	createReadOnlyTools,
+	createReadTool,
+	createWriteTool,
+	// Discovery
+	discoverAuthStorage,
+	discoverContextFiles,
+	discoverCustomTools,
+	discoverHooks,
+	discoverModels,
+	discoverSkills,
+	discoverSlashCommands,
+	type FileSlashCommand,
+	// Hook types
+	type HookAPI,
+	type HookContext,
+	type HookFactory,
+	loadSettings,
+	// Pre-built tools (use process.cwd())
+	readOnlyTools,
+} from "./core/sdk";
+export {
+	type BranchSummaryEntry,
+	buildSessionContext,
+	type CompactionEntry,
+	CURRENT_SESSION_VERSION,
+	type CustomEntry,
+	type CustomMessageEntry,
+	type FileEntry,
+	getLatestCompactionEntry,
+	type ModelChangeEntry,
+	migrateSessionEntries,
+	type NewSessionOptions,
+	parseSessionEntries,
+	type SessionContext,
+	type SessionEntry,
+	type SessionEntryBase,
+	type SessionHeader,
+	type SessionInfo,
+	SessionManager,
+	type SessionMessageEntry,
+	type ThinkingLevelChangeEntry,
+} from "./core/session-manager";
+export {
+	type CompactionSettings,
+	type LspSettings,
+	type RetrySettings,
+	type Settings,
+	SettingsManager,
+	type SkillsSettings,
+} from "./core/settings-manager";
+// Skills
+export {
+	formatSkillsForPrompt,
+	type LoadSkillsFromDirOptions,
+	type LoadSkillsResult,
+	loadSkills,
+	loadSkillsFromDir,
+	type Skill,
+	type SkillFrontmatter,
+	type SkillWarning,
+} from "./core/skills";
+// Tools
+export {
+	type BashToolDetails,
+	bashTool,
+	type CodingToolsOptions,
+	codingTools,
+	editTool,
+	type FindToolDetails,
+	findTool,
+	type GrepToolDetails,
+	grepTool,
+	type LsToolDetails,
+	lsTool,
+	type ReadToolDetails,
+	readTool,
+	type TruncationResult,
+	type WriteToolDetails,
+	type WriteToolOptions,
+	writeTool,
+} from "./core/tools/index";
+export type { FileDiagnosticsResult } from "./core/tools/lsp/index";
+// Main entry point
+export { main } from "./main";
+// UI components for hooks and custom tools
+export { BorderedLoader } from "./modes/interactive/components/bordered-loader";
+// Theme utilities for custom tools
+export { getMarkdownTheme } from "./modes/interactive/theme/theme";
+
+// TypeBox helper for string enums (convenience for custom tools)
+import { type TSchema, Type } from "@sinclair/typebox";
+export function StringEnum<T extends readonly string[]>(
+	values: T,
+	options?: { description?: string; default?: T[number] },
+): TSchema {
+	return Type.Union(
+		values.map((v) => Type.Literal(v)),
+		options,
+	);
+}