@cnstra/persist-postgres 0.1.0

package/dist/index.mjs

@@ -0,0 +1,326 @@
+import {
+  DEFAULT_PREFIX,
+  __name,
+  __publicField,
+  assertSafePrefix,
+  ensureSchema,
+  persistPostgresSchemaSql
+} from "./chunk-U4IKAHS4.mjs";
+
+// src/CNSPostgresStimulationRepository.ts
+import { Pool } from "pg";
+var j = /* @__PURE__ */ __name((v) => JSON.stringify(v != null ? v : null), "j");
+var num = /* @__PURE__ */ __name((v) => Number(v), "num");
+var numOrNull = /* @__PURE__ */ __name((v) => v == null ? null : Number(v), "numOrNull");
+var _CNSPostgresStimulationRepository = class _CNSPostgresStimulationRepository {
+  constructor(opts) {
+    __publicField(this, "db");
+    __publicField(this, "pool");
+    __publicField(this, "ownsPool");
+    __publicField(this, "p");
+    __publicField(this, "autoMigrate");
+    __publicField(this, "ready");
+    var _a, _b;
+    if (opts.pool) {
+      this.pool = opts.pool;
+      this.db = opts.pool;
+      this.ownsPool = false;
+    } else if (opts.connectionString) {
+      this.pool = new Pool({
+        connectionString: opts.connectionString
+      });
+      this.db = this.pool;
+      this.ownsPool = true;
+    } else if (opts.queryable) {
+      this.db = opts.queryable;
+      this.ownsPool = false;
+    } else {
+      throw new Error("[@cnstra/persist-postgres] needs `connectionString`, `pool`, or `queryable`");
+    }
+    this.p = assertSafePrefix((_a = opts.tablePrefix) != null ? _a : DEFAULT_PREFIX);
+    this.autoMigrate = (_b = opts.autoMigrate) != null ? _b : true;
+  }
+  ensure() {
+    var _a;
+    if (!this.autoMigrate) return Promise.resolve();
+    return (_a = this.ready) != null ? _a : this.ready = ensureSchema(this.db, this.p);
+  }
+  /** Run `fn` in a transaction when a Pool is available; else best-effort inline. */
+  async tx(fn) {
+    if (!this.pool) return fn(this.db);
+    const client = await this.pool.connect();
+    try {
+      await client.query("BEGIN");
+      const out = await fn(client);
+      await client.query("COMMIT");
+      return out;
+    } catch (e) {
+      await client.query("ROLLBACK");
+      throw e;
+    } finally {
+      client.release();
+    }
+  }
+  // ── write ──
+  async saveStimulation(s) {
+    await this.ensure();
+    const { p } = this;
+    await this.tx(async (db) => {
+      var _a;
+      await db.query(`INSERT INTO ${p}stimulation (stimulation_id, entry, status, scope_name, updated_at)
+                 VALUES ($1, $2::jsonb, $3, $4, now())
+                 ON CONFLICT (stimulation_id)
+                 DO UPDATE SET entry = EXCLUDED.entry, status = EXCLUDED.status,
+                    scope_name = EXCLUDED.scope_name, updated_at = now()`, [
+        s.stimulationId,
+        j(s.entry),
+        s.status,
+        (_a = s.scopeName) != null ? _a : null
+      ]);
+      await db.query(`DELETE FROM ${p}stimulation_frontier WHERE stimulation_id = $1`, [
+        s.stimulationId
+      ]);
+      for (let i = 0; i < s.progress.tasks.length; i++) {
+        const t = s.progress.tasks[i];
+        await db.query(`INSERT INTO ${p}stimulation_frontier
+                        (stimulation_id, ord, neuron_name, dendrite_collateral_name, input)
+                     VALUES ($1, $2, $3, $4, $5::jsonb)`, [
+          s.stimulationId,
+          i,
+          t.neuronName,
+          t.dendriteCollateralName,
+          t.input ? j(t.input) : null
+        ]);
+      }
+      await db.query(`DELETE FROM ${p}stimulation_context WHERE stimulation_id = $1`, [
+        s.stimulationId
+      ]);
+      for (const [neuronName, value] of Object.entries(s.progress.context)) {
+        await db.query(`INSERT INTO ${p}stimulation_context (stimulation_id, neuron_name, value)
+                     VALUES ($1, $2, $3::jsonb)`, [
+          s.stimulationId,
+          neuronName,
+          j(value)
+        ]);
+      }
+    });
+  }
+  async saveAttempt(a) {
+    await this.ensure();
+    const { p } = this;
+    await this.db.query(`INSERT INTO ${p}stimulation_attempt
+                (stimulation_attempt_id, stimulation_id, attempt_number, status, started_at,
+                 completed_at, hop_count, has_error, replay_of, entry)
+             VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9,$10::jsonb)
+             ON CONFLICT (stimulation_attempt_id) DO UPDATE SET
+                status = EXCLUDED.status, completed_at = EXCLUDED.completed_at,
+                hop_count = EXCLUDED.hop_count, has_error = EXCLUDED.has_error,
+                replay_of = EXCLUDED.replay_of, entry = EXCLUDED.entry`, [
+      a.stimulationAttemptId,
+      a.stimulationId,
+      a.attemptNumber,
+      a.status,
+      a.startedAt,
+      a.completedAt,
+      a.hopCount,
+      a.hasError,
+      a.replayOf,
+      j(a.entry)
+    ]);
+  }
+  async appendTask(t) {
+    await this.ensure();
+    const { p } = this;
+    await this.db.query(`INSERT INTO ${p}task
+                (stimulation_attempt_id, "index", neuron_name, dendrite_collateral_name,
+                 input_index, output, status, error, started_at, duration)
+             VALUES ($1,$2,$3,$4,$5,$6::jsonb,$7,$8,$9,$10)
+             ON CONFLICT (stimulation_attempt_id, "index") DO NOTHING`, [
+      t.stimulationAttemptId,
+      t.index,
+      t.neuronName,
+      t.dendriteCollateralName,
+      t.inputIndex,
+      t.output ? j(t.output) : null,
+      t.status,
+      t.error,
+      t.startedAt,
+      t.duration
+    ]);
+  }
+  // ── read ──
+  async getStimulation(stimulationId) {
+    await this.ensure();
+    const { p } = this;
+    const stimRes = await this.db.query(`SELECT entry, status, scope_name FROM ${p}stimulation WHERE stimulation_id = $1`, [
+      stimulationId
+    ]);
+    const row = stimRes.rows[0];
+    if (!row) return void 0;
+    const [frontier, context] = await Promise.all([
+      this.db.query(`SELECT ord, neuron_name, dendrite_collateral_name, input
+                 FROM ${p}stimulation_frontier WHERE stimulation_id = $1 ORDER BY ord`, [
+        stimulationId
+      ]),
+      this.db.query(`SELECT neuron_name, value FROM ${p}stimulation_context WHERE stimulation_id = $1`, [
+        stimulationId
+      ])
+    ]);
+    return {
+      stimulationId,
+      entry: row.entry,
+      status: row.status,
+      progress: this.buildProgress(frontier.rows, context.rows),
+      ...row.scope_name != null ? {
+        scopeName: row.scope_name
+      } : {}
+    };
+  }
+  async listStimulations(filter) {
+    var _a;
+    await this.ensure();
+    const { p } = this;
+    const scope = (_a = filter == null ? void 0 : filter.scopeName) != null ? _a : null;
+    const stimsRes = await this.db.query(`SELECT stimulation_id, entry, status, scope_name FROM ${p}stimulation
+             WHERE ($1::text IS NULL OR scope_name = $1)
+             ORDER BY updated_at DESC`, [
+      scope
+    ]);
+    const stims = stimsRes.rows;
+    if (stims.length === 0) return [];
+    const ids = stims.map((r) => r.stimulation_id);
+    const [frontierRes, contextRes] = await Promise.all([
+      this.db.query(`SELECT stimulation_id, ord, neuron_name, dendrite_collateral_name, input
+                 FROM ${p}stimulation_frontier WHERE stimulation_id = ANY($1) ORDER BY stimulation_id, ord`, [
+        ids
+      ]),
+      this.db.query(`SELECT stimulation_id, neuron_name, value
+                 FROM ${p}stimulation_context WHERE stimulation_id = ANY($1)`, [
+        ids
+      ])
+    ]);
+    const frontierBy = groupBy(frontierRes.rows, (r) => r.stimulation_id);
+    const contextBy = groupBy(contextRes.rows, (r) => r.stimulation_id);
+    return stims.map((r) => {
+      var _a2, _b;
+      return {
+        stimulationId: r.stimulation_id,
+        entry: r.entry,
+        status: r.status,
+        progress: this.buildProgress((_a2 = frontierBy.get(r.stimulation_id)) != null ? _a2 : [], (_b = contextBy.get(r.stimulation_id)) != null ? _b : []),
+        ...r.scope_name != null ? {
+          scopeName: r.scope_name
+        } : {}
+      };
+    });
+  }
+  async getAttempts(stimulationId) {
+    await this.ensure();
+    const { p } = this;
+    const res = await this.db.query(`SELECT * FROM ${p}stimulation_attempt WHERE stimulation_id = $1 ORDER BY attempt_number`, [
+      stimulationId
+    ]);
+    return res.rows.map(rowToAttempt);
+  }
+  async getTasks(stimulationAttemptId) {
+    await this.ensure();
+    const { p } = this;
+    const res = await this.db.query(`SELECT * FROM ${p}task WHERE stimulation_attempt_id = $1 ORDER BY "index"`, [
+      stimulationAttemptId
+    ]);
+    return res.rows.map(rowToTask);
+  }
+  // ── lifecycle ──
+  async delete(stimulationId) {
+    await this.ensure();
+    const { p } = this;
+    await this.tx(async (db) => {
+      await db.query(`DELETE FROM ${p}task WHERE stimulation_attempt_id IN
+                    (SELECT stimulation_attempt_id FROM ${p}stimulation_attempt WHERE stimulation_id = $1)`, [
+        stimulationId
+      ]);
+      await db.query(`DELETE FROM ${p}stimulation_attempt WHERE stimulation_id = $1`, [
+        stimulationId
+      ]);
+      await db.query(`DELETE FROM ${p}stimulation WHERE stimulation_id = $1`, [
+        stimulationId
+      ]);
+    });
+  }
+  async close() {
+    if (this.ownsPool && this.pool) await this.pool.end();
+  }
+  // ── helpers ──
+  buildProgress(frontier, context) {
+    const tasks = [
+      ...frontier
+    ].sort((a, b) => a.ord - b.ord).map((f) => ({
+      neuronName: f.neuron_name,
+      dendriteCollateralName: f.dendrite_collateral_name,
+      ...f.input != null ? {
+        input: f.input
+      } : {}
+    }));
+    const ctx = {};
+    for (const c of context) ctx[c.neuron_name] = c.value;
+    return {
+      tasks,
+      context: ctx
+    };
+  }
+};
+__name(_CNSPostgresStimulationRepository, "CNSPostgresStimulationRepository");
+var CNSPostgresStimulationRepository = _CNSPostgresStimulationRepository;
+function rowToAttempt(a) {
+  var _a;
+  return {
+    stimulationAttemptId: a.stimulation_attempt_id,
+    stimulationId: a.stimulation_id,
+    attemptNumber: num(a.attempt_number),
+    status: a.status,
+    startedAt: num(a.started_at),
+    completedAt: numOrNull(a.completed_at),
+    hopCount: num(a.hop_count),
+    hasError: Boolean(a.has_error),
+    replayOf: (_a = a.replay_of) != null ? _a : null,
+    entry: a.entry
+  };
+}
+__name(rowToAttempt, "rowToAttempt");
+function rowToTask(t) {
+  var _a, _b;
+  return {
+    stimulationAttemptId: t.stimulation_attempt_id,
+    index: num(t.index),
+    neuronName: t.neuron_name,
+    dendriteCollateralName: t.dendrite_collateral_name,
+    inputIndex: num(t.input_index),
+    output: (_a = t.output) != null ? _a : null,
+    status: t.status,
+    error: (_b = t.error) != null ? _b : null,
+    startedAt: num(t.started_at),
+    duration: numOrNull(t.duration)
+  };
+}
+__name(rowToTask, "rowToTask");
+function groupBy(rows, key) {
+  const m = /* @__PURE__ */ new Map();
+  for (const r of rows) {
+    const k = key(r);
+    const b = m.get(k);
+    if (b) b.push(r);
+    else m.set(k, [
+      r
+    ]);
+  }
+  return m;
+}
+__name(groupBy, "groupBy");
+export {
+  CNSPostgresStimulationRepository,
+  DEFAULT_PREFIX,
+  assertSafePrefix,
+  ensureSchema,
+  persistPostgresSchemaSql
+};
+//# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/CNSPostgresStimulationRepository.ts"],"sourcesContent":["/**\n * Postgres relational store for the CNStra stimulation/attempt/task durable model\n * (`ICNSStimulationRepository`). The stimulation's `progress` is NORMALISED: the\n * outstanding frontier into `<prefix>stimulation_frontier` (one row per task) and\n * per-neuron context into `<prefix>stimulation_context` — rebuilt into `TCNSProgress`\n * on read.\n *\n * Reuses the `IPgQueryable` seam so `pg` is an optional peer: inject any pg\n * client/pool, or pass a `connectionString` for the repo to own a Pool. Behaviourally\n * matches `CNSInMemoryStimulationRepository` (listStimulations newest-first, getAttempts\n * by attemptNumber, getTasks by index, delete cascades).\n */\nimport { Pool } from 'pg';\nimport type { ICNSStimulationRepository } from '@cnstra/persist';\nimport type {\n    TCNSStimulationPersisted,\n    TCNSStimulationAttemptPersisted,\n    TCNSStimulationTaskPersisted,\n    TCNSProgress,\n    TCNSSerializedTask,\n    TCNSSignalRef,\n    TCNSStimulationStatus,\n    TCNSStimulationTaskStatus,\n} from '@cnstra/persist-dto';\nimport {\n    DEFAULT_PREFIX,\n    assertSafePrefix,\n    ensureSchema,\n    type IPgQueryable,\n} from './migrate';\n\nexport interface TCNSPostgresStimulationRepositoryOptions {\n    connectionString?: string;\n    /** Inject an existing pg Pool (transactions supported; not closed by `close()`). */\n    pool?: Pool;\n    /** Inject any queryable (reads + best-effort non-transactional writes). */\n    queryable?: IPgQueryable;\n    /** Table prefix (default `cns_`). */\n    tablePrefix?: string;\n    /** Lazily ensure the schema on first use (default true). */\n    autoMigrate?: boolean;\n}\n\nconst j = (v: unknown): string => JSON.stringify(v ?? null);\nconst num = (v: unknown): number => Number(v);\nconst numOrNull = (v: unknown): number | null => (v == null ? null : Number(v));\n\nexport class CNSPostgresStimulationRepository\n    implements ICNSStimulationRepository\n{\n    private readonly db: IPgQueryable;\n    private readonly pool?: Pool;\n    private readonly ownsPool: boolean;\n    private readonly p: string;\n    private readonly autoMigrate: boolean;\n    private ready: Promise<void> | undefined;\n\n    constructor(opts: TCNSPostgresStimulationRepositoryOptions) {\n        if (opts.pool) {\n            this.pool = opts.pool;\n            this.db = opts.pool;\n            this.ownsPool = false;\n        } else if (opts.connectionString) {\n            this.pool = new Pool({ connectionString: opts.connectionString });\n            this.db = this.pool;\n            this.ownsPool = true;\n        } else if (opts.queryable) {\n            this.db = opts.queryable;\n            this.ownsPool = false;\n        } else {\n            throw new Error(\n                '[@cnstra/persist-postgres] needs `connectionString`, `pool`, or `queryable`'\n            );\n        }\n        this.p = assertSafePrefix(opts.tablePrefix ?? DEFAULT_PREFIX);\n        this.autoMigrate = opts.autoMigrate ?? true;\n    }\n\n    private ensure(): Promise<void> {\n        if (!this.autoMigrate) return Promise.resolve();\n        return (this.ready ??= ensureSchema(this.db, this.p));\n    }\n\n    /** Run `fn` in a transaction when a Pool is available; else best-effort inline. */\n    private async tx<T>(fn: (db: IPgQueryable) => Promise<T>): Promise<T> {\n        if (!this.pool) return fn(this.db);\n        const client = await this.pool.connect();\n        try {\n            await client.query('BEGIN');\n            const out = await fn(client as unknown as IPgQueryable);\n            await client.query('COMMIT');\n            return out;\n        } catch (e) {\n            await client.query('ROLLBACK');\n            throw e;\n        } finally {\n            client.release();\n        }\n    }\n\n    // ── write ──\n\n    async saveStimulation(s: TCNSStimulationPersisted): Promise<void> {\n        await this.ensure();\n        const { p } = this;\n        await this.tx(async db => {\n            await db.query(\n                `INSERT INTO ${p}stimulation (stimulation_id, entry, status, scope_name, updated_at)\n                 VALUES ($1, $2::jsonb, $3, $4, now())\n                 ON CONFLICT (stimulation_id)\n                 DO UPDATE SET entry = EXCLUDED.entry, status = EXCLUDED.status,\n                    scope_name = EXCLUDED.scope_name, updated_at = now()`,\n                [s.stimulationId, j(s.entry), s.status, s.scopeName ?? null]\n            );\n            // Replace the normalised frontier.\n            await db.query(\n                `DELETE FROM ${p}stimulation_frontier WHERE stimulation_id = $1`,\n                [s.stimulationId]\n            );\n            for (let i = 0; i < s.progress.tasks.length; i++) {\n                const t = s.progress.tasks[i];\n                await db.query(\n                    `INSERT INTO ${p}stimulation_frontier\n                        (stimulation_id, ord, neuron_name, dendrite_collateral_name, input)\n                     VALUES ($1, $2, $3, $4, $5::jsonb)`,\n                    [\n                        s.stimulationId,\n                        i,\n                        t.neuronName,\n                        t.dendriteCollateralName,\n                        t.input ? j(t.input) : null,\n                    ]\n                );\n            }\n            // Replace the normalised context.\n            await db.query(\n                `DELETE FROM ${p}stimulation_context WHERE stimulation_id = $1`,\n                [s.stimulationId]\n            );\n            for (const [neuronName, value] of Object.entries(\n                s.progress.context\n            )) {\n                await db.query(\n                    `INSERT INTO ${p}stimulation_context (stimulation_id, neuron_name, value)\n                     VALUES ($1, $2, $3::jsonb)`,\n                    [s.stimulationId, neuronName, j(value)]\n                );\n            }\n        });\n    }\n\n    async saveAttempt(a: TCNSStimulationAttemptPersisted): Promise<void> {\n        await this.ensure();\n        const { p } = this;\n        await this.db.query(\n            `INSERT INTO ${p}stimulation_attempt\n                (stimulation_attempt_id, stimulation_id, attempt_number, status, started_at,\n                 completed_at, hop_count, has_error, replay_of, entry)\n             VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9,$10::jsonb)\n             ON CONFLICT (stimulation_attempt_id) DO UPDATE SET\n                status = EXCLUDED.status, completed_at = EXCLUDED.completed_at,\n                hop_count = EXCLUDED.hop_count, has_error = EXCLUDED.has_error,\n                replay_of = EXCLUDED.replay_of, entry = EXCLUDED.entry`,\n            [\n                a.stimulationAttemptId,\n                a.stimulationId,\n                a.attemptNumber,\n                a.status,\n                a.startedAt,\n                a.completedAt,\n                a.hopCount,\n                a.hasError,\n                a.replayOf,\n                j(a.entry),\n            ]\n        );\n    }\n\n    async appendTask(t: TCNSStimulationTaskPersisted): Promise<void> {\n        await this.ensure();\n        const { p } = this;\n        await this.db.query(\n            `INSERT INTO ${p}task\n                (stimulation_attempt_id, \"index\", neuron_name, dendrite_collateral_name,\n                 input_index, output, status, error, started_at, duration)\n             VALUES ($1,$2,$3,$4,$5,$6::jsonb,$7,$8,$9,$10)\n             ON CONFLICT (stimulation_attempt_id, \"index\") DO NOTHING`,\n            [\n                t.stimulationAttemptId,\n                t.index,\n                t.neuronName,\n                t.dendriteCollateralName,\n                t.inputIndex,\n                t.output ? j(t.output) : null,\n                t.status,\n                t.error,\n                t.startedAt,\n                t.duration,\n            ]\n        );\n    }\n\n    // ── read ──\n\n    async getStimulation(\n        stimulationId: string\n    ): Promise<TCNSStimulationPersisted | undefined> {\n        await this.ensure();\n        const { p } = this;\n        const stimRes = await this.db.query(\n            `SELECT entry, status, scope_name FROM ${p}stimulation WHERE stimulation_id = $1`,\n            [stimulationId]\n        );\n        const row = stimRes.rows[0] as\n            | {\n                  entry: TCNSSignalRef;\n                  status: TCNSStimulationStatus;\n                  scope_name: string | null;\n              }\n            | undefined;\n        if (!row) return undefined;\n        const [frontier, context] = await Promise.all([\n            this.db.query(\n                `SELECT ord, neuron_name, dendrite_collateral_name, input\n                 FROM ${p}stimulation_frontier WHERE stimulation_id = $1 ORDER BY ord`,\n                [stimulationId]\n            ),\n            this.db.query(\n                `SELECT neuron_name, value FROM ${p}stimulation_context WHERE stimulation_id = $1`,\n                [stimulationId]\n            ),\n        ]);\n        return {\n            stimulationId,\n            entry: row.entry,\n            status: row.status,\n            progress: this.buildProgress(\n                frontier.rows as TFrontierRow[],\n                context.rows as TContextRow[]\n            ),\n            ...(row.scope_name != null ? { scopeName: row.scope_name } : {}),\n        };\n    }\n\n    async listStimulations(filter?: {\n        scopeName?: string;\n    }): Promise<TCNSStimulationPersisted[]> {\n        await this.ensure();\n        const { p } = this;\n        const scope = filter?.scopeName ?? null;\n        const stimsRes = await this.db.query(\n            `SELECT stimulation_id, entry, status, scope_name FROM ${p}stimulation\n             WHERE ($1::text IS NULL OR scope_name = $1)\n             ORDER BY updated_at DESC`,\n            [scope]\n        );\n        const stims = stimsRes.rows as TStimulationRow[];\n        if (stims.length === 0) return [];\n        const ids = stims.map(r => r.stimulation_id);\n        const [frontierRes, contextRes] = await Promise.all([\n            this.db.query(\n                `SELECT stimulation_id, ord, neuron_name, dendrite_collateral_name, input\n                 FROM ${p}stimulation_frontier WHERE stimulation_id = ANY($1) ORDER BY stimulation_id, ord`,\n                [ids]\n            ),\n            this.db.query(\n                `SELECT stimulation_id, neuron_name, value\n                 FROM ${p}stimulation_context WHERE stimulation_id = ANY($1)`,\n                [ids]\n            ),\n        ]);\n        const frontierBy = groupBy(\n            frontierRes.rows as TFrontierRow[],\n            r => r.stimulation_id\n        );\n        const contextBy = groupBy(\n            contextRes.rows as TContextRow[],\n            r => r.stimulation_id\n        );\n        return stims.map(r => ({\n            stimulationId: r.stimulation_id,\n            entry: r.entry,\n            status: r.status,\n            progress: this.buildProgress(\n                frontierBy.get(r.stimulation_id) ?? [],\n                contextBy.get(r.stimulation_id) ?? []\n            ),\n            ...(r.scope_name != null ? { scopeName: r.scope_name } : {}),\n        }));\n    }\n\n    async getAttempts(\n        stimulationId: string\n    ): Promise<TCNSStimulationAttemptPersisted[]> {\n        await this.ensure();\n        const { p } = this;\n        const res = await this.db.query(\n            `SELECT * FROM ${p}stimulation_attempt WHERE stimulation_id = $1 ORDER BY attempt_number`,\n            [stimulationId]\n        );\n        return (res.rows as TAttemptRow[]).map(rowToAttempt);\n    }\n\n    async getTasks(\n        stimulationAttemptId: string\n    ): Promise<TCNSStimulationTaskPersisted[]> {\n        await this.ensure();\n        const { p } = this;\n        const res = await this.db.query(\n            `SELECT * FROM ${p}task WHERE stimulation_attempt_id = $1 ORDER BY \"index\"`,\n            [stimulationAttemptId]\n        );\n        return (res.rows as TTaskRow[]).map(rowToTask);\n    }\n\n    // ── lifecycle ──\n\n    async delete(stimulationId: string): Promise<void> {\n        await this.ensure();\n        const { p } = this;\n        await this.tx(async db => {\n            // tasks + attempts have no FK (see migrate.ts) → delete them explicitly,\n            // tasks first (they reference attempt ids), then the attempts.\n            await db.query(\n                `DELETE FROM ${p}task WHERE stimulation_attempt_id IN\n                    (SELECT stimulation_attempt_id FROM ${p}stimulation_attempt WHERE stimulation_id = $1)`,\n                [stimulationId]\n            );\n            await db.query(\n                `DELETE FROM ${p}stimulation_attempt WHERE stimulation_id = $1`,\n                [stimulationId]\n            );\n            // FK ON DELETE CASCADE removes frontier/context.\n            await db.query(\n                `DELETE FROM ${p}stimulation WHERE stimulation_id = $1`,\n                [stimulationId]\n            );\n        });\n    }\n\n    async close(): Promise<void> {\n        if (this.ownsPool && this.pool) await this.pool.end();\n    }\n\n    // ── helpers ──\n\n    private buildProgress(\n        frontier: TFrontierRow[],\n        context: TContextRow[]\n    ): TCNSProgress {\n        const tasks: TCNSSerializedTask[] = [...frontier]\n            .sort((a, b) => a.ord - b.ord)\n            .map(f => ({\n                neuronName: f.neuron_name,\n                dendriteCollateralName: f.dendrite_collateral_name,\n                ...(f.input != null ? { input: f.input } : {}),\n            }));\n        const ctx: Record<string, unknown> = {};\n        for (const c of context) ctx[c.neuron_name] = c.value;\n        return { tasks, context: ctx };\n    }\n}\n\n// ── row shapes (jsonb columns come back already parsed by node-pg) ──\ntype TStimulationRow = {\n    stimulation_id: string;\n    entry: TCNSSignalRef;\n    status: TCNSStimulationStatus;\n    scope_name: string | null;\n};\ntype TFrontierRow = {\n    stimulation_id: string;\n    ord: number;\n    neuron_name: string;\n    dendrite_collateral_name: string;\n    input: { collateralName: string; payload?: unknown } | null;\n};\ntype TContextRow = {\n    stimulation_id: string;\n    neuron_name: string;\n    value: unknown;\n};\ntype TAttemptRow = {\n    stimulation_attempt_id: string;\n    stimulation_id: string;\n    attempt_number: number;\n    status: TCNSStimulationStatus;\n    started_at: string | number;\n    completed_at: string | number | null;\n    hop_count: number;\n    has_error: boolean;\n    replay_of: string | null;\n    entry: TCNSSignalRef[];\n};\ntype TTaskRow = {\n    stimulation_attempt_id: string;\n    index: number;\n    neuron_name: string;\n    dendrite_collateral_name: string;\n    input_index: number;\n    output: TCNSSignalRef | null;\n    status: TCNSStimulationTaskStatus;\n    error: string | null;\n    started_at: string | number;\n    duration: number | null;\n};\n\nfunction rowToAttempt(a: TAttemptRow): TCNSStimulationAttemptPersisted {\n    return {\n        stimulationAttemptId: a.stimulation_attempt_id,\n        stimulationId: a.stimulation_id,\n        attemptNumber: num(a.attempt_number),\n        status: a.status,\n        startedAt: num(a.started_at),\n        completedAt: numOrNull(a.completed_at),\n        hopCount: num(a.hop_count),\n        hasError: Boolean(a.has_error),\n        replayOf: a.replay_of ?? null,\n        entry: a.entry,\n    };\n}\n\nfunction rowToTask(t: TTaskRow): TCNSStimulationTaskPersisted {\n    return {\n        stimulationAttemptId: t.stimulation_attempt_id,\n        index: num(t.index),\n        neuronName: t.neuron_name,\n        dendriteCollateralName: t.dendrite_collateral_name,\n        inputIndex: num(t.input_index),\n        output: t.output ?? null,\n        status: t.status,\n        error: t.error ?? null,\n        startedAt: num(t.started_at),\n        duration: numOrNull(t.duration),\n    };\n}\n\nfunction groupBy<T, K>(rows: T[], key: (r: T) => K): Map<K, T[]> {\n    const m = new Map<K, T[]>();\n    for (const r of rows) {\n        const k = key(r);\n        const b = m.get(k);\n        if (b) b.push(r);\n        else m.set(k, [r]);\n    }\n    return m;\n}\n"],"mappings":";;;;;;;;;;AAYA,SAASA,YAAY;AA+BrB,IAAMC,IAAI,wBAACC,MAAuBC,KAAKC,UAAUF,gBAAK,IAAA,GAA5C;AACV,IAAMG,MAAM,wBAACH,MAAuBI,OAAOJ,CAAAA,GAA/B;AACZ,IAAMK,YAAY,wBAACL,MAA+BA,KAAK,OAAO,OAAOI,OAAOJ,CAAAA,GAA1D;AAEX,IAAMM,oCAAN,MAAMA,kCAAAA;EAUT,YAAYC,MAAgD;AAP3CC;AACAC;AACAC;AACAC;AACAC;AACTC;AAvDZ;AA0DQ,QAAIN,KAAKE,MAAM;AACX,WAAKA,OAAOF,KAAKE;AACjB,WAAKD,KAAKD,KAAKE;AACf,WAAKC,WAAW;IACpB,WAAWH,KAAKO,kBAAkB;AAC9B,WAAKL,OAAO,IAAIM,KAAK;QAAED,kBAAkBP,KAAKO;MAAiB,CAAA;AAC/D,WAAKN,KAAK,KAAKC;AACf,WAAKC,WAAW;IACpB,WAAWH,KAAKS,WAAW;AACvB,WAAKR,KAAKD,KAAKS;AACf,WAAKN,WAAW;IACpB,OAAO;AACH,YAAM,IAAIO,MACN,6EAAA;IAER;AACA,SAAKN,IAAIO,kBAAiBX,UAAKY,gBAALZ,YAAoBa,cAAAA;AAC9C,SAAKR,eAAcL,UAAKK,gBAALL,YAAoB;EAC3C;EAEQc,SAAwB;AA9EpC;AA+EQ,QAAI,CAAC,KAAKT,YAAa,QAAOU,QAAQC,QAAO;AAC7C,YAAQ,UAAKV,UAAL,iBAAKA,QAAUW,aAAa,KAAKhB,IAAI,KAAKG,CAAC;EACvD;;EAGA,MAAcc,GAAMC,IAAkD;AAClE,QAAI,CAAC,KAAKjB,KAAM,QAAOiB,GAAG,KAAKlB,EAAE;AACjC,UAAMmB,SAAS,MAAM,KAAKlB,KAAKmB,QAAO;AACtC,QAAI;AACA,YAAMD,OAAOE,MAAM,OAAA;AACnB,YAAMC,MAAM,MAAMJ,GAAGC,MAAAA;AACrB,YAAMA,OAAOE,MAAM,QAAA;AACnB,aAAOC;IACX,SAASC,GAAG;AACR,YAAMJ,OAAOE,MAAM,UAAA;AACnB,YAAME;IACV,UAAA;AACIJ,aAAOK,QAAO;IAClB;EACJ;;EAIA,MAAMC,gBAAgBC,GAA4C;AAC9D,UAAM,KAAKb,OAAM;AACjB,UAAM,EAAEV,EAAC,IAAK;AACd,UAAM,KAAKc,GAAG,OAAMjB,OAAAA;AAzG5B;AA0GY,YAAMA,GAAGqB,MACL,eAAelB,CAAAA;;;;2EAKf;QAACuB,EAAEC;QAAepC,EAAEmC,EAAEE,KAAK;QAAGF,EAAEG;SAAQH,OAAEI,cAAFJ,YAAe;OAAK;AAGhE,YAAM1B,GAAGqB,MACL,eAAelB,CAAAA,kDACf;QAACuB,EAAEC;OAAc;AAErB,eAASI,IAAI,GAAGA,IAAIL,EAAEM,SAASC,MAAMC,QAAQH,KAAK;AAC9C,cAAMI,IAAIT,EAAEM,SAASC,MAAMF,CAAAA;AAC3B,cAAM/B,GAAGqB,MACL,eAAelB,CAAAA;;0DAGf;UACIuB,EAAEC;UACFI;UACAI,EAAEC;UACFD,EAAEE;UACFF,EAAEG,QAAQ/C,EAAE4C,EAAEG,KAAK,IAAI;SAC1B;MAET;AAEA,YAAMtC,GAAGqB,MACL,eAAelB,CAAAA,iDACf;QAACuB,EAAEC;OAAc;AAErB,iBAAW,CAACS,YAAYG,KAAAA,KAAUC,OAAOC,QACrCf,EAAEM,SAASU,OAAO,GACnB;AACC,cAAM1C,GAAGqB,MACL,eAAelB,CAAAA;kDAEf;UAACuB,EAAEC;UAAeS;UAAY7C,EAAEgD,KAAAA;SAAO;MAE/C;IACJ,CAAA;EACJ;EAEA,MAAMI,YAAYC,GAAmD;AACjE,UAAM,KAAK/B,OAAM;AACjB,UAAM,EAAEV,EAAC,IAAK;AACd,UAAM,KAAKH,GAAGqB,MACV,eAAelB,CAAAA;;;;;;;yEAQf;MACIyC,EAAEC;MACFD,EAAEjB;MACFiB,EAAEE;MACFF,EAAEf;MACFe,EAAEG;MACFH,EAAEI;MACFJ,EAAEK;MACFL,EAAEM;MACFN,EAAEO;MACF5D,EAAEqD,EAAEhB,KAAK;KACZ;EAET;EAEA,MAAMwB,WAAWjB,GAAgD;AAC7D,UAAM,KAAKtB,OAAM;AACjB,UAAM,EAAEV,EAAC,IAAK;AACd,UAAM,KAAKH,GAAGqB,MACV,eAAelB,CAAAA;;;;wEAKf;MACIgC,EAAEU;MACFV,EAAEkB;MACFlB,EAAEC;MACFD,EAAEE;MACFF,EAAEmB;MACFnB,EAAEoB,SAAShE,EAAE4C,EAAEoB,MAAM,IAAI;MACzBpB,EAAEN;MACFM,EAAEqB;MACFrB,EAAEY;MACFZ,EAAEsB;KACL;EAET;;EAIA,MAAMC,eACF/B,eAC6C;AAC7C,UAAM,KAAKd,OAAM;AACjB,UAAM,EAAEV,EAAC,IAAK;AACd,UAAMwD,UAAU,MAAM,KAAK3D,GAAGqB,MAC1B,yCAAyClB,CAAAA,yCACzC;MAACwB;KAAc;AAEnB,UAAMiC,MAAMD,QAAQE,KAAK,CAAA;AAOzB,QAAI,CAACD,IAAK,QAAOE;AACjB,UAAM,CAACC,UAAUrB,OAAAA,IAAW,MAAM5B,QAAQkD,IAAI;MAC1C,KAAKhE,GAAGqB,MACJ;wBACQlB,CAAAA,+DACR;QAACwB;OAAc;MAEnB,KAAK3B,GAAGqB,MACJ,kCAAkClB,CAAAA,iDAClC;QAACwB;OAAc;KAEtB;AACD,WAAO;MACHA;MACAC,OAAOgC,IAAIhC;MACXC,QAAQ+B,IAAI/B;MACZG,UAAU,KAAKiC,cACXF,SAASF,MACTnB,QAAQmB,IAAI;MAEhB,GAAID,IAAIM,cAAc,OAAO;QAAEpC,WAAW8B,IAAIM;MAAW,IAAI,CAAC;IAClE;EACJ;EAEA,MAAMC,iBAAiBC,QAEiB;AAtP5C;AAuPQ,UAAM,KAAKvD,OAAM;AACjB,UAAM,EAAEV,EAAC,IAAK;AACd,UAAMkE,SAAQD,sCAAQtC,cAARsC,YAAqB;AACnC,UAAME,WAAW,MAAM,KAAKtE,GAAGqB,MAC3B,yDAAyDlB,CAAAA;;wCAGzD;MAACkE;KAAM;AAEX,UAAME,QAAQD,SAAST;AACvB,QAAIU,MAAMrC,WAAW,EAAG,QAAO,CAAA;AAC/B,UAAMsC,MAAMD,MAAME,IAAIC,CAAAA,MAAKA,EAAEC,cAAc;AAC3C,UAAM,CAACC,aAAaC,UAAAA,IAAc,MAAM/D,QAAQkD,IAAI;MAChD,KAAKhE,GAAGqB,MACJ;wBACQlB,CAAAA,oFACR;QAACqE;OAAI;MAET,KAAKxE,GAAGqB,MACJ;wBACQlB,CAAAA,sDACR;QAACqE;OAAI;KAEZ;AACD,UAAMM,aAAaC,QACfH,YAAYf,MACZa,CAAAA,MAAKA,EAAEC,cAAc;AAEzB,UAAMK,YAAYD,QACdF,WAAWhB,MACXa,CAAAA,MAAKA,EAAEC,cAAc;AAEzB,WAAOJ,MAAME,IAAIC,CAAAA,MAAAA;AAvRzB,UAAAO,KAAA;AAuR+B;QACnBtD,eAAe+C,EAAEC;QACjB/C,OAAO8C,EAAE9C;QACTC,QAAQ6C,EAAE7C;QACVG,UAAU,KAAKiC,eACXa,MAAAA,WAAWI,IAAIR,EAAEC,cAAc,MAA/BG,OAAAA,MAAoC,CAAA,IACpCE,eAAUE,IAAIR,EAAEC,cAAc,MAA9BK,YAAmC,CAAA,CAAE;QAEzC,GAAIN,EAAER,cAAc,OAAO;UAAEpC,WAAW4C,EAAER;QAAW,IAAI,CAAC;MAC9D;KAAA;EACJ;EAEA,MAAMiB,YACFxD,eAC0C;AAC1C,UAAM,KAAKd,OAAM;AACjB,UAAM,EAAEV,EAAC,IAAK;AACd,UAAMiF,MAAM,MAAM,KAAKpF,GAAGqB,MACtB,iBAAiBlB,CAAAA,yEACjB;MAACwB;KAAc;AAEnB,WAAQyD,IAAIvB,KAAuBY,IAAIY,YAAAA;EAC3C;EAEA,MAAMC,SACFzC,sBACuC;AACvC,UAAM,KAAKhC,OAAM;AACjB,UAAM,EAAEV,EAAC,IAAK;AACd,UAAMiF,MAAM,MAAM,KAAKpF,GAAGqB,MACtB,iBAAiBlB,CAAAA,2DACjB;MAAC0C;KAAqB;AAE1B,WAAQuC,IAAIvB,KAAoBY,IAAIc,SAAAA;EACxC;;EAIA,MAAMC,OAAO7D,eAAsC;AAC/C,UAAM,KAAKd,OAAM;AACjB,UAAM,EAAEV,EAAC,IAAK;AACd,UAAM,KAAKc,GAAG,OAAMjB,OAAAA;AAGhB,YAAMA,GAAGqB,MACL,eAAelB,CAAAA;0DAC2BA,CAAAA,kDAC1C;QAACwB;OAAc;AAEnB,YAAM3B,GAAGqB,MACL,eAAelB,CAAAA,iDACf;QAACwB;OAAc;AAGnB,YAAM3B,GAAGqB,MACL,eAAelB,CAAAA,yCACf;QAACwB;OAAc;IAEvB,CAAA;EACJ;EAEA,MAAM8D,QAAuB;AACzB,QAAI,KAAKvF,YAAY,KAAKD,KAAM,OAAM,KAAKA,KAAKyF,IAAG;EACvD;;EAIQzB,cACJF,UACArB,SACY;AACZ,UAAMT,QAA8B;SAAI8B;MACnC4B,KAAK,CAAC/C,GAAGgD,MAAMhD,EAAEiD,MAAMD,EAAEC,GAAG,EAC5BpB,IAAIqB,CAAAA,OAAM;MACP1D,YAAY0D,EAAEC;MACd1D,wBAAwByD,EAAEE;MAC1B,GAAIF,EAAExD,SAAS,OAAO;QAAEA,OAAOwD,EAAExD;MAAM,IAAI,CAAC;IAChD,EAAA;AACJ,UAAM2D,MAA+B,CAAC;AACtC,eAAWC,KAAKxD,QAASuD,KAAIC,EAAEH,WAAW,IAAIG,EAAE3D;AAChD,WAAO;MAAEN;MAAOS,SAASuD;IAAI;EACjC;AACJ;AA1TanG;AAAN,IAAMA,mCAAN;AAwWP,SAASuF,aAAazC,GAAc;AAvZpC;AAwZI,SAAO;IACHC,sBAAsBD,EAAEuD;IACxBxE,eAAeiB,EAAE+B;IACjB7B,eAAenD,IAAIiD,EAAEwD,cAAc;IACnCvE,QAAQe,EAAEf;IACVkB,WAAWpD,IAAIiD,EAAEyD,UAAU;IAC3BrD,aAAanD,UAAU+C,EAAE0D,YAAY;IACrCrD,UAAUtD,IAAIiD,EAAE2D,SAAS;IACzBrD,UAAUsD,QAAQ5D,EAAE6D,SAAS;IAC7BtD,WAAUP,OAAE8D,cAAF9D,YAAe;IACzBhB,OAAOgB,EAAEhB;EACb;AACJ;AAbSyD;AAeT,SAASE,UAAUpD,GAAW;AAta9B;AAuaI,SAAO;IACHU,sBAAsBV,EAAEgE;IACxB9C,OAAO1D,IAAIwC,EAAEkB,KAAK;IAClBjB,YAAYD,EAAE4D;IACd1D,wBAAwBF,EAAE6D;IAC1B1C,YAAY3D,IAAIwC,EAAEwE,WAAW;IAC7BpD,SAAQpB,OAAEoB,WAAFpB,YAAY;IACpBN,QAAQM,EAAEN;IACV2B,QAAOrB,OAAEqB,UAAFrB,YAAW;IAClBY,WAAWpD,IAAIwC,EAAEkE,UAAU;IAC3B5C,UAAU5D,UAAUsC,EAAEsB,QAAQ;EAClC;AACJ;AAbS8B;AAeT,SAASR,QAAclB,MAAW+C,KAAgB;AAC9C,QAAMC,IAAI,oBAAIC,IAAAA;AACd,aAAWpC,KAAKb,MAAM;AAClB,UAAMkD,IAAIH,IAAIlC,CAAAA;AACd,UAAMkB,IAAIiB,EAAE3B,IAAI6B,CAAAA;AAChB,QAAInB,EAAGA,GAAEoB,KAAKtC,CAAAA;QACTmC,GAAEI,IAAIF,GAAG;MAACrC;KAAE;EACrB;AACA,SAAOmC;AACX;AATS9B;","names":["Pool","j","v","JSON","stringify","num","Number","numOrNull","CNSPostgresStimulationRepository","opts","db","pool","ownsPool","p","autoMigrate","ready","connectionString","Pool","queryable","Error","assertSafePrefix","tablePrefix","DEFAULT_PREFIX","ensure","Promise","resolve","ensureSchema","tx","fn","client","connect","query","out","e","release","saveStimulation","s","stimulationId","entry","status","scopeName","i","progress","tasks","length","t","neuronName","dendriteCollateralName","input","value","Object","entries","context","saveAttempt","a","stimulationAttemptId","attemptNumber","startedAt","completedAt","hopCount","hasError","replayOf","appendTask","index","inputIndex","output","error","duration","getStimulation","stimRes","row","rows","undefined","frontier","all","buildProgress","scope_name","listStimulations","filter","scope","stimsRes","stims","ids","map","r","stimulation_id","frontierRes","contextRes","frontierBy","groupBy","contextBy","_a","get","getAttempts","res","rowToAttempt","getTasks","rowToTask","delete","close","end","sort","b","ord","f","neuron_name","dendrite_collateral_name","ctx","c","stimulation_attempt_id","attempt_number","started_at","completed_at","hop_count","Boolean","has_error","replay_of","input_index","key","m","Map","k","push","set"]}

package/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "@cnstra/persist-postgres",
+  "version": "0.1.0",
+  "description": "Postgres relational store for CNStra durable execution (ICNSStimulationRepository: run/attempt/task) — owns its schema + migrations",
+  "keywords": ["cnstra", "postgres", "durable", "run", "attempt", "task", "persist"],
+  "main": "dist/index.js",
+  "module": "dist/index.mjs",
+  "types": "dist/index.d.ts",
+  "exports": { ".": { "types": "./dist/index.d.ts", "import": "./dist/index.mjs", "require": "./dist/index.js" } },
+  "bin": { "cnstra-persist-postgres": "./dist/cli.js" },
+  "sideEffects": false,
+  "files": ["dist"],
+  "scripts": {
+    "build": "tsup",
+    "test": "jest",
+    "test:types": "tsc --noEmit",
+    "clean": "rm -rf dist"
+  },
+  "peerDependencies": {
+    "@cnstra/persist": "^0.1.0",
+    "@cnstra/persist-dto": "^0.1.0",
+    "pg": ">=8"
+  },
+  "peerDependenciesMeta": { "pg": { "optional": true } },
+  "devDependencies": {
+    "@cnstra/persist": "^0.1.0",
+    "@cnstra/persist-dto": "^0.1.0",
+    "@types/jest": "29.5.12",
+    "@types/node": "^20.0.0",
+    "@types/pg": "^8.11.0",
+    "jest": "29.7.0",
+    "pg": "^8.11.0",
+    "ts-jest": "29.1.2",
+    "tsup": "8.5.1",
+    "typescript": "5.7.2"
+  },
+  "author": "Andrei Baikov <abaikov23@gmail.com>",
+  "license": "MIT",
+  "publishConfig": { "access": "public" }
+}