@rangerchaz/aimem 0.1.5 → 0.2.1

package/dist/db/index.js

@@ -2,7 +2,7 @@ import Database from 'better-sqlite3';
 import { homedir } from 'os';
 import { join } from 'path';
 import { mkdirSync, existsSync } from 'fs';
-import { SCHEMA, MIGRATIONS, COMMIT_LINKS_SCHEMA } from './schema.js';
+import { SCHEMA, MIGRATIONS, COMMIT_LINKS_SCHEMA, GUARDRAILS_SCHEMA } from './schema.js';
 function resolveDataDir() {
     return process.env.AIMEM_DATA_DIR || join(homedir(), '.aimem');
 }
@@ -28,6 +28,8 @@ function applyMigrations(database) {
     }
     // Apply commit_links schema
     database.exec(COMMIT_LINKS_SCHEMA);
+    // Apply guardrails schema
+    database.exec(GUARDRAILS_SCHEMA);
 }
 export function getDb() {
     if (!db) {
@@ -69,13 +71,13 @@ export function getAllProjects() {
 // File operations
 export function upsertFile(projectId, path, hash) {
     const db = getDb();
-    const stmt = db.prepare(`
-    INSERT INTO files (project_id, path, hash, last_indexed)
-    VALUES (?, ?, ?, datetime('now'))
-    ON CONFLICT(project_id, path) DO UPDATE SET
-      hash = excluded.hash,
-      last_indexed = datetime('now')
-    RETURNING *
+    const stmt = db.prepare(`
+    INSERT INTO files (project_id, path, hash, last_indexed)
+    VALUES (?, ?, ?, datetime('now'))
+    ON CONFLICT(project_id, path) DO UPDATE SET
+      hash = excluded.hash,
+      last_indexed = datetime('now')
+    RETURNING *
   `);
     return stmt.get(projectId, path, hash);
 }
@@ -98,10 +100,10 @@ export function getProjectFiles(projectId) {
 // Structure operations
 export function insertStructure(fileId, type, name, lineStart, lineEnd, signature, rawContent, metadata = {}) {
     const db = getDb();
-    const stmt = db.prepare(`
-    INSERT INTO structures (file_id, type, name, line_start, line_end, signature, raw_content, metadata)
-    VALUES (?, ?, ?, ?, ?, ?, ?, ?)
-    RETURNING *
+    const stmt = db.prepare(`
+    INSERT INTO structures (file_id, type, name, line_start, line_end, signature, raw_content, metadata)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+    RETURNING *
   `);
     return stmt.get(fileId, type, name, lineStart, lineEnd, signature, rawContent, JSON.stringify(metadata));
 }
@@ -112,30 +114,30 @@ export function deleteFileStructures(fileId) {
 export function searchStructures(query, limit = 20, projectId) {
     const db = getDb();
     if (projectId) {
-        return db.prepare(`
-      SELECT s.* FROM structures s
-      JOIN structures_fts fts ON s.id = fts.rowid
-      JOIN files f ON s.file_id = f.id
-      WHERE structures_fts MATCH ? AND f.project_id = ?
-      ORDER BY rank
-      LIMIT ?
+        return db.prepare(`
+      SELECT s.* FROM structures s
+      JOIN structures_fts fts ON s.id = fts.rowid
+      JOIN files f ON s.file_id = f.id
+      WHERE structures_fts MATCH ? AND f.project_id = ?
+      ORDER BY rank
+      LIMIT ?
     `).all(query, projectId, limit);
     }
-    return db.prepare(`
-    SELECT s.* FROM structures s
-    JOIN structures_fts fts ON s.id = fts.rowid
-    WHERE structures_fts MATCH ?
-    ORDER BY rank
-    LIMIT ?
+    return db.prepare(`
+    SELECT s.* FROM structures s
+    JOIN structures_fts fts ON s.id = fts.rowid
+    WHERE structures_fts MATCH ?
+    ORDER BY rank
+    LIMIT ?
   `).all(query, limit);
 }
 export function getStructuresByName(name, projectId) {
     const db = getDb();
     if (projectId) {
-        return db.prepare(`
-      SELECT s.* FROM structures s
-      JOIN files f ON s.file_id = f.id
-      WHERE s.name = ? AND f.project_id = ?
+        return db.prepare(`
+      SELECT s.* FROM structures s
+      JOIN files f ON s.file_id = f.id
+      WHERE s.name = ? AND f.project_id = ?
     `).all(name, projectId);
     }
     return db.prepare('SELECT * FROM structures WHERE name = ?').all(name);
@@ -155,39 +157,39 @@ export function findProjectForPath(targetPath) {
 // Conversation operations
 export function insertConversation(rawContent, projectId = null, model = null, tool = null, summary = null) {
     const db = getDb();
-    const stmt = db.prepare(`
-    INSERT INTO conversations (project_id, model, tool, summary, raw_content)
-    VALUES (?, ?, ?, ?, ?)
-    RETURNING *
+    const stmt = db.prepare(`
+    INSERT INTO conversations (project_id, model, tool, summary, raw_content)
+    VALUES (?, ?, ?, ?, ?)
+    RETURNING *
   `);
     return stmt.get(projectId, model, tool, summary, rawContent);
 }
 export function searchConversations(query, limit = 20, projectId) {
     const db = getDb();
     if (projectId) {
-        return db.prepare(`
-      SELECT c.* FROM conversations c
-      JOIN conversations_fts fts ON c.id = fts.rowid
-      WHERE conversations_fts MATCH ? AND c.project_id = ?
-      ORDER BY rank
-      LIMIT ?
+        return db.prepare(`
+      SELECT c.* FROM conversations c
+      JOIN conversations_fts fts ON c.id = fts.rowid
+      WHERE conversations_fts MATCH ? AND c.project_id = ?
+      ORDER BY rank
+      LIMIT ?
     `).all(query, projectId, limit);
     }
-    return db.prepare(`
-    SELECT c.* FROM conversations c
-    JOIN conversations_fts fts ON c.id = fts.rowid
-    WHERE conversations_fts MATCH ?
-    ORDER BY rank
-    LIMIT ?
+    return db.prepare(`
+    SELECT c.* FROM conversations c
+    JOIN conversations_fts fts ON c.id = fts.rowid
+    WHERE conversations_fts MATCH ?
+    ORDER BY rank
+    LIMIT ?
   `).all(query, limit);
 }
 // Link operations
 export function createLink(sourceType, sourceId, targetType, targetId, linkType) {
     const db = getDb();
-    const stmt = db.prepare(`
-    INSERT OR IGNORE INTO links (source_type, source_id, target_type, target_id, link_type)
-    VALUES (?, ?, ?, ?, ?)
-    RETURNING *
+    const stmt = db.prepare(`
+    INSERT OR IGNORE INTO links (source_type, source_id, target_type, target_id, link_type)
+    VALUES (?, ?, ?, ?, ?)
+    RETURNING *
   `);
     return stmt.get(sourceType, sourceId, targetType, targetId, linkType);
 }
@@ -202,10 +204,10 @@ export function getLinksTo(targetType, targetId) {
 // Extraction operations
 export function insertExtraction(conversationId, type, content, metadata = {}) {
     const db = getDb();
-    const stmt = db.prepare(`
-    INSERT INTO extractions (conversation_id, type, content, metadata)
-    VALUES (?, ?, ?, ?)
-    RETURNING *
+    const stmt = db.prepare(`
+    INSERT INTO extractions (conversation_id, type, content, metadata)
+    VALUES (?, ?, ?, ?)
+    RETURNING *
   `);
     return stmt.get(conversationId, type, content, JSON.stringify(metadata));
 }
@@ -221,30 +223,30 @@ export function searchExtractions(query, limit = 20, projectId) {
     const db = getDb();
     const searchTerm = `%${query}%`;
     if (projectId) {
-        return db.prepare(`
-      SELECT e.* FROM extractions e
-      JOIN conversations c ON e.conversation_id = c.id
-      WHERE e.content LIKE ? AND c.project_id = ?
-      ORDER BY e.id DESC
-      LIMIT ?
+        return db.prepare(`
+      SELECT e.* FROM extractions e
+      JOIN conversations c ON e.conversation_id = c.id
+      WHERE e.content LIKE ? AND c.project_id = ?
+      ORDER BY e.id DESC
+      LIMIT ?
     `).all(searchTerm, projectId, limit);
     }
-    return db.prepare(`
-    SELECT * FROM extractions
-    WHERE content LIKE ?
-    ORDER BY id DESC
-    LIMIT ?
+    return db.prepare(`
+    SELECT * FROM extractions
+    WHERE content LIKE ?
+    ORDER BY id DESC
+    LIMIT ?
   `).all(searchTerm, limit);
 }
 export function isDuplicateExtraction(content, projectId, windowSeconds = 300) {
     const db = getDb();
     // Check for duplicate extraction content within the last N seconds
-    const stmt = db.prepare(`
-    SELECT COUNT(*) as count FROM extractions e
-    JOIN conversations c ON e.conversation_id = c.id
-    WHERE e.content = ?
-    AND (c.project_id = ? OR (c.project_id IS NULL AND ? IS NULL))
-    AND datetime(c.timestamp) > datetime('now', '-' || ? || ' seconds')
+    const stmt = db.prepare(`
+    SELECT COUNT(*) as count FROM extractions e
+    JOIN conversations c ON e.conversation_id = c.id
+    WHERE e.content = ?
+    AND (c.project_id = ? OR (c.project_id IS NULL AND ? IS NULL))
+    AND datetime(c.timestamp) > datetime('now', '-' || ? || ' seconds')
   `);
     const result = stmt.get(content.trim(), projectId, projectId, windowSeconds);
     return result.count > 0;
@@ -278,80 +280,97 @@ export function getConversationById(id) {
 // Get full conversations for a project (for long-term memory)
 export function getFullConversations(projectId, limit = 50, offset = 0) {
     const db = getDb();
-    return db.prepare(`
-    SELECT * FROM conversations
-    WHERE project_id = ?
-    ORDER BY timestamp DESC
-    LIMIT ? OFFSET ?
+    return db.prepare(`
+    SELECT * FROM conversations
+    WHERE project_id = ?
+    ORDER BY timestamp DESC
+    LIMIT ? OFFSET ?
   `).all(projectId, limit, offset);
 }
+// Get recent conversations (optionally scoped to project)
+export function getRecentConversations(limit = 10, projectId) {
+    const db = getDb();
+    if (projectId) {
+        return db.prepare(`
+      SELECT * FROM conversations
+      WHERE project_id = ?
+      ORDER BY timestamp DESC
+      LIMIT ?
+    `).all(projectId, limit);
+    }
+    return db.prepare(`
+    SELECT * FROM conversations
+    ORDER BY timestamp DESC
+    LIMIT ?
+  `).all(limit);
+}
 // Search conversations and return full content
 export function searchFullConversations(query, limit = 20, projectId) {
     const db = getDb();
     if (projectId) {
-        return db.prepare(`
-      SELECT c.* FROM conversations c
-      JOIN conversations_fts fts ON c.id = fts.rowid
-      WHERE conversations_fts MATCH ? AND c.project_id = ?
-      ORDER BY rank
-      LIMIT ?
+        return db.prepare(`
+      SELECT c.* FROM conversations c
+      JOIN conversations_fts fts ON c.id = fts.rowid
+      WHERE conversations_fts MATCH ? AND c.project_id = ?
+      ORDER BY rank
+      LIMIT ?
     `).all(query, projectId, limit);
     }
-    return db.prepare(`
-    SELECT c.* FROM conversations c
-    JOIN conversations_fts fts ON c.id = fts.rowid
-    WHERE conversations_fts MATCH ?
-    ORDER BY rank
-    LIMIT ?
+    return db.prepare(`
+    SELECT c.* FROM conversations c
+    JOIN conversations_fts fts ON c.id = fts.rowid
+    WHERE conversations_fts MATCH ?
+    ORDER BY rank
+    LIMIT ?
   `).all(query, limit);
 }
 // Get all structures for a project with file paths
 export function getAllProjectStructures(projectId) {
     const db = getDb();
-    return db.prepare(`
-    SELECT s.*, f.path as file_path
-    FROM structures s
-    JOIN files f ON s.file_id = f.id
-    WHERE f.project_id = ?
-    ORDER BY f.path, s.line_start
+    return db.prepare(`
+    SELECT s.*, f.path as file_path
+    FROM structures s
+    JOIN files f ON s.file_id = f.id
+    WHERE f.project_id = ?
+    ORDER BY f.path, s.line_start
   `).all(projectId);
 }
 // Get all links for a project
 export function getAllProjectLinks(projectId) {
     const db = getDb();
-    return db.prepare(`
-    SELECT DISTINCT l.* FROM links l
-    LEFT JOIN structures s ON l.source_type = 'structure' AND l.source_id = s.id
-    LEFT JOIN files f ON s.file_id = f.id
-    LEFT JOIN conversations c ON l.source_type = 'conversation' AND l.source_id = c.id
-    WHERE f.project_id = ? OR c.project_id = ?
+    return db.prepare(`
+    SELECT DISTINCT l.* FROM links l
+    LEFT JOIN structures s ON l.source_type = 'structure' AND l.source_id = s.id
+    LEFT JOIN files f ON s.file_id = f.id
+    LEFT JOIN conversations c ON l.source_type = 'conversation' AND l.source_id = c.id
+    WHERE f.project_id = ? OR c.project_id = ?
   `).all(projectId, projectId);
 }
 // Get all extractions for a project
 export function getAllProjectExtractions(projectId) {
     const db = getDb();
-    return db.prepare(`
-    SELECT e.* FROM extractions e
-    JOIN conversations c ON e.conversation_id = c.id
-    WHERE c.project_id = ?
-    ORDER BY c.timestamp DESC
+    return db.prepare(`
+    SELECT e.* FROM extractions e
+    JOIN conversations c ON e.conversation_id = c.id
+    WHERE c.project_id = ?
+    ORDER BY c.timestamp DESC
   `).all(projectId);
 }
 // ============ Git Operations ============
 // Commit operations
 export function upsertCommit(projectId, hash, shortHash, authorName, authorEmail, timestamp, subject, body, parentHashes = []) {
     const db = getDb();
-    const stmt = db.prepare(`
-    INSERT INTO commits (project_id, hash, short_hash, author_name, author_email, timestamp, subject, body, parent_hashes)
-    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
-    ON CONFLICT(project_id, hash) DO UPDATE SET
-      short_hash = excluded.short_hash,
-      author_name = excluded.author_name,
-      author_email = excluded.author_email,
-      subject = excluded.subject,
-      body = excluded.body,
-      parent_hashes = excluded.parent_hashes
-    RETURNING *
+    const stmt = db.prepare(`
+    INSERT INTO commits (project_id, hash, short_hash, author_name, author_email, timestamp, subject, body, parent_hashes)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+    ON CONFLICT(project_id, hash) DO UPDATE SET
+      short_hash = excluded.short_hash,
+      author_name = excluded.author_name,
+      author_email = excluded.author_email,
+      subject = excluded.subject,
+      body = excluded.body,
+      parent_hashes = excluded.parent_hashes
+    RETURNING *
   `);
     return stmt.get(projectId, hash, shortHash, authorName, authorEmail, timestamp, subject, body, JSON.stringify(parentHashes));
 }
@@ -366,36 +385,36 @@ export function getCommitById(id) {
 export function searchCommits(query, limit = 20, projectId) {
     const db = getDb();
     if (projectId) {
-        return db.prepare(`
-      SELECT c.* FROM commits c
-      JOIN commits_fts fts ON c.id = fts.rowid
-      WHERE commits_fts MATCH ? AND c.project_id = ?
-      ORDER BY rank
-      LIMIT ?
+        return db.prepare(`
+      SELECT c.* FROM commits c
+      JOIN commits_fts fts ON c.id = fts.rowid
+      WHERE commits_fts MATCH ? AND c.project_id = ?
+      ORDER BY rank
+      LIMIT ?
     `).all(query, projectId, limit);
     }
-    return db.prepare(`
-    SELECT c.* FROM commits c
-    JOIN commits_fts fts ON c.id = fts.rowid
-    WHERE commits_fts MATCH ?
-    ORDER BY rank
-    LIMIT ?
+    return db.prepare(`
+    SELECT c.* FROM commits c
+    JOIN commits_fts fts ON c.id = fts.rowid
+    WHERE commits_fts MATCH ?
+    ORDER BY rank
+    LIMIT ?
   `).all(query, limit);
 }
 export function getRecentCommits(projectId, limit = 50) {
     const db = getDb();
-    return db.prepare(`
-    SELECT * FROM commits WHERE project_id = ?
-    ORDER BY timestamp DESC LIMIT ?
+    return db.prepare(`
+    SELECT * FROM commits WHERE project_id = ?
+    ORDER BY timestamp DESC LIMIT ?
   `).all(projectId, limit);
 }
 // Commit link operations
 export function createCommitLink(commitId, targetType, targetId, linkType) {
     const db = getDb();
-    const stmt = db.prepare(`
-    INSERT OR IGNORE INTO commit_links (commit_id, target_type, target_id, link_type)
-    VALUES (?, ?, ?, ?)
-    RETURNING *
+    const stmt = db.prepare(`
+    INSERT OR IGNORE INTO commit_links (commit_id, target_type, target_id, link_type)
+    VALUES (?, ?, ?, ?)
+    RETURNING *
   `);
     return stmt.get(commitId, targetType, targetId, linkType);
 }
@@ -409,31 +428,31 @@ export function getLinksToCommit(targetType, targetId) {
 }
 export function getCommitsForStructure(structureId) {
     const db = getDb();
-    return db.prepare(`
-    SELECT c.* FROM commits c
-    JOIN commit_links cl ON c.id = cl.commit_id
-    WHERE cl.target_type = 'structure' AND cl.target_id = ?
-    ORDER BY c.timestamp DESC
+    return db.prepare(`
+    SELECT c.* FROM commits c
+    JOIN commit_links cl ON c.id = cl.commit_id
+    WHERE cl.target_type = 'structure' AND cl.target_id = ?
+    ORDER BY c.timestamp DESC
   `).all(structureId);
 }
 export function getCommitsForExtraction(extractionId) {
     const db = getDb();
-    return db.prepare(`
-    SELECT c.* FROM commits c
-    JOIN commit_links cl ON c.id = cl.commit_id
-    WHERE cl.target_type = 'extraction' AND cl.target_id = ?
-    ORDER BY c.timestamp DESC
+    return db.prepare(`
+    SELECT c.* FROM commits c
+    JOIN commit_links cl ON c.id = cl.commit_id
+    WHERE cl.target_type = 'extraction' AND cl.target_id = ?
+    ORDER BY c.timestamp DESC
   `).all(extractionId);
 }
 // Update structure authorship
 export function updateStructureAuthorship(structureId, author, authorEmail, commitHash) {
     const db = getDb();
-    db.prepare(`
-    UPDATE structures SET
-      last_author = ?,
-      last_author_email = ?,
-      last_commit_hash = ?
-    WHERE id = ?
+    db.prepare(`
+    UPDATE structures SET
+      last_author = ?,
+      last_author_email = ?,
+      last_commit_hash = ?
+    WHERE id = ?
   `).run(author, authorEmail, commitHash, structureId);
 }
 // Get uncommitted decisions (extractions created since last commit)
@@ -441,28 +460,329 @@ export function getUncommittedExtractions(projectId, sinceCommitHash) {
     const db = getDb();
     if (sinceCommitHash) {
         // Get extractions created after the specified commit's timestamp
-        return db.prepare(`
-      SELECT e.* FROM extractions e
-      JOIN conversations c ON e.conversation_id = c.id
-      WHERE c.project_id = ?
-        AND c.timestamp > (
-          SELECT timestamp FROM commits WHERE project_id = ? AND hash = ?
-        )
-        AND e.id NOT IN (
-          SELECT target_id FROM commit_links WHERE target_type = 'extraction'
-        )
-      ORDER BY c.timestamp DESC
+        return db.prepare(`
+      SELECT e.* FROM extractions e
+      JOIN conversations c ON e.conversation_id = c.id
+      WHERE c.project_id = ?
+        AND c.timestamp > (
+          SELECT timestamp FROM commits WHERE project_id = ? AND hash = ?
+        )
+        AND e.id NOT IN (
+          SELECT target_id FROM commit_links WHERE target_type = 'extraction'
+        )
+      ORDER BY c.timestamp DESC
     `).all(projectId, projectId, sinceCommitHash);
     }
     // Get all extractions not linked to any commit
-    return db.prepare(`
-    SELECT e.* FROM extractions e
-    JOIN conversations c ON e.conversation_id = c.id
-    WHERE c.project_id = ?
-      AND e.id NOT IN (
-        SELECT target_id FROM commit_links WHERE target_type = 'extraction'
-      )
-    ORDER BY c.timestamp DESC
+    return db.prepare(`
+    SELECT e.* FROM extractions e
+    JOIN conversations c ON e.conversation_id = c.id
+    WHERE c.project_id = ?
+      AND e.id NOT IN (
+        SELECT target_id FROM commit_links WHERE target_type = 'extraction'
+      )
+    ORDER BY c.timestamp DESC
+  `).all(projectId);
+}
+// ============ Guardrails Operations (DIK) ============
+// Guardrail CRUD
+export function insertGuardrail(projectId, category, rule, rationale = null, severity = 'warn', source = 'explicit', sourceFile = null) {
+    const db = getDb();
+    const stmt = db.prepare(`
+    INSERT INTO guardrails (project_id, category, rule, rationale, severity, source, source_file)
+    VALUES (?, ?, ?, ?, ?, ?, ?)
+    RETURNING *
+  `);
+    return stmt.get(projectId, category, rule, rationale, severity, source, sourceFile);
+}
+export function getGuardrail(id) {
+    const db = getDb();
+    return db.prepare('SELECT * FROM guardrails WHERE id = ?').get(id);
+}
+export function getProjectGuardrails(projectId, options = {}) {
+    const db = getDb();
+    const { category, confirmedOnly = false, activeOnly = true } = options;
+    let sql = 'SELECT * FROM guardrails WHERE project_id = ?';
+    const params = [projectId];
+    if (category) {
+        sql += ' AND category = ?';
+        params.push(category);
+    }
+    if (confirmedOnly) {
+        sql += ' AND confirmed = 1';
+    }
+    if (activeOnly) {
+        sql += ' AND active = 1';
+    }
+    sql += ' ORDER BY created_at DESC';
+    return db.prepare(sql).all(...params);
+}
+export function confirmGuardrail(id) {
+    const db = getDb();
+    const result = db.prepare('UPDATE guardrails SET confirmed = 1 WHERE id = ?').run(id);
+    return result.changes > 0;
+}
+export function deactivateGuardrail(id) {
+    const db = getDb();
+    const result = db.prepare('UPDATE guardrails SET active = 0 WHERE id = ?').run(id);
+    return result.changes > 0;
+}
+export function deleteGuardrail(id) {
+    const db = getDb();
+    const result = db.prepare('DELETE FROM guardrails WHERE id = ?').run(id);
+    return result.changes > 0;
+}
+// Guardrail events
+export function insertGuardrailEvent(guardrailId, eventType, context = null, response = null, dikLevel = null) {
+    const db = getDb();
+    const stmt = db.prepare(`
+    INSERT INTO guardrail_events (guardrail_id, event_type, context, response, dik_level)
+    VALUES (?, ?, ?, ?, ?)
+    RETURNING *
+  `);
+    return stmt.get(guardrailId, eventType, context, response, dikLevel);
+}
+export function getGuardrailEvents(guardrailId) {
+    const db = getDb();
+    return db.prepare('SELECT * FROM guardrail_events WHERE guardrail_id = ? ORDER BY timestamp DESC').all(guardrailId);
+}
+export function getGuardrailEvent(id) {
+    const db = getDb();
+    return db.prepare('SELECT * FROM guardrail_events WHERE id = ?').get(id);
+}
+export function getOverrideEvents(projectId) {
+    const db = getDb();
+    return db.prepare(`
+    SELECT e.* FROM guardrail_events e
+    JOIN guardrails g ON e.guardrail_id = g.id
+    WHERE g.project_id = ? AND e.event_type = 'overridden'
+    ORDER BY e.timestamp DESC
+  `).all(projectId);
+}
+// Project DIK
+export function getOrCreateProjectDik(projectId) {
+    const db = getDb();
+    let dik = db.prepare('SELECT * FROM project_dik WHERE project_id = ?').get(projectId);
+    if (!dik) {
+        dik = db.prepare(`
+      INSERT INTO project_dik (project_id) VALUES (?) RETURNING *
+    `).get(projectId);
+    }
+    return dik;
+}
+export function updateProjectDik(projectId, updates) {
+    const db = getDb();
+    // Build dynamic update query
+    const fields = [];
+    const values = [];
+    if (updates.level !== undefined) {
+        fields.push('level = ?');
+        values.push(updates.level);
+    }
+    if (updates.rules_confirmed !== undefined) {
+        fields.push('rules_confirmed = ?');
+        values.push(updates.rules_confirmed);
+    }
+    if (updates.rules_inferred !== undefined) {
+        fields.push('rules_inferred = ?');
+        values.push(updates.rules_inferred);
+    }
+    if (updates.conversations !== undefined) {
+        fields.push('conversations = ?');
+        values.push(updates.conversations);
+    }
+    if (updates.corrections_made !== undefined) {
+        fields.push('corrections_made = ?');
+        values.push(updates.corrections_made);
+    }
+    if (updates.overrides_regretted !== undefined) {
+        fields.push('overrides_regretted = ?');
+        values.push(updates.overrides_regretted);
+    }
+    fields.push("last_updated = datetime('now')");
+    values.push(projectId);
+    const sql = `UPDATE project_dik SET ${fields.join(', ')} WHERE project_id = ? RETURNING *`;
+    return db.prepare(sql).get(...values);
+}
+export function incrementDikCounter(projectId, counter) {
+    const db = getDb();
+    // Ensure project_dik exists
+    getOrCreateProjectDik(projectId);
+    db.prepare(`UPDATE project_dik SET ${counter} = ${counter} + 1, last_updated = datetime('now') WHERE project_id = ?`).run(projectId);
+}
+// Get guardrail history for a specific rule (for response generation)
+export function getGuardrailHistory(guardrailId) {
+    const db = getDb();
+    const events = db.prepare(`
+    SELECT event_type, COUNT(*) as count FROM guardrail_events
+    WHERE guardrail_id = ?
+    GROUP BY event_type
+  `).all(guardrailId);
+    let overrides = 0;
+    let vindicated = false;
+    for (const e of events) {
+        if (e.event_type === 'overridden')
+            overrides = e.count;
+        if (e.event_type === 'vindicated')
+            vindicated = true;
+    }
+    return { overrides, vindicated };
+}
+// Toggle ambient personality mode
+export function setAmbientPersonality(projectId, enabled) {
+    const db = getDb();
+    // Ensure project_dik exists
+    getOrCreateProjectDik(projectId);
+    db.prepare(`UPDATE project_dik SET ambient_personality = ?, last_updated = datetime('now') WHERE project_id = ?`).run(enabled ? 1 : 0, projectId);
+}
+// Get guardrails config for a project
+export function getGuardrailsConfig(projectId) {
+    const dik = getOrCreateProjectDik(projectId);
+    return {
+        enabled: true, // Guardrails always enabled if project_dik exists
+        ambient_personality: dik.ambient_personality === 1,
+    };
+}
+// Manually set DIK level (overrides calculated value)
+export function setDikLevel(projectId, level) {
+    const db = getDb();
+    const clampedLevel = Math.max(1, Math.min(10, level));
+    getOrCreateProjectDik(projectId);
+    db.prepare(`UPDATE project_dik SET level = ?, last_updated = datetime('now') WHERE project_id = ?`).run(clampedLevel, projectId);
+}
+// Check if DIK level is manually set (level != 2 default and stats don't match)
+export function isDikManuallySet(projectId) {
+    const dik = getOrCreateProjectDik(projectId);
+    // If level is non-default but stats are zero, it's manually set
+    return dik.level !== 2 && dik.rules_confirmed === 0 && dik.corrections_made === 0 && dik.overrides_regretted === 0 && dik.conversations === 0;
+}
+// ============ Vindication Operations ============
+// Insert guardrail event with vindication context (enhanced version)
+export function insertGuardrailEventWithContext(guardrailId, eventType, context = null, response = null, dikLevel = null, vindicationContext) {
+    const db = getDb();
+    if (vindicationContext && eventType === 'overridden') {
+        const stmt = db.prepare(`
+      INSERT INTO guardrail_events (
+        guardrail_id, event_type, context, response, dik_level,
+        suggestion, code_context, file_path, line_start, line_end, content_hash, vindication_pending
+      )
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 1)
+      RETURNING *
+    `);
+        return stmt.get(guardrailId, eventType, context, response, dikLevel, vindicationContext.suggestion || null, vindicationContext.codeContext || null, vindicationContext.filePath || null, vindicationContext.lineStart || null, vindicationContext.lineEnd || null, vindicationContext.contentHash || null);
+    }
+    // Fall back to regular insert for non-override events
+    return insertGuardrailEvent(guardrailId, eventType, context, response, dikLevel);
+}
+// Get pending vindications for a project
+export function getPendingVindications(projectId) {
+    const db = getDb();
+    const rows = db.prepare(`
+    SELECT
+      e.id as eventId,
+      e.guardrail_id as guardrailId,
+      g.project_id as projectId,
+      e.suggestion,
+      e.file_path as filePath,
+      e.line_start as lineStart,
+      e.line_end as lineEnd,
+      e.code_context as originalCode,
+      e.content_hash as contentHash,
+      e.context as reason,
+      e.timestamp
+    FROM guardrail_events e
+    JOIN guardrails g ON e.guardrail_id = g.id
+    WHERE g.project_id = ?
+      AND e.event_type = 'overridden'
+      AND e.vindication_pending = 1
+      AND e.suggestion IS NOT NULL
+      AND e.file_path IS NOT NULL
+    ORDER BY e.timestamp DESC
+  `).all(projectId);
+    return rows;
+}
+// Get pending vindications for specific files
+export function getPendingVindicationsForFile(projectId, filePath) {
+    const db = getDb();
+    const rows = db.prepare(`
+    SELECT
+      e.id as eventId,
+      e.guardrail_id as guardrailId,
+      g.project_id as projectId,
+      e.suggestion,
+      e.file_path as filePath,
+      e.line_start as lineStart,
+      e.line_end as lineEnd,
+      e.code_context as originalCode,
+      e.content_hash as contentHash,
+      e.context as reason,
+      e.timestamp
+    FROM guardrail_events e
+    JOIN guardrails g ON e.guardrail_id = g.id
+    WHERE g.project_id = ?
+      AND e.file_path = ?
+      AND e.event_type = 'overridden'
+      AND e.vindication_pending = 1
+      AND e.suggestion IS NOT NULL
+    ORDER BY e.timestamp DESC
+  `).all(projectId, filePath);
+    return rows;
+}
+// Mark a vindication as checked (not vindicated)
+export function markVindicationChecked(eventId) {
+    const db = getDb();
+    db.prepare(`
+    UPDATE guardrail_events
+    SET vindication_pending = 0, checked_at = datetime('now')
+    WHERE id = ?
+  `).run(eventId);
+}
+// Mark a vindication as complete (vindicated)
+export function markVindicated(eventId) {
+    const db = getDb();
+    db.prepare(`
+    UPDATE guardrail_events
+    SET vindication_pending = 0, checked_at = datetime('now')
+    WHERE id = ?
+  `).run(eventId);
+}
+// Expire old pending vindications
+export function expireOldVindications(projectId, maxAgeDays = 30) {
+    const db = getDb();
+    const result = db.prepare(`
+    UPDATE guardrail_events
+    SET vindication_pending = 0, checked_at = datetime('now')
+    WHERE id IN (
+      SELECT e.id FROM guardrail_events e
+      JOIN guardrails g ON e.guardrail_id = g.id
+      WHERE g.project_id = ?
+        AND e.vindication_pending = 1
+        AND datetime(e.timestamp) < datetime('now', '-' || ? || ' days')
+    )
+  `).run(projectId, maxAgeDays);
+    return result.changes;
+}
+// Get all vindicated events for a project (for CLI display)
+export function getVindicatedEvents(projectId) {
+    const db = getDb();
+    return db.prepare(`
+    SELECT e.* FROM guardrail_events e
+    JOIN guardrails g ON e.guardrail_id = g.id
+    WHERE g.project_id = ? AND e.event_type = 'vindicated'
+    ORDER BY e.timestamp DESC
   `).all(projectId);
 }
+export function getOverrideEventsWithRules(projectId, pendingOnly = false) {
+    const db = getDb();
+    let sql = `
+    SELECT e.*, g.rule, g.category FROM guardrail_events e
+    JOIN guardrails g ON e.guardrail_id = g.id
+    WHERE g.project_id = ? AND e.event_type = 'overridden'
+  `;
+    if (pendingOnly) {
+        sql += ' AND e.vindication_pending = 1';
+    }
+    sql += ' ORDER BY e.timestamp DESC';
+    return db.prepare(sql).all(projectId);
+}
 //# sourceMappingURL=index.js.map