opencodekit 0.17.13 → 0.18.0

package/dist/template/.opencode/plugin/lib/memory-helpers.ts

@@ -0,0 +1,202 @@
+/**
+ * Memory Plugin — Helpers
+ *
+ * Constants, compaction utilities, and tool formatting helpers.
+ * Pure functions — no plugin/closure dependencies.
+ */
+
+import { Database } from "bun:sqlite";
+import { readdir, readFile, stat } from "node:fs/promises";
+import path from "node:path";
+import type { ObservationType } from "./memory-db.js";
+
+// ============================================================================
+// Constants
+// ============================================================================
+
+export const VALID_TYPES: ObservationType[] = [
+	"decision",
+	"bugfix",
+	"feature",
+	"pattern",
+	"discovery",
+	"learning",
+	"warning",
+];
+
+export const TYPE_ICONS: Record<string, string> = {
+	decision: "\u2696\uFE0F",
+	bugfix: "\uD83D\uDC1B",
+	feature: "\u2728",
+	pattern: "\uD83D\uDD04",
+	discovery: "\uD83D\uDD2D",
+	learning: "\uD83D\uDCDA",
+	warning: "\u26A0\uFE0F",
+};
+
+export const FILE_REF_PATTERNS = [
+	/(?:^|\s)(\S+\.(?:ts|tsx|js|jsx|json|md|yaml|yml|toml|sql|sh|py|rs|go)):(\d+)/g,
+	/`([^`]+\.(?:ts|tsx|js|jsx|json|md|yaml|yml|toml))`/g,
+	/(?:^|\s)(src\/\S+)/gm,
+	/(?:^|\s)(\.opencode\/\S+)/gm,
+];
+
+// Compaction constants
+export const MAX_SESSION_CONTEXT_CHARS = 3000;
+export const MAX_PROJECT_FILES = 3;
+export const MAX_PROJECT_FILE_CHARS = 900;
+export const MAX_HANDOFF_CHARS = 2500;
+export const MAX_BEADS = 8;
+export const MAX_COMBINED_CONTEXT_CHARS = 10000;
+
+// ============================================================================
+// Compaction Helpers
+// ============================================================================
+
+export function truncate(text: string, maxChars: number): string {
+	if (text.length <= maxChars) return text;
+	return `${text.slice(0, maxChars)}\n...[truncated]`;
+}
+
+export async function safeReadFile(filePath: string): Promise<string> {
+	try {
+		return await readFile(filePath, "utf-8");
+	} catch {
+		return "";
+	}
+}
+
+export function renderSection(title: string, body: string): string {
+	if (!body.trim()) return "";
+	return `## ${title}\n${body.trim()}`;
+}
+
+export async function readProjectMemoryContext(
+	memoryDir: string,
+): Promise<string> {
+	const projectDir = path.join(memoryDir, "project");
+	let names: string[] = [];
+	try {
+		names = (await readdir(projectDir))
+			.filter((n) => n.endsWith(".md"))
+			.sort()
+			.slice(0, MAX_PROJECT_FILES);
+	} catch {
+		return "";
+	}
+
+	const chunks: string[] = [];
+	for (const name of names) {
+		const content = (await safeReadFile(path.join(projectDir, name))).trim();
+		if (!content) continue;
+		chunks.push(
+			`### ${name.replace(/\.md$/, "")}\n${truncate(content, MAX_PROJECT_FILE_CHARS)}`,
+		);
+	}
+	return chunks.join("\n\n");
+}
+
+export async function readLatestHandoff(handoffDir: string): Promise<string> {
+	let names: string[] = [];
+	try {
+		names = (await readdir(handoffDir)).filter((n) => n.endsWith(".md"));
+	} catch {
+		return "";
+	}
+	if (names.length === 0) return "";
+
+	const withMtime = await Promise.all(
+		names.map(async (name) => {
+			const fullPath = path.join(handoffDir, name);
+			try {
+				return { name, fullPath, mtimeMs: (await stat(fullPath)).mtimeMs };
+			} catch {
+				return { name, fullPath, mtimeMs: 0 };
+			}
+		}),
+	);
+	withMtime.sort((a, b) => b.mtimeMs - a.mtimeMs);
+	const latest = withMtime[0];
+	const content = (await safeReadFile(latest.fullPath)).trim();
+	if (!content) return "";
+	return `Source: ${latest.name}\n${truncate(content, MAX_HANDOFF_CHARS)}`;
+}
+
+export function readInProgressBeads(directory: string): string {
+	const dbPath = path.join(directory, ".beads", "beads.db");
+	let db: Database | undefined;
+	try {
+		db = new Database(dbPath, { readonly: true });
+		const rows = db
+			.query<{ id: string; title: string }, [number]>(
+				"SELECT id, title FROM issues WHERE status = 'in_progress' ORDER BY updated_at DESC LIMIT ?",
+			)
+			.all(MAX_BEADS);
+		return rows.length > 0
+			? rows.map((r) => `- ${r.id}: ${r.title}`).join("\n")
+			: "";
+	} catch {
+		return "";
+	} finally {
+		db?.close();
+	}
+}
+
+// ============================================================================
+// Tool Helpers
+// ============================================================================
+
+export function autoDetectFiles(text: string): string[] {
+	const files = new Set<string>();
+	for (const pattern of FILE_REF_PATTERNS) {
+		const regex = new RegExp(pattern.source, pattern.flags);
+		let match: RegExpExecArray | null;
+		while ((match = regex.exec(text)) !== null) {
+			files.add(match[1]);
+		}
+	}
+	return [...files];
+}
+
+export function parseCSV(value: string | undefined): string[] | undefined {
+	if (!value) return undefined;
+	return value
+		.split(",")
+		.map((s) => s.trim())
+		.filter((s) => s.length > 0);
+}
+
+export function formatObservation(obs: {
+	id: number;
+	type: string;
+	title: string;
+	subtitle?: string | null;
+	confidence?: string | null;
+	concepts?: string | null;
+	files_read?: string | null;
+	files_modified?: string | null;
+	facts?: string | null;
+	narrative?: string | null;
+	bead_id?: string | null;
+	supersedes?: number | null;
+	superseded_by?: number | null;
+	source?: string | null;
+	created_at?: string | null;
+}): string {
+	const icon = TYPE_ICONS[obs.type] ?? "\uD83D\uDCCC";
+	const lines = [`${icon} **#${obs.id}** [${obs.type}] ${obs.title}`];
+	if (obs.subtitle) lines.push(`  _${obs.subtitle}_`);
+	if (obs.confidence) lines.push(`  Confidence: ${obs.confidence}`);
+	if (obs.source && obs.source !== "manual")
+		lines.push(`  Source: ${obs.source}`);
+	if (obs.concepts) lines.push(`  Concepts: ${obs.concepts}`);
+	if (obs.files_read) lines.push(`  Files read: ${obs.files_read}`);
+	if (obs.files_modified) lines.push(`  Files modified: ${obs.files_modified}`);
+	if (obs.facts) lines.push(`  Facts: ${obs.facts}`);
+	if (obs.bead_id) lines.push(`  Bead: ${obs.bead_id}`);
+	if (obs.supersedes) lines.push(`  Supersedes: #${obs.supersedes}`);
+	if (obs.superseded_by) lines.push(`  Superseded by: #${obs.superseded_by}`);
+	if (obs.narrative) lines.push(`\n${obs.narrative}`);
+	if (obs.created_at) lines.push(`\n  _Created: ${obs.created_at}_`);
+	return lines.join("\n");
+}

package/dist/template/.opencode/plugin/lib/memory-hooks.ts

@@ -0,0 +1,240 @@
+/**
+ * Memory Plugin — Hooks
+ *
+ * All event handlers, transforms, and compaction logic.
+ * Uses factory pattern: createHooks(deps) returns hook definitions.
+ *
+ * Hook architecture (from @opencode-ai/plugin Hooks interface):
+ * - `event` — generic handler for ALL events (session.idle, message.updated, etc.)
+ * - Named hooks — separate handlers with (input, output) signature:
+ *   - "tool.execute.after", "chat.message", "experimental.chat.*", etc.
+ *
+ * Events NOT in the Hooks interface (handled via generic `event`):
+ * - session.idle, session.error, session.created, session.deleted
+ * - message.updated, message.removed, message.part.updated, message.part.removed
+ */
+
+import path from "node:path";
+import { captureMessageMeta, captureMessagePart } from "./capture.js";
+import { manageContext } from "./context.js";
+import { curateFromDistillations } from "./curator.js";
+import { distillSession } from "./distill.js";
+import { buildInjection } from "./inject.js";
+import {
+	checkFTS5Available,
+	checkpointWAL,
+	getDatabaseSizes,
+	optimizeFTS5,
+	searchObservationsFTS,
+} from "./memory-db.js";
+import {
+	MAX_COMBINED_CONTEXT_CHARS,
+	MAX_SESSION_CONTEXT_CHARS,
+	readInProgressBeads,
+	readLatestHandoff,
+	readProjectMemoryContext,
+	renderSection,
+	safeReadFile,
+	truncate,
+} from "./memory-helpers.js";
+
+interface HookDeps {
+	memoryDir: string;
+	handoffDir: string;
+	directory: string;
+	showToast: (
+		title: string,
+		message: string,
+		variant?: "info" | "warning",
+	) => Promise<void>;
+	log: (message: string, level?: "info" | "warn") => Promise<void>;
+}
+
+export function createHooks(deps: HookDeps) {
+	const { memoryDir, handoffDir, directory, showToast, log } = deps;
+
+	return {
+		// ================================================================
+		// Generic event handler — ALL events route through here
+		// Receives: { event: { type, properties? } }
+		// ================================================================
+		event: async (input: unknown) => {
+			const { event } = input as {
+				event: {
+					type?: string;
+					properties?: Record<string, unknown>;
+				};
+			};
+			if (!event?.type) return;
+
+			// --- Message capture ---
+			if (event.type === "message.updated") {
+				try {
+					captureMessageMeta(
+						event.properties as Parameters<typeof captureMessageMeta>[0],
+					);
+				} catch {
+					/* Non-fatal */
+				}
+			}
+
+			if (event.type === "message.part.updated") {
+				try {
+					captureMessagePart(
+						event.properties as Parameters<typeof captureMessagePart>[0],
+					);
+				} catch {
+					/* Non-fatal */
+				}
+			}
+
+			// --- Session idle: distill + curate + optimize ---
+			if (event.type === "session.idle") {
+				const sessionId =
+					(event.properties as { sessionID?: string })?.sessionID ??
+					(event as unknown as { sessionID?: string })?.sessionID;
+				try {
+					if (sessionId) distillSession(sessionId);
+					curateFromDistillations(sessionId, 5);
+					if (checkFTS5Available()) optimizeFTS5();
+					const sizes = getDatabaseSizes();
+					if (sizes.wal > 1024 * 1024) checkpointWAL();
+				} catch (err) {
+					const msg = err instanceof Error ? err.message : String(err);
+					await log(`Idle maintenance failed: ${msg}`, "warn");
+				}
+			}
+
+			// --- Session error: warn user ---
+			if (event.type === "session.error") {
+				await showToast(
+					"Session Error",
+					"Save important learnings with observation tool",
+					"warning",
+				);
+			}
+		},
+
+		// ================================================================
+		// Named hook: tool.execute.after
+		// Receives: (input: { tool, sessionID, callID, args }, output: { title, output, metadata })
+		// ================================================================
+		"tool.execute.after": async (input: {
+			tool?: string;
+			sessionID?: string;
+		}) => {
+			try {
+				if (input.tool === "observation" && typeof showToast === "function") {
+					await showToast("Saved", "Observation added to memory");
+				}
+			} catch {
+				/* Toast is cosmetic, never block tool execution */
+			}
+		},
+
+		// ================================================================
+		// LTM injection into system prompt
+		// ================================================================
+		"experimental.chat.system.transform": async (
+			_input: unknown,
+			output: { system: string[] },
+		) => {
+			try {
+				const injection = buildInjection(output.system);
+				if (injection) output.system.push(injection);
+			} catch {
+				/* Non-fatal */
+			}
+		},
+
+		// ================================================================
+		// Context window management
+		// ================================================================
+		"experimental.chat.messages.transform": async (
+			_input: unknown,
+			output: { messages: unknown[] },
+		) => {
+			try {
+				// eslint-disable-next-line @typescript-eslint/no-explicit-any
+				output.messages = manageContext(output.messages as any) as any;
+			} catch {
+				/* Non-fatal */
+			}
+		},
+
+		// ================================================================
+		// Compaction — inject session continuity context
+		// Receives: (input: { sessionID }, output: { context, prompt? })
+		// ================================================================
+		"experimental.session.compacting": async (
+			input: { sessionID?: string },
+			output: { context: string[]; prompt?: string },
+		) => {
+			const sessionContext = truncate(
+				(await safeReadFile(path.join(memoryDir, "session-context.md"))).trim(),
+				MAX_SESSION_CONTEXT_CHARS,
+			);
+
+			const [projectContext, handoffContext] = await Promise.all([
+				readProjectMemoryContext(memoryDir),
+				readLatestHandoff(handoffDir),
+			]);
+
+			const beadsContext = readInProgressBeads(directory);
+
+			// Add relevant observations for session
+			let knowledgeContext = "";
+			if (input.sessionID) {
+				try {
+					const recentObs = searchObservationsFTS("", { limit: 5 });
+					if (recentObs.length > 0) {
+						knowledgeContext = recentObs
+							.map((o) => `- [${o.type}] #${o.id}: ${o.title}`)
+							.join("\n");
+					}
+				} catch {
+					/* Non-fatal */
+				}
+			}
+
+			const combined = [
+				renderSection("Session Continuity", sessionContext),
+				renderSection("Active Beads", beadsContext),
+				renderSection("Previous Handoff", handoffContext),
+				renderSection("Recent Knowledge", knowledgeContext),
+				renderSection("Project Memory", projectContext),
+			]
+				.filter(Boolean)
+				.join("\n\n");
+
+			if (combined) {
+				output.context.push(
+					`## Session Context\n${truncate(combined, MAX_COMBINED_CONTEXT_CHARS)}\n`,
+				);
+			}
+
+			output.prompt = `${output.prompt ?? ""}
+
+<compaction_task>
+Summarize conversation state for reliable continuation after compaction.
+</compaction_task>
+
+<compaction_rules>
+- Preserve exact IDs, file paths, and unresolved constraints.
+- Distinguish completed work from current in-progress work.
+- Keep summary concise and execution-focused.
+- If critical context is missing, state uncertainty explicitly.
+</compaction_rules>
+
+<compaction_output>
+Include:
+- What was done
+- What is being worked on now
+- Files currently in play
+- Next actions
+- Persistent user constraints/preferences
+</compaction_output>
+`;
+		},
+	};
+}