@bacnh85/pi-serena 0.1.1 → 0.1.3

package/README.md

@@ -55,6 +55,20 @@ After install or update, restart Pi or run `/reload` in an existing Pi session.
 
 All tool outputs are truncated to 50KB / 2000 lines to match Pi-friendly output limits. Most tools accept optional `timeout_ms`.
 
+### Pattern search
+
+Use the Pi-facing `pattern` field with `serena_search_for_pattern`:
+
+```json
+{
+  "pattern": "USB_HOST_DEVICE_OBJ|USB_HOST_DEVICE_STATE_ERROR_HOLDING",
+  "relative_path": "AmazonFreeRTOS",
+  "paths_include_glob": "**/*.h"
+}
+```
+
+The extension maps `pattern` to Serena's backend `substring_pattern` parameter internally, so users do not need to call the Serena implementation detail directly.
+
 ## Commands
 
 - `/serena-status [project]`

package/index.ts

@@ -1,5 +1,7 @@
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { Type } from "typebox";
+import { withFileMutationQueue } from "@earendil-works/pi-coding-agent";
+import path from "node:path";
 import { SerenaWorkerClient, type SerenaWorkerResponse } from "./worker";
 
 const DEFAULT_CONTEXT = "ide";
@@ -102,6 +104,10 @@ const searchPatternSchema = Type.Object({
 	relative_path: Type.Optional(Type.String({ description: "Optional file or directory restriction relative to the Serena project root." })),
 	paths_include_glob: Type.Optional(Type.String({ description: "Optional glob for included paths." })),
 	paths_exclude_glob: Type.Optional(Type.String({ description: "Optional glob for excluded paths." })),
+	context_lines_before: Type.Optional(Type.Number({ description: "Number of context lines before each match." })),
+	context_lines_after: Type.Optional(Type.Number({ description: "Number of context lines after each match." })),
+	restrict_search_to_code_files: Type.Optional(Type.Boolean({ description: "Restrict search to source/code files when supported by Serena." })),
+	multiline: Type.Optional(Type.Boolean({ description: "Treat the pattern as a multiline regular expression when supported by Serena." })),
 	max_answer_chars: MAX_CHARS_PARAM,
 });
 
@@ -133,6 +139,15 @@ function stripControlParams(params: Record<string, unknown>): { project: string;
 	return { project: normalizeProject(project), context: normalizeContext(context), timeoutMs: normalizeTimeoutMs(timeout_ms), params: toolParams };
 }
 
+export function normalizeSearchPatternParams(params: Record<string, unknown>): Record<string, unknown> {
+	const normalized = { ...params };
+	if (normalized.substring_pattern === undefined && normalized.pattern !== undefined) {
+		normalized.substring_pattern = normalized.pattern;
+	}
+	delete normalized.pattern;
+	return normalized;
+}
+
 function truncateText(text: string): string {
 	const lines = text.split("\n");
 	let output = lines.slice(0, OUTPUT_MAX_LINES).join("\n");
@@ -214,15 +229,27 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		return worker;
 	};
 
-	const callSerena = async (ctx: any, tool: string, rawParams: Record<string, unknown>) => {
+	const callSerena = async (ctx: any, tool: string, rawParams: Record<string, unknown>, lockPath?: string) => {
 		const { project, context, timeoutMs, params } = stripControlParams(rawParams);
-		const response = await getWorker(ctx).request({ action: "call", project, context, tool, params }, timeoutMs);
-		return {
-			content: [{ type: "text" as const, text: resultText(response) }],
-			details: response,
+		const run = async () => {
+			const response = await getWorker(ctx).request({ action: "call", project, context, tool, params }, timeoutMs);
+			return {
+				content: [{ type: "text" as const, text: resultText(response) }],
+				details: response,
+			};
 		};
+		return lockPath ? withFileMutationQueue(lockPath, run) : run();
+	};
+
+	const lockPathForRelativeFile = (rawParams: Record<string, unknown>): string | undefined => {
+		const project = normalizeProject(rawParams.project);
+		return typeof rawParams.relative_path === "string" && rawParams.relative_path.trim()
+			? path.resolve(project, rawParams.relative_path)
+			: undefined;
 	};
 
+	const lockPathForProject = (rawParams: Record<string, unknown>): string => path.resolve(normalizeProject(rawParams.project));
+
 	pi.registerTool({
 		name: "serena_status",
 		label: "Serena Status",
@@ -299,7 +326,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_replace_symbol_body only after serena_find_symbol identifies the exact symbol to replace."],
 		parameters: replaceBodySchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "replace_symbol_body", params);
+			return callSerena(ctx, "replace_symbol_body", params, lockPathForRelativeFile(params));
 		},
 	});
 
@@ -311,7 +338,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_insert_before_symbol for symbol-adjacent code insertion after locating the target symbol."],
 		parameters: insertSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "insert_before_symbol", params);
+			return callSerena(ctx, "insert_before_symbol", params, lockPathForRelativeFile(params));
 		},
 	});
 
@@ -323,7 +350,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_insert_after_symbol for adding sibling/helper symbols after a located symbol."],
 		parameters: insertSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "insert_after_symbol", params);
+			return callSerena(ctx, "insert_after_symbol", params, lockPathForRelativeFile(params));
 		},
 	});
 
@@ -335,7 +362,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_rename_symbol for cross-file code renames after finding the exact symbol and references."],
 		parameters: renameSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "rename_symbol", params);
+			return callSerena(ctx, "rename_symbol", params, lockPathForProject(params));
 		},
 	});
 
@@ -347,7 +374,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_safe_delete_symbol instead of text deletion when remaining references matter."],
 		parameters: safeDeleteSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "safe_delete_symbol", params);
+			return callSerena(ctx, "safe_delete_symbol", params, lockPathForRelativeFile(params));
 		},
 	});
 
@@ -359,7 +386,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_search_for_pattern for project-scoped code searches when symbol lookup is not enough."],
 		parameters: searchPatternSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "search_for_pattern", params);
+			return callSerena(ctx, "search_for_pattern", normalizeSearchPatternParams(params));
 		},
 	});
 
@@ -371,7 +398,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Prefer symbol-aware Serena edit tools for whole symbols; use serena_replace_content for non-symbol scoped replacements supported by Serena."],
 		parameters: replaceContentSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "replace_content", params);
+			return callSerena(ctx, "replace_content", params, lockPathForRelativeFile(params));
 		},
 	});
 
@@ -455,7 +482,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_write_memory after onboarding or when durable verified project knowledge should be available to Serena."],
 		parameters: writeMemorySchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "write_memory", params);
+			return callSerena(ctx, "write_memory", params, lockPathForProject(params));
 		},
 	});
 
@@ -467,7 +494,7 @@ export default function serenaToolsExtension(pi: ExtensionAPI) {
 		promptGuidelines: ["Use serena_delete_memory only when the user explicitly asks to remove a Serena memory."],
 		parameters: memoryNameSchema,
 		async execute(_id, params, _signal, _onUpdate, ctx) {
-			return callSerena(ctx, "delete_memory", params);
+			return callSerena(ctx, "delete_memory", params, lockPathForProject(params));
 		},
 	});
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bacnh85/pi-serena",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "Pi extension that provides Serena semantic code tools through a persistent worker.",
   "license": "MIT",
   "publishConfig": {

package/worker.ts

@@ -21,8 +21,8 @@ function piConfigDirs(): string[] {
 	return process.env.PI_CODING_AGENT_DIR ? [process.env.PI_CODING_AGENT_DIR] : [path.join(os.homedir(), ".pi", "agent"), path.join(os.homedir(), ".pi", "agents")];
 }
 
-function loadEnv(cwd = process.cwd()): void {
-	for (const file of [path.resolve(cwd, ".env.local"), path.resolve(cwd, ".env"), ...piConfigDirs().flatMap((dir) => [path.join(dir, ".env.local"), path.join(dir, ".env")])]) {
+function loadEnv(): void {
+	for (const file of piConfigDirs().flatMap((dir) => [path.join(dir, ".env.local"), path.join(dir, ".env")])) {
 		let text: string;
 		try { text = fs.readFileSync(file, "utf8"); } catch (error: any) { if (error?.code === "ENOENT") continue; throw error; }
 		for (const rawLine of text.split(/\r?\n/)) {