pipework 0.1.0

package/dist/test/plugin.js

@@ -0,0 +1,63 @@
+import postgres from 'postgres';
+import { drizzle } from 'drizzle-orm/postgres-js';
+import { sql } from 'drizzle-orm';
+import { getTestDatabase } from './setup.js';
+import { createTestContext as createCtx } from '../context/create.js';
+import { runInContext } from '../context/store.js';
+import { randomUUID } from 'node:crypto';
+const contextConnections = new Map();
+export async function beginTestIsolation(instance) {
+    const dbConnections = {};
+    const contextId = randomUUID();
+    const connections = new Map();
+    for (const [name] of instance.config.databases()) {
+        const testDb = getTestDatabase(name);
+        if (!testDb) {
+            throw new Error(`[pipework] Test database not set up for "${name}". Did setupTestDatabases() run?`);
+        }
+        const conn = await createIsolatedConnection(testDb, connections, name);
+        dbConnections[name] = conn;
+    }
+    contextConnections.set(contextId, connections);
+    instance.isolation.markBegin();
+    return { id: contextId, db: dbConnections };
+}
+export async function endTestIsolation(instance, ctx) {
+    const connections = contextConnections.get(ctx.id);
+    if (!connections) {
+        instance.isolation.markEnd();
+        return;
+    }
+    for (const [, conn] of connections) {
+        try {
+            await conn.drizzle.execute(sql.raw('ROLLBACK'));
+        }
+        catch {
+            // Transaction may already be aborted
+        }
+        try {
+            await conn.client.end();
+        }
+        catch {
+            // Connection may already be closed
+        }
+    }
+    contextConnections.delete(ctx.id);
+    instance.isolation.markEnd();
+}
+async function createIsolatedConnection(testDb, connections, name) {
+    const client = postgres(testDb.url, { max: 1, onnotice: () => { } });
+    const db = drizzle(client);
+    await db.execute(sql.raw('BEGIN'));
+    connections.set(name, { client, drizzle: db });
+    return db;
+}
+export function createTestContext(instance, testConnections) {
+    const ctx = createCtx(instance, randomUUID());
+    for (const [name, drizzleDb] of Object.entries(testConnections.db)) {
+        ctx.transactions.set(name, { drizzle: drizzleDb, nonce: ctx._nonce, tenant: ctx.tenant });
+    }
+    return ctx;
+}
+export { runInContext };
+//# sourceMappingURL=plugin.js.map

package/dist/test/plugin.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"plugin.js","sourceRoot":"","sources":["../../src/test/plugin.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,UAAU,CAAA;AAC/B,OAAO,EAAE,OAAO,EAA2B,MAAM,yBAAyB,CAAA;AAC1E,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAA;AAEjC,OAAO,EAAE,eAAe,EAAqB,MAAM,YAAY,CAAA;AAC/D,OAAO,EAAE,iBAAiB,IAAI,SAAS,EAAE,MAAM,sBAAsB,CAAA;AACrE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAA;AAElD,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAgBxC,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAA6C,CAAA;AAE/E,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,QAAkB;IACzD,MAAM,aAAa,GAA4B,EAAE,CAAA;IACjD,MAAM,SAAS,GAAG,UAAU,EAAE,CAAA;IAC9B,MAAM,WAAW,GAAG,IAAI,GAAG,EAAgC,CAAA;IAE3D,KAAK,MAAM,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC;QACjD,MAAM,MAAM,GAAG,eAAe,CAAC,IAAI,CAAC,CAAA;QACpC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,4CAA4C,IAAI,kCAAkC,CAAC,CAAA;QACrG,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,WAAW,EAAE,IAAI,CAAC,CAAA;QACtE,aAAa,CAAC,IAAI,CAAC,GAAG,IAAI,CAAA;IAC5B,CAAC;IAED,kBAAkB,CAAC,GAAG,CAAC,SAAS,EAAE,WAAW,CAAC,CAAA;IAC9C,QAAQ,CAAC,SAAS,CAAC,SAAS,EAAE,CAAA;IAC9B,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,aAAa,EAAE,CAAA;AAC7C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,QAAkB,EAAE,GAAgB;IACzE,MAAM,WAAW,GAAG,kBAAkB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IAClD,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,CAAA;QAC5B,OAAM;IACR,CAAC;IAED,KAAK,MAAM,CAAC,EAAE,IAAI,CAAC,IAAI,WAAW,EAAE,CAAC;QACnC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAA;QACjD,CAAC;QAAC,MAAM,CAAC;YACP,qCAAqC;QACvC,CAAC;QACD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAA;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,mCAAmC;QACrC,CAAC;IACH,CAAC;IAED,kBAAkB,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IACjC,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,CAAA;AAC9B,CAAC;AAED,KAAK,UAAU,wBAAwB,CACrC,MAAoB,EACpB,WAA8C,EAC9C,IAAY;IAEZ,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC,CAAA;IACnE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAA;IAE1B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAA;IAElC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAA;IAC9C,OAAO,EAAE,CAAA;AACX,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,QAAkB,EAAE,eAA4B;IAChF,MAAM,GAAG,GAAG,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAA;IAC7C,KAAK,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC,EAAE,CAAC;QACnE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAA;IAC3F,CAAC;IACD,OAAO,GAAG,CAAA;AACZ,CAAC;AAED,OAAO,EAAE,YAAY,EAAE,CAAA"}

package/dist/test/setup.d.ts

@@ -0,0 +1,11 @@
+import type { Manifold } from '../pipework.js';
+export interface TestDatabase {
+    name: string;
+    url: string;
+    templateDb: string;
+}
+export declare function setupTestDatabases(instance: Manifold): Promise<Map<string, TestDatabase>>;
+export declare function teardownTestDatabases(): Promise<void>;
+export declare function getTestDatabase(name: string): TestDatabase | undefined;
+export declare function getTestDatabases(): ReadonlyMap<string, TestDatabase>;
+//# sourceMappingURL=setup.d.ts.map

package/dist/test/setup.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup.d.ts","sourceRoot":"","sources":["../../src/test/setup.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAA;AAI9C,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,EAAE,MAAM,CAAA;IACX,UAAU,EAAE,MAAM,CAAA;CACnB;AAeD,wBAAsB,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,CAO/F;AAwCD,wBAAsB,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC,CAkB3D;AAED,wBAAgB,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS,CAEtE;AAED,wBAAgB,gBAAgB,IAAI,WAAW,CAAC,MAAM,EAAE,YAAY,CAAC,CAEpE"}

package/dist/test/setup.js

@@ -0,0 +1,76 @@
+import postgres from 'postgres';
+import { drizzle } from 'drizzle-orm/postgres-js';
+import { sql } from 'drizzle-orm';
+import { migrate as drizzleMigrate } from 'drizzle-orm/postgres-js/migrator';
+import { quoteIdentifier } from '../db/identifiers.js';
+const testDatabases = new Map();
+let adminClient = null;
+function getAdminUrl(originalUrl) {
+    const url = new URL(originalUrl);
+    url.pathname = '/postgres';
+    return url.toString();
+}
+function testDbName(baseName) {
+    return `pipework_test_${baseName}_${process.pid}`;
+}
+export async function setupTestDatabases(instance) {
+    for (const [name, dbConfig] of instance.config.databases()) {
+        const testDb = await createTestDatabase(name, dbConfig);
+        testDatabases.set(name, testDb);
+    }
+    return testDatabases;
+}
+async function createTestDatabase(name, dbConfig) {
+    const adminUrl = getAdminUrl(dbConfig.url);
+    if (!adminClient) {
+        adminClient = postgres(adminUrl, { max: 1, onnotice: () => { } });
+    }
+    const dbName = testDbName(name);
+    const dbIdent = quoteIdentifier(dbName, 'database');
+    const admin = drizzle(adminClient);
+    await admin.execute(sql.raw(`DROP DATABASE IF EXISTS ${dbIdent}`));
+    await admin.execute(sql.raw(`CREATE DATABASE ${dbIdent}`));
+    const testUrl = new URL(dbConfig.url);
+    testUrl.pathname = `/${dbName}`;
+    const testUrlStr = testUrl.toString();
+    const testClient = postgres(testUrlStr, { max: 1, onnotice: () => { } });
+    const testDb = drizzle(testClient);
+    if (dbConfig.extensions.length > 0) {
+        for (const ext of dbConfig.extensions) {
+            await testDb.execute(sql.raw(`CREATE EXTENSION IF NOT EXISTS ${quoteIdentifier(ext, 'extension')}`));
+        }
+    }
+    if (dbConfig.migrations !== undefined) {
+        await drizzleMigrate(testDb, {
+            migrationsFolder: dbConfig.migrations,
+            migrationsTable: dbConfig.migrationTable,
+        });
+    }
+    await testClient.end();
+    return { name, url: testUrlStr, templateDb: dbName };
+}
+export async function teardownTestDatabases() {
+    if (!adminClient)
+        return;
+    const admin = drizzle(adminClient);
+    for (const [, testDb] of testDatabases) {
+        try {
+            const dbName = quoteIdentifier(testDb.templateDb, 'database');
+            await admin.execute(sql `SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = ${testDb.templateDb} AND pid != pg_backend_pid()`);
+            await admin.execute(sql.raw(`DROP DATABASE IF EXISTS ${dbName}`));
+        }
+        catch {
+            // Best effort cleanup
+        }
+    }
+    testDatabases.clear();
+    await adminClient.end();
+    adminClient = null;
+}
+export function getTestDatabase(name) {
+    return testDatabases.get(name);
+}
+export function getTestDatabases() {
+    return testDatabases;
+}
+//# sourceMappingURL=setup.js.map

package/dist/test/setup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup.js","sourceRoot":"","sources":["../../src/test/setup.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,UAAU,CAAA;AAC/B,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAA;AACjD,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAA;AACjC,OAAO,EAAE,OAAO,IAAI,cAAc,EAAE,MAAM,kCAAkC,CAAA;AAG5E,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAA;AAQtD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAwB,CAAA;AACrD,IAAI,WAAW,GAAwB,IAAI,CAAA;AAE3C,SAAS,WAAW,CAAC,WAAmB;IACtC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,CAAA;IAChC,GAAG,CAAC,QAAQ,GAAG,WAAW,CAAA;IAC1B,OAAO,GAAG,CAAC,QAAQ,EAAE,CAAA;AACvB,CAAC;AAED,SAAS,UAAU,CAAC,QAAgB;IAClC,OAAO,iBAAiB,QAAQ,IAAI,OAAO,CAAC,GAAG,EAAE,CAAA;AACnD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,QAAkB;IACzD,KAAK,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC;QAC3D,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;QACvD,aAAa,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;IACjC,CAAC;IAED,OAAO,aAAa,CAAA;AACtB,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,IAAY,EAAE,QAA0B;IACxE,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAA;IAC1C,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,WAAW,GAAG,QAAQ,CAAC,QAAQ,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC,CAAA;IAClE,CAAC;IAED,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAA;IAC/B,MAAM,OAAO,GAAG,eAAe,CAAC,MAAM,EAAE,UAAU,CAAC,CAAA;IACnD,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC,CAAA;IAElC,MAAM,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,2BAA2B,OAAO,EAAE,CAAC,CAAC,CAAA;IAClE,MAAM,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,mBAAmB,OAAO,EAAE,CAAC,CAAC,CAAA;IAE1D,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAA;IACrC,OAAO,CAAC,QAAQ,GAAG,IAAI,MAAM,EAAE,CAAA;IAC/B,MAAM,UAAU,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAA;IAErC,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAU,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC,CAAA;IACvE,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAAA;IAElC,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACnC,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YACtC,MAAM,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,kCAAkC,eAAe,CAAC,GAAG,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC,CAAA;QACtG,CAAC;IACH,CAAC;IAED,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QACtC,MAAM,cAAc,CAAC,MAAM,EAAE;YAC3B,gBAAgB,EAAE,QAAQ,CAAC,UAAU;YACrC,eAAe,EAAE,QAAQ,CAAC,cAAc;SACzC,CAAC,CAAA;IACJ,CAAC;IAED,MAAM,UAAU,CAAC,GAAG,EAAE,CAAA;IAEtB,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,EAAE,CAAA;AACtD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB;IACzC,IAAI,CAAC,WAAW;QAAE,OAAM;IAExB,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC,CAAA;IAElC,KAAK,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;QACvC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC,UAAU,EAAE,UAAU,CAAC,CAAA;YAC7D,MAAM,KAAK,CAAC,OAAO,CAAC,GAAG,CAAA,0EAA0E,MAAM,CAAC,UAAU,8BAA8B,CAAC,CAAA;YACjJ,MAAM,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,2BAA2B,MAAM,EAAE,CAAC,CAAC,CAAA;QACnE,CAAC;QAAC,MAAM,CAAC;YACP,sBAAsB;QACxB,CAAC;IACH,CAAC;IAED,aAAa,CAAC,KAAK,EAAE,CAAA;IACrB,MAAM,WAAW,CAAC,GAAG,EAAE,CAAA;IACvB,WAAW,GAAG,IAAI,CAAA;AACpB,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,IAAY;IAC1C,OAAO,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;AAChC,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,OAAO,aAAa,CAAA;AACtB,CAAC"}

package/dist/test/vitest.d.ts

@@ -0,0 +1,9 @@
+import type { DB } from '../db/index.js';
+import type { Blueprint } from '../config/index.js';
+import type { Manifold } from '../pipework.js';
+export { type TestContext, type TestDatabaseConnections } from './plugin.js';
+export declare function setupPipeworkTests(configOrInstance: Blueprint | Manifold): Manifold;
+export declare function useTestDb(name?: string): DB;
+export declare function useTestInstance(): Manifold;
+export declare function withFlow<T>(fn: () => T): T;
+//# sourceMappingURL=vitest.d.ts.map

package/dist/test/vitest.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vitest.d.ts","sourceRoot":"","sources":["../../src/test/vitest.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,gBAAgB,CAAA;AACxC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAA;AACnD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAA;AAK9C,OAAO,EAAE,KAAK,WAAW,EAAE,KAAK,uBAAuB,EAAE,MAAM,aAAa,CAAA;AAK5E,wBAAgB,kBAAkB,CAAC,gBAAgB,EAAE,SAAS,GAAG,QAAQ,GAAG,QAAQ,CAyBnF;AAED,wBAAgB,SAAS,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,EAAE,CAiB3C;AAED,wBAAgB,eAAe,IAAI,QAAQ,CAK1C;AAED,wBAAgB,QAAQ,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,CAM1C"}

package/dist/test/vitest.js

@@ -0,0 +1,58 @@
+import { beforeAll, afterAll, beforeEach, afterEach } from 'vitest';
+import { createPipework } from '../pipework.js';
+import { setupTestDatabases, teardownTestDatabases } from './setup.js';
+import { beginTestIsolation, endTestIsolation, createTestContext, runInContext } from './plugin.js';
+export {} from './plugin.js';
+let _testContext = null;
+let _instance = null;
+export function setupPipeworkTests(configOrInstance) {
+    const instance = isInstance(configOrInstance) ? configOrInstance : createPipework(configOrInstance);
+    _instance = instance;
+    beforeAll(async () => {
+        await setupTestDatabases(instance);
+    });
+    afterAll(async () => {
+        await instance.pool.closeAll();
+        await teardownTestDatabases();
+    });
+    beforeEach(async () => {
+        _testContext = await beginTestIsolation(instance);
+    });
+    afterEach(async () => {
+        if (_testContext) {
+            await endTestIsolation(instance, _testContext);
+            _testContext = null;
+        }
+    });
+    return instance;
+}
+export function useTestDb(name) {
+    if (!_testContext || !_instance) {
+        throw new Error('[pipework] No test context available. Are you inside a test that called setupPipeworkTests()?\n\n' +
+            '  Hint: Call setupPipeworkTests(config) at the top of your test file or in a setup file.\n');
+    }
+    const resolvedName = name ?? _instance.config.defaultDatabaseName();
+    const db = _testContext.db[resolvedName];
+    if (!db) {
+        const available = Object.keys(_testContext.db).join(', ');
+        throw new Error(`[pipework] No test database "${resolvedName}". Available: ${available}`);
+    }
+    return db;
+}
+export function useTestInstance() {
+    if (!_instance) {
+        throw new Error('[pipework] No test instance. Call setupPipeworkTests() first.');
+    }
+    return _instance;
+}
+export function withFlow(fn) {
+    if (!_testContext || !_instance) {
+        throw new Error('[pipework] No test context. Call setupPipeworkTests() first.');
+    }
+    const ctx = createTestContext(_instance, _testContext);
+    return runInContext(ctx, fn);
+}
+function isInstance(v) {
+    return 'pool' in v && 'config' in v;
+}
+//# sourceMappingURL=vitest.js.map

package/dist/test/vitest.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vitest.js","sourceRoot":"","sources":["../../src/test/vitest.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAA;AAInE,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAA;AAC/C,OAAO,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAA;AACtE,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,YAAY,EAAoB,MAAM,aAAa,CAAA;AAErH,OAAO,EAAkD,MAAM,aAAa,CAAA;AAE5E,IAAI,YAAY,GAAuB,IAAI,CAAA;AAC3C,IAAI,SAAS,GAAoB,IAAI,CAAA;AAErC,MAAM,UAAU,kBAAkB,CAAC,gBAAsC;IACvE,MAAM,QAAQ,GAAG,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAA;IACnG,SAAS,GAAG,QAAQ,CAAA;IAEpB,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAA;IACpC,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAA;QAC9B,MAAM,qBAAqB,EAAE,CAAA;IAC/B,CAAC,CAAC,CAAA;IAEF,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,YAAY,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAA;IACnD,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,gBAAgB,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAA;YAC9C,YAAY,GAAG,IAAI,CAAA;QACrB,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,OAAO,QAAQ,CAAA;AACjB,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,IAAa;IACrC,IAAI,CAAC,YAAY,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CACb,mGAAmG;YACnG,4FAA4F,CAC7F,CAAA;IACH,CAAC;IAED,MAAM,YAAY,GAAG,IAAI,IAAI,SAAS,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAA;IACnE,MAAM,EAAE,GAAG,YAAY,CAAC,EAAE,CAAC,YAAY,CAAC,CAAA;IACxC,IAAI,CAAC,EAAE,EAAE,CAAC;QACR,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACzD,MAAM,IAAI,KAAK,CACb,gCAAgC,YAAY,iBAAiB,SAAS,EAAE,CACzE,CAAA;IACH,CAAC;IACD,OAAO,EAAE,CAAA;AACX,CAAC;AAED,MAAM,UAAU,eAAe;IAC7B,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAA;IAClF,CAAC;IACD,OAAO,SAAS,CAAA;AAClB,CAAC;AAED,MAAM,UAAU,QAAQ,CAAI,EAAW;IACrC,IAAI,CAAC,YAAY,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAA;IACjF,CAAC;IACD,MAAM,GAAG,GAAG,iBAAiB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAA;IACtD,OAAO,YAAY,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;AAC9B,CAAC;AAED,SAAS,UAAU,CAAC,CAAuB;IACzC,OAAO,MAAM,IAAI,CAAC,IAAI,QAAQ,IAAI,CAAC,CAAA;AACrC,CAAC"}

package/dist/trunk.d.ts

@@ -0,0 +1,35 @@
+import type { ResolvedConfig } from './config/load.js';
+import type { ResolvedDatabase } from './config/load.js';
+import type { TrunkConfig } from './config/schema.js';
+import type { ManagedConnection } from './db/pool.js';
+export interface TrunkInstance {
+    readonly config: ResolvedConfig;
+    readonly pool: PoolState;
+    readonly isolation: IsolationState;
+    readonly connections: ConnectionTracker;
+}
+export interface PoolState {
+    getOrCreate(config: ResolvedDatabase): ManagedConnection;
+    get(name: string): ManagedConnection | undefined;
+    closeAll(): Promise<void>;
+    count(): number;
+}
+export interface IsolationState {
+    markBegin(): void;
+    markEnd(): void;
+    assertClean(): void;
+    stats(): {
+        active: number;
+        totalBegins: number;
+        totalEnds: number;
+    };
+    reset(): void;
+}
+export interface ConnectionTracker {
+    track(id: string, name: string, context: string): void;
+    release(id: string): void;
+    activeCount(): number;
+    assertNoLeaks(maxExpected: number): void;
+}
+export declare function createTrunk(raw: TrunkConfig): TrunkInstance;
+//# sourceMappingURL=trunk.d.ts.map

package/dist/trunk.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trunk.d.ts","sourceRoot":"","sources":["../src/trunk.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAA;AACtD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAA;AAExD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AACrD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAA;AAIrD,MAAM,WAAW,aAAa;IAC5B,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAA;IAC/B,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAA;IACxB,QAAQ,CAAC,SAAS,EAAE,cAAc,CAAA;IAClC,QAAQ,CAAC,WAAW,EAAE,iBAAiB,CAAA;CACxC;AAED,MAAM,WAAW,SAAS;IACxB,WAAW,CAAC,MAAM,EAAE,gBAAgB,GAAG,iBAAiB,CAAA;IACxD,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,iBAAiB,GAAG,SAAS,CAAA;IAChD,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,CAAA;IACzB,KAAK,IAAI,MAAM,CAAA;CAChB;AAED,MAAM,WAAW,cAAc;IAC7B,SAAS,IAAI,IAAI,CAAA;IACjB,OAAO,IAAI,IAAI,CAAA;IACf,WAAW,IAAI,IAAI,CAAA;IACnB,KAAK,IAAI;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAA;IACnE,KAAK,IAAI,IAAI,CAAA;CACd;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAA;IACtD,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI,CAAA;IACzB,WAAW,IAAI,MAAM,CAAA;IACrB,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI,CAAA;CACzC;AAED,wBAAgB,WAAW,CAAC,GAAG,EAAE,WAAW,GAAG,aAAa,CAO3D"}

package/dist/trunk.js

@@ -0,0 +1,102 @@
+import { loadConfig as parseConfig } from './config/load.js';
+import { createManagedConnection } from './db/pool.js';
+import { InvariantViolation } from './invariants/assert.js';
+export function createTrunk(raw) {
+    const config = parseConfig(raw);
+    const pool = createPoolState();
+    const isolation = createIsolationState();
+    const connections = createConnectionTracker();
+    return { config, pool, isolation, connections };
+}
+function createPoolState() {
+    const map = new Map();
+    return {
+        getOrCreate(config) {
+            const existing = map.get(config.name);
+            if (existing)
+                return existing;
+            const managed = createManagedConnection(config, () => map.delete(config.name));
+            map.set(config.name, managed);
+            return managed;
+        },
+        get(name) {
+            return map.get(name);
+        },
+        async closeAll() {
+            const closing = [...map.values()].map(c => c.close());
+            await Promise.all(closing);
+        },
+        count() {
+            return map.size;
+        },
+    };
+}
+function createIsolationState() {
+    let active = 0;
+    let totalBegins = 0;
+    let totalEnds = 0;
+    return {
+        markBegin() {
+            active++;
+            totalBegins++;
+        },
+        markEnd() {
+            if (active <= 0) {
+                throw new InvariantViolation('isolation-end-without-begin', {
+                    activeIsolationContexts: active,
+                    totalBegins,
+                    totalEnds,
+                });
+            }
+            active--;
+            totalEnds++;
+        },
+        assertClean() {
+            if (active !== 0) {
+                throw new InvariantViolation('isolation-context-leak', {
+                    activeIsolationContexts: active,
+                    totalBegins,
+                    totalEnds,
+                    delta: totalBegins - totalEnds,
+                });
+            }
+        },
+        stats() {
+            return { active, totalBegins, totalEnds };
+        },
+        reset() {
+            active = 0;
+            totalBegins = 0;
+            totalEnds = 0;
+        },
+    };
+}
+function createConnectionTracker() {
+    const tracked = new Map();
+    return {
+        track(id, name, context) {
+            tracked.set(id, { name, createdAt: Date.now(), context });
+        },
+        release(id) {
+            tracked.delete(id);
+        },
+        activeCount() {
+            return tracked.size;
+        },
+        assertNoLeaks(maxExpected) {
+            if (tracked.size > maxExpected) {
+                throw new InvariantViolation('connection-pool-bounded', {
+                    active: tracked.size,
+                    maxExpected,
+                    connections: [...tracked.entries()].map(([id, c]) => ({
+                        id,
+                        name: c.name,
+                        age_ms: Date.now() - c.createdAt,
+                        context: c.context,
+                    })),
+                });
+            }
+        },
+    };
+}
+//# sourceMappingURL=trunk.js.map

package/dist/trunk.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"trunk.js","sourceRoot":"","sources":["../src/trunk.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,IAAI,WAAW,EAAE,MAAM,kBAAkB,CAAA;AAG5D,OAAO,EAAE,uBAAuB,EAAE,MAAM,cAAc,CAAA;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAA;AA+B3D,MAAM,UAAU,WAAW,CAAC,GAAgB;IAC1C,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,CAAA;IAC/B,MAAM,IAAI,GAAG,eAAe,EAAE,CAAA;IAC9B,MAAM,SAAS,GAAG,oBAAoB,EAAE,CAAA;IACxC,MAAM,WAAW,GAAG,uBAAuB,EAAE,CAAA;IAE7C,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAA;AACjD,CAAC;AAED,SAAS,eAAe;IACtB,MAAM,GAAG,GAAG,IAAI,GAAG,EAA6B,CAAA;IAEhD,OAAO;QACL,WAAW,CAAC,MAAwB;YAClC,MAAM,QAAQ,GAAG,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YACrC,IAAI,QAAQ;gBAAE,OAAO,QAAQ,CAAA;YAE7B,MAAM,OAAO,GAAG,uBAAuB,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAA;YAC9E,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,CAAA;YAC7B,OAAO,OAAO,CAAA;QAChB,CAAC;QAED,GAAG,CAAC,IAAY;YACd,OAAO,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QACtB,CAAC;QAED,KAAK,CAAC,QAAQ;YACZ,MAAM,OAAO,GAAG,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAA;YACrD,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QAC5B,CAAC;QAED,KAAK;YACH,OAAO,GAAG,CAAC,IAAI,CAAA;QACjB,CAAC;KACF,CAAA;AACH,CAAC;AAED,SAAS,oBAAoB;IAC3B,IAAI,MAAM,GAAG,CAAC,CAAA;IACd,IAAI,WAAW,GAAG,CAAC,CAAA;IACnB,IAAI,SAAS,GAAG,CAAC,CAAA;IAEjB,OAAO;QACL,SAAS;YACP,MAAM,EAAE,CAAA;YACR,WAAW,EAAE,CAAA;QACf,CAAC;QAED,OAAO;YACL,IAAI,MAAM,IAAI,CAAC,EAAE,CAAC;gBAChB,MAAM,IAAI,kBAAkB,CAAC,6BAA6B,EAAE;oBAC1D,uBAAuB,EAAE,MAAM;oBAC/B,WAAW;oBACX,SAAS;iBACV,CAAC,CAAA;YACJ,CAAC;YACD,MAAM,EAAE,CAAA;YACR,SAAS,EAAE,CAAA;QACb,CAAC;QAED,WAAW;YACT,IAAI,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjB,MAAM,IAAI,kBAAkB,CAAC,wBAAwB,EAAE;oBACrD,uBAAuB,EAAE,MAAM;oBAC/B,WAAW;oBACX,SAAS;oBACT,KAAK,EAAE,WAAW,GAAG,SAAS;iBAC/B,CAAC,CAAA;YACJ,CAAC;QACH,CAAC;QAED,KAAK;YACH,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,CAAA;QAC3C,CAAC;QAED,KAAK;YACH,MAAM,GAAG,CAAC,CAAA;YACV,WAAW,GAAG,CAAC,CAAA;YACf,SAAS,GAAG,CAAC,CAAA;QACf,CAAC;KACF,CAAA;AACH,CAAC;AAED,SAAS,uBAAuB;IAC9B,MAAM,OAAO,GAAG,IAAI,GAAG,EAAgE,CAAA;IAEvF,OAAO;QACL,KAAK,CAAC,EAAU,EAAE,IAAY,EAAE,OAAe;YAC7C,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,CAAC,CAAA;QAC3D,CAAC;QAED,OAAO,CAAC,EAAU;YAChB,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;QACpB,CAAC;QAED,WAAW;YACT,OAAO,OAAO,CAAC,IAAI,CAAA;QACrB,CAAC;QAED,aAAa,CAAC,WAAmB;YAC/B,IAAI,OAAO,CAAC,IAAI,GAAG,WAAW,EAAE,CAAC;gBAC/B,MAAM,IAAI,kBAAkB,CAAC,yBAAyB,EAAE;oBACtD,MAAM,EAAE,OAAO,CAAC,IAAI;oBACpB,WAAW;oBACX,WAAW,EAAE,CAAC,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBACpD,EAAE;wBACF,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,SAAS;wBAChC,OAAO,EAAE,CAAC,CAAC,OAAO;qBACnB,CAAC,CAAC;iBACJ,CAAC,CAAA;YACJ,CAAC;QACH,CAAC;KACF,CAAA;AACH,CAAC"}

package/dist/vector/index.d.ts

@@ -0,0 +1,4 @@
+export { vector, tsvector } from './types.js';
+export { cosineDistance, cosineSimilarity, l2Distance, innerProduct, toTsvector, toTsquery, tsRank } from './query.js';
+export { validateVectorExtension } from './validate.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/vector/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/vector/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAC7C,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,UAAU,EAAE,YAAY,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,YAAY,CAAA;AACtH,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAA"}

package/dist/vector/index.js

@@ -0,0 +1,4 @@
+export { vector, tsvector } from './types.js';
+export { cosineDistance, cosineSimilarity, l2Distance, innerProduct, toTsvector, toTsquery, tsRank } from './query.js';
+export { validateVectorExtension } from './validate.js';
+//# sourceMappingURL=index.js.map

package/dist/vector/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/vector/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAC7C,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,UAAU,EAAE,YAAY,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,YAAY,CAAA;AACtH,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAA"}

package/dist/vector/namespace.d.ts

@@ -0,0 +1,16 @@
+import { vector as vectorColumn, tsvector } from './types.js';
+import { cosineDistance, cosineSimilarity, l2Distance, innerProduct, toTsvector, toTsquery, tsRank } from './query.js';
+import { validateVectorExtension } from './validate.js';
+export declare const vector: {
+    column: typeof vectorColumn;
+    tsvector: typeof tsvector;
+    cosine: typeof cosineDistance;
+    cosineSimilarity: typeof cosineSimilarity;
+    l2: typeof l2Distance;
+    innerProduct: typeof innerProduct;
+    toTsvector: typeof toTsvector;
+    toTsquery: typeof toTsquery;
+    rank: typeof tsRank;
+    validate: typeof validateVectorExtension;
+};
+//# sourceMappingURL=namespace.d.ts.map

package/dist/vector/namespace.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"namespace.d.ts","sourceRoot":"","sources":["../../src/vector/namespace.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,YAAY,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAC7D,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,UAAU,EAAE,YAAY,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,YAAY,CAAA;AACtH,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAA;AAEvD,eAAO,MAAM,MAAM;;;;;;;;;;;CAWlB,CAAA"}

package/dist/vector/namespace.js

@@ -0,0 +1,16 @@
+import { vector as vectorColumn, tsvector } from './types.js';
+import { cosineDistance, cosineSimilarity, l2Distance, innerProduct, toTsvector, toTsquery, tsRank } from './query.js';
+import { validateVectorExtension } from './validate.js';
+export const vector = {
+    column: vectorColumn,
+    tsvector,
+    cosine: cosineDistance,
+    cosineSimilarity,
+    l2: l2Distance,
+    innerProduct,
+    toTsvector,
+    toTsquery,
+    rank: tsRank,
+    validate: validateVectorExtension,
+};
+//# sourceMappingURL=namespace.js.map

package/dist/vector/namespace.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"namespace.js","sourceRoot":"","sources":["../../src/vector/namespace.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,YAAY,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAC7D,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,UAAU,EAAE,YAAY,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,YAAY,CAAA;AACtH,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAA;AAEvD,MAAM,CAAC,MAAM,MAAM,GAAG;IACpB,MAAM,EAAE,YAAY;IACpB,QAAQ;IACR,MAAM,EAAE,cAAc;IACtB,gBAAgB;IAChB,EAAE,EAAE,UAAU;IACd,YAAY;IACZ,UAAU;IACV,SAAS;IACT,IAAI,EAAE,MAAM;IACZ,QAAQ,EAAE,uBAAuB;CAClC,CAAA"}

package/dist/vector/query.d.ts

@@ -0,0 +1,9 @@
+import type { SQL, Column } from '../db/types.js';
+export declare function cosineDistance(column: Column, query: number[]): SQL<number>;
+export declare function cosineSimilarity(column: Column, query: number[]): SQL<number>;
+export declare function l2Distance(column: Column, query: number[]): SQL<number>;
+export declare function innerProduct(column: Column, query: number[]): SQL<number>;
+export declare function toTsvector(text: string | SQL, config?: string): SQL;
+export declare function toTsquery(query: string, config?: string): SQL;
+export declare function tsRank(column: Column, query: string, config?: string): SQL<number>;
+//# sourceMappingURL=query.d.ts.map

package/dist/vector/query.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query.d.ts","sourceRoot":"","sources":["../../src/vector/query.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AAEjD,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,CAG3E;AAED,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,CAG7E;AAED,wBAAgB,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,CAGvE;AAED,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,CAGzE;AAED,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,GAAG,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,GAAG,CAMnE;AAED,wBAAgB,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,GAAG,CAG7D;AAED,wBAAgB,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,CAGlF"}

package/dist/vector/query.js

@@ -0,0 +1,33 @@
+import { sql } from '../db/sql.js';
+export function cosineDistance(column, query) {
+    const vectorStr = `[${query.join(',')}]`;
+    return sql `${column} <=> ${vectorStr}::vector`;
+}
+export function cosineSimilarity(column, query) {
+    const vectorStr = `[${query.join(',')}]`;
+    return sql `1 - (${column} <=> ${vectorStr}::vector)`;
+}
+export function l2Distance(column, query) {
+    const vectorStr = `[${query.join(',')}]`;
+    return sql `${column} <-> ${vectorStr}::vector`;
+}
+export function innerProduct(column, query) {
+    const vectorStr = `[${query.join(',')}]`;
+    return sql `${column} <#> ${vectorStr}::vector`;
+}
+export function toTsvector(text, config) {
+    const tsConfig = config ?? 'english';
+    if (typeof text === 'string') {
+        return sql `to_tsvector(${tsConfig}, ${text})`;
+    }
+    return sql `to_tsvector(${tsConfig}, ${text})`;
+}
+export function toTsquery(query, config) {
+    const tsConfig = config ?? 'english';
+    return sql `to_tsquery(${tsConfig}, ${query})`;
+}
+export function tsRank(column, query, config) {
+    const tsConfig = config ?? 'english';
+    return sql `ts_rank(${column}, to_tsquery(${tsConfig}, ${query}))`;
+}
+//# sourceMappingURL=query.js.map

package/dist/vector/query.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query.js","sourceRoot":"","sources":["../../src/vector/query.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,cAAc,CAAA;AAGlC,MAAM,UAAU,cAAc,CAAC,MAAc,EAAE,KAAe;IAC5D,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAA;IACxC,OAAO,GAAG,CAAQ,GAAG,MAAM,QAAQ,SAAS,UAAU,CAAA;AACxD,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAc,EAAE,KAAe;IAC9D,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAA;IACxC,OAAO,GAAG,CAAQ,QAAQ,MAAM,QAAQ,SAAS,WAAW,CAAA;AAC9D,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,MAAc,EAAE,KAAe;IACxD,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAA;IACxC,OAAO,GAAG,CAAQ,GAAG,MAAM,QAAQ,SAAS,UAAU,CAAA;AACxD,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,MAAc,EAAE,KAAe;IAC1D,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAA;IACxC,OAAO,GAAG,CAAQ,GAAG,MAAM,QAAQ,SAAS,UAAU,CAAA;AACxD,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,IAAkB,EAAE,MAAe;IAC5D,MAAM,QAAQ,GAAG,MAAM,IAAI,SAAS,CAAA;IACpC,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO,GAAG,CAAA,eAAe,QAAQ,KAAK,IAAI,GAAG,CAAA;IAC/C,CAAC;IACD,OAAO,GAAG,CAAA,eAAe,QAAQ,KAAK,IAAI,GAAG,CAAA;AAC/C,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,KAAa,EAAE,MAAe;IACtD,MAAM,QAAQ,GAAG,MAAM,IAAI,SAAS,CAAA;IACpC,OAAO,GAAG,CAAA,cAAc,QAAQ,KAAK,KAAK,GAAG,CAAA;AAC/C,CAAC;AAED,MAAM,UAAU,MAAM,CAAC,MAAc,EAAE,KAAa,EAAE,MAAe;IACnE,MAAM,QAAQ,GAAG,MAAM,IAAI,SAAS,CAAA;IACpC,OAAO,GAAG,CAAQ,WAAW,MAAM,gBAAgB,QAAQ,KAAK,KAAK,IAAI,CAAA;AAC3E,CAAC"}

package/dist/vector/types.d.ts

@@ -0,0 +1,19 @@
+export declare function vector(name: string, config: {
+    dimensions: number;
+}): import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: number[];
+    driverParam: unknown;
+    enumValues: undefined;
+}>;
+export declare function tsvector(name: string): import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: unknown;
+    driverParam: unknown;
+    enumValues: undefined;
+}>;
+//# sourceMappingURL=types.d.ts.map

package/dist/vector/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/vector/types.ts"],"names":[],"mappings":"AAEA,wBAAgB,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE;IAAE,UAAU,EAAE,MAAM,CAAA;CAAE;;;;;;;GAgBlE;AAED,wBAAgB,QAAQ,CAAC,IAAI,EAAE,MAAM;;;;;;;GAMpC"}

package/dist/vector/types.js

@@ -0,0 +1,26 @@
+import { customType } from '../db/col/index.js';
+export function vector(name, config) {
+    return customType({
+        dataType() {
+            return `vector(${config.dimensions})`;
+        },
+        toDriver(value) {
+            return `[${value.join(',')}]`;
+        },
+        fromDriver(value) {
+            if (typeof value === 'string') {
+                const trimmed = value.startsWith('[') ? value.slice(1, -1) : value;
+                return trimmed.split(',').map(Number);
+            }
+            return value;
+        },
+    })(name);
+}
+export function tsvector(name) {
+    return customType({
+        dataType() {
+            return 'tsvector';
+        },
+    })(name);
+}
+//# sourceMappingURL=types.js.map

package/dist/vector/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/vector/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAA;AAE/C,MAAM,UAAU,MAAM,CAAC,IAAY,EAAE,MAA8B;IACjE,OAAO,UAAU,CAA0C;QACzD,QAAQ;YACN,OAAO,UAAU,MAAM,CAAC,UAAU,GAAG,CAAA;QACvC,CAAC;QACD,QAAQ,CAAC,KAAe;YACtB,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAA;QAC/B,CAAC;QACD,UAAU,CAAC,KAAc;YACvB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAA;gBAClE,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;YACvC,CAAC;YACD,OAAO,KAAiB,CAAA;QAC1B,CAAC;KACF,CAAC,CAAC,IAAI,CAAC,CAAA;AACV,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,IAAY;IACnC,OAAO,UAAU,CAA0C;QACzD,QAAQ;YACN,OAAO,UAAU,CAAA;QACnB,CAAC;KACF,CAAC,CAAC,IAAI,CAAC,CAAA;AACV,CAAC"}

package/dist/vector/validate.d.ts

@@ -0,0 +1,3 @@
+import type { DB } from '../db/index.js';
+export declare function validateVectorExtension(db: DB): Promise<void>;
+//# sourceMappingURL=validate.d.ts.map

package/dist/vector/validate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"validate.d.ts","sourceRoot":"","sources":["../../src/vector/validate.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,gBAAgB,CAAA;AAExC,wBAAsB,uBAAuB,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAYnE"}

package/dist/vector/validate.js

@@ -0,0 +1,10 @@
+import { sql } from '../db/sql.js';
+export async function validateVectorExtension(db) {
+    const result = await db.execute(sql `SELECT extname FROM pg_extension WHERE extname = 'vector'`);
+    if (result.length === 0) {
+        throw new Error('[pipework] pgvector extension not installed on this database.\n' +
+            "Run: CREATE EXTENSION IF NOT EXISTS vector;\n" +
+            "Or add 'vector' to your database's extensions list in pipework.config.ts.");
+    }
+}
+//# sourceMappingURL=validate.js.map

package/dist/vector/validate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validate.js","sourceRoot":"","sources":["../../src/vector/validate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,cAAc,CAAA;AAGlC,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,EAAM;IAClD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAC7B,GAAG,CAAA,2DAA2D,CAC/D,CAAA;IAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CACb,iEAAiE;YACjE,+CAA+C;YAC/C,2EAA2E,CAC5E,CAAA;IACH,CAAC;AACH,CAAC"}