@rol1n/pi-kit 0.0.0-stage → 0.1.1

package/prompt-snipaste/renderers.ts

@@ -0,0 +1,87 @@
+import type { ExtensionContext, Theme } from "@earendil-works/pi-coding-agent";
+import { truncateToWidth, visibleWidth, wrapTextWithAnsi } from "@earendil-works/pi-tui";
+import type { PromptCardData, Snippet } from "./types.ts";
+
+const WIDGET_ID = "prompt-snipaste";
+
+export function buildPromptCardComponent(
+	snippets: PromptCardData["snippets"],
+	theme: Theme,
+) {
+	const borderCol = (s: string) => theme.fg("borderAccent", s);
+	const dim = (s: string) => theme.fg("dim", s);
+	const bold = (s: string) => theme.bold(s);
+
+	const titleText = ` Prompt Snippets (${snippets.length}) `;
+
+	return {
+		render(width: number): string[] {
+			const effectiveWidth = Math.max(20, Math.min(width, 100));
+			const innerW = effectiveWidth - 4;
+
+			const titleVisW = visibleWidth(titleText);
+			const topDashLen = Math.max(1, innerW - titleVisW);
+			const topLine = borderCol("╭─") + bold(titleText) + borderCol("─".repeat(topDashLen) + "╮");
+			const botLine = borderCol("╰" + "─".repeat(innerW + 2) + "╯");
+
+			const lines: string[] = [];
+			lines.push(topLine);
+
+			snippets.forEach((item, idx) => {
+				if (idx > 0) {
+					lines.push(borderCol("├" + "─".repeat(innerW + 2) + "┤"));
+				}
+				const header = `● ${item.name}`;
+				const headerVisW = visibleWidth(header);
+				const headerPad = Math.max(0, innerW - headerVisW);
+				lines.push(borderCol("│ ") + bold(header) + " ".repeat(headerPad) + borderCol(" │"));
+
+				const bodyLines = item.body.split(/\r?\n/);
+				for (const line of bodyLines) {
+					if (!line.trim()) continue;
+					for (const wrapped of wrapTextWithAnsi(line, innerW)) {
+						const pad = Math.max(0, innerW - visibleWidth(wrapped));
+						lines.push(borderCol("│ ") + dim(wrapped) + " ".repeat(pad) + borderCol(" │"));
+					}
+				}
+			});
+
+			lines.push(botLine);
+			return lines.map((l) => truncateToWidth(l, width));
+		},
+		invalidate() {},
+	};
+}
+
+export function updateSnippetWidget(
+	ctx: ExtensionContext,
+	getSelection: () => { snippets: Snippet[]; enabled: Set<string> },
+) {
+	if (getSelection().enabled.size === 0) {
+		ctx.ui.setWidget(WIDGET_ID, undefined);
+		return;
+	}
+
+	ctx.ui.setWidget(WIDGET_ID, (_tui, theme) => ({
+		render(width: number): string[] {
+			const { snippets, enabled } = getSelection();
+			const active = snippets.filter((s) => enabled.has(s.id));
+			if (active.length === 0) return [];
+
+			const prepends = active.filter((s) => s.placement === "prepend");
+			const appends = active.filter((s) => s.placement === "append");
+
+			const parts: string[] = [];
+			if (prepends.length > 0) {
+				parts.push(theme.fg("accent", `↑ ${prepends.map((s) => s.name).join(", ")}`));
+			}
+			if (appends.length > 0) {
+				parts.push(theme.fg("warning", `↓ ${appends.map((s) => s.name).join(", ")}`));
+			}
+
+			const text = `[Prompts: ${parts.join(" · ")}]`;
+			return [truncateToWidth(text, width)];
+		},
+		invalidate() {},
+	}));
+}

package/prompt-snipaste/snippets.ts

@@ -0,0 +1,206 @@
+import { existsSync, readdirSync, readFileSync, statSync } from "node:fs";
+import { homedir } from "node:os";
+import { basename, dirname, isAbsolute, join, resolve } from "node:path";
+import { fileURLToPath } from "node:url";
+
+const BUNDLED_PROMPTS_DIR = join(dirname(fileURLToPath(import.meta.url)), "prompts");
+import { CONFIG_DIR_NAME, getAgentDir, type ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import type { Snippet } from "./types.ts";
+
+export function parseSnippet(filename: string, raw: string): Snippet | null {
+	const cleanRaw = raw.replace(/^\uFEFF/, "");
+	const match = cleanRaw.match(/^---\r?\n([\s\S]*?)\r?\n---\r?\n?([\s\S]*)$/);
+
+	let name = filename.replace(/\.md$/, "");
+	let description = "";
+	let placement: "prepend" | "append" = "append";
+	let order = 100;
+	let body = "";
+
+	if (match) {
+		const meta: Record<string, string> = {};
+		for (const line of match[1].split(/\r?\n/)) {
+			const kv = line.match(/^([A-Za-z][\w-]*)\s*:\s*(.*)$/);
+			if (kv) meta[kv[1].toLowerCase()] = kv[2].trim().replace(/^["']|["']$/g, "");
+		}
+		if (meta.name) name = meta.name;
+		if (meta.description) description = meta.description;
+		if (meta.placement === "prepend") placement = "prepend";
+		const parsedOrder = Number.parseInt(meta.order ?? "100", 10);
+		if (!Number.isNaN(parsedOrder)) order = parsedOrder;
+		body = match[2].trim();
+	} else {
+		body = cleanRaw.trim();
+		if (!body) return null;
+		const firstLine = body.split(/\r?\n/).find((l) => l.trim().length > 0) ?? "";
+		description = firstLine.replace(/^#+\s*/, "").slice(0, 80);
+	}
+
+	if (!body) return null;
+
+	return {
+		id: filename,
+		name,
+		description,
+		placement,
+		order,
+		body,
+	};
+}
+
+function resolveSnippetPath(rawPath: string, cwd: string): string {
+	const trimmed = rawPath.trim();
+	if (trimmed === "~" || trimmed.startsWith("~/") || (process.platform === "win32" && trimmed.startsWith("~\\"))) {
+		return resolve(homedir(), trimmed.replace(/^~[/\\]?/, ""));
+	}
+	if (isAbsolute(trimmed)) {
+		return resolve(trimmed);
+	}
+	const fromCwd = resolve(cwd, trimmed);
+	if (existsSync(fromCwd)) return fromCwd;
+	const fromAgent = resolve(getAgentDir(), trimmed);
+	if (existsSync(fromAgent)) return fromAgent;
+	return fromCwd;
+}
+
+function getSettingsPrompts(cwd: string, pi?: ExtensionAPI): string[] {
+	const results: string[] = [];
+
+	if (pi) {
+		try {
+			const settings = pi.getSettings();
+			if (Array.isArray(settings?.prompts)) {
+				results.push(...settings.prompts);
+				return results;
+			}
+		} catch {
+			// pi.getSettings() may throw before runner binds
+		}
+	}
+
+	const settingsFiles = [
+		join(getAgentDir(), "settings.json"),
+		join(cwd, CONFIG_DIR_NAME, "settings.json"),
+	];
+
+	for (const sf of settingsFiles) {
+		if (!existsSync(sf)) continue;
+		try {
+			const content = readFileSync(sf, "utf-8");
+			const parsed = JSON.parse(content.replace(/^\uFEFF/, ""));
+			if (Array.isArray(parsed?.prompts)) {
+				for (const p of parsed.prompts) {
+					if (typeof p === "string" && !results.includes(p)) {
+						results.push(p);
+					}
+				}
+			}
+		} catch {
+			// ignore JSON parse errors
+		}
+	}
+
+	return results;
+}
+
+export function loadSnippets(cwd: string, pi?: ExtensionAPI): Snippet[] {
+	const seen = new Map<string, Snippet>();
+	const excludedNames = new Set<string>();
+
+	const loadedPromptPaths = pi?.getCommands()
+		.filter((command) => command.source === "prompt")
+		.map((command) => command.sourceInfo.path);
+	const defaultDirs = loadedPromptPaths ? [] : [
+		BUNDLED_PROMPTS_DIR,
+		join(getAgentDir(), "prompts"),
+		join(cwd, CONFIG_DIR_NAME, "prompts"),
+	];
+	const settingsPrompts = loadedPromptPaths ?? getSettingsPrompts(cwd, pi);
+
+	const processDirectory = (dir: string) => {
+		if (!existsSync(dir)) return;
+		let files: string[];
+		try {
+			files = readdirSync(dir);
+		} catch (error) {
+			if (!(error instanceof Error) || !("code" in error)) throw error;
+			console.error(`Failed to read prompt directory ${dir}: ${error.message}`);
+			return;
+		}
+		for (const file of files) {
+			if (!file.endsWith(".md")) continue;
+			if (excludedNames.has(file)) continue;
+			const fullPath = join(dir, file);
+			try {
+				const snippet = parseSnippet(file, readFileSync(fullPath, "utf-8"));
+				if (snippet) seen.set(file, snippet);
+			} catch (error) {
+				if (!(error instanceof Error) || !("code" in error)) throw error;
+				console.error(`Failed to read prompt ${fullPath}: ${error.message}`);
+			}
+		}
+	};
+
+	const processFile = (filePath: string) => {
+		if (!existsSync(filePath) || !filePath.endsWith(".md")) return;
+		const filename = basename(filePath);
+		if (excludedNames.has(filename)) return;
+		try {
+			const snippet = parseSnippet(filename, readFileSync(filePath, "utf-8"));
+			if (snippet) seen.set(filename, snippet);
+		} catch (error) {
+			if (!(error instanceof Error) || !("code" in error)) throw error;
+			console.error(`Failed to read prompt file ${filePath}: ${error.message}`);
+		}
+	};
+
+	// Collect exclusions first
+	for (const raw of settingsPrompts) {
+		if (typeof raw !== "string") continue;
+		const trimmed = raw.trim();
+		if (trimmed.startsWith("-") || trimmed.startsWith("!")) {
+			const clean = trimmed.slice(1).trim();
+			excludedNames.add(basename(clean));
+		}
+	}
+
+	// 1. Load default prompt directories
+	for (const dir of defaultDirs) {
+		processDirectory(dir);
+	}
+
+	// 2. Load Pi's resolved prompts, or explicit settings paths without a runtime.
+	for (const raw of settingsPrompts) {
+		if (typeof raw !== "string") continue;
+		let trimmed = raw.trim();
+		if (trimmed.startsWith("-") || trimmed.startsWith("!")) continue;
+		if (trimmed.startsWith("+")) trimmed = trimmed.slice(1).trim();
+		if (!trimmed) continue;
+
+		const resolved = resolveSnippetPath(trimmed, cwd);
+		if (!existsSync(resolved)) continue;
+
+		try {
+			const stat = statSync(resolved);
+			if (stat.isDirectory()) {
+				processDirectory(resolved);
+			} else if (stat.isFile() && resolved.endsWith(".md")) {
+				processFile(resolved);
+			}
+		} catch (error) {
+			if (!(error instanceof Error) || !("code" in error)) throw error;
+			console.error(`Failed to inspect prompt path ${resolved}: ${error.message}`);
+		}
+	}
+
+	// Remove any excluded items
+	for (const ex of excludedNames) {
+		seen.delete(ex);
+	}
+
+	return Array.from(seen.values()).sort((a, b) => {
+		if (a.placement !== b.placement) return a.placement === "prepend" ? -1 : 1;
+		if (a.order !== b.order) return a.order - b.order;
+		return a.name.localeCompare(b.name);
+	});
+}

package/prompt-snipaste/types.ts

@@ -0,0 +1,18 @@
+export interface Snippet {
+	id: string;
+	name: string;
+	description: string;
+	placement: "prepend" | "append";
+	order: number;
+	body: string;
+}
+
+export interface PromptCardData {
+	snippets: {
+		id: string;
+		name: string;
+		placement: "prepend" | "append";
+		order: number;
+		body: string;
+	}[];
+}

package/screenshot/README.md

@@ -0,0 +1,14 @@
+# screenshot
+
+Insert the file path of the most recently modified Windows/WSL system screenshot into Pi's editor via `/screenshot`.
+
+Searches standard and localized `Pictures/Screenshots` and OneDrive screenshot directories under `USERPROFILE` on Windows or `/mnt/c/Users` on WSL.
+
+
+## Install
+
+```sh
+pi install npm:@rol1n/screenshot
+```
+
+Available as a standalone package (not enabled in `@rol1n/pi-kit` by default).

package/srp-theme/README.md

@@ -0,0 +1,18 @@
+# srp-theme
+
+Incremental Pi terminal presentation enhancement: adds a `belowEditor` last-prompt preview widget and a real-time streaming TPS (tokens per second) status meter while keeping Pi's native header and footer intact.
+
+## Install
+
+```sh
+pi install npm:@rol1n/srp-theme
+```
+
+Also included in `@rol1n/pi-kit`.
+
+## Commands
+
+```text
+/srp-theme on
+/srp-theme off
+```

package/srp-theme/footer.ts

@@ -0,0 +1,193 @@
+import type { ExtensionContext, Theme } from "@earendil-works/pi-coding-agent";
+import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+
+// ============================ Prompt Preview Widget ============================
+
+const POWERLINE_SEP_ANSI = "\x1b[38;5;244m";
+const ANSI_RESET = "\x1b[0m";
+
+/** Compact a multiline prompt into a single line, matching powerline's last-prompt behavior. */
+export function compactPrompt(prompt: string | null | undefined): string {
+  // Custom-message runs (extension `sendMessage`, auto-continue, retries) have no user prompt.
+  if (typeof prompt !== "string") return "";
+  return prompt.replace(/\s+/g, " ").trim();
+}
+
+/**
+ * Render last prompt line using powerline's sep color (ANSI 244), displaying '↳ prompt'.
+ */
+export function renderLastPromptLine(
+  lastUserPrompt: string | null | undefined,
+  width: number,
+): string[] {
+  const compact = compactPrompt(lastUserPrompt);
+  const prefix = ` ${POWERLINE_SEP_ANSI}↳${ANSI_RESET} `;
+  const availableWidth = width - visibleWidth(prefix);
+
+  if (!compact || availableWidth < 10) return [];
+
+  return [
+    truncateToWidth(
+      `${prefix}${POWERLINE_SEP_ANSI}${truncateToWidth(compact, availableWidth, "…")}${ANSI_RESET}`,
+      width,
+      "…",
+    ),
+  ];
+}
+
+
+
+// ============================ Incremental Time / Finished Widget ============================
+
+// const WEEK_DAYS = ["Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"];
+// type ClockVariant = "full" | "month_day" | "time_only";
+type FinishedVariant = "full" | "compact" | "duration_only";
+
+function formatDuration(ms: number): string {
+  if (ms < 0) return "0.0s";
+  if (ms < 60_000) return `${(ms / 1000).toFixed(1)}s`;
+  if (ms < 3600_000) {
+    const totalSecs = Math.round(ms / 1000);
+    const mins = Math.floor(totalSecs / 60);
+    const secs = totalSecs % 60;
+    return `${mins}m${secs < 10 ? "0" : ""}${secs}s`;
+  }
+  const totalMins = Math.round(ms / 60000);
+  const hours = Math.floor(totalMins / 60);
+  const mins = totalMins % 60;
+  return `${hours}h${mins < 10 ? "0" : ""}${mins}m`;
+}
+
+/*
+// 时间胶囊格式化已注释停用
+function formatFooterClockVariant(date: Date, variant: ClockVariant): string {
+  const pad = (value: number) => value.toString().padStart(2, "0");
+  const hours = pad(date.getHours());
+  const minutes = pad(date.getMinutes());
+
+  if (variant === "time_only") return `[${hours}:${minutes}]`;
+
+  const month = pad(date.getMonth() + 1);
+  const dayOfMonth = pad(date.getDate());
+  if (variant === "month_day") return `[${month}-${dayOfMonth} ${hours}:${minutes}]`;
+
+  return `[${date.getFullYear()}-${month}-${dayOfMonth} ${WEEK_DAYS[date.getDay()]} ${hours}:${minutes}]`;
+}
+*/
+
+function formatLoopEndVariant(
+  finished: Date,
+  now: Date,
+  durationMs: number | null,
+  variant: FinishedVariant,
+): string {
+  const pad = (value: number) => value.toString().padStart(2, "0");
+  const hasDuration = typeof durationMs === "number" && durationMs >= 0;
+
+  if (variant === "duration_only") {
+    return hasDuration ? `X ${formatDuration(durationMs)}` : "X";
+  }
+
+  const isCrossYear = finished.getFullYear() !== now.getFullYear();
+  const isCrossDay =
+    finished.getMonth() !== now.getMonth() || finished.getDate() !== now.getDate();
+  const dateOrTime = isCrossYear
+    ? `${finished.getFullYear()}`
+    : isCrossDay
+      ? `${pad(finished.getMonth() + 1)}-${pad(finished.getDate())}`
+      : `${pad(finished.getHours())}:${pad(finished.getMinutes())}`;
+
+  const details = hasDuration ? `${formatDuration(durationMs)} ${dateOrTime}` : dateOrTime;
+  return `X ${details}`;
+}
+
+export function getLastLoopInfoFromSession(ctx: ExtensionContext): {
+  endTime: Date;
+  durationMs: number | null;
+} | null {
+  try {
+    const entries = ctx.sessionManager?.getEntries?.() ?? [];
+    let assistantIndex = -1;
+    let endTime: Date | null = null;
+
+    for (let i = entries.length - 1; i >= 0; i--) {
+      const entry = entries[i] as any;
+      if (entry?.type !== "message" || entry?.message?.role !== "assistant" || !entry?.timestamp) continue;
+      const timestamp = new Date(entry.timestamp);
+      if (!Number.isNaN(timestamp.getTime())) {
+        assistantIndex = i;
+        endTime = timestamp;
+        break;
+      }
+    }
+
+    if (!endTime || assistantIndex < 0) return null;
+
+    let durationMs: number | null = null;
+    for (let i = assistantIndex - 1; i >= 0; i--) {
+      const entry = entries[i] as any;
+      if (entry?.type !== "message" || entry?.message?.role !== "user" || !entry?.timestamp) continue;
+      const startTime = new Date(entry.timestamp);
+      if (!Number.isNaN(startTime.getTime())) {
+        const duration = endTime.getTime() - startTime.getTime();
+        durationMs = duration >= 0 ? duration : null;
+        break;
+      }
+    }
+
+    return { endTime, durationMs };
+  } catch {
+    return null;
+  }
+}
+
+export function renderTimeAndFinishedLine(
+  lastLoopEndTime: Date | null,
+  lastLoopDurationMs: number | null,
+  width: number,
+  theme: Theme,
+): string[] {
+  if (!lastLoopEndTime) return [];
+
+  const now = new Date();
+  const candidates: string[] = [];
+
+  const fullFinished = formatLoopEndVariant(lastLoopEndTime, now, lastLoopDurationMs, "full");
+  const compactFinished = formatLoopEndVariant(lastLoopEndTime, now, lastLoopDurationMs, "compact");
+  const durationFinished = formatLoopEndVariant(lastLoopEndTime, now, lastLoopDurationMs, "duration_only");
+
+  candidates.push(fullFinished);
+  if (compactFinished !== fullFinished) {
+    candidates.push(compactFinished);
+  }
+  if (durationFinished && durationFinished !== compactFinished && durationFinished !== fullFinished) {
+    candidates.push(durationFinished);
+  }
+
+  /*
+  // 时间胶囊（已禁用/注释，仅保留 agent-loop 完成时间和用时）：
+  if (lastLoopEndTime) {
+    candidates.push(
+      `${fullFinished} ${formatFooterClockVariant(now, "full")}`,
+      `${fullFinished} ${formatFooterClockVariant(now, "month_day")}`,
+      `${fullFinished} ${formatFooterClockVariant(now, "time_only")}`,
+    );
+    if (compactFinished !== fullFinished) {
+      candidates.push(`${compactFinished} ${formatFooterClockVariant(now, "time_only")}`);
+    }
+    if (durationFinished && durationFinished !== compactFinished && durationFinished !== fullFinished) {
+      candidates.push(`${durationFinished} ${formatFooterClockVariant(now, "time_only")}`);
+    }
+  } else {
+    candidates.push(
+      formatFooterClockVariant(now, "full"),
+      formatFooterClockVariant(now, "month_day"),
+      formatFooterClockVariant(now, "time_only"),
+    );
+  }
+  */
+
+  const content = candidates.find((candidate) => visibleWidth(candidate) <= width) ?? candidates.at(-1)!;
+  const line = truncateToWidth(content, width, "…");
+  return [theme.fg("dim", line)];
+}