@henryqw/pi-herdr-rename 2.1.7 → 2.1.9

package/README.md

@@ -30,10 +30,10 @@ Requires Pi Coding Agent 0.84.3+.
 
 First real user prompt generates a title in the background after Pi expands skill and prompt-template shorthand. It does not delay main reply. Extension-injected prompts, empty prompts, and image-only input are ignored.
 
-Shared [`pi-task-models` config](../pi-task-models#config) at `~/.pi/agent/config/pi-task-models.json` routes task `pi-herdr-rename/rename` to the `fast` profile by default.
+Shared [`pi-task-models` config](../pi-task-models#config) at `~/.pi/agent/config/pi-task-models.json` can explicitly override the local `pi-herdr-rename/rename` declaration, which defaults to `fast`.
 
 Display titles are natural task phrases, preferably three or four words and always at most four words and 20 characters. Model classification stays internal: `refactor: update task logic` displays as `Update task logic` and maps to Git branch `refactor/update-task-logic`.
 
-In a linked worktree, a detached checkout or Herdr `worktree/...` branch is renamed; an existing non-generated branch stays. A generated workspace label such as `worktree-brave-meadow-4aa8` becomes display title; custom workspace names stay. Enclosing Herdr tab updates only when this pane is tab's only pane. Outside Herdr, only Pi session name changes.
+In a linked worktree, a detached checkout or Herdr `worktree/...` branch is renamed; an existing non-generated branch stays. A generated workspace label such as `worktree-brave-meadow-4aa8` becomes the display title automatically; `/rename` also replaces a custom workspace name. Enclosing Herdr tab updates only when this pane is tab's only pane. Outside Herdr, only Pi session name changes.
 
-Tries assigned profile primary, then fallback, while honoring configured thinking level. Never substitutes current session model. No viable route leaves titles unchanged. Resuming a session created by this version reapplies saved display title and semantic branch without another model request. Older titles receive no migration.
+Tries assigned profile primary, then fallback, while honoring configured thinking level. Never substitutes current session model. No viable route leaves titles unchanged. Resuming a session created by this version reapplies saved display title and semantic branch without another model request. Herdr and Git synchronization failures appear as warnings; cancellation by a newer rename remains silent. Older titles receive no migration.

package/extensions/rename.ts

@@ -3,10 +3,12 @@ import {
 	type ExtensionAPI,
 	type ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
-import { createHerdrClient } from "@henryqw/pi-herdr";
+import { createHerdrClient, withWorktreeLock } from "@henryqw/pi-herdr";
 import {
 	readTaskModelsConfig,
+	registerModelTask,
 	resolveConfiguredTaskRoutes,
+	type ModelTask,
 	type ResolvedTaskRoute,
 	type TaskRouteError,
 } from "@henryqw/pi-task-models";
@@ -18,7 +20,12 @@ const MAX_CONTEXT_CHARS = 2_000;
 const DISPLAY_MAX_WORDS = 4;
 const DISPLAY_MAX_CHARS = 20;
 const SEMANTIC_TYPE_MAX_CHARS = 12;
-const RENAME_TASK = "pi-herdr-rename/rename";
+export const RENAME_TASK = {
+	id: "pi-herdr-rename/rename",
+	label: "Conversation rename",
+	purpose: "Generate a short conversation title.",
+	defaultProfile: "fast",
+} as const satisfies ModelTask;
 const TITLE_STATE_TYPE = "pi-herdr-rename/title";
 const HERDR_DEFAULT_WORKTREE_NAME = /^(?:worktree[-/])?(?:brave|calm|clear|green|lucky|quiet|rapid|silver)-(?:river|cloud|field|forest|harbor|meadow|stone|valley)-[0-9a-f]{4}$/;
 const SEMANTIC_BRANCH = /^[a-z][a-z0-9-]{0,11}\/[a-z0-9]+(?:-[a-z0-9]+)*$/;
@@ -42,15 +49,22 @@ function configuredRenameRoutes(ctx: ExtensionContext): ResolvedTaskRoute[] {
 	}
 }
 
+function validSubject(subject: string): boolean {
+	return /^[a-z0-9]+(?: [a-z0-9]+)*$/.test(subject)
+		&& subject.length <= DISPLAY_MAX_CHARS
+		&& subject.split(" ").length <= DISPLAY_MAX_WORDS;
+}
+
+function isDisplayTitle(value: unknown): value is string {
+	if (typeof value !== "string" || !value) return false;
+	const subject = value[0].toLowerCase() + value.slice(1);
+	return validSubject(subject) && value === subject[0].toUpperCase() + subject.slice(1);
+}
+
 function parseGeneratedTitle(title: string): GeneratedTitle | undefined {
-	const match = /^([a-z][a-z0-9-]*): ([a-z0-9]+(?: [a-z0-9]+)*)$/.exec(title);
-	if (!match) return undefined;
+	const match = /^([a-z][a-z0-9-]*): (.+)$/.exec(title);
+	if (!match || match[1].length > SEMANTIC_TYPE_MAX_CHARS || !validSubject(match[2])) return undefined;
 	const subject = match[2];
-	if (
-		match[1].length > SEMANTIC_TYPE_MAX_CHARS ||
-		subject.length > DISPLAY_MAX_CHARS ||
-		subject.split(" ").length > DISPLAY_MAX_WORDS
-	) return undefined;
 	return {
 		display: subject[0].toUpperCase() + subject.slice(1),
 		branch: `${match[1]}/${subject.replaceAll(" ", "-")}`,
@@ -63,7 +77,7 @@ function savedTitle(ctx: ExtensionContext): GeneratedTitle | undefined {
 		.find((candidate) => candidate.type === "custom" && candidate.customType === TITLE_STATE_TYPE);
 	if (entry?.type !== "custom" || !entry.data || typeof entry.data !== "object" || Array.isArray(entry.data)) return undefined;
 	const { display, branch } = entry.data as { display?: unknown; branch?: unknown };
-	return typeof display === "string" && typeof branch === "string" && SEMANTIC_BRANCH.test(branch)
+	return isDisplayTitle(display) && typeof branch === "string" && SEMANTIC_BRANCH.test(branch)
 		? { display, branch }
 		: undefined;
 }
@@ -200,6 +214,7 @@ async function generateTitle(text: string, ctx: ExtensionContext, signal: AbortS
 }
 
 export default function herdrRenameExtension(pi: ExtensionAPI): void {
+	registerModelTask(pi, RENAME_TASK);
 	const herdr = createHerdrClient<{ signal: AbortSignal }>(pi.exec.bind(pi));
 	let latestUserText: string | undefined;
 	let automaticStarted = false;
@@ -216,6 +231,7 @@ export default function herdrRenameExtension(pi: ExtensionAPI): void {
 		displayTitle: string,
 		branchCandidate: string,
 		previousDisplayTitle: string | undefined,
+		forceWorkspaceRename: boolean,
 		request: number,
 		controller: AbortController,
 	): Promise<void> => {
@@ -247,6 +263,16 @@ export default function herdrRenameExtension(pi: ExtensionAPI): void {
 		const workspaceName = workspace?.label;
 		if (typeof workspaceName !== "string") throw new Error("Herdr workspace response omitted label.");
 		const worktree = workspace?.worktree;
+		if (
+			workspaceName !== displayTitle &&
+			(forceWorkspaceRename ||
+				(worktree?.is_linked_worktree === true &&
+					(HERDR_DEFAULT_WORKTREE_NAME.test(workspaceName) || workspaceName === previousDisplayTitle))) &&
+			isCurrent(request, controller)
+		) {
+			await herdr.run(["workspace", "rename", workspaceId, displayTitle], { signal: controller.signal });
+		}
+
 		const checkoutPath = worktree?.checkout_path;
 		if (worktree?.is_linked_worktree !== true || typeof checkoutPath !== "string" || !checkoutPath) return;
 
@@ -258,22 +284,18 @@ export default function herdrRenameExtension(pi: ExtensionAPI): void {
 			return result.stdout.trim();
 		};
 
-		const branch = await runGit(["branch", "--show-current"]);
-		if (!branch || branch.startsWith("worktree/")) {
+		await withWorktreeLock(checkoutPath, async () => {
 			if (!isCurrent(request, controller)) return;
-			const branches = (await runGit(["for-each-ref", "--format=%(refname:short)", "refs/heads"]))
-				.split("\n")
-				.filter(Boolean);
-			const semanticBranch = availableBranch(branchCandidate, branches);
-			await runGit(branch ? ["branch", "-m", semanticBranch] : ["switch", "-c", semanticBranch]);
-		}
-		if (
-			workspaceName !== displayTitle &&
-			(HERDR_DEFAULT_WORKTREE_NAME.test(workspaceName) || workspaceName === previousDisplayTitle) &&
-			isCurrent(request, controller)
-		) {
-			await herdr.run(["workspace", "rename", workspaceId, displayTitle], { signal: controller.signal });
-		}
+			const branch = await runGit(["branch", "--show-current"]);
+			if (!branch || branch.startsWith("worktree/")) {
+				if (!isCurrent(request, controller)) return;
+				const branches = (await runGit(["for-each-ref", "--format=%(refname:short)", "refs/heads"]))
+					.split("\n")
+					.filter(Boolean);
+				const semanticBranch = availableBranch(branchCandidate, branches);
+				await runGit(branch ? ["branch", "-m", semanticBranch] : ["switch", "-c", semanticBranch]);
+			}
+		});
 	};
 
 	const begin = () => {
@@ -300,10 +322,10 @@ export default function herdrRenameExtension(pi: ExtensionAPI): void {
 			const previousDisplayTitle = saved && pi.getSessionName() === saved.display ? saved.display : undefined;
 			pi.setSessionName(title.display);
 			pi.appendEntry(TITLE_STATE_TYPE, title);
-			await applyHerdr(title.display, title.branch, previousDisplayTitle, request, controller);
+			await applyHerdr(title.display, title.branch, previousDisplayTitle, manual, request, controller);
 			return title.display;
 		} catch (error) {
-			if (isCurrent(request, controller) && (manual || error instanceof RenameModelError)) {
+			if (isCurrent(request, controller)) {
 				ctx.ui.notify(error instanceof Error ? error.message : "Rename failed.", "warning");
 			}
 			return undefined;
@@ -328,7 +350,7 @@ export default function herdrRenameExtension(pi: ExtensionAPI): void {
 		latestUserText = latestSessionUserText(ctx);
 		try {
 			const taskModels = readTaskModelsConfig();
-			const profileName = taskModels.tasks[RENAME_TASK];
+			const profileName = taskModels.tasks[RENAME_TASK.id] ?? RENAME_TASK.defaultProfile;
 			if (!taskModels.profiles[profileName]) {
 				ctx.ui.notify(`Configure rename task profile ${profileName} with /task-models.`, "warning");
 			}
@@ -341,8 +363,12 @@ export default function herdrRenameExtension(pi: ExtensionAPI): void {
 		if (!title || title !== saved?.display) return;
 
 		const { request, controller } = begin();
-		void applyHerdr(title, saved.branch, saved.display, request, controller)
-			.catch(() => undefined)
+		void applyHerdr(title, saved.branch, saved.display, false, request, controller)
+			.catch((error) => {
+				if (isCurrent(request, controller)) {
+					ctx.ui.notify(error instanceof Error ? error.message : "Rename failed.", "warning");
+				}
+			})
 			.finally(() => finish(request, controller));
 	});
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@henryqw/pi-herdr-rename",
-  "version": "2.1.7",
+  "version": "2.1.9",
   "description": "Generate short Pi display titles and rename the current Herdr location.",
   "keywords": [
     "pi-package",
@@ -45,6 +45,6 @@
   },
   "dependencies": {
     "@henryqw/pi-herdr": "^0.3.0",
-    "@henryqw/pi-task-models": "^1.0.0"
+    "@henryqw/pi-task-models": "^3.0.0"
   }
 }