@umec/core 0.1.0-alpha.10 → 0.1.0-alpha.12

package/registry/disclosure/files/src/lib/disclosure-db.ts

@@ -0,0 +1,214 @@
+import type { ShopDatabase } from '@umec/core/db';
+
+export type DisclosureRequestStatus = 'received' | 'disclosed';
+
+export type DisclosureRequestRow = {
+  id: string;
+  requester_email: string;
+  status: DisclosureRequestStatus;
+  received_at: number;
+  disclosed_at: number | null;
+  disclosure_email_id: string | null;
+  send_attempts: number;
+  receipt_email_id: string | null;
+  operator_notified_at: number | null;
+  ip_hash: string | null;
+  user_agent: string | null;
+  created_at: number;
+};
+
+const SELECT_COLUMNS = `id, requester_email, status, received_at, disclosed_at, disclosure_email_id,
+       send_attempts, receipt_email_id, operator_notified_at, ip_hash, user_agent, created_at`;
+
+function asCount(row: { count?: unknown } | null): number {
+  const value = row?.count;
+  if (typeof value === 'number' && Number.isFinite(value)) return value;
+  if (typeof value === 'string' && value.trim() !== '') {
+    const parsed = Number(value);
+    return Number.isFinite(parsed) ? parsed : 0;
+  }
+  return 0;
+}
+
+export async function insertDisclosureRequest(
+  db: ShopDatabase,
+  input: {
+    id: string;
+    requesterEmail: string;
+    receivedAt: number;
+    ipHash: string | null;
+    userAgent: string | null;
+  },
+): Promise<void> {
+  await db
+    .prepare(
+      `INSERT INTO disclosure_requests (
+        id, requester_email, status, received_at, disclosed_at, disclosure_email_id,
+        send_attempts, receipt_email_id, operator_notified_at, ip_hash, user_agent, created_at
+      ) VALUES (?, ?, 'received', ?, NULL, NULL, 0, NULL, NULL, ?, ?, ?)`,
+    )
+    .bind(
+      input.id,
+      input.requesterEmail,
+      input.receivedAt,
+      input.ipHash,
+      input.userAgent,
+      input.receivedAt,
+    )
+    .run();
+}
+
+export async function countRequestsByEmailSince(
+  db: ShopDatabase,
+  email: string,
+  since: number,
+): Promise<number> {
+  const row = await db
+    .prepare(
+      `SELECT COUNT(*) AS count FROM disclosure_requests
+       WHERE requester_email = ? AND received_at >= ?`,
+    )
+    .bind(email, since)
+    .first<{ count: number }>();
+  return asCount(row);
+}
+
+export async function countRequestsByIpSince(
+  db: ShopDatabase,
+  ipHash: string,
+  since: number,
+): Promise<number> {
+  const row = await db
+    .prepare(
+      `SELECT COUNT(*) AS count FROM disclosure_requests
+       WHERE ip_hash = ? AND received_at >= ?`,
+    )
+    .bind(ipHash, since)
+    .first<{ count: number }>();
+  return asCount(row);
+}
+
+export async function countRequestsSince(db: ShopDatabase, since: number): Promise<number> {
+  const row = await db
+    .prepare(`SELECT COUNT(*) AS count FROM disclosure_requests WHERE received_at >= ?`)
+    .bind(since)
+    .first<{ count: number }>();
+  return asCount(row);
+}
+
+export async function markDisclosed(
+  db: ShopDatabase,
+  id: string,
+  input: { disclosedAt: number; disclosureEmailId: string },
+): Promise<void> {
+  await db
+    .prepare(
+      `UPDATE disclosure_requests
+       SET status = 'disclosed', disclosed_at = ?, disclosure_email_id = ?
+       WHERE id = ?`,
+    )
+    .bind(input.disclosedAt, input.disclosureEmailId, id)
+    .run();
+}
+
+export async function incrementSendAttempts(db: ShopDatabase, id: string): Promise<number> {
+  await db
+    .prepare(`UPDATE disclosure_requests SET send_attempts = send_attempts + 1 WHERE id = ?`)
+    .bind(id)
+    .run();
+  const row = await db
+    .prepare(`SELECT send_attempts FROM disclosure_requests WHERE id = ?`)
+    .bind(id)
+    .first<{ send_attempts: number }>();
+  return Number(row?.send_attempts ?? 0);
+}
+
+export async function markOperatorNotified(
+  db: ShopDatabase,
+  id: string,
+  notifiedAt: number,
+): Promise<void> {
+  await db
+    .prepare(`UPDATE disclosure_requests SET operator_notified_at = ? WHERE id = ?`)
+    .bind(notifiedAt, id)
+    .run();
+}
+
+export async function listRetryableUndisclosed(
+  db: ShopDatabase,
+  input: { maxAttempts: number; receivedAtOnOrBefore: number; limit?: number },
+): Promise<DisclosureRequestRow[]> {
+  const limit = input.limit ?? 25;
+  const result = await db
+    .prepare(
+      `SELECT ${SELECT_COLUMNS}
+       FROM disclosure_requests
+       WHERE disclosure_email_id IS NULL
+         AND send_attempts < ?
+         AND received_at <= ?
+       ORDER BY received_at ASC
+       LIMIT ?`,
+    )
+    .bind(input.maxAttempts, input.receivedAtOnOrBefore, limit)
+    .all<DisclosureRequestRow>();
+  return result.results ?? [];
+}
+
+export async function getDisclosureRequest(
+  db: ShopDatabase,
+  id: string,
+): Promise<DisclosureRequestRow | null> {
+  const row = await db
+    .prepare(`SELECT ${SELECT_COLUMNS} FROM disclosure_requests WHERE id = ?`)
+    .bind(id)
+    .first<DisclosureRequestRow>();
+  return row ?? null;
+}
+
+export async function listDisclosureRequests(
+  db: ShopDatabase,
+  input: { limit?: number } = {},
+): Promise<DisclosureRequestRow[]> {
+  const limit = input.limit ?? 100;
+  const result = await db
+    .prepare(
+      `SELECT ${SELECT_COLUMNS}
+       FROM disclosure_requests
+       ORDER BY CASE WHEN disclosure_email_id IS NULL THEN 0 ELSE 1 END,
+                received_at DESC
+       LIMIT ?`,
+    )
+    .bind(limit)
+    .all<DisclosureRequestRow>();
+  return result.results ?? [];
+}
+
+export async function listUndisclosed(
+  db: ShopDatabase,
+  input: { limit?: number } = {},
+): Promise<DisclosureRequestRow[]> {
+  const limit = input.limit ?? 100;
+  const result = await db
+    .prepare(
+      `SELECT ${SELECT_COLUMNS}
+       FROM disclosure_requests
+       WHERE disclosure_email_id IS NULL
+       ORDER BY received_at ASC
+       LIMIT ?`,
+    )
+    .bind(limit)
+    .all<DisclosureRequestRow>();
+  return result.results ?? [];
+}
+
+export async function listUndisclosedForReminder(db: ShopDatabase): Promise<DisclosureRequestRow[]> {
+  const result = await db
+    .prepare(
+      `SELECT ${SELECT_COLUMNS}
+       FROM disclosure_requests
+       WHERE disclosure_email_id IS NULL
+       ORDER BY received_at ASC`,
+    )
+    .all<DisclosureRequestRow>();
+  return result.results ?? [];
+}