@yandy0725/pi-memory 1.3.4 → 2.0.0

package/src/snapshot.ts

@@ -0,0 +1,68 @@
+import { cp, mkdir, readdir, rm, stat } from "node:fs/promises";
+import { basename, join } from "node:path";
+
+export interface SnapshotOptions {
+	keep: number;
+	now?: () => Date;
+}
+
+/** 可字典序排序的快照时间戳（ISO 8601，`:` 与 `.` 换为 `-`）。 */
+export function snapshotStamp(now: () => Date): string {
+	return now().toISOString().replace(/[:.]/g, "-");
+}
+
+async function uniqueDir(backupRoot: string, base: string): Promise<string> {
+	await mkdir(backupRoot, { recursive: true });
+	for (let n = 1; ; n++) {
+		const candidate = n === 1 ? join(backupRoot, base) : join(backupRoot, `${base}-${n}`);
+		try {
+			await mkdir(candidate);
+			return candidate;
+		} catch (e) {
+			if ((e as NodeJS.ErrnoException).code !== "EEXIST") throw e;
+		}
+	}
+}
+
+/** 将给定的 memory 目录相对路径复制到新快照目录；不存在的文件被跳过。 */
+export async function createSnapshot(
+	backupRoot: string,
+	label: string,
+	files: string[],
+	memoryDir: string,
+	options: SnapshotOptions,
+): Promise<string> {
+	const dir = await uniqueDir(backupRoot, `${snapshotStamp(options.now ?? (() => new Date()))}-${label}`);
+	for (const file of files) {
+		try {
+			await cp(join(memoryDir, file), join(dir, basename(file)));
+		} catch (e) {
+			if ((e as NodeJS.ErrnoException).code === "ENOENT") continue;
+			throw e;
+		}
+	}
+	await pruneSnapshots(backupRoot, options.keep);
+	return dir;
+}
+
+/** 保留名字序最新的 keep 个快照；`migrate-` 开头的目录永不参与裁剪。 */
+export async function pruneSnapshots(backupRoot: string, keep: number): Promise<void> {
+	let names: string[];
+	try {
+		names = await readdir(backupRoot);
+	} catch {
+		return;
+	}
+
+	const dirs: string[] = [];
+	for (const name of names) {
+		if (name.startsWith("migrate-")) continue;
+		const info = await stat(join(backupRoot, name)).catch(() => null);
+		if (info?.isDirectory()) dirs.push(name);
+	}
+
+	dirs.sort();
+	for (const name of dirs.slice(0, Math.max(0, dirs.length - Math.max(0, keep)))) {
+		await rm(join(backupRoot, name), { recursive: true, force: true });
+	}
+}

package/src/index-file.ts

@@ -1,91 +0,0 @@
-export interface IndexEntry {
-	name: string; // 原名 title，取自 frontmatter name
-	topic: string; // 文件名
-	hook: string; // 一行描述
-	raw: string; // 原始行文本
-}
-
-export interface IndexFile {
-	entries: IndexEntry[];
-	raw: string;
-}
-
-// Matches: - [Name](topic.md) — hook
-const LINE_RE = /^-\s+\[([^\]]+)\]\(([^)]+)\)\s*—\s*(.*)$/;
-
-export function parseIndex(content: string): IndexFile {
-	const entries: IndexEntry[] = [];
-	for (const line of content.split("\n")) {
-		const m = line.match(LINE_RE);
-		if (m) {
-			entries.push({
-				name: m[1].trim(),
-				topic: m[2].trim(),
-				hook: m[3].trim(),
-				raw: line,
-			});
-		}
-	}
-	return { entries, raw: content };
-}
-
-export function serializeIndex(entries: IndexEntry[]): string {
-	return entries.map((e) => `- [${e.name}](${e.topic}) — ${e.hook}`).join("\n");
-}
-
-export function upsertEntryByTopic(entries: IndexEntry[], entry: IndexEntry): IndexEntry[] {
-	const idx = entries.findIndex((e) => e.topic === entry.topic);
-	if (idx === -1) return [...entries, entry];
-	const next = [...entries];
-	next[idx] = entry;
-	return next;
-}
-
-export function removeEntryByTopic(entries: IndexEntry[], topic: string): IndexEntry[] {
-	const idx = entries.findIndex((e) => e.topic === topic);
-	if (idx === -1) throw new Error(`Topic "${topic}" not found in index`);
-	const next = [...entries];
-	next.splice(idx, 1);
-	return next;
-}
-
-export function findEntryByTopic(entries: IndexEntry[], topic: string): IndexEntry | null {
-	return entries.find((e) => e.topic === topic) ?? null;
-}
-
-export function updateHook(entries: IndexEntry[], topic: string, hook: string): IndexEntry[] {
-	const idx = entries.findIndex((e) => e.topic === topic);
-	if (idx === -1) throw new Error(`Topic "${topic}" not found in index`);
-	const next = [...entries];
-	next[idx] = { ...next[idx], hook, raw: "" };
-	return next;
-}
-
-export function truncateForInjection(
-	content: string,
-	maxLines: number,
-	maxBytes: number,
-): { ok: boolean; content: string; truncated: boolean } {
-	const lines = content.split("\n");
-	let out = content;
-	let truncated = false;
-	if (lines.length > maxLines) {
-		out = lines.slice(0, maxLines).join("\n");
-		truncated = true;
-	}
-	if (Buffer.byteLength(out, "utf8") > maxBytes) {
-		let cut = out;
-		while (Buffer.byteLength(cut, "utf8") > maxBytes && cut.length > 0) cut = cut.slice(0, -1);
-		out = cut;
-		truncated = true;
-	}
-	if (truncated) out += `\n[truncated: memory index exceeds injection limit]`;
-	return { ok: !truncated, content: out, truncated };
-}
-
-export function checkCapacity(entries: IndexEntry[], maxLines: number, maxBytes: number): boolean {
-	const serialized = serializeIndex(entries);
-	if (entries.length > maxLines) return false;
-	if (Buffer.byteLength(serialized, "utf8") > maxBytes) return false;
-	return true;
-}

package/src/topic-file.ts

@@ -1,119 +0,0 @@
-export interface TopicMeta {
-	name: string;
-	description: string;
-	type: string;
-	updated: string;
-}
-
-export const ALLOWED_TYPES = ["user", "feedback", "project", "reference"] as const;
-
-export function buildFrontmatter(meta: TopicMeta): string {
-	return [
-		"---",
-		`name: ${meta.name}`,
-		`description: ${meta.description}`,
-		`type: ${meta.type}`,
-		`updated: ${meta.updated}`,
-		"---",
-		"",
-		"",
-	].join("\n");
-}
-
-export function parseFrontmatter(raw: string): TopicMeta | null {
-	if (!raw.startsWith("---\n")) return null;
-	const endIdx = raw.indexOf("\n---\n", 4);
-	if (endIdx === -1) return null;
-	const block = raw.slice(4, endIdx);
-	const meta: Record<string, string> = {};
-	for (const line of block.split("\n")) {
-		const m = line.match(/^(\w+):\s*(.*)$/);
-		if (m) meta[m[1]] = m[2].trim();
-	}
-	if (!meta.name || !meta.description || !meta.type || !meta.updated) return null;
-	if (!(ALLOWED_TYPES as readonly string[]).includes(meta.type)) return null;
-	return meta as unknown as TopicMeta;
-}
-
-export function appendContent(existing: string | null, entryTitle: string, content: string): string {
-	const section = `## ${entryTitle}\n\n${content}`;
-	if (!existing || existing.trim() === "") return section;
-	return `${existing.trimEnd()}\n\n${section}\n`;
-}
-
-export function updateFrontmatterDate(raw: string, date: string): string {
-	return raw.replace(/^(---\n(?:.*\n)*?)updated: .+(\n---)/m, `$1updated: ${date}$2`);
-}
-
-export function replaceFrontmatterField(raw: string, field: string, value: string): string {
-	const regex = new RegExp(`^(---\n(?:.*\n)*?)${field}: .+(\n)`, "m");
-	return raw.replace(regex, (_full: string, prefix: string, nl: string) => `${prefix}${field}: ${value}${nl}`);
-}
-
-export function removeEntrySection(raw: string, title: string): string {
-	const marker = `## ${title}`;
-	// find start of this entry block
-	const startIdx = raw.indexOf(`\n${marker}\n`) !== -1 ? raw.indexOf(`\n${marker}\n`) : raw.indexOf(marker);
-	if (startIdx === -1) throw new Error(`Entry "${title}" not found`);
-
-	// find start of next ## or EOF
-	const afterHeader = raw.indexOf("\n", startIdx + marker.length);
-	const nextH2 = raw.indexOf("\n## ", afterHeader + 1);
-	const endIdx = nextH2 === -1 ? raw.length : nextH2;
-
-	// remove including preceding blank lines
-	let cutStart = startIdx;
-	while (cutStart > 0 && raw[cutStart - 1] === "\n") cutStart--;
-	// also strip one more \n if present (the blank line separator)
-	if (cutStart > 0 && raw[cutStart - 1] === "\n") cutStart--;
-
-	let result = raw.slice(0, cutStart) + raw.slice(endIdx);
-	// ensure exactly one trailing newline
-	result = result.replace(/\n{3,}$/, "\n\n").replace(/\n{2,}$/, "\n");
-	if (!result.endsWith("\n")) result += "\n";
-	return result;
-}
-
-export function hasEntries(raw: string): boolean {
-	return /^## /m.test(raw);
-}
-
-export interface ParsedEntry {
-	title: string;
-	content: string;
-}
-
-export function parseEntries(raw: string): ParsedEntry[] {
-	const entries: ParsedEntry[] = [];
-	const lines = raw.split("\n");
-	let currentTitle = "";
-	let currentContent: string[] = [];
-	let inEntry = false;
-	let inFrontmatter = false;
-
-	for (const line of lines) {
-		if (line === "---") {
-			inFrontmatter = !inFrontmatter;
-			continue;
-		}
-		if (inFrontmatter) continue;
-
-		const h2 = line.match(/^## (.+)$/);
-		if (h2) {
-			if (inEntry) {
-				entries.push({ title: currentTitle, content: currentContent.join("\n").trim() });
-			}
-			currentTitle = h2[1];
-			currentContent = [];
-			inEntry = true;
-			continue;
-		}
-		if (inEntry) {
-			currentContent.push(line);
-		}
-	}
-	if (inEntry) {
-		entries.push({ title: currentTitle, content: currentContent.join("\n").trim() });
-	}
-	return entries;
-}