@pasko70/pibo 1.9.9 → 1.9.10

package/dist/plugins/context-files-store.js

@@ -0,0 +1,354 @@
+import { createHash, randomUUID } from "node:crypto";
+import { existsSync, mkdirSync, readFileSync, statSync } from "node:fs";
+import { dirname, resolve } from "node:path";
+import { DatabaseSync } from "node:sqlite";
+function readLegacyManagedStore(storePath) {
+    if (!existsSync(storePath))
+        return { files: [] };
+    const parsed = JSON.parse(readFileSync(storePath, "utf8"));
+    if (!Array.isArray(parsed.files))
+        return { files: [] };
+    return {
+        files: parsed.files.flatMap((file) => {
+            if (!file || typeof file !== "object")
+                return [];
+            const candidate = file;
+            if (typeof candidate.key !== "string" || typeof candidate.path !== "string")
+                return [];
+            const label = typeof candidate.label === "string" && candidate.label.trim() ? candidate.label.trim() : candidate.key;
+            const scope = candidate.scope === "agent" ? "agent" : "global";
+            const agentProfileName = scope === "agent" && typeof candidate.agentProfileName === "string"
+                ? candidate.agentProfileName
+                : undefined;
+            if (scope === "agent" && !agentProfileName)
+                return [];
+            return [{
+                    key: candidate.key,
+                    label,
+                    path: candidate.path,
+                    scope,
+                    ...(agentProfileName ? { agentProfileName } : {}),
+                }];
+        }),
+    };
+}
+function fileRowToRecord(row) {
+    return {
+        key: row.key,
+        label: row.label,
+        managedPath: row.managed_path,
+        scope: row.scope,
+        sourceType: row.source_type,
+        ...(row.agent_profile_name ? { agentProfileName: row.agent_profile_name } : {}),
+        ...(row.active_revision_id ? { activeRevisionId: row.active_revision_id } : {}),
+        ...(row.source_ref ? { sourceRef: row.source_ref } : {}),
+        ...(row.source_hash ? { sourceHash: row.source_hash } : {}),
+        createdAt: row.created_at,
+        updatedAt: row.updated_at,
+    };
+}
+function revisionRowToRecord(row) {
+    return {
+        id: row.id,
+        contextFileKey: row.context_file_key,
+        kind: row.kind,
+        contentHash: row.content_hash,
+        content: row.content,
+        createdAt: row.created_at,
+        ...(row.actor_id ? { actorId: row.actor_id } : {}),
+        ...(row.based_on_revision_id ? { basedOnRevisionId: row.based_on_revision_id } : {}),
+        ...(row.source_hash_at_creation ? { sourceHashAtCreation: row.source_hash_at_creation } : {}),
+        ...(row.note ? { note: row.note } : {}),
+    };
+}
+export class ContextFileMetadataStore {
+    legacyStorePath;
+    db;
+    constructor(path, legacyStorePath) {
+        this.legacyStorePath = legacyStorePath;
+        const resolvedPath = resolve(path);
+        mkdirSync(dirname(resolvedPath), { recursive: true });
+        this.db = new DatabaseSync(resolvedPath);
+        this.db.exec("PRAGMA busy_timeout = 5000");
+        this.db.exec("PRAGMA journal_mode = WAL");
+        this.db.exec(`
+			CREATE TABLE IF NOT EXISTS context_files (
+				key TEXT PRIMARY KEY,
+				label TEXT NOT NULL,
+				managed_path TEXT NOT NULL,
+				scope TEXT NOT NULL,
+				source_type TEXT NOT NULL,
+				agent_profile_name TEXT,
+				active_revision_id TEXT,
+				source_ref TEXT,
+				source_hash TEXT,
+				created_at TEXT NOT NULL,
+				updated_at TEXT NOT NULL
+			);
+
+			CREATE INDEX IF NOT EXISTS idx_context_files_scope
+				ON context_files(scope, updated_at);
+
+			CREATE TABLE IF NOT EXISTS context_file_revisions (
+				id TEXT PRIMARY KEY,
+				context_file_key TEXT NOT NULL,
+				kind TEXT NOT NULL,
+				content_hash TEXT NOT NULL,
+				content TEXT NOT NULL,
+				created_at TEXT NOT NULL,
+				actor_id TEXT,
+				based_on_revision_id TEXT,
+				source_hash_at_creation TEXT,
+				note TEXT
+			);
+
+			CREATE INDEX IF NOT EXISTS idx_context_file_revisions_key
+				ON context_file_revisions(context_file_key, created_at DESC);
+		`);
+        this.migrateLegacyStore();
+    }
+    listFiles() {
+        const rows = this.db.prepare("SELECT * FROM context_files ORDER BY updated_at DESC").all();
+        return rows.map(fileRowToRecord);
+    }
+    getFile(key) {
+        const row = this.db.prepare("SELECT * FROM context_files WHERE key = ?").get(key);
+        return row ? fileRowToRecord(row) : undefined;
+    }
+    createFile(input) {
+        const createdAt = input.createdAt ?? new Date().toISOString();
+        const updatedAt = input.updatedAt ?? createdAt;
+        this.db
+            .prepare(`
+				INSERT INTO context_files (
+					key,
+					label,
+					managed_path,
+					scope,
+					source_type,
+					agent_profile_name,
+					active_revision_id,
+					source_ref,
+					source_hash,
+					created_at,
+					updated_at
+				) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+			`)
+            .run(input.key, input.label, input.managedPath, input.scope, "managed", input.agentProfileName ?? null, null, input.sourceRef ?? null, input.sourceHash ?? null, createdAt, updatedAt);
+        const record = this.getFile(input.key);
+        if (!record)
+            throw new Error(`Failed to create context file record "${input.key}"`);
+        return record;
+    }
+    updateFile(input) {
+        this.db
+            .prepare(`
+				UPDATE context_files SET
+					label = ?,
+					managed_path = ?,
+					scope = ?,
+					source_type = ?,
+					agent_profile_name = ?,
+					active_revision_id = ?,
+					source_ref = ?,
+					source_hash = ?,
+					created_at = ?,
+					updated_at = ?
+				WHERE key = ?
+			`)
+            .run(input.label, input.managedPath, input.scope, "managed", input.agentProfileName ?? null, input.activeRevisionId ?? null, input.sourceRef ?? null, input.sourceHash ?? null, input.createdAt, input.updatedAt, input.key);
+        const record = this.getFile(input.key);
+        if (!record)
+            throw new Error(`Failed to update context file record "${input.key}"`);
+        return record;
+    }
+    deleteFile(key) {
+        this.db.exec("BEGIN");
+        try {
+            this.db.prepare("DELETE FROM context_file_revisions WHERE context_file_key = ?").run(key);
+            this.db.prepare("DELETE FROM context_files WHERE key = ?").run(key);
+            this.db.exec("COMMIT");
+        }
+        catch (error) {
+            this.db.exec("ROLLBACK");
+            throw error;
+        }
+    }
+    appendRevision(input) {
+        const id = `rev_${randomUUID()}`;
+        const createdAt = input.createdAt ?? new Date().toISOString();
+        this.db
+            .prepare(`
+				INSERT INTO context_file_revisions (
+					id,
+					context_file_key,
+					kind,
+					content_hash,
+					content,
+					created_at,
+					actor_id,
+					based_on_revision_id,
+					source_hash_at_creation,
+					note
+				) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+			`)
+            .run(id, input.contextFileKey, input.kind, input.contentHash, input.content, createdAt, input.actorId ?? null, input.basedOnRevisionId ?? null, input.sourceHashAtCreation ?? null, input.note ?? null);
+        const revision = this.getRevision(id);
+        if (!revision)
+            throw new Error(`Failed to create context file revision "${id}"`);
+        return revision;
+    }
+    getRevision(id) {
+        const row = this.db.prepare("SELECT * FROM context_file_revisions WHERE id = ?").get(id);
+        return row ? revisionRowToRecord(row) : undefined;
+    }
+    listRevisions(contextFileKey) {
+        const rows = this.db
+            .prepare("SELECT * FROM context_file_revisions WHERE context_file_key = ? ORDER BY created_at DESC")
+            .all(contextFileKey);
+        return rows.map(revisionRowToRecord);
+    }
+    findLatestSourceSnapshot(contextFileKey, sourceHash) {
+        const row = sourceHash
+            ? this.db
+                .prepare(`
+					SELECT * FROM context_file_revisions
+					WHERE context_file_key = ? AND kind = 'source-snapshot' AND content_hash = ?
+					ORDER BY created_at DESC
+					LIMIT 1
+				`)
+                .get(contextFileKey, sourceHash)
+            : this.db
+                .prepare(`
+					SELECT * FROM context_file_revisions
+					WHERE context_file_key = ? AND kind = 'source-snapshot'
+					ORDER BY created_at DESC
+					LIMIT 1
+				`)
+                .get(contextFileKey);
+        return row ? revisionRowToRecord(row) : undefined;
+    }
+    close() {
+        this.db.close();
+    }
+    migrateLegacyStore() {
+        if (!this.legacyStorePath)
+            return;
+        const count = this.db.prepare("SELECT COUNT(*) AS count FROM context_files").get();
+        if (Number(count.count) > 0)
+            return;
+        const legacyStore = readLegacyManagedStore(this.legacyStorePath);
+        if (legacyStore.files.length === 0)
+            return;
+        this.db.exec("BEGIN");
+        try {
+            for (const file of legacyStore.files) {
+                const resolvedPath = resolve(file.path);
+                const createdAt = existsSync(resolvedPath)
+                    ? statSync(resolvedPath).mtime.toISOString()
+                    : new Date().toISOString();
+                this.db
+                    .prepare(`
+						INSERT INTO context_files (
+							key,
+							label,
+							managed_path,
+							scope,
+							source_type,
+							agent_profile_name,
+							active_revision_id,
+							source_ref,
+							source_hash,
+							created_at,
+							updated_at
+						) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+					`)
+                    .run(file.key, file.label, resolvedPath, file.scope, "managed", file.agentProfileName ?? null, null, null, null, createdAt, createdAt);
+                if (!existsSync(resolvedPath))
+                    continue;
+                const content = readFileSync(resolvedPath, "utf8");
+                const contentHash = hashContextFileContent(content);
+                const revisionId = `rev_${randomUUID()}`;
+                this.db
+                    .prepare(`
+						INSERT INTO context_file_revisions (
+							id,
+							context_file_key,
+							kind,
+							content_hash,
+							content,
+							created_at,
+							actor_id,
+							based_on_revision_id,
+							source_hash_at_creation,
+							note
+						) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+					`)
+                    .run(revisionId, file.key, "working", contentHash, content, createdAt, null, null, null, "Legacy managed file import");
+                this.db
+                    .prepare("UPDATE context_files SET active_revision_id = ?, updated_at = ? WHERE key = ?")
+                    .run(revisionId, createdAt, file.key);
+            }
+            this.db.exec("COMMIT");
+        }
+        catch (error) {
+            this.db.exec("ROLLBACK");
+            throw error;
+        }
+    }
+}
+export function hashContextFileContent(content) {
+    return `sha256:${createHash("sha256").update(content).digest("hex")}`;
+}
+function splitLines(content) {
+    return content.split("\n");
+}
+export function buildContextFileDiff(baseContent, targetContent) {
+    const baseLines = splitLines(baseContent);
+    const targetLines = splitLines(targetContent);
+    const rows = baseLines.length;
+    const cols = targetLines.length;
+    const lcs = Array.from({ length: rows + 1 }, () => Array(cols + 1).fill(0));
+    for (let left = rows - 1; left >= 0; left -= 1) {
+        for (let right = cols - 1; right >= 0; right -= 1) {
+            lcs[left][right] = baseLines[left] === targetLines[right]
+                ? lcs[left + 1][right + 1] + 1
+                : Math.max(lcs[left + 1][right], lcs[left][right + 1]);
+        }
+    }
+    const chunks = [];
+    let left = 0;
+    let right = 0;
+    while (left < rows && right < cols) {
+        if (baseLines[left] === targetLines[right]) {
+            pushDiffChunk(chunks, "equal", baseLines[left]);
+            left += 1;
+            right += 1;
+            continue;
+        }
+        if (lcs[left + 1][right] >= lcs[left][right + 1]) {
+            pushDiffChunk(chunks, "remove", baseLines[left]);
+            left += 1;
+            continue;
+        }
+        pushDiffChunk(chunks, "add", targetLines[right]);
+        right += 1;
+    }
+    while (left < rows) {
+        pushDiffChunk(chunks, "remove", baseLines[left]);
+        left += 1;
+    }
+    while (right < cols) {
+        pushDiffChunk(chunks, "add", targetLines[right]);
+        right += 1;
+    }
+    return chunks;
+}
+function pushDiffChunk(chunks, type, line) {
+    const previous = chunks[chunks.length - 1];
+    if (previous && previous.type === type) {
+        previous.lines.push(line);
+        return;
+    }
+    chunks.push({ type, lines: [line] });
+}