gsd-pi 2.38.0-dev.eeb3520 → 2.39.0-dev.38f11d1

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);
+}
+

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

@@ -0,0 +1,167 @@
+import { join } from "node:path";
+
+import type { ExtensionAPI, ExtensionContext } from "@gsd/pi-coding-agent";
+import { isToolCallEventType } from "@gsd/pi-coding-agent";
+
+import { buildMilestoneFileName, resolveMilestonePath, resolveSliceFile, resolveSlicePath } from "../paths.js";
+import { buildBeforeAgentStartResult } from "./system-context.js";
+import { handleAgentEnd } from "./agent-end-recovery.js";
+import { clearDiscussionFlowState, isDepthVerified, isQueuePhaseActive, markDepthVerified, resetWriteGateState, shouldBlockContextWrite } from "./write-gate.js";
+import { getDiscussionMilestoneId } from "../guided-flow.js";
+import { loadToolApiKeys } from "../commands-config.js";
+import { loadFile, saveFile, formatContinue } from "../files.js";
+import { deriveState } from "../state.js";
+import { getAutoDashboardData, isAutoActive, isAutoPaused, markToolEnd, markToolStart } from "../auto.js";
+import { isParallelActive, shutdownParallel } from "../parallel-orchestrator.js";
+import { saveActivityLog } from "../activity-log.js";
+import { maybeRenderGsdHeader } from "./register-shortcuts.js";
+
+export function registerHooks(pi: ExtensionAPI): void {
+  pi.on("session_start", async (_event, ctx) => {
+    resetWriteGateState();
+    maybeRenderGsdHeader(ctx);
+    loadToolApiKeys();
+    try {
+      const [{ getRemoteConfigStatus }, { getLatestPromptSummary }] = await Promise.all([
+        import("../../remote-questions/config.js"),
+        import("../../remote-questions/status.js"),
+      ]);
+      const status = getRemoteConfigStatus();
+      const latest = getLatestPromptSummary();
+      if (!status.includes("not configured")) {
+        const suffix = latest ? `\nLast remote prompt: ${latest.id} (${latest.status})` : "";
+        ctx.ui.notify(`${status}${suffix}`, status.includes("disabled") ? "warning" : "info");
+      }
+    } catch {
+      // ignore
+    }
+  });
+
+  pi.on("before_agent_start", async (event, ctx: ExtensionContext) => {
+    return buildBeforeAgentStartResult(event, ctx);
+  });
+
+  pi.on("agent_end", async (event, ctx: ExtensionContext) => {
+    await handleAgentEnd(pi, event, ctx);
+  });
+
+  pi.on("session_before_compact", async () => {
+    if (isAutoActive() || isAutoPaused()) {
+      return { cancel: true };
+    }
+    const basePath = process.cwd();
+    const state = await deriveState(basePath);
+    if (!state.activeMilestone || !state.activeSlice || !state.activeTask) return;
+    if (state.phase !== "executing") return;
+
+    const sliceDir = resolveSlicePath(basePath, state.activeMilestone.id, state.activeSlice.id);
+    if (!sliceDir) return;
+
+    const existingFile = resolveSliceFile(basePath, state.activeMilestone.id, state.activeSlice.id, "CONTINUE");
+    if (existingFile && await loadFile(existingFile)) return;
+    const legacyContinue = join(sliceDir, "continue.md");
+    if (await loadFile(legacyContinue)) return;
+
+    const continuePath = join(sliceDir, `${state.activeSlice.id}-CONTINUE.md`);
+    await saveFile(continuePath, formatContinue({
+      frontmatter: {
+        milestone: state.activeMilestone.id,
+        slice: state.activeSlice.id,
+        task: state.activeTask.id,
+        step: 0,
+        totalSteps: 0,
+        status: "compacted" as const,
+        savedAt: new Date().toISOString(),
+      },
+      completedWork: `Task ${state.activeTask.id} (${state.activeTask.title}) was in progress when compaction occurred.`,
+      remainingWork: "Check the task plan for remaining steps.",
+      decisions: "Check task summary files for prior decisions.",
+      context: "Session was auto-compacted by Pi. Resume with /gsd.",
+      nextAction: `Resume task ${state.activeTask.id}: ${state.activeTask.title}.`,
+    }));
+  });
+
+  pi.on("session_shutdown", async (_event, ctx: ExtensionContext) => {
+    if (isParallelActive()) {
+      try {
+        await shutdownParallel(process.cwd());
+      } catch {
+        // best-effort
+      }
+    }
+    if (!isAutoActive() && !isAutoPaused()) return;
+    const dash = getAutoDashboardData();
+    if (dash.currentUnit) {
+      saveActivityLog(ctx, dash.basePath, dash.currentUnit.type, dash.currentUnit.id);
+    }
+  });
+
+  pi.on("tool_call", async (event) => {
+    if (!isToolCallEventType("write", event)) return;
+    const result = shouldBlockContextWrite(
+      event.toolName,
+      event.input.path,
+      getDiscussionMilestoneId(),
+      isDepthVerified(),
+      isQueuePhaseActive(),
+    );
+    if (result.block) return result;
+  });
+
+  pi.on("tool_result", async (event) => {
+    if (event.toolName !== "ask_user_questions") return;
+    const milestoneId = getDiscussionMilestoneId();
+    if (!milestoneId) return;
+
+    const details = event.details as any;
+    if (details?.cancelled || !details?.response) return;
+
+    const questions: any[] = (event.input as any)?.questions ?? [];
+    for (const question of questions) {
+      if (typeof question.id === "string" && question.id.includes("depth_verification")) {
+        markDepthVerified();
+        break;
+      }
+    }
+
+    const basePath = process.cwd();
+    const milestoneDir = resolveMilestonePath(basePath, milestoneId);
+    if (!milestoneDir) return;
+
+    const discussionPath = join(milestoneDir, buildMilestoneFileName(milestoneId, "DISCUSSION"));
+    const timestamp = new Date().toISOString();
+    const lines: string[] = [`## Exchange — ${timestamp}`, ""];
+    for (const question of questions) {
+      lines.push(`### ${question.header ?? "Question"}`, "", question.question ?? "");
+      if (Array.isArray(question.options)) {
+        lines.push("");
+        for (const opt of question.options) {
+          lines.push(`- **${opt.label}** — ${opt.description ?? ""}`);
+        }
+      }
+      const answer = details.response?.answers?.[question.id];
+      if (answer) {
+        lines.push("");
+        const selected = Array.isArray(answer.selected) ? answer.selected.join(", ") : answer.selected;
+        lines.push(`**Selected:** ${selected}`);
+        if (answer.notes) {
+          lines.push(`**Notes:** ${answer.notes}`);
+        }
+      }
+      lines.push("");
+    }
+    lines.push("---", "");
+    const existing = await loadFile(discussionPath) ?? `# ${milestoneId} Discussion Log\n\n`;
+    await saveFile(discussionPath, existing + lines.join("\n"));
+  });
+
+  pi.on("tool_execution_start", async (event) => {
+    if (!isAutoActive()) return;
+    markToolStart(event.toolCallId);
+  });
+
+  pi.on("tool_execution_end", async (event) => {
+    markToolEnd(event.toolCallId);
+  });
+}
+

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

@@ -0,0 +1,55 @@
+import { existsSync } from "node:fs";
+import { join } from "node:path";
+
+import type { ExtensionAPI } from "@gsd/pi-coding-agent";
+import { Key, Text } from "@gsd/pi-tui";
+
+import { GSDDashboardOverlay } from "../dashboard-overlay.js";
+import { shortcutDesc } from "../../shared/mod.js";
+
+export const GSD_LOGO_LINES = [
+  "   ██████╗ ███████╗██████╗ ",
+  "  ██╔════╝ ██╔════╝██╔══██╗",
+  "  ██║  ███╗███████╗██║  ██║",
+  "  ██║   ██║╚════██║██║  ██║",
+  "  ╚██████╔╝███████║██████╔╝",
+  "   ╚═════╝ ╚══════╝╚═════╝ ",
+];
+
+export function registerShortcuts(pi: ExtensionAPI): void {
+  pi.registerShortcut(Key.ctrlAlt("g"), {
+    description: shortcutDesc("Open GSD dashboard", "/gsd status"),
+    handler: async (ctx) => {
+      if (!existsSync(join(process.cwd(), ".gsd"))) {
+        ctx.ui.notify("No .gsd/ directory found. Run /gsd to start.", "info");
+        return;
+      }
+      await ctx.ui.custom<void>(
+        (tui, theme, _kb, done) => new GSDDashboardOverlay(tui, theme, () => done()),
+        {
+          overlay: true,
+          overlayOptions: {
+            width: "90%",
+            minWidth: 80,
+            maxHeight: "92%",
+            anchor: "center",
+          },
+        },
+      );
+    },
+  });
+}
+
+export function maybeRenderGsdHeader(ctx: { ui: any }): void {
+  try {
+    const theme = ctx.ui.theme;
+    const version = process.env.GSD_VERSION || "0.0.0";
+    const logoText = GSD_LOGO_LINES.map((line) => theme.fg("accent", line)).join("\n");
+    const titleLine = `  ${theme.bold("Get Shit Done")} ${theme.fg("dim", `v${version}`)}`;
+    const headerContent = `${logoText}\n${titleLine}`;
+    ctx.ui.setHeader((_ui: unknown, _theme: unknown) => new Text(headerContent, 1, 0));
+  } catch {
+    // no TUI
+  }
+}
+