pi-crew 0.1.43 → 0.1.45

package/src/ui/syntax-highlight.ts

@@ -1,116 +1,116 @@
-import { supportsLanguage, highlight } from "cli-highlight";
-import type { CrewTheme } from "./theme-adapter.ts";
-import { asCrewTheme } from "./theme-adapter.ts";
-
-function buildCliTheme(theme: CrewTheme): Record<string, (text: string) => string> {
-	return {
-		keyword: (text) => theme.fg("syntaxKeyword", text),
-		built_in: (text) => theme.fg("syntaxType", text),
-		literal: (text) => theme.fg("syntaxNumber", text),
-		number: (text) => theme.fg("syntaxNumber", text),
-		string: (text) => theme.fg("syntaxString", text),
-		comment: (text) => theme.fg("syntaxComment", text),
-		function: (text) => theme.fg("syntaxFunction", text),
-		title: (text) => theme.fg("syntaxFunction", text),
-		class: (text) => theme.fg("syntaxType", text),
-		type: (text) => theme.fg("syntaxType", text),
-		attr: (text) => theme.fg("syntaxVariable", text),
-		variable: (text) => theme.fg("syntaxVariable", text),
-		params: (text) => theme.fg("syntaxVariable", text),
-		operator: (text) => theme.fg("syntaxOperator", text),
-		punctuation: (text) => theme.fg("syntaxPunctuation", text),
-	};
-}
-
-export function detectLanguageFromPath(filePath: string): string | undefined {
-	const ext = filePath.split(".").pop()?.toLowerCase();
-	if (!ext) return undefined;
-	return languageMap[ext];
-}
-
-export const languageMap: Record<string, string> = {
-	ts: "typescript",
-	tsx: "typescript",
-	js: "javascript",
-	jsx: "javascript",
-	mjs: "javascript",
-	cjs: "javascript",
-	py: "python",
-	md: "markdown",
-	markdown: "markdown",
-	json: "json",
-	yml: "yaml",
-	yaml: "yaml",
-	toml: "yaml",
-	html: "html",
-	htm: "html",
-	css: "css",
-	scss: "scss",
-	sass: "sass",
-	bash: "bash",
-	sh: "bash",
-	zsh: "bash",
-	fish: "bash",
-	ps1: "powershell",
-	sql: "sql",
-	rust: "rust",
-	rb: "ruby",
-	go: "go",
-	java: "java",
-	kt: "kotlin",
-	cpp: "cpp",
-	cc: "cpp",
-	cxx: "cpp",
-	hpp: "cpp",
-	c: "c",
-	h: "c",
-	cs: "csharp",
-	php: "php",
-};
-
-export function highlightCode(code: string, language: string | undefined, themeLike: unknown = undefined): string {
-	const theme = asCrewTheme(themeLike);
-	const validLanguage = language && supportsLanguage(language) ? language : undefined;
-	if (!validLanguage) {
-		return code
-			.split("\n")
-			.map((line) => theme.fg("mdCodeBlock", line))
-			.join("\n");
-	}
-	try {
-		return highlight(code, {
-			language: validLanguage,
-			ignoreIllegals: true,
-			theme: buildCliTheme(theme),
-		}).trimEnd();
-	} catch {
-		return code
-			.split("\n")
-			.map((line) => theme.fg("mdCodeBlock", line))
-			.join("\n");
-	}
-}
-
-export function highlightJson(payload: string, themeLike: unknown = undefined): string {
-	const theme = asCrewTheme(themeLike);
-	try {
-		return highlight(payload, {
-			language: "json",
-			ignoreIllegals: true,
-			theme: buildCliTheme(theme),
-		}).trimEnd();
-	} catch {
-		try {
-			const parsed = JSON.parse(payload);
-			return JSON.stringify(parsed, null, 2)
-				.split("\n")
-				.map((line) => theme.fg("mdCodeBlock", line))
-				.join("\n");
-		} catch {
-			return payload
-				.split("\n")
-				.map((line) => theme.fg("mdCodeBlock", line))
-				.join("\n");
-		}
-	}
-}
+import { supportsLanguage, highlight } from "cli-highlight";
+import type { CrewTheme } from "./theme-adapter.ts";
+import { asCrewTheme } from "./theme-adapter.ts";
+
+function buildCliTheme(theme: CrewTheme): Record<string, (text: string) => string> {
+	return {
+		keyword: (text) => theme.fg("syntaxKeyword", text),
+		built_in: (text) => theme.fg("syntaxType", text),
+		literal: (text) => theme.fg("syntaxNumber", text),
+		number: (text) => theme.fg("syntaxNumber", text),
+		string: (text) => theme.fg("syntaxString", text),
+		comment: (text) => theme.fg("syntaxComment", text),
+		function: (text) => theme.fg("syntaxFunction", text),
+		title: (text) => theme.fg("syntaxFunction", text),
+		class: (text) => theme.fg("syntaxType", text),
+		type: (text) => theme.fg("syntaxType", text),
+		attr: (text) => theme.fg("syntaxVariable", text),
+		variable: (text) => theme.fg("syntaxVariable", text),
+		params: (text) => theme.fg("syntaxVariable", text),
+		operator: (text) => theme.fg("syntaxOperator", text),
+		punctuation: (text) => theme.fg("syntaxPunctuation", text),
+	};
+}
+
+export function detectLanguageFromPath(filePath: string): string | undefined {
+	const ext = filePath.split(".").pop()?.toLowerCase();
+	if (!ext) return undefined;
+	return languageMap[ext];
+}
+
+export const languageMap: Record<string, string> = {
+	ts: "typescript",
+	tsx: "typescript",
+	js: "javascript",
+	jsx: "javascript",
+	mjs: "javascript",
+	cjs: "javascript",
+	py: "python",
+	md: "markdown",
+	markdown: "markdown",
+	json: "json",
+	yml: "yaml",
+	yaml: "yaml",
+	toml: "yaml",
+	html: "html",
+	htm: "html",
+	css: "css",
+	scss: "scss",
+	sass: "sass",
+	bash: "bash",
+	sh: "bash",
+	zsh: "bash",
+	fish: "bash",
+	ps1: "powershell",
+	sql: "sql",
+	rust: "rust",
+	rb: "ruby",
+	go: "go",
+	java: "java",
+	kt: "kotlin",
+	cpp: "cpp",
+	cc: "cpp",
+	cxx: "cpp",
+	hpp: "cpp",
+	c: "c",
+	h: "c",
+	cs: "csharp",
+	php: "php",
+};
+
+export function highlightCode(code: string, language: string | undefined, themeLike: unknown = undefined): string {
+	const theme = asCrewTheme(themeLike);
+	const validLanguage = language && supportsLanguage(language) ? language : undefined;
+	if (!validLanguage) {
+		return code
+			.split("\n")
+			.map((line) => theme.fg("mdCodeBlock", line))
+			.join("\n");
+	}
+	try {
+		return highlight(code, {
+			language: validLanguage,
+			ignoreIllegals: true,
+			theme: buildCliTheme(theme),
+		}).trimEnd();
+	} catch {
+		return code
+			.split("\n")
+			.map((line) => theme.fg("mdCodeBlock", line))
+			.join("\n");
+	}
+}
+
+export function highlightJson(payload: string, themeLike: unknown = undefined): string {
+	const theme = asCrewTheme(themeLike);
+	try {
+		return highlight(payload, {
+			language: "json",
+			ignoreIllegals: true,
+			theme: buildCliTheme(theme),
+		}).trimEnd();
+	} catch {
+		try {
+			const parsed = JSON.parse(payload);
+			return JSON.stringify(parsed, null, 2)
+				.split("\n")
+				.map((line) => theme.fg("mdCodeBlock", line))
+				.join("\n");
+		} catch {
+			return payload
+				.split("\n")
+				.map((line) => theme.fg("mdCodeBlock", line))
+				.join("\n");
+		}
+	}
+}

package/src/utils/atomic-write.ts

@@ -0,0 +1,33 @@
+import * as fs from "node:fs";
+import * as path from "node:path";
+
+/**
+ * Write JSON data to a file atomically.
+ * Uses write-to-temp + rename to avoid torn writes on crash.
+ */
+export function writeAtomicJson(filePath: string, data: unknown, pretty = false): void {
+	const dir = path.dirname(filePath);
+	const content = pretty ? JSON.stringify(data, null, 2) : JSON.stringify(data);
+	const tmpPath = filePath + ".tmp";
+	fs.writeFileSync(tmpPath, content, "utf-8");
+	fs.renameSync(tmpPath, filePath);
+}
+
+/**
+ * Read and parse JSON from a file. Returns undefined on any error.
+ */
+export function readJsonFile<T = unknown>(filePath: string): T | undefined {
+	try {
+		return JSON.parse(fs.readFileSync(filePath, "utf-8")) as T;
+	} catch {
+		return undefined;
+	}
+}
+
+/**
+ * Append a JSON line to a JSONL file atomically per line.
+ */
+export function appendJsonlLine(filePath: string, data: unknown): void {
+	const line = JSON.stringify(data) + "\n";
+	fs.appendFileSync(filePath, line, "utf-8");
+}

package/src/utils/completion-dedupe.ts

@@ -1,63 +1,63 @@
-interface CompletionDataLike {
-	id?: unknown;
-	agent?: unknown;
-	timestamp?: unknown;
-	sessionId?: unknown;
-	taskIndex?: unknown;
-	totalTasks?: unknown;
-	success?: unknown;
-}
-
-function asNonEmptyString(value: unknown): string | undefined {
-	if (typeof value !== "string") return undefined;
-	const trimmed = value.trim();
-	return trimmed.length > 0 ? trimmed : undefined;
-}
-
-function asFiniteNumber(value: unknown): number | undefined {
-	if (typeof value !== "number") return undefined;
-	return Number.isFinite(value) ? value : undefined;
-}
-
-export function buildCompletionKey(data: CompletionDataLike, fallback: string): string {
-	const id = asNonEmptyString(data.id);
-	if (id) return `id:${id}`;
-	const sessionId = asNonEmptyString(data.sessionId) ?? "no-session";
-	const agent = asNonEmptyString(data.agent) ?? "unknown";
-	const timestamp = asFiniteNumber(data.timestamp);
-	const taskIndex = asFiniteNumber(data.taskIndex);
-	const totalTasks = asFiniteNumber(data.totalTasks);
-	const success = typeof data.success === "boolean" ? (data.success ? "1" : "0") : "?";
-	return [
-		"meta",
-		sessionId,
-		agent,
-		timestamp !== undefined ? String(timestamp) : "no-ts",
-		taskIndex !== undefined ? String(taskIndex) : "-",
-		totalTasks !== undefined ? String(totalTasks) : "-",
-		success,
-		fallback,
-	].join(":");
-}
-
-export function pruneSeenMap(seen: Map<string, number>, now: number, ttlMs: number): void {
-	for (const [key, ts] of seen.entries()) {
-		if (now - ts > ttlMs) seen.delete(key);
-	}
-}
-
-export function markSeenWithTtl(seen: Map<string, number>, key: string, now: number, ttlMs: number): boolean {
-	pruneSeenMap(seen, now, ttlMs);
-	if (seen.has(key)) return true;
-	seen.set(key, now);
-	return false;
-}
-
-export function getGlobalSeenMap(storeKey: string): Map<string, number> {
-	const globalStore = globalThis as Record<string, unknown>;
-	const existing = globalStore[storeKey];
-	if (existing instanceof Map) return existing as Map<string, number>;
-	const map = new Map<string, number>();
-	globalStore[storeKey] = map;
-	return map;
-}
+interface CompletionDataLike {
+	id?: unknown;
+	agent?: unknown;
+	timestamp?: unknown;
+	sessionId?: unknown;
+	taskIndex?: unknown;
+	totalTasks?: unknown;
+	success?: unknown;
+}
+
+function asNonEmptyString(value: unknown): string | undefined {
+	if (typeof value !== "string") return undefined;
+	const trimmed = value.trim();
+	return trimmed.length > 0 ? trimmed : undefined;
+}
+
+function asFiniteNumber(value: unknown): number | undefined {
+	if (typeof value !== "number") return undefined;
+	return Number.isFinite(value) ? value : undefined;
+}
+
+export function buildCompletionKey(data: CompletionDataLike, fallback: string): string {
+	const id = asNonEmptyString(data.id);
+	if (id) return `id:${id}`;
+	const sessionId = asNonEmptyString(data.sessionId) ?? "no-session";
+	const agent = asNonEmptyString(data.agent) ?? "unknown";
+	const timestamp = asFiniteNumber(data.timestamp);
+	const taskIndex = asFiniteNumber(data.taskIndex);
+	const totalTasks = asFiniteNumber(data.totalTasks);
+	const success = typeof data.success === "boolean" ? (data.success ? "1" : "0") : "?";
+	return [
+		"meta",
+		sessionId,
+		agent,
+		timestamp !== undefined ? String(timestamp) : "no-ts",
+		taskIndex !== undefined ? String(taskIndex) : "-",
+		totalTasks !== undefined ? String(totalTasks) : "-",
+		success,
+		fallback,
+	].join(":");
+}
+
+export function pruneSeenMap(seen: Map<string, number>, now: number, ttlMs: number): void {
+	for (const [key, ts] of seen.entries()) {
+		if (now - ts > ttlMs) seen.delete(key);
+	}
+}
+
+export function markSeenWithTtl(seen: Map<string, number>, key: string, now: number, ttlMs: number): boolean {
+	pruneSeenMap(seen, now, ttlMs);
+	if (seen.has(key)) return true;
+	seen.set(key, now);
+	return false;
+}
+
+export function getGlobalSeenMap(storeKey: string): Map<string, number> {
+	const globalStore = globalThis as Record<string, unknown>;
+	const existing = globalStore[storeKey];
+	if (existing instanceof Map) return existing as Map<string, number>;
+	const map = new Map<string, number>();
+	globalStore[storeKey] = map;
+	return map;
+}

package/src/utils/frontmatter.ts

@@ -1,68 +1,68 @@
-export interface ParsedFrontmatter {
-	frontmatter: Record<string, string>;
-	body: string;
-}
-
-export function parseFrontmatter(content: string): ParsedFrontmatter {
-	if (!content.startsWith("---\n") && !content.startsWith("---\r\n")) {
-		return { frontmatter: {}, body: content };
-	}
-
-	const normalized = content.replaceAll("\r\n", "\n");
-	const end = normalized.indexOf("\n---\n", 4);
-	if (end === -1) {
-		// Support frontmatter that ends at EOF without trailing newline after ---.
-		const altEnd = normalized.indexOf("\n---", 4);
-		if (altEnd !== -1 && altEnd + 4 === normalized.length) {
-			const raw = normalized.slice(4, altEnd);
-			const frontmatter = parseLines(raw);
-			return { frontmatter, body: "" };
-		}
-		return { frontmatter: {}, body: content };
-	}
-
-	const raw = normalized.slice(4, end);
-	const body = normalized.slice(end + "\n---\n".length);
-	const frontmatter = parseLines(raw);
-	return { frontmatter, body };
-}
-
-function parseLines(raw: string): Record<string, string> {
-	const frontmatter: Record<string, string> = {};
-	for (const line of raw.split("\n")) {
-		const trimmed = line.trim();
-		if (!trimmed || trimmed.startsWith("#")) continue;
-		const separator = trimmed.indexOf(":");
-		if (separator === -1) continue;
-		const key = trimmed.slice(0, separator).trim();
-		const value = trimmed.slice(separator + 1).trim();
-		if (key) frontmatter[key] = value;
-	}
-	return frontmatter;
-}
-
-export function parseCsv(value: string | undefined): string[] | undefined {
-	if (value === undefined) return undefined;
-	// Handle quoted values with commas inside.
-	const values = splitCsv(value).map((item) => item.trim()).filter(Boolean);
-	return values.length > 0 ? [...new Set(values)] : undefined;
-}
-
-function splitCsv(input: string): string[] {
-	const result: string[] = [];
-	let current = "";
-	let inQuotes = false;
-	for (let i = 0; i < input.length; i++) {
-		const char = input[i];
-		if (char === '"') {
-			inQuotes = !inQuotes;
-		} else if (char === "," && !inQuotes) {
-			result.push(current);
-			current = "";
-		} else {
-			current += char;
-		}
-	}
-	result.push(current);
-	return result;
-}
+export interface ParsedFrontmatter {
+	frontmatter: Record<string, string>;
+	body: string;
+}
+
+export function parseFrontmatter(content: string): ParsedFrontmatter {
+	if (!content.startsWith("---\n") && !content.startsWith("---\r\n")) {
+		return { frontmatter: {}, body: content };
+	}
+
+	const normalized = content.replaceAll("\r\n", "\n");
+	const end = normalized.indexOf("\n---\n", 4);
+	if (end === -1) {
+		// Support frontmatter that ends at EOF without trailing newline after ---.
+		const altEnd = normalized.indexOf("\n---", 4);
+		if (altEnd !== -1 && altEnd + 4 === normalized.length) {
+			const raw = normalized.slice(4, altEnd);
+			const frontmatter = parseLines(raw);
+			return { frontmatter, body: "" };
+		}
+		return { frontmatter: {}, body: content };
+	}
+
+	const raw = normalized.slice(4, end);
+	const body = normalized.slice(end + "\n---\n".length);
+	const frontmatter = parseLines(raw);
+	return { frontmatter, body };
+}
+
+function parseLines(raw: string): Record<string, string> {
+	const frontmatter: Record<string, string> = {};
+	for (const line of raw.split("\n")) {
+		const trimmed = line.trim();
+		if (!trimmed || trimmed.startsWith("#")) continue;
+		const separator = trimmed.indexOf(":");
+		if (separator === -1) continue;
+		const key = trimmed.slice(0, separator).trim();
+		const value = trimmed.slice(separator + 1).trim();
+		if (key) frontmatter[key] = value;
+	}
+	return frontmatter;
+}
+
+export function parseCsv(value: string | undefined): string[] | undefined {
+	if (value === undefined) return undefined;
+	// Handle quoted values with commas inside.
+	const values = splitCsv(value).map((item) => item.trim()).filter(Boolean);
+	return values.length > 0 ? [...new Set(values)] : undefined;
+}
+
+function splitCsv(input: string): string[] {
+	const result: string[] = [];
+	let current = "";
+	let inQuotes = false;
+	for (let i = 0; i < input.length; i++) {
+		const char = input[i];
+		if (char === '"') {
+			inQuotes = !inQuotes;
+		} else if (char === "," && !inQuotes) {
+			result.push(current);
+			current = "";
+		} else {
+			current += char;
+		}
+	}
+	result.push(current);
+	return result;
+}