@fitzzero/quickdraw-core 3.6.0 → 3.8.0

package/dist/server/testing-prisma.d.ts

@@ -0,0 +1,101 @@
+/**
+ * Prisma test-database helpers (dual-mode: real PostgreSQL or PGlite).
+ *
+ * Exported as `@fitzzero/quickdraw-core/server/testing/prisma`.
+ *
+ * Real-PostgreSQL mode: a template database is migrated once per run, then
+ * per-worker databases are cloned from it (`CREATE DATABASE … TEMPLATE …`) so
+ * vitest workers never share state. PGlite mode (no TEST_DATABASE_URL):
+ * migrations run once into an in-memory PGlite instance whose data dir is
+ * dumped to a gzip cache keyed by a migrations fingerprint — later test
+ * processes boot from the dump in milliseconds with no PostgreSQL at all.
+ *
+ * `pg` and `@electric-sql/pglite` are optional peers, loaded via dynamic
+ * import only inside the mode that needs them. App-specific pieces (the
+ * generated PrismaClient, adapter choice, seed helpers) stay in the app.
+ */
+interface RawSqlClient {
+    $executeRawUnsafe(sql: string): Promise<unknown>;
+}
+/**
+ * Reset a test database by truncating all public tables.
+ *
+ * Dynamic TRUNCATE discovers all tables so it never falls out of sync with
+ * schema changes. Retries on deadlock (surfaced by Prisma as P2010), which can
+ * occur when parallel test files within the same fork race on TRUNCATE vs
+ * active queries.
+ */
+declare function resetDatabase(client: RawSqlClient): Promise<void>;
+/**
+ * Rewrite a database URL to point at a per-worker database
+ * (e.g. ".../app_test" → ".../test_worker_3").
+ */
+declare function workerDatabaseUrl(baseUrl: string, poolId: string, prefix?: string): string;
+/**
+ * Hash of all migration.sql contents — cache key for the PGlite template.
+ */
+declare function computeMigrationsFingerprint(migrationsDir: string): string;
+/** All migration.sql contents in execution order. */
+declare function readMigrationSql(migrationsDir: string): string[];
+interface PgliteTemplateOptions {
+    /** Absolute path to the Prisma migrations directory. */
+    migrationsDir: string;
+    /** Cache directory for the template dump (e.g. node_modules/.cache). */
+    cacheDir: string;
+    /** Base name for the cache files (e.g. "myapp-test-template"). */
+    templateName: string;
+}
+interface PgliteTemplatePaths {
+    templatePath: string;
+    fingerprintPath: string;
+}
+/** Resolve the cache file locations for a PGlite template. */
+declare function pgliteTemplatePaths(options: PgliteTemplateOptions): PgliteTemplatePaths;
+/**
+ * Build (or reuse) the gzip'd PGlite data-dir template with all migrations
+ * applied. Returns the template path; skips the build when the cached
+ * fingerprint matches the current migrations.
+ */
+declare function buildPgliteTemplate(options: PgliteTemplateOptions): Promise<{
+    templatePath: string;
+    rebuilt: boolean;
+}>;
+interface PostgresWorkerOptions {
+    /** Connection URL whose database segment names the template database. */
+    testDatabaseUrl: string;
+    /** Template database name (e.g. "myapp_test"). */
+    templateDbName: string;
+    /** Absolute path to the Prisma migrations directory. */
+    migrationsDir: string;
+    /** Number of per-worker databases to clone. Default: 8. */
+    workerCount?: number;
+    /** Worker database name prefix. Default: "test_worker_". */
+    workerPrefix?: string;
+    /** Seconds to wait for PostgreSQL to accept connections. Default: 30. */
+    maxWaitSeconds?: number;
+}
+/**
+ * Prepare real-PostgreSQL test databases: wait for the server, ensure the
+ * template database exists, reset + migrate it, then clone one database per
+ * worker from the template.
+ */
+declare function setupPostgresWorkerDatabases(options: PostgresWorkerOptions): Promise<void>;
+interface PrismaTestGlobalSetupOptions {
+    /** Absolute path to the Prisma migrations directory. */
+    migrationsDir: string;
+    /** Template database name for real-PostgreSQL mode. */
+    templateDbName: string;
+    /** Cache name for the PGlite template dump (e.g. "myapp-test-template"). */
+    templateName: string;
+    /** Cache directory. Default: <cwd>/node_modules/.cache. */
+    cacheDir?: string;
+    /** Per-worker database count for real-PostgreSQL mode. Default: 8. */
+    workerCount?: number;
+}
+/**
+ * Build a vitest globalSetup function that prepares the test databases.
+ * Picks real-PostgreSQL mode when TEST_DATABASE_URL is set, PGlite otherwise.
+ */
+declare function createPrismaTestGlobalSetup(options: PrismaTestGlobalSetupOptions): () => Promise<void>;
+
+export { type PgliteTemplateOptions, type PgliteTemplatePaths, type PostgresWorkerOptions, type PrismaTestGlobalSetupOptions, type RawSqlClient, buildPgliteTemplate, computeMigrationsFingerprint, createPrismaTestGlobalSetup, pgliteTemplatePaths, readMigrationSql, resetDatabase, setupPostgresWorkerDatabases, workerDatabaseUrl };

package/dist/server/testing-prisma.js

@@ -0,0 +1,172 @@
+import { createHash } from 'crypto';
+import { existsSync, readFileSync, mkdirSync, writeFileSync, readdirSync } from 'fs';
+import { resolve } from 'path';
+
+// src/server/testing-prisma.ts
+async function resetDatabase(client) {
+  for (let attempt = 0; attempt < 3; attempt++) {
+    try {
+      await client.$executeRawUnsafe(`
+        DO $$
+        DECLARE _tables TEXT;
+        BEGIN
+          SELECT string_agg('"' || tablename || '"', ', ')
+          INTO _tables
+          FROM pg_tables
+          WHERE schemaname = 'public' AND tablename != '_prisma_migrations';
+          IF _tables IS NOT NULL THEN
+            EXECUTE 'TRUNCATE TABLE ' || _tables || ' CASCADE';
+          END IF;
+        END $$
+      `);
+      return;
+    } catch (error) {
+      const isDeadlock = error instanceof Error && "code" in error && error.code === "P2010";
+      if (!isDeadlock || attempt === 2) throw error;
+      await new Promise((r) => {
+        setTimeout(r, 50 * (attempt + 1));
+      });
+    }
+  }
+}
+function workerDatabaseUrl(baseUrl, poolId, prefix = "test_worker_") {
+  return baseUrl.replace(/\/[^/?]+(\?|$)/, `/${prefix}${poolId}$1`);
+}
+function migrationDirNames(migrationsDir) {
+  return readdirSync(migrationsDir, { withFileTypes: true }).filter((d) => d.isDirectory()).map((d) => d.name).sort();
+}
+function computeMigrationsFingerprint(migrationsDir) {
+  const hash = createHash("sha256");
+  for (const dir of migrationDirNames(migrationsDir)) {
+    const sqlPath = resolve(migrationsDir, dir, "migration.sql");
+    if (existsSync(sqlPath)) {
+      hash.update(readFileSync(sqlPath));
+    }
+  }
+  return hash.digest("hex");
+}
+function readMigrationSql(migrationsDir) {
+  const sqls = [];
+  for (const dir of migrationDirNames(migrationsDir)) {
+    const sqlPath = resolve(migrationsDir, dir, "migration.sql");
+    if (existsSync(sqlPath)) {
+      sqls.push(readFileSync(sqlPath, "utf-8"));
+    }
+  }
+  return sqls;
+}
+function pgliteTemplatePaths(options) {
+  return {
+    templatePath: resolve(options.cacheDir, `${options.templateName}.tar.gz`),
+    fingerprintPath: resolve(options.cacheDir, `${options.templateName}.fingerprint`)
+  };
+}
+async function buildPgliteTemplate(options) {
+  const { templatePath, fingerprintPath } = pgliteTemplatePaths(options);
+  const fingerprint = computeMigrationsFingerprint(options.migrationsDir);
+  mkdirSync(options.cacheDir, { recursive: true });
+  if (existsSync(templatePath) && existsSync(fingerprintPath)) {
+    const cached = readFileSync(fingerprintPath, "utf-8").trim();
+    if (cached === fingerprint) {
+      return { templatePath, rebuilt: false };
+    }
+  }
+  const { PGlite } = await import('@electric-sql/pglite');
+  const pg = new PGlite();
+  for (const sql of readMigrationSql(options.migrationsDir)) {
+    await pg.exec(sql);
+  }
+  const dump = await pg.dumpDataDir("gzip");
+  const buffer = Buffer.from(await dump.arrayBuffer());
+  writeFileSync(templatePath, buffer);
+  writeFileSync(fingerprintPath, fingerprint);
+  await pg.close();
+  return { templatePath, rebuilt: true };
+}
+function withDatabase(url, database) {
+  return url.replace(/\/[^/?]+(\?|$)/, `/${database}$1`);
+}
+async function setupPostgresWorkerDatabases(options) {
+  const { default: pg } = await import('pg');
+  const workerCount = options.workerCount ?? 8;
+  const workerPrefix = options.workerPrefix ?? "test_worker_";
+  const maxAttempts = options.maxWaitSeconds ?? 30;
+  const adminUrl = withDatabase(options.testDatabaseUrl, "postgres");
+  const templateUrl = withDatabase(options.testDatabaseUrl, options.templateDbName);
+  for (let i = 0; i < maxAttempts; i++) {
+    try {
+      const client2 = new pg.Client({ connectionString: adminUrl });
+      await client2.connect();
+      await client2.end();
+      break;
+    } catch {
+      if (i === maxAttempts - 1) {
+        throw new Error(`PostgreSQL not ready after ${maxAttempts}s`);
+      }
+      await new Promise((res) => {
+        setTimeout(res, 1e3);
+      });
+    }
+  }
+  {
+    const client2 = new pg.Client({ connectionString: adminUrl });
+    await client2.connect();
+    try {
+      const res = await client2.query("SELECT 1 FROM pg_database WHERE datname = $1", [
+        options.templateDbName
+      ]);
+      if (res.rowCount === 0) {
+        await client2.query(`CREATE DATABASE ${options.templateDbName}`);
+      }
+    } finally {
+      await client2.end();
+    }
+  }
+  const client = new pg.Client({ connectionString: templateUrl });
+  await client.connect();
+  try {
+    await client.query(
+      `SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = $1 AND pid <> pg_backend_pid()`,
+      [options.templateDbName]
+    );
+    await client.query("DROP SCHEMA public CASCADE");
+    await client.query("CREATE SCHEMA public");
+    for (const sql of readMigrationSql(options.migrationsDir)) {
+      await client.query(sql);
+    }
+    for (let i = 0; i < workerCount; i++) {
+      const workerDb = `${workerPrefix}${i}`;
+      await client.query(
+        `SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = $1 AND pid <> pg_backend_pid()`,
+        [workerDb]
+      );
+      await client.query(`DROP DATABASE IF EXISTS ${workerDb}`);
+      await client.query(`CREATE DATABASE ${workerDb} TEMPLATE ${options.templateDbName}`);
+    }
+  } finally {
+    await client.end();
+  }
+}
+function createPrismaTestGlobalSetup(options) {
+  return async () => {
+    const testDatabaseUrl = process.env.TEST_DATABASE_URL;
+    if (testDatabaseUrl) {
+      await setupPostgresWorkerDatabases({
+        testDatabaseUrl,
+        templateDbName: options.templateDbName,
+        migrationsDir: options.migrationsDir,
+        workerCount: options.workerCount
+      });
+    } else {
+      await buildPgliteTemplate({
+        migrationsDir: options.migrationsDir,
+        cacheDir: options.cacheDir ?? resolve(process.cwd(), "node_modules/.cache"),
+        templateName: options.templateName
+      });
+    }
+  };
+}
+
+export { buildPgliteTemplate, computeMigrationsFingerprint, createPrismaTestGlobalSetup, pgliteTemplatePaths, readMigrationSql, resetDatabase, setupPostgresWorkerDatabases, workerDatabaseUrl };
+//# sourceMappingURL=testing-prisma.js.map
+//# sourceMappingURL=testing-prisma.js.map

package/dist/server/testing-prisma.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/server/testing-prisma.ts"],"names":["client"],"mappings":";;;;;AAiCA,eAAsB,cAAc,MAAA,EAAqC;AACvE,EAAA,KAAA,IAAS,OAAA,GAAU,CAAA,EAAG,OAAA,GAAU,CAAA,EAAG,OAAA,EAAA,EAAW;AAC5C,IAAA,IAAI;AACF,MAAA,MAAM,OAAO,iBAAA,CAAkB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,MAAA,CAY9B,CAAA;AACD,MAAA;AAAA,IACF,SAAS,KAAA,EAAgB;AACvB,MAAA,MAAM,aACJ,KAAA,YAAiB,KAAA,IAAS,MAAA,IAAU,KAAA,IAAU,MAA2B,IAAA,KAAS,OAAA;AACpF,MAAA,IAAI,CAAC,UAAA,IAAc,OAAA,KAAY,CAAA,EAAG,MAAM,KAAA;AACxC,MAAA,MAAM,IAAI,OAAA,CAAc,CAAC,CAAA,KAAM;AAC7B,QAAA,UAAA,CAAW,CAAA,EAAG,EAAA,IAAM,OAAA,GAAU,CAAA,CAAE,CAAA;AAAA,MAClC,CAAC,CAAA;AAAA,IACH;AAAA,EACF;AACF;AAMO,SAAS,iBAAA,CACd,OAAA,EACA,MAAA,EACA,MAAA,GAAiB,cAAA,EACT;AACR,EAAA,OAAO,QAAQ,OAAA,CAAQ,gBAAA,EAAkB,IAAI,MAAM,CAAA,EAAG,MAAM,CAAA,EAAA,CAAI,CAAA;AAClE;AAGA,SAAS,kBAAkB,aAAA,EAAiC;AAC1D,EAAA,OAAO,WAAA,CAAY,eAAe,EAAE,aAAA,EAAe,MAAM,CAAA,CACtD,OAAO,CAAC,CAAA,KAAM,EAAE,WAAA,EAAa,EAC7B,GAAA,CAAI,CAAC,MAAM,CAAA,CAAE,IAAI,EACjB,IAAA,EAAK;AACV;AAKO,SAAS,6BAA6B,aAAA,EAA+B;AAC1E,EAAA,MAAM,IAAA,GAAO,WAAW,QAAQ,CAAA;AAChC,EAAA,KAAA,MAAW,GAAA,IAAO,iBAAA,CAAkB,aAAa,CAAA,EAAG;AAClD,IAAA,MAAM,OAAA,GAAU,OAAA,CAAQ,aAAA,EAAe,GAAA,EAAK,eAAe,CAAA;AAC3D,IAAA,IAAI,UAAA,CAAW,OAAO,CAAA,EAAG;AACvB,MAAA,IAAA,CAAK,MAAA,CAAO,YAAA,CAAa,OAAO,CAAC,CAAA;AAAA,IACnC;AAAA,EACF;AACA,EAAA,OAAO,IAAA,CAAK,OAAO,KAAK,CAAA;AAC1B;AAGO,SAAS,iBAAiB,aAAA,EAAiC;AAChE,EAAA,MAAM,OAAiB,EAAC;AACxB,EAAA,KAAA,MAAW,GAAA,IAAO,iBAAA,CAAkB,aAAa,CAAA,EAAG;AAClD,IAAA,MAAM,OAAA,GAAU,OAAA,CAAQ,aAAA,EAAe,GAAA,EAAK,eAAe,CAAA;AAC3D,IAAA,IAAI,UAAA,CAAW,OAAO,CAAA,EAAG;AACvB,MAAA,IAAA,CAAK,IAAA,CAAK,YAAA,CAAa,OAAA,EAAS,OAAO,CAAC,CAAA;AAAA,IAC1C;AAAA,EACF;AACA,EAAA,OAAO,IAAA;AACT;AAiBO,SAAS,oBAAoB,OAAA,EAAqD;AACvF,EAAA,OAAO;AAAA,IACL,cAAc,OAAA,CAAQ,OAAA,CAAQ,UAAU,CAAA,EAAG,OAAA,CAAQ,YAAY,CAAA,OAAA,CAAS,CAAA;AAAA,IACxE,iBAAiB,OAAA,CAAQ,OAAA,CAAQ,UAAU,CAAA,EAAG,OAAA,CAAQ,YAAY,CAAA,YAAA,CAAc;AAAA,GAClF;AACF;AAOA,eAAsB,oBACpB,OAAA,EACqD;AACrD,EAAA,MAAM,EAAE,YAAA,EAAc,eAAA,EAAgB,GAAI,oBAAoB,OAAO,CAAA;AACrE,EAAA,MAAM,WAAA,GAAc,4BAAA,CAA6B,OAAA,CAAQ,aAAa,CAAA;AAEtE,EAAA,SAAA,CAAU,OAAA,CAAQ,QAAA,EAAU,EAAE,SAAA,EAAW,MAAM,CAAA;AAE/C,EAAA,IAAI,UAAA,CAAW,YAAY,CAAA,IAAK,UAAA,CAAW,eAAe,CAAA,EAAG;AAC3D,IAAA,MAAM,MAAA,GAAS,YAAA,CAAa,eAAA,EAAiB,OAAO,EAAE,IAAA,EAAK;AAC3D,IAAA,IAAI,WAAW,WAAA,EAAa;AAC1B,MAAA,OAAO,EAAE,YAAA,EAAc,OAAA,EAAS,KAAA,EAAM;AAAA,IACxC;AAAA,EACF;AAEA,EAAA,MAAM,EAAE,MAAA,EAAO,GAAI,MAAM,OAAO,sBAAsB,CAAA;AACtD,EAAA,MAAM,EAAA,GAAK,IAAI,MAAA,EAAO;AAEtB,EAAA,KAAA,MAAW,GAAA,IAAO,gBAAA,CAAiB,OAAA,CAAQ,aAAa,CAAA,EAAG;AACzD,IAAA,MAAM,EAAA,CAAG,KAAK,GAAG,CAAA;AAAA,EACnB;AAEA,EAAA,MAAM,IAAA,GAAO,MAAM,EAAA,CAAG,WAAA,CAAY,MAAM,CAAA;AACxC,EAAA,MAAM,SAAS,MAAA,CAAO,IAAA,CAAK,MAAM,IAAA,CAAK,aAAa,CAAA;AACnD,EAAA,aAAA,CAAc,cAAc,MAAM,CAAA;AAClC,EAAA,aAAA,CAAc,iBAAiB,WAAW,CAAA;AAE1C,EAAA,MAAM,GAAG,KAAA,EAAM;AACf,EAAA,OAAO,EAAE,YAAA,EAAc,OAAA,EAAS,IAAA,EAAK;AACvC;AAkBA,SAAS,YAAA,CAAa,KAAa,QAAA,EAA0B;AAC3D,EAAA,OAAO,GAAA,CAAI,OAAA,CAAQ,gBAAA,EAAkB,CAAA,CAAA,EAAI,QAAQ,CAAA,EAAA,CAAI,CAAA;AACvD;AAOA,eAAsB,6BAA6B,OAAA,EAA+C;AAChG,EAAA,MAAM,EAAE,OAAA,EAAS,EAAA,EAAG,GAAI,MAAM,OAAO,IAAI,CAAA;AACzC,EAAA,MAAM,WAAA,GAAc,QAAQ,WAAA,IAAe,CAAA;AAC3C,EAAA,MAAM,YAAA,GAAe,QAAQ,YAAA,IAAgB,cAAA;AAC7C,EAAA,MAAM,WAAA,GAAc,QAAQ,cAAA,IAAkB,EAAA;AAC9C,EAAA,MAAM,QAAA,GAAW,YAAA,CAAa,OAAA,CAAQ,eAAA,EAAiB,UAAU,CAAA;AACjE,EAAA,MAAM,WAAA,GAAc,YAAA,CAAa,OAAA,CAAQ,eAAA,EAAiB,QAAQ,cAAc,CAAA;AAIhF,EAAA,KAAA,IAAS,CAAA,GAAI,CAAA,EAAG,CAAA,GAAI,WAAA,EAAa,CAAA,EAAA,EAAK;AACpC,IAAA,IAAI;AACF,MAAA,MAAMA,UAAS,IAAI,EAAA,CAAG,OAAO,EAAE,gBAAA,EAAkB,UAAU,CAAA;AAC3D,MAAA,MAAMA,QAAO,OAAA,EAAQ;AACrB,MAAA,MAAMA,QAAO,GAAA,EAAI;AACjB,MAAA;AAAA,IACF,CAAA,CAAA,MAAQ;AACN,MAAA,IAAI,CAAA,KAAM,cAAc,CAAA,EAAG;AACzB,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,2BAAA,EAA8B,WAAW,CAAA,CAAA,CAAG,CAAA;AAAA,MAC9D;AACA,MAAA,MAAM,IAAI,OAAA,CAAc,CAAC,GAAA,KAAQ;AAC/B,QAAA,UAAA,CAAW,KAAK,GAAI,CAAA;AAAA,MACtB,CAAC,CAAA;AAAA,IACH;AAAA,EACF;AAGA,EAAA;AACE,IAAA,MAAMA,UAAS,IAAI,EAAA,CAAG,OAAO,EAAE,gBAAA,EAAkB,UAAU,CAAA;AAC3D,IAAA,MAAMA,QAAO,OAAA,EAAQ;AACrB,IAAA,IAAI;AACF,MAAA,MAAM,GAAA,GAAM,MAAMA,OAAAA,CAAO,KAAA,CAAM,8CAAA,EAAgD;AAAA,QAC7E,OAAA,CAAQ;AAAA,OACT,CAAA;AACD,MAAA,IAAI,GAAA,CAAI,aAAa,CAAA,EAAG;AACtB,QAAA,MAAMA,OAAAA,CAAO,KAAA,CAAM,CAAA,gBAAA,EAAmB,OAAA,CAAQ,cAAc,CAAA,CAAE,CAAA;AAAA,MAChE;AAAA,IACF,CAAA,SAAE;AACA,MAAA,MAAMA,QAAO,GAAA,EAAI;AAAA,IACnB;AAAA,EACF;AAGA,EAAA,MAAM,SAAS,IAAI,EAAA,CAAG,OAAO,EAAE,gBAAA,EAAkB,aAAa,CAAA;AAC9D,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,IAAI;AACF,IAAA,MAAM,MAAA,CAAO,KAAA;AAAA,MACX,CAAA,qGAAA,CAAA;AAAA,MACA,CAAC,QAAQ,cAAc;AAAA,KACzB;AAEA,IAAA,MAAM,MAAA,CAAO,MAAM,4BAA4B,CAAA;AAC/C,IAAA,MAAM,MAAA,CAAO,MAAM,sBAAsB,CAAA;AAEzC,IAAA,KAAA,MAAW,GAAA,IAAO,gBAAA,CAAiB,OAAA,CAAQ,aAAa,CAAA,EAAG;AACzD,MAAA,MAAM,MAAA,CAAO,MAAM,GAAG,CAAA;AAAA,IACxB;AAEA,IAAA,KAAA,IAAS,CAAA,GAAI,CAAA,EAAG,CAAA,GAAI,WAAA,EAAa,CAAA,EAAA,EAAK;AACpC,MAAA,MAAM,QAAA,GAAW,CAAA,EAAG,YAAY,CAAA,EAAG,CAAC,CAAA,CAAA;AACpC,MAAA,MAAM,MAAA,CAAO,KAAA;AAAA,QACX,CAAA,qGAAA,CAAA;AAAA,QACA,CAAC,QAAQ;AAAA,OACX;AACA,MAAA,MAAM,MAAA,CAAO,KAAA,CAAM,CAAA,wBAAA,EAA2B,QAAQ,CAAA,CAAE,CAAA;AACxD,MAAA,MAAM,OAAO,KAAA,CAAM,CAAA,gBAAA,EAAmB,QAAQ,CAAA,UAAA,EAAa,OAAA,CAAQ,cAAc,CAAA,CAAE,CAAA;AAAA,IACrF;AAAA,EACF,CAAA,SAAE;AACA,IAAA,MAAM,OAAO,GAAA,EAAI;AAAA,EACnB;AACF;AAmBO,SAAS,4BACd,OAAA,EACqB;AACrB,EAAA,OAAO,YAAY;AACjB,IAAA,MAAM,eAAA,GAAkB,QAAQ,GAAA,CAAI,iBAAA;AACpC,IAAA,IAAI,eAAA,EAAiB;AACnB,MAAA,MAAM,4BAAA,CAA6B;AAAA,QACjC,eAAA;AAAA,QACA,gBAAgB,OAAA,CAAQ,cAAA;AAAA,QACxB,eAAe,OAAA,CAAQ,aAAA;AAAA,QACvB,aAAa,OAAA,CAAQ;AAAA,OACtB,CAAA;AAAA,IACH,CAAA,MAAO;AACL,MAAA,MAAM,mBAAA,CAAoB;AAAA,QACxB,eAAe,OAAA,CAAQ,aAAA;AAAA,QACvB,UAAU,OAAA,CAAQ,QAAA,IAAY,QAAQ,OAAA,CAAQ,GAAA,IAAO,qBAAqB,CAAA;AAAA,QAC1E,cAAc,OAAA,CAAQ;AAAA,OACvB,CAAA;AAAA,IACH;AAAA,EACF,CAAA;AACF","file":"testing-prisma.js","sourcesContent":["/**\n * Prisma test-database helpers (dual-mode: real PostgreSQL or PGlite).\n *\n * Exported as `@fitzzero/quickdraw-core/server/testing/prisma`.\n *\n * Real-PostgreSQL mode: a template database is migrated once per run, then\n * per-worker databases are cloned from it (`CREATE DATABASE … TEMPLATE …`) so\n * vitest workers never share state. PGlite mode (no TEST_DATABASE_URL):\n * migrations run once into an in-memory PGlite instance whose data dir is\n * dumped to a gzip cache keyed by a migrations fingerprint — later test\n * processes boot from the dump in milliseconds with no PostgreSQL at all.\n *\n * `pg` and `@electric-sql/pglite` are optional peers, loaded via dynamic\n * import only inside the mode that needs them. App-specific pieces (the\n * generated PrismaClient, adapter choice, seed helpers) stay in the app.\n */\n\nimport { createHash } from \"node:crypto\";\nimport { existsSync, mkdirSync, readFileSync, readdirSync, writeFileSync } from \"node:fs\";\nimport { resolve } from \"node:path\";\n\nexport interface RawSqlClient {\n  $executeRawUnsafe(sql: string): Promise<unknown>;\n}\n\n/**\n * Reset a test database by truncating all public tables.\n *\n * Dynamic TRUNCATE discovers all tables so it never falls out of sync with\n * schema changes. Retries on deadlock (surfaced by Prisma as P2010), which can\n * occur when parallel test files within the same fork race on TRUNCATE vs\n * active queries.\n */\nexport async function resetDatabase(client: RawSqlClient): Promise<void> {\n  for (let attempt = 0; attempt < 3; attempt++) {\n    try {\n      await client.$executeRawUnsafe(`\n        DO $$\n        DECLARE _tables TEXT;\n        BEGIN\n          SELECT string_agg('\"' || tablename || '\"', ', ')\n          INTO _tables\n          FROM pg_tables\n          WHERE schemaname = 'public' AND tablename != '_prisma_migrations';\n          IF _tables IS NOT NULL THEN\n            EXECUTE 'TRUNCATE TABLE ' || _tables || ' CASCADE';\n          END IF;\n        END $$\n      `);\n      return;\n    } catch (error: unknown) {\n      const isDeadlock =\n        error instanceof Error && \"code\" in error && (error as { code: string }).code === \"P2010\";\n      if (!isDeadlock || attempt === 2) throw error;\n      await new Promise<void>((r) => {\n        setTimeout(r, 50 * (attempt + 1));\n      });\n    }\n  }\n}\n\n/**\n * Rewrite a database URL to point at a per-worker database\n * (e.g. \".../app_test\" → \".../test_worker_3\").\n */\nexport function workerDatabaseUrl(\n  baseUrl: string,\n  poolId: string,\n  prefix: string = \"test_worker_\",\n): string {\n  return baseUrl.replace(/\\/[^/?]+(\\?|$)/, `/${prefix}${poolId}$1`);\n}\n\n/** Sorted migration directory names under a Prisma migrations dir. */\nfunction migrationDirNames(migrationsDir: string): string[] {\n  return readdirSync(migrationsDir, { withFileTypes: true })\n    .filter((d) => d.isDirectory())\n    .map((d) => d.name)\n    .sort();\n}\n\n/**\n * Hash of all migration.sql contents — cache key for the PGlite template.\n */\nexport function computeMigrationsFingerprint(migrationsDir: string): string {\n  const hash = createHash(\"sha256\");\n  for (const dir of migrationDirNames(migrationsDir)) {\n    const sqlPath = resolve(migrationsDir, dir, \"migration.sql\");\n    if (existsSync(sqlPath)) {\n      hash.update(readFileSync(sqlPath));\n    }\n  }\n  return hash.digest(\"hex\");\n}\n\n/** All migration.sql contents in execution order. */\nexport function readMigrationSql(migrationsDir: string): string[] {\n  const sqls: string[] = [];\n  for (const dir of migrationDirNames(migrationsDir)) {\n    const sqlPath = resolve(migrationsDir, dir, \"migration.sql\");\n    if (existsSync(sqlPath)) {\n      sqls.push(readFileSync(sqlPath, \"utf-8\"));\n    }\n  }\n  return sqls;\n}\n\nexport interface PgliteTemplateOptions {\n  /** Absolute path to the Prisma migrations directory. */\n  migrationsDir: string;\n  /** Cache directory for the template dump (e.g. node_modules/.cache). */\n  cacheDir: string;\n  /** Base name for the cache files (e.g. \"myapp-test-template\"). */\n  templateName: string;\n}\n\nexport interface PgliteTemplatePaths {\n  templatePath: string;\n  fingerprintPath: string;\n}\n\n/** Resolve the cache file locations for a PGlite template. */\nexport function pgliteTemplatePaths(options: PgliteTemplateOptions): PgliteTemplatePaths {\n  return {\n    templatePath: resolve(options.cacheDir, `${options.templateName}.tar.gz`),\n    fingerprintPath: resolve(options.cacheDir, `${options.templateName}.fingerprint`),\n  };\n}\n\n/**\n * Build (or reuse) the gzip'd PGlite data-dir template with all migrations\n * applied. Returns the template path; skips the build when the cached\n * fingerprint matches the current migrations.\n */\nexport async function buildPgliteTemplate(\n  options: PgliteTemplateOptions,\n): Promise<{ templatePath: string; rebuilt: boolean }> {\n  const { templatePath, fingerprintPath } = pgliteTemplatePaths(options);\n  const fingerprint = computeMigrationsFingerprint(options.migrationsDir);\n\n  mkdirSync(options.cacheDir, { recursive: true });\n\n  if (existsSync(templatePath) && existsSync(fingerprintPath)) {\n    const cached = readFileSync(fingerprintPath, \"utf-8\").trim();\n    if (cached === fingerprint) {\n      return { templatePath, rebuilt: false };\n    }\n  }\n\n  const { PGlite } = await import(\"@electric-sql/pglite\");\n  const pg = new PGlite();\n\n  for (const sql of readMigrationSql(options.migrationsDir)) {\n    await pg.exec(sql);\n  }\n\n  const dump = await pg.dumpDataDir(\"gzip\");\n  const buffer = Buffer.from(await dump.arrayBuffer());\n  writeFileSync(templatePath, buffer);\n  writeFileSync(fingerprintPath, fingerprint);\n\n  await pg.close();\n  return { templatePath, rebuilt: true };\n}\n\nexport interface PostgresWorkerOptions {\n  /** Connection URL whose database segment names the template database. */\n  testDatabaseUrl: string;\n  /** Template database name (e.g. \"myapp_test\"). */\n  templateDbName: string;\n  /** Absolute path to the Prisma migrations directory. */\n  migrationsDir: string;\n  /** Number of per-worker databases to clone. Default: 8. */\n  workerCount?: number;\n  /** Worker database name prefix. Default: \"test_worker_\". */\n  workerPrefix?: string;\n  /** Seconds to wait for PostgreSQL to accept connections. Default: 30. */\n  maxWaitSeconds?: number;\n}\n\n/** Swap the database segment of a connection URL. */\nfunction withDatabase(url: string, database: string): string {\n  return url.replace(/\\/[^/?]+(\\?|$)/, `/${database}$1`);\n}\n\n/**\n * Prepare real-PostgreSQL test databases: wait for the server, ensure the\n * template database exists, reset + migrate it, then clone one database per\n * worker from the template.\n */\nexport async function setupPostgresWorkerDatabases(options: PostgresWorkerOptions): Promise<void> {\n  const { default: pg } = await import(\"pg\");\n  const workerCount = options.workerCount ?? 8;\n  const workerPrefix = options.workerPrefix ?? \"test_worker_\";\n  const maxAttempts = options.maxWaitSeconds ?? 30;\n  const adminUrl = withDatabase(options.testDatabaseUrl, \"postgres\");\n  const templateUrl = withDatabase(options.testDatabaseUrl, options.templateDbName);\n\n  // Wait for PostgreSQL to accept connections — guards against timing races\n  // when tests start before a sidecar container is fully ready.\n  for (let i = 0; i < maxAttempts; i++) {\n    try {\n      const client = new pg.Client({ connectionString: adminUrl });\n      await client.connect();\n      await client.end();\n      break;\n    } catch {\n      if (i === maxAttempts - 1) {\n        throw new Error(`PostgreSQL not ready after ${maxAttempts}s`);\n      }\n      await new Promise<void>((res) => {\n        setTimeout(res, 1000);\n      });\n    }\n  }\n\n  // Ensure the template database exists (self-heals fresh environments).\n  {\n    const client = new pg.Client({ connectionString: adminUrl });\n    await client.connect();\n    try {\n      const res = await client.query(\"SELECT 1 FROM pg_database WHERE datname = $1\", [\n        options.templateDbName,\n      ]);\n      if (res.rowCount === 0) {\n        await client.query(`CREATE DATABASE ${options.templateDbName}`);\n      }\n    } finally {\n      await client.end();\n    }\n  }\n\n  // Reset + migrate the template, then clone per-worker databases from it.\n  const client = new pg.Client({ connectionString: templateUrl });\n  await client.connect();\n  try {\n    await client.query(\n      `SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = $1 AND pid <> pg_backend_pid()`,\n      [options.templateDbName],\n    );\n\n    await client.query(\"DROP SCHEMA public CASCADE\");\n    await client.query(\"CREATE SCHEMA public\");\n\n    for (const sql of readMigrationSql(options.migrationsDir)) {\n      await client.query(sql);\n    }\n\n    for (let i = 0; i < workerCount; i++) {\n      const workerDb = `${workerPrefix}${i}`;\n      await client.query(\n        `SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = $1 AND pid <> pg_backend_pid()`,\n        [workerDb],\n      );\n      await client.query(`DROP DATABASE IF EXISTS ${workerDb}`);\n      await client.query(`CREATE DATABASE ${workerDb} TEMPLATE ${options.templateDbName}`);\n    }\n  } finally {\n    await client.end();\n  }\n}\n\nexport interface PrismaTestGlobalSetupOptions {\n  /** Absolute path to the Prisma migrations directory. */\n  migrationsDir: string;\n  /** Template database name for real-PostgreSQL mode. */\n  templateDbName: string;\n  /** Cache name for the PGlite template dump (e.g. \"myapp-test-template\"). */\n  templateName: string;\n  /** Cache directory. Default: <cwd>/node_modules/.cache. */\n  cacheDir?: string;\n  /** Per-worker database count for real-PostgreSQL mode. Default: 8. */\n  workerCount?: number;\n}\n\n/**\n * Build a vitest globalSetup function that prepares the test databases.\n * Picks real-PostgreSQL mode when TEST_DATABASE_URL is set, PGlite otherwise.\n */\nexport function createPrismaTestGlobalSetup(\n  options: PrismaTestGlobalSetupOptions,\n): () => Promise<void> {\n  return async () => {\n    const testDatabaseUrl = process.env.TEST_DATABASE_URL;\n    if (testDatabaseUrl) {\n      await setupPostgresWorkerDatabases({\n        testDatabaseUrl,\n        templateDbName: options.templateDbName,\n        migrationsDir: options.migrationsDir,\n        workerCount: options.workerCount,\n      });\n    } else {\n      await buildPgliteTemplate({\n        migrationsDir: options.migrationsDir,\n        cacheDir: options.cacheDir ?? resolve(process.cwd(), \"node_modules/.cache\"),\n        templateName: options.templateName,\n      });\n    }\n  };\n}\n"]}

package/dist/server/testing.d.ts

@@ -1,5 +1,5 @@
 import { Socket } from 'socket.io-client';
-import { B as BaseServiceInstance, Q as QuickdrawServerResult } from '../types-DniMxkQs.js';
+import { B as BaseServiceInstance, Q as QuickdrawServerResult } from '../types-ClgSr90l.js';
 import 'socket.io';
 import 'http';
 import 'zod';

package/dist/server/testing.js

@@ -6,6 +6,9 @@ import { Server } from 'socket.io';
 // src/server/testing.ts
 
 // src/shared/types.ts
+function channelEventName(serviceName, channelName) {
+  return `channel:${serviceName}:${channelName}`;
+}
 var consoleLogger = {
   info: (message, meta) => console.log(`[INFO] ${message}`, meta ?? ""),
   warn: (message, meta) => console.warn(`[WARN] ${message}`, meta ?? ""),
@@ -15,6 +18,8 @@ var consoleLogger = {
 };
 
 // src/server/ServiceRegistry.ts
+var CHANNEL_ABUSE_WINDOW_MS = 1e4;
+var CHANNEL_ABUSE_MULTIPLIER = 100;
 var ServiceRegistry = class {
   io;
   services = /* @__PURE__ */ new Map();
@@ -25,6 +30,8 @@ var ServiceRegistry = class {
   connectionHandlerInstalled = false;
   // Store method metadata for deferred registration
   methodRegistry = /* @__PURE__ */ new Map();
+  // Store channel metadata for deferred registration
+  channelRegistry = /* @__PURE__ */ new Map();
   constructor(io, options) {
     this.io = io;
     this.logger = options?.logger?.child({ service: "ServiceRegistry" }) ?? consoleLogger.child({ service: "ServiceRegistry" });
@@ -58,22 +65,13 @@ var ServiceRegistry = class {
     for (const [eventName, { method, service }] of this.methodRegistry) {
       this.registerMethodListener(socket, eventName, method, service);
     }
+    for (const [eventName, { channel, service }] of this.channelRegistry) {
+      this.registerChannelListener(socket, eventName, channel, service);
+    }
     for (const [serviceName, service] of this.services) {
-      this.registerSubscriptionListener(
-        socket,
-        `${serviceName}:subscribe`,
-        service
-      );
-      this.registerBatchSubscriptionListener(
-        socket,
-        `${serviceName}:batchSubscribe`,
-        service
-      );
-      this.registerUnsubscriptionListener(
-        socket,
-        `${serviceName}:unsubscribe`,
-        service
-      );
+      this.registerSubscriptionListener(socket, `${serviceName}:subscribe`, service);
+      this.registerBatchSubscriptionListener(socket, `${serviceName}:batchSubscribe`, service);
+      this.registerUnsubscriptionListener(socket, `${serviceName}:unsubscribe`, service);
     }
   }
   /**
@@ -86,6 +84,7 @@ var ServiceRegistry = class {
     }
     this.logger.info(`Registered service: ${serviceName}`);
     this.discoverServiceMethods(serviceName, service);
+    this.discoverServiceChannels(serviceName, service);
     this.installConnectionHandler();
   }
   /**
@@ -103,6 +102,77 @@ var ServiceRegistry = class {
       });
     }
   }
+  /**
+   * Discover and store all channels from a service.
+   */
+  discoverServiceChannels(serviceName, service) {
+    const channels = service.getPublicChannels?.() ?? [];
+    for (const channel of channels) {
+      const eventName = channelEventName(serviceName, channel.name);
+      this.logger.info(
+        `Registering channel event: ${eventName} (${channel.ratePerSecond}/s, burst ${channel.burst})`
+      );
+      this.channelRegistry.set(eventName, {
+        serviceName,
+        channel,
+        service
+      });
+    }
+  }
+  /**
+   * Register a channel as a socket event listener.
+   *
+   * This is the hot path for high-frequency traffic: no ack, no async access
+   * checks, no per-message logging. Order of operations per message:
+   * token bucket → schema validation → in-memory access check → handler.
+   * Any failure silently drops the message; sustained extreme flooding
+   * disconnects the socket.
+   */
+  registerChannelListener(socket, eventName, channel, service) {
+    let tokens = channel.burst;
+    let lastRefillAt = Date.now();
+    let abuseWindowStart = lastRefillAt;
+    let droppedInWindow = 0;
+    socket.on(eventName, (payload) => {
+      const now = Date.now();
+      tokens = Math.min(
+        channel.burst,
+        tokens + (now - lastRefillAt) / 1e3 * channel.ratePerSecond
+      );
+      lastRefillAt = now;
+      if (tokens < 1) {
+        if (now - abuseWindowStart > CHANNEL_ABUSE_WINDOW_MS) {
+          abuseWindowStart = now;
+          droppedInWindow = 0;
+        }
+        droppedInWindow++;
+        if (droppedInWindow > channel.ratePerSecond * CHANNEL_ABUSE_MULTIPLIER) {
+          this.logger.warn(
+            `Disconnecting socket ${socket.id} for sustained channel flooding on ${eventName}`,
+            { userId: socket.userId, droppedInWindow }
+          );
+          socket.disconnect(true);
+        }
+        return;
+      }
+      tokens -= 1;
+      const parsed = channel.schema.safeParse(payload);
+      if (!parsed.success) return;
+      if (!service.checkChannelAccess?.(channel, socket, parsed.data)) return;
+      try {
+        channel.handler(parsed.data, {
+          userId: socket.userId,
+          socketId: socket.id,
+          serviceAccess: socket.serviceAccess ?? {}
+        });
+      } catch (error) {
+        this.logger.error(`Channel handler error for ${eventName}`, {
+          userId: socket.userId,
+          error: error instanceof Error ? error.message : "Unknown error"
+        });
+      }
+    });
+  }
   /**
    * Register a method as a socket event listener.
    */
@@ -206,10 +276,7 @@ var ServiceRegistry = class {
           if (this.logPayloads) {
             logContext.payload = payload;
           }
-          this.logger.error(
-            `User ${userIdShort} ${serviceName}.${methodName} -> fail`,
-            logContext
-          );
+          this.logger.error(`User ${userIdShort} ${serviceName}.${methodName} -> fail`, logContext);
           const errorResponse = {
             success: false,
             error: error instanceof Error ? error.message : "Internal error",
@@ -407,9 +474,7 @@ function createQuickdrawServer(options) {
   });
   httpServer.listen(options.port, "0.0.0.0", () => {
     serverLogger.info(`Server listening on port ${options.port}`);
-    serverLogger.info(
-      `Registered services: ${registry.getServices().join(", ")}`
-    );
+    serverLogger.info(`Registered services: ${registry.getServices().join(", ")}`);
   });
   const shutdown = async (signal) => {
     serverLogger.info(`Received ${signal}, starting graceful shutdown...`);