@henryqw/pi-codegraph 0.1.4 → 0.2.1

package/README.md

@@ -1,4 +1,4 @@
-# @henryqw/pi-codegraph
+# `@henryqw/pi-codegraph`
 
 Get a separate [CodeGraph](https://github.com/colbymchenry/codegraph) index when you start Pi in a new Git worktree. Use CodeGraph's explore tool through pi-mcp-adapter from Pi.
 
@@ -10,7 +10,14 @@ pi install npm:pi-mcp-adapter
 pi install npm:@henryqw/pi-codegraph
 ```
 
-Requires `codegraph` CLI on `PATH`, and [pi-mcp-adapter](https://github.com/nicobailon/pi-mcp-adapter).
+Requires the `codegraph` CLI on `PATH` and [pi-mcp-adapter](https://github.com/nicobailon/pi-mcp-adapter).
+
+## Works with
+
+| Package | Relationship | Purpose |
+| --- | --- | --- |
+| [`@colbymchenry/codegraph`](https://github.com/colbymchenry/codegraph) | Required | Builds the indexes and serves CodeGraph tools over MCP. |
+| [`pi-mcp-adapter`](https://github.com/nicobailon/pi-mcp-adapter) | Required | Connects the MCP server to Pi. |
 
 ## Use
 
@@ -20,7 +27,15 @@ From the primary checkout, initialize CodeGraph once:
 codegraph init --yes
 ```
 
-Then launch `pi` in a linked worktree. The adapter server is `henryqw_pi-codegraph__codegraph`. It discovers tools lazily; other CodeGraph tools remain available through `mcp`.
+Then launch `pi` in a linked worktree. When the primary checkout has an index, pi-codegraph initializes a separate worktree index and exposes CodeGraph tools through pi-mcp-adapter.
+
+| Surface | Type | Purpose |
+| --- | --- | --- |
+| `codegraph_explore` | tool | Lets agents explore the indexed code through CodeGraph. |
+| `mcp` | tool | Lets agents call other CodeGraph tools through the `henryqw_pi-codegraph__codegraph` adapter server. |
+| CodeGraph setup status | ui | Shows users setup progress and errors, and confirms a new worktree index is ready. |
+
+The adapter discovers tools lazily.
 
 ## Flow
 
@@ -28,6 +43,7 @@ Then launch `pi` in a linked worktree. The adapter server is `henryqw_pi-codegra
 - Initialization happens when Pi launches, not when Git creates the worktree.
 - Each linked worktree keeps its own `.codegraph`; databases are never copied or shared between branches.
 - Non-Git directories and repositories without a primary index are not initialized.
+- When loaded, the extension reports whether the local index is `indexed`, `missing`, or `indexing…` through Pi's status channel. The `pi-footer` extension places a compact `CG` badge first on its third line and temporarily shows `● CG` during direct CodeGraph tool calls.
 
 ## State and storage
 
@@ -35,7 +51,7 @@ The extension relies on CodeGraph's own state — each worktree maintains its ow
 
 ## Limits and recovery
 
-At startup, the extension checks for a loaded adapter and runs `codegraph --version`. If either prerequisite is unavailable, Pi warns and skips setup. Restart Pi or run `/reload` after fixing prerequisites.
+At startup, the extension checks for a loaded adapter and runs `codegraph --version`. If either prerequisite is unavailable, Pi warns, reports `prerequisites missing`, and skips setup. Restart Pi or run `/reload` after fixing prerequisites.
 
 Initialization uses an exclusive `pi-codegraph-init.lock` in the worktree's Git metadata. Failed or interrupted initialization keeps the lock so a partial database is not accepted. To recover: remove the lock directory with `rmdir` only after `codegraph index` succeeds, then run `/reload`.
 

package/extensions/codegraph.ts

@@ -37,12 +37,12 @@ async function initialize(pi: ExtensionAPI, ctx: ExtensionContext): Promise<void
 	}
 	if (prerequisites.length) {
 		const message = `pi-codegraph: setup skipped.\n${prerequisites.join("\n")}\nThen restart Pi or /reload.`;
-		ctx.ui.setStatus("pi-codegraph", "CodeGraph: prerequisites missing");
+		ctx.ui.setStatus("pi-codegraph", "pi-codegraph: prerequisites missing");
 		if (ctx.hasUI) ctx.ui.notify(message, "warning");
 		else console.warn(message);
 		return;
 	}
-	ctx.ui.setStatus("pi-codegraph", undefined);
+	ctx.ui.setStatus("pi-codegraph", "pi-codegraph: missing");
 	const rootResult = await pi.exec("git", ["rev-parse", "--show-toplevel"], { cwd: ctx.cwd, timeout: 10_000 });
 	if (rootResult.code !== 0 || rootResult.killed) {
 		if (!rootResult.killed && rootResult.stderr.includes("not a git repository")) return;
@@ -59,7 +59,8 @@ async function initialize(pi: ExtensionAPI, ctx: ExtensionContext): Promise<void
 	// Check the lock before the database: an in-progress/failed init can leave a partial DB.
 	const recovery = `Inspect CodeGraph in ${root}. If no initializer is running, run codegraph index in that directory, then remove ${lock} with rmdir and /reload.`;
 	const deadline = Date.now() + LOCK_WAIT_MS;
-	ctx.ui.setStatus("pi-codegraph", "CodeGraph: checking index…");
+	ctx.ui.setStatus("pi-codegraph", "pi-codegraph: checking index…");
+	let indexed = false;
 	try {
 		while (true) {
 			try {
@@ -68,26 +69,30 @@ async function initialize(pi: ExtensionAPI, ctx: ExtensionContext): Promise<void
 			} catch (error) {
 				if ((error as NodeJS.ErrnoException).code !== "EEXIST") throw error;
 				if (Date.now() >= deadline) throw new Error(`Timed out waiting for CodeGraph initialization. ${recovery}`);
-				ctx.ui.setStatus("pi-codegraph", "CodeGraph: waiting for initialization…");
+				ctx.ui.setStatus("pi-codegraph", "pi-codegraph: indexing…");
 				await delay(250);
 			}
 		}
 		let attempted = false;
 		let succeeded = false;
 		try {
-			if (await hasIndex(root)) return;
+			if (await hasIndex(root)) {
+				indexed = true;
+				return;
+			}
 			const worktrees = await git(pi, root, ["worktree", "list", "--porcelain", "-z"]);
 			const first = worktrees.split("\0", 1)[0];
 			if (!first?.startsWith("worktree ")) throw new Error("Git returned an invalid primary worktree.");
 			const primary = first.slice("worktree ".length);
 			if (!primary || primary === root || !(await hasIndex(primary))) return;
-			ctx.ui.setStatus("pi-codegraph", "CodeGraph: building worktree index…");
+			ctx.ui.setStatus("pi-codegraph", "pi-codegraph: indexing…");
 			attempted = true;
 			const result = await pi.exec("codegraph", ["init", "--yes", root], { cwd: root, timeout: INIT_TIMEOUT_MS });
 			if (result.code !== 0 || result.killed || !(await hasIndex(root))) {
 				throw new Error(`CodeGraph init failed (${result.killed ? "timed out or killed" : `exit ${result.code}`}): ${(result.stderr || result.stdout).trim().slice(-2000)}`);
 			}
 			succeeded = true;
+			indexed = true;
 			if (ctx.hasUI) {
 				ctx.ui.setWidget(WIDGET_KEY, ["pi-codegraph: index ready"]);
 				setTimeout(() => ctx.ui.setWidget(WIDGET_KEY, undefined), SUCCESS_TTL_MS);
@@ -102,7 +107,7 @@ async function initialize(pi: ExtensionAPI, ctx: ExtensionContext): Promise<void
 			if (!attempted || succeeded) await rmdir(lock);
 		}
 	} finally {
-		ctx.ui.setStatus("pi-codegraph", undefined);
+		ctx.ui.setStatus("pi-codegraph", indexed ? "pi-codegraph: indexed" : "pi-codegraph: missing");
 	}
 }
 
@@ -112,7 +117,7 @@ export default function codegraphExtension(pi: ExtensionAPI): void {
 			await initialize(pi, ctx);
 		} catch (error) {
 			const message = `pi-codegraph: ${error instanceof Error ? error.message : String(error)}`;
-			ctx.ui.setStatus("pi-codegraph", "CodeGraph: setup failed");
+			ctx.ui.setStatus("pi-codegraph", "pi-codegraph: setup failed");
 			if (ctx.hasUI) ctx.ui.notify(message, "error");
 			else throw new Error(message, { cause: error });
 		}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@henryqw/pi-codegraph",
-  "version": "0.1.4",
+  "version": "0.2.1",
   "description": "Initialize CodeGraph indexes in opted-in Git worktrees and expose CodeGraph through pi-mcp-adapter.",
   "keywords": [
     "pi-package",