@rei-standard/amsg-server 2.6.0-next.8 → 3.0.0

package/dist/chunk-TKDISKCF.cjs

@@ -0,0 +1,163 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }// src/server/adapters/pg-shared.js
+async function claimTask(query, taskId, expectedNextSendAt, leaseUntil, serializeGroup = null) {
+  const grouped = typeof serializeGroup === "string" && serializeGroup.length > 0;
+  const params = [leaseUntil, taskId, expectedNextSendAt];
+  let setClause = "lease_until = $1, updated_at = NOW()";
+  let groupGuard = "";
+  if (grouped) {
+    params.push(serializeGroup);
+    setClause = "lease_until = $1, serialize_group = $4, updated_at = NOW()";
+    groupGuard = `
+        AND NOT EXISTS (
+          SELECT 1 FROM scheduled_messages busy
+           WHERE busy.serialize_group = $4 AND busy.id <> $2
+             AND busy.status = 'pending' AND busy.lease_until > NOW()
+        )`;
+  }
+  const rows = await query(
+    `UPDATE scheduled_messages
+        SET ${setClause}
+      WHERE id = $2 AND status = 'pending'
+        AND date_trunc('milliseconds', next_send_at)
+          = date_trunc('milliseconds', $3::timestamptz)
+        AND (lease_until IS NULL OR lease_until <= NOW())${groupGuard}
+     RETURNING id`,
+    params
+  );
+  if (rows.length === 0) return false;
+  if (!grouped) return true;
+  const conflict = await query(
+    `SELECT 1 FROM scheduled_messages busy
+      WHERE busy.serialize_group = $1 AND busy.id <> $2
+        AND busy.status = 'pending' AND busy.lease_until > NOW()
+      LIMIT 1`,
+    [serializeGroup, taskId]
+  );
+  if (conflict.length === 0) return true;
+  await query(
+    "UPDATE scheduled_messages SET lease_until = NULL, updated_at = NOW() WHERE id = $1",
+    [taskId]
+  );
+  return false;
+}
+async function renewTaskLease(query, taskId, leaseUntil) {
+  const rows = await query(
+    `UPDATE scheduled_messages SET lease_until = $1
+      WHERE id = $2 AND status = 'pending' AND lease_until IS NOT NULL
+     RETURNING id`,
+    [leaseUntil, taskId]
+  );
+  return rows.length > 0;
+}
+async function createTaskSuperseding(query, params, supersedesUuid) {
+  const rows = await query(
+    `WITH deleted AS (
+       DELETE FROM scheduled_messages
+        WHERE uuid = $6 AND user_id = $1
+       RETURNING id
+     ), inserted AS (
+       INSERT INTO scheduled_messages
+         (user_id, uuid, encrypted_payload, next_send_at, message_type, status, retry_count, created_at, updated_at)
+       VALUES ($1, $2, $3, $4, $5, 'pending', 0, NOW(), NOW())
+       RETURNING id, uuid, next_send_at, status, created_at
+     )
+     SELECT inserted.*, (SELECT count(*) FROM deleted) > 0 AS superseded
+       FROM inserted`,
+    [
+      params.user_id,
+      params.uuid,
+      params.encrypted_payload,
+      params.next_send_at,
+      params.message_type,
+      supersedesUuid
+    ]
+  );
+  const row = rows[0];
+  if (!row) return null;
+  return { ...row, superseded: row.superseded === true };
+}
+async function getTaskStatusInfo(query, uuid, userId) {
+  const rows = await query(
+    "SELECT status, last_error FROM scheduled_messages WHERE uuid = $1 AND user_id = $2 LIMIT 1",
+    [uuid, userId]
+  );
+  if (rows.length === 0) return null;
+  return { status: rows[0].status, last_error: _nullishCoalesce(rows[0].last_error, () => ( null)) };
+}
+async function getPushSubscription(query, userId) {
+  const rows = await query(
+    "SELECT subscription, updated_at FROM push_subscriptions WHERE user_id = $1 LIMIT 1",
+    [userId]
+  );
+  if (rows.length === 0) return null;
+  return { subscription: rows[0].subscription, updated_at: Number(rows[0].updated_at) };
+}
+async function upsertPushSubscription(query, userId, encryptedSubscription, updatedAt) {
+  const rows = await query(
+    `INSERT INTO push_subscriptions (user_id, subscription, updated_at)
+     VALUES ($1, $2, $3)
+     ON CONFLICT (user_id) DO UPDATE SET
+       subscription = EXCLUDED.subscription,
+       updated_at = EXCLUDED.updated_at
+     RETURNING user_id`,
+    [userId, encryptedSubscription, updatedAt]
+  );
+  return rows.length > 0;
+}
+async function deletePushSubscription(query, userId) {
+  const rows = await query(
+    "DELETE FROM push_subscriptions WHERE user_id = $1 RETURNING user_id",
+    [userId]
+  );
+  return rows.length > 0;
+}
+async function upsertLlmCredentials(query, userId, entries, now) {
+  let upserted = 0;
+  for (const entry of entries) {
+    const rows = await query(
+      `INSERT INTO llm_credentials (user_id, cred_id, encrypted_value, created_at, updated_at)
+       VALUES ($1, $2, $3, $4, $4)
+       ON CONFLICT (user_id, cred_id) DO UPDATE SET
+         encrypted_value = EXCLUDED.encrypted_value,
+         updated_at = EXCLUDED.updated_at
+       RETURNING cred_id`,
+      [userId, entry.credId, entry.encryptedValue, now]
+    );
+    if (rows.length > 0) upserted++;
+  }
+  return upserted;
+}
+async function getLlmCredentials(query, userId, credIds) {
+  if (!credIds || credIds.length === 0) return [];
+  return query(
+    `SELECT cred_id, encrypted_value, updated_at
+     FROM llm_credentials
+     WHERE user_id = $1 AND cred_id = ANY($2)`,
+    [userId, credIds]
+  );
+}
+async function listLlmCredentials(query, userId) {
+  return query(
+    "SELECT cred_id, updated_at FROM llm_credentials WHERE user_id = $1 ORDER BY cred_id ASC",
+    [userId]
+  );
+}
+async function deleteLlmCredentials(query, userId, credIds = null) {
+  if (credIds !== null && credIds.length === 0) return 0;
+  const rows = credIds === null ? await query("DELETE FROM llm_credentials WHERE user_id = $1 RETURNING cred_id", [userId]) : await query("DELETE FROM llm_credentials WHERE user_id = $1 AND cred_id = ANY($2) RETURNING cred_id", [userId, credIds]);
+  return rows.length;
+}
+
+
+
+
+
+
+
+
+
+
+
+
+
+exports.claimTask = claimTask; exports.renewTaskLease = renewTaskLease; exports.createTaskSuperseding = createTaskSuperseding; exports.getTaskStatusInfo = getTaskStatusInfo; exports.getPushSubscription = getPushSubscription; exports.upsertPushSubscription = upsertPushSubscription; exports.deletePushSubscription = deletePushSubscription; exports.upsertLlmCredentials = upsertLlmCredentials; exports.getLlmCredentials = getLlmCredentials; exports.listLlmCredentials = listLlmCredentials; exports.deleteLlmCredentials = deleteLlmCredentials;

package/dist/chunk-XKUWMYVF.mjs

@@ -0,0 +1,163 @@
+// src/server/adapters/pg-shared.js
+async function claimTask(query, taskId, expectedNextSendAt, leaseUntil, serializeGroup = null) {
+  const grouped = typeof serializeGroup === "string" && serializeGroup.length > 0;
+  const params = [leaseUntil, taskId, expectedNextSendAt];
+  let setClause = "lease_until = $1, updated_at = NOW()";
+  let groupGuard = "";
+  if (grouped) {
+    params.push(serializeGroup);
+    setClause = "lease_until = $1, serialize_group = $4, updated_at = NOW()";
+    groupGuard = `
+        AND NOT EXISTS (
+          SELECT 1 FROM scheduled_messages busy
+           WHERE busy.serialize_group = $4 AND busy.id <> $2
+             AND busy.status = 'pending' AND busy.lease_until > NOW()
+        )`;
+  }
+  const rows = await query(
+    `UPDATE scheduled_messages
+        SET ${setClause}
+      WHERE id = $2 AND status = 'pending'
+        AND date_trunc('milliseconds', next_send_at)
+          = date_trunc('milliseconds', $3::timestamptz)
+        AND (lease_until IS NULL OR lease_until <= NOW())${groupGuard}
+     RETURNING id`,
+    params
+  );
+  if (rows.length === 0) return false;
+  if (!grouped) return true;
+  const conflict = await query(
+    `SELECT 1 FROM scheduled_messages busy
+      WHERE busy.serialize_group = $1 AND busy.id <> $2
+        AND busy.status = 'pending' AND busy.lease_until > NOW()
+      LIMIT 1`,
+    [serializeGroup, taskId]
+  );
+  if (conflict.length === 0) return true;
+  await query(
+    "UPDATE scheduled_messages SET lease_until = NULL, updated_at = NOW() WHERE id = $1",
+    [taskId]
+  );
+  return false;
+}
+async function renewTaskLease(query, taskId, leaseUntil) {
+  const rows = await query(
+    `UPDATE scheduled_messages SET lease_until = $1
+      WHERE id = $2 AND status = 'pending' AND lease_until IS NOT NULL
+     RETURNING id`,
+    [leaseUntil, taskId]
+  );
+  return rows.length > 0;
+}
+async function createTaskSuperseding(query, params, supersedesUuid) {
+  const rows = await query(
+    `WITH deleted AS (
+       DELETE FROM scheduled_messages
+        WHERE uuid = $6 AND user_id = $1
+       RETURNING id
+     ), inserted AS (
+       INSERT INTO scheduled_messages
+         (user_id, uuid, encrypted_payload, next_send_at, message_type, status, retry_count, created_at, updated_at)
+       VALUES ($1, $2, $3, $4, $5, 'pending', 0, NOW(), NOW())
+       RETURNING id, uuid, next_send_at, status, created_at
+     )
+     SELECT inserted.*, (SELECT count(*) FROM deleted) > 0 AS superseded
+       FROM inserted`,
+    [
+      params.user_id,
+      params.uuid,
+      params.encrypted_payload,
+      params.next_send_at,
+      params.message_type,
+      supersedesUuid
+    ]
+  );
+  const row = rows[0];
+  if (!row) return null;
+  return { ...row, superseded: row.superseded === true };
+}
+async function getTaskStatusInfo(query, uuid, userId) {
+  const rows = await query(
+    "SELECT status, last_error FROM scheduled_messages WHERE uuid = $1 AND user_id = $2 LIMIT 1",
+    [uuid, userId]
+  );
+  if (rows.length === 0) return null;
+  return { status: rows[0].status, last_error: rows[0].last_error ?? null };
+}
+async function getPushSubscription(query, userId) {
+  const rows = await query(
+    "SELECT subscription, updated_at FROM push_subscriptions WHERE user_id = $1 LIMIT 1",
+    [userId]
+  );
+  if (rows.length === 0) return null;
+  return { subscription: rows[0].subscription, updated_at: Number(rows[0].updated_at) };
+}
+async function upsertPushSubscription(query, userId, encryptedSubscription, updatedAt) {
+  const rows = await query(
+    `INSERT INTO push_subscriptions (user_id, subscription, updated_at)
+     VALUES ($1, $2, $3)
+     ON CONFLICT (user_id) DO UPDATE SET
+       subscription = EXCLUDED.subscription,
+       updated_at = EXCLUDED.updated_at
+     RETURNING user_id`,
+    [userId, encryptedSubscription, updatedAt]
+  );
+  return rows.length > 0;
+}
+async function deletePushSubscription(query, userId) {
+  const rows = await query(
+    "DELETE FROM push_subscriptions WHERE user_id = $1 RETURNING user_id",
+    [userId]
+  );
+  return rows.length > 0;
+}
+async function upsertLlmCredentials(query, userId, entries, now) {
+  let upserted = 0;
+  for (const entry of entries) {
+    const rows = await query(
+      `INSERT INTO llm_credentials (user_id, cred_id, encrypted_value, created_at, updated_at)
+       VALUES ($1, $2, $3, $4, $4)
+       ON CONFLICT (user_id, cred_id) DO UPDATE SET
+         encrypted_value = EXCLUDED.encrypted_value,
+         updated_at = EXCLUDED.updated_at
+       RETURNING cred_id`,
+      [userId, entry.credId, entry.encryptedValue, now]
+    );
+    if (rows.length > 0) upserted++;
+  }
+  return upserted;
+}
+async function getLlmCredentials(query, userId, credIds) {
+  if (!credIds || credIds.length === 0) return [];
+  return query(
+    `SELECT cred_id, encrypted_value, updated_at
+     FROM llm_credentials
+     WHERE user_id = $1 AND cred_id = ANY($2)`,
+    [userId, credIds]
+  );
+}
+async function listLlmCredentials(query, userId) {
+  return query(
+    "SELECT cred_id, updated_at FROM llm_credentials WHERE user_id = $1 ORDER BY cred_id ASC",
+    [userId]
+  );
+}
+async function deleteLlmCredentials(query, userId, credIds = null) {
+  if (credIds !== null && credIds.length === 0) return 0;
+  const rows = credIds === null ? await query("DELETE FROM llm_credentials WHERE user_id = $1 RETURNING cred_id", [userId]) : await query("DELETE FROM llm_credentials WHERE user_id = $1 AND cred_id = ANY($2) RETURNING cred_id", [userId, credIds]);
+  return rows.length;
+}
+
+export {
+  claimTask,
+  renewTaskLease,
+  createTaskSuperseding,
+  getTaskStatusInfo,
+  getPushSubscription,
+  upsertPushSubscription,
+  deletePushSubscription,
+  upsertLlmCredentials,
+  getLlmCredentials,
+  listLlmCredentials,
+  deleteLlmCredentials
+};