nemoris 0.1.0 → 0.1.2

package/src/runtime/envelope.js

@@ -1,114 +1,114 @@
-import { ulid } from "./ulid.js";
-
-export class EnvelopeStore {
-  constructor(db, stateStore = null) {
-    this.db = db;
-    this.stateStore = stateStore;
-    this._ensureSchema();
-  }
-
-  _ensureSchema() {
-    this.db.exec(`
-      CREATE TABLE IF NOT EXISTS envelopes (
-        id              TEXT PRIMARY KEY,
-        correlation_id  TEXT,
-        source_agent    TEXT,
-        criticality     TEXT NOT NULL,
-        created_at      TEXT NOT NULL,
-        ack_status      TEXT NOT NULL DEFAULT 'pending',
-        ack_updated_at  TEXT,
-        payload_type    TEXT NOT NULL,
-        payload         TEXT NOT NULL
-      );
-      CREATE INDEX IF NOT EXISTS idx_envelopes_correlation ON envelopes(correlation_id);
-      CREATE INDEX IF NOT EXISTS idx_envelopes_ack_status ON envelopes(ack_status);
-      CREATE INDEX IF NOT EXISTS idx_envelopes_source_agent ON envelopes(source_agent);
-    `);
-  }
-
-  create({ sourceAgent, criticality, payloadType, payload, correlationId = null }) {
-    const id = ulid();
-    const now = new Date().toISOString();
-    const payloadJson = JSON.stringify(payload);
-    this.db.prepare(`
-      INSERT INTO envelopes (id, correlation_id, source_agent, criticality, created_at, ack_status, payload_type, payload)
-      VALUES (?, ?, ?, ?, ?, 'pending', ?, ?)
-    `).run(id, correlationId, sourceAgent, criticality, now, payloadType, payloadJson);
-    return this.get(id);
-  }
-
-  get(id) {
-    return this.db.prepare("SELECT * FROM envelopes WHERE id = ?").get(id) || null;
-  }
-
-  list({ limit = 50, ackStatus = null } = {}) {
-    if (ackStatus) {
-      return this.db.prepare("SELECT * FROM envelopes WHERE ack_status = ? ORDER BY created_at DESC LIMIT ?").all(ackStatus, limit);
-    }
-    return this.db.prepare("SELECT * FROM envelopes ORDER BY created_at DESC LIMIT ?").all(limit);
-  }
-
-  updateAck(id, newStatus) {
-    const now = new Date().toISOString();
-    const result = this.db.prepare(`
-      UPDATE envelopes SET ack_status = ?, ack_updated_at = ?
-      WHERE id = ? AND ack_status != 'dead_letter'
-    `).run(newStatus, now, id);
-    return result.changes > 0;
-  }
-
-  prune(retentionDays = 30) {
-    const cutoff = new Date(Date.now() - retentionDays * 24 * 60 * 60 * 1000).toISOString();
-    const result = this.db.prepare(`
-      DELETE FROM envelopes
-      WHERE created_at < ? AND ack_status IN ('delivered', 'responded')
-    `).run(cutoff);
-    return result.changes;
-  }
-
-  getSchemaVersion() {
-    if (!this.stateStore) return 1;
-    const row = this.stateStore.db.prepare(
-      "SELECT value_json FROM scheduler_meta WHERE key = ?"
-    ).get("schema_version");
-    return row ? parseInt(JSON.parse(row.value_json), 10) : 1;
-  }
-
-  setSchemaVersion(version) {
-    if (!this.stateStore) return;
-    this.stateStore.db.prepare(`
-      INSERT INTO scheduler_meta(key, value_json) VALUES (?, ?)
-      ON CONFLICT(key) DO UPDATE SET value_json = excluded.value_json
-    `).run("schema_version", JSON.stringify(version));
-  }
-
-  runBackfill(stateStore, taskContract) {
-    const version = this.getSchemaVersion();
-    if (version >= 2) return { backfilled: 0, skipped: true };
-
-    let backfilled = 0;
-    try {
-      const inFlight = stateStore.db.prepare(`
-        SELECT * FROM interactive_jobs
-        WHERE triggered_by IS NOT NULL AND status IN ('queued', 'running')
-      `).all();
-
-      for (const job of inFlight) {
-        if (taskContract) {
-          try {
-            taskContract.createTask({
-              ownerAgent: job.triggered_by,
-              assignedAgent: job.agent_id,
-              objective: (job.input || "migrated task").slice(0, 500),
-              deadlineMinutes: 30,
-            });
-            backfilled++;
-          } catch { /* skip if creation fails */ }
-        }
-      }
-    } catch { /* table may not have these columns */ }
-
-    this.setSchemaVersion(2);
-    return { backfilled, skipped: false };
-  }
-}
+import { ulid } from "./ulid.js";
+
+export class EnvelopeStore {
+  constructor(db, stateStore = null) {
+    this.db = db;
+    this.stateStore = stateStore;
+    this._ensureSchema();
+  }
+
+  _ensureSchema() {
+    this.db.exec(`
+      CREATE TABLE IF NOT EXISTS envelopes (
+        id              TEXT PRIMARY KEY,
+        correlation_id  TEXT,
+        source_agent    TEXT,
+        criticality     TEXT NOT NULL,
+        created_at      TEXT NOT NULL,
+        ack_status      TEXT NOT NULL DEFAULT 'pending',
+        ack_updated_at  TEXT,
+        payload_type    TEXT NOT NULL,
+        payload         TEXT NOT NULL
+      );
+      CREATE INDEX IF NOT EXISTS idx_envelopes_correlation ON envelopes(correlation_id);
+      CREATE INDEX IF NOT EXISTS idx_envelopes_ack_status ON envelopes(ack_status);
+      CREATE INDEX IF NOT EXISTS idx_envelopes_source_agent ON envelopes(source_agent);
+    `);
+  }
+
+  create({ sourceAgent, criticality, payloadType, payload, correlationId = null }) {
+    const id = ulid();
+    const now = new Date().toISOString();
+    const payloadJson = JSON.stringify(payload);
+    this.db.prepare(`
+      INSERT INTO envelopes (id, correlation_id, source_agent, criticality, created_at, ack_status, payload_type, payload)
+      VALUES (?, ?, ?, ?, ?, 'pending', ?, ?)
+    `).run(id, correlationId, sourceAgent, criticality, now, payloadType, payloadJson);
+    return this.get(id);
+  }
+
+  get(id) {
+    return this.db.prepare("SELECT * FROM envelopes WHERE id = ?").get(id) || null;
+  }
+
+  list({ limit = 50, ackStatus = null } = {}) {
+    if (ackStatus) {
+      return this.db.prepare("SELECT * FROM envelopes WHERE ack_status = ? ORDER BY created_at DESC LIMIT ?").all(ackStatus, limit);
+    }
+    return this.db.prepare("SELECT * FROM envelopes ORDER BY created_at DESC LIMIT ?").all(limit);
+  }
+
+  updateAck(id, newStatus) {
+    const now = new Date().toISOString();
+    const result = this.db.prepare(`
+      UPDATE envelopes SET ack_status = ?, ack_updated_at = ?
+      WHERE id = ? AND ack_status != 'dead_letter'
+    `).run(newStatus, now, id);
+    return result.changes > 0;
+  }
+
+  prune(retentionDays = 30) {
+    const cutoff = new Date(Date.now() - retentionDays * 24 * 60 * 60 * 1000).toISOString();
+    const result = this.db.prepare(`
+      DELETE FROM envelopes
+      WHERE created_at < ? AND ack_status IN ('delivered', 'responded')
+    `).run(cutoff);
+    return result.changes;
+  }
+
+  getSchemaVersion() {
+    if (!this.stateStore) return 1;
+    const row = this.stateStore.db.prepare(
+      "SELECT value_json FROM scheduler_meta WHERE key = ?"
+    ).get("schema_version");
+    return row ? parseInt(JSON.parse(row.value_json), 10) : 1;
+  }
+
+  setSchemaVersion(version) {
+    if (!this.stateStore) return;
+    this.stateStore.db.prepare(`
+      INSERT INTO scheduler_meta(key, value_json) VALUES (?, ?)
+      ON CONFLICT(key) DO UPDATE SET value_json = excluded.value_json
+    `).run("schema_version", JSON.stringify(version));
+  }
+
+  runBackfill(stateStore, taskContract) {
+    const version = this.getSchemaVersion();
+    if (version >= 2) return { backfilled: 0, skipped: true };
+
+    let backfilled = 0;
+    try {
+      const inFlight = stateStore.db.prepare(`
+        SELECT * FROM interactive_jobs
+        WHERE triggered_by IS NOT NULL AND status IN ('queued', 'running')
+      `).all();
+
+      for (const job of inFlight) {
+        if (taskContract) {
+          try {
+            taskContract.createTask({
+              ownerAgent: job.triggered_by,
+              assignedAgent: job.agent_id,
+              objective: (job.input || "migrated task").slice(0, 500),
+              deadlineMinutes: 30,
+            });
+            backfilled++;
+          } catch { /* skip if creation fails */ }
+        }
+      }
+    } catch { /* table may not have these columns */ }
+
+    this.setSchemaVersion(2);
+    return { backfilled, skipped: false };
+  }
+}