databricks-axi 0.1.0 → 0.3.0

package/dist/src/commands/shared.d.ts

@@ -0,0 +1,28 @@
+import { AxiError } from "axi-sdk-js";
+import type { RunDatabricksOptions } from "../databricks.js";
+export type AxiStructuredOutput = Record<string, unknown>;
+export type AxiRenderable = string | AxiStructuredOutput;
+export type FlagSpec = Record<string, "value" | "boolean">;
+export type Flags = Map<string, string | boolean>;
+export declare function spawnOpts(flags: Flags): RunDatabricksOptions;
+/** Suffix for suggested follow-up commands so they hit the same workspace. */
+export declare function profileSuffix(profile: unknown): string;
+/** Guard for endpoints whose result gets dereferenced — empty stdout (null)
+ * becomes a structured UPSTREAM_ERROR instead of a raw TypeError. */
+export declare function assertObject<T>(parsed: unknown, label: string): T;
+/**
+ * The Go CLI prints either a bare item array (>= 0.298) or the response
+ * object ({items, ...}) depending on version — tolerate both.
+ */
+export declare function asList(parsed: unknown, key: string): AxiStructuredOutput[];
+/** Helpers whose usage errors point at `databricks-axi <domain> --help`. */
+export declare function domainHelpers(domain: string): {
+    usage: (message: string, extraHelp?: string[]) => AxiError;
+    parseArgs: (args: string[], spec: FlagSpec) => {
+        positional: string[];
+        flags: Flags;
+    };
+    parseIntFlag: (flags: Flags, name: string, fallback: number, min?: number) => number;
+    requireId: (positional: string[], usageText: string, pattern?: RegExp) => string;
+    renderRows: (items: AxiStructuredOutput[], flags: Flags, defaults: string[]) => AxiStructuredOutput[];
+};

package/dist/src/commands/shared.js

@@ -0,0 +1,105 @@
+import { parseArgs as nodeParseArgs } from "node:util";
+import { AxiError } from "axi-sdk-js";
+export function spawnOpts(flags) {
+    const profile = flags.get("profile");
+    return typeof profile === "string" ? { profile } : {};
+}
+/** Suffix for suggested follow-up commands so they hit the same workspace. */
+export function profileSuffix(profile) {
+    return typeof profile === "string" ? ` --profile ${profile}` : "";
+}
+/** Guard for endpoints whose result gets dereferenced — empty stdout (null)
+ * becomes a structured UPSTREAM_ERROR instead of a raw TypeError. */
+export function assertObject(parsed, label) {
+    if (parsed === null || typeof parsed !== "object") {
+        throw new AxiError(`${label} returned an empty response`, "UPSTREAM_ERROR", ["Retry, or check workspace availability"]);
+    }
+    return parsed;
+}
+/**
+ * The Go CLI prints either a bare item array (>= 0.298) or the response
+ * object ({items, ...}) depending on version — tolerate both.
+ */
+export function asList(parsed, key) {
+    if (Array.isArray(parsed)) {
+        return parsed;
+    }
+    const obj = (parsed ?? {});
+    return obj[key] ?? [];
+}
+/** Helpers whose usage errors point at `databricks-axi <domain> --help`. */
+export function domainHelpers(domain) {
+    const usage = (message, extraHelp = []) => new AxiError(message, "VALIDATION_ERROR", [
+        ...extraHelp,
+        `Run \`databricks-axi ${domain} --help\``,
+    ]);
+    const parseArgs = (args, spec) => {
+        const options = Object.fromEntries(Object.entries(spec).map(([name, kind]) => [
+            name,
+            { type: kind === "value" ? "string" : "boolean" },
+        ]));
+        try {
+            const { values, positionals } = nodeParseArgs({
+                args,
+                options,
+                strict: true,
+                allowPositionals: true,
+            });
+            return {
+                positional: positionals,
+                flags: new Map(Object.entries(values)),
+            };
+        }
+        catch (error) {
+            const valid = Object.keys(spec)
+                .map((f) => `--${f}`)
+                .join(", ");
+            // First sentence only — node appends a long "--" placement hint,
+            // sometimes as extra lines (e.g. the ambiguous-option error).
+            const message = error.message.split("\n")[0].split(". ")[0];
+            throw usage(message, [`Valid flags: ${valid}`]);
+        }
+    };
+    const parseIntFlag = (flags, name, fallback, min = 1) => {
+        const raw = flags.get(name);
+        if (raw === undefined) {
+            return fallback;
+        }
+        const value = Number(raw);
+        if (!Number.isInteger(value) || value < min) {
+            const kind = min === 0 ? "non-negative" : "positive";
+            throw usage(`--${name} must be a ${kind} integer, got: ${String(raw)}`);
+        }
+        return value;
+    };
+    const requireId = (positional, usageText, pattern) => {
+        const id = positional[0];
+        if (!id || positional.length > 1 || (pattern && !pattern.test(id))) {
+            throw usage(`Usage: databricks-axi ${usageText}`);
+        }
+        return id;
+    };
+    /** Apply --fields (raw top-level keys) or the default field list. */
+    const renderRows = (items, flags, defaults) => {
+        const spec = flags.get("fields");
+        const fields = typeof spec === "string"
+            ? spec
+                .split(",")
+                .map((f) => f.trim())
+                .filter(Boolean)
+            : defaults;
+        if (typeof spec === "string" && items.length > 0) {
+            const known = new Set(items.flatMap((item) => Object.keys(item)));
+            for (const field of fields) {
+                if (!known.has(field)) {
+                    throw usage(`Unknown field: ${field}`, [
+                        `Available fields: ${[...known].sort().join(", ")}`,
+                    ]);
+                }
+            }
+        }
+        return items.map((item) => Object.fromEntries(fields.map((field) => [field, item[field] ?? ""])));
+    };
+    return { usage, parseArgs, parseIntFlag, requireId, renderRows };
+}
+//# sourceMappingURL=shared.js.map

package/dist/src/commands/shared.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shared.js","sourceRoot":"","sources":["../../../src/commands/shared.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,IAAI,aAAa,EAAE,MAAM,WAAW,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAWtC,MAAM,UAAU,SAAS,CAAC,KAAY;IACpC,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACrC,OAAO,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;AACxD,CAAC;AAED,8EAA8E;AAC9E,MAAM,UAAU,aAAa,CAAC,OAAgB;IAC5C,OAAO,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,cAAc,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;AACpE,CAAC;AAED;qEACqE;AACrE,MAAM,UAAU,YAAY,CAAI,MAAe,EAAE,KAAa;IAC5D,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAClD,MAAM,IAAI,QAAQ,CAChB,GAAG,KAAK,6BAA6B,EACrC,gBAAgB,EAChB,CAAC,wCAAwC,CAAC,CAC3C,CAAC;IACJ,CAAC;IACD,OAAO,MAAW,CAAC;AACrB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,MAAM,CAAC,MAAe,EAAE,GAAW;IACjD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,MAA+B,CAAC;IACzC,CAAC;IACD,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,EAAE,CAAwB,CAAC;IAClD,OAAQ,GAAG,CAAC,GAAG,CAAuC,IAAI,EAAE,CAAC;AAC/D,CAAC;AAED,4EAA4E;AAC5E,MAAM,UAAU,aAAa,CAAC,MAAc;IAC1C,MAAM,KAAK,GAAG,CAAC,OAAe,EAAE,YAAsB,EAAE,EAAY,EAAE,CACpE,IAAI,QAAQ,CAAC,OAAO,EAAE,kBAAkB,EAAE;QACxC,GAAG,SAAS;QACZ,wBAAwB,MAAM,WAAW;KAC1C,CAAC,CAAC;IAEL,MAAM,SAAS,GAAG,CAChB,IAAc,EACd,IAAc,EAC0B,EAAE;QAC1C,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAChC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;YACzC,IAAI;YACJ,EAAE,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAE,QAAkB,CAAC,CAAC,CAAE,SAAmB,EAAE;SACxE,CAAC,CACH,CAAC;QACF,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,aAAa,CAAC;gBAC5C,IAAI;gBACJ,OAAO;gBACP,MAAM,EAAE,IAAI;gBACZ,gBAAgB,EAAE,IAAI;aACvB,CAAC,CAAC;YACH,OAAO;gBACL,UAAU,EAAE,WAAW;gBACvB,KAAK,EAAE,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAiC,CAAC;aACvE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;iBAC5B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;iBACpB,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,iEAAiE;YACjE,8DAA8D;YAC9D,MAAM,OAAO,GAAI,KAAe,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YACvE,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC,gBAAgB,KAAK,EAAE,CAAC,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG,CACnB,KAAY,EACZ,IAAY,EACZ,QAAgB,EAChB,GAAG,GAAG,CAAC,EACC,EAAE;QACV,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC5B,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACtB,OAAO,QAAQ,CAAC;QAClB,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC1B,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,GAAG,EAAE,CAAC;YAC5C,MAAM,IAAI,GAAG,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,UAAU,CAAC;YACrD,MAAM,KAAK,CAAC,KAAK,IAAI,cAAc,IAAI,kBAAkB,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1E,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC,CAAC;IAEF,MAAM,SAAS,GAAG,CAChB,UAAoB,EACpB,SAAiB,EACjB,OAAgB,EACR,EAAE;QACV,MAAM,EAAE,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,CAAC,EAAE,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACnE,MAAM,KAAK,CAAC,yBAAyB,SAAS,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;IAEF,qEAAqE;IACrE,MAAM,UAAU,GAAG,CACjB,KAA4B,EAC5B,KAAY,EACZ,QAAkB,EACK,EAAE;QACzB,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,MAAM,MAAM,GACV,OAAO,IAAI,KAAK,QAAQ;YACtB,CAAC,CAAC,IAAI;iBACD,KAAK,CAAC,GAAG,CAAC;iBACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;iBACpB,MAAM,CAAC,OAAO,CAAC;YACpB,CAAC,CAAC,QAAQ,CAAC;QACf,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjD,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClE,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAC3B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;oBACtB,MAAM,KAAK,CAAC,kBAAkB,KAAK,EAAE,EAAE;wBACrC,qBAAqB,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;qBACpD,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACxB,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CACtE,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,YAAY,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;AACnE,CAAC"}

package/dist/src/commands/sql.d.ts

@@ -0,0 +1,6 @@
+import { type AxiRenderable } from "./shared.js";
+export declare const SQL_HELP = "usage: databricks-axi sql <subcommand> [args] [flags]\nsubcommands[6]:\n  warehouses [--fields a,b]\n  warehouses view <id>\n  warehouses start <id> [--wait]\n  warehouses stop <id> [--wait]\n  exec \"<query>\" [--warehouse <id>] [--limit N] [--timeout S] [--full]\n  statement view <statement_id>\nflags:\n  --profile <name>  databricks auth profile passthrough\nexamples:\n  databricks-axi sql warehouses\n  databricks-axi sql exec \"SELECT count(*) FROM workspace.default.my_table\"\n  databricks-axi sql statement view <statement_id>\nnotes:\n  exec picks the warehouse automatically when the workspace has exactly one\n  exec waits up to --timeout seconds (default 120), then exits 0 with the\n  statement id \u2014 resume with sql statement view\n  warehouse start/stop are async by default; --wait blocks up to ~20 min\n  never put secret values in the query text: it lands on child argv\n  (visible in ps) via the api substrate\n";
+export declare const sqlPoll: {
+    sleep: (ms: number) => Promise<void>;
+};
+export declare function sqlCommand(args: string[]): Promise<AxiRenderable>;

package/dist/src/commands/sql.js

@@ -0,0 +1,268 @@
+import { AxiError } from "axi-sdk-js";
+import { runDatabricks, runDatabricksApi, } from "../databricks.js";
+import { redactSecrets } from "../errors.js";
+import { asList, assertObject, domainHelpers, profileSuffix, spawnOpts, } from "./shared.js";
+const { usage, parseArgs, parseIntFlag, requireId, renderRows } = domainHelpers("sql");
+export const SQL_HELP = `usage: databricks-axi sql <subcommand> [args] [flags]
+subcommands[6]:
+  warehouses [--fields a,b]
+  warehouses view <id>
+  warehouses start <id> [--wait]
+  warehouses stop <id> [--wait]
+  exec "<query>" [--warehouse <id>] [--limit N] [--timeout S] [--full]
+  statement view <statement_id>
+flags:
+  --profile <name>  databricks auth profile passthrough
+examples:
+  databricks-axi sql warehouses
+  databricks-axi sql exec "SELECT count(*) FROM workspace.default.my_table"
+  databricks-axi sql statement view <statement_id>
+notes:
+  exec picks the warehouse automatically when the workspace has exactly one
+  exec waits up to --timeout seconds (default 120), then exits 0 with the
+  statement id — resume with sql statement view
+  warehouse start/stop are async by default; --wait blocks up to ~20 min
+  never put secret values in the query text: it lands on child argv
+  (visible in ps) via the api substrate
+`;
+// Injectable so poll tests run at full speed.
+export const sqlPoll = {
+    sleep: (ms) => new Promise((r) => setTimeout(r, ms)),
+};
+const STATEMENTS_PATH = "/api/2.0/sql/statements";
+const SUBMIT_TIMEOUT_MS = 60_000; // must exceed the API-side 50s wait_timeout
+const WAIT_TIMEOUT_MS = 25 * 60_000; // upstream blocks up to 20 min on --wait
+const POLL_INTERVAL_MS = 2_000;
+const DEFAULT_ROW_LIMIT = 100;
+const DEFAULT_BUDGET_S = 120;
+export async function sqlCommand(args) {
+    const [sub, ...rest] = args;
+    switch (sub) {
+        case "warehouses":
+            switch (rest[0]) {
+                case "view":
+                    return warehousesView(rest.slice(1));
+                case "start":
+                case "stop":
+                    return warehousesStartStop(rest[0], rest.slice(1));
+                default:
+                    return warehousesList(rest);
+            }
+        case "exec":
+            return sqlExec(rest);
+        case "statement":
+            if (rest[0] !== "view") {
+                throw usage("Usage: databricks-axi sql statement view <statement_id>");
+            }
+            return statementView(rest.slice(1));
+        default:
+            throw usage(sub ? `Unknown sql subcommand: ${sub}` : "sql requires a subcommand");
+    }
+}
+// --- warehouses ---
+function warehouseSize(w) {
+    const size = w.cluster_size ?? "";
+    return w.enable_serverless_compute ? `${size} (serverless)`.trim() : size;
+}
+async function listWarehouses(opts) {
+    const parsed = await runDatabricks(["warehouses", "list"], opts);
+    return asList(parsed, "warehouses");
+}
+async function warehousesList(args) {
+    const { positional, flags } = parseArgs(args, {
+        profile: "value",
+        fields: "value",
+    });
+    if (positional.length > 0) {
+        throw usage(`sql warehouses takes no arguments, got: ${positional[0]}`);
+    }
+    const items = await listWarehouses(spawnOpts(flags));
+    const p = profileSuffix(flags.get("profile"));
+    if (items.length === 0) {
+        return {
+            warehouses: [],
+            status: "no SQL warehouses in this workspace",
+            help: ["Create one in the workspace UI: SQL Warehouses > Create"],
+        };
+    }
+    const flattened = items.map((w) => ({ ...w, size: warehouseSize(w) }));
+    const rows = renderRows(flattened, flags, ["id", "name", "state", "size"]);
+    const help = [`databricks-axi sql exec "<query>" --warehouse <id>${p}`];
+    const stopped = items.find((w) => w.state === "STOPPED");
+    if (stopped) {
+        help.push(`databricks-axi sql warehouses start ${stopped.id}${p}`);
+    }
+    return { warehouses: rows, count: rows.length, help };
+}
+async function warehousesView(args) {
+    const { positional, flags } = parseArgs(args, { profile: "value" });
+    const id = requireId(positional, "sql warehouses view <id>");
+    const w = assertObject(await runDatabricks(["warehouses", "get", id], spawnOpts(flags)), "warehouses get");
+    const p = profileSuffix(flags.get("profile"));
+    return {
+        id: w.id,
+        name: w.name,
+        state: w.state,
+        size: warehouseSize(w),
+        auto_stop_mins: w.auto_stop_mins,
+        creator_name: w.creator_name,
+        help: [
+            w.state === "RUNNING"
+                ? `databricks-axi sql warehouses stop ${id}${p}`
+                : `databricks-axi sql warehouses start ${id}${p}`,
+            `databricks-axi sql exec "<query>" --warehouse ${id}${p}`,
+        ],
+    };
+}
+async function warehousesStartStop(verb, args) {
+    const { positional, flags } = parseArgs(args, {
+        profile: "value",
+        wait: "boolean",
+    });
+    const id = requireId(positional, `sql warehouses ${verb} <id> [--wait]`);
+    const wait = flags.get("wait") === true;
+    const p = profileSuffix(flags.get("profile"));
+    const argv = ["warehouses", verb, id];
+    if (!wait) {
+        argv.push("--no-wait");
+    }
+    // Upstream is already idempotent here: start-on-RUNNING / stop-on-STOPPED
+    // exit 0 with no output (pinned live 2026-07-07), so no no-op mapping.
+    await runDatabricks(argv, {
+        ...spawnOpts(flags),
+        ...(wait ? { timeoutMs: WAIT_TIMEOUT_MS } : {}),
+        timeoutHelp: [
+            `The ${verb} may have applied despite the timeout — check state:`,
+            `databricks-axi sql warehouses view ${id}${p}`,
+        ],
+    });
+    return {
+        id,
+        status: `${verb} requested`,
+        help: [`databricks-axi sql warehouses view ${id}${p}`],
+    };
+}
+// --- exec / statement view ---
+async function sqlExec(args) {
+    const { positional, flags } = parseArgs(args, {
+        profile: "value",
+        warehouse: "value",
+        limit: "value",
+        timeout: "value",
+        full: "boolean",
+    });
+    const query = positional[0];
+    if (!query || positional.length > 1) {
+        throw usage('Usage: databricks-axi sql exec "<query>" [--warehouse <id>]');
+    }
+    const limit = parseIntFlag(flags, "limit", DEFAULT_ROW_LIMIT);
+    const budgetS = parseIntFlag(flags, "timeout", DEFAULT_BUDGET_S, 0);
+    const full = flags.get("full") === true;
+    const opts = spawnOpts(flags);
+    const p = profileSuffix(flags.get("profile"));
+    const warehouseId = await resolveWarehouse(flags.get("warehouse"), opts);
+    const deadline = Date.now() + budgetS * 1000;
+    const waitTimeoutS = Math.max(5, Math.min(budgetS, 50));
+    let stmt = assertObject(await runDatabricksApi("post", STATEMENTS_PATH, JSON.stringify({
+        statement: query,
+        warehouse_id: warehouseId,
+        wait_timeout: `${waitTimeoutS}s`,
+        on_wait_timeout: "CONTINUE",
+        disposition: "INLINE",
+        format: "JSON_ARRAY",
+        row_limit: limit,
+    }), { ...opts, timeoutMs: SUBMIT_TIMEOUT_MS }), "sql statement submit");
+    while (isPendingState(stmt)) {
+        if (Date.now() >= deadline) {
+            return {
+                statement_id: stmt.statement_id,
+                state: stmt.status?.state,
+                status: `still running — stopped waiting after ${budgetS}s (statement continues server-side)`,
+                help: [`databricks-axi sql statement view ${stmt.statement_id}${p}`],
+            };
+        }
+        await sqlPoll.sleep(POLL_INTERVAL_MS);
+        stmt = assertObject(await runDatabricksApi("get", `${STATEMENTS_PATH}/${stmt.statement_id}`, undefined, opts), "sql statement poll");
+    }
+    return renderTerminalStatement(stmt, { full, limit, opts, p });
+}
+async function statementView(args) {
+    const { positional, flags } = parseArgs(args, {
+        profile: "value",
+        full: "boolean",
+    });
+    const id = requireId(positional, "sql statement view <statement_id>");
+    const opts = spawnOpts(flags);
+    const p = profileSuffix(flags.get("profile"));
+    const stmt = assertObject(await runDatabricksApi("get", `${STATEMENTS_PATH}/${id}`, undefined, opts), "sql statement get");
+    if (isPendingState(stmt)) {
+        return {
+            statement_id: stmt.statement_id ?? id,
+            state: stmt.status?.state,
+            help: [`databricks-axi sql statement view ${id}${p}`],
+        };
+    }
+    return renderTerminalStatement(stmt, {
+        full: flags.get("full") === true,
+        opts,
+        p,
+    });
+}
+function isPendingState(stmt) {
+    const state = stmt.status?.state;
+    return state === "PENDING" || state === "RUNNING";
+}
+async function resolveWarehouse(flagValue, opts) {
+    if (typeof flagValue === "string") {
+        return flagValue;
+    }
+    const warehouses = await listWarehouses(opts);
+    if (warehouses.length === 1 && warehouses[0].id) {
+        return warehouses[0].id;
+    }
+    if (warehouses.length === 0) {
+        throw new AxiError("no SQL warehouses in this workspace", "NOT_FOUND", [
+            "Create one in the workspace UI: SQL Warehouses > Create",
+        ]);
+    }
+    throw usage("multiple SQL warehouses — pick one with --warehouse <id>", warehouses.map((w) => `${w.id}: ${w.name}`));
+}
+async function renderTerminalStatement(stmt, ctx) {
+    const state = stmt.status?.state ?? "UNKNOWN";
+    const id = stmt.statement_id;
+    if (state !== "SUCCEEDED") {
+        const detail = stmt.status?.error?.message ?? `statement ${state}`;
+        throw new AxiError(redactSecrets(detail), "SQL_ERROR", id ? [`databricks-axi sql statement view ${id}${ctx.p}`] : []);
+    }
+    const manifest = stmt.manifest ?? {};
+    const columns = (manifest.schema?.columns ?? []).map((c) => `${c.name}:${c.type_text}`);
+    let rows = stmt.result?.data_array ?? [];
+    const totalChunks = manifest.total_chunk_count ?? 1;
+    const totalRows = manifest.total_row_count ?? rows.length;
+    if (ctx.full && totalChunks > 1 && id) {
+        for (let chunk = 1; chunk < totalChunks; chunk++) {
+            const next = assertObject(await runDatabricksApi("get", `${STATEMENTS_PATH}/${id}/result/chunks/${chunk}`, undefined, ctx.opts), "sql statement chunk fetch");
+            rows = rows.concat(next.data_array ?? []);
+        }
+    }
+    const out = {
+        statement_id: id,
+        columns,
+        rows,
+        total_row_count: totalRows,
+    };
+    const notes = [];
+    if (!ctx.full && totalChunks > 1) {
+        notes.push(`showing ${rows.length} of ${totalRows} rows — rerun with --full`);
+    }
+    if (manifest.truncated) {
+        // Server-side row_limit hit: --full can never exceed the submitted cap,
+        // so report what was actually delivered, not the true upstream total.
+        notes.push(`result capped at ${rows.length} rows — rerun with --limit ${(ctx.limit ?? rows.length) * 2}`);
+    }
+    if (notes.length > 0) {
+        out.truncated = notes.join("; ");
+    }
+    return out;
+}
+//# sourceMappingURL=sql.js.map

package/dist/src/commands/sql.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql.js","sourceRoot":"","sources":["../../../src/commands/sql.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,EACL,aAAa,EACb,gBAAgB,GAEjB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EACL,MAAM,EACN,YAAY,EACZ,aAAa,EACb,aAAa,EACb,SAAS,GAGV,MAAM,aAAa,CAAC;AAErB,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,YAAY,EAAE,SAAS,EAAE,UAAU,EAAE,GAC7D,aAAa,CAAC,KAAK,CAAC,CAAC;AAEvB,MAAM,CAAC,MAAM,QAAQ,GAAG;;;;;;;;;;;;;;;;;;;;;CAqBvB,CAAC;AAEF,8CAA8C;AAC9C,MAAM,CAAC,MAAM,OAAO,GAAG;IACrB,KAAK,EAAE,CAAC,EAAU,EAAE,EAAE,CAAC,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;CACnE,CAAC;AAEF,MAAM,eAAe,GAAG,yBAAyB,CAAC;AAClD,MAAM,iBAAiB,GAAG,MAAM,CAAC,CAAC,4CAA4C;AAC9E,MAAM,eAAe,GAAG,EAAE,GAAG,MAAM,CAAC,CAAC,yCAAyC;AAC9E,MAAM,gBAAgB,GAAG,KAAK,CAAC;AAC/B,MAAM,iBAAiB,GAAG,GAAG,CAAC;AAC9B,MAAM,gBAAgB,GAAG,GAAG,CAAC;AA4B7B,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,IAAc;IAC7C,MAAM,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC;IAC5B,QAAQ,GAAG,EAAE,CAAC;QACZ,KAAK,YAAY;YACf,QAAQ,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChB,KAAK,MAAM;oBACT,OAAO,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACvC,KAAK,OAAO,CAAC;gBACb,KAAK,MAAM;oBACT,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACrD;oBACE,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC;YAChC,CAAC;QACH,KAAK,MAAM;YACT,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;QACvB,KAAK,WAAW;YACd,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,MAAM,EAAE,CAAC;gBACvB,MAAM,KAAK,CAAC,yDAAyD,CAAC,CAAC;YACzE,CAAC;YACD,OAAO,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QACtC;YACE,MAAM,KAAK,CACT,GAAG,CAAC,CAAC,CAAC,2BAA2B,GAAG,EAAE,CAAC,CAAC,CAAC,2BAA2B,CACrE,CAAC;IACN,CAAC;AACH,CAAC;AAED,qBAAqB;AAErB,SAAS,aAAa,CAAC,CAAe;IACpC,MAAM,IAAI,GAAG,CAAC,CAAC,YAAY,IAAI,EAAE,CAAC;IAClC,OAAO,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,GAAG,IAAI,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AAC5E,CAAC;AAED,KAAK,UAAU,cAAc,CAC3B,IAA0B;IAE1B,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,CAAC,YAAY,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;IACjE,OAAO,MAAM,CAAC,MAAM,EAAE,YAAY,CAAmB,CAAC;AACxD,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,IAAc;IAC1C,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC,IAAI,EAAE;QAC5C,OAAO,EAAE,OAAO;QAChB,MAAM,EAAE,OAAO;KAChB,CAAC,CAAC;IACH,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1B,MAAM,KAAK,CAAC,2CAA2C,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC1E,CAAC;IACD,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;IACrD,MAAM,CAAC,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IAC9C,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO;YACL,UAAU,EAAE,EAAE;YACd,MAAM,EAAE,qCAAqC;YAC7C,IAAI,EAAE,CAAC,yDAAyD,CAAC;SAClE,CAAC;IACJ,CAAC;IACD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;IAC3E,MAAM,IAAI,GAAG,CAAC,qDAAqD,CAAC,EAAE,CAAC,CAAC;IACxE,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;IACzD,IAAI,OAAO,EAAE,CAAC;QACZ,IAAI,CAAC,IAAI,CAAC,uCAAuC,OAAO,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;IACrE,CAAC;IACD,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC;AACxD,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,IAAc;IAC1C,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;IACpE,MAAM,EAAE,GAAG,SAAS,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC;IAC7D,MAAM,CAAC,GAAG,YAAY,CACpB,MAAM,aAAa,CAAC,CAAC,YAAY,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC,EAChE,gBAAgB,CACjB,CAAC;IACF,MAAM,CAAC,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IAC9C,OAAO;QACL,EAAE,EAAE,CAAC,CAAC,EAAE;QACR,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,KAAK,EAAE,CAAC,CAAC,KAAK;QACd,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC;QACtB,cAAc,EAAE,CAAC,CAAC,cAAc;QAChC,YAAY,EAAE,CAAC,CAAC,YAAY;QAC5B,IAAI,EAAE;YACJ,CAAC,CAAC,KAAK,KAAK,SAAS;gBACnB,CAAC,CAAC,sCAAsC,EAAE,GAAG,CAAC,EAAE;gBAChD,CAAC,CAAC,uCAAuC,EAAE,GAAG,CAAC,EAAE;YACnD,iDAAiD,EAAE,GAAG,CAAC,EAAE;SAC1D;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,mBAAmB,CAChC,IAAsB,EACtB,IAAc;IAEd,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC,IAAI,EAAE;QAC5C,OAAO,EAAE,OAAO;QAChB,IAAI,EAAE,SAAS;KAChB,CAAC,CAAC;IACH,MAAM,EAAE,GAAG,SAAS,CAAC,UAAU,EAAE,kBAAkB,IAAI,gBAAgB,CAAC,CAAC;IACzE,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC;IACxC,MAAM,CAAC,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IAC9C,MAAM,IAAI,GAAG,CAAC,YAAY,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;IACtC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACzB,CAAC;IACD,0EAA0E;IAC1E,uEAAuE;IACvE,MAAM,aAAa,CAAC,IAAI,EAAE;QACxB,GAAG,SAAS,CAAC,KAAK,CAAC;QACnB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,WAAW,EAAE;YACX,OAAO,IAAI,sDAAsD;YACjE,sCAAsC,EAAE,GAAG,CAAC,EAAE;SAC/C;KACF,CAAC,CAAC;IACH,OAAO;QACL,EAAE;QACF,MAAM,EAAE,GAAG,IAAI,YAAY;QAC3B,IAAI,EAAE,CAAC,sCAAsC,EAAE,GAAG,CAAC,EAAE,CAAC;KACvD,CAAC;AACJ,CAAC;AAED,gCAAgC;AAEhC,KAAK,UAAU,OAAO,CAAC,IAAc;IACnC,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC,IAAI,EAAE;QAC5C,OAAO,EAAE,OAAO;QAChB,SAAS,EAAE,OAAO;QAClB,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,OAAO;QAChB,IAAI,EAAE,SAAS;KAChB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;IAC5B,IAAI,CAAC,KAAK,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpC,MAAM,KAAK,CAAC,6DAA6D,CAAC,CAAC;IAC7E,CAAC;IACD,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,iBAAiB,CAAC,CAAC;IAC9D,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC;IACpE,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC;IACxC,MAAM,IAAI,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;IAC9B,MAAM,CAAC,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IAC9C,MAAM,WAAW,GAAG,MAAM,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,CAAC;IAEzE,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,GAAG,IAAI,CAAC;IAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IACxD,IAAI,IAAI,GAAG,YAAY,CACrB,MAAM,gBAAgB,CACpB,MAAM,EACN,eAAe,EACf,IAAI,CAAC,SAAS,CAAC;QACb,SAAS,EAAE,KAAK;QAChB,YAAY,EAAE,WAAW;QACzB,YAAY,EAAE,GAAG,YAAY,GAAG;QAChC,eAAe,EAAE,UAAU;QAC3B,WAAW,EAAE,QAAQ;QACrB,MAAM,EAAE,YAAY;QACpB,SAAS,EAAE,KAAK;KACjB,CAAC,EACF,EAAE,GAAG,IAAI,EAAE,SAAS,EAAE,iBAAiB,EAAE,CAC1C,EACD,sBAAsB,CACvB,CAAC;IAEF,OAAO,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5B,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC3B,OAAO;gBACL,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,KAAK;gBACzB,MAAM,EAAE,yCAAyC,OAAO,qCAAqC;gBAC7F,IAAI,EAAE,CAAC,qCAAqC,IAAI,CAAC,YAAY,GAAG,CAAC,EAAE,CAAC;aACrE,CAAC;QACJ,CAAC;QACD,MAAM,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACtC,IAAI,GAAG,YAAY,CACjB,MAAM,gBAAgB,CACpB,KAAK,EACL,GAAG,eAAe,IAAI,IAAI,CAAC,YAAY,EAAE,EACzC,SAAS,EACT,IAAI,CACL,EACD,oBAAoB,CACrB,CAAC;IACJ,CAAC;IACD,OAAO,uBAAuB,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;AACjE,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,IAAc;IACzC,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC,IAAI,EAAE;QAC5C,OAAO,EAAE,OAAO;QAChB,IAAI,EAAE,SAAS;KAChB,CAAC,CAAC;IACH,MAAM,EAAE,GAAG,SAAS,CAAC,UAAU,EAAE,mCAAmC,CAAC,CAAC;IACtE,MAAM,IAAI,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;IAC9B,MAAM,CAAC,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IAC9C,MAAM,IAAI,GAAG,YAAY,CACvB,MAAM,gBAAgB,CAAC,KAAK,EAAE,GAAG,eAAe,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,EAC1E,mBAAmB,CACpB,CAAC;IACF,IAAI,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QACzB,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,EAAE;YACrC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,KAAK;YACzB,IAAI,EAAE,CAAC,qCAAqC,EAAE,GAAG,CAAC,EAAE,CAAC;SACtD,CAAC;IACJ,CAAC;IACD,OAAO,uBAAuB,CAAC,IAAI,EAAE;QACnC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI;QAChC,IAAI;QACJ,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,SAAS,cAAc,CAAC,IAAkB;IACxC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC;IACjC,OAAO,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,SAAS,CAAC;AACpD,CAAC;AAED,KAAK,UAAU,gBAAgB,CAC7B,SAAkB,EAClB,IAA0B;IAE1B,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;QAClC,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;IAC9C,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QAChD,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC1B,CAAC;IACD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,QAAQ,CAAC,qCAAqC,EAAE,WAAW,EAAE;YACrE,yDAAyD;SAC1D,CAAC,CAAC;IACL,CAAC;IACD,MAAM,KAAK,CACT,0DAA0D,EAC1D,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAC5C,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,IAAkB,EAClB,GAKC;IAED,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,SAAS,CAAC;IAC9C,MAAM,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC;IAC7B,IAAI,KAAK,KAAK,WAAW,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,IAAI,aAAa,KAAK,EAAE,CAAC;QACnE,MAAM,IAAI,QAAQ,CAChB,aAAa,CAAC,MAAM,CAAC,EACrB,WAAW,EACX,EAAE,CAAC,CAAC,CAAC,CAAC,qCAAqC,EAAE,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAC9D,CAAC;IACJ,CAAC;IACD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;IACrC,MAAM,OAAO,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC,GAAG,CAClD,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,SAAS,EAAE,CAClC,CAAC;IACF,IAAI,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE,UAAU,IAAI,EAAE,CAAC;IACzC,MAAM,WAAW,GAAG,QAAQ,CAAC,iBAAiB,IAAI,CAAC,CAAC;IACpD,MAAM,SAAS,GAAG,QAAQ,CAAC,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC;IAC1D,IAAI,GAAG,CAAC,IAAI,IAAI,WAAW,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC;QACtC,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,WAAW,EAAE,KAAK,EAAE,EAAE,CAAC;YACjD,MAAM,IAAI,GAAG,YAAY,CACvB,MAAM,gBAAgB,CACpB,KAAK,EACL,GAAG,eAAe,IAAI,EAAE,kBAAkB,KAAK,EAAE,EACjD,SAAS,EACT,GAAG,CAAC,IAAI,CACT,EACD,2BAA2B,CAC5B,CAAC;YACF,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IACD,MAAM,GAAG,GAAwB;QAC/B,YAAY,EAAE,EAAE;QAChB,OAAO;QACP,IAAI;QACJ,eAAe,EAAE,SAAS;KAC3B,CAAC;IACF,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACjC,KAAK,CAAC,IAAI,CACR,WAAW,IAAI,CAAC,MAAM,OAAO,SAAS,2BAA2B,CAClE,CAAC;IACJ,CAAC;IACD,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;QACvB,wEAAwE;QACxE,sEAAsE;QACtE,KAAK,CAAC,IAAI,CACR,oBAAoB,IAAI,CAAC,MAAM,8BAC7B,CAAC,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAC/B,EAAE,CACH,CAAC;IACJ,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrB,GAAG,CAAC,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC"}

package/dist/src/databricks.d.ts

@@ -0,0 +1,18 @@
+export type RunDatabricksOptions = {
+    profile?: string;
+    timeoutMs?: number;
+    /** TIMEOUT suggestions — mutations pass a state check, not "retry". */
+    timeoutHelp?: string[];
+};
+/**
+ * Run the official databricks CLI and return its parsed JSON output.
+ * Array argv only (never a shell), stdin ignored, hard timeout, always
+ * `-o json`. All failures surface as AxiError.
+ */
+export declare function runDatabricks(args: string[], opts?: RunDatabricksOptions): Promise<unknown>;
+/**
+ * REST passthrough over the CLI's `api` subcommand. Rule for all callers:
+ * `body` lands on child argv (visible in `ps`) — secret-bearing bodies are
+ * forbidden (the secrets domain uses upstream's stdin mechanism instead).
+ */
+export declare function runDatabricksApi(method: string, path: string, body?: string, opts?: RunDatabricksOptions): Promise<unknown>;

package/dist/src/databricks.js

@@ -0,0 +1,105 @@
+import { spawn } from "node:child_process";
+import { AxiError } from "axi-sdk-js";
+import { mapUpstreamError, redactSecrets } from "./errors.js";
+const DEFAULT_TIMEOUT_MS = 30_000;
+const MIN_MINOR_VERSION = 298; // databricks CLI floor: 0.298 (pre-0.298 pagination flags differ)
+const INSTALL_HELP = "Install it: https://docs.databricks.com/dev-tools/cli/install";
+/**
+ * Run the official databricks CLI and return its parsed JSON output.
+ * Array argv only (never a shell), stdin ignored, hard timeout, always
+ * `-o json`. All failures surface as AxiError.
+ */
+export async function runDatabricks(args, opts = {}) {
+    const argv = [
+        ...(opts.profile ? ["-p", opts.profile] : []),
+        ...args,
+        "-o",
+        "json",
+    ];
+    const timeoutMs = opts.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+    const result = await spawnCollect(argv, timeoutMs);
+    if (result.enoent) {
+        throw new AxiError("databricks CLI not found on PATH", "CLI_MISSING", [
+            INSTALL_HELP,
+        ]);
+    }
+    if (result.timedOut) {
+        throw new AxiError(`databricks ${args[0] ?? ""} timed out after ${timeoutMs}ms`, "TIMEOUT", opts.timeoutHelp ?? ["Retry, or check workspace availability"]);
+    }
+    if (result.code !== 0) {
+        throw await diagnoseFailure(result.stderr);
+    }
+    const trimmed = result.stdout.trim();
+    if (!trimmed) {
+        return null;
+    }
+    try {
+        // int64 ids (job_id/run_id) can exceed 2^53, where JSON.parse silently
+        // rounds; quote 16+-digit id values so they stay exact as strings.
+        return JSON.parse(trimmed.replace(/"(\w*_id)"\s*:\s*(\d{16,})(?=\s*[,}])/g, '"$1":"$2"'));
+    }
+    catch {
+        throw new AxiError(`databricks returned invalid JSON: ${redactSecrets(trimmed).slice(0, 120)}`, "UPSTREAM_ERROR");
+    }
+}
+/**
+ * REST passthrough over the CLI's `api` subcommand. Rule for all callers:
+ * `body` lands on child argv (visible in `ps`) — secret-bearing bodies are
+ * forbidden (the secrets domain uses upstream's stdin mechanism instead).
+ */
+export async function runDatabricksApi(method, path, body, opts = {}) {
+    return runDatabricks(["api", method, path, ...(body !== undefined ? ["--json", body] : [])], opts);
+}
+function spawnCollect(argv, timeoutMs) {
+    return new Promise((resolve) => {
+        const child = spawn("databricks", argv, {
+            stdio: ["ignore", "pipe", "pipe"],
+        });
+        let stdout = "";
+        let stderr = "";
+        let timedOut = false;
+        const timer = setTimeout(() => {
+            timedOut = true;
+            child.kill("SIGKILL");
+        }, timeoutMs);
+        child.stdout.setEncoding("utf8");
+        child.stderr.setEncoding("utf8");
+        child.stdout.on("data", (chunk) => (stdout += chunk));
+        child.stderr.on("data", (chunk) => (stderr += chunk));
+        child.on("error", (error) => {
+            clearTimeout(timer);
+            resolve({
+                code: null,
+                stdout,
+                stderr: stderr || error.message,
+                enoent: error.code === "ENOENT",
+                timedOut,
+            });
+        });
+        child.on("close", (code) => {
+            clearTimeout(timer);
+            resolve({ code, stdout, stderr, enoent: false, timedOut });
+        });
+    });
+}
+// ponytail: version guard runs only on the failure path — a pre-flight
+// `databricks -v` would tax every happy-path invocation for nothing.
+async function diagnoseFailure(stderr) {
+    if (/unknown (command|flag|shorthand)|no such (option|command)/i.test(stderr)) {
+        const version = await detectVersion();
+        if (version && version.major === 0 && version.minor < MIN_MINOR_VERSION) {
+            return new AxiError(`databricks CLI ${version.raw} is too old (need >= 0.298)`, "CLI_TOO_OLD", [`Upgrade: https://docs.databricks.com/dev-tools/cli/install`]);
+        }
+    }
+    return mapUpstreamError(stderr);
+}
+async function detectVersion() {
+    const result = await spawnCollect(["-v"], 5_000);
+    // Legacy CLIs print the version to stderr.
+    const match = /v?(\d+)\.(\d+)\.(\d+)/.exec(result.stdout + result.stderr);
+    if (!match) {
+        return null;
+    }
+    return { major: Number(match[1]), minor: Number(match[2]), raw: match[0] };
+}
+//# sourceMappingURL=databricks.js.map

package/dist/src/databricks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"databricks.js","sourceRoot":"","sources":["../../src/databricks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE9D,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAClC,MAAM,iBAAiB,GAAG,GAAG,CAAC,CAAC,kEAAkE;AACjG,MAAM,YAAY,GAChB,+DAA+D,CAAC;AAiBlE;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,IAAc,EACd,OAA6B,EAAE;IAE/B,MAAM,IAAI,GAAG;QACX,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7C,GAAG,IAAI;QACP,IAAI;QACJ,MAAM;KACP,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,kBAAkB,CAAC;IACvD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACnD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QAClB,MAAM,IAAI,QAAQ,CAAC,kCAAkC,EAAE,aAAa,EAAE;YACpE,YAAY;SACb,CAAC,CAAC;IACL,CAAC;IACD,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpB,MAAM,IAAI,QAAQ,CAChB,cAAc,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,oBAAoB,SAAS,IAAI,EAC5D,SAAS,EACT,IAAI,CAAC,WAAW,IAAI,CAAC,wCAAwC,CAAC,CAC/D,CAAC;IACJ,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QACtB,MAAM,MAAM,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC7C,CAAC;IACD,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IACrC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,CAAC;QACH,uEAAuE;QACvE,mEAAmE;QACnE,OAAO,IAAI,CAAC,KAAK,CACf,OAAO,CAAC,OAAO,CAAC,wCAAwC,EAAE,WAAW,CAAC,CAC5D,CAAC;IACf,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,QAAQ,CAChB,qCAAqC,aAAa,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,EAC3E,gBAAgB,CACjB,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,MAAc,EACd,IAAY,EACZ,IAAa,EACb,OAA6B,EAAE;IAE/B,OAAO,aAAa,CAClB,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EACtE,IAAI,CACL,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,IAAc,EAAE,SAAiB;IACrD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,KAAK,GAAG,KAAK,CAAC,YAAY,EAAE,IAAI,EAAE;YACtC,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;SAClC,CAAC,CAAC;QACH,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,QAAQ,GAAG,IAAI,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxB,CAAC,EAAE,SAAS,CAAC,CAAC;QACd,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACjC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACjC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC;QAC9D,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC;QAC9D,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAA4B,EAAE,EAAE;YACjD,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,OAAO,CAAC;gBACN,IAAI,EAAE,IAAI;gBACV,MAAM;gBACN,MAAM,EAAE,MAAM,IAAI,KAAK,CAAC,OAAO;gBAC/B,MAAM,EAAE,KAAK,CAAC,IAAI,KAAK,QAAQ;gBAC/B,QAAQ;aACT,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YACzB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,uEAAuE;AACvE,qEAAqE;AACrE,KAAK,UAAU,eAAe,CAAC,MAAc;IAC3C,IACE,4DAA4D,CAAC,IAAI,CAAC,MAAM,CAAC,EACzE,CAAC;QACD,MAAM,OAAO,GAAG,MAAM,aAAa,EAAE,CAAC;QACtC,IAAI,OAAO,IAAI,OAAO,CAAC,KAAK,KAAK,CAAC,IAAI,OAAO,CAAC,KAAK,GAAG,iBAAiB,EAAE,CAAC;YACxE,OAAO,IAAI,QAAQ,CACjB,kBAAkB,OAAO,CAAC,GAAG,6BAA6B,EAC1D,aAAa,EACb,CAAC,4DAA4D,CAAC,CAC/D,CAAC;QACJ,CAAC;IACH,CAAC;IACD,OAAO,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAClC,CAAC;AAED,KAAK,UAAU,aAAa;IAK1B,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,CAAC,CAAC;IACjD,2CAA2C;IAC3C,MAAM,KAAK,GAAG,uBAAuB,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;IAC1E,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;AAC7E,CAAC"}

package/dist/src/errors.d.ts

@@ -0,0 +1,11 @@
+import { AxiError } from "axi-sdk-js";
+/**
+ * Strip token-shaped strings from upstream text before it can reach stdout.
+ * Order matters: dapi tokens first (they are also long hex).
+ */
+export declare function redactSecrets(text: string): string;
+/**
+ * Pattern-match the Go CLI's plain-text stderr into the AXI taxonomy.
+ * Returns (never throws) so callers decide whether to throw or inspect.
+ */
+export declare function mapUpstreamError(stderr: string): AxiError;

package/dist/src/errors.js

@@ -0,0 +1,37 @@
+import { AxiError } from "axi-sdk-js";
+const AUTH_HELP = [
+    "Ask the user to run: databricks auth login --host <workspace-url>",
+];
+/**
+ * Strip token-shaped strings from upstream text before it can reach stdout.
+ * Order matters: dapi tokens first (they are also long hex).
+ */
+export function redactSecrets(text) {
+    return text
+        .replace(/dapi[0-9a-f]{16,}/gi, "[redacted]")
+        .replace(/\b[0-9a-f]{32,}\b/gi, "[redacted]")
+        .replace(/\b[A-Za-z0-9+=_-]{40,}\b/g, "[redacted]"); // no "/" so workspace paths stay readable
+}
+/**
+ * Pattern-match the Go CLI's plain-text stderr into the AXI taxonomy.
+ * Returns (never throws) so callers decide whether to throw or inspect.
+ */
+export function mapUpstreamError(stderr) {
+    const text = redactSecrets(stderr.trim()) ||
+        "databricks CLI failed with no error output";
+    const firstLine = text.split("\n", 1)[0] ?? text;
+    if (/\b401\b|unauthorized|token.{0,20}expired|cannot configure default credentials|oauth/i.test(text)) {
+        return new AxiError(firstLine, "AUTH_ERROR", AUTH_HELP);
+    }
+    if (/\b403\b|PERMISSION_DENIED/i.test(text)) {
+        return new AxiError(firstLine, "PERMISSION_DENIED");
+    }
+    if (/RESOURCE_DOES_NOT_EXIST|\b404\b|does not exist/i.test(text)) {
+        return new AxiError(firstLine, "NOT_FOUND");
+    }
+    if (/INVALID_STATE/.test(text)) {
+        return new AxiError(firstLine, "INVALID_STATE");
+    }
+    return new AxiError(firstLine, "UPSTREAM_ERROR");
+}
+//# sourceMappingURL=errors.js.map

package/dist/src/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../src/errors.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAEtC,MAAM,SAAS,GAAG;IAChB,mEAAmE;CACpE,CAAC;AAEF;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,IAAY;IACxC,OAAO,IAAI;SACR,OAAO,CAAC,qBAAqB,EAAE,YAAY,CAAC;SAC5C,OAAO,CAAC,qBAAqB,EAAE,YAAY,CAAC;SAC5C,OAAO,CAAC,2BAA2B,EAAE,YAAY,CAAC,CAAC,CAAC,0CAA0C;AACnG,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAAC,MAAc;IAC7C,MAAM,IAAI,GACR,aAAa,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC5B,4CAA4C,CAAC;IAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;IACjD,IACE,sFAAsF,CAAC,IAAI,CACzF,IAAI,CACL,EACD,CAAC;QACD,OAAO,IAAI,QAAQ,CAAC,SAAS,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IAC1D,CAAC;IACD,IAAI,4BAA4B,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5C,OAAO,IAAI,QAAQ,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;IACtD,CAAC;IACD,IAAI,iDAAiD,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACjE,OAAO,IAAI,QAAQ,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IAC9C,CAAC;IACD,IAAI,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/B,OAAO,IAAI,QAAQ,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,IAAI,QAAQ,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;AACnD,CAAC"}