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

package/src/core/tools/lsp/config.ts

@@ -0,0 +1,147 @@
+import { existsSync, readFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { extname, join } from "node:path";
+import type { ServerConfig } from "./types.js";
+
+export interface LspConfig {
+	servers: Record<string, ServerConfig>;
+}
+
+// Predefined server configurations with capabilities
+export const SERVERS: Record<string, ServerConfig> = {
+	"rust-analyzer": {
+		command: "rust-analyzer",
+		args: [],
+		fileTypes: [".rs"],
+		rootMarkers: ["Cargo.toml", "rust-analyzer.toml"],
+		initOptions: {
+			checkOnSave: { command: "clippy" },
+			cargo: { allFeatures: true },
+			procMacro: { enable: true },
+		},
+		capabilities: {
+			flycheck: true,
+			ssr: true,
+			expandMacro: true,
+			runnables: true,
+			relatedTests: true,
+		},
+	},
+	"typescript-language-server": {
+		command: "typescript-language-server",
+		args: ["--stdio"],
+		fileTypes: [".ts", ".tsx", ".js", ".jsx"],
+		rootMarkers: ["package.json", "tsconfig.json", "jsconfig.json"],
+	},
+	gopls: {
+		command: "gopls",
+		args: ["serve"],
+		fileTypes: [".go"],
+		rootMarkers: ["go.mod", "go.work"],
+	},
+	pyright: {
+		command: "pyright-langserver",
+		args: ["--stdio"],
+		fileTypes: [".py"],
+		rootMarkers: ["pyproject.toml", "setup.py", "requirements.txt", "Pipfile"],
+	},
+	zls: {
+		command: "zls",
+		args: [],
+		fileTypes: [".zig"],
+		rootMarkers: ["build.zig", "build.zig.zon", "zls.json"],
+	},
+	clangd: {
+		command: "clangd",
+		args: ["--background-index"],
+		fileTypes: [".c", ".cpp", ".cc", ".cxx", ".h", ".hpp"],
+		rootMarkers: ["compile_commands.json", "CMakeLists.txt", ".clangd"],
+	},
+	"lua-language-server": {
+		command: "lua-language-server",
+		args: [],
+		fileTypes: [".lua"],
+		rootMarkers: [".luarc.json", ".luarc.jsonc", ".luacheckrc"],
+	},
+};
+
+/**
+ * Check if any root marker file exists in the directory
+ */
+export function hasRootMarkers(cwd: string, markers: string[]): boolean {
+	return markers.some((marker) => existsSync(join(cwd, marker)));
+}
+
+/**
+ * Load LSP configuration.
+ *
+ * Priority:
+ * 1. Project-level config from .pi/lsp.json in cwd
+ * 2. User-level config from ~/.pi/lsp.json
+ * 3. Auto-detect from project markers + available binaries
+ */
+export function loadConfig(cwd: string): LspConfig {
+	// Try to load user config
+	const configPaths = [join(cwd, ".pi", "lsp.json"), join(homedir(), ".pi", "lsp.json")];
+
+	for (const configPath of configPaths) {
+		if (existsSync(configPath)) {
+			try {
+				const content = readFileSync(configPath, "utf-8");
+				const parsed = JSON.parse(content);
+				const servers = parsed.servers || parsed;
+
+				// Filter to only enabled servers with available commands
+				const available: Record<string, ServerConfig> = {};
+				for (const [name, config] of Object.entries(servers) as [string, ServerConfig][]) {
+					if (config.disabled) continue;
+					if (!Bun.which(config.command)) continue;
+					available[name] = config;
+				}
+
+				return { servers: available };
+			} catch {
+				// Ignore parse errors, fall through to auto-detect
+			}
+		}
+	}
+
+	// Auto-detect: find servers based on project markers AND available binaries
+	const detected: Record<string, ServerConfig> = {};
+
+	for (const [name, config] of Object.entries(SERVERS)) {
+		// Check if project has root markers for this language
+		if (!hasRootMarkers(cwd, config.rootMarkers)) continue;
+
+		// Check if the language server binary is available
+		if (!Bun.which(config.command)) continue;
+
+		detected[name] = config;
+	}
+
+	return { servers: detected };
+}
+
+/**
+ * Find the appropriate server for a file based on extension
+ */
+export function getServerForFile(config: LspConfig, filePath: string): [string, ServerConfig] | null {
+	const ext = extname(filePath).toLowerCase();
+
+	for (const [name, serverConfig] of Object.entries(config.servers)) {
+		if (serverConfig.fileTypes.includes(ext)) {
+			return [name, serverConfig];
+		}
+	}
+	return null;
+}
+
+/**
+ * Check if a server has a specific capability
+ */
+export function hasCapability(
+	config: ServerConfig,
+	capability: keyof NonNullable<ServerConfig["capabilities"]>,
+): boolean {
+	return config.capabilities?.[capability] === true;
+}

package/src/core/tools/lsp/edits.ts

@@ -0,0 +1,101 @@
+import { mkdir, rename, rm } from "node:fs/promises";
+import path from "node:path";
+import type { CreateFile, DeleteFile, RenameFile, TextDocumentEdit, TextEdit, WorkspaceEdit } from "./types.js";
+import { uriToFile } from "./utils.js";
+
+// =============================================================================
+// Text Edit Application
+// =============================================================================
+
+/**
+ * Apply text edits to a file.
+ * Edits are applied in reverse order (bottom-to-top) to preserve line/character indices.
+ */
+export async function applyTextEdits(filePath: string, edits: TextEdit[]): Promise<void> {
+	const content = await Bun.file(filePath).text();
+	const lines = content.split("\n");
+
+	// Sort edits in reverse order (bottom-to-top, right-to-left)
+	const sortedEdits = [...edits].sort((a, b) => {
+		if (a.range.start.line !== b.range.start.line) {
+			return b.range.start.line - a.range.start.line;
+		}
+		return b.range.start.character - a.range.start.character;
+	});
+
+	for (const edit of sortedEdits) {
+		const { start, end } = edit.range;
+
+		// Single-line edit: replace substring within same line
+		if (start.line === end.line) {
+			const line = lines[start.line] || "";
+			lines[start.line] = line.slice(0, start.character) + edit.newText + line.slice(end.character);
+		} else {
+			// Multi-line edit: splice across multiple lines
+			const startLine = lines[start.line] || "";
+			const endLine = lines[end.line] || "";
+			const newContent = startLine.slice(0, start.character) + edit.newText + endLine.slice(end.character);
+			lines.splice(start.line, end.line - start.line + 1, ...newContent.split("\n"));
+		}
+	}
+
+	await Bun.write(filePath, lines.join("\n"));
+}
+
+// =============================================================================
+// Workspace Edit Application
+// =============================================================================
+
+/**
+ * Apply a workspace edit (collection of file changes).
+ * Returns array of applied change descriptions.
+ */
+export async function applyWorkspaceEdit(edit: WorkspaceEdit, cwd: string): Promise<string[]> {
+	const applied: string[] = [];
+
+	// Handle changes map (legacy format)
+	if (edit.changes) {
+		for (const [uri, textEdits] of Object.entries(edit.changes)) {
+			const filePath = uriToFile(uri);
+			await applyTextEdits(filePath, textEdits);
+			applied.push(`Applied ${textEdits.length} edit(s) to ${path.relative(cwd, filePath)}`);
+		}
+	}
+
+	// Handle documentChanges array (modern format)
+	if (edit.documentChanges) {
+		for (const change of edit.documentChanges) {
+			if ("textDocument" in change && change.textDocument && "edits" in change && change.edits) {
+				// TextDocumentEdit
+				const docChange = change as TextDocumentEdit;
+				const filePath = uriToFile(docChange.textDocument.uri);
+				const textEdits = docChange.edits.filter((e): e is TextEdit => "range" in e && "newText" in e);
+				await applyTextEdits(filePath, textEdits);
+				applied.push(`Applied ${textEdits.length} edit(s) to ${path.relative(cwd, filePath)}`);
+			} else if ("kind" in change && change.kind) {
+				// Resource operations
+				if (change.kind === "create") {
+					const createOp = change as CreateFile;
+					const filePath = uriToFile(createOp.uri);
+					await mkdir(path.dirname(filePath), { recursive: true });
+					await Bun.write(filePath, "");
+					applied.push(`Created ${path.relative(cwd, filePath)}`);
+				} else if (change.kind === "rename") {
+					const renameOp = change as RenameFile;
+					const oldPath = uriToFile(renameOp.oldUri);
+					const newPath = uriToFile(renameOp.newUri);
+					await mkdir(path.dirname(newPath), { recursive: true });
+					await rename(oldPath, newPath);
+					applied.push(`Renamed ${path.relative(cwd, oldPath)} → ${path.relative(cwd, newPath)}`);
+				} else if (change.kind === "delete") {
+					const deleteOp = change as DeleteFile;
+					const filePath = uriToFile(deleteOp.uri);
+					await rm(filePath, { recursive: true });
+					applied.push(`Deleted ${path.relative(cwd, filePath)}`);
+				}
+			}
+		}
+	}
+
+	return applied;
+}