callman-core 1.44.0 → 1.45.0

package/dist-cjs/host/db/dbSessionManager.js

@@ -0,0 +1,172 @@
+"use strict";
+// Run-scoped DB sessions for scenario db nodes (`useSession`), on every SQL
+// engine.
+//
+// One live connection per (runId, connection identity). All session-joined
+// nodes of a run share it, so transactions, temp tables, package state and
+// session variables survive between nodes — the foundation of the
+// seed → call procedure → assert → ROLLBACK testing pattern.
+//
+// Modeled on callman-backend's kafka-client-pool: lazy open, per-session
+// serialization, idle sweeper, explicit shutdown hook.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.__getDbSessionCountForTests = exports.closeAllDbSessions = exports.releaseDbSessionsForRun = exports.executeInDbSession = void 0;
+const mssqlAdapter_js_1 = require("./mssqlAdapter.js");
+const mysqlAdapter_js_1 = require("./mysqlAdapter.js");
+const oracleAdapter_js_1 = require("./oracleAdapter.js");
+const pgSession_js_1 = require("./pgSession.js");
+const IDLE_SWEEP_INTERVAL_MS = 60_000;
+const IDLE_SESSION_TTL_MS = 5 * 60_000;
+const sessions = new Map();
+let sweeper = null;
+// Connection identity participates in the key so two different connections
+// in one run never share a client, even if their `id`s degenerate to the
+// same normalized placeholder.
+const sessionKey = (runId, connection) => [
+    runId,
+    connection.type,
+    connection.id,
+    connection.connectionString ?? "",
+    connection.host,
+    String(connection.port),
+    connection.database ?? "",
+    connection.username,
+].join("::");
+const ensureSweeper = () => {
+    if (sweeper)
+        return;
+    sweeper = setInterval(() => {
+        const now = Date.now();
+        for (const [key, entry] of sessions) {
+            if (now - entry.lastUsedAtMs > IDLE_SESSION_TTL_MS) {
+                sessions.delete(key);
+                void destroyEntry(entry, "rollback");
+            }
+        }
+        if (sessions.size === 0 && sweeper) {
+            clearInterval(sweeper);
+            sweeper = null;
+        }
+    }, IDLE_SWEEP_INTERVAL_MS);
+    // Never keep the host process alive just for the sweeper.
+    sweeper.unref?.();
+};
+const destroyEntry = async (entry, outcome) => {
+    try {
+        const client = await entry.clientPromise;
+        if (entry.transactionOpen && !entry.poisoned) {
+            try {
+                await (outcome === "commit" ? client.commit() : client.rollback());
+            }
+            catch {
+                // The transaction dies with the connection either way.
+            }
+        }
+        try {
+            await client.close();
+        }
+        catch {
+            // Best-effort teardown.
+        }
+    }
+    catch {
+        // Client never connected — nothing to tear down.
+    }
+};
+const openClient = (connection, entry) => {
+    switch (connection.type) {
+        case "postgres":
+            return (0, pgSession_js_1.openPgSession)(connection, (reason) => {
+                entry.poisoned = reason;
+            });
+        case "oracle":
+            return (0, oracleAdapter_js_1.openOracleSession)(connection);
+        case "mysql":
+            return (0, mysqlAdapter_js_1.openMysqlSession)(connection);
+        case "mssql":
+            return (0, mssqlAdapter_js_1.openMssqlSession)(connection);
+        default:
+            return Promise.reject(new Error(`Scenario DB sessions are only supported for SQL connections (got "${connection.type}"). Disable 'Run in scenario transaction' on this node.`));
+    }
+};
+const openEntry = (runId, connection) => {
+    const entry = {
+        chain: Promise.resolve(),
+        transactionOpen: false,
+        poisoned: null,
+        lastUsedAtMs: Date.now(),
+        runId,
+    };
+    entry.clientPromise = openClient(connection, entry);
+    // A failed connect reaches the caller through the chain below.
+    entry.clientPromise.catch(() => undefined);
+    return entry;
+};
+const executeInDbSession = async (input) => {
+    const key = sessionKey(input.runId, input.connection);
+    let entry = sessions.get(key);
+    if (!entry) {
+        entry = openEntry(input.runId, input.connection);
+        sessions.set(key, entry);
+        ensureSweeper();
+    }
+    const run = entry.chain.then(async () => {
+        const client = await entry.clientPromise;
+        if (entry.poisoned) {
+            throw new Error(`${client.label} session is unusable: ${entry.poisoned}`);
+        }
+        const inTransaction = input.transactionMode !== "none";
+        if (inTransaction && !entry.transactionOpen) {
+            await client.begin();
+            entry.transactionOpen = true;
+        }
+        const { connection: _connection, runId: _runId, transactionMode: _mode, query, ...options } = input;
+        try {
+            return await client.exec(query, options, inTransaction);
+        }
+        catch (error) {
+            if (error?.dbTimeout) {
+                entry.poisoned = error instanceof Error ? error.message : "statement cancelled";
+            }
+            throw error;
+        }
+    });
+    // The chain must survive a failed query (SQL errors are normal in
+    // expectError flows) — swallow for chaining purposes only.
+    entry.chain = run.then(() => undefined, () => undefined).then(() => {
+        entry.lastUsedAtMs = Date.now();
+    });
+    return run;
+};
+exports.executeInDbSession = executeInDbSession;
+// Close every session belonging to `runId` — COMMIT/ROLLBACK per the run
+// outcome, then close the connection. Queued statements finish (or fail) first
+// thanks to the serialization chain.
+const releaseDbSessionsForRun = async (runId, outcome) => {
+    const owned = [];
+    for (const [key, entry] of sessions) {
+        if (entry.runId === runId)
+            owned.push([key, entry]);
+    }
+    await Promise.all(owned.map(async ([key, entry]) => {
+        sessions.delete(key);
+        await entry.chain;
+        await destroyEntry(entry, outcome);
+    }));
+};
+exports.releaseDbSessionsForRun = releaseDbSessionsForRun;
+// Shutdown hook for hosts (backend process exit, CLI end, electron quit).
+const closeAllDbSessions = async () => {
+    const all = Array.from(sessions.values());
+    sessions.clear();
+    if (sweeper) {
+        clearInterval(sweeper);
+        sweeper = null;
+    }
+    await Promise.all(all.map((entry) => destroyEntry(entry, "rollback")));
+};
+exports.closeAllDbSessions = closeAllDbSessions;
+// Introspection for tests.
+const __getDbSessionCountForTests = () => sessions.size;
+exports.__getDbSessionCountForTests = __getDbSessionCountForTests;
+//# sourceMappingURL=dbSessionManager.js.map

package/dist-cjs/host/db/dbSessionManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dbSessionManager.js","sourceRoot":"","sources":["../../../src/host/db/dbSessionManager.ts"],"names":[],"mappings":";AAAA,4EAA4E;AAC5E,UAAU;AACV,EAAE;AACF,2EAA2E;AAC3E,2EAA2E;AAC3E,kEAAkE;AAClE,6DAA6D;AAC7D,EAAE;AACF,yEAAyE;AACzE,uDAAuD;;;AAEvD,uDAAqD;AACrD,uDAAqD;AACrD,yDAAuD;AACvD,iDAA+C;AA4B/C,MAAM,sBAAsB,GAAG,MAAM,CAAC;AACtC,MAAM,mBAAmB,GAAG,CAAC,GAAG,MAAM,CAAC;AAEvC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA0B,CAAC;AACnD,IAAI,OAAO,GAA0C,IAAI,CAAC;AAE1D,2EAA2E;AAC3E,yEAAyE;AACzE,+BAA+B;AAC/B,MAAM,UAAU,GAAG,CAAC,KAAa,EAAE,UAAwB,EAAU,EAAE,CACrE;IACE,KAAK;IACL,UAAU,CAAC,IAAI;IACf,UAAU,CAAC,EAAE;IACb,UAAU,CAAC,gBAAgB,IAAI,EAAE;IACjC,UAAU,CAAC,IAAI;IACf,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC;IACvB,UAAU,CAAC,QAAQ,IAAI,EAAE;IACzB,UAAU,CAAC,QAAQ;CACpB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEf,MAAM,aAAa,GAAG,GAAS,EAAE;IAC/B,IAAI,OAAO;QAAE,OAAO;IACpB,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,QAAQ,EAAE,CAAC;YACpC,IAAI,GAAG,GAAG,KAAK,CAAC,YAAY,GAAG,mBAAmB,EAAE,CAAC;gBACnD,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACrB,KAAK,YAAY,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,IAAI,OAAO,EAAE,CAAC;YACnC,aAAa,CAAC,OAAO,CAAC,CAAC;YACvB,OAAO,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC,EAAE,sBAAsB,CAAC,CAAC;IAC3B,0DAA0D;IACzD,OAAkC,CAAC,KAAK,EAAE,EAAE,CAAC;AAChD,CAAC,CAAC;AAEF,MAAM,YAAY,GAAG,KAAK,EACxB,KAAqB,EACrB,OAA8B,EACf,EAAE;IACjB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC;QACzC,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,MAAM,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;YACrE,CAAC;YAAC,MAAM,CAAC;gBACP,uDAAuD;YACzD,CAAC;QACH,CAAC;QACD,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACvB,CAAC;QAAC,MAAM,CAAC;YACP,wBAAwB;QAC1B,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,iDAAiD;IACnD,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,UAAU,GAAG,CAAC,UAAwB,EAAE,KAAqB,EAA4B,EAAE;IAC/F,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;QACxB,KAAK,UAAU;YACb,OAAO,IAAA,4BAAa,EAAC,UAAU,EAAE,CAAC,MAAM,EAAE,EAAE;gBAC1C,KAAK,CAAC,QAAQ,GAAG,MAAM,CAAC;YAC1B,CAAC,CAAC,CAAC;QACL,KAAK,QAAQ;YACX,OAAO,IAAA,oCAAiB,EAAC,UAAU,CAAC,CAAC;QACvC,KAAK,OAAO;YACV,OAAO,IAAA,kCAAgB,EAAC,UAAU,CAAC,CAAC;QACtC,KAAK,OAAO;YACV,OAAO,IAAA,kCAAgB,EAAC,UAAU,CAAC,CAAC;QACtC;YACE,OAAO,OAAO,CAAC,MAAM,CACnB,IAAI,KAAK,CACP,qEAAqE,UAAU,CAAC,IAAI,yDAAyD,CAC9I,CACF,CAAC;IACN,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,SAAS,GAAG,CAAC,KAAa,EAAE,UAAwB,EAAkB,EAAE;IAC5E,MAAM,KAAK,GAAG;QACZ,KAAK,EAAE,OAAO,CAAC,OAAO,EAAE;QACxB,eAAe,EAAE,KAAK;QACtB,QAAQ,EAAE,IAAI;QACd,YAAY,EAAE,IAAI,CAAC,GAAG,EAAE;QACxB,KAAK;KACuB,CAAC;IAC/B,KAAK,CAAC,aAAa,GAAG,UAAU,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IACpD,+DAA+D;IAC/D,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC3C,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AASK,MAAM,kBAAkB,GAAG,KAAK,EACrC,KAA4B,EACA,EAAE;IAC9B,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;IACtD,IAAI,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC9B,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;QACjD,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACzB,aAAa,EAAE,CAAC;IAClB,CAAC;IAED,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;QACtC,MAAM,MAAM,GAAG,MAAM,KAAM,CAAC,aAAa,CAAC;QAC1C,IAAI,KAAM,CAAC,QAAQ,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,GAAG,MAAM,CAAC,KAAK,yBAAyB,KAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC7E,CAAC;QACD,MAAM,aAAa,GAAG,KAAK,CAAC,eAAe,KAAK,MAAM,CAAC;QACvD,IAAI,aAAa,IAAI,CAAC,KAAM,CAAC,eAAe,EAAE,CAAC;YAC7C,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,KAAM,CAAC,eAAe,GAAG,IAAI,CAAC;QAChC,CAAC;QACD,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,eAAe,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,GAAG,KAAK,CAAC;QACpG,IAAI,CAAC;YACH,OAAO,MAAM,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;QAC1D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAK,KAAwC,EAAE,SAAS,EAAE,CAAC;gBACzD,KAAM,CAAC,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,qBAAqB,CAAC;YACnF,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,kEAAkE;IAClE,2DAA2D;IAC3D,KAAK,CAAC,KAAK,GAAG,GAAG,CAAC,IAAI,CACpB,GAAG,EAAE,CAAC,SAAS,EACf,GAAG,EAAE,CAAC,SAAS,CAChB,CAAC,IAAI,CAAC,GAAG,EAAE;QACV,KAAM,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,OAAO,GAAG,CAAC;AACb,CAAC,CAAC;AA1CW,QAAA,kBAAkB,sBA0C7B;AAEF,yEAAyE;AACzE,+EAA+E;AAC/E,qCAAqC;AAC9B,MAAM,uBAAuB,GAAG,KAAK,EAC1C,KAAa,EACb,OAA8B,EACf,EAAE;IACjB,MAAM,KAAK,GAAoC,EAAE,CAAC;IAClD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,QAAQ,EAAE,CAAC;QACpC,IAAI,KAAK,CAAC,KAAK,KAAK,KAAK;YAAE,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC;IACtD,CAAC;IACD,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;QAC/B,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACrB,MAAM,KAAK,CAAC,KAAK,CAAC;QAClB,MAAM,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACrC,CAAC,CAAC,CACH,CAAC;AACJ,CAAC,CAAC;AAfW,QAAA,uBAAuB,2BAelC;AAEF,0EAA0E;AACnE,MAAM,kBAAkB,GAAG,KAAK,IAAmB,EAAE;IAC1D,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1C,QAAQ,CAAC,KAAK,EAAE,CAAC;IACjB,IAAI,OAAO,EAAE,CAAC;QACZ,aAAa,CAAC,OAAO,CAAC,CAAC;QACvB,OAAO,GAAG,IAAI,CAAC;IACjB,CAAC;IACD,MAAM,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,YAAY,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;AACzE,CAAC,CAAC;AARW,QAAA,kBAAkB,sBAQ7B;AAEF,2BAA2B;AACpB,MAAM,2BAA2B,GAAG,GAAW,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;AAA1D,QAAA,2BAA2B,+BAA+B"}

package/dist-cjs/host/db/introspection/index.js

@@ -18,7 +18,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.introspectSchema = exports.formatOracleType = exports.formatMssqlType = exports.bsonTypeOf = exports.fieldsFromSample = exports.tableKey = exports.tableMatchesFilter = void 0;
+exports.introspectSchema = exports.formatOracleType = exports.formatMssqlType = exports.bsonTypeOf = exports.fieldsFromSample = exports.tableKey = exports.tableMatchesFilter = exports.introspectRoutines = exports.buildRoutineCall = void 0;
 const queryExecutor_js_1 = require("../queryExecutor.js");
 const mongo_js_1 = require("./mongo.js");
 const mssql_js_1 = require("./mssql.js");
@@ -27,6 +27,9 @@ const oracle_js_1 = require("./oracle.js");
 const postgres_js_1 = require("./postgres.js");
 const shared_js_1 = require("./shared.js");
 __exportStar(require("./types.js"), exports);
+var routines_js_1 = require("./routines.js");
+Object.defineProperty(exports, "buildRoutineCall", { enumerable: true, get: function () { return routines_js_1.buildRoutineCall; } });
+Object.defineProperty(exports, "introspectRoutines", { enumerable: true, get: function () { return routines_js_1.introspectRoutines; } });
 var shared_js_2 = require("./shared.js");
 Object.defineProperty(exports, "tableMatchesFilter", { enumerable: true, get: function () { return shared_js_2.tableMatchesFilter; } });
 Object.defineProperty(exports, "tableKey", { enumerable: true, get: function () { return shared_js_2.tableKey; } });

package/dist-cjs/host/db/introspection/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/host/db/introspection/index.ts"],"names":[],"mappings":";AAAA,yEAAyE;AACzE,yEAAyE;AACzE,0EAA0E;AAC1E,uDAAuD;;;;;;;;;;;;;;;;;AAEvD,0DAA+D;AAE/D,yCAA6C;AAC7C,yCAA6C;AAC7C,yCAA6C;AAC7C,2CAA+C;AAC/C,+CAAmD;AACnD,2CAAuD;AAGvD,6CAA2B;AAC3B,yCAA2D;AAAlD,+GAAA,kBAAkB,OAAA;AAAE,qGAAA,QAAQ,OAAA;AACrC,uCAA0D;AAAjD,4GAAA,gBAAgB,OAAA;AAAE,sGAAA,UAAU,OAAA;AACrC,uCAA6C;AAApC,2GAAA,eAAe,OAAA;AACxB,yCAA+C;AAAtC,6GAAA,gBAAgB,OAAA;AAElB,MAAM,gBAAgB,GAAG,KAAK,EAAE,eAAkC,EAAE,UAA+B,EAAE,EAAqB,EAAE;IACjI,MAAM,UAAU,GAAG,IAAA,2CAAwB,EAAC,EAAE,GAAG,eAAe,EAAE,EAAE,EAAE,eAAe,CAAC,EAAE,IAAI,YAAY,EAAE,CAAC,CAAC;IAC5G,MAAM,QAAQ,GAAG,IAAA,oCAAwB,EAAC,OAAO,CAAC,CAAC;IACnD,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;QACxB,KAAK,UAAU;YACb,OAAO,IAAA,gCAAkB,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAClD,KAAK,OAAO;YACV,OAAO,IAAA,0BAAe,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC/C,KAAK,OAAO;YACV,OAAO,IAAA,0BAAe,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC/C,KAAK,OAAO;YACV,OAAO,IAAA,0BAAe,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC/C,KAAK,QAAQ;YACX,OAAO,IAAA,4BAAgB,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAChD;YACE,MAAM,IAAI,KAAK,CAAC,8CAA8C,MAAM,CAAE,UAA+B,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACjI,CAAC;AACH,CAAC,CAAC;AAjBW,QAAA,gBAAgB,oBAiB3B"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/host/db/introspection/index.ts"],"names":[],"mappings":";AAAA,yEAAyE;AACzE,yEAAyE;AACzE,0EAA0E;AAC1E,uDAAuD;;;;;;;;;;;;;;;;;AAEvD,0DAA+D;AAE/D,yCAA6C;AAC7C,yCAA6C;AAC7C,yCAA6C;AAC7C,2CAA+C;AAC/C,+CAAmD;AACnD,2CAAuD;AAGvD,6CAA2B;AAC3B,6CAOuB;AANrB,+GAAA,gBAAgB,OAAA;AAChB,iHAAA,kBAAkB,OAAA;AAMpB,yCAA2D;AAAlD,+GAAA,kBAAkB,OAAA;AAAE,qGAAA,QAAQ,OAAA;AACrC,uCAA0D;AAAjD,4GAAA,gBAAgB,OAAA;AAAE,sGAAA,UAAU,OAAA;AACrC,uCAA6C;AAApC,2GAAA,eAAe,OAAA;AACxB,yCAA+C;AAAtC,6GAAA,gBAAgB,OAAA;AAElB,MAAM,gBAAgB,GAAG,KAAK,EAAE,eAAkC,EAAE,UAA+B,EAAE,EAAqB,EAAE;IACjI,MAAM,UAAU,GAAG,IAAA,2CAAwB,EAAC,EAAE,GAAG,eAAe,EAAE,EAAE,EAAE,eAAe,CAAC,EAAE,IAAI,YAAY,EAAE,CAAC,CAAC;IAC5G,MAAM,QAAQ,GAAG,IAAA,oCAAwB,EAAC,OAAO,CAAC,CAAC;IACnD,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;QACxB,KAAK,UAAU;YACb,OAAO,IAAA,gCAAkB,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAClD,KAAK,OAAO;YACV,OAAO,IAAA,0BAAe,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC/C,KAAK,OAAO;YACV,OAAO,IAAA,0BAAe,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC/C,KAAK,OAAO;YACV,OAAO,IAAA,0BAAe,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC/C,KAAK,QAAQ;YACX,OAAO,IAAA,4BAAgB,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAChD;YACE,MAAM,IAAI,KAAK,CAAC,8CAA8C,MAAM,CAAE,UAA+B,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACjI,CAAC;AACH,CAAC,CAAC;AAjBW,QAAA,gBAAgB,oBAiB3B"}

package/dist-cjs/host/db/introspection/routines.js

@@ -0,0 +1,444 @@
+"use strict";
+// Stored routine catalog: the procedures and functions a connection can call,
+// with their arguments, read from the database's own dictionary. What a
+// routine does is the database's business; this only says how to call it.
+//
+// `buildRoutineCall` turns a routine into a ready statement plus its binds, so
+// a test can be started from the catalog on any engine.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.buildRoutineCall = exports.introspectRoutines = void 0;
+const mssqlAdapter_js_1 = require("../mssqlAdapter.js");
+const mysqlAdapter_js_1 = require("../mysqlAdapter.js");
+const oracleAdapter_js_1 = require("../oracleAdapter.js");
+const postgresAdapter_js_1 = require("../postgresAdapter.js");
+const queryExecutor_js_1 = require("../queryExecutor.js");
+const DEFAULT_ROUTINE_LIMIT = 500;
+const text = (value) => (value === null || value === undefined ? "" : String(value));
+// ── Oracle ──────────────────────────────────────────────────────────────────
+const ORACLE_ROUTINES_SQL = `
+SELECT OWNER, OBJECT_NAME, PROCEDURE_NAME, OBJECT_TYPE, OVERLOAD
+FROM ALL_PROCEDURES
+WHERE OWNER = :owner
+  AND ((OBJECT_TYPE = 'PACKAGE' AND PROCEDURE_NAME IS NOT NULL)
+       OR OBJECT_TYPE IN ('PROCEDURE', 'FUNCTION'))
+ORDER BY OBJECT_NAME, PROCEDURE_NAME, OVERLOAD`;
+// DATA_LEVEL 0 = the arguments themselves, not the fields of a record type.
+const ORACLE_ARGUMENTS_SQL = `
+SELECT PACKAGE_NAME, OBJECT_NAME, OVERLOAD, ARGUMENT_NAME, POSITION, DATA_TYPE, IN_OUT, DEFAULTED,
+       TYPE_OWNER, TYPE_NAME, TYPE_SUBNAME
+FROM ALL_ARGUMENTS
+WHERE OWNER = :owner AND DATA_LEVEL = 0 AND DATA_TYPE IS NOT NULL
+ORDER BY PACKAGE_NAME, OBJECT_NAME, OVERLOAD, POSITION`;
+const oracleBindType = (dataType) => {
+    const type = dataType.toUpperCase();
+    if (type === "REF CURSOR")
+        return "cursor";
+    if (["NUMBER", "FLOAT", "BINARY_INTEGER", "PLS_INTEGER", "BINARY_FLOAT", "BINARY_DOUBLE"].includes(type))
+        return "numeric";
+    if (type === "DATE")
+        return "date";
+    if (type.startsWith("TIMESTAMP"))
+        return "timestamp";
+    if (type === "CLOB" || type === "NCLOB")
+        return "clob";
+    if (type === "BLOB")
+        return "blob";
+    if (type === "PL/SQL BOOLEAN" || type === "BOOLEAN")
+        return "boolean";
+    if (["PL/SQL TABLE", "PL/SQL RECORD", "OBJECT", "TABLE", "VARRAY"].includes(type))
+        return "object";
+    return "text";
+};
+const oracleDirection = (inOut) => inOut === "OUT" ? "out" : inOut === "IN/OUT" ? "inout" : "in";
+const oracleTypeName = (row) => {
+    const parts = [row.TYPE_OWNER, row.TYPE_NAME, row.TYPE_SUBNAME].map(text).filter(Boolean);
+    return parts.length > 0 ? parts.join(".") : undefined;
+};
+const listOracleRoutines = (connection, options) => (0, oracleAdapter_js_1.withOracleConnection)(connection, async (client, oracledb) => {
+    const outFormat = { outFormat: oracledb.OUT_FORMAT_OBJECT };
+    let owners = options.schemas?.map((owner) => owner.toUpperCase());
+    if (!owners || owners.length === 0) {
+        const current = await client.execute("SELECT SYS_CONTEXT('USERENV', 'CURRENT_SCHEMA') AS SCH FROM DUAL", [], outFormat);
+        owners = [text(current.rows[0]?.SCH)];
+    }
+    const routines = [];
+    for (const owner of owners) {
+        const objects = ((await client.execute(ORACLE_ROUTINES_SQL, { owner }, outFormat)).rows ?? []);
+        const argumentRows = ((await client.execute(ORACLE_ARGUMENTS_SQL, { owner }, outFormat)).rows ?? []);
+        const argsByRoutine = new Map();
+        for (const row of argumentRows) {
+            const key = [text(row.PACKAGE_NAME), text(row.OBJECT_NAME), text(row.OVERLOAD)].join("|");
+            argsByRoutine.set(key, [...(argsByRoutine.get(key) ?? []), row]);
+        }
+        for (const object of objects) {
+            const isPackage = text(object.OBJECT_TYPE) === "PACKAGE";
+            const name = text(isPackage ? object.PROCEDURE_NAME : object.OBJECT_NAME);
+            const pkg = isPackage ? text(object.OBJECT_NAME) : "";
+            const overload = text(object.OVERLOAD);
+            const rows = argsByRoutine.get([pkg, name, overload].join("|")) ?? [];
+            // Position 0 is a function's return value.
+            const returned = rows.find((row) => Number(row.POSITION) === 0 && !row.ARGUMENT_NAME);
+            routines.push({
+                schema: owner,
+                ...(pkg ? { package: pkg } : {}),
+                name,
+                kind: returned || text(object.OBJECT_TYPE) === "FUNCTION" ? "function" : "procedure",
+                ...(overload ? { overload } : {}),
+                arguments: rows
+                    .filter((row) => row !== returned && row.ARGUMENT_NAME)
+                    .map((row) => {
+                    const dataType = text(row.DATA_TYPE);
+                    const typeName = oracleBindType(dataType) === "object" ? oracleTypeName(row) : undefined;
+                    return {
+                        name: text(row.ARGUMENT_NAME),
+                        position: Number(row.POSITION),
+                        direction: oracleDirection(text(row.IN_OUT)),
+                        dataType,
+                        bindType: oracleBindType(dataType),
+                        ...(typeName ? { typeName } : {}),
+                        hasDefault: text(row.DEFAULTED) === "Y",
+                    };
+                }),
+                ...(returned
+                    ? {
+                        returnType: text(returned.DATA_TYPE),
+                        returnBindType: oracleBindType(text(returned.DATA_TYPE)),
+                        ...(oracleBindType(text(returned.DATA_TYPE)) === "object" && oracleTypeName(returned)
+                            ? { returnTypeName: oracleTypeName(returned) }
+                            : {}),
+                    }
+                    : {}),
+            });
+        }
+    }
+    return routines;
+});
+// ── PostgreSQL ──────────────────────────────────────────────────────────────
+const POSTGRES_ROUTINES_SQL = `
+SELECT n.nspname AS schema, p.proname AS name, p.prokind AS kind, p.oid::text AS overload,
+       p.proretset AS returns_set, format_type(p.prorettype, NULL) AS return_type,
+       p.pronargdefaults AS defaults, p.pronargs AS in_args,
+       p.proargnames AS names, p.proargmodes::text[] AS modes,
+       ARRAY(SELECT format_type(t, NULL)
+             FROM unnest(COALESCE(p.proallargtypes, p.proargtypes::oid[])) WITH ORDINALITY AS a(t, i)
+             ORDER BY a.i) AS types
+FROM pg_proc p
+JOIN pg_namespace n ON n.oid = p.pronamespace
+WHERE p.prokind IN ('f', 'p')
+  AND n.nspname NOT IN ('pg_catalog', 'information_schema')
+  AND n.nspname NOT LIKE 'pg\\_%'
+  AND p.prorettype <> 'trigger'::regtype
+ORDER BY 1, 2`;
+const postgresBindType = (dataType) => {
+    const type = dataType.toLowerCase();
+    if (type === "refcursor")
+        return "cursor";
+    if (/^(smallint|integer|bigint)$/.test(type))
+        return "int";
+    if (/^(numeric|real|double precision|money)/.test(type))
+        return "numeric";
+    if (type === "boolean")
+        return "boolean";
+    if (type === "date")
+        return "date";
+    if (type.startsWith("timestamp"))
+        return "timestamp";
+    if (type === "json" || type === "jsonb")
+        return "json";
+    return "text";
+};
+const listPostgresRoutines = (connection, options) => (0, postgresAdapter_js_1.withPostgresClient)(connection, async (client) => {
+    const { rows } = (await client.query(POSTGRES_ROUTINES_SQL));
+    return rows
+        .filter((row) => !options.schemas?.length || options.schemas.includes(text(row.schema)))
+        .map((row) => {
+        const types = (row.types ?? []);
+        const names = (row.names ?? []);
+        const modes = (row.modes ?? null);
+        const defaults = Number(row.defaults ?? 0);
+        const inTotal = types.filter((_, index) => {
+            const mode = modes?.[index] ?? "i";
+            return mode === "i" || mode === "b" || mode === "v";
+        }).length;
+        let inSeen = 0;
+        const args = types.flatMap((dataType, index) => {
+            const mode = modes?.[index] ?? "i";
+            // "t" columns of RETURNS TABLE are not arguments.
+            if (mode === "t")
+                return [];
+            const isIn = mode === "i" || mode === "b" || mode === "v";
+            if (isIn)
+                inSeen += 1;
+            return [
+                {
+                    name: names[index] || `arg${index + 1}`,
+                    position: index + 1,
+                    direction: mode === "o" ? "out" : mode === "b" ? "inout" : "in",
+                    dataType,
+                    bindType: postgresBindType(dataType),
+                    hasDefault: isIn && inSeen > inTotal - defaults,
+                },
+            ];
+        });
+        const isFunction = text(row.kind) === "f";
+        const returnType = text(row.return_type);
+        return {
+            schema: text(row.schema),
+            name: text(row.name),
+            kind: isFunction ? "function" : "procedure",
+            overload: text(row.overload),
+            arguments: args,
+            ...(isFunction && returnType !== "void"
+                ? { returnType, returnBindType: postgresBindType(returnType) }
+                : {}),
+            ...(row.returns_set === true || returnType === "record" ? { returnsTable: true } : {}),
+        };
+    });
+});
+// ── MySQL ───────────────────────────────────────────────────────────────────
+const MYSQL_ROUTINES_SQL = `
+SELECT r.ROUTINE_SCHEMA AS sch, r.ROUTINE_NAME AS name, r.ROUTINE_TYPE AS kind,
+       r.DTD_IDENTIFIER AS return_type,
+       p.PARAMETER_NAME AS arg_name, p.ORDINAL_POSITION AS pos, p.PARAMETER_MODE AS mode, p.DTD_IDENTIFIER AS arg_type
+FROM information_schema.ROUTINES r
+LEFT JOIN information_schema.PARAMETERS p
+  ON p.SPECIFIC_SCHEMA = r.ROUTINE_SCHEMA AND p.SPECIFIC_NAME = r.ROUTINE_NAME AND p.ROUTINE_TYPE = r.ROUTINE_TYPE AND p.ORDINAL_POSITION > 0
+WHERE r.ROUTINE_SCHEMA = COALESCE(?, DATABASE())
+ORDER BY r.ROUTINE_NAME, p.ORDINAL_POSITION`;
+const mysqlBindType = (dataType) => {
+    const type = dataType.toLowerCase();
+    if (/^(tinyint\(1\)|bool)/.test(type))
+        return "boolean";
+    if (/^(tinyint|smallint|mediumint|int|bigint)/.test(type))
+        return "int";
+    if (/^(decimal|numeric|float|double)/.test(type))
+        return "numeric";
+    if (type.startsWith("date") && !type.startsWith("datetime"))
+        return "date";
+    if (/^(datetime|timestamp)/.test(type))
+        return "timestamp";
+    if (type.startsWith("json"))
+        return "json";
+    return "text";
+};
+const listMysqlRoutines = (connection, options) => (0, mysqlAdapter_js_1.withMysqlConnection)(connection, async (conn) => {
+    const schemas = options.schemas?.length ? options.schemas : [null];
+    const routines = new Map();
+    for (const schema of schemas) {
+        const [rows] = (await conn.query(MYSQL_ROUTINES_SQL, [schema]));
+        for (const row of rows) {
+            const key = `${text(row.sch)}.${text(row.name)}.${text(row.kind)}`;
+            const isFunction = text(row.kind) === "FUNCTION";
+            let routine = routines.get(key);
+            if (!routine) {
+                routine = {
+                    schema: text(row.sch),
+                    name: text(row.name),
+                    kind: isFunction ? "function" : "procedure",
+                    arguments: [],
+                    ...(isFunction
+                        ? { returnType: text(row.return_type), returnBindType: mysqlBindType(text(row.return_type)) }
+                        : {}),
+                };
+                routines.set(key, routine);
+            }
+            if (row.arg_name) {
+                const dataType = text(row.arg_type);
+                routine.arguments.push({
+                    name: text(row.arg_name),
+                    position: Number(row.pos),
+                    direction: row.mode === "OUT" ? "out" : row.mode === "INOUT" ? "inout" : "in",
+                    dataType,
+                    bindType: mysqlBindType(dataType),
+                    hasDefault: false,
+                });
+            }
+        }
+    }
+    return [...routines.values()];
+});
+// ── SQL Server ──────────────────────────────────────────────────────────────
+const MSSQL_ROUTINES_SQL = `
+SELECT s.name AS sch, o.name AS name, RTRIM(o.type) AS kind, o.object_id AS id,
+       p.name AS arg_name, p.parameter_id AS pos, p.is_output, p.has_default_value,
+       TYPE_NAME(p.user_type_id) AS arg_type
+FROM sys.objects o
+JOIN sys.schemas s ON s.schema_id = o.schema_id
+LEFT JOIN sys.parameters p ON p.object_id = o.object_id
+WHERE o.type IN ('P', 'FN', 'IF', 'TF') AND o.is_ms_shipped = 0
+ORDER BY s.name, o.name, p.parameter_id`;
+const mssqlBindType = (dataType) => {
+    const type = dataType.toLowerCase();
+    if (/^(tinyint|smallint|int|bigint)$/.test(type))
+        return "int";
+    if (/^(decimal|numeric|money|smallmoney|float|real)$/.test(type))
+        return "numeric";
+    if (type === "bit")
+        return "boolean";
+    if (type === "date")
+        return "date";
+    if (/^(datetime|datetime2|smalldatetime|datetimeoffset)$/.test(type))
+        return "timestamp";
+    if (/^(varbinary|binary|image)$/.test(type))
+        return "blob";
+    return "text";
+};
+const listMssqlRoutines = (connection, options) => (0, mssqlAdapter_js_1.withMssqlPool)(connection, async (pool) => {
+    const { recordset } = (await pool.request().query(MSSQL_ROUTINES_SQL));
+    const routines = new Map();
+    for (const row of recordset) {
+        if (options.schemas?.length && !options.schemas.includes(text(row.sch)))
+            continue;
+        const key = text(row.id);
+        const kind = text(row.kind);
+        let routine = routines.get(key);
+        if (!routine) {
+            routine = {
+                schema: text(row.sch),
+                name: text(row.name),
+                kind: kind === "P" ? "procedure" : "function",
+                arguments: [],
+                ...(kind === "IF" || kind === "TF" ? { returnsTable: true } : {}),
+            };
+            routines.set(key, routine);
+        }
+        const argName = text(row.arg_name).replace(/^@/, "");
+        const position = Number(row.pos ?? 0);
+        const dataType = text(row.arg_type);
+        if (position === 0 && !argName) {
+            // Parameter 0 of a scalar function is its return value.
+            if (dataType) {
+                routine.returnType = dataType;
+                routine.returnBindType = mssqlBindType(dataType);
+            }
+        }
+        else if (argName) {
+            routine.arguments.push({
+                name: argName,
+                position,
+                // T-SQL has no pure OUT: every OUTPUT parameter can also take a value.
+                direction: row.is_output === true ? "inout" : "in",
+                dataType,
+                bindType: mssqlBindType(dataType),
+                // Set only for CLR routines: SQL Server does not report whether a
+                // T-SQL parameter has a default, so such a call lists it too.
+                hasDefault: row.has_default_value === true,
+            });
+        }
+    }
+    return [...routines.values()];
+});
+// ── Entry points ────────────────────────────────────────────────────────────
+const qualifiedName = (routine) => [routine.schema, routine.package, routine.name].filter(Boolean).join(".");
+const introspectRoutines = async (connectionInput, options = {}) => {
+    const connection = (0, queryExecutor_js_1.normalizeConnectionInput)({ ...connectionInput, id: connectionInput.id ?? "introspect" });
+    const load = () => {
+        switch (connection.type) {
+            case "oracle":
+                return listOracleRoutines(connection, options);
+            case "postgres":
+                return listPostgresRoutines(connection, options);
+            case "mysql":
+                return listMysqlRoutines(connection, options);
+            case "mssql":
+                return listMssqlRoutines(connection, options);
+            default:
+                throw new Error(`"${connection.type}" connections have no stored procedures to list.`);
+        }
+    };
+    const search = options.search?.trim().toLowerCase();
+    const matching = (await load()).filter((routine) => !search || qualifiedName(routine).toLowerCase().includes(search));
+    const limit = options.limit && options.limit > 0 ? options.limit : DEFAULT_ROUTINE_LIMIT;
+    return {
+        dialect: connection.type,
+        routines: matching.slice(0, limit),
+        truncated: matching.length > limit,
+    };
+};
+exports.introspectRoutines = introspectRoutines;
+// The statement and binds that call a routine. Arguments with a default are
+// left out, so the call starts from what the routine requires; a tester adds
+// any of them back by editing the statement.
+const buildRoutineCall = (dialect, routine) => {
+    const args = routine.arguments.filter((arg) => !arg.hasDefault || arg.direction !== "in");
+    const name = qualifiedName(routine);
+    const toBind = (arg) => ({
+        name: dialect === "mysql" || dialect === "postgres" ? arg.name : arg.name.toLowerCase(),
+        value: "",
+        type: arg.bindType,
+        direction: arg.bindType === "cursor" ? "out" : arg.direction,
+        ...(arg.typeName ? { typeName: arg.typeName } : {}),
+    });
+    const binds = args.map(toBind);
+    const returnBind = routine.returnBindType
+        ? {
+            name: "return_value",
+            value: "",
+            type: routine.returnBindType,
+            direction: "out",
+            ...(routine.returnTypeName ? { typeName: routine.returnTypeName } : {}),
+        }
+        : null;
+    switch (dialect) {
+        case "oracle": {
+            const list = args.map((arg) => `${arg.name} => :${arg.name.toLowerCase()}`).join(", ");
+            const call = `${name}${list ? `(${list})` : ""}`;
+            return routine.kind === "function" && returnBind
+                ? { query: `BEGIN\n  :return_value := ${call};\nEND;`, binds: [returnBind, ...binds] }
+                : { query: `BEGIN\n  ${call};\nEND;`, binds };
+        }
+        case "postgres": {
+            // OUT arguments take NULL in CALL and nothing in a function call; either
+            // way they come back as columns of the returned row.
+            let slot = 0;
+            if (routine.kind === "procedure") {
+                const list = args
+                    .map((arg) => (arg.direction === "out" ? "NULL" : `$${(slot += 1)}`))
+                    .join(", ");
+                return { query: `CALL ${name}(${list})`, binds };
+            }
+            const list = args
+                .filter((arg) => arg.direction !== "out")
+                .map(() => `$${(slot += 1)}`)
+                .join(", ");
+            return {
+                query: `SELECT * FROM ${name}(${list})`,
+                binds: returnBind && routine.returnBindType === "cursor"
+                    ? [...binds, { ...returnBind, name: routine.name.toLowerCase() }]
+                    : binds,
+            };
+        }
+        case "mysql": {
+            if (routine.kind === "function") {
+                return {
+                    query: `SELECT ${name}(${args.map(() => "?").join(", ")}) AS return_value`,
+                    binds,
+                };
+            }
+            return {
+                query: `CALL ${name}(${args
+                    .map((arg) => (arg.direction === "in" ? "?" : `@${arg.name}`))
+                    .join(", ")})`,
+                binds,
+            };
+        }
+        default: {
+            const quoted = [routine.schema, routine.name].filter(Boolean).map((part) => `[${part}]`).join(".");
+            if (routine.kind === "function") {
+                const list = args.map((arg) => `@${arg.name.toLowerCase()}`).join(", ");
+                return {
+                    query: routine.returnsTable
+                        ? `SELECT * FROM ${quoted}(${list})`
+                        : `SELECT ${quoted}(${list}) AS return_value`,
+                    binds,
+                };
+            }
+            const list = args
+                .map((arg) => `@${arg.name} = @${arg.name.toLowerCase()}${arg.direction === "in" ? "" : " OUTPUT"}`)
+                .join(", ");
+            return { query: `EXEC ${quoted}${list ? ` ${list}` : ""}`, binds };
+        }
+    }
+};
+exports.buildRoutineCall = buildRoutineCall;
+//# sourceMappingURL=routines.js.map