@simonsbs/keylore 1.0.0-rc4 → 1.0.0-rc6

package/dist/services/core-mode-service.js

@@ -99,7 +99,12 @@ export class CoreModeService {
             expiresAt: parsed.expiresAt,
             rotationPolicy: parsed.rotationPolicy,
             lastValidatedAt: null,
-            selectionNotes: parsed.selectionNotes,
+            selectionNotes: parsed.llmContext?.trim() || parsed.selectionNotes?.trim() || "",
+            userContext: parsed.userContext?.trim() ||
+                parsed.llmContext?.trim() ||
+                parsed.selectionNotes?.trim() ||
+                "",
+            llmContext: parsed.llmContext?.trim() || parsed.selectionNotes?.trim() || "",
             binding,
             tags: parsed.tags,
             status: parsed.status,
@@ -138,6 +143,17 @@ export class CoreModeService {
         await this.syncCoreAllowRule(updated);
         return updated;
     }
+    async replaceLocalSecret(context, credentialId, secretValue) {
+        const existing = await this.broker.getCredential(context, credentialId);
+        if (!existing) {
+            throw new Error(`Credential ${credentialId} not found.`);
+        }
+        if (existing.owner !== "local") {
+            throw new Error("Only locally stored tokens can be replaced through the admin UI.");
+        }
+        await this.localSecrets.put(`local:${existing.tenantId}:${existing.id}`, secretValue);
+        return existing;
+    }
     async deleteCredential(context, credentialId) {
         const existing = await this.broker.getCredential(context, credentialId);
         if (!existing) {

package/dist/storage/database.js

@@ -1,3 +1,6 @@
+import fs from "node:fs/promises";
+import path from "node:path";
+import { PGlite } from "@electric-sql/pglite";
 import { Pool } from "pg";
 class PostgresDatabase {
     pool;
@@ -7,11 +10,20 @@ class PostgresDatabase {
     async query(text, params) {
         return this.pool.query(text, params);
     }
+    async exec(text) {
+        await this.pool.query(text);
+    }
     async withTransaction(fn) {
         const client = await this.pool.connect();
         try {
             await client.query("BEGIN");
-            const result = await fn(client);
+            const transactionalClient = {
+                query: (text, params) => client.query(text, params),
+                exec: async (text) => {
+                    await client.query(text);
+                },
+            };
+            const result = await fn(transactionalClient);
             await client.query("COMMIT");
             return result;
         }
@@ -30,10 +42,81 @@ class PostgresDatabase {
         await this.pool.end();
     }
 }
-export function createPostgresDatabase(config) {
+class LocalDatabase {
+    database;
+    queue = Promise.resolve();
+    constructor(database) {
+        this.database = database;
+    }
+    enqueue(fn) {
+        const next = this.queue.then(fn, fn);
+        this.queue = next.then(() => undefined, () => undefined);
+        return next;
+    }
+    normalize(result) {
+        const rows = result.rows;
+        return {
+            command: "",
+            rowCount: result.affectedRows && result.affectedRows > 0 ? result.affectedRows : rows.length,
+            oid: 0,
+            rows,
+            fields: (result.fields ?? []),
+        };
+    }
+    async query(text, params) {
+        return this.enqueue(async () => this.normalize(await this.database.query(text, params)));
+    }
+    async exec(text) {
+        await this.enqueue(async () => {
+            await this.database.exec(text);
+        });
+    }
+    async withTransaction(fn) {
+        return this.enqueue(async () => {
+            await this.database.exec("BEGIN");
+            const client = {
+                query: async (text, params) => this.normalize(await this.database.query(text, params)),
+                exec: async (text) => {
+                    await this.database.exec(text);
+                },
+            };
+            try {
+                const result = await fn(client);
+                await this.database.exec("COMMIT");
+                return result;
+            }
+            catch (error) {
+                await this.database.exec("ROLLBACK");
+                throw error;
+            }
+        });
+    }
+    async healthcheck() {
+        await this.enqueue(async () => {
+            await this.database.query("SELECT 1");
+        });
+    }
+    async close() {
+        await this.enqueue(async () => {
+            await this.database.close();
+        });
+    }
+}
+function createPostgresDatabase(config) {
     const pool = new Pool({
         connectionString: config.databaseUrl,
         max: config.databasePoolMax,
     });
     return new PostgresDatabase(pool);
 }
+async function createLocalDatabase(config) {
+    await fs.mkdir(path.dirname(config.localDatabasePath), { recursive: true });
+    const database = new PGlite(config.localDatabasePath);
+    return new LocalDatabase(database);
+}
+export async function createSqlDatabase(config) {
+    if (config.databaseMode === "postgres") {
+        return createPostgresDatabase(config);
+    }
+    return createLocalDatabase(config);
+}

package/dist/storage/in-memory-database.js

@@ -7,11 +7,20 @@ class InMemoryDatabase {
     async query(text, params) {
         return this.pool.query(text, params);
     }
+    async exec(text) {
+        await this.pool.query(text);
+    }
     async withTransaction(fn) {
         const client = await this.pool.connect();
         try {
             await client.query("BEGIN");
-            const result = await fn(client);
+            const transactionalClient = {
+                query: (text, params) => client.query(text, params),
+                exec: async (text) => {
+                    await client.query(text);
+                },
+            };
+            const result = await fn(transactionalClient);
             await client.query("COMMIT");
             return result;
         }

package/dist/storage/migrations.js

@@ -1,7 +1,7 @@
 import fs from "node:fs/promises";
 import path from "node:path";
 export async function runMigrations(database, migrationsDir) {
-    await database.query(`
+    await database.exec(`
     CREATE TABLE IF NOT EXISTS schema_migrations (
       version TEXT PRIMARY KEY,
       applied_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
@@ -20,7 +20,7 @@ export async function runMigrations(database, migrationsDir) {
         }
         const sql = await fs.readFile(path.join(migrationsDir, fileName), "utf8");
         await database.withTransaction(async (client) => {
-            await client.query(sql);
+            await client.exec(sql);
             await client.query("INSERT INTO schema_migrations(version) VALUES ($1)", [version]);
         });
     }

package/migrations/009_v1_context_split.sql

@@ -0,0 +1,30 @@
+ALTER TABLE credentials
+  ADD COLUMN IF NOT EXISTS user_context TEXT;
+
+ALTER TABLE credentials
+  ADD COLUMN IF NOT EXISTS llm_context TEXT;
+
+UPDATE credentials
+SET
+  llm_context = CASE
+    WHEN llm_context IS NULL OR llm_context = '' THEN selection_notes
+    ELSE llm_context
+  END,
+  user_context = CASE
+    WHEN user_context IS NULL OR user_context = '' THEN
+      CASE
+        WHEN llm_context IS NULL OR llm_context = '' THEN selection_notes
+        ELSE llm_context
+      END
+    ELSE user_context
+  END
+WHERE llm_context IS NULL
+   OR llm_context = ''
+   OR user_context IS NULL
+   OR user_context = '';
+
+ALTER TABLE credentials
+  ALTER COLUMN llm_context SET NOT NULL;
+
+ALTER TABLE credentials
+  ALTER COLUMN user_context SET NOT NULL;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@simonsbs/keylore",
-  "version": "1.0.0-rc4",
+  "version": "1.0.0-rc6",
   "description": "MCP credential broker and searchable credential catalogue for LLM coding tools.",
   "type": "module",
   "main": "dist/index.js",
@@ -63,6 +63,7 @@
     "test": "node --test --import tsx src/test/**/*.test.ts"
   },
   "dependencies": {
+    "@electric-sql/pglite": "^0.3.16",
     "@modelcontextprotocol/sdk": "^1.27.1",
     "pg": "^8.20.0",
     "pg-mem": "^3.0.14",