@askdb/studio 0.2.0-beta.24 → 0.2.0-beta.25

package/dist/client/index.html

@@ -6,8 +6,8 @@
     <title>AskDB Studio</title>
     <link rel="icon" type="image/png" media="(prefers-color-scheme: light)" href="/assets/brand/light-icon.png" />
     <link rel="icon" type="image/png" media="(prefers-color-scheme: dark)" href="/assets/brand/dark-icon.png" />
-    <script type="module" crossorigin src="/assets/index-_lnkmWaK.js"></script>
-    <link rel="stylesheet" crossorigin href="/assets/index-DY6BxO-7.css">
+    <script type="module" crossorigin src="/assets/index-DYBXDl3K.js"></script>
+    <link rel="stylesheet" crossorigin href="/assets/index-Bj6qGfT6.css">
   </head>
   <body>
     <div id="root"></div>

package/dist/execute-registry.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Studio execute-driver registry.
+ *
+ * Each entry describes a live execute provider — how to dynamically import the
+ * driver package, what connection shape it expects, and how to normalize its
+ * results into the shared `ExecuteResponse` DTO.
+ *
+ * Drivers are optional peers. The registry never imports a driver at module
+ * load time; every import lives inside an `executeXxx()` function so a missing
+ * package only fails when the user actually tries to execute a query.
+ */
+import type { ExecuteResponse } from "./shared/api.js";
+export type StudioExecuteProvider = "postgres" | "mysql" | "sqlite" | "sqlserver";
+export type StudioExecuteInput = {
+    connectionString?: string;
+    file?: string;
+    sql: string;
+    params: unknown[];
+};
+export type StudioDriverDefinition = {
+    provider: StudioExecuteProvider;
+    label: string;
+    packageName: "pg" | "mysql2" | "better-sqlite3" | "mssql";
+    installCommand: string;
+    execute(input: StudioExecuteInput): Promise<ExecuteResponse>;
+};
+export declare const EXECUTE_DRIVER_REGISTRY: Record<StudioExecuteProvider, StudioDriverDefinition>;
+/**
+ * Check whether a driver package is importable at runtime.
+ * Returns `true` when the import succeeds (even if the value is empty),
+ * `false` when it throws (module not found).
+ */
+export declare function isDriverInstalled(packageName: string): Promise<boolean>;
+//# sourceMappingURL=execute-registry.d.ts.map

package/dist/execute-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"execute-registry.d.ts","sourceRoot":"","sources":["../src/execute-registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAMvD,MAAM,MAAM,qBAAqB,GAAG,UAAU,GAAG,OAAO,GAAG,QAAQ,GAAG,WAAW,CAAC;AAElF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,OAAO,EAAE,CAAC;CACnB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,QAAQ,EAAE,qBAAqB,CAAC;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,IAAI,GAAG,QAAQ,GAAG,gBAAgB,GAAG,OAAO,CAAC;IAC1D,cAAc,EAAE,MAAM,CAAC;IACvB,OAAO,CAAC,KAAK,EAAE,kBAAkB,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;CAC9D,CAAC;AA6PF,eAAO,MAAM,uBAAuB,EAAE,MAAM,CAAC,qBAAqB,EAAE,sBAAsB,CA6BzF,CAAC;AAEF;;;;GAIG;AACH,wBAAsB,iBAAiB,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAO7E"}

package/dist/execute-registry.js

@@ -0,0 +1,240 @@
+/**
+ * Studio execute-driver registry.
+ *
+ * Each entry describes a live execute provider — how to dynamically import the
+ * driver package, what connection shape it expects, and how to normalize its
+ * results into the shared `ExecuteResponse` DTO.
+ *
+ * Drivers are optional peers. The registry never imports a driver at module
+ * load time; every import lives inside an `executeXxx()` function so a missing
+ * package only fails when the user actually tries to execute a query.
+ */
+// ---------------------------------------------------------------------------
+// Normalization helpers
+// ---------------------------------------------------------------------------
+const MAX_ROWS = 500;
+function buildOkResponse(columns, rows, rowCount, durationMs) {
+    const truncated = rows.length > MAX_ROWS;
+    return {
+        ok: true,
+        columns,
+        rows: rows.slice(0, MAX_ROWS),
+        rowCount,
+        durationMs,
+        truncated,
+    };
+}
+async function executePostgres(input) {
+    const { connectionString, sql, params } = input;
+    if (!connectionString) {
+        return { ok: false, error: "No connection URL configured for Postgres execute. Set ASKDB_STUDIO_DATABASE_URL or introspection.providerConfig.postgres.databaseUrl." };
+    }
+    let pgMod;
+    try {
+        const mod = await import("pg");
+        pgMod = (mod.default ?? mod);
+    }
+    catch {
+        return { ok: false, error: "The `pg` package is not installed. Install it with `pnpm add pg`." };
+    }
+    const client = new pgMod.Client({ connectionString });
+    try {
+        await client.connect();
+    }
+    catch (err) {
+        return { ok: false, error: err instanceof Error ? err.message : String(err) };
+    }
+    const startMs = Date.now();
+    try {
+        await client.query("BEGIN READ ONLY");
+        const result = await client.query({ text: sql, values: params });
+        await client.query("COMMIT");
+        const durationMs = Date.now() - startMs;
+        const columns = result.fields.map((f) => f.name);
+        const rows = result.rows.map((r) => columns.map((c) => r[c]));
+        return buildOkResponse(columns, rows, result.rowCount ?? rows.length, durationMs);
+    }
+    catch (err) {
+        await client.query("ROLLBACK").catch(() => { });
+        return { ok: false, error: err instanceof Error ? err.message : String(err) };
+    }
+    finally {
+        await client.end().catch(() => { });
+    }
+}
+async function executeMySQL(input) {
+    const { connectionString, sql, params } = input;
+    if (!connectionString) {
+        return { ok: false, error: "No connection URL configured for MySQL execute. Set ASKDB_STUDIO_DATABASE_URL or introspection.providerConfig.mysql.databaseUrl." };
+    }
+    let mysql2Mod;
+    try {
+        const mod = await import("mysql2/promise");
+        mysql2Mod = (mod.default ?? mod);
+    }
+    catch {
+        return { ok: false, error: "The `mysql2` package is not installed. Install it with `pnpm add mysql2`." };
+    }
+    let conn;
+    try {
+        conn = await mysql2Mod.createConnection({ uri: connectionString });
+    }
+    catch (err) {
+        return { ok: false, error: err instanceof Error ? err.message : String(err) };
+    }
+    const startMs = Date.now();
+    try {
+        await conn.query("SET SESSION TRANSACTION READ ONLY");
+        await conn.query("START TRANSACTION");
+        const [rowsRaw, fields] = await conn.execute(sql, params);
+        await conn.query("ROLLBACK");
+        const durationMs = Date.now() - startMs;
+        const columns = fields.map((f) => f.name);
+        const rows = rowsRaw.map((r) => columns.map((c) => r[c]));
+        return buildOkResponse(columns, rows, rows.length, durationMs);
+    }
+    catch (err) {
+        await conn.query("ROLLBACK").catch(() => { });
+        return { ok: false, error: err instanceof Error ? err.message : String(err) };
+    }
+    finally {
+        await conn.end().catch(() => { });
+    }
+}
+async function executeSQLite(input) {
+    const { file, sql, params } = input;
+    if (!file) {
+        return { ok: false, error: "No SQLite file path configured for execute. Set ASKDB_STUDIO_SQLITE_FILE or introspection.providerConfig.sqlite.file." };
+    }
+    let Sqlite3;
+    try {
+        const mod = await import("better-sqlite3");
+        Sqlite3 = mod.default;
+    }
+    catch {
+        return { ok: false, error: "The `better-sqlite3` package is not installed. Install it with `pnpm add better-sqlite3`." };
+    }
+    const startMs = Date.now();
+    const db = new Sqlite3(file, { readonly: true, fileMustExist: true });
+    try {
+        const stmt = db.prepare(sql);
+        const columns = stmt.columns().map((c) => c.name);
+        const rowsRaw = stmt.all(...params);
+        const durationMs = Date.now() - startMs;
+        const rows = rowsRaw.map((r) => columns.map((c) => r[c]));
+        return buildOkResponse(columns, rows, rows.length, durationMs);
+    }
+    catch (err) {
+        return { ok: false, error: err instanceof Error ? err.message : String(err) };
+    }
+    finally {
+        db.close();
+    }
+}
+/**
+ * Rewrite `$1`, `$2`, ... or bare `?` positional params to MSSQL `@p0`, `@p1`, ...
+ * and return an object of named bindings.
+ */
+function rewriteSqlServerParams(sql, params) {
+    if (params.length === 0)
+        return { sql, bindings: {} };
+    const bindings = {};
+    params.forEach((value, idx) => {
+        bindings[`p${idx}`] = value;
+    });
+    // Replace $N placeholders (1-indexed) first, then bare ? placeholders.
+    let idx = 0;
+    const rewritten = sql
+        .replace(/\$\d+/g, () => `@p${idx++}`)
+        .replace(/\?/g, () => `@p${idx++}`);
+    return { sql: rewritten, bindings };
+}
+async function executeSQLServer(input) {
+    const { connectionString, sql, params } = input;
+    if (!connectionString) {
+        return { ok: false, error: "No connection URL configured for SQL Server execute. Set ASKDB_STUDIO_DATABASE_URL or introspection.providerConfig.sqlserver.databaseUrl." };
+    }
+    let mssqlMod;
+    try {
+        const mod = await import("mssql");
+        mssqlMod = (mod.default ?? mod);
+    }
+    catch {
+        return { ok: false, error: "The `mssql` package is not installed. Install it with `pnpm add mssql`." };
+    }
+    let pool;
+    try {
+        pool = await new mssqlMod.ConnectionPool(connectionString).connect();
+    }
+    catch (err) {
+        return { ok: false, error: err instanceof Error ? err.message : String(err) };
+    }
+    const startMs = Date.now();
+    try {
+        const { sql: rewrittenSql, bindings } = rewriteSqlServerParams(sql, params);
+        const req = pool.request();
+        for (const [name, value] of Object.entries(bindings)) {
+            req.input(name, value);
+        }
+        const result = await req.query(rewrittenSql);
+        const durationMs = Date.now() - startMs;
+        const rowsRaw = result.recordset ?? [];
+        const columns = rowsRaw.length > 0 ? Object.keys(rowsRaw[0]) : [];
+        const rows = rowsRaw.map((r) => columns.map((c) => r[c]));
+        return buildOkResponse(columns, rows, rows.length, durationMs);
+    }
+    catch (err) {
+        return { ok: false, error: err instanceof Error ? err.message : String(err) };
+    }
+    finally {
+        await pool.close();
+    }
+}
+// ---------------------------------------------------------------------------
+// Registry
+// ---------------------------------------------------------------------------
+export const EXECUTE_DRIVER_REGISTRY = {
+    postgres: {
+        provider: "postgres",
+        label: "Postgres",
+        packageName: "pg",
+        installCommand: "pnpm add pg",
+        execute: executePostgres,
+    },
+    mysql: {
+        provider: "mysql",
+        label: "MySQL",
+        packageName: "mysql2",
+        installCommand: "pnpm add mysql2",
+        execute: executeMySQL,
+    },
+    sqlite: {
+        provider: "sqlite",
+        label: "SQLite",
+        packageName: "better-sqlite3",
+        installCommand: "pnpm add better-sqlite3",
+        execute: executeSQLite,
+    },
+    sqlserver: {
+        provider: "sqlserver",
+        label: "SQL Server",
+        packageName: "mssql",
+        installCommand: "pnpm add mssql",
+        execute: executeSQLServer,
+    },
+};
+/**
+ * Check whether a driver package is importable at runtime.
+ * Returns `true` when the import succeeds (even if the value is empty),
+ * `false` when it throws (module not found).
+ */
+export async function isDriverInstalled(packageName) {
+    try {
+        await import(packageName);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+//# sourceMappingURL=execute-registry.js.map

package/dist/execute-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"execute-registry.js","sourceRoot":"","sources":["../src/execute-registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAyBH,8EAA8E;AAC9E,wBAAwB;AACxB,8EAA8E;AAE9E,MAAM,QAAQ,GAAG,GAAG,CAAC;AAErB,SAAS,eAAe,CACtB,OAAiB,EACjB,IAAiB,EACjB,QAAgB,EAChB,UAAkB;IAElB,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC;IACzC,OAAO;QACL,EAAE,EAAE,IAAI;QACR,OAAO;QACP,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC;QAC7B,QAAQ;QACR,UAAU;QACV,SAAS;KACV,CAAC;AACJ,CAAC;AAkBD,KAAK,UAAU,eAAe,CAAC,KAAyB;IACtD,MAAM,EAAE,gBAAgB,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IAChD,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,wIAAwI,EAAE,CAAC;IACxK,CAAC;IAED,IAAI,KAAY,CAAC;IACjB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QAC/B,KAAK,GAAG,CAAE,GAAsC,CAAC,OAAO,IAAI,GAAG,CAAU,CAAC;IAC5E,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,mEAAmE,EAAE,CAAC;IACnG,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC;IACtD,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IACzB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;IAChF,CAAC;IACD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC3B,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACjE,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;QACxC,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,OAAO,eAAe,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACpF,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAO,MAAM,CAAC,KAAK,CAAC,UAAU,CAAsB,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACrE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;IAChF,CAAC;YAAS,CAAC;QACT,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACrC,CAAC;AACH,CAAC;AAeD,KAAK,UAAU,YAAY,CAAC,KAAyB;IACnD,MAAM,EAAE,gBAAgB,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IAChD,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,kIAAkI,EAAE,CAAC;IAClK,CAAC;IAED,IAAI,SAAoB,CAAC;IACzB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAC3C,SAAS,GAAG,CAAE,GAA0C,CAAC,OAAO,IAAI,GAAG,CAAc,CAAC;IACxF,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,2EAA2E,EAAE,CAAC;IAC3G,CAAC;IAED,IAAI,IAAsB,CAAC;IAC3B,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,SAAS,CAAC,gBAAgB,CAAC,EAAE,GAAG,EAAE,gBAAgB,EAAE,CAAC,CAAC;IACrE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;IAChF,CAAC;IACD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC3B,IAAI,CAAC;QACH,MAAM,IAAI,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACtD,MAAM,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC1D,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;QACxC,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,IAAI,GAAI,OAA0C,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9F,OAAO,eAAe,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACjE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC7C,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;IAChF,CAAC;YAAS,CAAC;QACT,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACnC,CAAC;AACH,CAAC;AAkBD,KAAK,UAAU,aAAa,CAAC,KAAyB;IACpD,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IACpC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,uHAAuH,EAAE,CAAC;IACvJ,CAAC;IAED,IAAI,OAAoC,CAAC;IACzC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAC3C,OAAO,GAAI,GAAmC,CAAC,OAAO,CAAC;IACzD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,2FAA2F,EAAE,CAAC;IAC3H,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC3B,MAAM,EAAE,GAAG,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;IACtE,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAmC,CAAC;QACtE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;QACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1D,OAAO,eAAe,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACjE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;IAChF,CAAC;YAAS,CAAC;QACT,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC;AACH,CAAC;AAkBD;;;GAGG;AACH,SAAS,sBAAsB,CAAC,GAAW,EAAE,MAAiB;IAC5D,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IACtD,MAAM,QAAQ,GAA4B,EAAE,CAAC;IAC7C,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QAC5B,QAAQ,CAAC,IAAI,GAAG,EAAE,CAAC,GAAG,KAAK,CAAC;IAC9B,CAAC,CAAC,CAAC;IACH,uEAAuE;IACvE,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,MAAM,SAAS,GAAG,GAAG;SAClB,OAAO,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,KAAK,GAAG,EAAE,EAAE,CAAC;SACrC,OAAO,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,KAAK,GAAG,EAAE,EAAE,CAAC,CAAC;IACtC,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;AACtC,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,KAAyB;IACvD,MAAM,EAAE,gBAAgB,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IAChD,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,2IAA2I,EAAE,CAAC;IAC3K,CAAC;IAED,IAAI,QAAkB,CAAC;IACvB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAClC,QAAQ,GAAG,CAAE,GAAyC,CAAC,OAAO,IAAI,GAAG,CAAa,CAAC;IACrF,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,yEAAyE,EAAE,CAAC;IACzG,CAAC;IAED,IAAI,IAAe,CAAC;IACpB,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,IAAI,QAAQ,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC,OAAO,EAAE,CAAC;IACvE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;IAChF,CAAC;IACD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC3B,IAAI,CAAC;QACH,MAAM,EAAE,GAAG,EAAE,YAAY,EAAE,QAAQ,EAAE,GAAG,sBAAsB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC5E,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;QAC3B,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YACrD,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACzB,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC7C,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;QACxC,MAAM,OAAO,GAAG,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAClE,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1D,OAAO,eAAe,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACjE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;IAChF,CAAC;YAAS,CAAC;QACT,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;AACH,CAAC;AAED,8EAA8E;AAC9E,WAAW;AACX,8EAA8E;AAE9E,MAAM,CAAC,MAAM,uBAAuB,GAA0D;IAC5F,QAAQ,EAAE;QACR,QAAQ,EAAE,UAAU;QACpB,KAAK,EAAE,UAAU;QACjB,WAAW,EAAE,IAAI;QACjB,cAAc,EAAE,aAAa;QAC7B,OAAO,EAAE,eAAe;KACzB;IACD,KAAK,EAAE;QACL,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,OAAO;QACd,WAAW,EAAE,QAAQ;QACrB,cAAc,EAAE,iBAAiB;QACjC,OAAO,EAAE,YAAY;KACtB;IACD,MAAM,EAAE;QACN,QAAQ,EAAE,QAAQ;QAClB,KAAK,EAAE,QAAQ;QACf,WAAW,EAAE,gBAAgB;QAC7B,cAAc,EAAE,yBAAyB;QACzC,OAAO,EAAE,aAAa;KACvB;IACD,SAAS,EAAE;QACT,QAAQ,EAAE,WAAW;QACrB,KAAK,EAAE,YAAY;QACnB,WAAW,EAAE,OAAO;QACpB,cAAc,EAAE,gBAAgB;QAChC,OAAO,EAAE,gBAAgB;KAC1B;CACF,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,WAAmB;IACzD,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QAC1B,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}

package/dist/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAA6C,MAAM,WAAW,CAAC;AA8BpF,OAAO,EAOL,mBAAmB,EAOpB,MAAM,YAAY,CAAC;AACpB,OAAO,EAaL,KAAK,SAAS,EACf,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EASV,kBAAkB,EAGnB,MAAM,iBAAiB,CAAC;AAMzB,MAAM,MAAM,aAAa,GAAG;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAyD3D,wBAAgB,qCAAqC,CACnD,OAAO,EAAE,OAAO,mBAAmB,GAAG,SAAS,GAC9C,IAAI,CAEN;AAED,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,aAAa,GAAG,YAAY,CAqGvE;AAED,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,SAAS,GAAG,kBAAkB,CAyC3E"}
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAA6C,MAAM,WAAW,CAAC;AA+BpF,OAAO,EAOL,mBAAmB,EAOpB,MAAM,YAAY,CAAC;AACpB,OAAO,EAaL,KAAK,SAAS,EACf,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAYV,kBAAkB,EAGnB,MAAM,iBAAiB,CAAC;AAQzB,MAAM,MAAM,aAAa,GAAG;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAyD3D,wBAAgB,qCAAqC,CACnD,OAAO,EAAE,OAAO,mBAAmB,GAAG,SAAS,GAC9C,IAAI,CAEN;AAED,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,aAAa,GAAG,YAAY,CAiHvE;AAED,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,SAAS,GAAG,kBAAkB,CAyC3E"}

package/dist/server.js

@@ -1,7 +1,8 @@
 import { randomUUID } from "node:crypto";
 import { existsSync, readFileSync, rmSync, writeFileSync } from "node:fs";
 import { createServer } from "node:http";
-import { extname, relative, join, resolve } from "node:path";
+import { extname, relative, join, resolve, dirname } from "node:path";
+import { spawn } from "node:child_process";
 import { fileURLToPath } from "node:url";
 import { generateText as defaultGenerateText } from "ai";
 import { getAskDbRuntimeConfig } from "@askdb/config";
@@ -13,6 +14,7 @@ import { openaiProvider } from "@askdb/ai-openai";
 import { ask, isBuiltInDialectId, loadSchema, suggestEnrichment, tenantScopeSchema, writeTenantPolicyMarkdown, tenantPolicyFrontmatterSchema, } from "@askdb/core";
 import { buildSchemaIndex, chunkContentHash, chunkSchema, createAiSdkEmbedder, createFileStore, createMemoryStore, createPgvectorStore, createRetriever, loadChunkerSourcesFromDir, } from "@askdb/rag";
 import { buildDefaultTableBody, buildFrontmatter, buildSuggestionContext, buildSuggestionTarget, buildTableDraft, loadWorkspace, replaceH2Section, replaceTableDescription, saveConcepts, saveTable, } from "@askdb/enrich";
+import { EXECUTE_DRIVER_REGISTRY, isDriverInstalled } from "./execute-registry.js";
 const ai = createAiRegistry([openaiProvider, azureProvider, googleProvider, anthropicProvider]);
 const CLIENT_DIR = fileURLToPath(new URL("./client/", import.meta.url));
 const STUDIO_RAG_MOCK_DIMENSIONS = 64;
@@ -105,6 +107,18 @@ export function createStudioServer(options) {
                 const result = deletePlaygroundHistoryEntry(state.schemaDir, id);
                 return writeJson(res, 200, result);
             }
+            if (req.method === "GET" && url.pathname === "/api/execute/status") {
+                const result = await getExecuteStatus();
+                return writeJson(res, 200, result);
+            }
+            if (req.method === "POST" && url.pathname === "/api/execute/install-driver") {
+                if (!isLoopbackRequest(req)) {
+                    return writeJson(res, 403, { error: { message: "Driver install is only available from loopback clients." } });
+                }
+                const body = await readJson(req);
+                const result = await installExecuteDriver(body, state.schemaDir);
+                return writeJson(res, 200, result);
+            }
             if (req.method === "POST" && url.pathname === "/api/execute") {
                 const body = await readJson(req);
                 const result = await executeQuery(body);
@@ -865,56 +879,171 @@ function parsePlaygroundHistoryEntry(body) {
     return body;
 }
 // ---------------------------------------------------------------------------
-// Execute endpoint helper
+// Execute endpoint helpers
 // ---------------------------------------------------------------------------
-async function executeQuery(body) {
-    const databaseUrl = getAskDbRuntimeConfig().studio.execute.databaseUrl;
-    if (!databaseUrl) {
-        return { ok: false, error: "ASKDB_STUDIO_DATABASE_URL is not configured" };
+async function getExecuteStatus() {
+    const rt = getAskDbRuntimeConfig();
+    const { provider, databaseUrl, file } = rt.studio.execute;
+    const def = EXECUTE_DRIVER_REGISTRY[provider];
+    const installed = await isDriverInstalled(def.packageName);
+    const connectionKind = provider === "sqlite" ? "file" : "url";
+    const configured = connectionKind === "file" ? Boolean(file) : Boolean(databaseUrl);
+    return {
+        provider,
+        label: def.label,
+        configured,
+        connectionKind,
+        packageName: def.packageName,
+        installed,
+        installCommand: def.installCommand,
+        canInstallFromStudio: true,
+        manualInstallReason: null,
+    };
+}
+async function installExecuteDriver(body, schemaDir) {
+    const rt = getAskDbRuntimeConfig();
+    const configuredProvider = rt.studio.execute.provider;
+    // Accept the configured provider or an explicit allowlisted provider.
+    let provider = configuredProvider;
+    if (isRecord(body) && typeof body.provider === "string") {
+        const requested = body.provider;
+        if (!(requested in EXECUTE_DRIVER_REGISTRY)) {
+            return {
+                ok: false,
+                provider: configuredProvider,
+                packageName: EXECUTE_DRIVER_REGISTRY[configuredProvider].packageName,
+                command: [],
+                error: `Unknown provider: ${body.provider}`,
+                installed: false,
+            };
+        }
+        provider = requested;
     }
-    if (!isRecord(body) || typeof body.sql !== "string" || body.sql.trim() === "") {
-        return { ok: false, error: "`sql` is required." };
+    const def = EXECUTE_DRIVER_REGISTRY[provider];
+    const { packageManager, command, args, manualReason } = detectPackageManager(schemaDir, def.packageName);
+    if (manualReason) {
+        return {
+            ok: false,
+            provider,
+            packageName: def.packageName,
+            command: [],
+            error: manualReason,
+            installed: false,
+        };
     }
-    const sql = body.sql;
-    const params = Array.isArray(body.params) ? body.params : [];
-    let pgMod;
-    try {
-        const mod = await import("pg");
-        pgMod = (mod.default ?? mod);
+    // Run the install.
+    const { stdout, stderr, code } = await spawnCommand(command, args, schemaDir);
+    const installed = await isDriverInstalled(def.packageName);
+    return {
+        ok: code === 0,
+        provider,
+        packageName: def.packageName,
+        command: [command, ...args],
+        stdout,
+        stderr,
+        error: code !== 0 ? `${packageManager} exited with code ${code}` : undefined,
+        installed,
+    };
+}
+function detectPackageManager(cwd, packageName) {
+    // Walk up from schemaDir to find the project root's lockfile.
+    const projectRoot = findProjectRoot(cwd);
+    if (!projectRoot) {
+        return {
+            packageManager: "",
+            command: "",
+            args: [],
+            manualReason: `Could not locate a package.json above ${cwd}. Run the install manually.`,
+        };
     }
-    catch {
-        return { ok: false, error: "The `pg` package is required for query execution. Install it with `pnpm add pg`." };
+    if (existsSync(join(projectRoot, "pnpm-lock.yaml"))) {
+        return {
+            packageManager: "pnpm",
+            command: "pnpm",
+            args: ["add", packageName],
+            manualReason: null,
+        };
     }
-    const client = new pgMod.Client({ connectionString: databaseUrl });
-    await client.connect();
-    const startMs = Date.now();
-    try {
-        await client.query("BEGIN READ ONLY");
-        const result = await client.query({ text: sql, values: params });
-        await client.query("COMMIT");
-        const durationMs = Date.now() - startMs;
-        const columns = result.fields.map((f) => f.name);
-        const rows = result.rows.map((r) => columns.map((c) => r[c]));
-        const truncated = rows.length > 500;
+    if (existsSync(join(projectRoot, "package-lock.json"))) {
         return {
-            ok: true,
-            columns,
-            rows: rows.slice(0, 500),
-            rowCount: result.rowCount ?? rows.length,
-            durationMs,
-            truncated,
+            packageManager: "npm",
+            command: "npm",
+            args: ["install", packageName],
+            manualReason: null,
         };
     }
-    catch (err) {
-        await client.query("ROLLBACK").catch(() => { });
+    if (existsSync(join(projectRoot, "yarn.lock"))) {
         return {
-            ok: false,
-            error: err instanceof Error ? err.message : String(err),
+            packageManager: "yarn",
+            command: "yarn",
+            args: ["add", packageName],
+            manualReason: null,
         };
     }
-    finally {
-        await client.end();
+    if (existsSync(join(projectRoot, "bun.lockb")) || existsSync(join(projectRoot, "bun.lock"))) {
+        return {
+            packageManager: "bun",
+            command: "bun",
+            args: ["add", packageName],
+            manualReason: null,
+        };
     }
+    return {
+        packageManager: "",
+        command: "",
+        args: [],
+        manualReason: `No recognized lockfile found in ${projectRoot}. Run: pnpm add ${packageName}`,
+    };
+}
+function findProjectRoot(startDir) {
+    let current = resolve(startDir);
+    for (let i = 0; i < 20; i++) {
+        if (existsSync(join(current, "package.json")))
+            return current;
+        const parent = dirname(current);
+        if (parent === current)
+            return null;
+        current = parent;
+    }
+    return null;
+}
+function spawnCommand(command, args, cwd) {
+    return new Promise((resolve) => {
+        const proc = spawn(command, args, { cwd, shell: false, stdio: ["ignore", "pipe", "pipe"] });
+        const stdoutChunks = [];
+        const stderrChunks = [];
+        proc.stdout.on("data", (chunk) => stdoutChunks.push(chunk));
+        proc.stderr.on("data", (chunk) => stderrChunks.push(chunk));
+        proc.on("close", (code) => {
+            resolve({
+                stdout: Buffer.concat(stdoutChunks).toString("utf8").slice(0, 4096),
+                stderr: Buffer.concat(stderrChunks).toString("utf8").slice(0, 4096),
+                code: code ?? 1,
+            });
+        });
+        proc.on("error", (err) => {
+            resolve({ stdout: "", stderr: err.message, code: 1 });
+        });
+    });
+}
+async function executeQuery(body) {
+    const rt = getAskDbRuntimeConfig();
+    const { provider, databaseUrl, file } = rt.studio.execute;
+    if (!isRecord(body) || typeof body.sql !== "string" || body.sql.trim() === "") {
+        return { ok: false, error: "`sql` is required." };
+    }
+    const sql = body.sql;
+    const params = Array.isArray(body.params) ? body.params : [];
+    const def = EXECUTE_DRIVER_REGISTRY[provider];
+    return def.execute({ connectionString: databaseUrl, file, sql, params });
+}
+function isLoopbackRequest(req) {
+    const socket = req.socket;
+    const remoteAddress = socket?.remoteAddress ?? "";
+    return (remoteAddress === "127.0.0.1" ||
+        remoteAddress === "::1" ||
+        remoteAddress === "::ffff:127.0.0.1" ||
+        remoteAddress === "localhost");
 }
 function readOptionalJson(path) {
     if (!existsSync(path))