pi-ui-extend 0.1.5 → 0.1.8

package/external/pi-tools-suite/src/lsp/markdown-diagnostics.ts

@@ -0,0 +1,157 @@
+import path from "node:path";
+import fs from "node:fs";
+import type { Diagnostic } from "vscode-languageserver-protocol";
+
+interface LineOffsets {
+  readonly text: string;
+  readonly starts: number[];
+}
+
+function lineOffsets(text: string): LineOffsets {
+  const starts = [0];
+  for (let index = 0; index < text.length; index += 1) {
+    if (text[index] === "\n") starts.push(index + 1);
+  }
+  return { text, starts };
+}
+
+function positionAt(offsets: LineOffsets, offset: number): { line: number; character: number } {
+  let low = 0;
+  let high = offsets.starts.length - 1;
+  while (low <= high) {
+    const mid = Math.floor((low + high) / 2);
+    if (offsets.starts[mid] <= offset) low = mid + 1;
+    else high = mid - 1;
+  }
+  const line = Math.max(0, high);
+  return { line, character: Math.max(0, offset - offsets.starts[line]) };
+}
+
+function diagnostic(offsets: LineOffsets, start: number, end: number, message: string, code: string, severity: Diagnostic["severity"] = 1): Diagnostic {
+  return {
+    severity,
+    source: "pi-markdown",
+    code,
+    message,
+    range: {
+      start: positionAt(offsets, start),
+      end: positionAt(offsets, Math.max(end, start + 1)),
+    },
+  };
+}
+
+function addMarkdownLinkDiagnostics(file: string, text: string, offsets: LineOffsets, out: Diagnostic[]): void {
+  const definitions = new Map<string, Array<{ start: number; end: number }>>();
+  const usedReferences = new Set<string>();
+
+  for (const match of text.matchAll(/^\s{0,3}\[([^\]\r\n]+)\]:\s*(\S+)/gm)) {
+    const ref = match[1].trim().toLocaleLowerCase();
+    const start = (match.index ?? 0) + match[0].indexOf(match[1]);
+    const end = start + match[1].length;
+    const existing = definitions.get(ref) ?? [];
+    existing.push({ start, end });
+    definitions.set(ref, existing);
+  }
+
+  for (const [ref, locations] of definitions) {
+    if (locations.length <= 1) continue;
+    for (const location of locations) {
+      out.push(diagnostic(offsets, location.start, location.end, `Duplicate link definition: '${ref}'`, "link.duplicate-definition", 2));
+    }
+  }
+
+  for (const match of text.matchAll(/(?<!!)(?:\[[^\]\r\n]+\]\[([^\]\r\n]*)\]|\[([^\]\r\n]+)\]\[\])/g)) {
+    const full = match[0];
+    const explicit = match[1];
+    const collapsed = match[2];
+    const ref = (explicit === "" ? collapsed : explicit)?.trim();
+    if (!ref) continue;
+    const normalized = ref.toLocaleLowerCase();
+    usedReferences.add(normalized);
+    if (definitions.has(normalized)) continue;
+    const start = (match.index ?? 0) + full.lastIndexOf(ref);
+    out.push(diagnostic(offsets, start, start + ref.length, `No link definition found: '${ref}'`, "link.no-such-reference"));
+  }
+
+  for (const [ref, locations] of definitions) {
+    if (usedReferences.has(ref)) continue;
+    for (const location of locations) {
+      out.push(diagnostic(offsets, location.start, location.end, "Link definition is unused", "link.unused-definition", 4));
+    }
+  }
+
+  const fileLinkPattern = /(?<!!)\[[^\]\r\n]+\]\(([^)\s]+)(?:\s+[^)]*)?\)/g;
+  for (const match of text.matchAll(fileLinkPattern)) {
+    const href = match[1];
+    if (!href || /^[a-z][a-z0-9+.-]*:/i.test(href) || href.startsWith("#")) continue;
+    const [targetPath] = href.split("#", 1);
+    if (!targetPath || targetPath.startsWith("mailto:")) continue;
+    const decoded = decodeURIComponent(targetPath);
+    const absolute = path.resolve(path.dirname(file), decoded);
+    try {
+      if (!fs.existsSync(absolute)) {
+        const start = (match.index ?? 0) + match[0].indexOf(href);
+        out.push(diagnostic(offsets, start, start + href.length, `File does not exist: '${href}'`, "link.no-such-file"));
+      }
+    } catch {
+      // Ignore malformed/unsupported local paths.
+    }
+  }
+}
+
+const mermaidStarters = /^(?:---|graph\s+(?:TB|BT|RL|LR|TD)|flowchart\s+(?:TB|BT|RL|LR|TD)|sequenceDiagram|classDiagram(?:-v2)?|stateDiagram(?:-v2)?|erDiagram|journey|gantt|pie(?:\s+title\b)?|gitGraph|mindmap|timeline|quadrantChart|requirementDiagram|C4Context|C4Container|C4Component|C4Dynamic|sankey-beta|xyChart-beta|block-beta|packet-beta|architecture-beta)\b/i;
+
+function mermaidBlocks(text: string): Array<{ content: string; startOffset: number; fenceStart: number; fenceEnd: number }> {
+  const blocks: Array<{ content: string; startOffset: number; fenceStart: number; fenceEnd: number }> = [];
+  const fencePattern = /^(```|~~~)\s*(?:mermaid|mmd)\b[^\r\n]*\r?\n([\s\S]*?)^\1\s*$/gim;
+  for (const match of text.matchAll(fencePattern)) {
+    const fenceStart = match.index ?? 0;
+    const contentStart = fenceStart + match[0].indexOf(match[2]);
+    blocks.push({ content: match[2], startOffset: contentStart, fenceStart, fenceEnd: fenceStart + match[0].length });
+  }
+  return blocks;
+}
+
+function addMermaidDiagnostics(file: string, text: string, offsets: LineOffsets, out: Diagnostic[]): void {
+  const extension = path.extname(file).toLocaleLowerCase();
+  const blocks = [".mmd", ".mermaid"].includes(extension)
+    ? [{ content: text, startOffset: 0, fenceStart: 0, fenceEnd: text.length }]
+    : mermaidBlocks(text);
+
+  for (const block of blocks) {
+    const lines = block.content.split(/\r?\n/);
+    const firstIndex = lines.findIndex((line) => {
+      const trimmed = line.trim();
+      return trimmed.length > 0 && !trimmed.startsWith("%%");
+    });
+    if (firstIndex === -1) {
+      out.push(diagnostic(offsets, block.fenceStart, block.fenceEnd, "Mermaid diagram is empty", "mermaid.empty"));
+      continue;
+    }
+
+    const beforeFirst = lines.slice(0, firstIndex).join("\n");
+    const firstOffset = block.startOffset + (beforeFirst ? beforeFirst.length + 1 : 0) + (lines[firstIndex].match(/^\s*/)?.[0].length ?? 0);
+    const firstLine = lines[firstIndex].trim();
+    if (!mermaidStarters.test(firstLine)) {
+      out.push(diagnostic(offsets, firstOffset, firstOffset + firstLine.length, "Mermaid diagram should start with a supported diagram type such as 'flowchart TD'", "mermaid.missing-diagram-type"));
+      continue;
+    }
+
+    let runningOffset = block.startOffset;
+    for (const line of lines) {
+      const arrow = /\b[A-Za-z0-9_]+\s*->\s*[A-Za-z0-9_]+\b/.exec(line);
+      if (arrow) {
+        out.push(diagnostic(offsets, runningOffset + arrow.index, runningOffset + arrow.index + arrow[0].length, "Mermaid flowchart arrows use '-->' or another Mermaid arrow form, not '->'", "mermaid.invalid-arrow"));
+      }
+      runningOffset += line.length + 1;
+    }
+  }
+}
+
+export function localMarkdownDiagnostics(file: string, text: string): Diagnostic[] {
+  const offsets = lineOffsets(text);
+  const diagnostics: Diagnostic[] = [];
+  addMarkdownLinkDiagnostics(file, text, offsets, diagnostics);
+  addMermaidDiagnostics(file, text, offsets, diagnostics);
+  return diagnostics;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ui-extend",
-  "version": "0.1.5",
+  "version": "0.1.8",
   "private": false,
   "type": "module",
   "bin": {

package/external/pi-tools-suite/src/terminal-bell/index.ts

@@ -1,309 +0,0 @@
-import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
-import { spawn } from "node:child_process";
-import { existsSync } from "node:fs";
-import { delimiter, isAbsolute, join } from "node:path";
-
-const BELL = "\x07";
-const DEFAULT_IDLE_DELAY_MS = 250;
-const IDLE_RETRY_DELAY_MS = 100;
-const MAX_IDLE_RETRIES = 40;
-const SUBAGENTS_LIVE_COUNT_EVENT = "pi-tools-suite:async-subagents:live-count";
-const DEFAULT_NOTIFICATION_TITLE = "Pi";
-const DEFAULT_NOTIFICATION_MESSAGE = "Session stopped";
-const DEFAULT_ASK_USER_NOTIFICATION_MESSAGE = "Waiting for your answer";
-const DEFAULT_MAC_SOUND = "Glass";
-
-const TERM_PROGRAM_BUNDLE_IDS: Record<string, string> = {
-	Apple_Terminal: "com.apple.Terminal",
-	iTerm: "com.googlecode.iterm2",
-	"iTerm.app": "com.googlecode.iterm2",
-	WezTerm: "com.github.wez.wezterm",
-	WarpTerminal: "dev.warp.Warp-Stable",
-	ghostty: "com.mitchellh.ghostty",
-	Ghostty: "com.mitchellh.ghostty",
-	kitty: "net.kovidgoyal.kitty",
-	Alacritty: "org.alacritty",
-	vscode: "com.microsoft.VSCode",
-	"vscode-insiders": "com.microsoft.VSCodeInsiders",
-	zed: "dev.zed.Zed",
-};
-
-type Timer = ReturnType<typeof setTimeout>;
-
-type SubagentsLiveCountEvent = {
-	count?: unknown;
-};
-
-function parseDelayMs(value: string | undefined): number {
-	if (value === undefined || value.trim() === "") return DEFAULT_IDLE_DELAY_MS;
-	const parsed = Number(value);
-	return Number.isFinite(parsed) && parsed >= 0 ? parsed : DEFAULT_IDLE_DELAY_MS;
-}
-
-function isTruthyEnv(value: string | undefined): boolean {
-	if (value === undefined) return false;
-	return ["1", "true", "yes", "on"].includes(value.trim().toLowerCase());
-}
-
-function extensionDisabled(): boolean {
-	return isTruthyEnv(process.env.HEADLESS) || isTruthyEnv(process.env.PI_TERMINAL_BELL_DISABLED);
-}
-
-function canRingTerminal(): boolean {
-	if (process.env.PI_TERMINAL_BELL === "0") return false;
-	if (process.env.PI_TERMINAL_BELL_FORCE === "1") return true;
-	return Boolean(process.stdout.isTTY || process.stderr.isTTY);
-}
-
-function writeBell(): void {
-	const stream = process.stdout.isTTY || !process.stderr.isTTY ? process.stdout : process.stderr;
-	stream.write(BELL);
-}
-
-function soundEnabled(ctx: ExtensionContext): boolean {
-	if (process.env.PI_TERMINAL_BELL_SOUND === "0") return false;
-	if (process.env.PI_TERMINAL_BELL_SOUND === "1") return true;
-	return ctx.hasUI === true;
-}
-
-function notificationsEnabled(ctx: ExtensionContext): boolean {
-	if (process.env.PI_TERMINAL_BELL_NOTIFY === "0") return false;
-	if (process.env.PI_TERMINAL_BELL_NOTIFY === "1") return true;
-	return ctx.hasUI === true;
-}
-
-function appleScriptString(value: string): string {
-	return `"${value.replace(/\\/g, "\\\\").replace(/"/g, '\\"')}"`;
-}
-
-function shellSingleQuote(value: string): string {
-	return `'${value.replace(/'/g, `'"'"'`)}'`;
-}
-
-function spawnDetached(command: string, args: string[]): void {
-	try {
-		const child = spawn(command, args, { detached: true, stdio: "ignore" });
-		child.on("error", () => {});
-		child.unref();
-	} catch {
-		// Best-effort user attention signal. Missing notification backends should not
-		// affect the agent loop or suppress the terminal bell.
-	}
-}
-
-function findExecutable(command: string): string | undefined {
-	if (command.includes("/")) return existsSync(command) ? command : undefined;
-	for (const dir of (process.env.PATH ?? "").split(delimiter)) {
-		if (!dir) continue;
-		const candidate = join(dir, command);
-		if (existsSync(candidate)) return candidate;
-	}
-	return undefined;
-}
-
-function resolveMacSoundPath(sound: string): string {
-	if (isAbsolute(sound)) return sound;
-	const fileName = sound.endsWith(".aiff") ? sound : `${sound}.aiff`;
-	return `/System/Library/Sounds/${fileName}`;
-}
-
-function detectMacActivationBundleId(): string | undefined {
-	const explicit = process.env.PI_TERMINAL_BELL_NOTIFY_ACTIVATE;
-	if (explicit === "0" || explicit === "false") return undefined;
-	if (explicit && explicit.trim() !== "") return explicit.trim();
-
-	// GUI apps that launch shells on macOS commonly export their own bundle id.
-	// This catches Terminal.app, iTerm2, Zed's terminal, VS Code terminals, etc.
-	const inheritedBundleId = process.env.__CFBundleIdentifier;
-	if (inheritedBundleId && inheritedBundleId.trim() !== "") return inheritedBundleId.trim();
-
-	const termProgram = process.env.TERM_PROGRAM;
-	if (!termProgram) return undefined;
-	return TERM_PROGRAM_BUNDLE_IDS[termProgram];
-}
-
-function playAttentionSound(ctx: ExtensionContext): void {
-	if (!soundEnabled(ctx)) return;
-	if (process.platform !== "darwin") return;
-	const sound = process.env.PI_TERMINAL_BELL_SOUND && process.env.PI_TERMINAL_BELL_SOUND !== "1"
-		? process.env.PI_TERMINAL_BELL_SOUND
-		: DEFAULT_MAC_SOUND;
-	const soundPath = resolveMacSoundPath(sound);
-	if (!existsSync(soundPath)) return;
-	spawnDetached("/usr/bin/afplay", [soundPath]);
-}
-
-function notifySessionStopped(
-	ctx: ExtensionContext,
-	macActivationBundleId: string | undefined,
-	message = process.env.PI_TERMINAL_BELL_NOTIFY_MESSAGE ?? DEFAULT_NOTIFICATION_MESSAGE,
-): void {
-	if (!notificationsEnabled(ctx)) return;
-	const title = process.env.PI_TERMINAL_BELL_NOTIFY_TITLE ?? DEFAULT_NOTIFICATION_TITLE;
-
-	if (process.platform === "darwin") {
-		const terminalNotifier = findExecutable(process.env.PI_TERMINAL_BELL_NOTIFIER ?? "terminal-notifier");
-		if (terminalNotifier) {
-			const args = ["-title", title, "-message", message];
-			const activate = macActivationBundleId;
-			if (activate) {
-				args.push("-activate", activate);
-				args.push("-execute", `/usr/bin/open -b ${shellSingleQuote(activate)}`);
-
-				// Do not pass -sender by default. On recent macOS versions it can make the
-				// notification look like it came from the target app, but then clicking the
-				// “Show” button may be handled as that app's own notification instead of
-				// terminal-notifier's -activate/-execute action.
-				if (process.env.PI_TERMINAL_BELL_NOTIFY_SENDER === "1") {
-					args.push("-sender", activate);
-				}
-			}
-			spawnDetached(terminalNotifier, args);
-			return;
-		}
-
-		// Bare osascript notifications are sent by Script Editor/osascript on macOS;
-		// clicking them can open Script Editor's file picker. Keep that backend opt-in
-		// and prefer terminal-notifier for clickable system notifications.
-		if (process.env.PI_TERMINAL_BELL_NOTIFY_OSASCRIPT !== "1") return;
-		spawnDetached("/usr/bin/osascript", [
-			"-e",
-			`display notification ${appleScriptString(message)} with title ${appleScriptString(title)}`,
-		]);
-		return;
-	}
-
-	if (process.platform === "linux") {
-		spawnDetached("notify-send", [title, message]);
-	}
-}
-
-const ASK_USER_TOOL_NAMES = new Set(["ask_user", "ask_user_question", "question"]);
-
-function isAskUserToolName(toolName: string): boolean {
-	return ASK_USER_TOOL_NAMES.has(toolName);
-}
-
-function isSubagentsWaitTool(toolName: string, args: unknown): boolean {
-	if (toolName === "async_subagents_wait") return true;
-	if (toolName !== "subagents") return false;
-	if (!args || typeof args !== "object") return false;
-	const action = (args as { action?: unknown }).action;
-	return typeof action === "string" && action.trim().toLowerCase() === "wait";
-}
-
-function normalizeLiveCount(event: SubagentsLiveCountEvent): number | undefined {
-	if (typeof event.count !== "number" || !Number.isFinite(event.count)) return undefined;
-	return Math.max(0, Math.floor(event.count));
-}
-
-export default function terminalBell(pi: ExtensionAPI) {
-	if (extensionDisabled()) return;
-
-	let timer: Timer | undefined;
-	let lastCtx: ExtensionContext | undefined;
-	let deferredUntilSubagentsFinish = false;
-	let liveSubagentCount = 0;
-	const activeSubagentWaitToolCallIds = new Set<string>();
-	const notifiedAskUserToolCallIds = new Set<string>();
-	const idleDelayMs = parseDelayMs(process.env.PI_TERMINAL_BELL_DELAY_MS);
-	const macActivationBundleId = process.platform === "darwin" ? detectMacActivationBundleId() : undefined;
-
-	function clearTimer(): void {
-		if (!timer) return;
-		clearTimeout(timer);
-		timer = undefined;
-	}
-
-	function hasSubagentWork(): boolean {
-		return liveSubagentCount > 0 || activeSubagentWaitToolCallIds.size > 0;
-	}
-
-	function notifyAttention(ctx: ExtensionContext, message?: string): void {
-		if (canRingTerminal()) writeBell();
-		playAttentionSound(ctx);
-		notifySessionStopped(ctx, macActivationBundleId, message);
-	}
-
-	function attemptBell(ctx: ExtensionContext, attempt: number): void {
-		timer = undefined;
-
-		if (!ctx.isIdle()) {
-			if (attempt < MAX_IDLE_RETRIES) scheduleBell(ctx, IDLE_RETRY_DELAY_MS, attempt + 1);
-			return;
-		}
-
-		if (ctx.hasPendingMessages()) return;
-
-		if (hasSubagentWork()) {
-			deferredUntilSubagentsFinish = true;
-			return;
-		}
-
-		deferredUntilSubagentsFinish = false;
-		notifyAttention(ctx);
-	}
-
-	function scheduleBell(ctx: ExtensionContext, delayMs = idleDelayMs, attempt = 0): void {
-		lastCtx = ctx;
-		clearTimer();
-		timer = setTimeout(() => attemptBell(ctx, attempt), delayMs);
-		timer.unref?.();
-	}
-
-	function notifyAskUserWaiting(toolCallId: string, ctx: ExtensionContext): void {
-		if (notifiedAskUserToolCallIds.has(toolCallId)) return;
-		notifiedAskUserToolCallIds.add(toolCallId);
-		notifyAttention(ctx, process.env.PI_TERMINAL_BELL_ASK_USER_NOTIFY_MESSAGE ?? DEFAULT_ASK_USER_NOTIFICATION_MESSAGE);
-	}
-
-	pi.events.on(SUBAGENTS_LIVE_COUNT_EVENT, (data: unknown) => {
-		const event = data && typeof data === "object" ? data as SubagentsLiveCountEvent : {};
-		const count = normalizeLiveCount(event);
-		if (count === undefined) return;
-		liveSubagentCount = count;
-		if (count === 0 && deferredUntilSubagentsFinish && lastCtx) {
-			scheduleBell(lastCtx);
-		}
-	});
-
-	pi.on("agent_start", async () => {
-		clearTimer();
-		deferredUntilSubagentsFinish = false;
-		activeSubagentWaitToolCallIds.clear();
-		notifiedAskUserToolCallIds.clear();
-	});
-
-	pi.on("tool_execution_start", async (event, ctx) => {
-		if (isSubagentsWaitTool(event.toolName, event.args)) {
-			activeSubagentWaitToolCallIds.add(event.toolCallId);
-		}
-		if (isAskUserToolName(event.toolName)) {
-			notifyAskUserWaiting(event.toolCallId, ctx);
-		}
-	});
-
-	pi.on("tool_call", async (event, ctx) => {
-		if (isAskUserToolName(event.toolName)) {
-			notifyAskUserWaiting(event.toolCallId, ctx);
-		}
-	});
-
-	pi.on("tool_execution_end", async (event) => {
-		activeSubagentWaitToolCallIds.delete(event.toolCallId);
-		notifiedAskUserToolCallIds.delete(event.toolCallId);
-	});
-
-	pi.on("agent_end", async (_event, ctx) => {
-		scheduleBell(ctx);
-	});
-
-	pi.on("session_shutdown", async () => {
-		clearTimer();
-		lastCtx = undefined;
-		deferredUntilSubagentsFinish = false;
-		liveSubagentCount = 0;
-		activeSubagentWaitToolCallIds.clear();
-		notifiedAskUserToolCallIds.clear();
-	});
-}