lattice-graph 0.2.0 → 0.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "lattice-graph",
-	"version": "0.2.0",
+	"version": "0.3.0",
 	"description": "Knowledge graph CLI for coding agents — navigate code through flows, not grep.",
 	"module": "src/main.ts",
 	"type": "module",
@@ -9,10 +9,12 @@
 	},
 	"files": [
 		"src",
+		"scripts",
 		"README.md",
 		"LICENSE"
 	],
 	"scripts": {
+		"postinstall": "bun scripts/postinstall.ts",
 		"dev": "bun src/main.ts",
 		"test": "bun test",
 		"lint": "bunx biome check src/ tests/",
@@ -44,11 +46,10 @@
 		"@types/bun": "latest",
 		"lefthook": "^2.1.4"
 	},
-	"peerDependencies": {
-		"typescript": "^5"
-	},
 	"dependencies": {
 		"commander": "^14.0.3",
-		"smol-toml": "^1.6.0"
+		"smol-toml": "^1.6.0",
+		"typescript": "^5",
+		"typescript-language-server": "^4"
 	}
 }

package/scripts/postinstall.ts

@@ -0,0 +1,69 @@
+#!/usr/bin/env bun
+/**
+ * Postinstall script — creates a minimal Python venv and installs zuban into it.
+ * This gives zuban a proper Python environment with typeshed stubs.
+ */
+import { existsSync } from "node:fs";
+import { join } from "node:path";
+
+const VENDOR_DIR = join(import.meta.dir, "..", "vendor");
+const VENV_DIR = join(VENDOR_DIR, "venv");
+
+function findPython(): string | undefined {
+	for (const cmd of ["python3", "python"]) {
+		if (Bun.which(cmd)) return cmd;
+	}
+	return undefined;
+}
+
+async function main() {
+	const isWindows = process.platform === "win32";
+	const zubanBin = join(VENV_DIR, isWindows ? "Scripts" : "bin", isWindows ? "zubanls.exe" : "zubanls");
+
+	if (existsSync(zubanBin)) {
+		console.log("zuban already installed");
+		return;
+	}
+
+	const python = findPython();
+	if (!python) {
+		console.warn("Warning: Python not found. Python support requires python3 in PATH.");
+		return;
+	}
+
+	console.log("Installing zuban for Python support...");
+
+	try {
+		// Create venv
+		const venvResult = Bun.spawnSync([python, "-m", "venv", VENV_DIR], {
+			stdout: "ignore",
+			stderr: "pipe",
+		});
+		if (venvResult.exitCode !== 0) {
+			throw new Error(`Failed to create venv: ${venvResult.stderr.toString()}`);
+		}
+
+		// Install zuban via pip
+		const pip = join(VENV_DIR, isWindows ? "Scripts" : "bin", "pip");
+		const pipResult = Bun.spawnSync([pip, "install", "zuban", "--quiet"], {
+			stdout: "ignore",
+			stderr: "pipe",
+		});
+		if (pipResult.exitCode !== 0) {
+			throw new Error(`Failed to install zuban: ${pipResult.stderr.toString()}`);
+		}
+
+		if (!existsSync(zubanBin)) {
+			throw new Error("zubanls binary not found after installation");
+		}
+
+		console.log("zuban installed successfully");
+	} catch (error) {
+		console.warn(
+			`Warning: Failed to install zuban: ${error instanceof Error ? error.message : error}. ` +
+				"Python support requires: pip install zuban",
+		);
+	}
+}
+
+main();

package/src/commands/init.ts

@@ -1,5 +1,5 @@
 import { existsSync, mkdirSync, writeFileSync } from "node:fs";
-import { join } from "node:path";
+import { join, resolve } from "node:path";
 import { err, ok, type Result } from "../types/result.ts";
 
 /**
@@ -102,25 +102,29 @@ function generateToml(languages: readonly string[], root: string): string {
 	return lines.join("\n");
 }
 
-/** Checks if language server binaries are available. */
+/** Checks if language server binaries are available (bundled or in PATH). */
 function checkLspAvailability(languages: readonly string[]): readonly string[] {
 	const warnings: string[] = [];
-	const commands: Record<string, string> = {
-		typescript: "typescript-language-server",
-		python: "zubanls",
-		go: "gopls",
+	const latticeRoot = resolve(import.meta.dir, "..", "..");
+
+	const checks: Record<string, { bundled: string; system: string; install: string }> = {
+		typescript: {
+			bundled: join(latticeRoot, "node_modules", ".bin", "typescript-language-server"),
+			system: "typescript-language-server",
+			install: "reinstall lattice-graph",
+		},
+		python: {
+			bundled: join(latticeRoot, "vendor", "venv", "bin", "zubanls"),
+			system: "zubanls",
+			install: "reinstall lattice-graph (or pip install zuban)",
+		},
 	};
+
 	for (const lang of languages) {
-		const cmd = commands[lang];
-		if (!cmd) continue;
-		if (!Bun.which(cmd)) {
-			const installHint =
-				lang === "python"
-					? "pip install zuban"
-					: lang === "typescript"
-						? "npm install -g typescript-language-server typescript"
-						: `install ${cmd}`;
-			warnings.push(`Warning: ${cmd} not found. Run '${installHint}' before 'lattice build'.`);
+		const check = checks[lang];
+		if (!check) continue;
+		if (!existsSync(check.bundled) && !Bun.which(check.system)) {
+			warnings.push(`Warning: ${check.system} not found. ${check.install}`);
 		}
 	}
 	return warnings;

package/src/lsp/builder.ts

@@ -1,5 +1,6 @@
 import type { Database } from "bun:sqlite";
-import { relative, resolve } from "node:path";
+import { existsSync } from "node:fs";
+import { join, relative, resolve } from "node:path";
 import { scanTags } from "../extract/tag-scanner.ts";
 import { discoverFiles } from "../files.ts";
 import {
@@ -39,18 +40,30 @@ type BuildStats = {
 	readonly durationMs: number;
 };
 
-/** Default LSP server commands per language. */
-const DEFAULT_LSP: Record<
-	string,
-	{ command: string; args: readonly string[]; languageId: string }
-> = {
-	typescript: {
-		command: "typescript-language-server",
-		args: ["--stdio"],
-		languageId: "typescript",
-	},
-	python: { command: "zubanls", args: [], languageId: "python" },
-};
+/** Resolves the LSP server binary for a language, checking bundled paths first. */
+function resolveLspServer(
+	language: string,
+): { command: string; args: readonly string[]; languageId: string } | undefined {
+	if (language === "typescript") {
+		// Check node_modules/.bin/ first (bundled with lattice-graph)
+		const bundled = join(
+			import.meta.dir,
+			"..",
+			"..",
+			"node_modules",
+			".bin",
+			"typescript-language-server",
+		);
+		const command = existsSync(bundled) ? bundled : "typescript-language-server";
+		return { command, args: ["--stdio"], languageId: "typescript" };
+	}
+	if (language === "python") {
+		const bundled = join(import.meta.dir, "..", "..", "vendor", "venv", "bin", "zubanls");
+		const command = existsSync(bundled) ? bundled : "zubanls";
+		return { command, args: [], languageId: "python" };
+	}
+	return undefined;
+}
 
 /**
  * Builds the knowledge graph by querying LSP servers for symbols and call hierarchy,
@@ -81,7 +94,7 @@ async function buildGraph(opts: BuildGraphOptions): Promise<BuildStats> {
 		if (files.length === 0) continue;
 		totalFiles += files.length;
 
-		const lsp = DEFAULT_LSP[langConfig.language];
+		const lsp = resolveLspServer(langConfig.language);
 		if (!lsp) continue;
 
 		const client = await createLspClient({

package/src/main.ts

@@ -37,7 +37,7 @@ import {
 import type { Node } from "./types/graph.ts";
 import { isOk, unwrap } from "./types/result.ts";
 
-const VERSION = "0.2.0";
+const VERSION = "0.3.0";
 
 const program = new Command();
 program.name("lattice").description("Knowledge graph CLI for coding agents").version(VERSION);