gsd-pi 2.39.0 → 2.40.0-dev.4a93031

package/src/resources/extensions/gsd/auto-post-unit.ts

@@ -168,8 +168,12 @@ export async function postUnitPreVerification(pctx: PostUnitContext, opts?: PreV
       const sliceTerminalUnits = new Set(["complete-slice", "run-uat"]);
       const effectiveFixLevel = sliceTerminalUnits.has(s.currentUnit.type) ? "all" as const : "task" as const;
       const report = await runGSDDoctor(s.basePath, { fix: true, scope: doctorScope, fixLevel: effectiveFixLevel });
+      // Human-readable fix notification with details
       if (report.fixesApplied.length > 0) {
-        ctx.ui.notify(`Post-hook: applied ${report.fixesApplied.length} fix(es).`, "info");
+        const fixSummary = report.fixesApplied.length <= 2
+          ? report.fixesApplied.join("; ")
+          : `${report.fixesApplied[0]}; +${report.fixesApplied.length - 1} more`;
+        ctx.ui.notify(`Doctor: ${fixSummary}`, "info");
       }
 
       // Proactive health tracking — filter to current milestone to avoid
@@ -181,7 +185,11 @@ export async function postUnitPreVerification(pctx: PostUnitContext, opts?: PreV
             i.unitId.startsWith(`${currentMilestoneId}/`))
         : report.issues;
       const summary = summarizeDoctorIssues(milestoneIssues);
-      recordHealthSnapshot(summary.errors, summary.warnings, report.fixesApplied.length);
+      // Pass issue details + scope for real-time visibility in the progress widget
+      const issueDetails = milestoneIssues
+        .filter(i => i.severity === "error" || i.severity === "warning")
+        .map(i => ({ code: i.code, message: i.message, severity: i.severity, unitId: i.unitId }));
+      recordHealthSnapshot(summary.errors, summary.warnings, report.fixesApplied.length, issueDetails, report.fixesApplied, doctorScope);
 
       // Check if we should escalate to LLM-assisted heal
       if (summary.errors > 0) {

package/src/resources/extensions/gsd/auto-prompts.ts

@@ -759,8 +759,8 @@ export async function checkNeedsRunUat(
     if (hasResult) return null;
   }
 
-  // Classify UAT type; unknown type → treat as human-experience (human review)
-  const uatType = extractUatType(uatContent) ?? "human-experience";
+  // Classify UAT type; default to artifact-driven (LLM-executed UATs are always artifact-driven)
+  const uatType = extractUatType(uatContent) ?? "artifact-driven";
 
   return { sliceId: sid, uatType };
 }
@@ -1403,7 +1403,7 @@ export async function buildRunUatPrompt(
   const inlinedContext = capPreamble(`## Inlined Context (preloaded — do not re-read these files)\n\n${inlined.join("\n\n---\n\n")}`);
 
   const uatResultPath = join(base, relSliceFile(base, mid, sliceId, "UAT-RESULT"));
-  const uatType = extractUatType(uatContent) ?? "human-experience";
+  const uatType = extractUatType(uatContent) ?? "artifact-driven";
 
   return loadPrompt("run-uat", {
     workingDirectory: base,

package/src/resources/extensions/gsd/auto-start.ts

@@ -56,7 +56,7 @@ import { readResourceVersion } from "./auto-worktree-sync.js";
 import { initMetrics } from "./metrics.js";
 import { initRoutingHistory } from "./routing-history.js";
 import { restoreHookState, resetHookState } from "./post-unit-hooks.js";
-import { resetProactiveHealing } from "./doctor-proactive.js";
+import { resetProactiveHealing, setLevelChangeCallback } from "./doctor-proactive.js";
 import { snapshotSkills } from "./skill-discovery.js";
 import { isDbAvailable } from "./gsd-db.js";
 import { hideFooter } from "./auto-dashboard.js";
@@ -415,6 +415,11 @@ export async function bootstrapAutoSession(
     resetHookState();
     restoreHookState(base);
     resetProactiveHealing();
+    // Notify user on health level transitions (green→yellow→red and back)
+    setLevelChangeCallback((_from, to, summary) => {
+      const level = to === "red" ? "error" : to === "yellow" ? "warning" : "info";
+      ctx.ui.notify(summary, level as "info" | "warning" | "error");
+    });
     s.autoStartTime = Date.now();
     s.resourceVersionOnStart = readResourceVersion();
     s.completedUnits = [];

package/src/resources/extensions/gsd/auto.ts

@@ -111,6 +111,7 @@ import {
   recordHealthSnapshot,
   checkHealEscalation,
   resetProactiveHealing,
+  setLevelChangeCallback,
   formatHealthSummary,
   getConsecutiveErrorUnits,
 } from "./doctor-proactive.js";
@@ -195,7 +196,7 @@ import {
   postUnitPostVerification,
 } from "./auto-post-unit.js";
 import { bootstrapAutoSession, type BootstrapDeps } from "./auto-start.js";
-import { autoLoop, resolveAgentEnd, type LoopDeps } from "./auto-loop.js";
+import { autoLoop, resolveAgentEnd, isSessionSwitchInFlight, type LoopDeps } from "./auto-loop.js";
 import {
   WorktreeResolver,
   type WorktreeResolverDeps,
@@ -687,6 +688,7 @@ export async function stopAuto(
     clearInFlightTools();
     clearSliceProgressCache();
     clearActivityLogState();
+    setLevelChangeCallback(null);
     resetProactiveHealing();
 
     // UI cleanup
@@ -1129,6 +1131,7 @@ const widgetStateAccessors: WidgetStateAccessors = {
   getCmdCtx: () => s.cmdCtx,
   getBasePath: () => s.basePath,
   isVerbose: () => s.verbose,
+  isSessionSwitching: isSessionSwitchInFlight,
 };
 
 // ─── Preconditions ────────────────────────────────────────────────────────────
@@ -1183,15 +1186,6 @@ function buildRecoveryContext(): import("./auto-timeout-recovery.js").RecoveryCo
   };
 }
 
-// Re-export recovery functions for external consumers
-export {
-  resolveExpectedArtifactPath,
-  verifyExpectedArtifact,
-  writeBlockerPlaceholder,
-  skipExecuteTask,
-  buildLoopRemediationSteps,
-} from "./auto-recovery.js";
-
 /**
  * Test-only: expose skip-loop state for unit tests.
  * Not part of the public API.
@@ -1327,3 +1321,12 @@ export async function dispatchHookUnit(
 
 // Direct phase dispatch → auto-direct-dispatch.ts
 export { dispatchDirectPhase } from "./auto-direct-dispatch.js";
+
+// Re-export recovery functions for external consumers
+export {
+  resolveExpectedArtifactPath,
+  verifyExpectedArtifact,
+  writeBlockerPlaceholder,
+  skipExecuteTask,
+  buildLoopRemediationSteps,
+} from "./auto-recovery.js";

package/src/resources/extensions/gsd/bootstrap/agent-end-recovery.ts

@@ -0,0 +1,142 @@
+import type { ExtensionAPI, ExtensionContext } from "@gsd/pi-coding-agent";
+
+import { checkAutoStartAfterDiscuss } from "../guided-flow.js";
+import { getAutoDashboardData, getAutoModeStartModel, isAutoActive, pauseAuto } from "../auto.js";
+import { getNextFallbackModel, isTransientNetworkError, resolveModelWithFallbacksForUnit } from "../preferences.js";
+import { classifyProviderError, pauseAutoForProviderError } from "../provider-error-pause.js";
+import { isSessionSwitchInFlight, resolveAgentEnd } from "../auto-loop.js";
+import { clearDiscussionFlowState } from "./write-gate.js";
+
+const networkRetryCounters = new Map<string, number>();
+const MAX_TRANSIENT_AUTO_RESUMES = 3;
+let consecutiveTransientErrors = 0;
+
+export async function handleAgentEnd(
+  pi: ExtensionAPI,
+  event: { messages: any[] },
+  ctx: ExtensionContext,
+): Promise<void> {
+  if (checkAutoStartAfterDiscuss()) {
+    clearDiscussionFlowState();
+    return;
+  }
+  if (!isAutoActive()) return;
+  if (isSessionSwitchInFlight()) return;
+
+  const lastMsg = event.messages[event.messages.length - 1];
+  if (lastMsg && "stopReason" in lastMsg && lastMsg.stopReason === "aborted") {
+    await pauseAuto(ctx, pi);
+    return;
+  }
+  if (lastMsg && "stopReason" in lastMsg && lastMsg.stopReason === "error") {
+    const errorDetail = "errorMessage" in lastMsg && lastMsg.errorMessage ? `: ${lastMsg.errorMessage}` : "";
+    const errorMsg = ("errorMessage" in lastMsg && lastMsg.errorMessage) ? String(lastMsg.errorMessage) : "";
+
+    if (isTransientNetworkError(errorMsg)) {
+      const currentModelId = ctx.model?.id ?? "unknown";
+      const retryKey = `network-retry:${currentModelId}`;
+      const currentRetries = networkRetryCounters.get(retryKey) ?? 0;
+      const maxRetries = 2;
+      if (currentRetries < maxRetries) {
+        networkRetryCounters.set(retryKey, currentRetries + 1);
+        const attempt = currentRetries + 1;
+        const delayMs = attempt * 3000;
+        ctx.ui.notify(`Network error on ${currentModelId}${errorDetail}. Retry ${attempt}/${maxRetries} in ${delayMs / 1000}s...`, "warning");
+        setTimeout(() => {
+          pi.sendMessage(
+            { customType: "gsd-auto-timeout-recovery", content: "Continue execution — retrying after transient network error.", display: false },
+            { triggerTurn: true },
+          );
+        }, delayMs);
+        return;
+      }
+      networkRetryCounters.delete(retryKey);
+      ctx.ui.notify(`Network retries exhausted for ${currentModelId}. Attempting model fallback.`, "warning");
+    }
+
+    const dash = getAutoDashboardData();
+    if (dash.currentUnit) {
+      const modelConfig = resolveModelWithFallbacksForUnit(dash.currentUnit.type);
+      if (modelConfig && modelConfig.fallbacks.length > 0) {
+        const availableModels = ctx.modelRegistry.getAvailable();
+        const nextModelId = getNextFallbackModel(ctx.model?.id, modelConfig);
+        if (nextModelId) {
+          networkRetryCounters.clear();
+          const slashIdx = nextModelId.indexOf("/");
+          const modelToSet = slashIdx !== -1
+            ? availableModels.find((m) => m.provider.toLowerCase() === nextModelId.substring(0, slashIdx).toLowerCase() && m.id.toLowerCase() === nextModelId.substring(slashIdx + 1).toLowerCase())
+            : (availableModels.find((m) => m.id === nextModelId && m.provider === ctx.model?.provider) ?? availableModels.find((m) => m.id === nextModelId));
+          if (modelToSet) {
+            const ok = await pi.setModel(modelToSet, { persist: false });
+            if (ok) {
+              ctx.ui.notify(`Model error${errorDetail}. Switched to fallback: ${nextModelId} and resuming.`, "warning");
+              pi.sendMessage({ customType: "gsd-auto-timeout-recovery", content: "Continue execution.", display: false }, { triggerTurn: true });
+              return;
+            }
+          }
+        }
+      }
+    }
+
+    const sessionModel = getAutoModeStartModel();
+    if (sessionModel) {
+      if (ctx.model?.id !== sessionModel.id || ctx.model?.provider !== sessionModel.provider) {
+        const startModel = ctx.modelRegistry.getAvailable().find((m) => m.provider === sessionModel.provider && m.id === sessionModel.id);
+        if (startModel) {
+          const ok = await pi.setModel(startModel, { persist: false });
+          if (ok) {
+            networkRetryCounters.clear();
+            ctx.ui.notify(`Model error${errorDetail}. Restored session model: ${sessionModel.provider}/${sessionModel.id} and resuming.`, "warning");
+            pi.sendMessage({ customType: "gsd-auto-timeout-recovery", content: "Continue execution.", display: false }, { triggerTurn: true });
+            return;
+          }
+        }
+      }
+    }
+
+    const classification = classifyProviderError(errorMsg);
+    const explicitRetryAfterMs = ("retryAfterMs" in lastMsg && typeof lastMsg.retryAfterMs === "number") ? lastMsg.retryAfterMs : undefined;
+    if (classification.isTransient) {
+      consecutiveTransientErrors += 1;
+    } else {
+      consecutiveTransientErrors = 0;
+    }
+    const baseRetryAfterMs = explicitRetryAfterMs ?? classification.suggestedDelayMs;
+    const retryAfterMs = classification.isTransient
+      ? baseRetryAfterMs * 2 ** Math.max(0, consecutiveTransientErrors - 1)
+      : baseRetryAfterMs;
+    const allowAutoResume = classification.isTransient && consecutiveTransientErrors <= MAX_TRANSIENT_AUTO_RESUMES;
+    if (classification.isTransient && !allowAutoResume) {
+      ctx.ui.notify(`Transient provider errors persisted after ${MAX_TRANSIENT_AUTO_RESUMES} auto-resume attempts. Pausing for manual review.`, "warning");
+    }
+    await pauseAutoForProviderError(ctx.ui, errorDetail, () => pauseAuto(ctx, pi), {
+      isRateLimit: classification.isRateLimit,
+      isTransient: allowAutoResume,
+      retryAfterMs,
+      resume: allowAutoResume
+        ? () => {
+          pi.sendMessage(
+            { customType: "gsd-auto-timeout-recovery", content: "Continue execution — provider error recovery delay elapsed.", display: false },
+            { triggerTurn: true },
+          );
+        }
+        : undefined,
+    });
+    return;
+  }
+
+  try {
+    consecutiveTransientErrors = 0;
+    networkRetryCounters.clear();
+    resolveAgentEnd(event);
+  } catch (err) {
+    const message = err instanceof Error ? err.message : String(err);
+    ctx.ui.notify(`Auto-mode error in agent_end handler: ${message}. Stopping auto-mode.`, "error");
+    try {
+      await pauseAuto(ctx, pi);
+    } catch {
+      // best-effort
+    }
+  }
+}
+

package/src/resources/extensions/gsd/bootstrap/db-tools.ts

@@ -0,0 +1,238 @@
+import { Type } from "@sinclair/typebox";
+import type { ExtensionAPI } from "@gsd/pi-coding-agent";
+
+import { findMilestoneIds, nextMilestoneId } from "../guided-flow.js";
+import { loadEffectiveGSDPreferences } from "../preferences.js";
+import { ensureDbOpen } from "./dynamic-tools.js";
+
+export function registerDbTools(pi: ExtensionAPI): void {
+  pi.registerTool({
+    name: "gsd_save_decision",
+    label: "Save Decision",
+    description:
+      "Record a project decision to the GSD database and regenerate DECISIONS.md. " +
+      "Decision IDs are auto-assigned — never provide an ID manually.",
+    promptSnippet: "Record a project decision to the GSD database (auto-assigns ID, regenerates DECISIONS.md)",
+    promptGuidelines: [
+      "Use gsd_save_decision when recording an architectural, pattern, library, or observability decision.",
+      "Decision IDs are auto-assigned (D001, D002, ...) — never guess or provide an ID.",
+      "All fields except revisable and when_context are required.",
+      "The tool writes to the DB and regenerates .gsd/DECISIONS.md automatically.",
+    ],
+    parameters: Type.Object({
+      scope: Type.String({ description: "Scope of the decision (e.g. 'architecture', 'library', 'observability')" }),
+      decision: Type.String({ description: "What is being decided" }),
+      choice: Type.String({ description: "The choice made" }),
+      rationale: Type.String({ description: "Why this choice was made" }),
+      revisable: Type.Optional(Type.String({ description: "Whether this can be revisited (default: 'Yes')" })),
+      when_context: Type.Optional(Type.String({ description: "When/context for the decision (e.g. milestone ID)" })),
+    }),
+    async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
+      const dbAvailable = await ensureDbOpen();
+      if (!dbAvailable) {
+        return {
+          content: [{ type: "text" as const, text: "Error: GSD database is not available. Cannot save decision." }],
+          details: { operation: "save_decision", error: "db_unavailable" } as any,
+        };
+      }
+      try {
+        const { saveDecisionToDb } = await import("../db-writer.js");
+        const { id } = await saveDecisionToDb(
+          {
+            scope: params.scope,
+            decision: params.decision,
+            choice: params.choice,
+            rationale: params.rationale,
+            revisable: params.revisable,
+            when_context: params.when_context,
+          },
+          process.cwd(),
+        );
+        return {
+          content: [{ type: "text" as const, text: `Saved decision ${id}` }],
+          details: { operation: "save_decision", id } as any,
+        };
+      } catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        process.stderr.write(`gsd-db: gsd_save_decision tool failed: ${msg}\n`);
+        return {
+          content: [{ type: "text" as const, text: `Error saving decision: ${msg}` }],
+          details: { operation: "save_decision", error: msg } as any,
+        };
+      }
+    },
+  });
+
+  pi.registerTool({
+    name: "gsd_update_requirement",
+    label: "Update Requirement",
+    description:
+      "Update an existing requirement in the GSD database and regenerate REQUIREMENTS.md. " +
+      "Provide the requirement ID (e.g. R001) and any fields to update.",
+    promptSnippet: "Update an existing GSD requirement by ID (regenerates REQUIREMENTS.md)",
+    promptGuidelines: [
+      "Use gsd_update_requirement to change status, validation, notes, or other fields on an existing requirement.",
+      "The id parameter is required — it must be an existing RXXX identifier.",
+      "All other fields are optional — only provided fields are updated.",
+      "The tool verifies the requirement exists before updating.",
+    ],
+    parameters: Type.Object({
+      id: Type.String({ description: "The requirement ID (e.g. R001, R014)" }),
+      status: Type.Optional(Type.String({ description: "New status (e.g. 'active', 'validated', 'deferred')" })),
+      validation: Type.Optional(Type.String({ description: "Validation criteria or proof" })),
+      notes: Type.Optional(Type.String({ description: "Additional notes" })),
+      description: Type.Optional(Type.String({ description: "Updated description" })),
+      primary_owner: Type.Optional(Type.String({ description: "Primary owning slice" })),
+      supporting_slices: Type.Optional(Type.String({ description: "Supporting slices" })),
+    }),
+    async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
+      const dbAvailable = await ensureDbOpen();
+      if (!dbAvailable) {
+        return {
+          content: [{ type: "text" as const, text: "Error: GSD database is not available. Cannot update requirement." }],
+          details: { operation: "update_requirement", id: params.id, error: "db_unavailable" } as any,
+        };
+      }
+      try {
+        const db = await import("../gsd-db.js");
+        const existing = db.getRequirementById(params.id);
+        if (!existing) {
+          return {
+            content: [{ type: "text" as const, text: `Error: Requirement ${params.id} not found.` }],
+            details: { operation: "update_requirement", id: params.id, error: "not_found" } as any,
+          };
+        }
+        const { updateRequirementInDb } = await import("../db-writer.js");
+        const updates: Record<string, string | undefined> = {};
+        if (params.status !== undefined) updates.status = params.status;
+        if (params.validation !== undefined) updates.validation = params.validation;
+        if (params.notes !== undefined) updates.notes = params.notes;
+        if (params.description !== undefined) updates.description = params.description;
+        if (params.primary_owner !== undefined) updates.primary_owner = params.primary_owner;
+        if (params.supporting_slices !== undefined) updates.supporting_slices = params.supporting_slices;
+        await updateRequirementInDb(params.id, updates, process.cwd());
+        return {
+          content: [{ type: "text" as const, text: `Updated requirement ${params.id}` }],
+          details: { operation: "update_requirement", id: params.id } as any,
+        };
+      } catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        process.stderr.write(`gsd-db: gsd_update_requirement tool failed: ${msg}\n`);
+        return {
+          content: [{ type: "text" as const, text: `Error updating requirement: ${msg}` }],
+          details: { operation: "update_requirement", id: params.id, error: msg } as any,
+        };
+      }
+    },
+  });
+
+  pi.registerTool({
+    name: "gsd_save_summary",
+    label: "Save Summary",
+    description:
+      "Save a summary, research, context, or assessment artifact to the GSD database and write it to disk. " +
+      "Computes the file path from milestone/slice/task IDs automatically.",
+    promptSnippet: "Save a GSD artifact (summary/research/context/assessment) to DB and disk",
+    promptGuidelines: [
+      "Use gsd_save_summary to persist structured artifacts (SUMMARY, RESEARCH, CONTEXT, ASSESSMENT).",
+      "milestone_id is required. slice_id and task_id are optional — they determine the file path.",
+      "The tool computes the relative path automatically: milestones/M001/M001-SUMMARY.md, milestones/M001/slices/S01/S01-SUMMARY.md, etc.",
+      "artifact_type must be one of: SUMMARY, RESEARCH, CONTEXT, ASSESSMENT.",
+    ],
+    parameters: Type.Object({
+      milestone_id: Type.String({ description: "Milestone ID (e.g. M001)" }),
+      slice_id: Type.Optional(Type.String({ description: "Slice ID (e.g. S01)" })),
+      task_id: Type.Optional(Type.String({ description: "Task ID (e.g. T01)" })),
+      artifact_type: Type.String({ description: "One of: SUMMARY, RESEARCH, CONTEXT, ASSESSMENT" }),
+      content: Type.String({ description: "The full markdown content of the artifact" }),
+    }),
+    async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
+      const dbAvailable = await ensureDbOpen();
+      if (!dbAvailable) {
+        return {
+          content: [{ type: "text" as const, text: "Error: GSD database is not available. Cannot save artifact." }],
+          details: { operation: "save_summary", error: "db_unavailable" } as any,
+        };
+      }
+      const validTypes = ["SUMMARY", "RESEARCH", "CONTEXT", "ASSESSMENT"];
+      if (!validTypes.includes(params.artifact_type)) {
+        return {
+          content: [{ type: "text" as const, text: `Error: Invalid artifact_type "${params.artifact_type}". Must be one of: ${validTypes.join(", ")}` }],
+          details: { operation: "save_summary", error: "invalid_artifact_type" } as any,
+        };
+      }
+      try {
+        let relativePath: string;
+        if (params.task_id && params.slice_id) {
+          relativePath = `milestones/${params.milestone_id}/slices/${params.slice_id}/tasks/${params.task_id}-${params.artifact_type}.md`;
+        } else if (params.slice_id) {
+          relativePath = `milestones/${params.milestone_id}/slices/${params.slice_id}/${params.slice_id}-${params.artifact_type}.md`;
+        } else {
+          relativePath = `milestones/${params.milestone_id}/${params.milestone_id}-${params.artifact_type}.md`;
+        }
+        const { saveArtifactToDb } = await import("../db-writer.js");
+        await saveArtifactToDb(
+          {
+            path: relativePath,
+            artifact_type: params.artifact_type,
+            content: params.content,
+            milestone_id: params.milestone_id,
+            slice_id: params.slice_id,
+            task_id: params.task_id,
+          },
+          process.cwd(),
+        );
+        return {
+          content: [{ type: "text" as const, text: `Saved ${params.artifact_type} artifact to ${relativePath}` }],
+          details: { operation: "save_summary", path: relativePath, artifact_type: params.artifact_type } as any,
+        };
+      } catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        process.stderr.write(`gsd-db: gsd_save_summary tool failed: ${msg}\n`);
+        return {
+          content: [{ type: "text" as const, text: `Error saving artifact: ${msg}` }],
+          details: { operation: "save_summary", error: msg } as any,
+        };
+      }
+    },
+  });
+
+  const reservedMilestoneIds = new Set<string>();
+  pi.registerTool({
+    name: "gsd_generate_milestone_id",
+    label: "Generate Milestone ID",
+    description:
+      "Generate the next milestone ID for a new GSD milestone. " +
+      "Scans existing milestones on disk and respects the unique_milestone_ids preference. " +
+      "Always use this tool when creating a new milestone — never invent milestone IDs manually.",
+    promptSnippet: "Generate a valid milestone ID (respects unique_milestone_ids preference)",
+    promptGuidelines: [
+      "ALWAYS call gsd_generate_milestone_id before creating a new milestone directory or writing milestone files.",
+      "Never invent or hardcode milestone IDs like M001, M002 — always use this tool.",
+      "Call it once per milestone you need to create. For multi-milestone projects, call it once for each milestone in sequence.",
+      "The tool returns the correct format based on project preferences (e.g. M001 or M001-r5jzab).",
+    ],
+    parameters: Type.Object({}),
+    async execute(_toolCallId, _params, _signal, _onUpdate, _ctx) {
+      try {
+        const basePath = process.cwd();
+        const existingIds = findMilestoneIds(basePath);
+        const uniqueEnabled = !!loadEffectiveGSDPreferences()?.preferences?.unique_milestone_ids;
+        const allIds = [...new Set([...existingIds, ...reservedMilestoneIds])];
+        const newId = nextMilestoneId(allIds, uniqueEnabled);
+        reservedMilestoneIds.add(newId);
+        return {
+          content: [{ type: "text" as const, text: newId }],
+          details: { operation: "generate_milestone_id", id: newId, existingCount: existingIds.length, reservedCount: reservedMilestoneIds.size, uniqueEnabled } as any,
+        };
+      } catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        return {
+          content: [{ type: "text" as const, text: `Error generating milestone ID: ${msg}` }],
+          details: { operation: "generate_milestone_id", error: msg } as any,
+        };
+      }
+    },
+  });
+}
+

package/src/resources/extensions/gsd/bootstrap/dynamic-tools.ts

@@ -0,0 +1,90 @@
+import { existsSync } from "node:fs";
+import { join } from "node:path";
+
+import type { ExtensionAPI } from "@gsd/pi-coding-agent";
+import { createBashTool, createEditTool, createReadTool, createWriteTool } from "@gsd/pi-coding-agent";
+
+import { DEFAULT_BASH_TIMEOUT_SECS } from "../constants.js";
+
+export async function ensureDbOpen(): Promise<boolean> {
+  try {
+    const db = await import("../gsd-db.js");
+    if (db.isDbAvailable()) return true;
+    const dbPath = join(process.cwd(), ".gsd", "gsd.db");
+    if (existsSync(dbPath)) {
+      return db.openDatabase(dbPath);
+    }
+    return false;
+  } catch {
+    return false;
+  }
+}
+
+export function registerDynamicTools(pi: ExtensionAPI): void {
+  const baseBash = createBashTool(process.cwd(), {
+    spawnHook: (ctx) => ({ ...ctx, cwd: process.cwd() }),
+  });
+  const dynamicBash = {
+    ...baseBash,
+    execute: async (
+      toolCallId: string,
+      params: { command: string; timeout?: number },
+      signal?: AbortSignal,
+      onUpdate?: unknown,
+      ctx?: unknown,
+    ) => {
+      const paramsWithTimeout = {
+        ...params,
+        timeout: params.timeout ?? DEFAULT_BASH_TIMEOUT_SECS,
+      };
+      return (baseBash as any).execute(toolCallId, paramsWithTimeout, signal, onUpdate, ctx);
+    },
+  };
+  pi.registerTool(dynamicBash as any);
+
+  const baseWrite = createWriteTool(process.cwd());
+  pi.registerTool({
+    ...baseWrite,
+    execute: async (
+      toolCallId: string,
+      params: { path: string; content: string },
+      signal?: AbortSignal,
+      onUpdate?: unknown,
+      ctx?: unknown,
+    ) => {
+      const fresh = createWriteTool(process.cwd());
+      return (fresh as any).execute(toolCallId, params, signal, onUpdate, ctx);
+    },
+  } as any);
+
+  const baseRead = createReadTool(process.cwd());
+  pi.registerTool({
+    ...baseRead,
+    execute: async (
+      toolCallId: string,
+      params: { path: string; offset?: number; limit?: number },
+      signal?: AbortSignal,
+      onUpdate?: unknown,
+      ctx?: unknown,
+    ) => {
+      const fresh = createReadTool(process.cwd());
+      return (fresh as any).execute(toolCallId, params, signal, onUpdate, ctx);
+    },
+  } as any);
+
+  const baseEdit = createEditTool(process.cwd());
+  pi.registerTool({
+    ...baseEdit,
+    execute: async (
+      toolCallId: string,
+      params: { path: string; oldText: string; newText: string },
+      signal?: AbortSignal,
+      onUpdate?: unknown,
+      ctx?: unknown,
+    ) => {
+      const fresh = createEditTool(process.cwd());
+      return (fresh as any).execute(toolCallId, params, signal, onUpdate, ctx);
+    },
+  } as any);
+}
+

package/src/resources/extensions/gsd/bootstrap/register-extension.ts

@@ -0,0 +1,46 @@
+import type { ExtensionAPI, ExtensionCommandContext } from "@gsd/pi-coding-agent";
+
+import { registerGSDCommand } from "../commands.js";
+import { registerExitCommand } from "../exit-command.js";
+import { registerWorktreeCommand } from "../worktree-command.js";
+import { registerDbTools } from "./db-tools.js";
+import { registerDynamicTools } from "./dynamic-tools.js";
+import { registerHooks } from "./register-hooks.js";
+import { registerShortcuts } from "./register-shortcuts.js";
+
+function installEpipeGuard(): void {
+  if (!process.listeners("uncaughtException").some((listener) => listener.name === "_gsdEpipeGuard")) {
+    const _gsdEpipeGuard = (err: Error): void => {
+      if ((err as NodeJS.ErrnoException).code === "EPIPE") {
+        process.exit(0);
+      }
+      if ((err as NodeJS.ErrnoException).code === "ENOENT" && (err as any).syscall?.startsWith("spawn")) {
+        process.stderr.write(`[gsd] spawn ENOENT: ${(err as any).path ?? "unknown"} — command not found\n`);
+        return;
+      }
+      throw err;
+    };
+    process.on("uncaughtException", _gsdEpipeGuard);
+  }
+}
+
+export function registerGsdExtension(pi: ExtensionAPI): void {
+  registerGSDCommand(pi);
+  registerWorktreeCommand(pi);
+  registerExitCommand(pi);
+
+  installEpipeGuard();
+
+  pi.registerCommand("kill", {
+    description: "Exit GSD immediately (no cleanup)",
+    handler: async (_args: string, _ctx: ExtensionCommandContext) => {
+      process.exit(0);
+    },
+  });
+
+  registerDynamicTools(pi);
+  registerDbTools(pi);
+  registerShortcuts(pi);
+  registerHooks(pi);
+}
+