gsd-pi 2.33.1 → 2.34.0-dev.6769913

package/dist/resources/extensions/aws-auth/index.ts

@@ -1,144 +0,0 @@
-/**
- * AWS Auth Refresh Extension
- *
- * Automatically refreshes AWS credentials when Bedrock API requests fail
- * with authentication/token errors, then retries the user's message.
- *
- * ## How it works
- *
- * Hooks into `agent_end` to check if the last assistant message failed with
- * an AWS auth error (expired SSO token, missing credentials, etc.). If so:
- *
- *   1. Runs the configured `awsAuthRefresh` command (e.g. `aws sso login`)
- *   2. Streams the SSO auth URL and verification code to the TUI so users
- *      can copy/paste if the browser doesn't auto-open
- *   3. Calls `retryLastTurn()` which removes the failed assistant response
- *      and re-runs the agent from the user's original message
- *
- * ## Activation
- *
- * This extension is completely inert unless BOTH conditions are met:
- *   1. A Bedrock API request fails with a recognized AWS auth error
- *   2. `awsAuthRefresh` is configured in settings.json
- *
- * Non-Bedrock users and Bedrock users without `awsAuthRefresh` configured
- * are not affected in any way.
- *
- * ## Setup
- *
- * Add to ~/.gsd/agent/settings.json (or project-level .gsd/settings.json):
- *
- *   { "awsAuthRefresh": "aws sso login --profile my-profile" }
- *
- * ## Matched error patterns
- *
- * The extension recognizes errors from the AWS SDK, Bedrock, and SSO
- * credential providers including:
- *   - ExpiredTokenException / ExpiredToken
- *   - The security token included in the request is expired
- *   - The SSO session associated with this profile has expired or is invalid
- *   - Unable to locate credentials / Could not load credentials
- *   - UnrecognizedClientException
- *   - Error loading SSO Token / Token does not exist
- *   - SSOTokenProviderFailure
- */
-
-import { exec } from "node:child_process";
-import { existsSync, readFileSync } from "node:fs";
-import { join } from "node:path";
-import { homedir } from "node:os";
-import type { ExtensionAPI } from "@gsd/pi-coding-agent";
-
-/** Matches AWS SDK / Bedrock / SSO credential and token errors. */
-const AWS_AUTH_ERROR_RE =
-	/ExpiredToken|security token.*expired|unable to locate credentials|SSO.*(?:session|token).*(?:expired|not found|invalid)|UnrecognizedClient|Could not load credentials|Invalid identity token|token is expired|credentials.*(?:could not|cannot|failed to).*(?:load|resolve|find)|The.*token.*is.*not.*valid|token has expired|SSOTokenProviderFailure|Error loading SSO Token|Token.*does not exist/i;
-
-/**
- * Reads the `awsAuthRefresh` command from settings.json.
- * Checks project-level first, then global (~/.gsd/agent/settings.json).
- */
-function getAwsAuthRefreshCommand(): string | undefined {
-	const configDir = process.env.PI_CONFIG_DIR || ".gsd";
-	const paths = [
-		join(process.cwd(), configDir, "settings.json"),
-		join(homedir(), configDir, "agent", "settings.json"),
-	];
-	for (const settingsPath of paths) {
-		if (!existsSync(settingsPath)) continue;
-		try {
-			const settings = JSON.parse(readFileSync(settingsPath, "utf-8"));
-			if (settings.awsAuthRefresh) return settings.awsAuthRefresh;
-		} catch {}
-	}
-	return undefined;
-}
-
-/**
- * Runs the refresh command with a 2-minute timeout (for SSO browser flows).
- * Streams stdout/stderr to capture and display the SSO auth URL and
- * verification code in real-time via TUI notifications.
- */
-async function runRefresh(
-	command: string,
-	notify: (msg: string, level: "info" | "warning" | "error") => void,
-): Promise<boolean> {
-	notify("Refreshing AWS credentials...", "info");
-	try {
-		await new Promise<void>((resolve, reject) => {
-			const child = exec(command, { timeout: 120_000, env: { ...process.env } });
-			const onData = (data: Buffer | string) => {
-				const text = data.toString();
-				const urlMatch = text.match(/https?:\/\/\S+/);
-				if (urlMatch) {
-					notify(`Open this URL if the browser didn't launch: ${urlMatch[0]}`, "warning");
-				}
-				const codeMatch = text.match(/code[:\s]+([A-Z]{4}-[A-Z]{4})/i);
-				if (codeMatch) {
-					notify(`Verification code: ${codeMatch[1]}`, "info");
-				}
-			};
-			child.stdout?.on("data", onData);
-			child.stderr?.on("data", onData);
-			child.on("close", (code) => {
-				if (code === 0) resolve();
-				else reject(new Error(`Refresh command exited with code ${code}`));
-			});
-			child.on("error", reject);
-		});
-		notify("AWS credentials refreshed successfully ✓", "info");
-		return true;
-	} catch (error) {
-		const msg = error instanceof Error ? error.message : String(error);
-		const isTimeout = /timed out|ETIMEDOUT|killed/i.test(msg);
-		if (isTimeout) {
-			notify("AWS credential refresh timed out. The SSO login may have been cancelled or the browser window was closed.", "error");
-		} else {
-			notify(`AWS credential refresh failed: ${msg}`, "error");
-		}
-		return false;
-	}
-}
-
-export default function (pi: ExtensionAPI) {
-	pi.on("agent_end", async (event, ctx) => {
-		const refreshCommand = getAwsAuthRefreshCommand();
-		if (!refreshCommand) return;
-
-		const messages = event.messages;
-		const lastAssistant = messages[messages.length - 1];
-		if (
-			!lastAssistant ||
-			lastAssistant.role !== "assistant" ||
-			!("errorMessage" in lastAssistant) ||
-			!lastAssistant.errorMessage ||
-			!AWS_AUTH_ERROR_RE.test(lastAssistant.errorMessage)
-		) {
-			return;
-		}
-
-		const refreshed = await runRefresh(refreshCommand, (m, level) => ctx.ui.notify(m, level));
-		if (!refreshed) return;
-
-		pi.retryLastTurn();
-	});
-}

package/dist/resources/extensions/bg-shell/bg-shell-command.ts

@@ -1,219 +0,0 @@
-/**
- * /bg slash command registration — interactive process manager overlay and CLI subcommands.
- */
-
-import type { ExtensionAPI } from "@gsd/pi-coding-agent";
-import { Key } from "@gsd/pi-tui";
-import { shortcutDesc } from "../shared/terminal.js";
-
-import {
-	processes,
-	killProcess,
-	getGroupStatus,
-	cleanupAll,
-} from "./process-manager.js";
-import {
-	generateDigest,
-	getOutput,
-	formatDigestText,
-} from "./output-formatter.js";
-import { formatUptime } from "./utilities.js";
-import { BgManagerOverlay } from "./overlay.js";
-
-import type { BgShellSharedState } from "./index.js";
-
-export function registerBgShellCommand(pi: ExtensionAPI, state: BgShellSharedState): void {
-	pi.registerCommand("bg", {
-		description: "Manage background processes: /bg [list|output|kill|killall|groups] [id]",
-
-		getArgumentCompletions: (prefix: string) => {
-			const subcommands = ["list", "output", "kill", "killall", "groups", "digest"];
-			const parts = prefix.trim().split(/\s+/);
-
-			if (parts.length <= 1) {
-				return subcommands
-					.filter(cmd => cmd.startsWith(parts[0] ?? ""))
-					.map(cmd => ({ value: cmd, label: cmd }));
-			}
-
-			if (parts[0] === "output" || parts[0] === "kill" || parts[0] === "digest") {
-				const idPrefix = parts[1] ?? "";
-				return Array.from(processes.values())
-					.filter(p => p.id.startsWith(idPrefix))
-					.map(p => ({
-						value: `${parts[0]} ${p.id}`,
-						label: `${p.id} — ${p.label}`,
-					}));
-			}
-
-			return [];
-		},
-
-		handler: async (args, ctx) => {
-			const parts = args.trim().split(/\s+/);
-			const sub = parts[0] || "list";
-
-			if (sub === "list" || sub === "") {
-				if (processes.size === 0) {
-					ctx.ui.notify("No background processes.", "info");
-					return;
-				}
-
-				if (!ctx.hasUI) {
-					const lines = Array.from(processes.values()).map(p => {
-						const statusIcon = p.alive
-							? (p.status === "ready" ? "✓" : p.status === "error" ? "✗" : "⋯")
-							: "○";
-						const uptime = formatUptime(Date.now() - p.startedAt);
-						const portInfo = p.ports.length > 0 ? ` :${p.ports.join(",")}` : "";
-						return `${p.id}  ${statusIcon} ${p.status}  ${uptime}  ${p.label}  [${p.processType}]${portInfo}`;
-					});
-					ctx.ui.notify(lines.join("\n"), "info");
-					return;
-				}
-
-				await ctx.ui.custom<void>(
-					(tui, theme, _kb, done) => {
-						return new BgManagerOverlay(tui, theme, () => {
-							done();
-							state.refreshWidget();
-						});
-					},
-					{
-						overlay: true,
-						overlayOptions: {
-							width: "60%",
-							minWidth: 50,
-							maxHeight: "70%",
-							anchor: "center",
-						},
-					},
-				);
-				return;
-			}
-
-			if (sub === "output" || sub === "digest") {
-				const id = parts[1];
-				if (!id) {
-					ctx.ui.notify(`Usage: /bg ${sub} <id>`, "error");
-					return;
-				}
-				const bg = processes.get(id);
-				if (!bg) {
-					ctx.ui.notify(`No process with id '${id}'`, "error");
-					return;
-				}
-
-				if (!ctx.hasUI) {
-					if (sub === "digest") {
-						const digest = generateDigest(bg);
-						ctx.ui.notify(formatDigestText(bg, digest), "info");
-					} else {
-						const output = getOutput(bg, { stream: "both", tail: 50 });
-						ctx.ui.notify(output || "(no output)", "info");
-					}
-					return;
-				}
-
-				await ctx.ui.custom<void>(
-					(tui, theme, _kb, done) => {
-						const overlay = new BgManagerOverlay(tui, theme, () => {
-							done();
-							state.refreshWidget();
-						});
-						const procs = Array.from(processes.values());
-						const idx = procs.findIndex(p => p.id === id);
-						if (idx >= 0) overlay.selectAndView(idx);
-						return overlay;
-					},
-					{
-						overlay: true,
-						overlayOptions: {
-							width: "60%",
-							minWidth: 50,
-							maxHeight: "70%",
-							anchor: "center",
-						},
-					},
-				);
-				return;
-			}
-
-			if (sub === "kill") {
-				const id = parts[1];
-				if (!id) {
-					ctx.ui.notify("Usage: /bg kill <id>", "error");
-					return;
-				}
-				const bg = processes.get(id);
-				if (!bg) {
-					ctx.ui.notify(`No process with id '${id}'`, "error");
-					return;
-				}
-				killProcess(id, "SIGTERM");
-				await new Promise(r => setTimeout(r, 300));
-				if (bg.alive) {
-					killProcess(id, "SIGKILL");
-					await new Promise(r => setTimeout(r, 200));
-				}
-				if (!bg.alive) processes.delete(id);
-				ctx.ui.notify(`Killed process ${id} (${bg.label})`, "info");
-				return;
-			}
-
-			if (sub === "killall") {
-				const count = processes.size;
-				cleanupAll();
-				ctx.ui.notify(`Killed ${count} background process(es)`, "info");
-				return;
-			}
-
-			if (sub === "groups") {
-				const groups = new Set<string>();
-				for (const p of processes.values()) {
-					if (p.group) groups.add(p.group);
-				}
-				if (groups.size === 0) {
-					ctx.ui.notify("No process groups defined.", "info");
-					return;
-				}
-				const lines = Array.from(groups).map(g => {
-					const gs = getGroupStatus(g);
-					const icon = gs.healthy ? "✓" : "✗";
-					const procs = gs.processes.map(p => `${p.id}(${p.status})`).join(", ");
-					return `${icon} ${g}: ${procs}`;
-				});
-				ctx.ui.notify(lines.join("\n"), "info");
-				return;
-			}
-
-			ctx.ui.notify("Usage: /bg [list|output|digest|kill|killall|groups] [id]", "info");
-		},
-	});
-
-	// ── Ctrl+Alt+B shortcut ──────────────────────────────────────────────
-
-	pi.registerShortcut(Key.ctrlAlt("b"), {
-		description: shortcutDesc("Open background process manager", "/bg"),
-		handler: async (ctx) => {
-			state.latestCtx = ctx;
-			await ctx.ui.custom<void>(
-				(tui, theme, _kb, done) => {
-					return new BgManagerOverlay(tui, theme, () => {
-						done();
-						state.refreshWidget();
-					});
-				},
-				{
-					overlay: true,
-					overlayOptions: {
-						width: "60%",
-						minWidth: 50,
-						maxHeight: "70%",
-						anchor: "center",
-					},
-				},
-			);
-		},
-	});
-}