@vrplatform/kysely 1.3.41 → 1.3.45-1

package/build/module/query/tenant/billing-partner.d.ts

@@ -0,0 +1,4 @@
+export declare function getCalculatedBillingPartnerId({ billingPartnerId, partnerId, }: {
+    billingPartnerId?: string | null;
+    partnerId?: string | null;
+}): string;

package/build/module/query/tenant/billing-partner.js

@@ -0,0 +1,5 @@
+const VRP_TENANT_ID = '7bfec41f-aa3f-4602-95d2-f7996e4f0e59';
+export function getCalculatedBillingPartnerId({ billingPartnerId, partnerId, }) {
+    return billingPartnerId ?? partnerId ?? VRP_TENANT_ID;
+}
+//# sourceMappingURL=billing-partner.js.map

package/build/module/query/tenant/billing-partner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"billing-partner.js","sourceRoot":"src/","sources":["query/tenant/billing-partner.ts"],"names":[],"mappings":"AAAA,MAAM,aAAa,GAAG,sCAAsC,CAAC;AAE7D,MAAM,UAAU,6BAA6B,CAAC,EAC5C,gBAAgB,EAChB,SAAS,GAIV;IACC,OAAO,gBAAgB,IAAI,SAAS,IAAI,aAAa,CAAC;AACxD,CAAC","sourcesContent":["const VRP_TENANT_ID = '7bfec41f-aa3f-4602-95d2-f7996e4f0e59';\n\nexport function getCalculatedBillingPartnerId({\n  billingPartnerId,\n  partnerId,\n}: {\n  billingPartnerId?: string | null;\n  partnerId?: string | null;\n}) {\n  return billingPartnerId ?? partnerId ?? VRP_TENANT_ID;\n}\n"]}

package/build/module/query/tenant/index.d.ts

@@ -0,0 +1,2 @@
+export * from './billing-partner';
+export * from './status';

package/build/module/query/tenant/index.js

@@ -0,0 +1,3 @@
+export * from './billing-partner';
+export * from './status';
+//# sourceMappingURL=index.js.map

package/build/module/query/tenant/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["query/tenant/index.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC;AAClC,cAAc,UAAU,CAAC","sourcesContent":["export * from './billing-partner';\nexport * from './status';\n"]}

package/build/module/query/tenant/status.d.ts

@@ -0,0 +1,14 @@
+import { type DB, type ExpressionBuilder, type PublicTenant, type SqlBool } from '../../isomorphic';
+type TenantExpressionBuilder = ExpressionBuilder<DB & {
+    tenant: PublicTenant;
+}, 'tenant'>;
+export declare function whereTenantActive(eb: TenantExpressionBuilder): import("kysely").ExpressionWrapper<DB & {
+    tenant: PublicTenant;
+}, "tenant", SqlBool>;
+export declare function whereTenantNotTestTeam(testTeamId: string): import("kysely").RawBuilder<SqlBool>;
+export declare function whereTenantShouldBePaid(eb: TenantExpressionBuilder, { includePartnerTeams, }?: {
+    includePartnerTeams?: boolean;
+}): import("kysely").ExpressionWrapper<DB & {
+    tenant: PublicTenant;
+}, "tenant", SqlBool>;
+export {};

package/build/module/query/tenant/status.js

@@ -0,0 +1,35 @@
+import { sql, } from '../../isomorphic';
+import { getCalculatedBillingPartnerId } from './billing-partner';
+export function whereTenantActive(eb) {
+    return eb.and([
+        eb('tenant.calculatedStatus', '=', 'active'),
+        eb.not(eb.exists(eb
+            .selectFrom('public.tenant as billingPartner')
+            .select('billingPartner.id')
+            .where('billingPartner.id', '=', eb.fn.coalesce('tenant.calculatedBillingPartnerId', eb.val(getCalculatedBillingPartnerId({}))))
+            .where('billingPartner.calculatedStatus', 'is distinct from', 'active'))),
+    ]);
+}
+export function whereTenantNotTestTeam(testTeamId) {
+    return sql `(
+    ${sql.ref('tenant.id')} != ${testTeamId}
+    and (
+      ${sql.ref('tenant.partnerId')} is null
+      or ${sql.ref('tenant.partnerId')} != ${testTeamId}
+    )
+    and (
+      ${sql.ref('tenant.billingPartnerId')} is null
+      or ${sql.ref('tenant.billingPartnerId')} != ${testTeamId}
+    )
+  )`;
+}
+export function whereTenantShouldBePaid(eb, { includePartnerTeams = false, } = {}) {
+    return eb.and([
+        whereTenantActive(eb),
+        eb('tenant.isOnboarding', '=', false),
+        ...(includePartnerTeams
+            ? []
+            : [eb('tenant.calculatedIsBillable', '=', true)]),
+    ]);
+}
+//# sourceMappingURL=status.js.map

package/build/module/query/tenant/status.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"status.js","sourceRoot":"src/","sources":["query/tenant/status.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,GAAG,GAKJ,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,6BAA6B,EAAE,MAAM,mBAAmB,CAAC;AAOlE,MAAM,UAAU,iBAAiB,CAAC,EAA2B;IAC3D,OAAO,EAAE,CAAC,GAAG,CAAC;QACZ,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE,QAAQ,CAAC;QAC5C,EAAE,CAAC,GAAG,CACJ,EAAE,CAAC,MAAM,CACP,EAAE;aACC,UAAU,CAAC,iCAAiC,CAAC;aAC7C,MAAM,CAAC,mBAAmB,CAAC;aAC3B,KAAK,CACJ,mBAAmB,EACnB,GAAG,EACH,EAAE,CAAC,EAAE,CAAC,QAAQ,CACZ,mCAAmC,EACnC,EAAE,CAAC,GAAG,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC,CAC1C,CACF;aACA,KAAK,CACJ,iCAAiC,EACjC,kBAAkB,EAClB,QAAQ,CACT,CACJ,CACF;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,UAAkB;IACvD,OAAO,GAAG,CAAS;MACf,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,OAAO,UAAU;;QAEnC,GAAG,CAAC,GAAG,CAAC,kBAAkB,CAAC;WACxB,GAAG,CAAC,GAAG,CAAC,kBAAkB,CAAC,OAAO,UAAU;;;QAG/C,GAAG,CAAC,GAAG,CAAC,yBAAyB,CAAC;WAC/B,GAAG,CAAC,GAAG,CAAC,yBAAyB,CAAC,OAAO,UAAU;;IAE1D,CAAC;AACL,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,EAA2B,EAC3B,EACE,mBAAmB,GAAG,KAAK,MAGzB,EAAE;IAEN,OAAO,EAAE,CAAC,GAAG,CAAC;QACZ,iBAAiB,CAAC,EAAE,CAAC;QACrB,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE,KAAK,CAAC;QACrC,GAAG,CAAC,mBAAmB;YACrB,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,CAAC,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC;KACpD,CAAC,CAAC;AACL,CAAC","sourcesContent":["import {\n  sql,\n  type DB,\n  type ExpressionBuilder,\n  type PublicTenant,\n  type SqlBool,\n} from '../../isomorphic';\nimport { getCalculatedBillingPartnerId } from './billing-partner';\n\ntype TenantExpressionBuilder = ExpressionBuilder<\n  DB & { tenant: PublicTenant },\n  'tenant'\n>;\n\nexport function whereTenantActive(eb: TenantExpressionBuilder) {\n  return eb.and([\n    eb('tenant.calculatedStatus', '=', 'active'),\n    eb.not(\n      eb.exists(\n        eb\n          .selectFrom('public.tenant as billingPartner')\n          .select('billingPartner.id')\n          .where(\n            'billingPartner.id',\n            '=',\n            eb.fn.coalesce(\n              'tenant.calculatedBillingPartnerId',\n              eb.val(getCalculatedBillingPartnerId({}))\n            )\n          )\n          .where(\n            'billingPartner.calculatedStatus',\n            'is distinct from',\n            'active'\n          )\n      )\n    ),\n  ]);\n}\n\nexport function whereTenantNotTestTeam(testTeamId: string) {\n  return sql<SqlBool>`(\n    ${sql.ref('tenant.id')} != ${testTeamId}\n    and (\n      ${sql.ref('tenant.partnerId')} is null\n      or ${sql.ref('tenant.partnerId')} != ${testTeamId}\n    )\n    and (\n      ${sql.ref('tenant.billingPartnerId')} is null\n      or ${sql.ref('tenant.billingPartnerId')} != ${testTeamId}\n    )\n  )`;\n}\n\nexport function whereTenantShouldBePaid(\n  eb: TenantExpressionBuilder,\n  {\n    includePartnerTeams = false,\n  }: {\n    includePartnerTeams?: boolean;\n  } = {}\n) {\n  return eb.and([\n    whereTenantActive(eb),\n    eb('tenant.isOnboarding', '=', false),\n    ...(includePartnerTeams\n      ? []\n      : [eb('tenant.calculatedIsBillable', '=', true)]),\n  ]);\n}\n"]}

package/build/module/regional.d.ts

@@ -0,0 +1,60 @@
+import type { ControlPlaneKysely, Kysely } from './index';
+export declare const tenantDataRegions: readonly ["ap", "crunchy", "eu", "hostaway", "us"];
+export type TenantDataRegion = (typeof tenantDataRegions)[number];
+export type RegionEnvironment = Record<string, string | undefined>;
+export declare function assertTenantDataRegion(value: string | undefined): asserts value is TenantDataRegion;
+export declare function resolveTenantDatabaseUrl(dataRegion: TenantDataRegion, env: RegionEnvironment): string;
+export declare function resolveTenantApiBaseUrl(dataRegion: TenantDataRegion, env: RegionEnvironment): string;
+export declare function resolveConfiguredTenantDataRegions(env: RegionEnvironment): ("ap" | "crunchy" | "eu" | "hostaway" | "us")[];
+type UseAsyncRegionalKysely = (connectionString: string, options?: {
+    repositoryName?: string;
+    probe?: true;
+}) => Promise<Kysely>;
+type UseAsyncControlPlaneKysely = (connectionString: string, options?: {
+    repositoryName?: string;
+    probe?: true;
+}) => Promise<ControlPlaneKysely>;
+export declare function useTenantDatabaseKysely(dataRegion: TenantDataRegion, repositoryName: string, env: RegionEnvironment, options?: {
+    useAsyncKysely?: UseAsyncRegionalKysely;
+    probe?: true;
+}): Promise<Kysely>;
+export type TenantKysely = {
+    id: string;
+    slug: string;
+    dataRegion: TenantDataRegion;
+    kysely: Kysely;
+};
+/**
+ * Resolve a tenant's regional Kysely connection from a connection id, slug,
+ * or tenant id. The control-plane connection is opened and destroyed
+ * internally; the returned regional connection must be destroyed by the
+ * caller.
+ */
+export declare function resolveTenantKysely({ connectionId, slug, tenantId, repositoryName, env, ...options }: {
+    connectionId?: string;
+    slug?: string;
+    tenantId?: string;
+    repositoryName: string;
+    env: RegionEnvironment;
+    useAsyncControlPlaneKysely?: UseAsyncControlPlaneKysely;
+    useAsyncKysely?: UseAsyncRegionalKysely;
+    probe?: true;
+}): Promise<TenantKysely>;
+export type RegionRunError = {
+    dataRegion: TenantDataRegion;
+    error: string;
+};
+/**
+ * Run a job once per configured tenant data region with per-region error
+ * isolation: one region's outage must not stall the others (regions iterate
+ * in a fixed order, so a failing first region would otherwise starve the
+ * rest on every attempt). Throws only when every configured region failed.
+ */
+export declare function forEachConfiguredRegion<T>(run: (dataRegion: TenantDataRegion) => Promise<T>, env: RegionEnvironment): Promise<{
+    results: {
+        dataRegion: TenantDataRegion;
+        result: T;
+    }[];
+    errors: RegionRunError[];
+}>;
+export {};

package/build/module/regional.js

@@ -0,0 +1,162 @@
+import { KyselyError } from './error';
+// The tenant-data-region contract lives here — the region set plus how a
+// region resolves to its database url, API base url, and how to fan a job out
+// across regions. Every runtime resolves regions through these functions;
+// @vrplatform/common re-exports them for app-facing imports, and the trigger
+// runtime wraps them only to default `env` to process.env.
+//
+// Region → resource keys are derived from the region name
+// (`DATABASE_URL_VRTRUST_<REGION>`, `API_BASE_URL_<REGION>`), so adding a
+// region means extending this array plus its env values and migrations — not
+// editing per-region branches.
+export const tenantDataRegions = [
+    'ap',
+    'crunchy',
+    'eu',
+    'hostaway',
+    'us',
+];
+export function assertTenantDataRegion(value) {
+    if (!tenantDataRegions.includes(value)) {
+        throw new KyselyError('Missing or invalid tenant data region', { value });
+    }
+}
+function databaseUrlEnvKey(dataRegion) {
+    return `DATABASE_URL_VRTRUST_${dataRegion.toUpperCase()}`;
+}
+function apiBaseUrlEnvKey(dataRegion) {
+    return `API_BASE_URL_${dataRegion.toUpperCase()}`;
+}
+export function resolveTenantDatabaseUrl(dataRegion, env) {
+    const databaseUrl = env[databaseUrlEnvKey(dataRegion)];
+    if (!databaseUrl) {
+        throw new KyselyError('Tenant database URL is missing', {
+            dataRegion,
+            key: databaseUrlEnvKey(dataRegion),
+        });
+    }
+    return databaseUrl;
+}
+export function resolveTenantApiBaseUrl(dataRegion, env) {
+    const key = apiBaseUrlEnvKey(dataRegion);
+    const baseUrl = env[key];
+    if (!baseUrl) {
+        throw new KyselyError('Tenant API base URL is missing', {
+            dataRegion,
+            key,
+        });
+    }
+    return baseUrl;
+}
+export function resolveConfiguredTenantDataRegions(env) {
+    const configured = tenantDataRegions.filter((region) => Boolean(env[databaseUrlEnvKey(region)]));
+    if (!configured.length) {
+        throw new Error('No tenant data region database urls configured');
+    }
+    // Two regions sharing one URL means every region-fan-out job scans the
+    // same physical DB twice (e.g. a stale dashboard env var re-merged into a
+    // deploy). Regions must stay unset until they point at distinct databases.
+    const regionsByUrl = new Map();
+    for (const region of configured) {
+        const url = env[databaseUrlEnvKey(region)];
+        regionsByUrl.set(url, [...(regionsByUrl.get(url) ?? []), region]);
+    }
+    for (const aliased of regionsByUrl.values()) {
+        if (aliased.length > 1) {
+            throw new KyselyError('Tenant data regions share the same database URL', {
+                dataRegions: aliased,
+            });
+        }
+    }
+    return configured;
+}
+async function defaultUseAsyncKysely(connectionString, options) {
+    const { useAsyncKysely } = await import('./index');
+    return await useAsyncKysely(connectionString, options);
+}
+async function defaultUseAsyncControlPlaneKysely(connectionString, options) {
+    const { useAsyncControlPlaneKysely } = await import('./index');
+    return await useAsyncControlPlaneKysely(connectionString, options);
+}
+export async function useTenantDatabaseKysely(dataRegion, repositoryName, env, options = {}) {
+    const open = options.useAsyncKysely ?? defaultUseAsyncKysely;
+    return await open(resolveTenantDatabaseUrl(dataRegion, env), {
+        repositoryName,
+        probe: options.probe,
+    });
+}
+/**
+ * Resolve a tenant's regional Kysely connection from a connection id, slug,
+ * or tenant id. The control-plane connection is opened and destroyed
+ * internally; the returned regional connection must be destroyed by the
+ * caller.
+ */
+export async function resolveTenantKysely({ connectionId, slug, tenantId, repositoryName, env, ...options }) {
+    if (!connectionId && !slug && !tenantId) {
+        throw new Error('connectionId, slug, or tenantId is required');
+    }
+    const controlPlaneDatabaseUrl = env.CONTROL_PLANE_DATABASE_URL;
+    if (!controlPlaneDatabaseUrl)
+        throw new Error('CONTROL_PLANE_DATABASE_URL is required');
+    const controlPlane = await (options.useAsyncControlPlaneKysely ?? defaultUseAsyncControlPlaneKysely)(controlPlaneDatabaseUrl, {
+        repositoryName: `${repositoryName}/control-plane`,
+        probe: options.probe,
+    });
+    let tenant;
+    try {
+        const query = controlPlane
+            .selectFrom('controlPlane.tenant as tenant')
+            .select([
+            'tenant.id as id',
+            'tenant.slug as slug',
+            'tenant.dataRegion as dataRegion',
+        ]);
+        tenant = await (slug
+            ? query.where('tenant.slug', '=', slug)
+            : connectionId
+                ? query
+                    .innerJoin('controlPlane.connectionRoute as connectionRoute', 'connectionRoute.tenantId', 'tenant.id')
+                    .where('connectionRoute.connectionId', '=', connectionId)
+                : query.where('tenant.id', '=', tenantId)).executeTakeFirst();
+    }
+    finally {
+        await controlPlane.destroy();
+    }
+    if (!tenant)
+        throw new Error('Tenant not found in control plane');
+    assertTenantDataRegion(tenant.dataRegion);
+    return {
+        id: tenant.id,
+        slug: tenant.slug,
+        dataRegion: tenant.dataRegion,
+        kysely: await useTenantDatabaseKysely(tenant.dataRegion, repositoryName, env, options),
+    };
+}
+/**
+ * Run a job once per configured tenant data region with per-region error
+ * isolation: one region's outage must not stall the others (regions iterate
+ * in a fixed order, so a failing first region would otherwise starve the
+ * rest on every attempt). Throws only when every configured region failed.
+ */
+export async function forEachConfiguredRegion(run, env) {
+    const results = [];
+    const errors = [];
+    for (const dataRegion of resolveConfiguredTenantDataRegions(env)) {
+        try {
+            results.push({ dataRegion, result: await run(dataRegion) });
+        }
+        catch (error) {
+            errors.push({
+                dataRegion,
+                error: error instanceof Error ? error.message : String(error),
+            });
+        }
+    }
+    if (!results.length && errors.length) {
+        throw new KyselyError('All configured tenant data regions failed', {
+            errors,
+        });
+    }
+    return { results, errors };
+}
+//# sourceMappingURL=regional.js.map

package/build/module/regional.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"regional.js","sourceRoot":"src/","sources":["regional.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAEtC,yEAAyE;AACzE,8EAA8E;AAC9E,0EAA0E;AAC1E,6EAA6E;AAC7E,2DAA2D;AAC3D,EAAE;AACF,0DAA0D;AAC1D,0EAA0E;AAC1E,6EAA6E;AAC7E,+BAA+B;AAC/B,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,IAAI;IACJ,SAAS;IACT,IAAI;IACJ,UAAU;IACV,IAAI;CACI,CAAC;AAKX,MAAM,UAAU,sBAAsB,CACpC,KAAyB;IAEzB,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,KAAyB,CAAC,EAAE,CAAC;QAC3D,MAAM,IAAI,WAAW,CAAC,uCAAuC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;IAC5E,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,UAA4B;IACrD,OAAO,wBAAwB,UAAU,CAAC,WAAW,EAAE,EAAE,CAAC;AAC5D,CAAC;AAED,SAAS,gBAAgB,CAAC,UAA4B;IACpD,OAAO,gBAAgB,UAAU,CAAC,WAAW,EAAE,EAAE,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,UAA4B,EAC5B,GAAsB;IAEtB,MAAM,WAAW,GAAG,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC,CAAC;IACvD,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,MAAM,IAAI,WAAW,CAAC,gCAAgC,EAAE;YACtD,UAAU;YACV,GAAG,EAAE,iBAAiB,CAAC,UAAU,CAAC;SACnC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,UAA4B,EAC5B,GAAsB;IAEtB,MAAM,GAAG,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAC;IACzC,MAAM,OAAO,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;IACzB,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,WAAW,CAAC,gCAAgC,EAAE;YACtD,UAAU;YACV,GAAG;SACJ,CAAC,CAAC;IACL,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,kCAAkC,CAAC,GAAsB;IACvE,MAAM,UAAU,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CACrD,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,CACxC,CAAC;IAEF,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACpE,CAAC;IAED,uEAAuE;IACvE,0EAA0E;IAC1E,2EAA2E;IAC3E,MAAM,YAAY,GAAG,IAAI,GAAG,EAA8B,CAAC;IAC3D,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;QAChC,MAAM,GAAG,GAAG,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAE,CAAC;QAC5C,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;IACpE,CAAC;IACD,KAAK,MAAM,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;QAC5C,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,WAAW,CAAC,iDAAiD,EAAE;gBACvE,WAAW,EAAE,OAAO;aACrB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAkBD,KAAK,UAAU,qBAAqB,CAClC,gBAAwB,EACxB,OAA+C;IAE/C,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;IACnD,OAAO,MAAM,cAAc,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAC;AACzD,CAAC;AAED,KAAK,UAAU,iCAAiC,CAC9C,gBAAwB,EACxB,OAAmD;IAEnD,MAAM,EAAE,0BAA0B,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;IAC/D,OAAO,MAAM,0BAA0B,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAC;AACrE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,UAA4B,EAC5B,cAAsB,EACtB,GAAsB,EACtB,UAAqE,EAAE;IAEvE,MAAM,IAAI,GAAG,OAAO,CAAC,cAAc,IAAI,qBAAqB,CAAC;IAC7D,OAAO,MAAM,IAAI,CAAC,wBAAwB,CAAC,UAAU,EAAE,GAAG,CAAC,EAAE;QAC3D,cAAc;QACd,KAAK,EAAE,OAAO,CAAC,KAAK;KACrB,CAAC,CAAC;AACL,CAAC;AASD;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EACxC,YAAY,EACZ,IAAI,EACJ,QAAQ,EACR,cAAc,EACd,GAAG,EACH,GAAG,OAAO,EAUX;IACC,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;QACxC,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,uBAAuB,GAAG,GAAG,CAAC,0BAA0B,CAAC;IAC/D,IAAI,CAAC,uBAAuB;QAC1B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAE5D,MAAM,YAAY,GAAG,MAAM,CACzB,OAAO,CAAC,0BAA0B,IAAI,iCAAiC,CACxE,CAAC,uBAAuB,EAAE;QACzB,cAAc,EAAE,GAAG,cAAc,gBAAgB;QACjD,KAAK,EAAE,OAAO,CAAC,KAAK;KACrB,CAAC,CAAC;IACH,IAAI,MAAoE,CAAC;IACzE,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,YAAY;aACvB,UAAU,CAAC,+BAA+B,CAAC;aAC3C,MAAM,CAAC;YACN,iBAAiB;YACjB,qBAAqB;YACrB,iCAAiC;SAClC,CAAC,CAAC;QACL,MAAM,GAAG,MAAM,CACb,IAAI;YACF,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,EAAE,GAAG,EAAE,IAAI,CAAC;YACvC,CAAC,CAAC,YAAY;gBACZ,CAAC,CAAC,KAAK;qBACF,SAAS,CACR,iDAAiD,EACjD,0BAA0B,EAC1B,WAAW,CACZ;qBACA,KAAK,CAAC,8BAA8B,EAAE,GAAG,EAAE,YAAY,CAAC;gBAC7D,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,GAAG,EAAE,QAAS,CAAC,CAC/C,CAAC,gBAAgB,EAAE,CAAC;IACvB,CAAC;YAAS,CAAC;QACT,MAAM,YAAY,CAAC,OAAO,EAAE,CAAC;IAC/B,CAAC;IACD,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IAElE,sBAAsB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAC1C,OAAO;QACL,EAAE,EAAE,MAAM,CAAC,EAAE;QACb,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,MAAM,EAAE,MAAM,uBAAuB,CACnC,MAAM,CAAC,UAAU,EACjB,cAAc,EACd,GAAG,EACH,OAAO,CACR;KACF,CAAC;AACJ,CAAC;AAOD;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,GAAiD,EACjD,GAAsB;IAKtB,MAAM,OAAO,GAAkD,EAAE,CAAC;IAClE,MAAM,MAAM,GAAqB,EAAE,CAAC;IAEpC,KAAK,MAAM,UAAU,IAAI,kCAAkC,CAAC,GAAG,CAAC,EAAE,CAAC;QACjE,IAAI,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,IAAI,CAAC;gBACV,UAAU;gBACV,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QACrC,MAAM,IAAI,WAAW,CAAC,2CAA2C,EAAE;YACjE,MAAM;SACP,CAAC,CAAC;IACL,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AAC7B,CAAC","sourcesContent":["import type { ControlPlaneKysely, Kysely } from './index';\nimport { KyselyError } from './error';\n\n// The tenant-data-region contract lives here — the region set plus how a\n// region resolves to its database url, API base url, and how to fan a job out\n// across regions. Every runtime resolves regions through these functions;\n// @vrplatform/common re-exports them for app-facing imports, and the trigger\n// runtime wraps them only to default `env` to process.env.\n//\n// Region → resource keys are derived from the region name\n// (`DATABASE_URL_VRTRUST_<REGION>`, `API_BASE_URL_<REGION>`), so adding a\n// region means extending this array plus its env values and migrations — not\n// editing per-region branches.\nexport const tenantDataRegions = [\n  'ap',\n  'crunchy',\n  'eu',\n  'hostaway',\n  'us',\n] as const;\nexport type TenantDataRegion = (typeof tenantDataRegions)[number];\n\nexport type RegionEnvironment = Record<string, string | undefined>;\n\nexport function assertTenantDataRegion(\n  value: string | undefined\n): asserts value is TenantDataRegion {\n  if (!tenantDataRegions.includes(value as TenantDataRegion)) {\n    throw new KyselyError('Missing or invalid tenant data region', { value });\n  }\n}\n\nfunction databaseUrlEnvKey(dataRegion: TenantDataRegion) {\n  return `DATABASE_URL_VRTRUST_${dataRegion.toUpperCase()}`;\n}\n\nfunction apiBaseUrlEnvKey(dataRegion: TenantDataRegion) {\n  return `API_BASE_URL_${dataRegion.toUpperCase()}`;\n}\n\nexport function resolveTenantDatabaseUrl(\n  dataRegion: TenantDataRegion,\n  env: RegionEnvironment\n) {\n  const databaseUrl = env[databaseUrlEnvKey(dataRegion)];\n  if (!databaseUrl) {\n    throw new KyselyError('Tenant database URL is missing', {\n      dataRegion,\n      key: databaseUrlEnvKey(dataRegion),\n    });\n  }\n  return databaseUrl;\n}\n\nexport function resolveTenantApiBaseUrl(\n  dataRegion: TenantDataRegion,\n  env: RegionEnvironment\n) {\n  const key = apiBaseUrlEnvKey(dataRegion);\n  const baseUrl = env[key];\n  if (!baseUrl) {\n    throw new KyselyError('Tenant API base URL is missing', {\n      dataRegion,\n      key,\n    });\n  }\n  return baseUrl;\n}\n\nexport function resolveConfiguredTenantDataRegions(env: RegionEnvironment) {\n  const configured = tenantDataRegions.filter((region) =>\n    Boolean(env[databaseUrlEnvKey(region)])\n  );\n\n  if (!configured.length) {\n    throw new Error('No tenant data region database urls configured');\n  }\n\n  // Two regions sharing one URL means every region-fan-out job scans the\n  // same physical DB twice (e.g. a stale dashboard env var re-merged into a\n  // deploy). Regions must stay unset until they point at distinct databases.\n  const regionsByUrl = new Map<string, TenantDataRegion[]>();\n  for (const region of configured) {\n    const url = env[databaseUrlEnvKey(region)]!;\n    regionsByUrl.set(url, [...(regionsByUrl.get(url) ?? []), region]);\n  }\n  for (const aliased of regionsByUrl.values()) {\n    if (aliased.length > 1) {\n      throw new KyselyError('Tenant data regions share the same database URL', {\n        dataRegions: aliased,\n      });\n    }\n  }\n\n  return configured;\n}\n\ntype UseAsyncRegionalKysely = (\n  connectionString: string,\n  options?: {\n    repositoryName?: string;\n    probe?: true;\n  }\n) => Promise<Kysely>;\n\ntype UseAsyncControlPlaneKysely = (\n  connectionString: string,\n  options?: {\n    repositoryName?: string;\n    probe?: true;\n  }\n) => Promise<ControlPlaneKysely>;\n\nasync function defaultUseAsyncKysely(\n  connectionString: string,\n  options?: Parameters<UseAsyncRegionalKysely>[1]\n) {\n  const { useAsyncKysely } = await import('./index');\n  return await useAsyncKysely(connectionString, options);\n}\n\nasync function defaultUseAsyncControlPlaneKysely(\n  connectionString: string,\n  options?: Parameters<UseAsyncControlPlaneKysely>[1]\n) {\n  const { useAsyncControlPlaneKysely } = await import('./index');\n  return await useAsyncControlPlaneKysely(connectionString, options);\n}\n\nexport async function useTenantDatabaseKysely(\n  dataRegion: TenantDataRegion,\n  repositoryName: string,\n  env: RegionEnvironment,\n  options: { useAsyncKysely?: UseAsyncRegionalKysely; probe?: true } = {}\n) {\n  const open = options.useAsyncKysely ?? defaultUseAsyncKysely;\n  return await open(resolveTenantDatabaseUrl(dataRegion, env), {\n    repositoryName,\n    probe: options.probe,\n  });\n}\n\nexport type TenantKysely = {\n  id: string;\n  slug: string;\n  dataRegion: TenantDataRegion;\n  kysely: Kysely;\n};\n\n/**\n * Resolve a tenant's regional Kysely connection from a connection id, slug,\n * or tenant id. The control-plane connection is opened and destroyed\n * internally; the returned regional connection must be destroyed by the\n * caller.\n */\nexport async function resolveTenantKysely({\n  connectionId,\n  slug,\n  tenantId,\n  repositoryName,\n  env,\n  ...options\n}: {\n  connectionId?: string;\n  slug?: string;\n  tenantId?: string;\n  repositoryName: string;\n  env: RegionEnvironment;\n  useAsyncControlPlaneKysely?: UseAsyncControlPlaneKysely;\n  useAsyncKysely?: UseAsyncRegionalKysely;\n  probe?: true;\n}): Promise<TenantKysely> {\n  if (!connectionId && !slug && !tenantId) {\n    throw new Error('connectionId, slug, or tenantId is required');\n  }\n\n  const controlPlaneDatabaseUrl = env.CONTROL_PLANE_DATABASE_URL;\n  if (!controlPlaneDatabaseUrl)\n    throw new Error('CONTROL_PLANE_DATABASE_URL is required');\n\n  const controlPlane = await (\n    options.useAsyncControlPlaneKysely ?? defaultUseAsyncControlPlaneKysely\n  )(controlPlaneDatabaseUrl, {\n    repositoryName: `${repositoryName}/control-plane`,\n    probe: options.probe,\n  });\n  let tenant: { id: string; slug: string; dataRegion: string } | undefined;\n  try {\n    const query = controlPlane\n      .selectFrom('controlPlane.tenant as tenant')\n      .select([\n        'tenant.id as id',\n        'tenant.slug as slug',\n        'tenant.dataRegion as dataRegion',\n      ]);\n    tenant = await (\n      slug\n        ? query.where('tenant.slug', '=', slug)\n        : connectionId\n          ? query\n              .innerJoin(\n                'controlPlane.connectionRoute as connectionRoute',\n                'connectionRoute.tenantId',\n                'tenant.id'\n              )\n              .where('connectionRoute.connectionId', '=', connectionId)\n          : query.where('tenant.id', '=', tenantId!)\n    ).executeTakeFirst();\n  } finally {\n    await controlPlane.destroy();\n  }\n  if (!tenant) throw new Error('Tenant not found in control plane');\n\n  assertTenantDataRegion(tenant.dataRegion);\n  return {\n    id: tenant.id,\n    slug: tenant.slug,\n    dataRegion: tenant.dataRegion,\n    kysely: await useTenantDatabaseKysely(\n      tenant.dataRegion,\n      repositoryName,\n      env,\n      options\n    ),\n  };\n}\n\nexport type RegionRunError = {\n  dataRegion: TenantDataRegion;\n  error: string;\n};\n\n/**\n * Run a job once per configured tenant data region with per-region error\n * isolation: one region's outage must not stall the others (regions iterate\n * in a fixed order, so a failing first region would otherwise starve the\n * rest on every attempt). Throws only when every configured region failed.\n */\nexport async function forEachConfiguredRegion<T>(\n  run: (dataRegion: TenantDataRegion) => Promise<T>,\n  env: RegionEnvironment\n): Promise<{\n  results: { dataRegion: TenantDataRegion; result: T }[];\n  errors: RegionRunError[];\n}> {\n  const results: { dataRegion: TenantDataRegion; result: T }[] = [];\n  const errors: RegionRunError[] = [];\n\n  for (const dataRegion of resolveConfiguredTenantDataRegions(env)) {\n    try {\n      results.push({ dataRegion, result: await run(dataRegion) });\n    } catch (error) {\n      errors.push({\n        dataRegion,\n        error: error instanceof Error ? error.message : String(error),\n      });\n    }\n  }\n\n  if (!results.length && errors.length) {\n    throw new KyselyError('All configured tenant data regions failed', {\n      errors,\n    });\n  }\n\n  return { results, errors };\n}\n"]}

package/build/module/test/index.d.ts

@@ -1,17 +1,6 @@
-import type { Kysely } from 'kysely';
-import { type UseLocalKyselyOptions } from '../local';
-import type { DB } from '../v1.generated';
-declare global {
-    var __vrplatform_testSharedKysely: Promise<Kysely<DB>> | Kysely<DB> | undefined;
-}
-/**
- * @deprecated Use `useLocalKyselyResource` instead.
- */
-export declare function useTestKysely(cleanupStrategy: 'afterAll', args?: UseLocalKyselyOptions): Promise<{
-    kysely: Kysely<DB>;
-}>;
-export declare function useTestAutoCleanupKysely(args?: UseLocalKyselyOptions): Promise<Kysely<DB>>;
-type ReturnType = Kysely<DB> & {
+import { type LocalKysely, type UseLocalKyselyOptions } from '../local';
+export declare function useTestAutoCleanupKysely(args?: UseLocalKyselyOptions): Promise<LocalKysely>;
+type ReturnType = LocalKysely & {
     [Symbol.dispose]: () => Promise<void>;
 };
 export declare function useLocalKyselyResource(args?: UseLocalKyselyOptions): Promise<ReturnType>;

package/build/module/test/index.js

@@ -1,28 +1,6 @@
 import { afterAll } from 'bun:test';
-import { useLocalKysely } from '../local';
-/**
- * @deprecated Use `useLocalKyselyResource` instead.
- */
-export async function useTestKysely(cleanupStrategy, args) {
-    const kysely = await useLocalKysely(args);
-    if (cleanupStrategy === 'afterAll')
-        afterAll(async () => {
-            await kysely.destroy();
-        });
-    return {
-        kysely,
-    };
-}
+import { useLocalKysely, } from '../local';
 export async function useTestAutoCleanupKysely(args) {
-    if (process.env.VR_TEST_SHARED_KYSELY === '1') {
-        const existing = globalThis.__vrplatform_testSharedKysely;
-        if (!existing) {
-            const created = useLocalKysely(args);
-            globalThis.__vrplatform_testSharedKysely = created;
-            return created;
-        }
-        return await existing;
-    }
     const kysely = await useLocalKysely(args);
     afterAll(async () => {
         await kysely.destroy();

package/build/module/test/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["test/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAEpC,OAAO,EAA8B,cAAc,EAAE,MAAM,UAAU,CAAC;AAYtE;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,eAA2B,EAC3B,IAA4B,EAC5B;IACA,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;IAE1C,IAAI,eAAe,KAAK,UAAU;QAChC,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC;YACnB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QAAA,CACxB,CAAC,CAAC;IAEL,OAAO;QACL,MAAM;KACP,CAAC;AAAA,CACH;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,IAA4B,EAAE;IAC3E,IAAI,OAAO,CAAC,GAAG,CAAC,qBAAqB,KAAK,GAAG,EAAE,CAAC;QAC9C,MAAM,QAAQ,GAAG,UAAU,CAAC,6BAA6B,CAAC;QAC1D,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,OAAO,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;YACrC,UAAU,CAAC,6BAA6B,GAAG,OAAO,CAAC;YACnD,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,OAAO,MAAM,QAAQ,CAAC;IACxB,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;IAE1C,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC;QACnB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IAAA,CACxB,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAAA,CACf;AAMD,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,IAA4B,EACP;IACrB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;IAE1C,OAAO,IAAI,KAAK,CACd,EAAE,EACF;QACE,GAAG,CAAC,OAAO,EAAE,IAAI,EAAE;YACjB,8CAA8C;YAC9C,IAAI,IAAI,KAAK,MAAM,CAAC,YAAY,EAAE,CAAC;gBACjC,OAAO,KAAK,IAAI,EAAE,CAAC;oBACjB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;gBAAA,CACxB,CAAC;YACJ,CAAC;YACD,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,IAAI,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACvC,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC;QAAA,CACrB;KACF,CACY,CAAC;AAAA,CACjB","sourcesContent":["import { afterAll } from 'bun:test';\nimport type { Kysely } from 'kysely';\nimport { type UseLocalKyselyOptions, useLocalKysely } from '../local';\nimport type { DB } from '../v1.generated';\n\ndeclare global {\n  // Shared Kysely instance for test processes that run multiple files in one Bun process.\n  // This must be process-global to allow cross-file reuse.\n  var __vrplatform_testSharedKysely:\n    | Promise<Kysely<DB>>\n    | Kysely<DB>\n    | undefined;\n}\n\n/**\n * @deprecated Use `useLocalKyselyResource` instead.\n */\nexport async function useTestKysely(\n  cleanupStrategy: 'afterAll',\n  args?: UseLocalKyselyOptions\n) {\n  const kysely = await useLocalKysely(args);\n\n  if (cleanupStrategy === 'afterAll')\n    afterAll(async () => {\n      await kysely.destroy();\n    });\n\n  return {\n    kysely,\n  };\n}\n\nexport async function useTestAutoCleanupKysely(args?: UseLocalKyselyOptions) {\n  if (process.env.VR_TEST_SHARED_KYSELY === '1') {\n    const existing = globalThis.__vrplatform_testSharedKysely;\n    if (!existing) {\n      const created = useLocalKysely(args);\n      globalThis.__vrplatform_testSharedKysely = created;\n      return created;\n    }\n    return await existing;\n  }\n\n  const kysely = await useLocalKysely(args);\n\n  afterAll(async () => {\n    await kysely.destroy();\n  });\n\n  return kysely;\n}\n\ntype ReturnType = Kysely<DB> & {\n  [Symbol.dispose]: () => Promise<void>;\n};\n\nexport async function useLocalKyselyResource(\n  args?: UseLocalKyselyOptions\n): Promise<ReturnType> {\n  const kysely = await useLocalKysely(args);\n\n  return new Proxy(\n    {},\n    {\n      get(_target, prop) {\n        // somehow not working if using Symbol.dispose\n        if (prop === Symbol.asyncDispose) {\n          return async () => {\n            await kysely.destroy();\n          };\n        }\n        if (prop === 'kysely') {\n          return kysely;\n        }\n        if (typeof kysely[prop] === 'function') {\n          return kysely[prop].bind(kysely);\n        }\n        return kysely[prop];\n      },\n    }\n  ) as ReturnType;\n}\n"]}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["test/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AACpC,OAAO,EAGL,cAAc,GACf,MAAM,UAAU,CAAC;AAElB,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,IAA4B;IACzE,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;IAE1C,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,IAA4B;IAE5B,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;IAE1C,OAAO,IAAI,KAAK,CACd,EAAE,EACF;QACE,GAAG,CAAC,OAAO,EAAE,IAAI;YACf,8CAA8C;YAC9C,IAAI,IAAI,KAAK,MAAM,CAAC,YAAY,EAAE,CAAC;gBACjC,OAAO,KAAK,IAAI,EAAE;oBAChB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;gBACzB,CAAC,CAAC;YACJ,CAAC;YACD,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,IAAI,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACvC,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;KACF,CACY,CAAC;AAClB,CAAC","sourcesContent":["import { afterAll } from 'bun:test';\nimport {\n  type LocalKysely,\n  type UseLocalKyselyOptions,\n  useLocalKysely,\n} from '../local';\n\nexport async function useTestAutoCleanupKysely(args?: UseLocalKyselyOptions) {\n  const kysely = await useLocalKysely(args);\n\n  afterAll(async () => {\n    await kysely.destroy();\n  });\n\n  return kysely;\n}\n\ntype ReturnType = LocalKysely & {\n  [Symbol.dispose]: () => Promise<void>;\n};\n\nexport async function useLocalKyselyResource(\n  args?: UseLocalKyselyOptions\n): Promise<ReturnType> {\n  const kysely = await useLocalKysely(args);\n\n  return new Proxy(\n    {},\n    {\n      get(_target, prop) {\n        // somehow not working if using Symbol.dispose\n        if (prop === Symbol.asyncDispose) {\n          return async () => {\n            await kysely.destroy();\n          };\n        }\n        if (prop === 'kysely') {\n          return kysely;\n        }\n        if (typeof kysely[prop] === 'function') {\n          return kysely[prop].bind(kysely);\n        }\n        return kysely[prop];\n      },\n    }\n  ) as ReturnType;\n}\n"]}

package/build/module/transaction.d.ts

@@ -1,7 +1,7 @@
-import { type ControlledTransaction, type Kysely, type Transaction } from './isomorphic';
+import { type ControlledTransaction, type DatabaseKysely, type DatabaseTransaction, type Transaction } from './isomorphic';
 export type ActiveTransaction = Transaction | ControlledTransaction;
-export declare function isActiveTransaction(db: Kysely | ActiveTransaction): db is ActiveTransaction;
-export declare function startControlledTransaction(db: Kysely): Promise<import("kysely").ControlledTransaction<import("./v1.generated").DB, []>>;
+export declare function isActiveTransaction<Database>(db: DatabaseKysely<Database> | DatabaseTransaction<Database>): db is DatabaseTransaction<Database>;
+export declare function startControlledTransaction<Database>(db: DatabaseKysely<Database>): Promise<import("kysely").ControlledTransaction<Database, []>>;
 export declare function createTransactionSavepoint(trx: ActiveTransaction, savepointName: string): Promise<ActiveTransaction>;
 export declare function rollbackTransactionSavepoint(trx: ActiveTransaction, savepointName: string): Promise<ActiveTransaction>;
 export declare function releaseTransactionSavepoint(trx: ActiveTransaction, savepointName: string): Promise<ActiveTransaction>;

package/build/module/transaction.js.map

@@ -1 +1 @@
-{"version":3,"file":"transaction.js","sourceRoot":"src/","sources":["transaction.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,GAAG,GAEJ,MAAM,cAAc,CAAC;AAItB,MAAM,UAAU,mBAAmB,CACjC,EAA8B,EACL;IACzB,OAAO,EAAE,EAAE,aAAa,KAAK,IAAI,CAAC;AAAA,CACnC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAAU,EAAE;IAC3D,OAAO,EAAE,CAAC,gBAAgB,EAAE,CAAC,OAAO,EAAE,CAAC;AAAA,CACxC;AAED,uEAAuE;AACvE,kDAAkD;AAClD,SAAS,mBAAmB,CAC1B,GAAgB,EAChB,OAAoE,EACpE,aAAqB,EACrB;IACA,OAAO,GAAG,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;AAAA,CAC5D;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,GAAsB,EACtB,aAAqB,EACO;IAC5B,MAAM,mBAAmB,CAAC,GAAG,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IAC3D,OAAO,GAAG,CAAC;AAAA,CACZ;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,GAAsB,EACtB,aAAqB,EACO;IAC5B,MAAM,mBAAmB,CAAC,GAAG,EAAE,uBAAuB,EAAE,aAAa,CAAC,CAAC;IACvE,OAAO,GAAG,CAAC;AAAA,CACZ;AAED,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAC/C,GAAsB,EACtB,aAAqB,EACO;IAC5B,MAAM,mBAAmB,CAAC,GAAG,EAAE,mBAAmB,EAAE,aAAa,CAAC,CAAC;IACnE,OAAO,GAAG,CAAC;AAAA,CACZ","sourcesContent":["import {\n  type ControlledTransaction,\n  type Kysely,\n  sql,\n  type Transaction,\n} from './isomorphic';\n\nexport type ActiveTransaction = Transaction | ControlledTransaction;\n\nexport function isActiveTransaction(\n  db: Kysely | ActiveTransaction\n): db is ActiveTransaction {\n  return db?.isTransaction === true;\n}\n\nexport async function startControlledTransaction(db: Kysely) {\n  return db.startTransaction().execute();\n}\n\n// Our dialect stack supports controlled transactions, but not Kysely's\n// savepoint commands consistently across drivers.\nfunction runSavepointCommand(\n  trx: Transaction,\n  command: 'savepoint' | 'rollback to savepoint' | 'release savepoint',\n  savepointName: string\n) {\n  return sql.raw(`${command} ${savepointName}`).execute(trx);\n}\n\nexport async function createTransactionSavepoint(\n  trx: ActiveTransaction,\n  savepointName: string\n): Promise<ActiveTransaction> {\n  await runSavepointCommand(trx, 'savepoint', savepointName);\n  return trx;\n}\n\nexport async function rollbackTransactionSavepoint(\n  trx: ActiveTransaction,\n  savepointName: string\n): Promise<ActiveTransaction> {\n  await runSavepointCommand(trx, 'rollback to savepoint', savepointName);\n  return trx;\n}\n\nexport async function releaseTransactionSavepoint(\n  trx: ActiveTransaction,\n  savepointName: string\n): Promise<ActiveTransaction> {\n  await runSavepointCommand(trx, 'release savepoint', savepointName);\n  return trx;\n}\n"]}
+{"version":3,"file":"transaction.js","sourceRoot":"src/","sources":["transaction.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,GAAG,GAEJ,MAAM,cAAc,CAAC;AAItB,MAAM,UAAU,mBAAmB,CACjC,EAA4D;IAE5D,OAAO,EAAE,EAAE,aAAa,KAAK,IAAI,CAAC;AACpC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,EAA4B;IAE5B,OAAO,EAAE,CAAC,gBAAgB,EAAE,CAAC,OAAO,EAAE,CAAC;AACzC,CAAC;AAED,uEAAuE;AACvE,kDAAkD;AAClD,SAAS,mBAAmB,CAC1B,GAAgB,EAChB,OAAoE,EACpE,aAAqB;IAErB,OAAO,GAAG,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;AAC7D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,GAAsB,EACtB,aAAqB;IAErB,MAAM,mBAAmB,CAAC,GAAG,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IAC3D,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,GAAsB,EACtB,aAAqB;IAErB,MAAM,mBAAmB,CAAC,GAAG,EAAE,uBAAuB,EAAE,aAAa,CAAC,CAAC;IACvE,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAC/C,GAAsB,EACtB,aAAqB;IAErB,MAAM,mBAAmB,CAAC,GAAG,EAAE,mBAAmB,EAAE,aAAa,CAAC,CAAC;IACnE,OAAO,GAAG,CAAC;AACb,CAAC","sourcesContent":["import {\n  type ControlledTransaction,\n  type DatabaseKysely,\n  type DatabaseTransaction,\n  sql,\n  type Transaction,\n} from './isomorphic';\n\nexport type ActiveTransaction = Transaction | ControlledTransaction;\n\nexport function isActiveTransaction<Database>(\n  db: DatabaseKysely<Database> | DatabaseTransaction<Database>\n): db is DatabaseTransaction<Database> {\n  return db?.isTransaction === true;\n}\n\nexport async function startControlledTransaction<Database>(\n  db: DatabaseKysely<Database>\n) {\n  return db.startTransaction().execute();\n}\n\n// Our dialect stack supports controlled transactions, but not Kysely's\n// savepoint commands consistently across drivers.\nfunction runSavepointCommand(\n  trx: Transaction,\n  command: 'savepoint' | 'rollback to savepoint' | 'release savepoint',\n  savepointName: string\n) {\n  return sql.raw(`${command} ${savepointName}`).execute(trx);\n}\n\nexport async function createTransactionSavepoint(\n  trx: ActiveTransaction,\n  savepointName: string\n): Promise<ActiveTransaction> {\n  await runSavepointCommand(trx, 'savepoint', savepointName);\n  return trx;\n}\n\nexport async function rollbackTransactionSavepoint(\n  trx: ActiveTransaction,\n  savepointName: string\n): Promise<ActiveTransaction> {\n  await runSavepointCommand(trx, 'rollback to savepoint', savepointName);\n  return trx;\n}\n\nexport async function releaseTransactionSavepoint(\n  trx: ActiveTransaction,\n  savepointName: string\n): Promise<ActiveTransaction> {\n  await runSavepointCommand(trx, 'release savepoint', savepointName);\n  return trx;\n}\n"]}