@databricks/appkit 0.38.1 → 0.40.0

package/dist/appkit/package.js

@@ -1,6 +1,6 @@
 //#region package.json
 var name = "@databricks/appkit";
-var version = "0.38.1";
+var version = "0.40.0";
 
 //#endregion
 export { name, version };

package/dist/cli/commands/generate-types.js

@@ -1,15 +1,33 @@
+import { acquireSpawnLock, getSpawnLockPath, releaseSpawnLock } from "./spawn-lock.js";
 import fs from "node:fs";
 import path from "node:path";
-import { Command } from "commander";
+import { Command, Option } from "commander";
+import { spawn } from "node:child_process";
 
 //#region src/cli/commands/generate-types.ts
 /**
-* Generate types command implementation
+* Resolve the typegen pre-flight mode for the CLI. Defaults to "non-blocking" —
+* a one-shot CLI can't describe in the background, so by default it never
+* describes at all: it skips the warehouse probe AND every DESCRIBE, emits
+* best-available types (cache where the SQL hash matches, else `result: unknown`)
+* and returns immediately, never blocking on — or failing because of — a
+* warehouse, even a RUNNING one. Pass `--wait` (commander sets `wait: true`)
+* for a deliberate/CI invocation that should wait for a starting warehouse and
+* fail fast on a stopped one.
+*/
+function resolveTypegenMode(options) {
+	return options?.wait ? "blocking" : "non-blocking";
+}
+/**
+* Generate types command implementation. Runs the library generate (which, in
+* non-blocking mode, writes degraded types and returns immediately). This is the
+* SAME work the worker performs in blocking mode in the background.
 */
 async function runGenerateTypes(rootDir, outFile, warehouseId, options) {
 	try {
 		const resolvedRootDir = rootDir || process.cwd();
 		const noCache = options?.noCache || false;
+		const mode = resolveTypegenMode(options);
 		const typeGen = await import("@databricks/appkit/type-generator");
 		const resolvedWarehouseId = warehouseId || process.env.DATABRICKS_WAREHOUSE_ID;
 		if (resolvedWarehouseId) {
@@ -20,7 +38,8 @@ async function runGenerateTypes(rootDir, outFile, warehouseId, options) {
 					queryFolder,
 					outFile: resolvedOutFile,
 					warehouseId: resolvedWarehouseId,
-					noCache
+					noCache,
+					mode
 				});
 				console.log(`Generated query types: ${resolvedOutFile}`);
 			}
@@ -37,15 +56,105 @@ async function runGenerateTypes(rootDir, outFile, warehouseId, options) {
 			console.error("Please install @databricks/appkit to use this command.");
 			process.exit(1);
 		}
+		if (error instanceof Error && (error.name === "TypegenSyntaxError" || error.name === "TypegenFatalError")) {
+			console.error(error.message);
+			process.exit(1);
+		}
 		throw error;
 	}
 }
-const generateTypesCommand = new Command("generate-types").description("Generate TypeScript types from SQL queries").argument("[rootDir]", "Root directory of the project", process.cwd()).argument("[outFile]", "Output file path", path.join(process.cwd(), "shared/appkit-types/analytics.d.ts")).argument("[warehouseId]", "Databricks warehouse ID").option("--no-cache", "Disable caching for type generation").addHelpText("after", `
+/**
+* Spawn the detached blocking worker that refreshes real types in the background
+* after the foreground non-blocking generate has already written degraded types.
+*
+* Re-invokes THIS CLI (`process.execPath` + `process.argv[1]` — the bin entry
+* that launched us) with `generate-types --wait --worker-lock <lockPath>` plus
+* the same positional target options the foreground used, so the worker writes
+* to the same out file / reads the same query folder. The worker is:
+*  - `detached: true` + `.unref()` so it outlives this process (install/dev-setup
+*    can finish and exit while the worker keeps warming the warehouse).
+*  - `stdio: "ignore"` so it never holds the parent's pipes open or interleaves
+*    output into the install/dev log.
+*
+* Spawning is wrapped so any failure is non-fatal: the caller still has degraded
+* types and exits 0.
+*
+* @param lockPath - the acquired single-flight lock; passed to the worker so it
+*   releases the SAME lock when it finishes.
+* @param targets - the foreground's positional args, forwarded verbatim.
+* @returns true if the worker was spawned, false if spawning threw.
+*/
+function spawnTypegenWorker(lockPath, targets) {
+	const cliEntry = process.argv[1];
+	const positionals = [];
+	for (const value of [
+		targets.rootDir,
+		targets.outFile,
+		targets.warehouseId
+	]) {
+		if (value === void 0) break;
+		positionals.push(value);
+	}
+	const args = [
+		...process.execArgv,
+		cliEntry,
+		"generate-types",
+		"--wait",
+		"--worker-lock",
+		lockPath,
+		...positionals
+	];
+	try {
+		spawn(process.execPath, args, {
+			detached: true,
+			stdio: "ignore"
+		}).unref();
+		return true;
+	} catch (error) {
+		console.error(`Could not start background type refresh: ${error instanceof Error ? error.message : String(error)}`);
+		return false;
+	}
+}
+/**
+* The command action. Orchestrates the non-blocking foreground contract:
+*  1. Run the library generate (writes degraded types immediately in non-blocking
+*     mode; does the full blocking lifecycle when this is the worker).
+*  2. If this is a non-blocking, non-worker invocation, try to spawn the detached
+*     blocking worker behind the single-flight lock. If the lock is already held
+*     by a live worker, skip (single-flight) with a one-line note. Either way the
+*     foreground returns normally (exit 0).
+*  3. If this IS the worker (`--worker-lock` present), it ran blocking above and
+*     releases the lock here (and via a process-exit guard, so a hard failure /
+*     process.exit still frees it).
+*/
+async function generateTypesAction(rootDir, outFile, warehouseId, options) {
+	const isWorker = typeof options.workerLock === "string";
+	if (isWorker && options.workerLock) {
+		const lockPath = options.workerLock;
+		process.once("exit", () => releaseSpawnLock(lockPath));
+	}
+	try {
+		await runGenerateTypes(rootDir, outFile, warehouseId, options);
+	} finally {
+		if (isWorker && options.workerLock) releaseSpawnLock(options.workerLock);
+	}
+	if (!isWorker && resolveTypegenMode(options) === "non-blocking") {
+		const lockPath = getSpawnLockPath(rootDir || process.cwd());
+		if (acquireSpawnLock(lockPath)) spawnTypegenWorker(lockPath, {
+			rootDir,
+			outFile,
+			warehouseId
+		});
+		else console.log("Type refresh already in progress, skipping.");
+	}
+}
+const generateTypesCommand = new Command("generate-types").description("Generate TypeScript types from SQL queries").argument("[rootDir]", "Root directory of the project", process.cwd()).argument("[outFile]", "Output file path", path.join(process.cwd(), "shared/appkit-types/analytics.d.ts")).argument("[warehouseId]", "Databricks warehouse ID").option("--no-cache", "Disable caching for type generation").option("--wait", "Wait for warehouse readiness instead of degrading (use for CI)").addOption(new Option("--worker-lock <path>", "Internal: detached worker lock path").hideHelp()).addHelpText("after", `
 Examples:
   $ appkit generate-types
   $ appkit generate-types . shared/appkit-types/analytics.d.ts
   $ appkit generate-types . shared/appkit-types/analytics.d.ts my-warehouse-id
-  $ appkit generate-types --no-cache`).action(runGenerateTypes);
+  $ appkit generate-types --no-cache
+  $ appkit generate-types --wait   # CI: wait for the warehouse and fail on a cold one`).action(generateTypesAction);
 
 //#endregion
 export { generateTypesCommand };

package/dist/cli/commands/generate-types.js.map

@@ -1 +1 @@
-{"version":3,"file":"generate-types.js","names":[],"sources":["../../../src/cli/commands/generate-types.ts"],"sourcesContent":["import fs from \"node:fs\";\nimport path from \"node:path\";\nimport { Command } from \"commander\";\n\n/**\n * Generate types command implementation\n */\nasync function runGenerateTypes(\n  rootDir?: string,\n  outFile?: string,\n  warehouseId?: string,\n  options?: { noCache?: boolean },\n) {\n  try {\n    const resolvedRootDir = rootDir || process.cwd();\n    const noCache = options?.noCache || false;\n\n    const typeGen = await import(\"@databricks/appkit/type-generator\");\n\n    // Generate analytics query types (requires warehouse ID)\n    const resolvedWarehouseId =\n      warehouseId || process.env.DATABRICKS_WAREHOUSE_ID;\n\n    if (resolvedWarehouseId) {\n      const resolvedOutFile =\n        outFile ||\n        path.join(process.cwd(), \"shared/appkit-types/analytics.d.ts\");\n\n      const queryFolder = path.join(resolvedRootDir, \"config/queries\");\n      if (fs.existsSync(queryFolder)) {\n        await typeGen.generateFromEntryPoint({\n          queryFolder,\n          outFile: resolvedOutFile,\n          warehouseId: resolvedWarehouseId,\n          noCache,\n        });\n        console.log(`Generated query types: ${resolvedOutFile}`);\n      }\n    } else {\n      console.error(\n        \"Skipping query type generation: no warehouse ID. Set DATABRICKS_WAREHOUSE_ID or pass as argument.\",\n      );\n    }\n\n    // Generate serving endpoint types (no warehouse required)\n    const servingOutFile = path.join(\n      process.cwd(),\n      \"shared/appkit-types/serving.d.ts\",\n    );\n    await typeGen.generateServingTypes({\n      outFile: servingOutFile,\n      noCache,\n    });\n    console.log(`Generated serving types: ${servingOutFile}`);\n  } catch (error) {\n    if (\n      error instanceof Error &&\n      error.message.includes(\"Cannot find module\")\n    ) {\n      console.error(\n        \"Error: The 'generate-types' command is only available in @databricks/appkit.\",\n      );\n      console.error(\"Please install @databricks/appkit to use this command.\");\n      process.exit(1);\n    }\n    throw error;\n  }\n}\n\nexport const generateTypesCommand = new Command(\"generate-types\")\n  .description(\"Generate TypeScript types from SQL queries\")\n  .argument(\"[rootDir]\", \"Root directory of the project\", process.cwd())\n  .argument(\n    \"[outFile]\",\n    \"Output file path\",\n    path.join(process.cwd(), \"shared/appkit-types/analytics.d.ts\"),\n  )\n  .argument(\"[warehouseId]\", \"Databricks warehouse ID\")\n  .option(\"--no-cache\", \"Disable caching for type generation\")\n  .addHelpText(\n    \"after\",\n    `\nExamples:\n  $ appkit generate-types\n  $ appkit generate-types . shared/appkit-types/analytics.d.ts\n  $ appkit generate-types . shared/appkit-types/analytics.d.ts my-warehouse-id\n  $ appkit generate-types --no-cache`,\n  )\n  .action(runGenerateTypes);\n"],"mappings":";;;;;;;;AAOA,eAAe,iBACb,SACA,SACA,aACA,SACA;AACA,KAAI;EACF,MAAM,kBAAkB,WAAW,QAAQ,KAAK;EAChD,MAAM,UAAU,SAAS,WAAW;EAEpC,MAAM,UAAU,MAAM,OAAO;EAG7B,MAAM,sBACJ,eAAe,QAAQ,IAAI;AAE7B,MAAI,qBAAqB;GACvB,MAAM,kBACJ,WACA,KAAK,KAAK,QAAQ,KAAK,EAAE,qCAAqC;GAEhE,MAAM,cAAc,KAAK,KAAK,iBAAiB,iBAAiB;AAChE,OAAI,GAAG,WAAW,YAAY,EAAE;AAC9B,UAAM,QAAQ,uBAAuB;KACnC;KACA,SAAS;KACT,aAAa;KACb;KACD,CAAC;AACF,YAAQ,IAAI,0BAA0B,kBAAkB;;QAG1D,SAAQ,MACN,oGACD;EAIH,MAAM,iBAAiB,KAAK,KAC1B,QAAQ,KAAK,EACb,mCACD;AACD,QAAM,QAAQ,qBAAqB;GACjC,SAAS;GACT;GACD,CAAC;AACF,UAAQ,IAAI,4BAA4B,iBAAiB;UAClD,OAAO;AACd,MACE,iBAAiB,SACjB,MAAM,QAAQ,SAAS,qBAAqB,EAC5C;AACA,WAAQ,MACN,+EACD;AACD,WAAQ,MAAM,yDAAyD;AACvE,WAAQ,KAAK,EAAE;;AAEjB,QAAM;;;AAIV,MAAa,uBAAuB,IAAI,QAAQ,iBAAiB,CAC9D,YAAY,6CAA6C,CACzD,SAAS,aAAa,iCAAiC,QAAQ,KAAK,CAAC,CACrE,SACC,aACA,oBACA,KAAK,KAAK,QAAQ,KAAK,EAAE,qCAAqC,CAC/D,CACA,SAAS,iBAAiB,0BAA0B,CACpD,OAAO,cAAc,sCAAsC,CAC3D,YACC,SACA;;;;;sCAMD,CACA,OAAO,iBAAiB"}
+{"version":3,"file":"generate-types.js","names":[],"sources":["../../../src/cli/commands/generate-types.ts"],"sourcesContent":["import { spawn } from \"node:child_process\";\nimport fs from \"node:fs\";\nimport path from \"node:path\";\nimport { Command, Option } from \"commander\";\nimport {\n  acquireSpawnLock,\n  getSpawnLockPath,\n  releaseSpawnLock,\n} from \"./spawn-lock.js\";\n\n/**\n * Resolve the typegen pre-flight mode for the CLI. Defaults to \"non-blocking\" —\n * a one-shot CLI can't describe in the background, so by default it never\n * describes at all: it skips the warehouse probe AND every DESCRIBE, emits\n * best-available types (cache where the SQL hash matches, else `result: unknown`)\n * and returns immediately, never blocking on — or failing because of — a\n * warehouse, even a RUNNING one. Pass `--wait` (commander sets `wait: true`)\n * for a deliberate/CI invocation that should wait for a starting warehouse and\n * fail fast on a stopped one.\n */\nexport function resolveTypegenMode(options?: {\n  wait?: boolean;\n}): \"non-blocking\" | \"blocking\" {\n  return options?.wait ? \"blocking\" : \"non-blocking\";\n}\n\n/** Options parsed by commander for the generate-types command. */\ninterface GenerateTypesOptions {\n  noCache?: boolean;\n  wait?: boolean;\n  /**\n   * Internal: present only on the detached worker invocation. Carries the path\n   * of the single-flight lock this worker must release when it finishes. Its\n   * presence is what marks an invocation as \"the worker\" — workers always run\n   * with `--wait`, so they never spawn another worker (only non-blocking runs\n   * spawn), which terminates the recursion.\n   */\n  workerLock?: string;\n}\n\n/**\n * Generate types command implementation. Runs the library generate (which, in\n * non-blocking mode, writes degraded types and returns immediately). This is the\n * SAME work the worker performs in blocking mode in the background.\n */\nasync function runGenerateTypes(\n  rootDir?: string,\n  outFile?: string,\n  warehouseId?: string,\n  options?: GenerateTypesOptions,\n) {\n  try {\n    const resolvedRootDir = rootDir || process.cwd();\n    const noCache = options?.noCache || false;\n    const mode = resolveTypegenMode(options);\n\n    const typeGen = await import(\"@databricks/appkit/type-generator\");\n\n    // Generate analytics query types (requires warehouse ID)\n    const resolvedWarehouseId =\n      warehouseId || process.env.DATABRICKS_WAREHOUSE_ID;\n\n    if (resolvedWarehouseId) {\n      const resolvedOutFile =\n        outFile ||\n        path.join(process.cwd(), \"shared/appkit-types/analytics.d.ts\");\n\n      const queryFolder = path.join(resolvedRootDir, \"config/queries\");\n      if (fs.existsSync(queryFolder)) {\n        await typeGen.generateFromEntryPoint({\n          queryFolder,\n          outFile: resolvedOutFile,\n          warehouseId: resolvedWarehouseId,\n          noCache,\n          mode,\n        });\n        console.log(`Generated query types: ${resolvedOutFile}`);\n      }\n    } else {\n      console.error(\n        \"Skipping query type generation: no warehouse ID. Set DATABRICKS_WAREHOUSE_ID or pass as argument.\",\n      );\n    }\n\n    // Generate serving endpoint types (no warehouse required)\n    const servingOutFile = path.join(\n      process.cwd(),\n      \"shared/appkit-types/serving.d.ts\",\n    );\n    await typeGen.generateServingTypes({\n      outFile: servingOutFile,\n      noCache,\n    });\n    console.log(`Generated serving types: ${servingOutFile}`);\n  } catch (error) {\n    if (\n      error instanceof Error &&\n      error.message.includes(\"Cannot find module\")\n    ) {\n      console.error(\n        \"Error: The 'generate-types' command is only available in @databricks/appkit.\",\n      );\n      console.error(\"Please install @databricks/appkit to use this command.\");\n      process.exit(1);\n    }\n    // TypegenSyntaxError / TypegenFatalError carry a complete, actionable\n    // message (which queries failed and how to debug them). The stack trace\n    // points into appkit internals and is noise for app developers, so print\n    // only the message and exit non-zero instead of letting it bubble up.\n    if (\n      error instanceof Error &&\n      (error.name === \"TypegenSyntaxError\" ||\n        error.name === \"TypegenFatalError\")\n    ) {\n      console.error(error.message);\n      process.exit(1);\n    }\n    throw error;\n  }\n}\n\n/**\n * Spawn the detached blocking worker that refreshes real types in the background\n * after the foreground non-blocking generate has already written degraded types.\n *\n * Re-invokes THIS CLI (`process.execPath` + `process.argv[1]` — the bin entry\n * that launched us) with `generate-types --wait --worker-lock <lockPath>` plus\n * the same positional target options the foreground used, so the worker writes\n * to the same out file / reads the same query folder. The worker is:\n *  - `detached: true` + `.unref()` so it outlives this process (install/dev-setup\n *    can finish and exit while the worker keeps warming the warehouse).\n *  - `stdio: \"ignore\"` so it never holds the parent's pipes open or interleaves\n *    output into the install/dev log.\n *\n * Spawning is wrapped so any failure is non-fatal: the caller still has degraded\n * types and exits 0.\n *\n * @param lockPath - the acquired single-flight lock; passed to the worker so it\n *   releases the SAME lock when it finishes.\n * @param targets - the foreground's positional args, forwarded verbatim.\n * @returns true if the worker was spawned, false if spawning threw.\n */\nexport function spawnTypegenWorker(\n  lockPath: string,\n  targets: { rootDir?: string; outFile?: string; warehouseId?: string },\n): boolean {\n  // The script the runtime launched us with (the `appkit` bin shim). Re-running\n  // it under the same node binary reproduces this exact CLI in the worker.\n  const cliEntry = process.argv[1];\n\n  // Forward the positionals in declaration order (rootDir, outFile,\n  // warehouseId). Stop at the first undefined so we never pass a literal\n  // \"undefined\" — commander would treat it as a positional value. (rootDir is\n  // effectively always set by commander's default, but guard anyway.)\n  const positionals: string[] = [];\n  for (const value of [targets.rootDir, targets.outFile, targets.warehouseId]) {\n    if (value === undefined) break;\n    positionals.push(value);\n  }\n\n  const args = [\n    // Forward the parent's node/loader flags so the worker runs under the same\n    // runtime. Critically this carries tsx's `--require`/`--import` when the CLI\n    // is run from source (`tsx index.ts …`); without them the worker would be\n    // `node index.ts …`, which can't parse TypeScript and dies silently — the\n    // degraded types would then never refresh. Empty for the built bin (plain\n    // `node bin/appkit.js`), so production behaviour is unchanged.\n    ...process.execArgv,\n    cliEntry,\n    \"generate-types\",\n    \"--wait\",\n    \"--worker-lock\",\n    lockPath,\n    ...positionals,\n  ];\n\n  try {\n    const child = spawn(process.execPath, args, {\n      detached: true,\n      stdio: \"ignore\",\n    });\n    child.unref();\n    return true;\n  } catch (error) {\n    // Non-fatal: the foreground already wrote degraded types. Log and move on.\n    console.error(\n      `Could not start background type refresh: ${\n        error instanceof Error ? error.message : String(error)\n      }`,\n    );\n    return false;\n  }\n}\n\n/**\n * The command action. Orchestrates the non-blocking foreground contract:\n *  1. Run the library generate (writes degraded types immediately in non-blocking\n *     mode; does the full blocking lifecycle when this is the worker).\n *  2. If this is a non-blocking, non-worker invocation, try to spawn the detached\n *     blocking worker behind the single-flight lock. If the lock is already held\n *     by a live worker, skip (single-flight) with a one-line note. Either way the\n *     foreground returns normally (exit 0).\n *  3. If this IS the worker (`--worker-lock` present), it ran blocking above and\n *     releases the lock here (and via a process-exit guard, so a hard failure /\n *     process.exit still frees it).\n */\nasync function generateTypesAction(\n  rootDir: string | undefined,\n  outFile: string | undefined,\n  warehouseId: string | undefined,\n  options: GenerateTypesOptions,\n) {\n  const isWorker = typeof options.workerLock === \"string\";\n\n  // A worker must always free its lock, even if the blocking generate throws or\n  // calls process.exit (TypegenFatalError → exit 1). The exit handler covers the\n  // process.exit / uncaught paths; the finally covers the normal return.\n  if (isWorker && options.workerLock) {\n    const lockPath = options.workerLock;\n    process.once(\"exit\", () => releaseSpawnLock(lockPath));\n  }\n\n  try {\n    await runGenerateTypes(rootDir, outFile, warehouseId, options);\n  } finally {\n    if (isWorker && options.workerLock) {\n      releaseSpawnLock(options.workerLock);\n    }\n  }\n\n  // Only a non-blocking, non-worker invocation spawns. A worker is always\n  // --wait (so resolveTypegenMode → \"blocking\"), which both prevents recursion\n  // and means we never get here for a worker.\n  if (!isWorker && resolveTypegenMode(options) === \"non-blocking\") {\n    const resolvedRootDir = rootDir || process.cwd();\n    const lockPath = getSpawnLockPath(resolvedRootDir);\n\n    if (acquireSpawnLock(lockPath)) {\n      spawnTypegenWorker(lockPath, { rootDir, outFile, warehouseId });\n    } else {\n      console.log(\"Type refresh already in progress, skipping.\");\n    }\n  }\n}\n\nexport const generateTypesCommand = new Command(\"generate-types\")\n  .description(\"Generate TypeScript types from SQL queries\")\n  .argument(\"[rootDir]\", \"Root directory of the project\", process.cwd())\n  .argument(\n    \"[outFile]\",\n    \"Output file path\",\n    path.join(process.cwd(), \"shared/appkit-types/analytics.d.ts\"),\n  )\n  .argument(\"[warehouseId]\", \"Databricks warehouse ID\")\n  .option(\"--no-cache\", \"Disable caching for type generation\")\n  .option(\n    \"--wait\",\n    \"Wait for warehouse readiness instead of degrading (use for CI)\",\n  )\n  // Internal: marks the detached background worker and carries the lock it must\n  // release. Hidden from --help; users should never pass it.\n  .addOption(\n    new Option(\n      \"--worker-lock <path>\",\n      \"Internal: detached worker lock path\",\n    ).hideHelp(),\n  )\n  .addHelpText(\n    \"after\",\n    `\nExamples:\n  $ appkit generate-types\n  $ appkit generate-types . shared/appkit-types/analytics.d.ts\n  $ appkit generate-types . shared/appkit-types/analytics.d.ts my-warehouse-id\n  $ appkit generate-types --no-cache\n  $ appkit generate-types --wait   # CI: wait for the warehouse and fail on a cold one`,\n  )\n  .action(generateTypesAction);\n"],"mappings":";;;;;;;;;;;;;;;;;AAoBA,SAAgB,mBAAmB,SAEH;AAC9B,QAAO,SAAS,OAAO,aAAa;;;;;;;AAsBtC,eAAe,iBACb,SACA,SACA,aACA,SACA;AACA,KAAI;EACF,MAAM,kBAAkB,WAAW,QAAQ,KAAK;EAChD,MAAM,UAAU,SAAS,WAAW;EACpC,MAAM,OAAO,mBAAmB,QAAQ;EAExC,MAAM,UAAU,MAAM,OAAO;EAG7B,MAAM,sBACJ,eAAe,QAAQ,IAAI;AAE7B,MAAI,qBAAqB;GACvB,MAAM,kBACJ,WACA,KAAK,KAAK,QAAQ,KAAK,EAAE,qCAAqC;GAEhE,MAAM,cAAc,KAAK,KAAK,iBAAiB,iBAAiB;AAChE,OAAI,GAAG,WAAW,YAAY,EAAE;AAC9B,UAAM,QAAQ,uBAAuB;KACnC;KACA,SAAS;KACT,aAAa;KACb;KACA;KACD,CAAC;AACF,YAAQ,IAAI,0BAA0B,kBAAkB;;QAG1D,SAAQ,MACN,oGACD;EAIH,MAAM,iBAAiB,KAAK,KAC1B,QAAQ,KAAK,EACb,mCACD;AACD,QAAM,QAAQ,qBAAqB;GACjC,SAAS;GACT;GACD,CAAC;AACF,UAAQ,IAAI,4BAA4B,iBAAiB;UAClD,OAAO;AACd,MACE,iBAAiB,SACjB,MAAM,QAAQ,SAAS,qBAAqB,EAC5C;AACA,WAAQ,MACN,+EACD;AACD,WAAQ,MAAM,yDAAyD;AACvE,WAAQ,KAAK,EAAE;;AAMjB,MACE,iBAAiB,UAChB,MAAM,SAAS,wBACd,MAAM,SAAS,sBACjB;AACA,WAAQ,MAAM,MAAM,QAAQ;AAC5B,WAAQ,KAAK,EAAE;;AAEjB,QAAM;;;;;;;;;;;;;;;;;;;;;;;;AAyBV,SAAgB,mBACd,UACA,SACS;CAGT,MAAM,WAAW,QAAQ,KAAK;CAM9B,MAAM,cAAwB,EAAE;AAChC,MAAK,MAAM,SAAS;EAAC,QAAQ;EAAS,QAAQ;EAAS,QAAQ;EAAY,EAAE;AAC3E,MAAI,UAAU,OAAW;AACzB,cAAY,KAAK,MAAM;;CAGzB,MAAM,OAAO;EAOX,GAAG,QAAQ;EACX;EACA;EACA;EACA;EACA;EACA,GAAG;EACJ;AAED,KAAI;AAKF,EAJc,MAAM,QAAQ,UAAU,MAAM;GAC1C,UAAU;GACV,OAAO;GACR,CAAC,CACI,OAAO;AACb,SAAO;UACA,OAAO;AAEd,UAAQ,MACN,4CACE,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,GAEzD;AACD,SAAO;;;;;;;;;;;;;;;AAgBX,eAAe,oBACb,SACA,SACA,aACA,SACA;CACA,MAAM,WAAW,OAAO,QAAQ,eAAe;AAK/C,KAAI,YAAY,QAAQ,YAAY;EAClC,MAAM,WAAW,QAAQ;AACzB,UAAQ,KAAK,cAAc,iBAAiB,SAAS,CAAC;;AAGxD,KAAI;AACF,QAAM,iBAAiB,SAAS,SAAS,aAAa,QAAQ;WACtD;AACR,MAAI,YAAY,QAAQ,WACtB,kBAAiB,QAAQ,WAAW;;AAOxC,KAAI,CAAC,YAAY,mBAAmB,QAAQ,KAAK,gBAAgB;EAE/D,MAAM,WAAW,iBADO,WAAW,QAAQ,KAAK,CACE;AAElD,MAAI,iBAAiB,SAAS,CAC5B,oBAAmB,UAAU;GAAE;GAAS;GAAS;GAAa,CAAC;MAE/D,SAAQ,IAAI,8CAA8C;;;AAKhE,MAAa,uBAAuB,IAAI,QAAQ,iBAAiB,CAC9D,YAAY,6CAA6C,CACzD,SAAS,aAAa,iCAAiC,QAAQ,KAAK,CAAC,CACrE,SACC,aACA,oBACA,KAAK,KAAK,QAAQ,KAAK,EAAE,qCAAqC,CAC/D,CACA,SAAS,iBAAiB,0BAA0B,CACpD,OAAO,cAAc,sCAAsC,CAC3D,OACC,UACA,iEACD,CAGA,UACC,IAAI,OACF,wBACA,sCACD,CAAC,UAAU,CACb,CACA,YACC,SACA;;;;;;wFAOD,CACA,OAAO,oBAAoB"}

package/dist/cli/commands/spawn-lock.js

@@ -0,0 +1,116 @@
+import fs from "node:fs";
+import path from "node:path";
+
+//#region src/cli/commands/spawn-lock.ts
+/**
+* How long a spawn lock is considered fresh. A held lock newer than this means a
+* background worker is genuinely in flight, so the foreground skips spawning;
+* older than this the lock is presumed orphaned (the worker crashed/was killed
+* before it could release) and is stolen.
+*
+* Must comfortably exceed the worker's worst-case runtime: the blocking preflight
+* wait cap (PREFLIGHT_WAIT_MAX_MS = 5 min in the type-generator) plus a DESCRIBE
+* budget. Six minutes leaves ~1 min of headroom over a worker that waits the full
+* preflight window and then describes.
+*/
+const SPAWN_LOCK_STALE_MS = 360 * 1e3;
+/**
+* Resolve the on-disk path of the single-flight spawn lock for a project.
+*
+* Lives alongside the type-generator cache (`node_modules/.databricks/appkit/`)
+* so it shares the same already-creatable, gitignored, per-project location and
+* doesn't introduce a new directory. The lock is keyed only by project root, so
+* concurrent `generate-types` invocations for the same project (postinstall +
+* predev, say) contend for one lock and only one wins the spawn.
+*
+* @param rootDir - project root (the resolved first CLI argument / cwd).
+* @returns absolute path to the lock file.
+*/
+function getSpawnLockPath(rootDir) {
+	return path.join(rootDir, "node_modules", ".databricks", "appkit", ".appkit-typegen-worker.lock");
+}
+/**
+* Try to acquire the single-flight spawn lock.
+*
+* Atomic create via `fs.writeFileSync(lockPath, ..., { flag: "wx" })` — `wx`
+* fails (EEXIST) if the file already exists, so the create itself is the
+* mutual-exclusion primitive (no check-then-create race between two foreground
+* processes). The lock body records pid + timestamp purely for debugging.
+*
+* On EEXIST we stat the existing lock:
+*  - fresh (mtime within {@link staleMs}) → a worker is in flight, return false.
+*  - stale (mtime older than staleMs) → presumed orphaned; unlink and recreate.
+*    The recreate also uses `wx`, so if a competing process steals it first we
+*    lose the race cleanly and return false.
+*
+* Any unexpected error (permission, ENOENT on a missing parent dir we couldn't
+* create, …) is swallowed and reported as "not acquired": failing to take the
+* lock must never break the foreground — at worst we skip the background refresh.
+*
+* @param lockPath - path returned by {@link getSpawnLockPath}.
+* @param staleMs - age beyond which a held lock is stolen. Defaults to
+*   {@link SPAWN_LOCK_STALE_MS}.
+* @returns true if this caller now owns the lock (and must release it), false if
+*   another live worker holds it or the lock couldn't be taken.
+*/
+function acquireSpawnLock(lockPath, staleMs = SPAWN_LOCK_STALE_MS) {
+	const body = `${process.pid} ${Date.now()}\n`;
+	try {
+		fs.mkdirSync(path.dirname(lockPath), { recursive: true });
+	} catch {}
+	try {
+		fs.writeFileSync(lockPath, body, { flag: "wx" });
+		return true;
+	} catch (error) {
+		if (!isErrnoException(error) || error.code !== "EEXIST") return false;
+	}
+	let mtimeMs;
+	try {
+		mtimeMs = fs.statSync(lockPath).mtimeMs;
+	} catch {
+		return tryCreate(lockPath, body);
+	}
+	if (Date.now() - mtimeMs < staleMs) return false;
+	try {
+		fs.unlinkSync(lockPath);
+	} catch (error) {
+		if (isErrnoException(error) && error.code !== "ENOENT") return false;
+	}
+	return tryCreate(lockPath, body);
+}
+/**
+* Release the spawn lock. Unlink, ignoring ENOENT (already gone — e.g. it was
+* stolen as stale by another process, or never existed). Any other error is
+* swallowed: a failed release at worst leaves a stale lock that the next caller
+* will steal after {@link SPAWN_LOCK_STALE_MS}.
+*
+* @param lockPath - path returned by {@link getSpawnLockPath}.
+*/
+function releaseSpawnLock(lockPath) {
+	try {
+		fs.unlinkSync(lockPath);
+	} catch {}
+}
+/**
+* Attempt an atomic `wx` create, returning whether it succeeded. EEXIST (a
+* racing creator beat us) and any other error map to false.
+*/
+function tryCreate(lockPath, body) {
+	try {
+		fs.writeFileSync(lockPath, body, { flag: "wx" });
+		return true;
+	} catch {
+		return false;
+	}
+}
+/**
+* Narrow an unknown caught value to a Node errno exception so `.code` is safe to
+* read.
+*/
+function isErrnoException(error) {
+	return error instanceof Error && "code" in error;
+}
+
+//#endregion
+export { acquireSpawnLock, getSpawnLockPath, releaseSpawnLock };
+//# sourceMappingURL=spawn-lock.js.map

package/dist/cli/commands/spawn-lock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"spawn-lock.js","names":[],"sources":["../../../src/cli/commands/spawn-lock.ts"],"sourcesContent":["import fs from \"node:fs\";\nimport path from \"node:path\";\n\n/**\n * How long a spawn lock is considered fresh. A held lock newer than this means a\n * background worker is genuinely in flight, so the foreground skips spawning;\n * older than this the lock is presumed orphaned (the worker crashed/was killed\n * before it could release) and is stolen.\n *\n * Must comfortably exceed the worker's worst-case runtime: the blocking preflight\n * wait cap (PREFLIGHT_WAIT_MAX_MS = 5 min in the type-generator) plus a DESCRIBE\n * budget. Six minutes leaves ~1 min of headroom over a worker that waits the full\n * preflight window and then describes.\n */\nexport const SPAWN_LOCK_STALE_MS = 6 * 60 * 1000;\n\n/**\n * Resolve the on-disk path of the single-flight spawn lock for a project.\n *\n * Lives alongside the type-generator cache (`node_modules/.databricks/appkit/`)\n * so it shares the same already-creatable, gitignored, per-project location and\n * doesn't introduce a new directory. The lock is keyed only by project root, so\n * concurrent `generate-types` invocations for the same project (postinstall +\n * predev, say) contend for one lock and only one wins the spawn.\n *\n * @param rootDir - project root (the resolved first CLI argument / cwd).\n * @returns absolute path to the lock file.\n */\nexport function getSpawnLockPath(rootDir: string): string {\n  return path.join(\n    rootDir,\n    \"node_modules\",\n    \".databricks\",\n    \"appkit\",\n    \".appkit-typegen-worker.lock\",\n  );\n}\n\n/**\n * Try to acquire the single-flight spawn lock.\n *\n * Atomic create via `fs.writeFileSync(lockPath, ..., { flag: \"wx\" })` — `wx`\n * fails (EEXIST) if the file already exists, so the create itself is the\n * mutual-exclusion primitive (no check-then-create race between two foreground\n * processes). The lock body records pid + timestamp purely for debugging.\n *\n * On EEXIST we stat the existing lock:\n *  - fresh (mtime within {@link staleMs}) → a worker is in flight, return false.\n *  - stale (mtime older than staleMs) → presumed orphaned; unlink and recreate.\n *    The recreate also uses `wx`, so if a competing process steals it first we\n *    lose the race cleanly and return false.\n *\n * Any unexpected error (permission, ENOENT on a missing parent dir we couldn't\n * create, …) is swallowed and reported as \"not acquired\": failing to take the\n * lock must never break the foreground — at worst we skip the background refresh.\n *\n * @param lockPath - path returned by {@link getSpawnLockPath}.\n * @param staleMs - age beyond which a held lock is stolen. Defaults to\n *   {@link SPAWN_LOCK_STALE_MS}.\n * @returns true if this caller now owns the lock (and must release it), false if\n *   another live worker holds it or the lock couldn't be taken.\n */\nexport function acquireSpawnLock(\n  lockPath: string,\n  staleMs: number = SPAWN_LOCK_STALE_MS,\n): boolean {\n  const body = `${process.pid} ${Date.now()}\\n`;\n\n  try {\n    fs.mkdirSync(path.dirname(lockPath), { recursive: true });\n  } catch {\n    // Parent dir creation is best-effort; the create below will surface any real\n    // problem and we'll treat it as \"not acquired\".\n  }\n\n  try {\n    fs.writeFileSync(lockPath, body, { flag: \"wx\" });\n    return true;\n  } catch (error) {\n    if (!isErrnoException(error) || error.code !== \"EEXIST\") {\n      // Unexpected failure — don't let lock IO break the foreground.\n      return false;\n    }\n  }\n\n  // Lock exists — decide fresh vs stale.\n  let mtimeMs: number;\n  try {\n    mtimeMs = fs.statSync(lockPath).mtimeMs;\n  } catch {\n    // It vanished between the failed create and the stat (released by the\n    // worker). Try once more to take it.\n    return tryCreate(lockPath, body);\n  }\n\n  if (Date.now() - mtimeMs < staleMs) {\n    // A worker is genuinely in flight.\n    return false;\n  }\n\n  // Stale: steal it. Unlink (ignore ENOENT — someone else may have cleaned up)\n  // then re-create with `wx` so we still lose cleanly to a racing stealer.\n  try {\n    fs.unlinkSync(lockPath);\n  } catch (error) {\n    if (isErrnoException(error) && error.code !== \"ENOENT\") {\n      return false;\n    }\n  }\n  return tryCreate(lockPath, body);\n}\n\n/**\n * Release the spawn lock. Unlink, ignoring ENOENT (already gone — e.g. it was\n * stolen as stale by another process, or never existed). Any other error is\n * swallowed: a failed release at worst leaves a stale lock that the next caller\n * will steal after {@link SPAWN_LOCK_STALE_MS}.\n *\n * @param lockPath - path returned by {@link getSpawnLockPath}.\n */\nexport function releaseSpawnLock(lockPath: string): void {\n  try {\n    fs.unlinkSync(lockPath);\n  } catch {\n    // ENOENT or any other error — releasing is best-effort.\n  }\n}\n\n/**\n * Attempt an atomic `wx` create, returning whether it succeeded. EEXIST (a\n * racing creator beat us) and any other error map to false.\n */\nfunction tryCreate(lockPath: string, body: string): boolean {\n  try {\n    fs.writeFileSync(lockPath, body, { flag: \"wx\" });\n    return true;\n  } catch {\n    return false;\n  }\n}\n\n/**\n * Narrow an unknown caught value to a Node errno exception so `.code` is safe to\n * read.\n */\nfunction isErrnoException(error: unknown): error is NodeJS.ErrnoException {\n  return error instanceof Error && \"code\" in error;\n}\n"],"mappings":";;;;;;;;;;;;;;;AAcA,MAAa,sBAAsB,MAAS;;;;;;;;;;;;;AAc5C,SAAgB,iBAAiB,SAAyB;AACxD,QAAO,KAAK,KACV,SACA,gBACA,eACA,UACA,8BACD;;;;;;;;;;;;;;;;;;;;;;;;;;AA2BH,SAAgB,iBACd,UACA,UAAkB,qBACT;CACT,MAAM,OAAO,GAAG,QAAQ,IAAI,GAAG,KAAK,KAAK,CAAC;AAE1C,KAAI;AACF,KAAG,UAAU,KAAK,QAAQ,SAAS,EAAE,EAAE,WAAW,MAAM,CAAC;SACnD;AAKR,KAAI;AACF,KAAG,cAAc,UAAU,MAAM,EAAE,MAAM,MAAM,CAAC;AAChD,SAAO;UACA,OAAO;AACd,MAAI,CAAC,iBAAiB,MAAM,IAAI,MAAM,SAAS,SAE7C,QAAO;;CAKX,IAAI;AACJ,KAAI;AACF,YAAU,GAAG,SAAS,SAAS,CAAC;SAC1B;AAGN,SAAO,UAAU,UAAU,KAAK;;AAGlC,KAAI,KAAK,KAAK,GAAG,UAAU,QAEzB,QAAO;AAKT,KAAI;AACF,KAAG,WAAW,SAAS;UAChB,OAAO;AACd,MAAI,iBAAiB,MAAM,IAAI,MAAM,SAAS,SAC5C,QAAO;;AAGX,QAAO,UAAU,UAAU,KAAK;;;;;;;;;;AAWlC,SAAgB,iBAAiB,UAAwB;AACvD,KAAI;AACF,KAAG,WAAW,SAAS;SACjB;;;;;;AASV,SAAS,UAAU,UAAkB,MAAuB;AAC1D,KAAI;AACF,KAAG,cAAc,UAAU,MAAM,EAAE,MAAM,MAAM,CAAC;AAChD,SAAO;SACD;AACN,SAAO;;;;;;;AAQX,SAAS,iBAAiB,OAAgD;AACxE,QAAO,iBAAiB,SAAS,UAAU"}

package/dist/cli/index.js

@@ -22,7 +22,7 @@ cmd.addCommand(lintCommand);
 cmd.addCommand(docsCommand);
 cmd.addCommand(pluginCommand);
 cmd.addCommand(codemodCommand);
-cmd.parse();
+await cmd.parseAsync();
 
 //#endregion
 export {  };

package/dist/cli/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../../src/cli/index.ts"],"sourcesContent":["#!/usr/bin/env node\nimport \"dotenv/config\";\nimport { readFileSync } from \"node:fs\";\nimport { dirname, join } from \"node:path\";\nimport { fileURLToPath } from \"node:url\";\nimport { Command } from \"commander\";\nimport { codemodCommand } from \"./commands/codemod/index.js\";\nimport { docsCommand } from \"./commands/docs.js\";\nimport { generateTypesCommand } from \"./commands/generate-types.js\";\nimport { lintCommand } from \"./commands/lint.js\";\nimport { pluginCommand } from \"./commands/plugin/index.js\";\nimport { setupCommand } from \"./commands/setup.js\";\n\nconst __dirname = dirname(fileURLToPath(import.meta.url));\nconst pkgPath = join(__dirname, \"../../package.json\");\nconst pkg = JSON.parse(readFileSync(pkgPath, \"utf-8\"));\n\nconst cmd = new Command();\n\ncmd\n  .name(\"appkit\")\n  .description(\"CLI tools for Databricks AppKit\")\n  .version(pkg.version);\n\ncmd.addCommand(setupCommand);\ncmd.addCommand(generateTypesCommand);\ncmd.addCommand(lintCommand);\ncmd.addCommand(docsCommand);\ncmd.addCommand(pluginCommand);\ncmd.addCommand(codemodCommand);\n\ncmd.parse();\n"],"mappings":";;;;;;;;;;;;;;AAcA,MAAM,UAAU,KADE,QAAQ,cAAc,OAAO,KAAK,IAAI,CAAC,EACzB,qBAAqB;AACrD,MAAM,MAAM,KAAK,MAAM,aAAa,SAAS,QAAQ,CAAC;AAEtD,MAAM,MAAM,IAAI,SAAS;AAEzB,IACG,KAAK,SAAS,CACd,YAAY,kCAAkC,CAC9C,QAAQ,IAAI,QAAQ;AAEvB,IAAI,WAAW,aAAa;AAC5B,IAAI,WAAW,qBAAqB;AACpC,IAAI,WAAW,YAAY;AAC3B,IAAI,WAAW,YAAY;AAC3B,IAAI,WAAW,cAAc;AAC7B,IAAI,WAAW,eAAe;AAE9B,IAAI,OAAO"}
+{"version":3,"file":"index.js","names":[],"sources":["../../src/cli/index.ts"],"sourcesContent":["#!/usr/bin/env node\nimport \"dotenv/config\";\nimport { readFileSync } from \"node:fs\";\nimport { dirname, join } from \"node:path\";\nimport { fileURLToPath } from \"node:url\";\nimport { Command } from \"commander\";\nimport { codemodCommand } from \"./commands/codemod/index.js\";\nimport { docsCommand } from \"./commands/docs.js\";\nimport { generateTypesCommand } from \"./commands/generate-types.js\";\nimport { lintCommand } from \"./commands/lint.js\";\nimport { pluginCommand } from \"./commands/plugin/index.js\";\nimport { setupCommand } from \"./commands/setup.js\";\n\nconst __dirname = dirname(fileURLToPath(import.meta.url));\nconst pkgPath = join(__dirname, \"../../package.json\");\nconst pkg = JSON.parse(readFileSync(pkgPath, \"utf-8\"));\n\nconst cmd = new Command();\n\ncmd\n  .name(\"appkit\")\n  .description(\"CLI tools for Databricks AppKit\")\n  .version(pkg.version);\n\ncmd.addCommand(setupCommand);\ncmd.addCommand(generateTypesCommand);\ncmd.addCommand(lintCommand);\ncmd.addCommand(docsCommand);\ncmd.addCommand(pluginCommand);\ncmd.addCommand(codemodCommand);\n\nawait cmd.parseAsync();\n"],"mappings":";;;;;;;;;;;;;;AAcA,MAAM,UAAU,KADE,QAAQ,cAAc,OAAO,KAAK,IAAI,CAAC,EACzB,qBAAqB;AACrD,MAAM,MAAM,KAAK,MAAM,aAAa,SAAS,QAAQ,CAAC;AAEtD,MAAM,MAAM,IAAI,SAAS;AAEzB,IACG,KAAK,SAAS,CACd,YAAY,kCAAkC,CAC9C,QAAQ,IAAI,QAAQ;AAEvB,IAAI,WAAW,aAAa;AAC5B,IAAI,WAAW,qBAAqB;AACpC,IAAI,WAAW,YAAY;AAC3B,IAAI,WAAW,YAAY;AAC3B,IAAI,WAAW,cAAc;AAC7B,IAAI,WAAW,eAAe;AAE9B,MAAM,IAAI,YAAY"}

package/dist/connectors/index.js

@@ -11,7 +11,7 @@ import "./jobs/index.js";
 import { buildMcpHostPolicy } from "./mcp/host-policy.js";
 import { AppKitMcpClient } from "./mcp/client.js";
 import "./mcp/index.js";
-import { SQLWarehouseConnector } from "./sql-warehouse/client.js";
+import { DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS, SQLWarehouseConnector } from "./sql-warehouse/client.js";
 import "./sql-warehouse/index.js";
 import "./vector-search/index.js";
 

package/dist/connectors/sql-warehouse/client.js

@@ -1,5 +1,6 @@
 import { createLogger } from "../../logging/logger.js";
 import { AppKitError } from "../../errors/base.js";
+import { ConfigurationError } from "../../errors/configuration.js";
 import { ConnectionError } from "../../errors/connection.js";
 import { ExecutionError } from "../../errors/execution.js";
 import { ValidationError } from "../../errors/validation.js";
@@ -8,14 +9,38 @@ import { TelemetryManager } from "../../telemetry/telemetry-manager.js";
 import { SpanKind, SpanStatusCode } from "../../telemetry/index.js";
 import { ArrowStreamProcessor } from "../../stream/arrow-stream-processor.js";
 import { executeStatementDefaults } from "./defaults.js";
+import { WarehousePollBackoff } from "./warehouse-poll-backoff.js";
+import { WarehouseStatusEmitter } from "./warehouse-status-emitter.js";
 import { Context } from "@databricks/sdk-experimental";
 
 //#region src/connectors/sql-warehouse/client.ts
 const logger = createLogger("connectors:sql-warehouse");
+/**
+* Default ceiling for how long {@link SQLWarehouseConnector.ensureWarehouseRunning}
+* will wait for a warehouse to reach the RUNNING state before giving up.
+*
+* Five minutes covers a cold-start of a classic warehouse on most workspaces;
+* serverless typically reaches RUNNING within ~30s.
+*/
+const DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS = 300 * 1e3;
+/**
+* Window during which a recent `RUNNING` observation lets subsequent calls
+* to {@link SQLWarehouseConnector.ensureWarehouseRunning} short-circuit
+* without making any SDK calls. Sized to roughly the upper bound of how
+* long Databricks keeps a warehouse "stickily" available between requests
+* — past that, we re-verify.
+*/
+const WAREHOUSE_RUNNING_CACHE_TTL_MS = 3e4;
 var SQLWarehouseConnector = class {
 	name = "sql-warehouse";
 	config;
 	_arrowProcessor = null;
+	/**
+	* Per-warehouse cache of the last RUNNING observation timestamp. Used by
+	* {@link ensureWarehouseRunning} to short-circuit warm-path callers; see
+	* {@link WAREHOUSE_RUNNING_CACHE_TTL_MS}.
+	*/
+	_recentlyRunning = /* @__PURE__ */ new Map();
 	telemetry;
 	telemetryMetrics;
 	constructor(config) {
@@ -157,6 +182,148 @@ var SQLWarehouseConnector = class {
 			includePrefix: true
 		});
 	}
+	/**
+	* Wait until the SQL warehouse is in the `RUNNING` state, auto-starting it
+	* if currently `STOPPED`. Emits a {@link WarehouseStatusUpdate} whenever
+	* the observed state changes, so callers can surface progress (e.g. over
+	* SSE) instead of letting the UI freeze on a cold warehouse. Equal
+	* successive observations are de-duplicated.
+	*
+	* Fast path: if this connector recently observed the warehouse RUNNING
+	* (within {@link WAREHOUSE_RUNNING_CACHE_TTL_MS}), the call returns
+	* immediately without any SDK round-trip or status emission. This keeps
+	* cache-hit analytics requests off the Databricks control plane.
+	*
+	* Behaviour by initial state:
+	* - `RUNNING`: emits one update, caches the observation, returns.
+	* - `STOPPED`: emits a synthetic `STARTING` update, calls
+	*   `workspaceClient.warehouses.start`, then polls until `RUNNING`.
+	*   When `autoStart: false`, throws `ConfigurationError` instead.
+	* - `STARTING` / `STOPPING`: polls until `RUNNING`.
+	* - `DELETED` / `DELETING`: throws `ConfigurationError.resourceNotFound`.
+	*
+	* Aborts and timeouts surface as `ExecutionError`. The poll loop uses
+	* exponential backoff with ±15% jitter (1s → 30s cap) and runs inside a
+	* `sql.warehouseReady` telemetry span.
+	*/
+	async ensureWarehouseRunning(workspaceClient, warehouseId, opts) {
+		const { onStatus, signal, timeoutMs = DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS, autoStart = true } = opts;
+		if (signal?.aborted) throw ExecutionError.canceled();
+		if (!warehouseId) throw ValidationError.missingField("warehouse_id");
+		if (this._isRecentlyRunning(warehouseId)) return;
+		return this.telemetry.startActiveSpan("sql.warehouseReady", {
+			kind: SpanKind.CLIENT,
+			attributes: {
+				"db.system": "databricks",
+				"db.warehouse_id": warehouseId,
+				"db.warehouse.startup_timeout_ms": timeoutMs
+			}
+		}, async (span) => {
+			const startTime = Date.now();
+			const emitter = new WarehouseStatusEmitter(span, startTime, onStatus);
+			let didStart = false;
+			const backoff = new WarehousePollBackoff();
+			try {
+				while (true) {
+					if (signal?.aborted) throw ExecutionError.canceled();
+					if (Date.now() - startTime > timeoutMs) throw ExecutionError.statementFailed(`SQL warehouse did not reach RUNNING within ${timeoutMs}ms`);
+					const info = await workspaceClient.warehouses.get({ id: warehouseId }, this._createContext(signal));
+					const state = info?.state;
+					const summary = info?.health?.summary;
+					switch (state) {
+						case "RUNNING":
+							emitter.emit(state, summary);
+							this._recentlyRunning.set(warehouseId, Date.now());
+							span.setAttribute("db.warehouse.attempts", emitter.attempt);
+							span.setStatus({ code: SpanStatusCode.OK });
+							return;
+						case "DELETED":
+						case "DELETING": throw ConfigurationError.resourceNotFound("Warehouse ID", `The configured SQL warehouse is ${state}. Update DATABRICKS_WAREHOUSE_ID to point at an active warehouse.`);
+						case "STOPPED":
+							if (!autoStart) throw new ConfigurationError("The configured SQL warehouse is STOPPED and analytics auto-start is disabled. Start the warehouse manually or set analytics.autoStartWarehouse=true.");
+							if (!didStart) {
+								emitter.emit("STARTING", summary);
+								await workspaceClient.warehouses.start({ id: warehouseId }, this._createContext(signal));
+								didStart = true;
+							} else emitter.emit(state, summary);
+							break;
+						case "STARTING":
+						case "STOPPING":
+							emitter.emit(state, summary);
+							break;
+						default: throw ExecutionError.unknownState(String(state ?? "unknown"));
+					}
+					const sleepMs = backoff.next();
+					if (Date.now() + sleepMs - startTime >= timeoutMs) throw ExecutionError.statementFailed(`SQL warehouse did not reach RUNNING within ${timeoutMs}ms`);
+					await this._sleepRespectingAbort(sleepMs, signal);
+				}
+			} catch (error) {
+				this._throwSanitizedReadinessError(error, warehouseId, span);
+			} finally {
+				span.end();
+			}
+		}, {
+			name: this.name,
+			includePrefix: true
+		});
+	}
+	/**
+	* `true` if this connector observed `warehouseId` in the RUNNING state
+	* within the recent-cache TTL.
+	*/
+	_isRecentlyRunning(warehouseId) {
+		const observedAt = this._recentlyRunning.get(warehouseId);
+		return observedAt !== void 0 && Date.now() - observedAt < WAREHOUSE_RUNNING_CACHE_TTL_MS;
+	}
+	/**
+	* Final landing for any error thrown by the readiness poll loop. Records
+	* the original on the span and (at debug level) on the logger, then
+	* rethrows either the structured AppKitError unchanged or a curated
+	* ExecutionError — never the raw SDK message, which can contain operator
+	* internals.
+	*
+	* The `logger.debug` covers environments where OTel isn't configured
+	* (common in dev) so the raw error isn't lost just because no span
+	* exporter is hooked up.
+	*/
+	_throwSanitizedReadinessError(error, warehouseId, span) {
+		if (error instanceof AppKitError) {
+			span.recordException(error);
+			span.setStatus({
+				code: SpanStatusCode.ERROR,
+				message: error.code
+			});
+			throw error;
+		}
+		span.recordException(error instanceof Error ? error : new Error(String(error)));
+		logger.debug("Warehouse readiness check raw error for %s: %O", warehouseId, error);
+		const wrapped = ExecutionError.statementFailed("Warehouse readiness check failed");
+		span.setStatus({
+			code: SpanStatusCode.ERROR,
+			message: wrapped.code
+		});
+		throw wrapped;
+	}
+	/**
+	* Sleep for `ms` milliseconds, but resolve early (and reject with
+	* `ExecutionError.canceled()`) if `signal` aborts mid-sleep. Used by the
+	* warehouse readiness loop so that client disconnects don't keep the loop
+	* polling for the full interval.
+	*/
+	_sleepRespectingAbort(ms, signal) {
+		if (signal?.aborted) return Promise.reject(ExecutionError.canceled());
+		return new Promise((resolve, reject) => {
+			const timer = setTimeout(() => {
+				signal?.removeEventListener("abort", onAbort);
+				resolve();
+			}, ms);
+			const onAbort = () => {
+				clearTimeout(timer);
+				reject(ExecutionError.canceled());
+			};
+			signal?.addEventListener("abort", onAbort, { once: true });
+		});
+	}
 	async _pollForStatementResult(workspaceClient, statementId, timeout = executeStatementDefaults.timeout, signal) {
 		return this.telemetry.startActiveSpan("sql.poll", { attributes: {
 			"db.statement_id": statementId,
@@ -323,5 +490,5 @@ var SQLWarehouseConnector = class {
 };
 
 //#endregion
-export { SQLWarehouseConnector };
+export { DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS, SQLWarehouseConnector };
 //# sourceMappingURL=client.js.map