pi-readseek 0.2.6 → 0.3.0

package/index.ts

@@ -6,6 +6,9 @@ import { registerSgTool, isSgAvailable } from "./src/sg.js";
 import { registerWriteTool } from "./src/write.js";
 import { registerLsTool } from "./src/ls.js";
 import { registerFindTool } from "./src/find.js";
+import { registerReadseekCommand } from "./src/readseek-command.js";
+import { hasReadseekDir } from "./src/readseek-repo.js";
+import { readseekUpdate } from "./src/readseek-client.js";
 import { applyContextHygieneStaleContext } from "./src/context-application.js";
 import {
   createContextHygieneTracker,
@@ -83,6 +86,13 @@ export default function piReadseekExtension(pi: ExtensionAPI): void {
   registerWriteTool(pi, { onFileAnchored: noteRead });
   registerLsTool(pi);
   registerFindTool(pi);
+  registerReadseekCommand(pi);
+
+  pi.on("session_start", async (_event, ctx) => {
+    if (hasReadseekDir(ctx.cwd)) {
+      readseekUpdate(ctx.cwd).catch(() => {});
+    }
+  });
 
   pi.on("tool_call", (event: any) => {
     recordToolCall(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-readseek",
-  "version": "0.2.6",
+  "version": "0.3.0",
   "description": "Pi extension for readseek-backed hash-anchored read/edit/grep, structural code maps, structural search, and file exploration",
   "type": "module",
   "exports": {
@@ -39,7 +39,7 @@
     "node": ">=20.0.0"
   },
   "dependencies": {
-    "@jarkkojs/readseek": "^0.2.9",
+    "@jarkkojs/readseek": "^0.3.0",
     "diff": "^8.0.3",
     "ignore": "^7.0.5",
     "picomatch": "^4.0.4",

package/src/readseek-client.ts

@@ -135,6 +135,8 @@ function readseekPackageDir(): string {
 }
 
 export function readseekBinaryPath(): string {
+	if (process.env.READSEEK_BIN) return process.env.READSEEK_BIN;
+
 	const platformPackage = (() => {
 		switch (process.platform) {
 			case "darwin":
@@ -166,8 +168,8 @@ interface RunReadseekOptions {
 	stdin?: string;
 }
 
-async function runReadseek(args: string[], options: RunReadseekOptions = {}): Promise<unknown> {
-	const stdout = await new Promise<string>((resolve, reject) => {
+async function runReadseekRaw(args: string[], options: RunReadseekOptions = {}): Promise<string> {
+	return new Promise<string>((resolve, reject) => {
 		const stdin = options.stdin;
 		const stdio: StdioOptions = [stdin === undefined ? "ignore" : "pipe", "pipe", "pipe"];
 		const child = spawn(readseekBinaryPath(), args, { stdio, signal: options.signal });
@@ -212,6 +214,10 @@ async function runReadseek(args: string[], options: RunReadseekOptions = {}): Pr
 			else reject(new Error((stderr || `readseek exited with status ${code}`).replace(/^error:\s*/i, "")));
 		});
 	});
+}
+
+async function runReadseek(args: string[], options: RunReadseekOptions = {}): Promise<unknown> {
+	const stdout = await runReadseekRaw(args, options);
 	return JSON.parse(stdout) as unknown;
 }
 
@@ -383,3 +389,14 @@ export async function readseekMapContent(
 	);
 	return fileMapFromReadseekOutput(output, filePath, Buffer.byteLength(content, "utf8"));
 }
+
+export interface ReadseekUpdateStats {
+	created: number;
+	removed: number;
+	unchanged: number;
+}
+
+export async function readseekUpdate(cwd: string): Promise<ReadseekUpdateStats> {
+	const stdout = await runReadseekRaw(["update", cwd]);
+	return JSON.parse(stdout) as ReadseekUpdateStats;
+}

package/src/readseek-command.ts

@@ -0,0 +1,170 @@
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import { Key, matchesKey, truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+import { existsSync, readFileSync, rmSync, writeFileSync } from "node:fs";
+import { dirname, join, relative } from "node:path";
+import { findReadseekDir, initReadseekDir } from "./readseek-repo.js";
+import { readseekUpdate } from "./readseek-client.js";
+
+type ReadseekAction = "init" | "deinit" | "update" | null;
+
+function stripFromGitignore(dir: string): void {
+	const gitignorePath = join(dir, ".gitignore");
+	if (!existsSync(gitignorePath)) return;
+	const lines = readFileSync(gitignorePath, "utf-8").split("\n");
+	const filtered = lines.filter((line) => line.trim() !== "/.readseek");
+	writeFileSync(gitignorePath, filtered.join("\n"), "utf-8");
+}
+
+async function deinit(ctx: ExtensionContext): Promise<void> {
+	const dir = findReadseekDir(ctx.cwd);
+	if (!dir) {
+		ctx.ui.notify("No .readseek directory found", "info");
+		return;
+	}
+	const projectDir = dirname(dir);
+	rmSync(dir, { recursive: true, force: true });
+	stripFromGitignore(projectDir);
+	ctx.ui.notify("Removed .readseek/", "info");
+}
+
+async function init(ctx: ExtensionContext): Promise<void> {
+	const projectDir = ctx.cwd;
+	if (findReadseekDir(projectDir)) {
+		ctx.ui.notify(".readseek/ already exists", "info");
+		return;
+	}
+	initReadseekDir(projectDir);
+	ctx.ui.notify("Initialized .readseek/", "info");
+	await readseekUpdate(projectDir);
+}
+
+export function registerReadseekCommand(pi: ExtensionAPI): void {
+	const maybePi = pi as ExtensionAPI & {
+		registerCommand?: ExtensionAPI["registerCommand"];
+	};
+
+	maybePi.registerCommand?.("readseek", {
+		description: "Manage .readseek/ map cache",
+		handler: async (_args, ctx) => {
+			if (!ctx.hasUI) return;
+
+			const action = await new Promise<ReadseekAction>((resolve) => {
+				const initialized = findReadseekDir(ctx.cwd) !== null;
+
+				void ctx.ui.custom<ReadseekAction>(
+					(tui, theme, _kb, done) => {
+						return {
+							render(width: number): string[] {
+								const innerW = Math.max(1, width - 4);
+								const border = theme.fg("border", "│");
+								const dim = (s: string) => theme.fg("dim", s);
+								const accent = (s: string) => theme.fg("accent", s);
+								const borderFg = (s: string) => theme.fg("border", s);
+
+								function row(content: string): string {
+									const line = truncateToWidth(content, innerW);
+									const pad = Math.max(0, innerW - visibleWidth(line));
+									return `${border} ${line}${" ".repeat(pad)} ${border}`;
+								}
+
+								const lines: string[] = [];
+
+								const label = " Readseek ";
+								const topFill = borderFg(
+									"─".repeat(Math.max(0, width - 4 - visibleWidth(label))),
+								);
+								lines.push(
+									`${borderFg("╭─")}${accent(label)}${topFill}${borderFg("─╮")}`,
+								);
+
+								lines.push(row(""));
+
+								const readseekDir = findReadseekDir(ctx.cwd);
+								const statusDot = initialized
+									? theme.fg("success", "●")
+									: theme.fg("warning", "●");
+								const statusLabel = initialized ? "Initialized" : "Not initialized";
+								const statusColor = initialized
+									? theme.fg("success", statusLabel)
+									: theme.fg("warning", statusLabel);
+								const pathText = readseekDir
+									? dim(relative(ctx.cwd, readseekDir) || readseekDir)
+									: dim(".readseek");
+								lines.push(
+									row(`${statusDot} ${statusColor} ${dim("·")} ${pathText}`),
+								);
+
+								lines.push(row(""));
+
+								if (initialized) {
+									lines.push(
+										row(
+											`${accent("▶")} ${dim("[u]")} Update  ${dim("refresh map cache")}`,
+										),
+									);
+									lines.push(
+										row(`  ${dim("[d]")} Deinit  ${dim("remove .readseek/")}`),
+									);
+								} else {
+									lines.push(
+										row(
+											`${accent("▶")} ${dim("[i]")} Init  ${dim("create .readseek/ map cache")}`,
+										),
+									);
+								}
+
+								lines.push(row(""));
+								lines.push(row(dim("esc close")));
+
+								lines.push(
+									`${borderFg("╰")}${borderFg("─".repeat(Math.max(0, width - 2)))}${borderFg("╯")}`,
+								);
+
+								return lines;
+							},
+
+							handleInput(data: string): void {
+								if (matchesKey(data, Key.escape) || matchesKey(data, Key.ctrl("c"))) {
+									done(null);
+									return;
+								}
+
+								if (!initialized && (data === "i" || data === "I")) {
+									done("init");
+									return;
+								}
+								if (initialized && (data === "d" || data === "D")) {
+									done("deinit");
+									return;
+								}
+								if (initialized && (data === "u" || data === "U")) {
+									done("update");
+									return;
+								}
+							},
+
+							invalidate(): void {},
+						};
+					},
+					{
+						overlay: true,
+						overlayOptions: {
+							anchor: "center",
+							width: 60,
+							margin: 2,
+						},
+					},
+				).then((result) => {
+					resolve(result ?? null);
+				});
+			});
+
+			if (action === "init") await init(ctx);
+			else if (action === "deinit") await deinit(ctx);
+			else if (action === "update") {
+				await readseekUpdate(ctx.cwd);
+				ctx.ui.notify("Map cache updated", "info");
+			}
+		},
+	});
+}

package/src/readseek-repo.ts

@@ -0,0 +1,50 @@
+import { appendFileSync, existsSync, mkdirSync, readFileSync } from "node:fs";
+import { dirname, join, resolve } from "node:path";
+
+export function findReadseekDir(cwd: string): string | null {
+	const abs = resolve(cwd);
+	for (let dir = abs; ; dir = dirname(dir)) {
+		const candidate = join(dir, ".readseek");
+		if (existsSync(candidate)) return candidate;
+		const parent = dirname(dir);
+		if (parent === dir) break;
+	}
+	return null;
+}
+
+export function hasReadseekDir(cwd: string): boolean {
+	return findReadseekDir(cwd) !== null;
+}
+
+export function initReadseekDir(cwd: string): string {
+	const abs = resolve(cwd);
+	const readseekDir = join(abs, ".readseek");
+	const mapsDir = join(readseekDir, "maps");
+
+	if (existsSync(readseekDir)) {
+		throw new Error(`.readseek/ already exists in ${abs}`);
+	}
+
+	mkdirSync(mapsDir, { recursive: true });
+
+	const gitignore = join(abs, ".gitignore");
+	const entry = "/.readseek";
+	const needsAppend = (() => {
+		if (!existsSync(gitignore)) return true;
+		const contents = readFileSync(gitignore, "utf-8");
+		return !contents.split("\n").some((line) => line.trim() === entry);
+	})();
+
+	if (needsAppend) {
+		let prefix = "";
+		if (existsSync(gitignore)) {
+			const contents = readFileSync(gitignore, "utf-8");
+			if (contents.length > 0 && !contents.endsWith("\n")) {
+				prefix = "\n";
+			}
+		}
+		appendFileSync(gitignore, `${prefix}${entry}\n`, "utf-8");
+	}
+
+	return readseekDir;
+}