@databricks/appkit 0.75.1 → 0.76.0

package/CLAUDE.md

@@ -59,7 +59,7 @@ npx @databricks/appkit docs <query>
 - [Plugin management](./docs/plugins/plugin-management.md): AppKit includes a CLI for managing plugins. All commands are available under npx @databricks/appkit plugin.
 - [Server plugin](./docs/plugins/server.md): Provides HTTP server capabilities with development and production modes.
 - [Plugin Stability Tiers](./docs/plugins/stability.md): AppKit plugins have a two-tier stability system that communicates API maturity and breaking-change expectations.
-- [Testing](./docs/plugins/testing.md): AppKit ships a testing kit at @databricks/appkit/testing so you can test a plugin — including its cross-plugin tool calls and streaming responses — without a live Databricks workspace, credentials, or network access. That makes plugin tests fast and lets them run in CI, where no workspace is available.
+- [Testing](./docs/plugins/testing.md): AppKit ships a testing kit at @databricks/appkit/testing so you can test a plugin, including its cross-plugin tool calls and streaming responses, without a live Databricks workspace, credentials, or network access. Plugin tests stay fast and run in CI, where no workspace is available.
 
 ## appkit API reference [collapsed]
 

package/dist/appkit/package.js

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

package/dist/cache/index.d.ts

@@ -139,6 +139,16 @@ declare class CacheManager {
   generateKey(parts: (string | number | object)[], userKey: string): string;
   /** Close the cache */
   close(): Promise<void>;
+  /**
+   * Drop the singleton so the next {@link getInstance} builds a fresh manager.
+   *
+   * Both fields must clear — `getInstance()` falls back to `initPromise` when
+   * `instance` is null. A pointer drop, not teardown: call {@link close} first
+   * or the old storage leaks (a `pg.Pool` under `PersistentStorage`).
+   *
+   * @internal
+   */
+  static reset(): void;
   /**
    * Check if the storage is healthy
    * @returns Promise of true if the storage is healthy, false otherwise

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","names":[],"sources":["../../src/cache/index.ts"],"mappings":";;;;;;;AAsDA;;;;;;;;;;;cAAa,YAAA;EAAA,wBACa,uBAAA;EAAA,wBACA,qBAAA;EAAA,iBACP,IAAA;EAAA,eACF,QAAA;EAAA,eACA,WAAA;EAAA,QAEP,OAAA;EAAA,QACA,MAAA;EAAA,QACA,gBAAA;EAAA,QACA,iBAAA;EAAA,QACA,kBAAA;EAAA,QAEA,SAAA;EAAA,QACA,gBAAA;EAAA,QAKD,WAAA,CAAA;EAlBiB;;;;;;EAAA,OA+CjB,eAAA,CAAA,GAAmB,YAAA;EAvClB;;;;;;;EAAA,OAyDK,WAAA,CACX,UAAA,GAAa,OAAA,CAAQ,WAAA,IACpB,OAAA,CAAQ,YAAA;EAFE;;;;;;;;;;;;EAAA,eA+BQ,MAAA;EA2EyB;;;;;;;;;;;;;;;;;;;;EAFxC,YAAA,GAAA,CACJ,GAAA,gCACA,EAAA,GAAK,YAAA,GAAe,WAAA,KAAgB,OAAA,CAAQ,CAAA,GAC5C,OAAA,UACA,OAAA;IAAY,GAAA;IAAc,YAAA,GAAe,WAAA;EAAA,IACxC,OAAA,CAAQ,CAAA;EAkRR;;;;;;;EAAA,QA7HK,iBAAA;EA6JgB;;;;;EAhGlB,GAAA,GAAA,CAAO,GAAA,WAAc,OAAA,CAAQ,CAAA;EA4H7B;;;;;;;EAAA,QA3GQ,QAAA;;UAYN,YAAA;;;;;;;;EA+BF,GAAA,GAAA,CACJ,GAAA,UACA,KAAA,EAAO,CAAA,EACP,OAAA;IAAY,GAAA;EAAA,IACX,OAAA;;;;;;EAaG,MAAA,CAAO,GAAA,WAAc,OAAA;;EAMrB,KAAA,CAAA,GAAS,OAAA;;;;;;EAaT,GAAA,CAAI,GAAA,WAAc,OAAA;;;;;;;EAaxB,WAAA,CAAY,KAAA,gCAAqC,OAAA;;EAO3C,KAAA,CAAA,GAAS,OAAA;;;;;EAQT,gBAAA,CAAA,GAAoB,OAAA;AAAA"}
+{"version":3,"file":"index.d.ts","names":[],"sources":["../../src/cache/index.ts"],"mappings":";;;;;;;AAsDA;;;;;;;;;;;cAAa,YAAA;EAAA,wBACa,uBAAA;EAAA,wBACA,qBAAA;EAAA,iBACP,IAAA;EAAA,eACF,QAAA;EAAA,eACA,WAAA;EAAA,QAEP,OAAA;EAAA,QACA,MAAA;EAAA,QACA,gBAAA;EAAA,QACA,iBAAA;EAAA,QACA,kBAAA;EAAA,QAEA,SAAA;EAAA,QACA,gBAAA;EAAA,QAKD,WAAA,CAAA;EAlBiB;;;;;;EAAA,OA+CjB,eAAA,CAAA,GAAmB,YAAA;EAvClB;;;;;;;EAAA,OAyDK,WAAA,CACX,UAAA,GAAa,OAAA,CAAQ,WAAA,IACpB,OAAA,CAAQ,YAAA;EAFE;;;;;;;;;;;;EAAA,eAmCQ,MAAA;EA2EyB;;;;;;;;;;;;;;;;;;;;EAFxC,YAAA,GAAA,CACJ,GAAA,gCACA,EAAA,GAAK,YAAA,GAAe,WAAA,KAAgB,OAAA,CAAQ,CAAA,GAC5C,OAAA,UACA,OAAA;IAAY,GAAA;IAAc,YAAA,GAAe,WAAA;EAAA,IACxC,OAAA,CAAQ,CAAA;EAkRR;;;;;;;EAAA,QA7HK,iBAAA;EA6JgB;;;;;EAhGlB,GAAA,GAAA,CAAO,GAAA,WAAc,OAAA,CAAQ,CAAA;EAiI5B;;;;;;;EAAA,QAhHO,QAAA;;UAYN,YAAA;;;;;;;;EA+BF,GAAA,GAAA,CACJ,GAAA,UACA,KAAA,EAAO,CAAA,EACP,OAAA;IAAY,GAAA;EAAA,IACX,OAAA;;;;;;EAaG,MAAA,CAAO,GAAA,WAAc,OAAA;;EAMrB,KAAA,CAAA,GAAS,OAAA;;;;;;EAaT,GAAA,CAAI,GAAA,WAAc,OAAA;;;;;;;EAaxB,WAAA,CAAY,KAAA,gCAAqC,OAAA;;EAO3C,KAAA,CAAA,GAAS,OAAA;;;;;;;;;;SAaR,KAAA,CAAA;;;;;EASD,gBAAA,CAAA,GAAoB,OAAA;AAAA"}

package/dist/cache/index.js

@@ -384,6 +384,19 @@ var CacheManager = class CacheManager {
 		await this.storage.close();
 	}
 	/**
+	* Drop the singleton so the next {@link getInstance} builds a fresh manager.
+	*
+	* Both fields must clear — `getInstance()` falls back to `initPromise` when
+	* `instance` is null. A pointer drop, not teardown: call {@link close} first
+	* or the old storage leaks (a `pg.Pool` under `PersistentStorage`).
+	*
+	* @internal
+	*/
+	static reset() {
+		CacheManager.instance = null;
+		CacheManager.initPromise = null;
+	}
+	/**
 	* Check if the storage is healthy
 	* @returns Promise of true if the storage is healthy, false otherwise
 	*/

package/dist/cache/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../../src/cache/index.ts"],"sourcesContent":["import { createHash } from \"node:crypto\";\n\nimport type { CacheConfig, CacheEntry, CacheStorage } from \"shared\";\n\nimport { createLakebasePool } from \"../connectors/lakebase\";\nimport { getClientOptions } from \"../context/client-options\";\nimport { AppKitError, ExecutionError, InitializationError } from \"../errors\";\nimport { createLogger } from \"../logging/logger\";\nimport type { Counter, TelemetryProvider } from \"../telemetry\";\nimport { SpanStatusCode, TelemetryManager } from \"../telemetry\";\nimport { deepMerge } from \"../utils\";\nimport { ApiError, createWorkspaceClient } from \"../workspace-client\";\nimport { cacheDefaults } from \"./defaults\";\nimport { InMemoryStorage, PersistentStorage } from \"./storage\";\n\nconst logger = createLogger(\"cache\");\n\n/**\n * Reference-counted in-flight cache execution entry.\n *\n * `sharedController` decouples the cached `fn()` from any single caller's\n * abort signal. Callers join an in-flight entry by incrementing `refCount`;\n * when a caller aborts, refCount is decremented. The shared controller is\n * aborted only when refCount drops to 0 — i.e. all callers have abandoned\n * the request. This prevents one caller's cancellation (e.g. React\n * StrictMode unmount) from poisoning the in-flight result for other still-\n * connected awaiters.\n */\ninterface InFlightEntry<T> {\n  promise: Promise<T>;\n  refCount: number;\n  sharedController: AbortController;\n  abortTimer?: ReturnType<typeof setTimeout>;\n}\n\nfunction createAbortError(signal: AbortSignal): unknown {\n  if (signal.reason !== undefined) return signal.reason;\n  return new DOMException(\"The operation was aborted.\", \"AbortError\");\n}\n\n/**\n * Cache manager class to handle cache operations.\n * Can be used with in-memory storage or persistent storage (Lakebase).\n *\n * The cache is automatically initialized by AppKit. Use `getInstanceSync()` to access\n * the singleton instance after initialization.\n *\n * @internal\n * @example\n * ```typescript\n * const cache = CacheManager.getInstanceSync();\n * const result = await cache.getOrExecute([\"users\", userId], () => fetchUser(userId), userKey);\n * ```\n */\nexport class CacheManager {\n  private static readonly MIN_CLEANUP_INTERVAL_MS = 60_000;\n  private static readonly ABORT_GRACE_PERIOD_MS = 100;\n  private readonly name: string = \"cache-manager\";\n  private static instance: CacheManager | null = null;\n  private static initPromise: Promise<CacheManager> | null = null;\n\n  private storage: CacheStorage;\n  private config: CacheConfig;\n  private inFlightRequests: Map<string, InFlightEntry<unknown>>;\n  private cleanupInProgress: boolean;\n  private lastCleanupAttempt: number;\n\n  private telemetry: TelemetryProvider;\n  private telemetryMetrics: {\n    cacheHitCount: Counter;\n    cacheMissCount: Counter;\n  };\n\n  private constructor(storage: CacheStorage, config: CacheConfig) {\n    this.storage = storage;\n    this.config = config;\n    this.inFlightRequests = new Map();\n    this.cleanupInProgress = false;\n    this.lastCleanupAttempt = 0;\n\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      cacheHitCount: this.telemetry.getMeter().createCounter(\"cache.hit\", {\n        description: \"Total number of cache hits\",\n        unit: \"1\",\n      }),\n      cacheMissCount: this.telemetry.getMeter().createCounter(\"cache.miss\", {\n        description: \"Total number of cache misses\",\n        unit: \"1\",\n      }),\n    };\n  }\n\n  /**\n   * Get the singleton instance of the cache manager (sync version).\n   *\n   * Throws if not initialized - ensure AppKit.create() has completed first.\n   * @returns CacheManager instance\n   */\n  static getInstanceSync(): CacheManager {\n    if (!CacheManager.instance) {\n      throw InitializationError.notInitialized(\n        \"CacheManager\",\n        \"Ensure AppKit.create() has completed before accessing the cache\",\n      );\n    }\n\n    return CacheManager.instance;\n  }\n\n  /**\n   * Initialize and get the singleton instance of the cache manager.\n   * Called internally by AppKit - prefer `getInstanceSync()` for plugin access.\n   * @param userConfig - User configuration for the cache manager\n   * @returns CacheManager instance\n   * @internal\n   */\n  static async getInstance(\n    userConfig?: Partial<CacheConfig>,\n  ): Promise<CacheManager> {\n    if (CacheManager.instance) {\n      return CacheManager.instance;\n    }\n\n    if (!CacheManager.initPromise) {\n      CacheManager.initPromise = CacheManager.create(userConfig).then(\n        (instance) => {\n          CacheManager.instance = instance;\n          return instance;\n        },\n      );\n    }\n\n    return CacheManager.initPromise;\n  }\n\n  /**\n   * Create a new cache manager instance\n   *\n   * Storage selection logic:\n   * 1. If `storage` provided and healthy → use provided storage\n   * 2. If `storage` provided but unhealthy → fallback to InMemory (or disable if strictPersistence)\n   * 3. If no `storage` provided and Lakebase available → use Lakebase\n   * 4. If no `storage` provided and Lakebase unavailable → fallback to InMemory (or disable if strictPersistence)\n   *\n   * @param userConfig - User configuration for the cache manager\n   * @returns CacheManager instance\n   */\n  private static async create(\n    userConfig?: Partial<CacheConfig>,\n  ): Promise<CacheManager> {\n    const config = deepMerge(cacheDefaults, userConfig);\n\n    if (config.storage) {\n      const isHealthy = await config.storage.healthCheck();\n      if (isHealthy) {\n        return new CacheManager(config.storage, config);\n      }\n\n      if (config.strictPersistence) {\n        const disabledConfig = { ...config, enabled: false };\n        return new CacheManager(\n          new InMemoryStorage(disabledConfig),\n          disabledConfig,\n        );\n      }\n\n      return new CacheManager(new InMemoryStorage(config), config);\n    }\n\n    // try to use lakebase storage\n    try {\n      const workspaceClient = createWorkspaceClient({\n        clientOptions: getClientOptions(),\n      }).toLegacyWorkspaceClient();\n      const pool = createLakebasePool({ workspaceClient });\n      const persistentStorage = new PersistentStorage(config, pool);\n\n      const isHealthy = await persistentStorage.healthCheck();\n      if (isHealthy) {\n        await persistentStorage.initialize();\n        return new CacheManager(persistentStorage, config);\n      }\n\n      // Health check failed, close the pool and fallback\n      await pool.end();\n    } catch {\n      // lakebase unavailable, continue with in-memory storage\n    }\n\n    if (config.strictPersistence) {\n      const disabledConfig = { ...config, enabled: false };\n      return new CacheManager(\n        new InMemoryStorage(disabledConfig),\n        disabledConfig,\n      );\n    }\n\n    return new CacheManager(new InMemoryStorage(config), config);\n  }\n\n  /**\n   * Get or execute a function and cache the result.\n   *\n   * Multiple concurrent callers with the same `cacheKey` are deduplicated\n   * onto a single in-flight execution. Each caller may pass its own\n   * `callerSignal`; the underlying `fn()` is run with a shared, internally\n   * managed `AbortSignal` that aborts only when *all* callers have\n   * abandoned the request (reference counted). This decouples a single\n   * caller's cancellation (e.g. React StrictMode unmount) from the shared\n   * result, so other still-connected callers receive the cached value\n   * normally.\n   *\n   * @param key - Cache key parts\n   * @param fn - Function to execute. Receives the cache-owned shared signal;\n   *   pass it through to the underlying I/O so the work is cancelled when\n   *   no caller is left waiting.\n   * @param userKey - User key for cache namespacing\n   * @param options - Options for the cache\n   * @returns Promise of the result\n   */\n  async getOrExecute<T>(\n    key: (string | number | object)[],\n    fn: (sharedSignal?: AbortSignal) => Promise<T>,\n    userKey: string,\n    options?: { ttl?: number; callerSignal?: AbortSignal },\n  ): Promise<T> {\n    if (!this.config.enabled) return fn(options?.callerSignal);\n\n    const callerSignal = options?.callerSignal;\n    if (callerSignal?.aborted) {\n      throw createAbortError(callerSignal);\n    }\n\n    const cacheKey = this.generateKey(key, userKey);\n\n    return this.telemetry.startActiveSpan(\n      \"cache.getOrExecute\",\n      {\n        attributes: {\n          \"cache.key\": cacheKey,\n          \"cache.enabled\": this.config.enabled,\n          \"cache.persistent\": this.storage.isPersistent(),\n        },\n      },\n      async (span) => {\n        try {\n          const cached = await this.getValid<T>(cacheKey);\n          if (cached !== null) {\n            span.setAttribute(\"cache.hit\", true);\n            span.setStatus({ code: SpanStatusCode.OK });\n            this.telemetryMetrics.cacheHitCount.add(1, {\n              \"cache.key\": cacheKey,\n            });\n\n            logger.event()?.setExecution({\n              cache_hit: true,\n              cache_key: cacheKey,\n            });\n\n            return cached.value;\n          }\n\n          // check if the value is being processed by another request — join\n          // the existing in-flight entry under reference counting so this\n          // caller's abort doesn't poison the shared result.\n          const existing = this.inFlightRequests.get(cacheKey) as\n            | InFlightEntry<T>\n            | undefined;\n          if (existing && !existing.sharedController.signal.aborted) {\n            existing.refCount++;\n            // Cancel any pending abort timer — a new caller has joined\n            if (existing.abortTimer) {\n              clearTimeout(existing.abortTimer);\n              existing.abortTimer = undefined;\n            }\n            span.setAttribute(\"cache.hit\", true);\n            span.setAttribute(\"cache.deduplication\", true);\n            span.addEvent(\"cache.deduplication_used\", {\n              \"cache.key\": cacheKey,\n            });\n            this.telemetryMetrics.cacheHitCount.add(1, {\n              \"cache.key\": cacheKey,\n              \"cache.deduplication\": \"true\",\n            });\n\n            logger.event()?.setExecution({\n              cache_hit: true,\n              cache_key: cacheKey,\n              cache_deduplication: true,\n            });\n\n            return await this._waitWithRefCount(existing, callerSignal);\n          }\n\n          // cache miss - execute function under a shared abort controller\n          span.setAttribute(\"cache.hit\", false);\n          span.addEvent(\"cache.miss\", { \"cache.key\": cacheKey });\n          this.telemetryMetrics.cacheMissCount.add(1, {\n            \"cache.key\": cacheKey,\n          });\n\n          logger.event()?.setExecution({\n            cache_hit: false,\n            cache_key: cacheKey,\n          });\n\n          const sharedController = new AbortController();\n          const entry: InFlightEntry<T> = {\n            promise: undefined as unknown as Promise<T>,\n            refCount: 1,\n            sharedController,\n          };\n\n          entry.promise = fn(sharedController.signal)\n            .then(async (result) => {\n              await this.set(cacheKey, result, options);\n              span.addEvent(\"cache.value_stored\", {\n                \"cache.key\": cacheKey,\n                \"cache.ttl\": options?.ttl ?? this.config.ttl ?? 3600,\n              });\n              return result;\n            })\n            .catch((error) => {\n              span.recordException(error);\n              span.setStatus({ code: SpanStatusCode.ERROR });\n              // If the shared controller aborted, all callers have already\n              // abandoned the request (or are about to via their own signals)\n              // — propagate the original error without wrapping. No live\n              // awaiter will observe this rejection.\n              if (sharedController.signal.aborted) {\n                throw error;\n              }\n              if (error instanceof AppKitError || error instanceof ApiError) {\n                throw error;\n              }\n              throw ExecutionError.statementFailed(\n                error instanceof Error ? error.message : String(error),\n              );\n            })\n            .finally(() => {\n              if (this.inFlightRequests.get(cacheKey) === entry) {\n                this.inFlightRequests.delete(cacheKey);\n              }\n            });\n\n          // Suppress unhandled rejection warnings when every caller bailed\n          // before fn() resolved (their own promises rejected via\n          // waitWithRefCount; the underlying entry.promise has no awaiter).\n          entry.promise.catch(() => {});\n\n          this.inFlightRequests.set(cacheKey, entry as InFlightEntry<unknown>);\n\n          const result = await this._waitWithRefCount(entry, callerSignal);\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          span.recordException(error as Error);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n          throw error;\n        } finally {\n          span.end();\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  /**\n   * Wait on an in-flight entry, racing the underlying promise against the\n   * caller's abort signal. When the caller aborts, the entry's refCount is\n   * decremented; if it hits zero the shared controller is aborted so the\n   * underlying `fn()` can stop. Other callers continue to await the same\n   * entry and receive the result when it arrives.\n   */\n  private _waitWithRefCount<T>(\n    entry: InFlightEntry<T>,\n    callerSignal?: AbortSignal,\n  ): Promise<T> {\n    if (!callerSignal) return entry.promise;\n\n    return new Promise<T>((resolve, reject) => {\n      let settled = false;\n\n      const release = () => {\n        if (entry.refCount > 0) entry.refCount--;\n        if (entry.refCount <= 0 && !entry.sharedController.signal.aborted) {\n          // Grace period: delay abort so a StrictMode remount can join\n          // the in-flight entry before the shared execution is cancelled.\n          entry.abortTimer = setTimeout(() => {\n            if (entry.refCount <= 0 && !entry.sharedController.signal.aborted) {\n              entry.sharedController.abort(\n                callerSignal.reason ?? \"all cache callers aborted\",\n              );\n            }\n          }, CacheManager.ABORT_GRACE_PERIOD_MS);\n        }\n      };\n\n      const onAbort = () => {\n        if (settled) return;\n        settled = true;\n        callerSignal.removeEventListener(\"abort\", onAbort);\n        release();\n        reject(createAbortError(callerSignal));\n      };\n\n      if (callerSignal.aborted) {\n        onAbort();\n        return;\n      }\n\n      callerSignal.addEventListener(\"abort\", onAbort, { once: true });\n\n      entry.promise.then(\n        (value) => {\n          if (settled) return;\n          settled = true;\n          callerSignal.removeEventListener(\"abort\", onAbort);\n          resolve(value);\n        },\n        (error) => {\n          if (settled) return;\n          settled = true;\n          callerSignal.removeEventListener(\"abort\", onAbort);\n          reject(error);\n        },\n      );\n    });\n  }\n\n  /**\n   * Get a cached value\n   * @param key - Cache key\n   * @returns Promise of the value or null if not found or expired\n   */\n  async get<T>(key: string): Promise<T | null> {\n    if (!this.config.enabled) return null;\n\n    // probabilistic cleanup trigger\n    this.maybeCleanup();\n\n    const entry = await this.getValid<T>(key);\n    return entry?.value ?? null;\n  }\n\n  /**\n   * Get a cached entry only if it has not expired.\n   * Returns null on miss or expired (and deletes the expired entry).\n   *\n   * Storage implementations return entries unconditionally — expiry handling\n   * lives at the CacheManager layer.\n   */\n  private async getValid<T>(key: string): Promise<CacheEntry<T> | null> {\n    const entry = await this.storage.get<T>(key);\n    if (!entry) return null;\n\n    if (Date.now() > entry.expiry) {\n      await this.storage.delete(key);\n      return null;\n    }\n    return entry;\n  }\n\n  /** Probabilistically trigger cleanup of expired entries (fire-and-forget) */\n  private maybeCleanup(): void {\n    if (this.cleanupInProgress) return;\n    if (!this.storage.isPersistent()) return;\n    const now = Date.now();\n    if (now - this.lastCleanupAttempt < CacheManager.MIN_CLEANUP_INTERVAL_MS)\n      return;\n\n    const probability = this.config.cleanupProbability ?? 0.01;\n\n    if (Math.random() > probability) return;\n\n    this.lastCleanupAttempt = now;\n\n    this.cleanupInProgress = true;\n    (this.storage as PersistentStorage)\n      .cleanupExpired()\n      .catch((error) => {\n        logger.debug(\"Error cleaning up expired entries: %O\", error);\n      })\n      .finally(() => {\n        this.cleanupInProgress = false;\n      });\n  }\n\n  /**\n   * Set a value in the cache\n   * @param key - Cache key\n   * @param value - Value to set\n   * @param options - Options for the cache\n   * @returns Promise of the result\n   */\n  async set<T>(\n    key: string,\n    value: T,\n    options?: { ttl?: number },\n  ): Promise<void> {\n    if (!this.config.enabled) return;\n\n    const ttl = options?.ttl ?? this.config.ttl ?? 3600;\n    const expiryTime = Date.now() + ttl * 1000;\n    await this.storage.set(key, { value, expiry: expiryTime });\n  }\n\n  /**\n   * Delete a value from the cache\n   * @param key - Cache key\n   * @returns Promise of the result\n   */\n  async delete(key: string): Promise<void> {\n    if (!this.config.enabled) return;\n    await this.storage.delete(key);\n  }\n\n  /** Clear the cache */\n  async clear(): Promise<void> {\n    await this.storage.clear();\n    for (const entry of this.inFlightRequests.values()) {\n      if (entry.abortTimer) clearTimeout(entry.abortTimer);\n    }\n    this.inFlightRequests.clear();\n  }\n\n  /**\n   * Check if a value exists in the cache\n   * @param key - Cache key\n   * @returns Promise of true if the value exists, false otherwise\n   */\n  async has(key: string): Promise<boolean> {\n    if (!this.config.enabled) return false;\n\n    const entry = await this.getValid(key);\n    return entry !== null;\n  }\n\n  /**\n   * Generate a cache key\n   * @param parts - Parts of the key\n   * @param userKey - User key\n   * @returns Cache key\n   */\n  generateKey(parts: (string | number | object)[], userKey: string): string {\n    const allParts = [userKey, ...parts];\n    const serialized = JSON.stringify(allParts);\n    return createHash(\"sha256\").update(serialized).digest(\"hex\");\n  }\n\n  /** Close the cache */\n  async close(): Promise<void> {\n    await this.storage.close();\n  }\n\n  /**\n   * Check if the storage is healthy\n   * @returns Promise of true if the storage is healthy, false otherwise\n   */\n  async isStorageHealthy(): Promise<boolean> {\n    return this.storage.healthCheck();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAeA,MAAM,SAAS,aAAa,QAAQ;AAoBpC,SAAS,iBAAiB,QAA8B;AACtD,KAAI,OAAO,WAAW,OAAW,QAAO,OAAO;AAC/C,QAAO,IAAI,aAAa,8BAA8B,aAAa;;;;;;;;;;;;;;;;AAiBrE,IAAa,eAAb,MAAa,aAAa;CACxB,OAAwB,0BAA0B;CAClD,OAAwB,wBAAwB;CAChD,AAAiB,OAAe;CAChC,OAAe,WAAgC;CAC/C,OAAe,cAA4C;CAE3D,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,AAAQ;CACR,AAAQ;CAKR,AAAQ,YAAY,SAAuB,QAAqB;AAC9D,OAAK,UAAU;AACf,OAAK,SAAS;AACd,OAAK,mCAAmB,IAAI,KAAK;AACjC,OAAK,oBAAoB;AACzB,OAAK,qBAAqB;AAE1B,OAAK,YAAY,iBAAiB,YAChC,KAAK,MACL,KAAK,OAAO,UACb;AACD,OAAK,mBAAmB;GACtB,eAAe,KAAK,UAAU,UAAU,CAAC,cAAc,aAAa;IAClE,aAAa;IACb,MAAM;IACP,CAAC;GACF,gBAAgB,KAAK,UAAU,UAAU,CAAC,cAAc,cAAc;IACpE,aAAa;IACb,MAAM;IACP,CAAC;GACH;;;;;;;;CASH,OAAO,kBAAgC;AACrC,MAAI,CAAC,aAAa,SAChB,OAAM,oBAAoB,eACxB,gBACA,kEACD;AAGH,SAAO,aAAa;;;;;;;;;CAUtB,aAAa,YACX,YACuB;AACvB,MAAI,aAAa,SACf,QAAO,aAAa;AAGtB,MAAI,CAAC,aAAa,YAChB,cAAa,cAAc,aAAa,OAAO,WAAW,CAAC,MACxD,aAAa;AACZ,gBAAa,WAAW;AACxB,UAAO;IAEV;AAGH,SAAO,aAAa;;;;;;;;;;;;;;CAetB,aAAqB,OACnB,YACuB;EACvB,MAAM,SAAS,UAAU,eAAe,WAAW;AAEnD,MAAI,OAAO,SAAS;AAElB,OADkB,MAAM,OAAO,QAAQ,aAAa,CAElD,QAAO,IAAI,aAAa,OAAO,SAAS,OAAO;AAGjD,OAAI,OAAO,mBAAmB;IAC5B,MAAM,iBAAiB;KAAE,GAAG;KAAQ,SAAS;KAAO;AACpD,WAAO,IAAI,aACT,IAAI,gBAAgB,eAAe,EACnC,eACD;;AAGH,UAAO,IAAI,aAAa,IAAI,gBAAgB,OAAO,EAAE,OAAO;;AAI9D,MAAI;GAIF,MAAM,OAAO,mBAAmB,EAAE,iBAHV,sBAAsB,EAC5C,eAAe,kBAAkB,EAClC,CAAC,CAAC,yBAAyB,EACuB,CAAC;GACpD,MAAM,oBAAoB,IAAI,kBAAkB,QAAQ,KAAK;AAG7D,OADkB,MAAM,kBAAkB,aAAa,EACxC;AACb,UAAM,kBAAkB,YAAY;AACpC,WAAO,IAAI,aAAa,mBAAmB,OAAO;;AAIpD,SAAM,KAAK,KAAK;UACV;AAIR,MAAI,OAAO,mBAAmB;GAC5B,MAAM,iBAAiB;IAAE,GAAG;IAAQ,SAAS;IAAO;AACpD,UAAO,IAAI,aACT,IAAI,gBAAgB,eAAe,EACnC,eACD;;AAGH,SAAO,IAAI,aAAa,IAAI,gBAAgB,OAAO,EAAE,OAAO;;;;;;;;;;;;;;;;;;;;;;CAuB9D,MAAM,aACJ,KACA,IACA,SACA,SACY;AACZ,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO,GAAG,SAAS,aAAa;EAE1D,MAAM,eAAe,SAAS;AAC9B,MAAI,cAAc,QAChB,OAAM,iBAAiB,aAAa;EAGtC,MAAM,WAAW,KAAK,YAAY,KAAK,QAAQ;AAE/C,SAAO,KAAK,UAAU,gBACpB,sBACA,EACE,YAAY;GACV,aAAa;GACb,iBAAiB,KAAK,OAAO;GAC7B,oBAAoB,KAAK,QAAQ,cAAc;GAChD,EACF,EACD,OAAO,SAAS;AACd,OAAI;IACF,MAAM,SAAS,MAAM,KAAK,SAAY,SAAS;AAC/C,QAAI,WAAW,MAAM;AACnB,UAAK,aAAa,aAAa,KAAK;AACpC,UAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,UAAK,iBAAiB,cAAc,IAAI,GAAG,EACzC,aAAa,UACd,CAAC;AAEF,YAAO,OAAO,EAAE,aAAa;MAC3B,WAAW;MACX,WAAW;MACZ,CAAC;AAEF,YAAO,OAAO;;IAMhB,MAAM,WAAW,KAAK,iBAAiB,IAAI,SAAS;AAGpD,QAAI,YAAY,CAAC,SAAS,iBAAiB,OAAO,SAAS;AACzD,cAAS;AAET,SAAI,SAAS,YAAY;AACvB,mBAAa,SAAS,WAAW;AACjC,eAAS,aAAa;;AAExB,UAAK,aAAa,aAAa,KAAK;AACpC,UAAK,aAAa,uBAAuB,KAAK;AAC9C,UAAK,SAAS,4BAA4B,EACxC,aAAa,UACd,CAAC;AACF,UAAK,iBAAiB,cAAc,IAAI,GAAG;MACzC,aAAa;MACb,uBAAuB;MACxB,CAAC;AAEF,YAAO,OAAO,EAAE,aAAa;MAC3B,WAAW;MACX,WAAW;MACX,qBAAqB;MACtB,CAAC;AAEF,YAAO,MAAM,KAAK,kBAAkB,UAAU,aAAa;;AAI7D,SAAK,aAAa,aAAa,MAAM;AACrC,SAAK,SAAS,cAAc,EAAE,aAAa,UAAU,CAAC;AACtD,SAAK,iBAAiB,eAAe,IAAI,GAAG,EAC1C,aAAa,UACd,CAAC;AAEF,WAAO,OAAO,EAAE,aAAa;KAC3B,WAAW;KACX,WAAW;KACZ,CAAC;IAEF,MAAM,mBAAmB,IAAI,iBAAiB;IAC9C,MAAM,QAA0B;KAC9B,SAAS;KACT,UAAU;KACV;KACD;AAED,UAAM,UAAU,GAAG,iBAAiB,OAAO,CACxC,KAAK,OAAO,WAAW;AACtB,WAAM,KAAK,IAAI,UAAU,QAAQ,QAAQ;AACzC,UAAK,SAAS,sBAAsB;MAClC,aAAa;MACb,aAAa,SAAS,OAAO,KAAK,OAAO,OAAO;MACjD,CAAC;AACF,YAAO;MACP,CACD,OAAO,UAAU;AAChB,UAAK,gBAAgB,MAAM;AAC3B,UAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAK9C,SAAI,iBAAiB,OAAO,QAC1B,OAAM;AAER,SAAI,iBAAiB,eAAe,iBAAiB,SACnD,OAAM;AAER,WAAM,eAAe,gBACnB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;MACD,CACD,cAAc;AACb,SAAI,KAAK,iBAAiB,IAAI,SAAS,KAAK,MAC1C,MAAK,iBAAiB,OAAO,SAAS;MAExC;AAKJ,UAAM,QAAQ,YAAY,GAAG;AAE7B,SAAK,iBAAiB,IAAI,UAAU,MAAgC;IAEpE,MAAM,SAAS,MAAM,KAAK,kBAAkB,OAAO,aAAa;AAChE,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAC9C,UAAM;aACE;AACR,SAAK,KAAK;;KAGd;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;;;;;;;;CAUH,AAAQ,kBACN,OACA,cACY;AACZ,MAAI,CAAC,aAAc,QAAO,MAAM;AAEhC,SAAO,IAAI,SAAY,SAAS,WAAW;GACzC,IAAI,UAAU;GAEd,MAAM,gBAAgB;AACpB,QAAI,MAAM,WAAW,EAAG,OAAM;AAC9B,QAAI,MAAM,YAAY,KAAK,CAAC,MAAM,iBAAiB,OAAO,QAGxD,OAAM,aAAa,iBAAiB;AAClC,SAAI,MAAM,YAAY,KAAK,CAAC,MAAM,iBAAiB,OAAO,QACxD,OAAM,iBAAiB,MACrB,aAAa,UAAU,4BACxB;OAEF,aAAa,sBAAsB;;GAI1C,MAAM,gBAAgB;AACpB,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,aAAS;AACT,WAAO,iBAAiB,aAAa,CAAC;;AAGxC,OAAI,aAAa,SAAS;AACxB,aAAS;AACT;;AAGF,gBAAa,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;AAE/D,SAAM,QAAQ,MACX,UAAU;AACT,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,YAAQ,MAAM;OAEf,UAAU;AACT,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,WAAO,MAAM;KAEhB;IACD;;;;;;;CAQJ,MAAM,IAAO,KAAgC;AAC3C,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO;AAGjC,OAAK,cAAc;AAGnB,UADc,MAAM,KAAK,SAAY,IAAI,GAC3B,SAAS;;;;;;;;;CAUzB,MAAc,SAAY,KAA4C;EACpE,MAAM,QAAQ,MAAM,KAAK,QAAQ,IAAO,IAAI;AAC5C,MAAI,CAAC,MAAO,QAAO;AAEnB,MAAI,KAAK,KAAK,GAAG,MAAM,QAAQ;AAC7B,SAAM,KAAK,QAAQ,OAAO,IAAI;AAC9B,UAAO;;AAET,SAAO;;;CAIT,AAAQ,eAAqB;AAC3B,MAAI,KAAK,kBAAmB;AAC5B,MAAI,CAAC,KAAK,QAAQ,cAAc,CAAE;EAClC,MAAM,MAAM,KAAK,KAAK;AACtB,MAAI,MAAM,KAAK,qBAAqB,aAAa,wBAC/C;EAEF,MAAM,cAAc,KAAK,OAAO,sBAAsB;AAEtD,MAAI,KAAK,QAAQ,GAAG,YAAa;AAEjC,OAAK,qBAAqB;AAE1B,OAAK,oBAAoB;AACzB,EAAC,KAAK,QACH,gBAAgB,CAChB,OAAO,UAAU;AAChB,UAAO,MAAM,yCAAyC,MAAM;IAC5D,CACD,cAAc;AACb,QAAK,oBAAoB;IACzB;;;;;;;;;CAUN,MAAM,IACJ,KACA,OACA,SACe;AACf,MAAI,CAAC,KAAK,OAAO,QAAS;EAE1B,MAAM,MAAM,SAAS,OAAO,KAAK,OAAO,OAAO;EAC/C,MAAM,aAAa,KAAK,KAAK,GAAG,MAAM;AACtC,QAAM,KAAK,QAAQ,IAAI,KAAK;GAAE;GAAO,QAAQ;GAAY,CAAC;;;;;;;CAQ5D,MAAM,OAAO,KAA4B;AACvC,MAAI,CAAC,KAAK,OAAO,QAAS;AAC1B,QAAM,KAAK,QAAQ,OAAO,IAAI;;;CAIhC,MAAM,QAAuB;AAC3B,QAAM,KAAK,QAAQ,OAAO;AAC1B,OAAK,MAAM,SAAS,KAAK,iBAAiB,QAAQ,CAChD,KAAI,MAAM,WAAY,cAAa,MAAM,WAAW;AAEtD,OAAK,iBAAiB,OAAO;;;;;;;CAQ/B,MAAM,IAAI,KAA+B;AACvC,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO;AAGjC,SADc,MAAM,KAAK,SAAS,IAAI,KACrB;;;;;;;;CASnB,YAAY,OAAqC,SAAyB;EACxE,MAAM,WAAW,CAAC,SAAS,GAAG,MAAM;EACpC,MAAM,aAAa,KAAK,UAAU,SAAS;AAC3C,SAAO,WAAW,SAAS,CAAC,OAAO,WAAW,CAAC,OAAO,MAAM;;;CAI9D,MAAM,QAAuB;AAC3B,QAAM,KAAK,QAAQ,OAAO;;;;;;CAO5B,MAAM,mBAAqC;AACzC,SAAO,KAAK,QAAQ,aAAa"}
+{"version":3,"file":"index.js","names":[],"sources":["../../src/cache/index.ts"],"sourcesContent":["import { createHash } from \"node:crypto\";\n\nimport type { CacheConfig, CacheEntry, CacheStorage } from \"shared\";\n\nimport { createLakebasePool } from \"../connectors/lakebase\";\nimport { getClientOptions } from \"../context/client-options\";\nimport { AppKitError, ExecutionError, InitializationError } from \"../errors\";\nimport { createLogger } from \"../logging/logger\";\nimport type { Counter, TelemetryProvider } from \"../telemetry\";\nimport { SpanStatusCode, TelemetryManager } from \"../telemetry\";\nimport { deepMerge } from \"../utils\";\nimport { ApiError, createWorkspaceClient } from \"../workspace-client\";\nimport { cacheDefaults } from \"./defaults\";\nimport { InMemoryStorage, PersistentStorage } from \"./storage\";\n\nconst logger = createLogger(\"cache\");\n\n/**\n * Reference-counted in-flight cache execution entry.\n *\n * `sharedController` decouples the cached `fn()` from any single caller's\n * abort signal. Callers join an in-flight entry by incrementing `refCount`;\n * when a caller aborts, refCount is decremented. The shared controller is\n * aborted only when refCount drops to 0 — i.e. all callers have abandoned\n * the request. This prevents one caller's cancellation (e.g. React\n * StrictMode unmount) from poisoning the in-flight result for other still-\n * connected awaiters.\n */\ninterface InFlightEntry<T> {\n  promise: Promise<T>;\n  refCount: number;\n  sharedController: AbortController;\n  abortTimer?: ReturnType<typeof setTimeout>;\n}\n\nfunction createAbortError(signal: AbortSignal): unknown {\n  if (signal.reason !== undefined) return signal.reason;\n  return new DOMException(\"The operation was aborted.\", \"AbortError\");\n}\n\n/**\n * Cache manager class to handle cache operations.\n * Can be used with in-memory storage or persistent storage (Lakebase).\n *\n * The cache is automatically initialized by AppKit. Use `getInstanceSync()` to access\n * the singleton instance after initialization.\n *\n * @internal\n * @example\n * ```typescript\n * const cache = CacheManager.getInstanceSync();\n * const result = await cache.getOrExecute([\"users\", userId], () => fetchUser(userId), userKey);\n * ```\n */\nexport class CacheManager {\n  private static readonly MIN_CLEANUP_INTERVAL_MS = 60_000;\n  private static readonly ABORT_GRACE_PERIOD_MS = 100;\n  private readonly name: string = \"cache-manager\";\n  private static instance: CacheManager | null = null;\n  private static initPromise: Promise<CacheManager> | null = null;\n\n  private storage: CacheStorage;\n  private config: CacheConfig;\n  private inFlightRequests: Map<string, InFlightEntry<unknown>>;\n  private cleanupInProgress: boolean;\n  private lastCleanupAttempt: number;\n\n  private telemetry: TelemetryProvider;\n  private telemetryMetrics: {\n    cacheHitCount: Counter;\n    cacheMissCount: Counter;\n  };\n\n  private constructor(storage: CacheStorage, config: CacheConfig) {\n    this.storage = storage;\n    this.config = config;\n    this.inFlightRequests = new Map();\n    this.cleanupInProgress = false;\n    this.lastCleanupAttempt = 0;\n\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      cacheHitCount: this.telemetry.getMeter().createCounter(\"cache.hit\", {\n        description: \"Total number of cache hits\",\n        unit: \"1\",\n      }),\n      cacheMissCount: this.telemetry.getMeter().createCounter(\"cache.miss\", {\n        description: \"Total number of cache misses\",\n        unit: \"1\",\n      }),\n    };\n  }\n\n  /**\n   * Get the singleton instance of the cache manager (sync version).\n   *\n   * Throws if not initialized - ensure AppKit.create() has completed first.\n   * @returns CacheManager instance\n   */\n  static getInstanceSync(): CacheManager {\n    if (!CacheManager.instance) {\n      throw InitializationError.notInitialized(\n        \"CacheManager\",\n        \"Ensure AppKit.create() has completed before accessing the cache\",\n      );\n    }\n\n    return CacheManager.instance;\n  }\n\n  /**\n   * Initialize and get the singleton instance of the cache manager.\n   * Called internally by AppKit - prefer `getInstanceSync()` for plugin access.\n   * @param userConfig - User configuration for the cache manager\n   * @returns CacheManager instance\n   * @internal\n   */\n  static async getInstance(\n    userConfig?: Partial<CacheConfig>,\n  ): Promise<CacheManager> {\n    if (CacheManager.instance) {\n      return CacheManager.instance;\n    }\n\n    if (!CacheManager.initPromise) {\n      CacheManager.initPromise = CacheManager.create(userConfig).then(\n        (instance) => {\n          // Publishes unconditionally: safe only because every getInstance() is\n          // awaited before any reset(), so a reset() can never land mid-init and\n          // this can never publish over it. A future unawaited-init caller would\n          // reintroduce that stale-publish race (the removed `generation` guard).\n          CacheManager.instance = instance;\n          return instance;\n        },\n      );\n    }\n\n    return CacheManager.initPromise;\n  }\n\n  /**\n   * Create a new cache manager instance\n   *\n   * Storage selection logic:\n   * 1. If `storage` provided and healthy → use provided storage\n   * 2. If `storage` provided but unhealthy → fallback to InMemory (or disable if strictPersistence)\n   * 3. If no `storage` provided and Lakebase available → use Lakebase\n   * 4. If no `storage` provided and Lakebase unavailable → fallback to InMemory (or disable if strictPersistence)\n   *\n   * @param userConfig - User configuration for the cache manager\n   * @returns CacheManager instance\n   */\n  private static async create(\n    userConfig?: Partial<CacheConfig>,\n  ): Promise<CacheManager> {\n    const config = deepMerge(cacheDefaults, userConfig);\n\n    if (config.storage) {\n      const isHealthy = await config.storage.healthCheck();\n      if (isHealthy) {\n        return new CacheManager(config.storage, config);\n      }\n\n      if (config.strictPersistence) {\n        const disabledConfig = { ...config, enabled: false };\n        return new CacheManager(\n          new InMemoryStorage(disabledConfig),\n          disabledConfig,\n        );\n      }\n\n      return new CacheManager(new InMemoryStorage(config), config);\n    }\n\n    // try to use lakebase storage\n    try {\n      const workspaceClient = createWorkspaceClient({\n        clientOptions: getClientOptions(),\n      }).toLegacyWorkspaceClient();\n      const pool = createLakebasePool({ workspaceClient });\n      const persistentStorage = new PersistentStorage(config, pool);\n\n      const isHealthy = await persistentStorage.healthCheck();\n      if (isHealthy) {\n        await persistentStorage.initialize();\n        return new CacheManager(persistentStorage, config);\n      }\n\n      // Health check failed, close the pool and fallback\n      await pool.end();\n    } catch {\n      // lakebase unavailable, continue with in-memory storage\n    }\n\n    if (config.strictPersistence) {\n      const disabledConfig = { ...config, enabled: false };\n      return new CacheManager(\n        new InMemoryStorage(disabledConfig),\n        disabledConfig,\n      );\n    }\n\n    return new CacheManager(new InMemoryStorage(config), config);\n  }\n\n  /**\n   * Get or execute a function and cache the result.\n   *\n   * Multiple concurrent callers with the same `cacheKey` are deduplicated\n   * onto a single in-flight execution. Each caller may pass its own\n   * `callerSignal`; the underlying `fn()` is run with a shared, internally\n   * managed `AbortSignal` that aborts only when *all* callers have\n   * abandoned the request (reference counted). This decouples a single\n   * caller's cancellation (e.g. React StrictMode unmount) from the shared\n   * result, so other still-connected callers receive the cached value\n   * normally.\n   *\n   * @param key - Cache key parts\n   * @param fn - Function to execute. Receives the cache-owned shared signal;\n   *   pass it through to the underlying I/O so the work is cancelled when\n   *   no caller is left waiting.\n   * @param userKey - User key for cache namespacing\n   * @param options - Options for the cache\n   * @returns Promise of the result\n   */\n  async getOrExecute<T>(\n    key: (string | number | object)[],\n    fn: (sharedSignal?: AbortSignal) => Promise<T>,\n    userKey: string,\n    options?: { ttl?: number; callerSignal?: AbortSignal },\n  ): Promise<T> {\n    if (!this.config.enabled) return fn(options?.callerSignal);\n\n    const callerSignal = options?.callerSignal;\n    if (callerSignal?.aborted) {\n      throw createAbortError(callerSignal);\n    }\n\n    const cacheKey = this.generateKey(key, userKey);\n\n    return this.telemetry.startActiveSpan(\n      \"cache.getOrExecute\",\n      {\n        attributes: {\n          \"cache.key\": cacheKey,\n          \"cache.enabled\": this.config.enabled,\n          \"cache.persistent\": this.storage.isPersistent(),\n        },\n      },\n      async (span) => {\n        try {\n          const cached = await this.getValid<T>(cacheKey);\n          if (cached !== null) {\n            span.setAttribute(\"cache.hit\", true);\n            span.setStatus({ code: SpanStatusCode.OK });\n            this.telemetryMetrics.cacheHitCount.add(1, {\n              \"cache.key\": cacheKey,\n            });\n\n            logger.event()?.setExecution({\n              cache_hit: true,\n              cache_key: cacheKey,\n            });\n\n            return cached.value;\n          }\n\n          // check if the value is being processed by another request — join\n          // the existing in-flight entry under reference counting so this\n          // caller's abort doesn't poison the shared result.\n          const existing = this.inFlightRequests.get(cacheKey) as\n            | InFlightEntry<T>\n            | undefined;\n          if (existing && !existing.sharedController.signal.aborted) {\n            existing.refCount++;\n            // Cancel any pending abort timer — a new caller has joined\n            if (existing.abortTimer) {\n              clearTimeout(existing.abortTimer);\n              existing.abortTimer = undefined;\n            }\n            span.setAttribute(\"cache.hit\", true);\n            span.setAttribute(\"cache.deduplication\", true);\n            span.addEvent(\"cache.deduplication_used\", {\n              \"cache.key\": cacheKey,\n            });\n            this.telemetryMetrics.cacheHitCount.add(1, {\n              \"cache.key\": cacheKey,\n              \"cache.deduplication\": \"true\",\n            });\n\n            logger.event()?.setExecution({\n              cache_hit: true,\n              cache_key: cacheKey,\n              cache_deduplication: true,\n            });\n\n            return await this._waitWithRefCount(existing, callerSignal);\n          }\n\n          // cache miss - execute function under a shared abort controller\n          span.setAttribute(\"cache.hit\", false);\n          span.addEvent(\"cache.miss\", { \"cache.key\": cacheKey });\n          this.telemetryMetrics.cacheMissCount.add(1, {\n            \"cache.key\": cacheKey,\n          });\n\n          logger.event()?.setExecution({\n            cache_hit: false,\n            cache_key: cacheKey,\n          });\n\n          const sharedController = new AbortController();\n          const entry: InFlightEntry<T> = {\n            promise: undefined as unknown as Promise<T>,\n            refCount: 1,\n            sharedController,\n          };\n\n          entry.promise = fn(sharedController.signal)\n            .then(async (result) => {\n              await this.set(cacheKey, result, options);\n              span.addEvent(\"cache.value_stored\", {\n                \"cache.key\": cacheKey,\n                \"cache.ttl\": options?.ttl ?? this.config.ttl ?? 3600,\n              });\n              return result;\n            })\n            .catch((error) => {\n              span.recordException(error);\n              span.setStatus({ code: SpanStatusCode.ERROR });\n              // If the shared controller aborted, all callers have already\n              // abandoned the request (or are about to via their own signals)\n              // — propagate the original error without wrapping. No live\n              // awaiter will observe this rejection.\n              if (sharedController.signal.aborted) {\n                throw error;\n              }\n              if (error instanceof AppKitError || error instanceof ApiError) {\n                throw error;\n              }\n              throw ExecutionError.statementFailed(\n                error instanceof Error ? error.message : String(error),\n              );\n            })\n            .finally(() => {\n              if (this.inFlightRequests.get(cacheKey) === entry) {\n                this.inFlightRequests.delete(cacheKey);\n              }\n            });\n\n          // Suppress unhandled rejection warnings when every caller bailed\n          // before fn() resolved (their own promises rejected via\n          // waitWithRefCount; the underlying entry.promise has no awaiter).\n          entry.promise.catch(() => {});\n\n          this.inFlightRequests.set(cacheKey, entry as InFlightEntry<unknown>);\n\n          const result = await this._waitWithRefCount(entry, callerSignal);\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          span.recordException(error as Error);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n          throw error;\n        } finally {\n          span.end();\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  /**\n   * Wait on an in-flight entry, racing the underlying promise against the\n   * caller's abort signal. When the caller aborts, the entry's refCount is\n   * decremented; if it hits zero the shared controller is aborted so the\n   * underlying `fn()` can stop. Other callers continue to await the same\n   * entry and receive the result when it arrives.\n   */\n  private _waitWithRefCount<T>(\n    entry: InFlightEntry<T>,\n    callerSignal?: AbortSignal,\n  ): Promise<T> {\n    if (!callerSignal) return entry.promise;\n\n    return new Promise<T>((resolve, reject) => {\n      let settled = false;\n\n      const release = () => {\n        if (entry.refCount > 0) entry.refCount--;\n        if (entry.refCount <= 0 && !entry.sharedController.signal.aborted) {\n          // Grace period: delay abort so a StrictMode remount can join\n          // the in-flight entry before the shared execution is cancelled.\n          entry.abortTimer = setTimeout(() => {\n            if (entry.refCount <= 0 && !entry.sharedController.signal.aborted) {\n              entry.sharedController.abort(\n                callerSignal.reason ?? \"all cache callers aborted\",\n              );\n            }\n          }, CacheManager.ABORT_GRACE_PERIOD_MS);\n        }\n      };\n\n      const onAbort = () => {\n        if (settled) return;\n        settled = true;\n        callerSignal.removeEventListener(\"abort\", onAbort);\n        release();\n        reject(createAbortError(callerSignal));\n      };\n\n      if (callerSignal.aborted) {\n        onAbort();\n        return;\n      }\n\n      callerSignal.addEventListener(\"abort\", onAbort, { once: true });\n\n      entry.promise.then(\n        (value) => {\n          if (settled) return;\n          settled = true;\n          callerSignal.removeEventListener(\"abort\", onAbort);\n          resolve(value);\n        },\n        (error) => {\n          if (settled) return;\n          settled = true;\n          callerSignal.removeEventListener(\"abort\", onAbort);\n          reject(error);\n        },\n      );\n    });\n  }\n\n  /**\n   * Get a cached value\n   * @param key - Cache key\n   * @returns Promise of the value or null if not found or expired\n   */\n  async get<T>(key: string): Promise<T | null> {\n    if (!this.config.enabled) return null;\n\n    // probabilistic cleanup trigger\n    this.maybeCleanup();\n\n    const entry = await this.getValid<T>(key);\n    return entry?.value ?? null;\n  }\n\n  /**\n   * Get a cached entry only if it has not expired.\n   * Returns null on miss or expired (and deletes the expired entry).\n   *\n   * Storage implementations return entries unconditionally — expiry handling\n   * lives at the CacheManager layer.\n   */\n  private async getValid<T>(key: string): Promise<CacheEntry<T> | null> {\n    const entry = await this.storage.get<T>(key);\n    if (!entry) return null;\n\n    if (Date.now() > entry.expiry) {\n      await this.storage.delete(key);\n      return null;\n    }\n    return entry;\n  }\n\n  /** Probabilistically trigger cleanup of expired entries (fire-and-forget) */\n  private maybeCleanup(): void {\n    if (this.cleanupInProgress) return;\n    if (!this.storage.isPersistent()) return;\n    const now = Date.now();\n    if (now - this.lastCleanupAttempt < CacheManager.MIN_CLEANUP_INTERVAL_MS)\n      return;\n\n    const probability = this.config.cleanupProbability ?? 0.01;\n\n    if (Math.random() > probability) return;\n\n    this.lastCleanupAttempt = now;\n\n    this.cleanupInProgress = true;\n    (this.storage as PersistentStorage)\n      .cleanupExpired()\n      .catch((error) => {\n        logger.debug(\"Error cleaning up expired entries: %O\", error);\n      })\n      .finally(() => {\n        this.cleanupInProgress = false;\n      });\n  }\n\n  /**\n   * Set a value in the cache\n   * @param key - Cache key\n   * @param value - Value to set\n   * @param options - Options for the cache\n   * @returns Promise of the result\n   */\n  async set<T>(\n    key: string,\n    value: T,\n    options?: { ttl?: number },\n  ): Promise<void> {\n    if (!this.config.enabled) return;\n\n    const ttl = options?.ttl ?? this.config.ttl ?? 3600;\n    const expiryTime = Date.now() + ttl * 1000;\n    await this.storage.set(key, { value, expiry: expiryTime });\n  }\n\n  /**\n   * Delete a value from the cache\n   * @param key - Cache key\n   * @returns Promise of the result\n   */\n  async delete(key: string): Promise<void> {\n    if (!this.config.enabled) return;\n    await this.storage.delete(key);\n  }\n\n  /** Clear the cache */\n  async clear(): Promise<void> {\n    await this.storage.clear();\n    for (const entry of this.inFlightRequests.values()) {\n      if (entry.abortTimer) clearTimeout(entry.abortTimer);\n    }\n    this.inFlightRequests.clear();\n  }\n\n  /**\n   * Check if a value exists in the cache\n   * @param key - Cache key\n   * @returns Promise of true if the value exists, false otherwise\n   */\n  async has(key: string): Promise<boolean> {\n    if (!this.config.enabled) return false;\n\n    const entry = await this.getValid(key);\n    return entry !== null;\n  }\n\n  /**\n   * Generate a cache key\n   * @param parts - Parts of the key\n   * @param userKey - User key\n   * @returns Cache key\n   */\n  generateKey(parts: (string | number | object)[], userKey: string): string {\n    const allParts = [userKey, ...parts];\n    const serialized = JSON.stringify(allParts);\n    return createHash(\"sha256\").update(serialized).digest(\"hex\");\n  }\n\n  /** Close the cache */\n  async close(): Promise<void> {\n    await this.storage.close();\n  }\n\n  /**\n   * Drop the singleton so the next {@link getInstance} builds a fresh manager.\n   *\n   * Both fields must clear — `getInstance()` falls back to `initPromise` when\n   * `instance` is null. A pointer drop, not teardown: call {@link close} first\n   * or the old storage leaks (a `pg.Pool` under `PersistentStorage`).\n   *\n   * @internal\n   */\n  static reset(): void {\n    CacheManager.instance = null;\n    CacheManager.initPromise = null;\n  }\n\n  /**\n   * Check if the storage is healthy\n   * @returns Promise of true if the storage is healthy, false otherwise\n   */\n  async isStorageHealthy(): Promise<boolean> {\n    return this.storage.healthCheck();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAeA,MAAM,SAAS,aAAa,QAAQ;AAoBpC,SAAS,iBAAiB,QAA8B;AACtD,KAAI,OAAO,WAAW,OAAW,QAAO,OAAO;AAC/C,QAAO,IAAI,aAAa,8BAA8B,aAAa;;;;;;;;;;;;;;;;AAiBrE,IAAa,eAAb,MAAa,aAAa;CACxB,OAAwB,0BAA0B;CAClD,OAAwB,wBAAwB;CAChD,AAAiB,OAAe;CAChC,OAAe,WAAgC;CAC/C,OAAe,cAA4C;CAE3D,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,AAAQ;CACR,AAAQ;CAKR,AAAQ,YAAY,SAAuB,QAAqB;AAC9D,OAAK,UAAU;AACf,OAAK,SAAS;AACd,OAAK,mCAAmB,IAAI,KAAK;AACjC,OAAK,oBAAoB;AACzB,OAAK,qBAAqB;AAE1B,OAAK,YAAY,iBAAiB,YAChC,KAAK,MACL,KAAK,OAAO,UACb;AACD,OAAK,mBAAmB;GACtB,eAAe,KAAK,UAAU,UAAU,CAAC,cAAc,aAAa;IAClE,aAAa;IACb,MAAM;IACP,CAAC;GACF,gBAAgB,KAAK,UAAU,UAAU,CAAC,cAAc,cAAc;IACpE,aAAa;IACb,MAAM;IACP,CAAC;GACH;;;;;;;;CASH,OAAO,kBAAgC;AACrC,MAAI,CAAC,aAAa,SAChB,OAAM,oBAAoB,eACxB,gBACA,kEACD;AAGH,SAAO,aAAa;;;;;;;;;CAUtB,aAAa,YACX,YACuB;AACvB,MAAI,aAAa,SACf,QAAO,aAAa;AAGtB,MAAI,CAAC,aAAa,YAChB,cAAa,cAAc,aAAa,OAAO,WAAW,CAAC,MACxD,aAAa;AAKZ,gBAAa,WAAW;AACxB,UAAO;IAEV;AAGH,SAAO,aAAa;;;;;;;;;;;;;;CAetB,aAAqB,OACnB,YACuB;EACvB,MAAM,SAAS,UAAU,eAAe,WAAW;AAEnD,MAAI,OAAO,SAAS;AAElB,OADkB,MAAM,OAAO,QAAQ,aAAa,CAElD,QAAO,IAAI,aAAa,OAAO,SAAS,OAAO;AAGjD,OAAI,OAAO,mBAAmB;IAC5B,MAAM,iBAAiB;KAAE,GAAG;KAAQ,SAAS;KAAO;AACpD,WAAO,IAAI,aACT,IAAI,gBAAgB,eAAe,EACnC,eACD;;AAGH,UAAO,IAAI,aAAa,IAAI,gBAAgB,OAAO,EAAE,OAAO;;AAI9D,MAAI;GAIF,MAAM,OAAO,mBAAmB,EAAE,iBAHV,sBAAsB,EAC5C,eAAe,kBAAkB,EAClC,CAAC,CAAC,yBAAyB,EACuB,CAAC;GACpD,MAAM,oBAAoB,IAAI,kBAAkB,QAAQ,KAAK;AAG7D,OADkB,MAAM,kBAAkB,aAAa,EACxC;AACb,UAAM,kBAAkB,YAAY;AACpC,WAAO,IAAI,aAAa,mBAAmB,OAAO;;AAIpD,SAAM,KAAK,KAAK;UACV;AAIR,MAAI,OAAO,mBAAmB;GAC5B,MAAM,iBAAiB;IAAE,GAAG;IAAQ,SAAS;IAAO;AACpD,UAAO,IAAI,aACT,IAAI,gBAAgB,eAAe,EACnC,eACD;;AAGH,SAAO,IAAI,aAAa,IAAI,gBAAgB,OAAO,EAAE,OAAO;;;;;;;;;;;;;;;;;;;;;;CAuB9D,MAAM,aACJ,KACA,IACA,SACA,SACY;AACZ,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO,GAAG,SAAS,aAAa;EAE1D,MAAM,eAAe,SAAS;AAC9B,MAAI,cAAc,QAChB,OAAM,iBAAiB,aAAa;EAGtC,MAAM,WAAW,KAAK,YAAY,KAAK,QAAQ;AAE/C,SAAO,KAAK,UAAU,gBACpB,sBACA,EACE,YAAY;GACV,aAAa;GACb,iBAAiB,KAAK,OAAO;GAC7B,oBAAoB,KAAK,QAAQ,cAAc;GAChD,EACF,EACD,OAAO,SAAS;AACd,OAAI;IACF,MAAM,SAAS,MAAM,KAAK,SAAY,SAAS;AAC/C,QAAI,WAAW,MAAM;AACnB,UAAK,aAAa,aAAa,KAAK;AACpC,UAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,UAAK,iBAAiB,cAAc,IAAI,GAAG,EACzC,aAAa,UACd,CAAC;AAEF,YAAO,OAAO,EAAE,aAAa;MAC3B,WAAW;MACX,WAAW;MACZ,CAAC;AAEF,YAAO,OAAO;;IAMhB,MAAM,WAAW,KAAK,iBAAiB,IAAI,SAAS;AAGpD,QAAI,YAAY,CAAC,SAAS,iBAAiB,OAAO,SAAS;AACzD,cAAS;AAET,SAAI,SAAS,YAAY;AACvB,mBAAa,SAAS,WAAW;AACjC,eAAS,aAAa;;AAExB,UAAK,aAAa,aAAa,KAAK;AACpC,UAAK,aAAa,uBAAuB,KAAK;AAC9C,UAAK,SAAS,4BAA4B,EACxC,aAAa,UACd,CAAC;AACF,UAAK,iBAAiB,cAAc,IAAI,GAAG;MACzC,aAAa;MACb,uBAAuB;MACxB,CAAC;AAEF,YAAO,OAAO,EAAE,aAAa;MAC3B,WAAW;MACX,WAAW;MACX,qBAAqB;MACtB,CAAC;AAEF,YAAO,MAAM,KAAK,kBAAkB,UAAU,aAAa;;AAI7D,SAAK,aAAa,aAAa,MAAM;AACrC,SAAK,SAAS,cAAc,EAAE,aAAa,UAAU,CAAC;AACtD,SAAK,iBAAiB,eAAe,IAAI,GAAG,EAC1C,aAAa,UACd,CAAC;AAEF,WAAO,OAAO,EAAE,aAAa;KAC3B,WAAW;KACX,WAAW;KACZ,CAAC;IAEF,MAAM,mBAAmB,IAAI,iBAAiB;IAC9C,MAAM,QAA0B;KAC9B,SAAS;KACT,UAAU;KACV;KACD;AAED,UAAM,UAAU,GAAG,iBAAiB,OAAO,CACxC,KAAK,OAAO,WAAW;AACtB,WAAM,KAAK,IAAI,UAAU,QAAQ,QAAQ;AACzC,UAAK,SAAS,sBAAsB;MAClC,aAAa;MACb,aAAa,SAAS,OAAO,KAAK,OAAO,OAAO;MACjD,CAAC;AACF,YAAO;MACP,CACD,OAAO,UAAU;AAChB,UAAK,gBAAgB,MAAM;AAC3B,UAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAK9C,SAAI,iBAAiB,OAAO,QAC1B,OAAM;AAER,SAAI,iBAAiB,eAAe,iBAAiB,SACnD,OAAM;AAER,WAAM,eAAe,gBACnB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;MACD,CACD,cAAc;AACb,SAAI,KAAK,iBAAiB,IAAI,SAAS,KAAK,MAC1C,MAAK,iBAAiB,OAAO,SAAS;MAExC;AAKJ,UAAM,QAAQ,YAAY,GAAG;AAE7B,SAAK,iBAAiB,IAAI,UAAU,MAAgC;IAEpE,MAAM,SAAS,MAAM,KAAK,kBAAkB,OAAO,aAAa;AAChE,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAC9C,UAAM;aACE;AACR,SAAK,KAAK;;KAGd;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;;;;;;;;CAUH,AAAQ,kBACN,OACA,cACY;AACZ,MAAI,CAAC,aAAc,QAAO,MAAM;AAEhC,SAAO,IAAI,SAAY,SAAS,WAAW;GACzC,IAAI,UAAU;GAEd,MAAM,gBAAgB;AACpB,QAAI,MAAM,WAAW,EAAG,OAAM;AAC9B,QAAI,MAAM,YAAY,KAAK,CAAC,MAAM,iBAAiB,OAAO,QAGxD,OAAM,aAAa,iBAAiB;AAClC,SAAI,MAAM,YAAY,KAAK,CAAC,MAAM,iBAAiB,OAAO,QACxD,OAAM,iBAAiB,MACrB,aAAa,UAAU,4BACxB;OAEF,aAAa,sBAAsB;;GAI1C,MAAM,gBAAgB;AACpB,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,aAAS;AACT,WAAO,iBAAiB,aAAa,CAAC;;AAGxC,OAAI,aAAa,SAAS;AACxB,aAAS;AACT;;AAGF,gBAAa,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;AAE/D,SAAM,QAAQ,MACX,UAAU;AACT,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,YAAQ,MAAM;OAEf,UAAU;AACT,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,WAAO,MAAM;KAEhB;IACD;;;;;;;CAQJ,MAAM,IAAO,KAAgC;AAC3C,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO;AAGjC,OAAK,cAAc;AAGnB,UADc,MAAM,KAAK,SAAY,IAAI,GAC3B,SAAS;;;;;;;;;CAUzB,MAAc,SAAY,KAA4C;EACpE,MAAM,QAAQ,MAAM,KAAK,QAAQ,IAAO,IAAI;AAC5C,MAAI,CAAC,MAAO,QAAO;AAEnB,MAAI,KAAK,KAAK,GAAG,MAAM,QAAQ;AAC7B,SAAM,KAAK,QAAQ,OAAO,IAAI;AAC9B,UAAO;;AAET,SAAO;;;CAIT,AAAQ,eAAqB;AAC3B,MAAI,KAAK,kBAAmB;AAC5B,MAAI,CAAC,KAAK,QAAQ,cAAc,CAAE;EAClC,MAAM,MAAM,KAAK,KAAK;AACtB,MAAI,MAAM,KAAK,qBAAqB,aAAa,wBAC/C;EAEF,MAAM,cAAc,KAAK,OAAO,sBAAsB;AAEtD,MAAI,KAAK,QAAQ,GAAG,YAAa;AAEjC,OAAK,qBAAqB;AAE1B,OAAK,oBAAoB;AACzB,EAAC,KAAK,QACH,gBAAgB,CAChB,OAAO,UAAU;AAChB,UAAO,MAAM,yCAAyC,MAAM;IAC5D,CACD,cAAc;AACb,QAAK,oBAAoB;IACzB;;;;;;;;;CAUN,MAAM,IACJ,KACA,OACA,SACe;AACf,MAAI,CAAC,KAAK,OAAO,QAAS;EAE1B,MAAM,MAAM,SAAS,OAAO,KAAK,OAAO,OAAO;EAC/C,MAAM,aAAa,KAAK,KAAK,GAAG,MAAM;AACtC,QAAM,KAAK,QAAQ,IAAI,KAAK;GAAE;GAAO,QAAQ;GAAY,CAAC;;;;;;;CAQ5D,MAAM,OAAO,KAA4B;AACvC,MAAI,CAAC,KAAK,OAAO,QAAS;AAC1B,QAAM,KAAK,QAAQ,OAAO,IAAI;;;CAIhC,MAAM,QAAuB;AAC3B,QAAM,KAAK,QAAQ,OAAO;AAC1B,OAAK,MAAM,SAAS,KAAK,iBAAiB,QAAQ,CAChD,KAAI,MAAM,WAAY,cAAa,MAAM,WAAW;AAEtD,OAAK,iBAAiB,OAAO;;;;;;;CAQ/B,MAAM,IAAI,KAA+B;AACvC,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO;AAGjC,SADc,MAAM,KAAK,SAAS,IAAI,KACrB;;;;;;;;CASnB,YAAY,OAAqC,SAAyB;EACxE,MAAM,WAAW,CAAC,SAAS,GAAG,MAAM;EACpC,MAAM,aAAa,KAAK,UAAU,SAAS;AAC3C,SAAO,WAAW,SAAS,CAAC,OAAO,WAAW,CAAC,OAAO,MAAM;;;CAI9D,MAAM,QAAuB;AAC3B,QAAM,KAAK,QAAQ,OAAO;;;;;;;;;;;CAY5B,OAAO,QAAc;AACnB,eAAa,WAAW;AACxB,eAAa,cAAc;;;;;;CAO7B,MAAM,mBAAqC;AACzC,SAAO,KAAK,QAAQ,aAAa"}

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

@@ -352,7 +352,19 @@ var SQLWarehouseConnector = class {
 		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));
+			let info;
+			try {
+				info = await workspaceClient.warehouses.get({ id: warehouseId }, this._createContext(signal));
+			} catch (error) {
+				if (signal?.aborted || error instanceof Error && error.name === "AbortError") throw ExecutionError.canceled();
+				span.addEvent("warehouse.status_probe_failed", {
+					"db.warehouse_id": warehouseId,
+					"error.message": error instanceof Error ? error.message : String(error)
+				});
+				logger.debug("Warehouse status probe failed for %s; skipping readiness gate and proceeding: %O", warehouseId, error);
+				span.setAttribute("db.warehouse.status_probe_failed", true);
+				return;
+			}
 			const state = info?.state;
 			const summary = info?.health?.summary;
 			switch (state) {

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

@@ -1 +1 @@
-{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/sql-warehouse/client.ts"],"sourcesContent":["import type { TelemetryOptions } from \"shared\";\n\nimport {\n  AppKitError,\n  ConfigurationError,\n  ConnectionError,\n  ExecutionError,\n  ValidationError,\n} from \"../../errors\";\nimport { createLogger } from \"../../logging/logger\";\nimport {\n  ArrowStreamProcessor,\n  type RefreshChunkLink,\n} from \"../../stream/arrow-stream-processor\";\nimport type { TelemetryProvider } from \"../../telemetry\";\nimport {\n  type Counter,\n  type Histogram,\n  type Span,\n  SpanKind,\n  SpanStatusCode,\n  TelemetryManager,\n} from \"../../telemetry\";\nimport {\n  Context,\n  type sql,\n  type WorkspaceClient,\n} from \"../../workspace-client\";\nimport { buildEmptyArrowIPCBase64 } from \"./arrow-schema\";\nimport { executeStatementDefaults } from \"./defaults\";\nimport { WarehousePollBackoff } from \"./warehouse-poll-backoff\";\nimport { WarehouseStatusEmitter } from \"./warehouse-status-emitter\";\n\nconst logger = createLogger(\"connectors:sql-warehouse\");\n\n/**\n * Real column names from a statement's result manifest (positional\n * `column_N` fallback for any blank ones). Databricks encodes ARROW_STREAM\n * schemas positionally (`col_0`, …); these names let consumers relabel an\n * Arrow result to match the JSON path. Returns `undefined` when the manifest\n * carries no columns.\n */\nfunction arrowColumnNames(\n  response: sql.StatementResponse,\n): string[] | undefined {\n  const cols = response.manifest?.schema?.columns;\n  if (!cols || cols.length === 0) return undefined;\n  return cols.map((c, i) =>\n    c.name && c.name.length > 0 ? c.name : `column_${i}`,\n  );\n}\n\n/**\n * Maximum size for inline Arrow IPC attachments (25 MiB decoded — the\n * Databricks Statement Execution API hard cap on INLINE responses).\n *\n * The analytics route streams the decoded attachment straight back on the\n * query response body; this cap bounds the single in-memory copy. Larger\n * results fall through to `disposition: \"EXTERNAL_LINKS\"`, which streams\n * chunk-by-chunk and is not bound by this limit.\n */\nconst MAX_INLINE_ATTACHMENT_BYTES = 25 * 1024 * 1024;\n\n/**\n * Safety cap on how many additional EXTERNAL_LINKS chunks\n * {@link SQLWarehouseConnector._resolveAllExternalLinks} will follow when the\n * manifest omits `total_chunk_count`. High enough to cover any real result;\n * only bounds a misbehaving warehouse with a cyclic `next_chunk_index`.\n */\nconst MAX_EXTERNAL_CHUNK_FOLLOWS = 10_000;\n\ninterface SQLWarehouseConfig {\n  timeout?: number;\n  telemetry?: TelemetryOptions;\n}\n\n/**\n * Default ceiling for how long {@link SQLWarehouseConnector.ensureWarehouseRunning}\n * will wait for a warehouse to reach the RUNNING state before giving up.\n *\n * Five minutes covers a cold-start of a classic warehouse on most workspaces;\n * serverless typically reaches RUNNING within ~30s.\n */\nexport const DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS = 5 * 60 * 1000;\n\n/**\n * Window during which a recent `RUNNING` observation lets subsequent calls\n * to {@link SQLWarehouseConnector.ensureWarehouseRunning} short-circuit\n * without making any SDK calls. Sized to roughly the upper bound of how\n * long Databricks keeps a warehouse \"stickily\" available between requests\n * — past that, we re-verify.\n */\nconst WAREHOUSE_RUNNING_CACHE_TTL_MS = 30_000;\n\n/**\n * A single observation of the warehouse state, emitted by\n * {@link SQLWarehouseConnector.ensureWarehouseRunning} so callers can stream\n * progress to clients (e.g. over SSE).\n *\n * Note: `health.summary` from the SDK is intentionally NOT exposed on this\n * type. It's free-form operator-oriented diagnostic text (cluster IDs,\n * capacity-failure reasons, internal RPC errors) that must not reach end\n * users. The raw value stays in the OTel span attributes for server-side\n * debugging only.\n */\nexport interface WarehouseStatusUpdate {\n  /** Current state from the SDK (RUNNING | STARTING | STOPPED | STOPPING | DELETED | DELETING). */\n  state: sql.State;\n  /** Milliseconds elapsed since `ensureWarehouseRunning` was called. */\n  elapsedMs: number;\n  /** 1-based attempt counter — useful for tests and telemetry. */\n  attempt: number;\n}\n\n/** Options for {@link SQLWarehouseConnector.ensureWarehouseRunning}. */\ninterface EnsureWarehouseRunningOptions {\n  /** Invoked every time the warehouse state changes during the wait. */\n  onStatus: (update: WarehouseStatusUpdate) => void;\n  /** Aborts the wait. The connector treats abort as `ExecutionError.canceled()`. */\n  signal?: AbortSignal;\n  /** Hard ceiling on the total wait. Defaults to {@link DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS}. */\n  timeoutMs?: number;\n  /**\n   * When `true` (default), a `STOPPED` warehouse is auto-started.\n   * When `false`, `STOPPED` surfaces as `ConfigurationError` immediately —\n   * useful for cost-controlled deployments that don't want any caller to\n   * trigger billable warehouse starts.\n   */\n  autoStart?: boolean;\n  /** @internal Fired once when `warehouses.start` is issued for this poll. */\n  onWarehouseStartIssued?: () => void;\n}\n\n/** In-flight warehouse readiness work shared across concurrent callers. */\ninterface WarehouseReadinessInFlight {\n  promise: Promise<void>;\n  refCount: number;\n  sharedController: AbortController;\n  subscribers: Set<(update: WarehouseStatusUpdate) => void>;\n  /** Last emitted update; replayed to late joiners. */\n  lastUpdate: WarehouseStatusUpdate | null;\n  /** `true` once this poll has called `warehouses.start`. */\n  warehouseStartIssued: boolean;\n}\n\nexport class SQLWarehouseConnector {\n  private readonly name = \"sql-warehouse\";\n\n  private config: SQLWarehouseConfig;\n\n  // Lazy-initialized: only created when Arrow format is used\n  private _arrowProcessor: ArrowStreamProcessor | null = null;\n\n  /**\n   * Per-warehouse cache of the last RUNNING observation timestamp. Used by\n   * {@link ensureWarehouseRunning} to short-circuit warm-path callers; see\n   * {@link WAREHOUSE_RUNNING_CACHE_TTL_MS}.\n   */\n  private _recentlyRunning = new Map<string, number>();\n  /** Per-warehouse readiness singleflight — one poll loop per cold start. */\n  private _readinessInFlight = new Map<string, WarehouseReadinessInFlight>();\n  // telemetry\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    queryCount: Counter;\n    queryDuration: Histogram;\n  };\n\n  constructor(config: SQLWarehouseConfig) {\n    this.config = config;\n\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      queryCount: this.telemetry.getMeter().createCounter(\"query.count\", {\n        description: \"Total number of queries executed\",\n        unit: \"1\",\n      }),\n      queryDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"query.duration\", {\n          description: \"Duration of queries executed\",\n          unit: \"ms\",\n        }),\n    };\n  }\n\n  /**\n   * Lazily initializes and returns the ArrowStreamProcessor.\n   * Only created on first Arrow format query to avoid unnecessary allocation.\n   */\n  private get arrowProcessor(): ArrowStreamProcessor {\n    if (!this._arrowProcessor) {\n      this._arrowProcessor = new ArrowStreamProcessor({\n        timeout: this.config.timeout || executeStatementDefaults.timeout,\n        retries: ArrowStreamProcessor.DEFAULT_RETRIES,\n      });\n    }\n    return this._arrowProcessor;\n  }\n\n  async executeStatement(\n    workspaceClient: WorkspaceClient,\n    input: sql.ExecuteStatementRequest,\n    signal?: AbortSignal,\n  ) {\n    const startTime = Date.now();\n    let success = false;\n\n    // if signal is aborted, throw an error\n    if (signal?.aborted) {\n      throw ExecutionError.canceled();\n    }\n\n    return this.telemetry.startActiveSpan(\n      \"sql.query\",\n      {\n        kind: SpanKind.CLIENT,\n        attributes: {\n          \"db.system\": \"databricks\",\n          \"db.warehouse_id\": input.warehouse_id || \"\",\n          \"db.catalog\": input.catalog ?? \"\",\n          \"db.schema\": input.schema ?? \"\",\n          \"db.statement\": input.statement?.substring(0, 500) || \"\",\n          \"db.has_parameters\": !!input.parameters,\n        },\n      },\n      async (span: Span) => {\n        let abortHandler: (() => void) | undefined;\n        let isAborted = false;\n\n        if (signal) {\n          abortHandler = () => {\n            // abort span if not recording\n            if (!span.isRecording()) return;\n            isAborted = true;\n            span.setAttribute(\"cancelled\", true);\n            span.setStatus({\n              code: SpanStatusCode.ERROR,\n              message: \"Query cancelled by client\",\n            });\n            span.end();\n          };\n          signal.addEventListener(\"abort\", abortHandler, { once: true });\n        }\n\n        try {\n          // validate required fields\n          if (!input.statement) {\n            throw ValidationError.missingField(\"statement\");\n          }\n\n          if (!input.warehouse_id) {\n            throw ValidationError.missingField(\"warehouse_id\");\n          }\n\n          const body: sql.ExecuteStatementRequest = {\n            statement: input.statement,\n            parameters: input.parameters,\n            warehouse_id: input.warehouse_id,\n            catalog: input.catalog,\n            schema: input.schema,\n            wait_timeout:\n              input.wait_timeout || executeStatementDefaults.wait_timeout,\n            disposition:\n              input.disposition || executeStatementDefaults.disposition,\n            format: input.format || executeStatementDefaults.format,\n            byte_limit: input.byte_limit,\n            row_limit: input.row_limit,\n            on_wait_timeout:\n              input.on_wait_timeout || executeStatementDefaults.on_wait_timeout,\n          };\n\n          span.addEvent(\"statement.submitting\", {\n            \"db.warehouse_id\": input.warehouse_id,\n          });\n\n          const response =\n            await workspaceClient.statementExecution.executeStatement(\n              body,\n              this._createContext(signal),\n            );\n\n          if (!response) {\n            throw ConnectionError.apiFailure(\"SQL Warehouse\");\n          }\n          const status = response.status;\n          const statementId = response.statement_id as string;\n\n          span.setAttribute(\"db.statement_id\", statementId);\n          span.addEvent(\"statement.submitted\", {\n            \"db.statement_id\": response.statement_id,\n            \"db.status\": status?.state,\n          });\n\n          let result:\n            | sql.StatementResponse\n            | { result: { statement_id: string; status: sql.StatementStatus } };\n\n          switch (status?.state) {\n            case \"RUNNING\":\n            case \"PENDING\":\n              span.addEvent(\"statement.polling_started\", {\n                \"db.status\": response.status?.state,\n              });\n              result = await this._pollForStatementResult(\n                workspaceClient,\n                statementId,\n                this.config.timeout,\n                signal,\n              );\n              break;\n            case \"SUCCEEDED\":\n              result = await this._transformDataArray(\n                response,\n                workspaceClient,\n                signal,\n              );\n              break;\n            case \"FAILED\":\n              throw ExecutionError.statementFailed(\n                status.error?.message,\n                status.error?.error_code,\n              );\n            case \"CANCELED\":\n              throw ExecutionError.canceled();\n            case \"CLOSED\":\n              throw ExecutionError.resultsClosed();\n            default:\n              throw ExecutionError.unknownState(\n                String(status?.state ?? \"unknown\"),\n              );\n          }\n\n          const resultData = result.result as any;\n          const rowCount =\n            resultData?.data?.length ?? resultData?.data_array?.length ?? 0;\n\n          if (rowCount > 0) {\n            span.setAttribute(\"db.result.row_count\", rowCount);\n          }\n\n          const duration = Date.now() - startTime;\n          logger.event()?.setContext(\"sql-warehouse\", {\n            warehouse_id: input.warehouse_id,\n            rows_returned: rowCount,\n            query_duration_ms: duration,\n          });\n\n          success = true;\n          // only set success status if not aborted\n          if (!isAborted) {\n            span.setStatus({ code: SpanStatusCode.OK });\n          }\n          return result;\n        } catch (error) {\n          // only record error if not already handled by abort\n          if (!isAborted) {\n            span.recordException(error as Error);\n            span.setStatus({\n              code: SpanStatusCode.ERROR,\n              message: error instanceof Error ? error.message : String(error),\n            });\n\n            logger.error(\n              \"Statement execution failed: %s\",\n              error instanceof Error ? error.message : String(error),\n            );\n          }\n\n          this._rethrowStatementError(error);\n        } finally {\n          // remove abort handler\n          if (abortHandler && signal) {\n            signal.removeEventListener(\"abort\", abortHandler);\n          }\n\n          const duration = Date.now() - startTime;\n\n          // end span if not already ended by abort handler\n          if (!isAborted) {\n            span.end();\n          }\n\n          const attributes = {\n            \"db.warehouse_id\": input.warehouse_id,\n            \"db.catalog\": input.catalog ?? \"\",\n            \"db.schema\": input.schema ?? \"\",\n            \"db.statement\": input.statement?.substring(0, 500) || \"\",\n            success: success.toString(),\n          };\n\n          this.telemetryMetrics.queryCount.add(1, attributes);\n          this.telemetryMetrics.queryDuration.record(duration, attributes);\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  /**\n   * Wait until the SQL warehouse is in the `RUNNING` state, auto-starting it\n   * if currently `STOPPED`. Emits a {@link WarehouseStatusUpdate} whenever\n   * the observed state changes, so callers can surface progress (e.g. over\n   * SSE) instead of letting the UI freeze on a cold warehouse. Equal\n   * successive observations are de-duplicated.\n   *\n   * Fast path: if this connector recently observed the warehouse RUNNING\n   * (within {@link WAREHOUSE_RUNNING_CACHE_TTL_MS}), the call returns\n   * immediately without any SDK round-trip or status emission. This keeps\n   * cache-hit analytics requests off the Databricks control plane.\n   *\n   * Concurrent callers for the same `warehouseId` share a single in-flight\n   * poll loop (singleflight): only the owner issues `get`/`start`; joiners\n   * receive broadcast `onStatus` updates and share the result promise.\n   * When every waiter disconnects before `warehouses.start`, the shared poll\n   * is aborted on the next microtask if still unclaimed (kills true orphans\n   * without a fixed grace window). After `start` is issued, the poll runs to\n   * completion even with no waiters.\n   *\n   * Behaviour by initial state:\n   * - `RUNNING`: emits one update, caches the observation, returns.\n   * - `STOPPED`: emits a synthetic `STARTING` update, calls\n   *   `workspaceClient.warehouses.start`, then polls until `RUNNING`.\n   *   When `autoStart: false`, throws `ConfigurationError` instead.\n   * - `STARTING` / `STOPPING`: polls until `RUNNING`.\n   * - `DELETED` / `DELETING`: throws `ConfigurationError.resourceNotFound`.\n   *\n   * Aborts and timeouts surface as `ExecutionError`. The poll loop uses\n   * exponential backoff with ±15% jitter (1s → 30s cap) and runs inside a\n   * `sql.warehouseReady` telemetry span.\n   */\n  async ensureWarehouseRunning(\n    workspaceClient: WorkspaceClient,\n    warehouseId: string,\n    opts: EnsureWarehouseRunningOptions,\n  ): Promise<void> {\n    const {\n      onStatus,\n      signal,\n      timeoutMs = DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS,\n      autoStart = true,\n    } = opts;\n\n    if (signal?.aborted) throw ExecutionError.canceled();\n    if (!warehouseId) throw ValidationError.missingField(\"warehouse_id\");\n    if (this._isRecentlyRunning(warehouseId)) return;\n\n    const existing = this._readinessInFlight.get(warehouseId);\n    if (existing && !existing.sharedController.signal.aborted) {\n      return this._waitOnReadiness(existing, onStatus, signal, true);\n    }\n\n    const entry = this._spawnReadinessInFlight(workspaceClient, warehouseId, {\n      timeoutMs,\n      autoStart,\n    });\n    this._readinessInFlight.set(warehouseId, entry);\n    return this._waitOnReadiness(entry, onStatus, signal, false);\n  }\n\n  private _waitOnReadiness(\n    entry: WarehouseReadinessInFlight,\n    onStatus: (update: WarehouseStatusUpdate) => void,\n    signal: AbortSignal | undefined,\n    join: boolean,\n  ): Promise<void> {\n    if (signal?.aborted) throw ExecutionError.canceled();\n    if (join) entry.refCount++;\n    entry.subscribers.add(onStatus);\n    if (entry.lastUpdate) onStatus(entry.lastUpdate);\n    return this._joinWarehouseReadiness(entry, signal);\n  }\n\n  private _spawnReadinessInFlight(\n    workspaceClient: WorkspaceClient,\n    warehouseId: string,\n    opts: { timeoutMs: number; autoStart: boolean },\n  ): WarehouseReadinessInFlight {\n    const sharedController = new AbortController();\n    const entry: WarehouseReadinessInFlight = {\n      refCount: 1,\n      sharedController,\n      subscribers: new Set(),\n      lastUpdate: null,\n      warehouseStartIssued: false,\n      promise: Promise.resolve(),\n    };\n\n    entry.promise = this.telemetry\n      .startActiveSpan(\n        \"sql.warehouseReady\",\n        {\n          kind: SpanKind.CLIENT,\n          attributes: {\n            \"db.system\": \"databricks\",\n            \"db.warehouse_id\": warehouseId,\n            \"db.warehouse.startup_timeout_ms\": opts.timeoutMs,\n          },\n        },\n        async (span: Span) => {\n          try {\n            await this._pollUntilWarehouseRunning(\n              workspaceClient,\n              warehouseId,\n              {\n                onStatus: (update) => this._broadcastReadiness(entry, update),\n                signal: sharedController.signal,\n                timeoutMs: opts.timeoutMs,\n                autoStart: opts.autoStart,\n                onWarehouseStartIssued: () => {\n                  entry.warehouseStartIssued = true;\n                },\n              },\n              span,\n            );\n            span.setStatus({ code: SpanStatusCode.OK });\n          } catch (error) {\n            this._throwSanitizedReadinessError(error, warehouseId, span);\n          } finally {\n            span.end();\n          }\n        },\n        { name: this.name, includePrefix: true },\n      )\n      .finally(() => {\n        if (this._readinessInFlight.get(warehouseId) === entry) {\n          this._readinessInFlight.delete(warehouseId);\n        }\n      });\n\n    entry.promise.catch(() => {});\n    return entry;\n  }\n\n  private _broadcastReadiness(\n    entry: WarehouseReadinessInFlight,\n    update: WarehouseStatusUpdate,\n  ): void {\n    entry.lastUpdate = update;\n    for (const subscriber of entry.subscribers) {\n      try {\n        subscriber(update);\n      } catch {\n        // One consumer must not block updates to the rest.\n      }\n    }\n  }\n\n  /** Ref-counted await; caller abort rejects locally. */\n  private _joinWarehouseReadiness(\n    entry: WarehouseReadinessInFlight,\n    callerSignal?: AbortSignal,\n  ): Promise<void> {\n    if (!callerSignal) return entry.promise;\n\n    return new Promise<void>((resolve, reject) => {\n      let done = false;\n      const finish = (outcome: \"ok\" | \"abort\" | \"err\", error?: unknown) => {\n        if (done) return;\n        done = true;\n        callerSignal.removeEventListener(\"abort\", onAbort);\n        if (outcome === \"ok\") resolve();\n        else if (outcome === \"abort\") reject(ExecutionError.canceled());\n        else reject(error);\n      };\n      const onAbort = () => {\n        this._releaseReadinessWaiter(entry);\n        finish(\"abort\");\n      };\n      callerSignal.addEventListener(\"abort\", onAbort, { once: true });\n      entry.promise.then(\n        () => finish(\"ok\"),\n        (error) => finish(\"err\", error),\n      );\n    });\n  }\n\n  private _releaseReadinessWaiter(entry: WarehouseReadinessInFlight): void {\n    if (entry.refCount > 0) entry.refCount--;\n    if (entry.refCount > 0 || entry.sharedController.signal.aborted) return;\n\n    if (entry.warehouseStartIssued) {\n      // Start already issued — finish warming for the process.\n      return;\n    }\n\n    // Defer abort to the next microtask so a synchronous StrictMode\n    // remount can rejoin before the shared poll is cancelled.\n    queueMicrotask(() => {\n      if (entry.refCount <= 0 && !entry.sharedController.signal.aborted) {\n        entry.sharedController.abort(\"all warehouse readiness waiters aborted\");\n      }\n    });\n  }\n\n  private async _pollUntilWarehouseRunning(\n    workspaceClient: WorkspaceClient,\n    warehouseId: string,\n    opts: EnsureWarehouseRunningOptions,\n    span: Span,\n  ): Promise<void> {\n    const {\n      onStatus,\n      signal,\n      timeoutMs = DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS,\n      autoStart = true,\n      onWarehouseStartIssued,\n    } = opts;\n    const startTime = Date.now();\n    const emitter = new WarehouseStatusEmitter(span, startTime, onStatus);\n    let didStart = false;\n    const backoff = new WarehousePollBackoff();\n\n    while (true) {\n      if (signal?.aborted) throw ExecutionError.canceled();\n      if (Date.now() - startTime > timeoutMs) {\n        throw ExecutionError.statementFailed(\n          `SQL warehouse did not reach RUNNING within ${timeoutMs}ms`,\n        );\n      }\n\n      const info = await workspaceClient.warehouses.get(\n        { id: warehouseId },\n        this._createContext(signal),\n      );\n      const state = info?.state;\n      const summary = info?.health?.summary;\n\n      switch (state) {\n        case \"RUNNING\":\n          emitter.emit(state, summary);\n          this._recentlyRunning.set(warehouseId, Date.now());\n          span.setAttribute(\"db.warehouse.attempts\", emitter.attempt);\n          return;\n        case \"DELETED\":\n        case \"DELETING\":\n          throw ConfigurationError.resourceNotFound(\n            \"Warehouse ID\",\n            `The configured SQL warehouse is ${state}. Update DATABRICKS_WAREHOUSE_ID to point at an active warehouse.`,\n          );\n        case \"STOPPED\":\n          if (!autoStart) {\n            throw new ConfigurationError(\n              \"The configured SQL warehouse is STOPPED and analytics auto-start is disabled. Start the warehouse manually or set analytics.autoStartWarehouse=true.\",\n            );\n          }\n          if (!didStart) {\n            emitter.emit(\"STARTING\", summary);\n            onWarehouseStartIssued?.();\n            await workspaceClient.warehouses.start(\n              { id: warehouseId },\n              this._createContext(signal),\n            );\n            didStart = true;\n          } else {\n            emitter.emit(state, summary);\n          }\n          break;\n        case \"STARTING\":\n        case \"STOPPING\":\n          emitter.emit(state, summary);\n          break;\n        default:\n          throw ExecutionError.unknownState(String(state ?? \"unknown\"));\n      }\n\n      const sleepMs = backoff.next();\n      if (Date.now() + sleepMs - startTime >= timeoutMs) {\n        throw ExecutionError.statementFailed(\n          `SQL warehouse did not reach RUNNING within ${timeoutMs}ms`,\n        );\n      }\n      await this._sleepRespectingAbort(sleepMs, signal);\n    }\n  }\n\n  /**\n   * `true` if this connector observed `warehouseId` in the RUNNING state\n   * within the recent-cache TTL.\n   */\n  private _isRecentlyRunning(warehouseId: string): boolean {\n    const observedAt = this._recentlyRunning.get(warehouseId);\n    return (\n      observedAt !== undefined &&\n      Date.now() - observedAt < WAREHOUSE_RUNNING_CACHE_TTL_MS\n    );\n  }\n\n  /**\n   * Final landing for any error thrown by the readiness poll loop. Records\n   * the original on the span and (at debug level) on the logger, then\n   * rethrows either the structured AppKitError unchanged or a curated\n   * ExecutionError — never the raw SDK message, which can contain operator\n   * internals.\n   *\n   * The `logger.debug` covers environments where OTel isn't configured\n   * (common in dev) so the raw error isn't lost just because no span\n   * exporter is hooked up.\n   */\n  private _throwSanitizedReadinessError(\n    error: unknown,\n    warehouseId: string,\n    span: Span,\n  ): never {\n    if (error instanceof AppKitError) {\n      span.recordException(error);\n      span.setStatus({ code: SpanStatusCode.ERROR, message: error.code });\n      throw error;\n    }\n    span.recordException(\n      error instanceof Error ? error : new Error(String(error)),\n    );\n    logger.debug(\n      \"Warehouse readiness check raw error for %s: %O\",\n      warehouseId,\n      error,\n    );\n    const wrapped = ExecutionError.statementFailed(\n      \"Warehouse readiness check failed\",\n    );\n    span.setStatus({ code: SpanStatusCode.ERROR, message: wrapped.code });\n    throw wrapped;\n  }\n\n  /**\n   * Sleep for `ms` milliseconds, but resolve early (and reject with\n   * `ExecutionError.canceled()`) if `signal` aborts mid-sleep. Used by the\n   * warehouse readiness loop so that client disconnects don't keep the loop\n   * polling for the full interval.\n   */\n  private _sleepRespectingAbort(\n    ms: number,\n    signal?: AbortSignal,\n  ): Promise<void> {\n    if (signal?.aborted) return Promise.reject(ExecutionError.canceled());\n    return new Promise<void>((resolve, reject) => {\n      const timer = setTimeout(() => {\n        signal?.removeEventListener(\"abort\", onAbort);\n        resolve();\n      }, ms);\n      const onAbort = () => {\n        clearTimeout(timer);\n        reject(ExecutionError.canceled());\n      };\n      signal?.addEventListener(\"abort\", onAbort, { once: true });\n    });\n  }\n\n  private async _pollForStatementResult(\n    workspaceClient: WorkspaceClient,\n    statementId: string,\n    timeout = executeStatementDefaults.timeout,\n    signal?: AbortSignal,\n  ) {\n    return this.telemetry.startActiveSpan(\n      \"sql.poll\",\n      {\n        attributes: {\n          \"db.statement_id\": statementId,\n          \"db.polling.timeout\": timeout,\n        },\n      },\n      async (span: Span) => {\n        try {\n          const startTime = Date.now();\n          let delay = 1000;\n          const maxDelayBetweenPolls = 5000; // max 5 seconds between polls\n          let pollCount = 0;\n\n          while (true) {\n            pollCount++;\n            span.setAttribute(\"db.polling.current_attempt\", pollCount);\n\n            // check if timeout exceeded\n            const elapsedTime = Date.now() - startTime;\n            if (elapsedTime > timeout) {\n              const error = ExecutionError.statementFailed(\n                `Polling timeout exceeded after ${timeout}ms (elapsed: ${elapsedTime}ms)`,\n              );\n              span.recordException(error);\n              span.setStatus({ code: SpanStatusCode.ERROR });\n              throw error;\n            }\n\n            if (signal?.aborted) {\n              const error = ExecutionError.canceled();\n              span.recordException(error);\n              span.setStatus({ code: SpanStatusCode.ERROR });\n              throw error;\n            }\n\n            span.addEvent(\"polling.attempt\", {\n              \"poll.attempt\": pollCount,\n              \"poll.delay_ms\": delay,\n              \"poll.elapsed_ms\": elapsedTime,\n            });\n\n            const response =\n              await workspaceClient.statementExecution.getStatement(\n                {\n                  statement_id: statementId,\n                },\n                this._createContext(signal),\n              );\n            if (!response) {\n              throw ConnectionError.apiFailure(\"SQL Warehouse\");\n            }\n\n            const status = response.status;\n\n            span.addEvent(\"polling.status_check\", {\n              \"db.status\": status?.state,\n              \"poll.attempt\": pollCount,\n            });\n\n            switch (status?.state) {\n              case \"PENDING\":\n              case \"RUNNING\":\n                // continue polling\n                break;\n              case \"SUCCEEDED\":\n                span.setAttribute(\"db.polling.attempts\", pollCount);\n                span.setAttribute(\"db.polling.total_duration_ms\", elapsedTime);\n                span.addEvent(\"polling.completed\", {\n                  \"poll.attempts\": pollCount,\n                  \"poll.duration_ms\": elapsedTime,\n                });\n                span.setStatus({ code: SpanStatusCode.OK });\n                return this._transformDataArray(\n                  response,\n                  workspaceClient,\n                  signal,\n                );\n              case \"FAILED\":\n                throw ExecutionError.statementFailed(\n                  status.error?.message,\n                  status.error?.error_code,\n                );\n              case \"CANCELED\":\n                throw ExecutionError.canceled();\n              case \"CLOSED\":\n                throw ExecutionError.resultsClosed();\n              default:\n                throw ExecutionError.unknownState(\n                  String(status?.state ?? \"unknown\"),\n                );\n            }\n\n            // continue polling after delay\n            await new Promise((resolve) => setTimeout(resolve, delay));\n            delay = Math.min(delay * 2, maxDelayBetweenPolls);\n          }\n        } catch (error) {\n          span.recordException(error as Error);\n          span.setStatus({\n            code: SpanStatusCode.ERROR,\n            message: error instanceof Error ? error.message : String(error),\n          });\n\n          // error logging is handled by executeStatement's catch block (gated on isAborted)\n          this._rethrowStatementError(error);\n        } finally {\n          span.end();\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  private async _transformDataArray(\n    response: sql.StatementResponse,\n    workspaceClient: WorkspaceClient,\n    signal?: AbortSignal,\n  ) {\n    if (response.manifest?.format === \"ARROW_STREAM\") {\n      const result = response.result as\n        | (sql.ResultData & { attachment?: string })\n        | undefined;\n\n      // Inline Arrow: pass the base64 IPC attachment through unmodified so\n      // the analytics route can stream it to the client, where the existing\n      // ArrowClient infrastructure decodes it into a Table. Validate size\n      // here to fail fast on runaway payloads.\n      if (result?.attachment) {\n        return this._validateArrowAttachment(response, result.attachment);\n      }\n\n      // External links: data fetched separately via statement_id. An empty\n      // array is a zero-row result (some warehouses emit `external_links: []`\n      // rather than omitting it) — it must NOT go down the streaming path\n      // (`streamChunks([])` rejects), so fall through to synthesize an empty\n      // Arrow table below.\n      if (result?.external_links && result.external_links.length > 0) {\n        return this.updateWithArrowStatus(response, workspaceClient, signal);\n      }\n\n      // Empty result with a known schema: synthesize a zero-row Arrow IPC\n      // attachment so the client always receives an Arrow Table for\n      // ARROW_STREAM, regardless of whether the warehouse returned data.\n      // Note: an empty array (`data_array: []`) is truthy, so length-check\n      // explicitly — otherwise zero-row responses fall through to the JSON\n      // row transform below and return `[]` JSON rows instead of an Arrow\n      // table.\n      const hasNoRows =\n        !result?.data_array ||\n        (Array.isArray(result.data_array) && result.data_array.length === 0);\n      if (hasNoRows && response.manifest?.schema?.columns) {\n        const synthesized = buildEmptyArrowIPCBase64(\n          response.manifest.schema.columns,\n        );\n        return {\n          ...response,\n          result: { ...(result ?? {}), attachment: synthesized },\n        };\n      }\n\n      // Inline data_array under ARROW_STREAM (rare): fall through to the\n      // row transform below. The hook will receive `type: \"result\"` rows;\n      // callers asking for ARROW_STREAM should not hit this path with\n      // current Databricks warehouses.\n    }\n\n    if (!response.result?.data_array || !response.manifest?.schema?.columns) {\n      return response;\n    }\n\n    const columns = response.manifest.schema.columns;\n\n    const transformedData = response.result.data_array.map((row) => {\n      const obj: Record<string, unknown> = {};\n      row.forEach((value, index) => {\n        const column = columns[index];\n        const columnName = column?.name || `column_${index}`;\n\n        // attempt to parse JSON strings for string columns\n        if (\n          column?.type_name === \"STRING\" &&\n          typeof value === \"string\" &&\n          value &&\n          (value[0] === \"{\" || value[0] === \"[\")\n        ) {\n          try {\n            obj[columnName] = JSON.parse(value);\n          } catch {\n            // if parsing fails, keep as string\n            obj[columnName] = value;\n          }\n        } else {\n          obj[columnName] = value;\n        }\n      });\n      return obj;\n    });\n\n    // remove data_array\n    const { data_array: _data_array, ...restResult } = response.result;\n    return {\n      ...response,\n      result: {\n        ...restResult,\n        data: transformedData,\n      },\n    };\n  }\n\n  /**\n   * Validate a base64 Arrow IPC attachment (INLINE ARROW_STREAM) and attach\n   * the real column names from the result manifest.\n   *\n   * Databricks warehouses encode the Arrow schema positionally (`col_0`, …);\n   * the real, aliased names live only in `manifest.schema.columns`. Rather\n   * than decode/re-encode the payload server-side, we carry the names on the\n   * result (`columnNames`) so the route can hand them to the client in a\n   * response header and the client relabels the decoded Table — the one\n   * mechanism used for both INLINE and EXTERNAL_LINKS.\n   */\n  private _validateArrowAttachment(\n    response: sql.StatementResponse,\n    attachment: string,\n  ) {\n    // Cap the size to protect against unbounded inline payloads from\n    // misbehaving warehouses. `MAX_INLINE_ATTACHMENT_BYTES` tracks the API's\n    // ~25 MiB inline hard cap and bounds memory if a server returns a runaway\n    // response.\n    //\n    // Strip whitespace (rare but legal in base64) and account for trailing\n    // `=` padding so the byte count is exact rather than an upper bound.\n    const stripped = attachment.replace(/\\s+/g, \"\");\n    const padding = stripped.endsWith(\"==\")\n      ? 2\n      : stripped.endsWith(\"=\")\n        ? 1\n        : 0;\n    const decodedSize = Math.floor((stripped.length * 3) / 4) - padding;\n    if (decodedSize > MAX_INLINE_ATTACHMENT_BYTES) {\n      throw ExecutionError.statementFailed(\n        `Inline Arrow attachment exceeds maximum size (${decodedSize} > ${MAX_INLINE_ATTACHMENT_BYTES} bytes)`,\n      );\n    }\n\n    const columnNames = arrowColumnNames(response);\n    if (columnNames) {\n      return {\n        ...response,\n        result: {\n          ...(response.result as sql.ResultData & {\n            attachment?: string;\n            columnNames?: string[];\n          }),\n          // `statement_id` is a top-level field, not on `ResultData` — carry it\n          // onto the result (as the EXTERNAL_LINKS path does) so the route can\n          // advertise it in `X-Appkit-Arrow-Columns-Ref` for wide inline schemas.\n          statement_id: response.statement_id,\n          columnNames,\n        },\n      };\n    }\n\n    return response;\n  }\n\n  private async updateWithArrowStatus(\n    response: sql.StatementResponse,\n    workspaceClient: WorkspaceClient,\n    signal?: AbortSignal,\n  ): Promise<{\n    result: {\n      statement_id: string;\n      status: sql.StatementStatus;\n      columnNames?: string[];\n      external_links?: sql.ExternalLink[];\n      refreshChunkLink?: RefreshChunkLink;\n    };\n  }> {\n    const statementId = response.statement_id as string;\n    return {\n      result: {\n        statement_id: statementId,\n        status: {\n          state: response.status?.state,\n          error: response.status?.error,\n        } as sql.StatementStatus,\n        columnNames: arrowColumnNames(response),\n        // Resolve the pre-signed links for EVERY chunk in the caller's own\n        // execution context. Streaming these directly (see\n        // {@link streamExternalLinks}) avoids a second `getStatement` under\n        // the ambient service-principal context — which would fail for\n        // `.obo.sql` (user-owned) statements.\n        external_links: await this._resolveAllExternalLinks(\n          workspaceClient,\n          statementId,\n          response,\n          signal,\n        ),\n        // Bound to this caller's identity so the streamer can re-mint an\n        // expired chunk link mid-download (links live <= 15 min; a large/slow\n        // result can outlast the earliest ones).\n        refreshChunkLink: this._makeChunkLinkRefresher(\n          workspaceClient,\n          statementId,\n        ),\n      },\n    };\n  }\n\n  /**\n   * Resolve pre-signed links for EVERY chunk of an EXTERNAL_LINKS result.\n   *\n   * The execute/getStatement response carries only the first chunk's links\n   * (each link, except the last, exposes `next_chunk_index`); the remaining\n   * chunks are fetched with `getStatementResultChunkN`. Runs in the caller's\n   * identity context (user creds for `.obo.sql`), so there is no cross-identity\n   * fetch. Only the tiny link metadata is resolved eagerly — the bytes still\n   * stream one chunk at a time downstream. Without this a multi-chunk result\n   * would be silently truncated to its first chunk.\n   */\n  private async _resolveAllExternalLinks(\n    workspaceClient: WorkspaceClient,\n    statementId: string,\n    response: sql.StatementResponse,\n    signal?: AbortSignal,\n  ): Promise<sql.ExternalLink[] | undefined> {\n    const first = response.result?.external_links;\n    if (!first || first.length === 0) return first;\n\n    const links: sql.ExternalLink[] = [...first];\n    // Bound the follow loop so a warehouse returning a cyclic/never-ending\n    // `next_chunk_index` can't spin forever. The manifest's chunk count is the\n    // natural bound; fall back to a generous safety cap if it's absent (real\n    // results still terminate earlier when `next_chunk_index` becomes null) so\n    // a missing count doesn't silently truncate a genuine multi-chunk result.\n    const maxFetches =\n      response.manifest?.total_chunk_count ?? MAX_EXTERNAL_CHUNK_FOLLOWS;\n    let next = this._nextChunkIndex(first);\n    for (let fetches = 0; next != null && fetches < maxFetches; fetches++) {\n      if (signal?.aborted) throw ExecutionError.canceled();\n      const chunk =\n        await workspaceClient.statementExecution.getStatementResultChunkN(\n          { statement_id: statementId, chunk_index: next },\n          this._createContext(signal),\n        );\n      const chunkLinks = chunk.external_links ?? [];\n      if (chunkLinks.length === 0) break;\n      links.push(...chunkLinks);\n      next = this._nextChunkIndex(chunkLinks);\n    }\n    return links;\n  }\n\n  /** The `next_chunk_index` advertised by a chunk's links, if any. */\n  private _nextChunkIndex(links: sql.ExternalLink[]): number | undefined {\n    for (const link of links) {\n      if (link.next_chunk_index != null) return link.next_chunk_index;\n    }\n    return undefined;\n  }\n\n  /**\n   * A closure that re-mints a single chunk's pre-signed link via\n   * `getStatementResultChunkN`, bound to the caller's workspace client +\n   * statement id. Created here (in the caller's identity context) so the\n   * streamer — which runs outside that context — can refresh an expired link\n   * for `.obo.sql` statements without a cross-identity `getStatement`.\n   */\n  private _makeChunkLinkRefresher(\n    workspaceClient: WorkspaceClient,\n    statementId: string,\n  ): RefreshChunkLink {\n    return async (chunkIndex, signal) => {\n      const chunk =\n        await workspaceClient.statementExecution.getStatementResultChunkN(\n          { statement_id: statementId, chunk_index: chunkIndex },\n          this._createContext(signal),\n        );\n      return chunk.external_links?.find((l) => l.chunk_index === chunkIndex);\n    };\n  }\n\n  /**\n   * Stream already-resolved EXTERNAL_LINKS chunks as raw Arrow IPC bytes, one\n   * chunk in memory at a time. Takes the `external_links` the caller already\n   * has from `executeStatement` (see {@link updateWithArrowStatus}), so there\n   * is no extra `getStatement` round-trip and no execution-context mismatch —\n   * the pre-signed URLs need no auth to download.\n   */\n  streamExternalLinks(\n    chunks: sql.ExternalLink[],\n    signal?: AbortSignal,\n    refresh?: RefreshChunkLink,\n  ): AsyncGenerator<Uint8Array, void, unknown> {\n    return this.arrowProcessor.streamChunks(chunks, signal, refresh);\n  }\n\n  /**\n   * Fetch just the real column names for a completed statement from its result\n   * manifest — no result data. Backs the `/columns/:statementId` fallback the\n   * client uses when a very wide schema's names exceed the response-header\n   * size limit. Stateless: re-derives from the warehouse, no server cache.\n   */\n  async getColumnNames(\n    workspaceClient: WorkspaceClient,\n    jobId: string,\n    signal?: AbortSignal,\n  ): Promise<string[] | undefined> {\n    const response = await workspaceClient.statementExecution.getStatement(\n      { statement_id: jobId },\n      this._createContext(signal),\n    );\n    return arrowColumnNames(response);\n  }\n\n  /**\n   * Normalize an error caught during statement execution/polling and rethrow.\n   * Abort and already-structured `AppKitError`s pass through untouched;\n   * anything else is wrapped as an `ExecutionError`, preserving the SDK's\n   * structured `ApiError.errorCode` (e.g. \"INVALID_PARAMETER_VALUE\",\n   * \"BAD_REQUEST\") so callers can branch on a stable identifier rather than\n   * substring-matching the message.\n   */\n  private _rethrowStatementError(error: unknown): never {\n    if (error instanceof Error && error.name === \"AbortError\") {\n      throw error;\n    }\n    if (error instanceof AppKitError) {\n      throw error;\n    }\n    const sdkErrorCode =\n      error && typeof error === \"object\" && \"errorCode\" in error\n        ? (error as { errorCode?: unknown }).errorCode\n        : undefined;\n    throw ExecutionError.statementFailed(\n      error instanceof Error ? error.message : String(error),\n      typeof sdkErrorCode === \"string\" ? sdkErrorCode : undefined,\n    );\n  }\n\n  // create context for cancellation token\n  private _createContext(signal?: AbortSignal) {\n    return new Context({\n      cancellationToken: {\n        isCancellationRequested: signal?.aborted ?? false,\n        onCancellationRequested: (cb: () => void) => {\n          signal?.addEventListener(\"abort\", cb, { once: true });\n        },\n      },\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAiCA,MAAM,SAAS,aAAa,2BAA2B;;;;;;;;AASvD,SAAS,iBACP,UACsB;CACtB,MAAM,OAAO,SAAS,UAAU,QAAQ;AACxC,KAAI,CAAC,QAAQ,KAAK,WAAW,EAAG,QAAO;AACvC,QAAO,KAAK,KAAK,GAAG,MAClB,EAAE,QAAQ,EAAE,KAAK,SAAS,IAAI,EAAE,OAAO,UAAU,IAClD;;;;;;;;;;;AAYH,MAAM,8BAA8B,KAAK,OAAO;;;;;;;AAQhD,MAAM,6BAA6B;;;;;;;;AAcnC,MAAa,uCAAuC,MAAS;;;;;;;;AAS7D,MAAM,iCAAiC;AAqDvC,IAAa,wBAAb,MAAmC;CACjC,AAAiB,OAAO;CAExB,AAAQ;CAGR,AAAQ,kBAA+C;;;;;;CAOvD,AAAQ,mCAAmB,IAAI,KAAqB;;CAEpD,AAAQ,qCAAqB,IAAI,KAAyC;CAE1E,AAAiB;CACjB,AAAiB;CAKjB,YAAY,QAA4B;AACtC,OAAK,SAAS;AAEd,OAAK,YAAY,iBAAiB,YAChC,KAAK,MACL,KAAK,OAAO,UACb;AACD,OAAK,mBAAmB;GACtB,YAAY,KAAK,UAAU,UAAU,CAAC,cAAc,eAAe;IACjE,aAAa;IACb,MAAM;IACP,CAAC;GACF,eAAe,KAAK,UACjB,UAAU,CACV,gBAAgB,kBAAkB;IACjC,aAAa;IACb,MAAM;IACP,CAAC;GACL;;;;;;CAOH,IAAY,iBAAuC;AACjD,MAAI,CAAC,KAAK,gBACR,MAAK,kBAAkB,IAAI,qBAAqB;GAC9C,SAAS,KAAK,OAAO,WAAW,yBAAyB;GACzD,SAAS,qBAAqB;GAC/B,CAAC;AAEJ,SAAO,KAAK;;CAGd,MAAM,iBACJ,iBACA,OACA,QACA;EACA,MAAM,YAAY,KAAK,KAAK;EAC5B,IAAI,UAAU;AAGd,MAAI,QAAQ,QACV,OAAM,eAAe,UAAU;AAGjC,SAAO,KAAK,UAAU,gBACpB,aACA;GACE,MAAM,SAAS;GACf,YAAY;IACV,aAAa;IACb,mBAAmB,MAAM,gBAAgB;IACzC,cAAc,MAAM,WAAW;IAC/B,aAAa,MAAM,UAAU;IAC7B,gBAAgB,MAAM,WAAW,UAAU,GAAG,IAAI,IAAI;IACtD,qBAAqB,CAAC,CAAC,MAAM;IAC9B;GACF,EACD,OAAO,SAAe;GACpB,IAAI;GACJ,IAAI,YAAY;AAEhB,OAAI,QAAQ;AACV,yBAAqB;AAEnB,SAAI,CAAC,KAAK,aAAa,CAAE;AACzB,iBAAY;AACZ,UAAK,aAAa,aAAa,KAAK;AACpC,UAAK,UAAU;MACb,MAAM,eAAe;MACrB,SAAS;MACV,CAAC;AACF,UAAK,KAAK;;AAEZ,WAAO,iBAAiB,SAAS,cAAc,EAAE,MAAM,MAAM,CAAC;;AAGhE,OAAI;AAEF,QAAI,CAAC,MAAM,UACT,OAAM,gBAAgB,aAAa,YAAY;AAGjD,QAAI,CAAC,MAAM,aACT,OAAM,gBAAgB,aAAa,eAAe;IAGpD,MAAM,OAAoC;KACxC,WAAW,MAAM;KACjB,YAAY,MAAM;KAClB,cAAc,MAAM;KACpB,SAAS,MAAM;KACf,QAAQ,MAAM;KACd,cACE,MAAM,gBAAgB,yBAAyB;KACjD,aACE,MAAM,eAAe,yBAAyB;KAChD,QAAQ,MAAM,UAAU,yBAAyB;KACjD,YAAY,MAAM;KAClB,WAAW,MAAM;KACjB,iBACE,MAAM,mBAAmB,yBAAyB;KACrD;AAED,SAAK,SAAS,wBAAwB,EACpC,mBAAmB,MAAM,cAC1B,CAAC;IAEF,MAAM,WACJ,MAAM,gBAAgB,mBAAmB,iBACvC,MACA,KAAK,eAAe,OAAO,CAC5B;AAEH,QAAI,CAAC,SACH,OAAM,gBAAgB,WAAW,gBAAgB;IAEnD,MAAM,SAAS,SAAS;IACxB,MAAM,cAAc,SAAS;AAE7B,SAAK,aAAa,mBAAmB,YAAY;AACjD,SAAK,SAAS,uBAAuB;KACnC,mBAAmB,SAAS;KAC5B,aAAa,QAAQ;KACtB,CAAC;IAEF,IAAI;AAIJ,YAAQ,QAAQ,OAAhB;KACE,KAAK;KACL,KAAK;AACH,WAAK,SAAS,6BAA6B,EACzC,aAAa,SAAS,QAAQ,OAC/B,CAAC;AACF,eAAS,MAAM,KAAK,wBAClB,iBACA,aACA,KAAK,OAAO,SACZ,OACD;AACD;KACF,KAAK;AACH,eAAS,MAAM,KAAK,oBAClB,UACA,iBACA,OACD;AACD;KACF,KAAK,SACH,OAAM,eAAe,gBACnB,OAAO,OAAO,SACd,OAAO,OAAO,WACf;KACH,KAAK,WACH,OAAM,eAAe,UAAU;KACjC,KAAK,SACH,OAAM,eAAe,eAAe;KACtC,QACE,OAAM,eAAe,aACnB,OAAO,QAAQ,SAAS,UAAU,CACnC;;IAGL,MAAM,aAAa,OAAO;IAC1B,MAAM,WACJ,YAAY,MAAM,UAAU,YAAY,YAAY,UAAU;AAEhE,QAAI,WAAW,EACb,MAAK,aAAa,uBAAuB,SAAS;IAGpD,MAAM,WAAW,KAAK,KAAK,GAAG;AAC9B,WAAO,OAAO,EAAE,WAAW,iBAAiB;KAC1C,cAAc,MAAM;KACpB,eAAe;KACf,mBAAmB;KACpB,CAAC;AAEF,cAAU;AAEV,QAAI,CAAC,UACH,MAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAE7C,WAAO;YACA,OAAO;AAEd,QAAI,CAAC,WAAW;AACd,UAAK,gBAAgB,MAAe;AACpC,UAAK,UAAU;MACb,MAAM,eAAe;MACrB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;MAChE,CAAC;AAEF,YAAO,MACL,kCACA,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;;AAGH,SAAK,uBAAuB,MAAM;aAC1B;AAER,QAAI,gBAAgB,OAClB,QAAO,oBAAoB,SAAS,aAAa;IAGnD,MAAM,WAAW,KAAK,KAAK,GAAG;AAG9B,QAAI,CAAC,UACH,MAAK,KAAK;IAGZ,MAAM,aAAa;KACjB,mBAAmB,MAAM;KACzB,cAAc,MAAM,WAAW;KAC/B,aAAa,MAAM,UAAU;KAC7B,gBAAgB,MAAM,WAAW,UAAU,GAAG,IAAI,IAAI;KACtD,SAAS,QAAQ,UAAU;KAC5B;AAED,SAAK,iBAAiB,WAAW,IAAI,GAAG,WAAW;AACnD,SAAK,iBAAiB,cAAc,OAAO,UAAU,WAAW;;KAGpE;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAmCH,MAAM,uBACJ,iBACA,aACA,MACe;EACf,MAAM,EACJ,UACA,QACA,YAAY,sCACZ,YAAY,SACV;AAEJ,MAAI,QAAQ,QAAS,OAAM,eAAe,UAAU;AACpD,MAAI,CAAC,YAAa,OAAM,gBAAgB,aAAa,eAAe;AACpE,MAAI,KAAK,mBAAmB,YAAY,CAAE;EAE1C,MAAM,WAAW,KAAK,mBAAmB,IAAI,YAAY;AACzD,MAAI,YAAY,CAAC,SAAS,iBAAiB,OAAO,QAChD,QAAO,KAAK,iBAAiB,UAAU,UAAU,QAAQ,KAAK;EAGhE,MAAM,QAAQ,KAAK,wBAAwB,iBAAiB,aAAa;GACvE;GACA;GACD,CAAC;AACF,OAAK,mBAAmB,IAAI,aAAa,MAAM;AAC/C,SAAO,KAAK,iBAAiB,OAAO,UAAU,QAAQ,MAAM;;CAG9D,AAAQ,iBACN,OACA,UACA,QACA,MACe;AACf,MAAI,QAAQ,QAAS,OAAM,eAAe,UAAU;AACpD,MAAI,KAAM,OAAM;AAChB,QAAM,YAAY,IAAI,SAAS;AAC/B,MAAI,MAAM,WAAY,UAAS,MAAM,WAAW;AAChD,SAAO,KAAK,wBAAwB,OAAO,OAAO;;CAGpD,AAAQ,wBACN,iBACA,aACA,MAC4B;EAC5B,MAAM,mBAAmB,IAAI,iBAAiB;EAC9C,MAAM,QAAoC;GACxC,UAAU;GACV;GACA,6BAAa,IAAI,KAAK;GACtB,YAAY;GACZ,sBAAsB;GACtB,SAAS,QAAQ,SAAS;GAC3B;AAED,QAAM,UAAU,KAAK,UAClB,gBACC,sBACA;GACE,MAAM,SAAS;GACf,YAAY;IACV,aAAa;IACb,mBAAmB;IACnB,mCAAmC,KAAK;IACzC;GACF,EACD,OAAO,SAAe;AACpB,OAAI;AACF,UAAM,KAAK,2BACT,iBACA,aACA;KACE,WAAW,WAAW,KAAK,oBAAoB,OAAO,OAAO;KAC7D,QAAQ,iBAAiB;KACzB,WAAW,KAAK;KAChB,WAAW,KAAK;KAChB,8BAA8B;AAC5B,YAAM,uBAAuB;;KAEhC,EACD,KACD;AACD,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;YACpC,OAAO;AACd,SAAK,8BAA8B,OAAO,aAAa,KAAK;aACpD;AACR,SAAK,KAAK;;KAGd;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC,CACA,cAAc;AACb,OAAI,KAAK,mBAAmB,IAAI,YAAY,KAAK,MAC/C,MAAK,mBAAmB,OAAO,YAAY;IAE7C;AAEJ,QAAM,QAAQ,YAAY,GAAG;AAC7B,SAAO;;CAGT,AAAQ,oBACN,OACA,QACM;AACN,QAAM,aAAa;AACnB,OAAK,MAAM,cAAc,MAAM,YAC7B,KAAI;AACF,cAAW,OAAO;UACZ;;;CAOZ,AAAQ,wBACN,OACA,cACe;AACf,MAAI,CAAC,aAAc,QAAO,MAAM;AAEhC,SAAO,IAAI,SAAe,SAAS,WAAW;GAC5C,IAAI,OAAO;GACX,MAAM,UAAU,SAAiC,UAAoB;AACnE,QAAI,KAAM;AACV,WAAO;AACP,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,QAAI,YAAY,KAAM,UAAS;aACtB,YAAY,QAAS,QAAO,eAAe,UAAU,CAAC;QAC1D,QAAO,MAAM;;GAEpB,MAAM,gBAAgB;AACpB,SAAK,wBAAwB,MAAM;AACnC,WAAO,QAAQ;;AAEjB,gBAAa,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;AAC/D,SAAM,QAAQ,WACN,OAAO,KAAK,GACjB,UAAU,OAAO,OAAO,MAAM,CAChC;IACD;;CAGJ,AAAQ,wBAAwB,OAAyC;AACvE,MAAI,MAAM,WAAW,EAAG,OAAM;AAC9B,MAAI,MAAM,WAAW,KAAK,MAAM,iBAAiB,OAAO,QAAS;AAEjE,MAAI,MAAM,qBAER;AAKF,uBAAqB;AACnB,OAAI,MAAM,YAAY,KAAK,CAAC,MAAM,iBAAiB,OAAO,QACxD,OAAM,iBAAiB,MAAM,0CAA0C;IAEzE;;CAGJ,MAAc,2BACZ,iBACA,aACA,MACA,MACe;EACf,MAAM,EACJ,UACA,QACA,YAAY,sCACZ,YAAY,MACZ,2BACE;EACJ,MAAM,YAAY,KAAK,KAAK;EAC5B,MAAM,UAAU,IAAI,uBAAuB,MAAM,WAAW,SAAS;EACrE,IAAI,WAAW;EACf,MAAM,UAAU,IAAI,sBAAsB;AAE1C,SAAO,MAAM;AACX,OAAI,QAAQ,QAAS,OAAM,eAAe,UAAU;AACpD,OAAI,KAAK,KAAK,GAAG,YAAY,UAC3B,OAAM,eAAe,gBACnB,8CAA8C,UAAU,IACzD;GAGH,MAAM,OAAO,MAAM,gBAAgB,WAAW,IAC5C,EAAE,IAAI,aAAa,EACnB,KAAK,eAAe,OAAO,CAC5B;GACD,MAAM,QAAQ,MAAM;GACpB,MAAM,UAAU,MAAM,QAAQ;AAE9B,WAAQ,OAAR;IACE,KAAK;AACH,aAAQ,KAAK,OAAO,QAAQ;AAC5B,UAAK,iBAAiB,IAAI,aAAa,KAAK,KAAK,CAAC;AAClD,UAAK,aAAa,yBAAyB,QAAQ,QAAQ;AAC3D;IACF,KAAK;IACL,KAAK,WACH,OAAM,mBAAmB,iBACvB,gBACA,mCAAmC,MAAM,mEAC1C;IACH,KAAK;AACH,SAAI,CAAC,UACH,OAAM,IAAI,mBACR,uJACD;AAEH,SAAI,CAAC,UAAU;AACb,cAAQ,KAAK,YAAY,QAAQ;AACjC,gCAA0B;AAC1B,YAAM,gBAAgB,WAAW,MAC/B,EAAE,IAAI,aAAa,EACnB,KAAK,eAAe,OAAO,CAC5B;AACD,iBAAW;WAEX,SAAQ,KAAK,OAAO,QAAQ;AAE9B;IACF,KAAK;IACL,KAAK;AACH,aAAQ,KAAK,OAAO,QAAQ;AAC5B;IACF,QACE,OAAM,eAAe,aAAa,OAAO,SAAS,UAAU,CAAC;;GAGjE,MAAM,UAAU,QAAQ,MAAM;AAC9B,OAAI,KAAK,KAAK,GAAG,UAAU,aAAa,UACtC,OAAM,eAAe,gBACnB,8CAA8C,UAAU,IACzD;AAEH,SAAM,KAAK,sBAAsB,SAAS,OAAO;;;;;;;CAQrD,AAAQ,mBAAmB,aAA8B;EACvD,MAAM,aAAa,KAAK,iBAAiB,IAAI,YAAY;AACzD,SACE,eAAe,UACf,KAAK,KAAK,GAAG,aAAa;;;;;;;;;;;;;CAe9B,AAAQ,8BACN,OACA,aACA,MACO;AACP,MAAI,iBAAiB,aAAa;AAChC,QAAK,gBAAgB,MAAM;AAC3B,QAAK,UAAU;IAAE,MAAM,eAAe;IAAO,SAAS,MAAM;IAAM,CAAC;AACnE,SAAM;;AAER,OAAK,gBACH,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,MAAM,CAAC,CAC1D;AACD,SAAO,MACL,kDACA,aACA,MACD;EACD,MAAM,UAAU,eAAe,gBAC7B,mCACD;AACD,OAAK,UAAU;GAAE,MAAM,eAAe;GAAO,SAAS,QAAQ;GAAM,CAAC;AACrE,QAAM;;;;;;;;CASR,AAAQ,sBACN,IACA,QACe;AACf,MAAI,QAAQ,QAAS,QAAO,QAAQ,OAAO,eAAe,UAAU,CAAC;AACrE,SAAO,IAAI,SAAe,SAAS,WAAW;GAC5C,MAAM,QAAQ,iBAAiB;AAC7B,YAAQ,oBAAoB,SAAS,QAAQ;AAC7C,aAAS;MACR,GAAG;GACN,MAAM,gBAAgB;AACpB,iBAAa,MAAM;AACnB,WAAO,eAAe,UAAU,CAAC;;AAEnC,WAAQ,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;IAC1D;;CAGJ,MAAc,wBACZ,iBACA,aACA,UAAU,yBAAyB,SACnC,QACA;AACA,SAAO,KAAK,UAAU,gBACpB,YACA,EACE,YAAY;GACV,mBAAmB;GACnB,sBAAsB;GACvB,EACF,EACD,OAAO,SAAe;AACpB,OAAI;IACF,MAAM,YAAY,KAAK,KAAK;IAC5B,IAAI,QAAQ;IACZ,MAAM,uBAAuB;IAC7B,IAAI,YAAY;AAEhB,WAAO,MAAM;AACX;AACA,UAAK,aAAa,8BAA8B,UAAU;KAG1D,MAAM,cAAc,KAAK,KAAK,GAAG;AACjC,SAAI,cAAc,SAAS;MACzB,MAAM,QAAQ,eAAe,gBAC3B,kCAAkC,QAAQ,eAAe,YAAY,KACtE;AACD,WAAK,gBAAgB,MAAM;AAC3B,WAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAC9C,YAAM;;AAGR,SAAI,QAAQ,SAAS;MACnB,MAAM,QAAQ,eAAe,UAAU;AACvC,WAAK,gBAAgB,MAAM;AAC3B,WAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAC9C,YAAM;;AAGR,UAAK,SAAS,mBAAmB;MAC/B,gBAAgB;MAChB,iBAAiB;MACjB,mBAAmB;MACpB,CAAC;KAEF,MAAM,WACJ,MAAM,gBAAgB,mBAAmB,aACvC,EACE,cAAc,aACf,EACD,KAAK,eAAe,OAAO,CAC5B;AACH,SAAI,CAAC,SACH,OAAM,gBAAgB,WAAW,gBAAgB;KAGnD,MAAM,SAAS,SAAS;AAExB,UAAK,SAAS,wBAAwB;MACpC,aAAa,QAAQ;MACrB,gBAAgB;MACjB,CAAC;AAEF,aAAQ,QAAQ,OAAhB;MACE,KAAK;MACL,KAAK,UAEH;MACF,KAAK;AACH,YAAK,aAAa,uBAAuB,UAAU;AACnD,YAAK,aAAa,gCAAgC,YAAY;AAC9D,YAAK,SAAS,qBAAqB;QACjC,iBAAiB;QACjB,oBAAoB;QACrB,CAAC;AACF,YAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,cAAO,KAAK,oBACV,UACA,iBACA,OACD;MACH,KAAK,SACH,OAAM,eAAe,gBACnB,OAAO,OAAO,SACd,OAAO,OAAO,WACf;MACH,KAAK,WACH,OAAM,eAAe,UAAU;MACjC,KAAK,SACH,OAAM,eAAe,eAAe;MACtC,QACE,OAAM,eAAe,aACnB,OAAO,QAAQ,SAAS,UAAU,CACnC;;AAIL,WAAM,IAAI,SAAS,YAAY,WAAW,SAAS,MAAM,CAAC;AAC1D,aAAQ,KAAK,IAAI,QAAQ,GAAG,qBAAqB;;YAE5C,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU;KACb,MAAM,eAAe;KACrB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;KAChE,CAAC;AAGF,SAAK,uBAAuB,MAAM;aAC1B;AACR,SAAK,KAAK;;KAGd;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;CAGH,MAAc,oBACZ,UACA,iBACA,QACA;AACA,MAAI,SAAS,UAAU,WAAW,gBAAgB;GAChD,MAAM,SAAS,SAAS;AAQxB,OAAI,QAAQ,WACV,QAAO,KAAK,yBAAyB,UAAU,OAAO,WAAW;AAQnE,OAAI,QAAQ,kBAAkB,OAAO,eAAe,SAAS,EAC3D,QAAO,KAAK,sBAAsB,UAAU,iBAAiB,OAAO;AAatE,QAFE,CAAC,QAAQ,cACR,MAAM,QAAQ,OAAO,WAAW,IAAI,OAAO,WAAW,WAAW,MACnD,SAAS,UAAU,QAAQ,SAAS;IACnD,MAAM,cAAc,yBAClB,SAAS,SAAS,OAAO,QAC1B;AACD,WAAO;KACL,GAAG;KACH,QAAQ;MAAE,GAAI,UAAU,EAAE;MAAG,YAAY;MAAa;KACvD;;;AASL,MAAI,CAAC,SAAS,QAAQ,cAAc,CAAC,SAAS,UAAU,QAAQ,QAC9D,QAAO;EAGT,MAAM,UAAU,SAAS,SAAS,OAAO;EAEzC,MAAM,kBAAkB,SAAS,OAAO,WAAW,KAAK,QAAQ;GAC9D,MAAM,MAA+B,EAAE;AACvC,OAAI,SAAS,OAAO,UAAU;IAC5B,MAAM,SAAS,QAAQ;IACvB,MAAM,aAAa,QAAQ,QAAQ,UAAU;AAG7C,QACE,QAAQ,cAAc,YACtB,OAAO,UAAU,YACjB,UACC,MAAM,OAAO,OAAO,MAAM,OAAO,KAElC,KAAI;AACF,SAAI,cAAc,KAAK,MAAM,MAAM;YAC7B;AAEN,SAAI,cAAc;;QAGpB,KAAI,cAAc;KAEpB;AACF,UAAO;IACP;EAGF,MAAM,EAAE,YAAY,aAAa,GAAG,eAAe,SAAS;AAC5D,SAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAG;IACH,MAAM;IACP;GACF;;;;;;;;;;;;;CAcH,AAAQ,yBACN,UACA,YACA;EAQA,MAAM,WAAW,WAAW,QAAQ,QAAQ,GAAG;EAC/C,MAAM,UAAU,SAAS,SAAS,KAAK,GACnC,IACA,SAAS,SAAS,IAAI,GACpB,IACA;EACN,MAAM,cAAc,KAAK,MAAO,SAAS,SAAS,IAAK,EAAE,GAAG;AAC5D,MAAI,cAAc,4BAChB,OAAM,eAAe,gBACnB,iDAAiD,YAAY,KAAK,4BAA4B,SAC/F;EAGH,MAAM,cAAc,iBAAiB,SAAS;AAC9C,MAAI,YACF,QAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAI,SAAS;IAOb,cAAc,SAAS;IACvB;IACD;GACF;AAGH,SAAO;;CAGT,MAAc,sBACZ,UACA,iBACA,QASC;EACD,MAAM,cAAc,SAAS;AAC7B,SAAO,EACL,QAAQ;GACN,cAAc;GACd,QAAQ;IACN,OAAO,SAAS,QAAQ;IACxB,OAAO,SAAS,QAAQ;IACzB;GACD,aAAa,iBAAiB,SAAS;GAMvC,gBAAgB,MAAM,KAAK,yBACzB,iBACA,aACA,UACA,OACD;GAID,kBAAkB,KAAK,wBACrB,iBACA,YACD;GACF,EACF;;;;;;;;;;;;;CAcH,MAAc,yBACZ,iBACA,aACA,UACA,QACyC;EACzC,MAAM,QAAQ,SAAS,QAAQ;AAC/B,MAAI,CAAC,SAAS,MAAM,WAAW,EAAG,QAAO;EAEzC,MAAM,QAA4B,CAAC,GAAG,MAAM;EAM5C,MAAM,aACJ,SAAS,UAAU,qBAAqB;EAC1C,IAAI,OAAO,KAAK,gBAAgB,MAAM;AACtC,OAAK,IAAI,UAAU,GAAG,QAAQ,QAAQ,UAAU,YAAY,WAAW;AACrE,OAAI,QAAQ,QAAS,OAAM,eAAe,UAAU;GAMpD,MAAM,cAJJ,MAAM,gBAAgB,mBAAmB,yBACvC;IAAE,cAAc;IAAa,aAAa;IAAM,EAChD,KAAK,eAAe,OAAO,CAC5B,EACsB,kBAAkB,EAAE;AAC7C,OAAI,WAAW,WAAW,EAAG;AAC7B,SAAM,KAAK,GAAG,WAAW;AACzB,UAAO,KAAK,gBAAgB,WAAW;;AAEzC,SAAO;;;CAIT,AAAQ,gBAAgB,OAA+C;AACrE,OAAK,MAAM,QAAQ,MACjB,KAAI,KAAK,oBAAoB,KAAM,QAAO,KAAK;;;;;;;;;CAYnD,AAAQ,wBACN,iBACA,aACkB;AAClB,SAAO,OAAO,YAAY,WAAW;AAMnC,WAJE,MAAM,gBAAgB,mBAAmB,yBACvC;IAAE,cAAc;IAAa,aAAa;IAAY,EACtD,KAAK,eAAe,OAAO,CAC5B,EACU,gBAAgB,MAAM,MAAM,EAAE,gBAAgB,WAAW;;;;;;;;;;CAW1E,oBACE,QACA,QACA,SAC2C;AAC3C,SAAO,KAAK,eAAe,aAAa,QAAQ,QAAQ,QAAQ;;;;;;;;CASlE,MAAM,eACJ,iBACA,OACA,QAC+B;AAK/B,SAAO,iBAJU,MAAM,gBAAgB,mBAAmB,aACxD,EAAE,cAAc,OAAO,EACvB,KAAK,eAAe,OAAO,CAC5B,CACgC;;;;;;;;;;CAWnC,AAAQ,uBAAuB,OAAuB;AACpD,MAAI,iBAAiB,SAAS,MAAM,SAAS,aAC3C,OAAM;AAER,MAAI,iBAAiB,YACnB,OAAM;EAER,MAAM,eACJ,SAAS,OAAO,UAAU,YAAY,eAAe,QAChD,MAAkC,YACnC;AACN,QAAM,eAAe,gBACnB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,EACtD,OAAO,iBAAiB,WAAW,eAAe,OACnD;;CAIH,AAAQ,eAAe,QAAsB;AAC3C,SAAO,IAAI,QAAQ,EACjB,mBAAmB;GACjB,yBAAyB,QAAQ,WAAW;GAC5C,0BAA0B,OAAmB;AAC3C,YAAQ,iBAAiB,SAAS,IAAI,EAAE,MAAM,MAAM,CAAC;;GAExD,EACF,CAAC"}
+{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/sql-warehouse/client.ts"],"sourcesContent":["import type { TelemetryOptions } from \"shared\";\n\nimport {\n  AppKitError,\n  ConfigurationError,\n  ConnectionError,\n  ExecutionError,\n  ValidationError,\n} from \"../../errors\";\nimport { createLogger } from \"../../logging/logger\";\nimport {\n  ArrowStreamProcessor,\n  type RefreshChunkLink,\n} from \"../../stream/arrow-stream-processor\";\nimport type { TelemetryProvider } from \"../../telemetry\";\nimport {\n  type Counter,\n  type Histogram,\n  type Span,\n  SpanKind,\n  SpanStatusCode,\n  TelemetryManager,\n} from \"../../telemetry\";\nimport {\n  Context,\n  type sql,\n  type WorkspaceClient,\n} from \"../../workspace-client\";\nimport { buildEmptyArrowIPCBase64 } from \"./arrow-schema\";\nimport { executeStatementDefaults } from \"./defaults\";\nimport { WarehousePollBackoff } from \"./warehouse-poll-backoff\";\nimport { WarehouseStatusEmitter } from \"./warehouse-status-emitter\";\n\nconst logger = createLogger(\"connectors:sql-warehouse\");\n\n/**\n * Real column names from a statement's result manifest (positional\n * `column_N` fallback for any blank ones). Databricks encodes ARROW_STREAM\n * schemas positionally (`col_0`, …); these names let consumers relabel an\n * Arrow result to match the JSON path. Returns `undefined` when the manifest\n * carries no columns.\n */\nfunction arrowColumnNames(\n  response: sql.StatementResponse,\n): string[] | undefined {\n  const cols = response.manifest?.schema?.columns;\n  if (!cols || cols.length === 0) return undefined;\n  return cols.map((c, i) =>\n    c.name && c.name.length > 0 ? c.name : `column_${i}`,\n  );\n}\n\n/**\n * Maximum size for inline Arrow IPC attachments (25 MiB decoded — the\n * Databricks Statement Execution API hard cap on INLINE responses).\n *\n * The analytics route streams the decoded attachment straight back on the\n * query response body; this cap bounds the single in-memory copy. Larger\n * results fall through to `disposition: \"EXTERNAL_LINKS\"`, which streams\n * chunk-by-chunk and is not bound by this limit.\n */\nconst MAX_INLINE_ATTACHMENT_BYTES = 25 * 1024 * 1024;\n\n/**\n * Safety cap on how many additional EXTERNAL_LINKS chunks\n * {@link SQLWarehouseConnector._resolveAllExternalLinks} will follow when the\n * manifest omits `total_chunk_count`. High enough to cover any real result;\n * only bounds a misbehaving warehouse with a cyclic `next_chunk_index`.\n */\nconst MAX_EXTERNAL_CHUNK_FOLLOWS = 10_000;\n\ninterface SQLWarehouseConfig {\n  timeout?: number;\n  telemetry?: TelemetryOptions;\n}\n\n/**\n * Default ceiling for how long {@link SQLWarehouseConnector.ensureWarehouseRunning}\n * will wait for a warehouse to reach the RUNNING state before giving up.\n *\n * Five minutes covers a cold-start of a classic warehouse on most workspaces;\n * serverless typically reaches RUNNING within ~30s.\n */\nexport const DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS = 5 * 60 * 1000;\n\n/**\n * Window during which a recent `RUNNING` observation lets subsequent calls\n * to {@link SQLWarehouseConnector.ensureWarehouseRunning} short-circuit\n * without making any SDK calls. Sized to roughly the upper bound of how\n * long Databricks keeps a warehouse \"stickily\" available between requests\n * — past that, we re-verify.\n */\nconst WAREHOUSE_RUNNING_CACHE_TTL_MS = 30_000;\n\n/**\n * A single observation of the warehouse state, emitted by\n * {@link SQLWarehouseConnector.ensureWarehouseRunning} so callers can stream\n * progress to clients (e.g. over SSE).\n *\n * Note: `health.summary` from the SDK is intentionally NOT exposed on this\n * type. It's free-form operator-oriented diagnostic text (cluster IDs,\n * capacity-failure reasons, internal RPC errors) that must not reach end\n * users. The raw value stays in the OTel span attributes for server-side\n * debugging only.\n */\nexport interface WarehouseStatusUpdate {\n  /** Current state from the SDK (RUNNING | STARTING | STOPPED | STOPPING | DELETED | DELETING). */\n  state: sql.State;\n  /** Milliseconds elapsed since `ensureWarehouseRunning` was called. */\n  elapsedMs: number;\n  /** 1-based attempt counter — useful for tests and telemetry. */\n  attempt: number;\n}\n\n/** Options for {@link SQLWarehouseConnector.ensureWarehouseRunning}. */\ninterface EnsureWarehouseRunningOptions {\n  /** Invoked every time the warehouse state changes during the wait. */\n  onStatus: (update: WarehouseStatusUpdate) => void;\n  /** Aborts the wait. The connector treats abort as `ExecutionError.canceled()`. */\n  signal?: AbortSignal;\n  /** Hard ceiling on the total wait. Defaults to {@link DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS}. */\n  timeoutMs?: number;\n  /**\n   * When `true` (default), a `STOPPED` warehouse is auto-started.\n   * When `false`, `STOPPED` surfaces as `ConfigurationError` immediately —\n   * useful for cost-controlled deployments that don't want any caller to\n   * trigger billable warehouse starts.\n   */\n  autoStart?: boolean;\n  /** @internal Fired once when `warehouses.start` is issued for this poll. */\n  onWarehouseStartIssued?: () => void;\n}\n\n/** In-flight warehouse readiness work shared across concurrent callers. */\ninterface WarehouseReadinessInFlight {\n  promise: Promise<void>;\n  refCount: number;\n  sharedController: AbortController;\n  subscribers: Set<(update: WarehouseStatusUpdate) => void>;\n  /** Last emitted update; replayed to late joiners. */\n  lastUpdate: WarehouseStatusUpdate | null;\n  /** `true` once this poll has called `warehouses.start`. */\n  warehouseStartIssued: boolean;\n}\n\nexport class SQLWarehouseConnector {\n  private readonly name = \"sql-warehouse\";\n\n  private config: SQLWarehouseConfig;\n\n  // Lazy-initialized: only created when Arrow format is used\n  private _arrowProcessor: ArrowStreamProcessor | null = null;\n\n  /**\n   * Per-warehouse cache of the last RUNNING observation timestamp. Used by\n   * {@link ensureWarehouseRunning} to short-circuit warm-path callers; see\n   * {@link WAREHOUSE_RUNNING_CACHE_TTL_MS}.\n   */\n  private _recentlyRunning = new Map<string, number>();\n  /** Per-warehouse readiness singleflight — one poll loop per cold start. */\n  private _readinessInFlight = new Map<string, WarehouseReadinessInFlight>();\n  // telemetry\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    queryCount: Counter;\n    queryDuration: Histogram;\n  };\n\n  constructor(config: SQLWarehouseConfig) {\n    this.config = config;\n\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      queryCount: this.telemetry.getMeter().createCounter(\"query.count\", {\n        description: \"Total number of queries executed\",\n        unit: \"1\",\n      }),\n      queryDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"query.duration\", {\n          description: \"Duration of queries executed\",\n          unit: \"ms\",\n        }),\n    };\n  }\n\n  /**\n   * Lazily initializes and returns the ArrowStreamProcessor.\n   * Only created on first Arrow format query to avoid unnecessary allocation.\n   */\n  private get arrowProcessor(): ArrowStreamProcessor {\n    if (!this._arrowProcessor) {\n      this._arrowProcessor = new ArrowStreamProcessor({\n        timeout: this.config.timeout || executeStatementDefaults.timeout,\n        retries: ArrowStreamProcessor.DEFAULT_RETRIES,\n      });\n    }\n    return this._arrowProcessor;\n  }\n\n  async executeStatement(\n    workspaceClient: WorkspaceClient,\n    input: sql.ExecuteStatementRequest,\n    signal?: AbortSignal,\n  ) {\n    const startTime = Date.now();\n    let success = false;\n\n    // if signal is aborted, throw an error\n    if (signal?.aborted) {\n      throw ExecutionError.canceled();\n    }\n\n    return this.telemetry.startActiveSpan(\n      \"sql.query\",\n      {\n        kind: SpanKind.CLIENT,\n        attributes: {\n          \"db.system\": \"databricks\",\n          \"db.warehouse_id\": input.warehouse_id || \"\",\n          \"db.catalog\": input.catalog ?? \"\",\n          \"db.schema\": input.schema ?? \"\",\n          \"db.statement\": input.statement?.substring(0, 500) || \"\",\n          \"db.has_parameters\": !!input.parameters,\n        },\n      },\n      async (span: Span) => {\n        let abortHandler: (() => void) | undefined;\n        let isAborted = false;\n\n        if (signal) {\n          abortHandler = () => {\n            // abort span if not recording\n            if (!span.isRecording()) return;\n            isAborted = true;\n            span.setAttribute(\"cancelled\", true);\n            span.setStatus({\n              code: SpanStatusCode.ERROR,\n              message: \"Query cancelled by client\",\n            });\n            span.end();\n          };\n          signal.addEventListener(\"abort\", abortHandler, { once: true });\n        }\n\n        try {\n          // validate required fields\n          if (!input.statement) {\n            throw ValidationError.missingField(\"statement\");\n          }\n\n          if (!input.warehouse_id) {\n            throw ValidationError.missingField(\"warehouse_id\");\n          }\n\n          const body: sql.ExecuteStatementRequest = {\n            statement: input.statement,\n            parameters: input.parameters,\n            warehouse_id: input.warehouse_id,\n            catalog: input.catalog,\n            schema: input.schema,\n            wait_timeout:\n              input.wait_timeout || executeStatementDefaults.wait_timeout,\n            disposition:\n              input.disposition || executeStatementDefaults.disposition,\n            format: input.format || executeStatementDefaults.format,\n            byte_limit: input.byte_limit,\n            row_limit: input.row_limit,\n            on_wait_timeout:\n              input.on_wait_timeout || executeStatementDefaults.on_wait_timeout,\n          };\n\n          span.addEvent(\"statement.submitting\", {\n            \"db.warehouse_id\": input.warehouse_id,\n          });\n\n          const response =\n            await workspaceClient.statementExecution.executeStatement(\n              body,\n              this._createContext(signal),\n            );\n\n          if (!response) {\n            throw ConnectionError.apiFailure(\"SQL Warehouse\");\n          }\n          const status = response.status;\n          const statementId = response.statement_id as string;\n\n          span.setAttribute(\"db.statement_id\", statementId);\n          span.addEvent(\"statement.submitted\", {\n            \"db.statement_id\": response.statement_id,\n            \"db.status\": status?.state,\n          });\n\n          let result:\n            | sql.StatementResponse\n            | { result: { statement_id: string; status: sql.StatementStatus } };\n\n          switch (status?.state) {\n            case \"RUNNING\":\n            case \"PENDING\":\n              span.addEvent(\"statement.polling_started\", {\n                \"db.status\": response.status?.state,\n              });\n              result = await this._pollForStatementResult(\n                workspaceClient,\n                statementId,\n                this.config.timeout,\n                signal,\n              );\n              break;\n            case \"SUCCEEDED\":\n              result = await this._transformDataArray(\n                response,\n                workspaceClient,\n                signal,\n              );\n              break;\n            case \"FAILED\":\n              throw ExecutionError.statementFailed(\n                status.error?.message,\n                status.error?.error_code,\n              );\n            case \"CANCELED\":\n              throw ExecutionError.canceled();\n            case \"CLOSED\":\n              throw ExecutionError.resultsClosed();\n            default:\n              throw ExecutionError.unknownState(\n                String(status?.state ?? \"unknown\"),\n              );\n          }\n\n          const resultData = result.result as any;\n          const rowCount =\n            resultData?.data?.length ?? resultData?.data_array?.length ?? 0;\n\n          if (rowCount > 0) {\n            span.setAttribute(\"db.result.row_count\", rowCount);\n          }\n\n          const duration = Date.now() - startTime;\n          logger.event()?.setContext(\"sql-warehouse\", {\n            warehouse_id: input.warehouse_id,\n            rows_returned: rowCount,\n            query_duration_ms: duration,\n          });\n\n          success = true;\n          // only set success status if not aborted\n          if (!isAborted) {\n            span.setStatus({ code: SpanStatusCode.OK });\n          }\n          return result;\n        } catch (error) {\n          // only record error if not already handled by abort\n          if (!isAborted) {\n            span.recordException(error as Error);\n            span.setStatus({\n              code: SpanStatusCode.ERROR,\n              message: error instanceof Error ? error.message : String(error),\n            });\n\n            logger.error(\n              \"Statement execution failed: %s\",\n              error instanceof Error ? error.message : String(error),\n            );\n          }\n\n          this._rethrowStatementError(error);\n        } finally {\n          // remove abort handler\n          if (abortHandler && signal) {\n            signal.removeEventListener(\"abort\", abortHandler);\n          }\n\n          const duration = Date.now() - startTime;\n\n          // end span if not already ended by abort handler\n          if (!isAborted) {\n            span.end();\n          }\n\n          const attributes = {\n            \"db.warehouse_id\": input.warehouse_id,\n            \"db.catalog\": input.catalog ?? \"\",\n            \"db.schema\": input.schema ?? \"\",\n            \"db.statement\": input.statement?.substring(0, 500) || \"\",\n            success: success.toString(),\n          };\n\n          this.telemetryMetrics.queryCount.add(1, attributes);\n          this.telemetryMetrics.queryDuration.record(duration, attributes);\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  /**\n   * Wait until the SQL warehouse is in the `RUNNING` state, auto-starting it\n   * if currently `STOPPED`. Emits a {@link WarehouseStatusUpdate} whenever\n   * the observed state changes, so callers can surface progress (e.g. over\n   * SSE) instead of letting the UI freeze on a cold warehouse. Equal\n   * successive observations are de-duplicated.\n   *\n   * Fast path: if this connector recently observed the warehouse RUNNING\n   * (within {@link WAREHOUSE_RUNNING_CACHE_TTL_MS}), the call returns\n   * immediately without any SDK round-trip or status emission. This keeps\n   * cache-hit analytics requests off the Databricks control plane.\n   *\n   * Concurrent callers for the same `warehouseId` share a single in-flight\n   * poll loop (singleflight): only the owner issues `get`/`start`; joiners\n   * receive broadcast `onStatus` updates and share the result promise.\n   * When every waiter disconnects before `warehouses.start`, the shared poll\n   * is aborted on the next microtask if still unclaimed (kills true orphans\n   * without a fixed grace window). After `start` is issued, the poll runs to\n   * completion even with no waiters.\n   *\n   * Behaviour by initial state:\n   * - `RUNNING`: emits one update, caches the observation, returns.\n   * - `STOPPED`: emits a synthetic `STARTING` update, calls\n   *   `workspaceClient.warehouses.start`, then polls until `RUNNING`.\n   *   When `autoStart: false`, throws `ConfigurationError` instead.\n   * - `STARTING` / `STOPPING`: polls until `RUNNING`.\n   * - `DELETED` / `DELETING`: throws `ConfigurationError.resourceNotFound`.\n   *\n   * Aborts and timeouts surface as `ExecutionError`. The poll loop uses\n   * exponential backoff with ±15% jitter (1s → 30s cap) and runs inside a\n   * `sql.warehouseReady` telemetry span.\n   */\n  async ensureWarehouseRunning(\n    workspaceClient: WorkspaceClient,\n    warehouseId: string,\n    opts: EnsureWarehouseRunningOptions,\n  ): Promise<void> {\n    const {\n      onStatus,\n      signal,\n      timeoutMs = DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS,\n      autoStart = true,\n    } = opts;\n\n    if (signal?.aborted) throw ExecutionError.canceled();\n    if (!warehouseId) throw ValidationError.missingField(\"warehouse_id\");\n    if (this._isRecentlyRunning(warehouseId)) return;\n\n    const existing = this._readinessInFlight.get(warehouseId);\n    if (existing && !existing.sharedController.signal.aborted) {\n      return this._waitOnReadiness(existing, onStatus, signal, true);\n    }\n\n    const entry = this._spawnReadinessInFlight(workspaceClient, warehouseId, {\n      timeoutMs,\n      autoStart,\n    });\n    this._readinessInFlight.set(warehouseId, entry);\n    return this._waitOnReadiness(entry, onStatus, signal, false);\n  }\n\n  private _waitOnReadiness(\n    entry: WarehouseReadinessInFlight,\n    onStatus: (update: WarehouseStatusUpdate) => void,\n    signal: AbortSignal | undefined,\n    join: boolean,\n  ): Promise<void> {\n    if (signal?.aborted) throw ExecutionError.canceled();\n    if (join) entry.refCount++;\n    entry.subscribers.add(onStatus);\n    if (entry.lastUpdate) onStatus(entry.lastUpdate);\n    return this._joinWarehouseReadiness(entry, signal);\n  }\n\n  private _spawnReadinessInFlight(\n    workspaceClient: WorkspaceClient,\n    warehouseId: string,\n    opts: { timeoutMs: number; autoStart: boolean },\n  ): WarehouseReadinessInFlight {\n    const sharedController = new AbortController();\n    const entry: WarehouseReadinessInFlight = {\n      refCount: 1,\n      sharedController,\n      subscribers: new Set(),\n      lastUpdate: null,\n      warehouseStartIssued: false,\n      promise: Promise.resolve(),\n    };\n\n    entry.promise = this.telemetry\n      .startActiveSpan(\n        \"sql.warehouseReady\",\n        {\n          kind: SpanKind.CLIENT,\n          attributes: {\n            \"db.system\": \"databricks\",\n            \"db.warehouse_id\": warehouseId,\n            \"db.warehouse.startup_timeout_ms\": opts.timeoutMs,\n          },\n        },\n        async (span: Span) => {\n          try {\n            await this._pollUntilWarehouseRunning(\n              workspaceClient,\n              warehouseId,\n              {\n                onStatus: (update) => this._broadcastReadiness(entry, update),\n                signal: sharedController.signal,\n                timeoutMs: opts.timeoutMs,\n                autoStart: opts.autoStart,\n                onWarehouseStartIssued: () => {\n                  entry.warehouseStartIssued = true;\n                },\n              },\n              span,\n            );\n            span.setStatus({ code: SpanStatusCode.OK });\n          } catch (error) {\n            this._throwSanitizedReadinessError(error, warehouseId, span);\n          } finally {\n            span.end();\n          }\n        },\n        { name: this.name, includePrefix: true },\n      )\n      .finally(() => {\n        if (this._readinessInFlight.get(warehouseId) === entry) {\n          this._readinessInFlight.delete(warehouseId);\n        }\n      });\n\n    entry.promise.catch(() => {});\n    return entry;\n  }\n\n  private _broadcastReadiness(\n    entry: WarehouseReadinessInFlight,\n    update: WarehouseStatusUpdate,\n  ): void {\n    entry.lastUpdate = update;\n    for (const subscriber of entry.subscribers) {\n      try {\n        subscriber(update);\n      } catch {\n        // One consumer must not block updates to the rest.\n      }\n    }\n  }\n\n  /** Ref-counted await; caller abort rejects locally. */\n  private _joinWarehouseReadiness(\n    entry: WarehouseReadinessInFlight,\n    callerSignal?: AbortSignal,\n  ): Promise<void> {\n    if (!callerSignal) return entry.promise;\n\n    return new Promise<void>((resolve, reject) => {\n      let done = false;\n      const finish = (outcome: \"ok\" | \"abort\" | \"err\", error?: unknown) => {\n        if (done) return;\n        done = true;\n        callerSignal.removeEventListener(\"abort\", onAbort);\n        if (outcome === \"ok\") resolve();\n        else if (outcome === \"abort\") reject(ExecutionError.canceled());\n        else reject(error);\n      };\n      const onAbort = () => {\n        this._releaseReadinessWaiter(entry);\n        finish(\"abort\");\n      };\n      callerSignal.addEventListener(\"abort\", onAbort, { once: true });\n      entry.promise.then(\n        () => finish(\"ok\"),\n        (error) => finish(\"err\", error),\n      );\n    });\n  }\n\n  private _releaseReadinessWaiter(entry: WarehouseReadinessInFlight): void {\n    if (entry.refCount > 0) entry.refCount--;\n    if (entry.refCount > 0 || entry.sharedController.signal.aborted) return;\n\n    if (entry.warehouseStartIssued) {\n      // Start already issued — finish warming for the process.\n      return;\n    }\n\n    // Defer abort to the next microtask so a synchronous StrictMode\n    // remount can rejoin before the shared poll is cancelled.\n    queueMicrotask(() => {\n      if (entry.refCount <= 0 && !entry.sharedController.signal.aborted) {\n        entry.sharedController.abort(\"all warehouse readiness waiters aborted\");\n      }\n    });\n  }\n\n  private async _pollUntilWarehouseRunning(\n    workspaceClient: WorkspaceClient,\n    warehouseId: string,\n    opts: EnsureWarehouseRunningOptions,\n    span: Span,\n  ): Promise<void> {\n    const {\n      onStatus,\n      signal,\n      timeoutMs = DEFAULT_WAREHOUSE_STARTUP_TIMEOUT_MS,\n      autoStart = true,\n      onWarehouseStartIssued,\n    } = opts;\n    const startTime = Date.now();\n    const emitter = new WarehouseStatusEmitter(span, startTime, onStatus);\n    let didStart = false;\n    const backoff = new WarehousePollBackoff();\n\n    while (true) {\n      if (signal?.aborted) throw ExecutionError.canceled();\n      if (Date.now() - startTime > timeoutMs) {\n        throw ExecutionError.statementFailed(\n          `SQL warehouse did not reach RUNNING within ${timeoutMs}ms`,\n        );\n      }\n\n      let info: Awaited<ReturnType<typeof workspaceClient.warehouses.get>>;\n      try {\n        info = await workspaceClient.warehouses.get(\n          { id: warehouseId },\n          this._createContext(signal),\n        );\n      } catch (error) {\n        // A real cancellation must still surface as canceled, not be swallowed\n        // as a probe failure.\n        if (\n          signal?.aborted ||\n          (error instanceof Error && error.name === \"AbortError\")\n        ) {\n          throw ExecutionError.canceled();\n        }\n        // The status probe itself failed for a non-abort reason — e.g. the\n        // caller can submit statements to this warehouse but lacks CAN_VIEW to\n        // read its status, or a transient control-plane error. Readiness is a\n        // UX/auto-start optimization, not a correctness gate: the Statement\n        // Execution API auto-starts and waits for the warehouse on its own. So\n        // rather than block the query on an unobservable status, record the\n        // probe failure and let execution proceed.\n        span.addEvent(\"warehouse.status_probe_failed\", {\n          \"db.warehouse_id\": warehouseId,\n          \"error.message\":\n            error instanceof Error ? error.message : String(error),\n        });\n        logger.debug(\n          \"Warehouse status probe failed for %s; skipping readiness gate and proceeding: %O\",\n          warehouseId,\n          error,\n        );\n        span.setAttribute(\"db.warehouse.status_probe_failed\", true);\n        return;\n      }\n      const state = info?.state;\n      const summary = info?.health?.summary;\n\n      switch (state) {\n        case \"RUNNING\":\n          emitter.emit(state, summary);\n          this._recentlyRunning.set(warehouseId, Date.now());\n          span.setAttribute(\"db.warehouse.attempts\", emitter.attempt);\n          return;\n        case \"DELETED\":\n        case \"DELETING\":\n          throw ConfigurationError.resourceNotFound(\n            \"Warehouse ID\",\n            `The configured SQL warehouse is ${state}. Update DATABRICKS_WAREHOUSE_ID to point at an active warehouse.`,\n          );\n        case \"STOPPED\":\n          if (!autoStart) {\n            throw new ConfigurationError(\n              \"The configured SQL warehouse is STOPPED and analytics auto-start is disabled. Start the warehouse manually or set analytics.autoStartWarehouse=true.\",\n            );\n          }\n          if (!didStart) {\n            emitter.emit(\"STARTING\", summary);\n            onWarehouseStartIssued?.();\n            await workspaceClient.warehouses.start(\n              { id: warehouseId },\n              this._createContext(signal),\n            );\n            didStart = true;\n          } else {\n            emitter.emit(state, summary);\n          }\n          break;\n        case \"STARTING\":\n        case \"STOPPING\":\n          emitter.emit(state, summary);\n          break;\n        default:\n          throw ExecutionError.unknownState(String(state ?? \"unknown\"));\n      }\n\n      const sleepMs = backoff.next();\n      if (Date.now() + sleepMs - startTime >= timeoutMs) {\n        throw ExecutionError.statementFailed(\n          `SQL warehouse did not reach RUNNING within ${timeoutMs}ms`,\n        );\n      }\n      await this._sleepRespectingAbort(sleepMs, signal);\n    }\n  }\n\n  /**\n   * `true` if this connector observed `warehouseId` in the RUNNING state\n   * within the recent-cache TTL.\n   */\n  private _isRecentlyRunning(warehouseId: string): boolean {\n    const observedAt = this._recentlyRunning.get(warehouseId);\n    return (\n      observedAt !== undefined &&\n      Date.now() - observedAt < WAREHOUSE_RUNNING_CACHE_TTL_MS\n    );\n  }\n\n  /**\n   * Final landing for any error thrown by the readiness poll loop. Records\n   * the original on the span and (at debug level) on the logger, then\n   * rethrows either the structured AppKitError unchanged or a curated\n   * ExecutionError — never the raw SDK message, which can contain operator\n   * internals.\n   *\n   * The `logger.debug` covers environments where OTel isn't configured\n   * (common in dev) so the raw error isn't lost just because no span\n   * exporter is hooked up.\n   */\n  private _throwSanitizedReadinessError(\n    error: unknown,\n    warehouseId: string,\n    span: Span,\n  ): never {\n    if (error instanceof AppKitError) {\n      span.recordException(error);\n      span.setStatus({ code: SpanStatusCode.ERROR, message: error.code });\n      throw error;\n    }\n    span.recordException(\n      error instanceof Error ? error : new Error(String(error)),\n    );\n    logger.debug(\n      \"Warehouse readiness check raw error for %s: %O\",\n      warehouseId,\n      error,\n    );\n    const wrapped = ExecutionError.statementFailed(\n      \"Warehouse readiness check failed\",\n    );\n    span.setStatus({ code: SpanStatusCode.ERROR, message: wrapped.code });\n    throw wrapped;\n  }\n\n  /**\n   * Sleep for `ms` milliseconds, but resolve early (and reject with\n   * `ExecutionError.canceled()`) if `signal` aborts mid-sleep. Used by the\n   * warehouse readiness loop so that client disconnects don't keep the loop\n   * polling for the full interval.\n   */\n  private _sleepRespectingAbort(\n    ms: number,\n    signal?: AbortSignal,\n  ): Promise<void> {\n    if (signal?.aborted) return Promise.reject(ExecutionError.canceled());\n    return new Promise<void>((resolve, reject) => {\n      const timer = setTimeout(() => {\n        signal?.removeEventListener(\"abort\", onAbort);\n        resolve();\n      }, ms);\n      const onAbort = () => {\n        clearTimeout(timer);\n        reject(ExecutionError.canceled());\n      };\n      signal?.addEventListener(\"abort\", onAbort, { once: true });\n    });\n  }\n\n  private async _pollForStatementResult(\n    workspaceClient: WorkspaceClient,\n    statementId: string,\n    timeout = executeStatementDefaults.timeout,\n    signal?: AbortSignal,\n  ) {\n    return this.telemetry.startActiveSpan(\n      \"sql.poll\",\n      {\n        attributes: {\n          \"db.statement_id\": statementId,\n          \"db.polling.timeout\": timeout,\n        },\n      },\n      async (span: Span) => {\n        try {\n          const startTime = Date.now();\n          let delay = 1000;\n          const maxDelayBetweenPolls = 5000; // max 5 seconds between polls\n          let pollCount = 0;\n\n          while (true) {\n            pollCount++;\n            span.setAttribute(\"db.polling.current_attempt\", pollCount);\n\n            // check if timeout exceeded\n            const elapsedTime = Date.now() - startTime;\n            if (elapsedTime > timeout) {\n              const error = ExecutionError.statementFailed(\n                `Polling timeout exceeded after ${timeout}ms (elapsed: ${elapsedTime}ms)`,\n              );\n              span.recordException(error);\n              span.setStatus({ code: SpanStatusCode.ERROR });\n              throw error;\n            }\n\n            if (signal?.aborted) {\n              const error = ExecutionError.canceled();\n              span.recordException(error);\n              span.setStatus({ code: SpanStatusCode.ERROR });\n              throw error;\n            }\n\n            span.addEvent(\"polling.attempt\", {\n              \"poll.attempt\": pollCount,\n              \"poll.delay_ms\": delay,\n              \"poll.elapsed_ms\": elapsedTime,\n            });\n\n            const response =\n              await workspaceClient.statementExecution.getStatement(\n                {\n                  statement_id: statementId,\n                },\n                this._createContext(signal),\n              );\n            if (!response) {\n              throw ConnectionError.apiFailure(\"SQL Warehouse\");\n            }\n\n            const status = response.status;\n\n            span.addEvent(\"polling.status_check\", {\n              \"db.status\": status?.state,\n              \"poll.attempt\": pollCount,\n            });\n\n            switch (status?.state) {\n              case \"PENDING\":\n              case \"RUNNING\":\n                // continue polling\n                break;\n              case \"SUCCEEDED\":\n                span.setAttribute(\"db.polling.attempts\", pollCount);\n                span.setAttribute(\"db.polling.total_duration_ms\", elapsedTime);\n                span.addEvent(\"polling.completed\", {\n                  \"poll.attempts\": pollCount,\n                  \"poll.duration_ms\": elapsedTime,\n                });\n                span.setStatus({ code: SpanStatusCode.OK });\n                return this._transformDataArray(\n                  response,\n                  workspaceClient,\n                  signal,\n                );\n              case \"FAILED\":\n                throw ExecutionError.statementFailed(\n                  status.error?.message,\n                  status.error?.error_code,\n                );\n              case \"CANCELED\":\n                throw ExecutionError.canceled();\n              case \"CLOSED\":\n                throw ExecutionError.resultsClosed();\n              default:\n                throw ExecutionError.unknownState(\n                  String(status?.state ?? \"unknown\"),\n                );\n            }\n\n            // continue polling after delay\n            await new Promise((resolve) => setTimeout(resolve, delay));\n            delay = Math.min(delay * 2, maxDelayBetweenPolls);\n          }\n        } catch (error) {\n          span.recordException(error as Error);\n          span.setStatus({\n            code: SpanStatusCode.ERROR,\n            message: error instanceof Error ? error.message : String(error),\n          });\n\n          // error logging is handled by executeStatement's catch block (gated on isAborted)\n          this._rethrowStatementError(error);\n        } finally {\n          span.end();\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  private async _transformDataArray(\n    response: sql.StatementResponse,\n    workspaceClient: WorkspaceClient,\n    signal?: AbortSignal,\n  ) {\n    if (response.manifest?.format === \"ARROW_STREAM\") {\n      const result = response.result as\n        | (sql.ResultData & { attachment?: string })\n        | undefined;\n\n      // Inline Arrow: pass the base64 IPC attachment through unmodified so\n      // the analytics route can stream it to the client, where the existing\n      // ArrowClient infrastructure decodes it into a Table. Validate size\n      // here to fail fast on runaway payloads.\n      if (result?.attachment) {\n        return this._validateArrowAttachment(response, result.attachment);\n      }\n\n      // External links: data fetched separately via statement_id. An empty\n      // array is a zero-row result (some warehouses emit `external_links: []`\n      // rather than omitting it) — it must NOT go down the streaming path\n      // (`streamChunks([])` rejects), so fall through to synthesize an empty\n      // Arrow table below.\n      if (result?.external_links && result.external_links.length > 0) {\n        return this.updateWithArrowStatus(response, workspaceClient, signal);\n      }\n\n      // Empty result with a known schema: synthesize a zero-row Arrow IPC\n      // attachment so the client always receives an Arrow Table for\n      // ARROW_STREAM, regardless of whether the warehouse returned data.\n      // Note: an empty array (`data_array: []`) is truthy, so length-check\n      // explicitly — otherwise zero-row responses fall through to the JSON\n      // row transform below and return `[]` JSON rows instead of an Arrow\n      // table.\n      const hasNoRows =\n        !result?.data_array ||\n        (Array.isArray(result.data_array) && result.data_array.length === 0);\n      if (hasNoRows && response.manifest?.schema?.columns) {\n        const synthesized = buildEmptyArrowIPCBase64(\n          response.manifest.schema.columns,\n        );\n        return {\n          ...response,\n          result: { ...(result ?? {}), attachment: synthesized },\n        };\n      }\n\n      // Inline data_array under ARROW_STREAM (rare): fall through to the\n      // row transform below. The hook will receive `type: \"result\"` rows;\n      // callers asking for ARROW_STREAM should not hit this path with\n      // current Databricks warehouses.\n    }\n\n    if (!response.result?.data_array || !response.manifest?.schema?.columns) {\n      return response;\n    }\n\n    const columns = response.manifest.schema.columns;\n\n    const transformedData = response.result.data_array.map((row) => {\n      const obj: Record<string, unknown> = {};\n      row.forEach((value, index) => {\n        const column = columns[index];\n        const columnName = column?.name || `column_${index}`;\n\n        // attempt to parse JSON strings for string columns\n        if (\n          column?.type_name === \"STRING\" &&\n          typeof value === \"string\" &&\n          value &&\n          (value[0] === \"{\" || value[0] === \"[\")\n        ) {\n          try {\n            obj[columnName] = JSON.parse(value);\n          } catch {\n            // if parsing fails, keep as string\n            obj[columnName] = value;\n          }\n        } else {\n          obj[columnName] = value;\n        }\n      });\n      return obj;\n    });\n\n    // remove data_array\n    const { data_array: _data_array, ...restResult } = response.result;\n    return {\n      ...response,\n      result: {\n        ...restResult,\n        data: transformedData,\n      },\n    };\n  }\n\n  /**\n   * Validate a base64 Arrow IPC attachment (INLINE ARROW_STREAM) and attach\n   * the real column names from the result manifest.\n   *\n   * Databricks warehouses encode the Arrow schema positionally (`col_0`, …);\n   * the real, aliased names live only in `manifest.schema.columns`. Rather\n   * than decode/re-encode the payload server-side, we carry the names on the\n   * result (`columnNames`) so the route can hand them to the client in a\n   * response header and the client relabels the decoded Table — the one\n   * mechanism used for both INLINE and EXTERNAL_LINKS.\n   */\n  private _validateArrowAttachment(\n    response: sql.StatementResponse,\n    attachment: string,\n  ) {\n    // Cap the size to protect against unbounded inline payloads from\n    // misbehaving warehouses. `MAX_INLINE_ATTACHMENT_BYTES` tracks the API's\n    // ~25 MiB inline hard cap and bounds memory if a server returns a runaway\n    // response.\n    //\n    // Strip whitespace (rare but legal in base64) and account for trailing\n    // `=` padding so the byte count is exact rather than an upper bound.\n    const stripped = attachment.replace(/\\s+/g, \"\");\n    const padding = stripped.endsWith(\"==\")\n      ? 2\n      : stripped.endsWith(\"=\")\n        ? 1\n        : 0;\n    const decodedSize = Math.floor((stripped.length * 3) / 4) - padding;\n    if (decodedSize > MAX_INLINE_ATTACHMENT_BYTES) {\n      throw ExecutionError.statementFailed(\n        `Inline Arrow attachment exceeds maximum size (${decodedSize} > ${MAX_INLINE_ATTACHMENT_BYTES} bytes)`,\n      );\n    }\n\n    const columnNames = arrowColumnNames(response);\n    if (columnNames) {\n      return {\n        ...response,\n        result: {\n          ...(response.result as sql.ResultData & {\n            attachment?: string;\n            columnNames?: string[];\n          }),\n          // `statement_id` is a top-level field, not on `ResultData` — carry it\n          // onto the result (as the EXTERNAL_LINKS path does) so the route can\n          // advertise it in `X-Appkit-Arrow-Columns-Ref` for wide inline schemas.\n          statement_id: response.statement_id,\n          columnNames,\n        },\n      };\n    }\n\n    return response;\n  }\n\n  private async updateWithArrowStatus(\n    response: sql.StatementResponse,\n    workspaceClient: WorkspaceClient,\n    signal?: AbortSignal,\n  ): Promise<{\n    result: {\n      statement_id: string;\n      status: sql.StatementStatus;\n      columnNames?: string[];\n      external_links?: sql.ExternalLink[];\n      refreshChunkLink?: RefreshChunkLink;\n    };\n  }> {\n    const statementId = response.statement_id as string;\n    return {\n      result: {\n        statement_id: statementId,\n        status: {\n          state: response.status?.state,\n          error: response.status?.error,\n        } as sql.StatementStatus,\n        columnNames: arrowColumnNames(response),\n        // Resolve the pre-signed links for EVERY chunk in the caller's own\n        // execution context. Streaming these directly (see\n        // {@link streamExternalLinks}) avoids a second `getStatement` under\n        // the ambient service-principal context — which would fail for\n        // `.obo.sql` (user-owned) statements.\n        external_links: await this._resolveAllExternalLinks(\n          workspaceClient,\n          statementId,\n          response,\n          signal,\n        ),\n        // Bound to this caller's identity so the streamer can re-mint an\n        // expired chunk link mid-download (links live <= 15 min; a large/slow\n        // result can outlast the earliest ones).\n        refreshChunkLink: this._makeChunkLinkRefresher(\n          workspaceClient,\n          statementId,\n        ),\n      },\n    };\n  }\n\n  /**\n   * Resolve pre-signed links for EVERY chunk of an EXTERNAL_LINKS result.\n   *\n   * The execute/getStatement response carries only the first chunk's links\n   * (each link, except the last, exposes `next_chunk_index`); the remaining\n   * chunks are fetched with `getStatementResultChunkN`. Runs in the caller's\n   * identity context (user creds for `.obo.sql`), so there is no cross-identity\n   * fetch. Only the tiny link metadata is resolved eagerly — the bytes still\n   * stream one chunk at a time downstream. Without this a multi-chunk result\n   * would be silently truncated to its first chunk.\n   */\n  private async _resolveAllExternalLinks(\n    workspaceClient: WorkspaceClient,\n    statementId: string,\n    response: sql.StatementResponse,\n    signal?: AbortSignal,\n  ): Promise<sql.ExternalLink[] | undefined> {\n    const first = response.result?.external_links;\n    if (!first || first.length === 0) return first;\n\n    const links: sql.ExternalLink[] = [...first];\n    // Bound the follow loop so a warehouse returning a cyclic/never-ending\n    // `next_chunk_index` can't spin forever. The manifest's chunk count is the\n    // natural bound; fall back to a generous safety cap if it's absent (real\n    // results still terminate earlier when `next_chunk_index` becomes null) so\n    // a missing count doesn't silently truncate a genuine multi-chunk result.\n    const maxFetches =\n      response.manifest?.total_chunk_count ?? MAX_EXTERNAL_CHUNK_FOLLOWS;\n    let next = this._nextChunkIndex(first);\n    for (let fetches = 0; next != null && fetches < maxFetches; fetches++) {\n      if (signal?.aborted) throw ExecutionError.canceled();\n      const chunk =\n        await workspaceClient.statementExecution.getStatementResultChunkN(\n          { statement_id: statementId, chunk_index: next },\n          this._createContext(signal),\n        );\n      const chunkLinks = chunk.external_links ?? [];\n      if (chunkLinks.length === 0) break;\n      links.push(...chunkLinks);\n      next = this._nextChunkIndex(chunkLinks);\n    }\n    return links;\n  }\n\n  /** The `next_chunk_index` advertised by a chunk's links, if any. */\n  private _nextChunkIndex(links: sql.ExternalLink[]): number | undefined {\n    for (const link of links) {\n      if (link.next_chunk_index != null) return link.next_chunk_index;\n    }\n    return undefined;\n  }\n\n  /**\n   * A closure that re-mints a single chunk's pre-signed link via\n   * `getStatementResultChunkN`, bound to the caller's workspace client +\n   * statement id. Created here (in the caller's identity context) so the\n   * streamer — which runs outside that context — can refresh an expired link\n   * for `.obo.sql` statements without a cross-identity `getStatement`.\n   */\n  private _makeChunkLinkRefresher(\n    workspaceClient: WorkspaceClient,\n    statementId: string,\n  ): RefreshChunkLink {\n    return async (chunkIndex, signal) => {\n      const chunk =\n        await workspaceClient.statementExecution.getStatementResultChunkN(\n          { statement_id: statementId, chunk_index: chunkIndex },\n          this._createContext(signal),\n        );\n      return chunk.external_links?.find((l) => l.chunk_index === chunkIndex);\n    };\n  }\n\n  /**\n   * Stream already-resolved EXTERNAL_LINKS chunks as raw Arrow IPC bytes, one\n   * chunk in memory at a time. Takes the `external_links` the caller already\n   * has from `executeStatement` (see {@link updateWithArrowStatus}), so there\n   * is no extra `getStatement` round-trip and no execution-context mismatch —\n   * the pre-signed URLs need no auth to download.\n   */\n  streamExternalLinks(\n    chunks: sql.ExternalLink[],\n    signal?: AbortSignal,\n    refresh?: RefreshChunkLink,\n  ): AsyncGenerator<Uint8Array, void, unknown> {\n    return this.arrowProcessor.streamChunks(chunks, signal, refresh);\n  }\n\n  /**\n   * Fetch just the real column names for a completed statement from its result\n   * manifest — no result data. Backs the `/columns/:statementId` fallback the\n   * client uses when a very wide schema's names exceed the response-header\n   * size limit. Stateless: re-derives from the warehouse, no server cache.\n   */\n  async getColumnNames(\n    workspaceClient: WorkspaceClient,\n    jobId: string,\n    signal?: AbortSignal,\n  ): Promise<string[] | undefined> {\n    const response = await workspaceClient.statementExecution.getStatement(\n      { statement_id: jobId },\n      this._createContext(signal),\n    );\n    return arrowColumnNames(response);\n  }\n\n  /**\n   * Normalize an error caught during statement execution/polling and rethrow.\n   * Abort and already-structured `AppKitError`s pass through untouched;\n   * anything else is wrapped as an `ExecutionError`, preserving the SDK's\n   * structured `ApiError.errorCode` (e.g. \"INVALID_PARAMETER_VALUE\",\n   * \"BAD_REQUEST\") so callers can branch on a stable identifier rather than\n   * substring-matching the message.\n   */\n  private _rethrowStatementError(error: unknown): never {\n    if (error instanceof Error && error.name === \"AbortError\") {\n      throw error;\n    }\n    if (error instanceof AppKitError) {\n      throw error;\n    }\n    const sdkErrorCode =\n      error && typeof error === \"object\" && \"errorCode\" in error\n        ? (error as { errorCode?: unknown }).errorCode\n        : undefined;\n    throw ExecutionError.statementFailed(\n      error instanceof Error ? error.message : String(error),\n      typeof sdkErrorCode === \"string\" ? sdkErrorCode : undefined,\n    );\n  }\n\n  // create context for cancellation token\n  private _createContext(signal?: AbortSignal) {\n    return new Context({\n      cancellationToken: {\n        isCancellationRequested: signal?.aborted ?? false,\n        onCancellationRequested: (cb: () => void) => {\n          signal?.addEventListener(\"abort\", cb, { once: true });\n        },\n      },\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAiCA,MAAM,SAAS,aAAa,2BAA2B;;;;;;;;AASvD,SAAS,iBACP,UACsB;CACtB,MAAM,OAAO,SAAS,UAAU,QAAQ;AACxC,KAAI,CAAC,QAAQ,KAAK,WAAW,EAAG,QAAO;AACvC,QAAO,KAAK,KAAK,GAAG,MAClB,EAAE,QAAQ,EAAE,KAAK,SAAS,IAAI,EAAE,OAAO,UAAU,IAClD;;;;;;;;;;;AAYH,MAAM,8BAA8B,KAAK,OAAO;;;;;;;AAQhD,MAAM,6BAA6B;;;;;;;;AAcnC,MAAa,uCAAuC,MAAS;;;;;;;;AAS7D,MAAM,iCAAiC;AAqDvC,IAAa,wBAAb,MAAmC;CACjC,AAAiB,OAAO;CAExB,AAAQ;CAGR,AAAQ,kBAA+C;;;;;;CAOvD,AAAQ,mCAAmB,IAAI,KAAqB;;CAEpD,AAAQ,qCAAqB,IAAI,KAAyC;CAE1E,AAAiB;CACjB,AAAiB;CAKjB,YAAY,QAA4B;AACtC,OAAK,SAAS;AAEd,OAAK,YAAY,iBAAiB,YAChC,KAAK,MACL,KAAK,OAAO,UACb;AACD,OAAK,mBAAmB;GACtB,YAAY,KAAK,UAAU,UAAU,CAAC,cAAc,eAAe;IACjE,aAAa;IACb,MAAM;IACP,CAAC;GACF,eAAe,KAAK,UACjB,UAAU,CACV,gBAAgB,kBAAkB;IACjC,aAAa;IACb,MAAM;IACP,CAAC;GACL;;;;;;CAOH,IAAY,iBAAuC;AACjD,MAAI,CAAC,KAAK,gBACR,MAAK,kBAAkB,IAAI,qBAAqB;GAC9C,SAAS,KAAK,OAAO,WAAW,yBAAyB;GACzD,SAAS,qBAAqB;GAC/B,CAAC;AAEJ,SAAO,KAAK;;CAGd,MAAM,iBACJ,iBACA,OACA,QACA;EACA,MAAM,YAAY,KAAK,KAAK;EAC5B,IAAI,UAAU;AAGd,MAAI,QAAQ,QACV,OAAM,eAAe,UAAU;AAGjC,SAAO,KAAK,UAAU,gBACpB,aACA;GACE,MAAM,SAAS;GACf,YAAY;IACV,aAAa;IACb,mBAAmB,MAAM,gBAAgB;IACzC,cAAc,MAAM,WAAW;IAC/B,aAAa,MAAM,UAAU;IAC7B,gBAAgB,MAAM,WAAW,UAAU,GAAG,IAAI,IAAI;IACtD,qBAAqB,CAAC,CAAC,MAAM;IAC9B;GACF,EACD,OAAO,SAAe;GACpB,IAAI;GACJ,IAAI,YAAY;AAEhB,OAAI,QAAQ;AACV,yBAAqB;AAEnB,SAAI,CAAC,KAAK,aAAa,CAAE;AACzB,iBAAY;AACZ,UAAK,aAAa,aAAa,KAAK;AACpC,UAAK,UAAU;MACb,MAAM,eAAe;MACrB,SAAS;MACV,CAAC;AACF,UAAK,KAAK;;AAEZ,WAAO,iBAAiB,SAAS,cAAc,EAAE,MAAM,MAAM,CAAC;;AAGhE,OAAI;AAEF,QAAI,CAAC,MAAM,UACT,OAAM,gBAAgB,aAAa,YAAY;AAGjD,QAAI,CAAC,MAAM,aACT,OAAM,gBAAgB,aAAa,eAAe;IAGpD,MAAM,OAAoC;KACxC,WAAW,MAAM;KACjB,YAAY,MAAM;KAClB,cAAc,MAAM;KACpB,SAAS,MAAM;KACf,QAAQ,MAAM;KACd,cACE,MAAM,gBAAgB,yBAAyB;KACjD,aACE,MAAM,eAAe,yBAAyB;KAChD,QAAQ,MAAM,UAAU,yBAAyB;KACjD,YAAY,MAAM;KAClB,WAAW,MAAM;KACjB,iBACE,MAAM,mBAAmB,yBAAyB;KACrD;AAED,SAAK,SAAS,wBAAwB,EACpC,mBAAmB,MAAM,cAC1B,CAAC;IAEF,MAAM,WACJ,MAAM,gBAAgB,mBAAmB,iBACvC,MACA,KAAK,eAAe,OAAO,CAC5B;AAEH,QAAI,CAAC,SACH,OAAM,gBAAgB,WAAW,gBAAgB;IAEnD,MAAM,SAAS,SAAS;IACxB,MAAM,cAAc,SAAS;AAE7B,SAAK,aAAa,mBAAmB,YAAY;AACjD,SAAK,SAAS,uBAAuB;KACnC,mBAAmB,SAAS;KAC5B,aAAa,QAAQ;KACtB,CAAC;IAEF,IAAI;AAIJ,YAAQ,QAAQ,OAAhB;KACE,KAAK;KACL,KAAK;AACH,WAAK,SAAS,6BAA6B,EACzC,aAAa,SAAS,QAAQ,OAC/B,CAAC;AACF,eAAS,MAAM,KAAK,wBAClB,iBACA,aACA,KAAK,OAAO,SACZ,OACD;AACD;KACF,KAAK;AACH,eAAS,MAAM,KAAK,oBAClB,UACA,iBACA,OACD;AACD;KACF,KAAK,SACH,OAAM,eAAe,gBACnB,OAAO,OAAO,SACd,OAAO,OAAO,WACf;KACH,KAAK,WACH,OAAM,eAAe,UAAU;KACjC,KAAK,SACH,OAAM,eAAe,eAAe;KACtC,QACE,OAAM,eAAe,aACnB,OAAO,QAAQ,SAAS,UAAU,CACnC;;IAGL,MAAM,aAAa,OAAO;IAC1B,MAAM,WACJ,YAAY,MAAM,UAAU,YAAY,YAAY,UAAU;AAEhE,QAAI,WAAW,EACb,MAAK,aAAa,uBAAuB,SAAS;IAGpD,MAAM,WAAW,KAAK,KAAK,GAAG;AAC9B,WAAO,OAAO,EAAE,WAAW,iBAAiB;KAC1C,cAAc,MAAM;KACpB,eAAe;KACf,mBAAmB;KACpB,CAAC;AAEF,cAAU;AAEV,QAAI,CAAC,UACH,MAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAE7C,WAAO;YACA,OAAO;AAEd,QAAI,CAAC,WAAW;AACd,UAAK,gBAAgB,MAAe;AACpC,UAAK,UAAU;MACb,MAAM,eAAe;MACrB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;MAChE,CAAC;AAEF,YAAO,MACL,kCACA,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;;AAGH,SAAK,uBAAuB,MAAM;aAC1B;AAER,QAAI,gBAAgB,OAClB,QAAO,oBAAoB,SAAS,aAAa;IAGnD,MAAM,WAAW,KAAK,KAAK,GAAG;AAG9B,QAAI,CAAC,UACH,MAAK,KAAK;IAGZ,MAAM,aAAa;KACjB,mBAAmB,MAAM;KACzB,cAAc,MAAM,WAAW;KAC/B,aAAa,MAAM,UAAU;KAC7B,gBAAgB,MAAM,WAAW,UAAU,GAAG,IAAI,IAAI;KACtD,SAAS,QAAQ,UAAU;KAC5B;AAED,SAAK,iBAAiB,WAAW,IAAI,GAAG,WAAW;AACnD,SAAK,iBAAiB,cAAc,OAAO,UAAU,WAAW;;KAGpE;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAmCH,MAAM,uBACJ,iBACA,aACA,MACe;EACf,MAAM,EACJ,UACA,QACA,YAAY,sCACZ,YAAY,SACV;AAEJ,MAAI,QAAQ,QAAS,OAAM,eAAe,UAAU;AACpD,MAAI,CAAC,YAAa,OAAM,gBAAgB,aAAa,eAAe;AACpE,MAAI,KAAK,mBAAmB,YAAY,CAAE;EAE1C,MAAM,WAAW,KAAK,mBAAmB,IAAI,YAAY;AACzD,MAAI,YAAY,CAAC,SAAS,iBAAiB,OAAO,QAChD,QAAO,KAAK,iBAAiB,UAAU,UAAU,QAAQ,KAAK;EAGhE,MAAM,QAAQ,KAAK,wBAAwB,iBAAiB,aAAa;GACvE;GACA;GACD,CAAC;AACF,OAAK,mBAAmB,IAAI,aAAa,MAAM;AAC/C,SAAO,KAAK,iBAAiB,OAAO,UAAU,QAAQ,MAAM;;CAG9D,AAAQ,iBACN,OACA,UACA,QACA,MACe;AACf,MAAI,QAAQ,QAAS,OAAM,eAAe,UAAU;AACpD,MAAI,KAAM,OAAM;AAChB,QAAM,YAAY,IAAI,SAAS;AAC/B,MAAI,MAAM,WAAY,UAAS,MAAM,WAAW;AAChD,SAAO,KAAK,wBAAwB,OAAO,OAAO;;CAGpD,AAAQ,wBACN,iBACA,aACA,MAC4B;EAC5B,MAAM,mBAAmB,IAAI,iBAAiB;EAC9C,MAAM,QAAoC;GACxC,UAAU;GACV;GACA,6BAAa,IAAI,KAAK;GACtB,YAAY;GACZ,sBAAsB;GACtB,SAAS,QAAQ,SAAS;GAC3B;AAED,QAAM,UAAU,KAAK,UAClB,gBACC,sBACA;GACE,MAAM,SAAS;GACf,YAAY;IACV,aAAa;IACb,mBAAmB;IACnB,mCAAmC,KAAK;IACzC;GACF,EACD,OAAO,SAAe;AACpB,OAAI;AACF,UAAM,KAAK,2BACT,iBACA,aACA;KACE,WAAW,WAAW,KAAK,oBAAoB,OAAO,OAAO;KAC7D,QAAQ,iBAAiB;KACzB,WAAW,KAAK;KAChB,WAAW,KAAK;KAChB,8BAA8B;AAC5B,YAAM,uBAAuB;;KAEhC,EACD,KACD;AACD,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;YACpC,OAAO;AACd,SAAK,8BAA8B,OAAO,aAAa,KAAK;aACpD;AACR,SAAK,KAAK;;KAGd;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC,CACA,cAAc;AACb,OAAI,KAAK,mBAAmB,IAAI,YAAY,KAAK,MAC/C,MAAK,mBAAmB,OAAO,YAAY;IAE7C;AAEJ,QAAM,QAAQ,YAAY,GAAG;AAC7B,SAAO;;CAGT,AAAQ,oBACN,OACA,QACM;AACN,QAAM,aAAa;AACnB,OAAK,MAAM,cAAc,MAAM,YAC7B,KAAI;AACF,cAAW,OAAO;UACZ;;;CAOZ,AAAQ,wBACN,OACA,cACe;AACf,MAAI,CAAC,aAAc,QAAO,MAAM;AAEhC,SAAO,IAAI,SAAe,SAAS,WAAW;GAC5C,IAAI,OAAO;GACX,MAAM,UAAU,SAAiC,UAAoB;AACnE,QAAI,KAAM;AACV,WAAO;AACP,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,QAAI,YAAY,KAAM,UAAS;aACtB,YAAY,QAAS,QAAO,eAAe,UAAU,CAAC;QAC1D,QAAO,MAAM;;GAEpB,MAAM,gBAAgB;AACpB,SAAK,wBAAwB,MAAM;AACnC,WAAO,QAAQ;;AAEjB,gBAAa,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;AAC/D,SAAM,QAAQ,WACN,OAAO,KAAK,GACjB,UAAU,OAAO,OAAO,MAAM,CAChC;IACD;;CAGJ,AAAQ,wBAAwB,OAAyC;AACvE,MAAI,MAAM,WAAW,EAAG,OAAM;AAC9B,MAAI,MAAM,WAAW,KAAK,MAAM,iBAAiB,OAAO,QAAS;AAEjE,MAAI,MAAM,qBAER;AAKF,uBAAqB;AACnB,OAAI,MAAM,YAAY,KAAK,CAAC,MAAM,iBAAiB,OAAO,QACxD,OAAM,iBAAiB,MAAM,0CAA0C;IAEzE;;CAGJ,MAAc,2BACZ,iBACA,aACA,MACA,MACe;EACf,MAAM,EACJ,UACA,QACA,YAAY,sCACZ,YAAY,MACZ,2BACE;EACJ,MAAM,YAAY,KAAK,KAAK;EAC5B,MAAM,UAAU,IAAI,uBAAuB,MAAM,WAAW,SAAS;EACrE,IAAI,WAAW;EACf,MAAM,UAAU,IAAI,sBAAsB;AAE1C,SAAO,MAAM;AACX,OAAI,QAAQ,QAAS,OAAM,eAAe,UAAU;AACpD,OAAI,KAAK,KAAK,GAAG,YAAY,UAC3B,OAAM,eAAe,gBACnB,8CAA8C,UAAU,IACzD;GAGH,IAAI;AACJ,OAAI;AACF,WAAO,MAAM,gBAAgB,WAAW,IACtC,EAAE,IAAI,aAAa,EACnB,KAAK,eAAe,OAAO,CAC5B;YACM,OAAO;AAGd,QACE,QAAQ,WACP,iBAAiB,SAAS,MAAM,SAAS,aAE1C,OAAM,eAAe,UAAU;AASjC,SAAK,SAAS,iCAAiC;KAC7C,mBAAmB;KACnB,iBACE,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;KACzD,CAAC;AACF,WAAO,MACL,oFACA,aACA,MACD;AACD,SAAK,aAAa,oCAAoC,KAAK;AAC3D;;GAEF,MAAM,QAAQ,MAAM;GACpB,MAAM,UAAU,MAAM,QAAQ;AAE9B,WAAQ,OAAR;IACE,KAAK;AACH,aAAQ,KAAK,OAAO,QAAQ;AAC5B,UAAK,iBAAiB,IAAI,aAAa,KAAK,KAAK,CAAC;AAClD,UAAK,aAAa,yBAAyB,QAAQ,QAAQ;AAC3D;IACF,KAAK;IACL,KAAK,WACH,OAAM,mBAAmB,iBACvB,gBACA,mCAAmC,MAAM,mEAC1C;IACH,KAAK;AACH,SAAI,CAAC,UACH,OAAM,IAAI,mBACR,uJACD;AAEH,SAAI,CAAC,UAAU;AACb,cAAQ,KAAK,YAAY,QAAQ;AACjC,gCAA0B;AAC1B,YAAM,gBAAgB,WAAW,MAC/B,EAAE,IAAI,aAAa,EACnB,KAAK,eAAe,OAAO,CAC5B;AACD,iBAAW;WAEX,SAAQ,KAAK,OAAO,QAAQ;AAE9B;IACF,KAAK;IACL,KAAK;AACH,aAAQ,KAAK,OAAO,QAAQ;AAC5B;IACF,QACE,OAAM,eAAe,aAAa,OAAO,SAAS,UAAU,CAAC;;GAGjE,MAAM,UAAU,QAAQ,MAAM;AAC9B,OAAI,KAAK,KAAK,GAAG,UAAU,aAAa,UACtC,OAAM,eAAe,gBACnB,8CAA8C,UAAU,IACzD;AAEH,SAAM,KAAK,sBAAsB,SAAS,OAAO;;;;;;;CAQrD,AAAQ,mBAAmB,aAA8B;EACvD,MAAM,aAAa,KAAK,iBAAiB,IAAI,YAAY;AACzD,SACE,eAAe,UACf,KAAK,KAAK,GAAG,aAAa;;;;;;;;;;;;;CAe9B,AAAQ,8BACN,OACA,aACA,MACO;AACP,MAAI,iBAAiB,aAAa;AAChC,QAAK,gBAAgB,MAAM;AAC3B,QAAK,UAAU;IAAE,MAAM,eAAe;IAAO,SAAS,MAAM;IAAM,CAAC;AACnE,SAAM;;AAER,OAAK,gBACH,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,MAAM,CAAC,CAC1D;AACD,SAAO,MACL,kDACA,aACA,MACD;EACD,MAAM,UAAU,eAAe,gBAC7B,mCACD;AACD,OAAK,UAAU;GAAE,MAAM,eAAe;GAAO,SAAS,QAAQ;GAAM,CAAC;AACrE,QAAM;;;;;;;;CASR,AAAQ,sBACN,IACA,QACe;AACf,MAAI,QAAQ,QAAS,QAAO,QAAQ,OAAO,eAAe,UAAU,CAAC;AACrE,SAAO,IAAI,SAAe,SAAS,WAAW;GAC5C,MAAM,QAAQ,iBAAiB;AAC7B,YAAQ,oBAAoB,SAAS,QAAQ;AAC7C,aAAS;MACR,GAAG;GACN,MAAM,gBAAgB;AACpB,iBAAa,MAAM;AACnB,WAAO,eAAe,UAAU,CAAC;;AAEnC,WAAQ,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;IAC1D;;CAGJ,MAAc,wBACZ,iBACA,aACA,UAAU,yBAAyB,SACnC,QACA;AACA,SAAO,KAAK,UAAU,gBACpB,YACA,EACE,YAAY;GACV,mBAAmB;GACnB,sBAAsB;GACvB,EACF,EACD,OAAO,SAAe;AACpB,OAAI;IACF,MAAM,YAAY,KAAK,KAAK;IAC5B,IAAI,QAAQ;IACZ,MAAM,uBAAuB;IAC7B,IAAI,YAAY;AAEhB,WAAO,MAAM;AACX;AACA,UAAK,aAAa,8BAA8B,UAAU;KAG1D,MAAM,cAAc,KAAK,KAAK,GAAG;AACjC,SAAI,cAAc,SAAS;MACzB,MAAM,QAAQ,eAAe,gBAC3B,kCAAkC,QAAQ,eAAe,YAAY,KACtE;AACD,WAAK,gBAAgB,MAAM;AAC3B,WAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAC9C,YAAM;;AAGR,SAAI,QAAQ,SAAS;MACnB,MAAM,QAAQ,eAAe,UAAU;AACvC,WAAK,gBAAgB,MAAM;AAC3B,WAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAC9C,YAAM;;AAGR,UAAK,SAAS,mBAAmB;MAC/B,gBAAgB;MAChB,iBAAiB;MACjB,mBAAmB;MACpB,CAAC;KAEF,MAAM,WACJ,MAAM,gBAAgB,mBAAmB,aACvC,EACE,cAAc,aACf,EACD,KAAK,eAAe,OAAO,CAC5B;AACH,SAAI,CAAC,SACH,OAAM,gBAAgB,WAAW,gBAAgB;KAGnD,MAAM,SAAS,SAAS;AAExB,UAAK,SAAS,wBAAwB;MACpC,aAAa,QAAQ;MACrB,gBAAgB;MACjB,CAAC;AAEF,aAAQ,QAAQ,OAAhB;MACE,KAAK;MACL,KAAK,UAEH;MACF,KAAK;AACH,YAAK,aAAa,uBAAuB,UAAU;AACnD,YAAK,aAAa,gCAAgC,YAAY;AAC9D,YAAK,SAAS,qBAAqB;QACjC,iBAAiB;QACjB,oBAAoB;QACrB,CAAC;AACF,YAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,cAAO,KAAK,oBACV,UACA,iBACA,OACD;MACH,KAAK,SACH,OAAM,eAAe,gBACnB,OAAO,OAAO,SACd,OAAO,OAAO,WACf;MACH,KAAK,WACH,OAAM,eAAe,UAAU;MACjC,KAAK,SACH,OAAM,eAAe,eAAe;MACtC,QACE,OAAM,eAAe,aACnB,OAAO,QAAQ,SAAS,UAAU,CACnC;;AAIL,WAAM,IAAI,SAAS,YAAY,WAAW,SAAS,MAAM,CAAC;AAC1D,aAAQ,KAAK,IAAI,QAAQ,GAAG,qBAAqB;;YAE5C,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU;KACb,MAAM,eAAe;KACrB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;KAChE,CAAC;AAGF,SAAK,uBAAuB,MAAM;aAC1B;AACR,SAAK,KAAK;;KAGd;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;CAGH,MAAc,oBACZ,UACA,iBACA,QACA;AACA,MAAI,SAAS,UAAU,WAAW,gBAAgB;GAChD,MAAM,SAAS,SAAS;AAQxB,OAAI,QAAQ,WACV,QAAO,KAAK,yBAAyB,UAAU,OAAO,WAAW;AAQnE,OAAI,QAAQ,kBAAkB,OAAO,eAAe,SAAS,EAC3D,QAAO,KAAK,sBAAsB,UAAU,iBAAiB,OAAO;AAatE,QAFE,CAAC,QAAQ,cACR,MAAM,QAAQ,OAAO,WAAW,IAAI,OAAO,WAAW,WAAW,MACnD,SAAS,UAAU,QAAQ,SAAS;IACnD,MAAM,cAAc,yBAClB,SAAS,SAAS,OAAO,QAC1B;AACD,WAAO;KACL,GAAG;KACH,QAAQ;MAAE,GAAI,UAAU,EAAE;MAAG,YAAY;MAAa;KACvD;;;AASL,MAAI,CAAC,SAAS,QAAQ,cAAc,CAAC,SAAS,UAAU,QAAQ,QAC9D,QAAO;EAGT,MAAM,UAAU,SAAS,SAAS,OAAO;EAEzC,MAAM,kBAAkB,SAAS,OAAO,WAAW,KAAK,QAAQ;GAC9D,MAAM,MAA+B,EAAE;AACvC,OAAI,SAAS,OAAO,UAAU;IAC5B,MAAM,SAAS,QAAQ;IACvB,MAAM,aAAa,QAAQ,QAAQ,UAAU;AAG7C,QACE,QAAQ,cAAc,YACtB,OAAO,UAAU,YACjB,UACC,MAAM,OAAO,OAAO,MAAM,OAAO,KAElC,KAAI;AACF,SAAI,cAAc,KAAK,MAAM,MAAM;YAC7B;AAEN,SAAI,cAAc;;QAGpB,KAAI,cAAc;KAEpB;AACF,UAAO;IACP;EAGF,MAAM,EAAE,YAAY,aAAa,GAAG,eAAe,SAAS;AAC5D,SAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAG;IACH,MAAM;IACP;GACF;;;;;;;;;;;;;CAcH,AAAQ,yBACN,UACA,YACA;EAQA,MAAM,WAAW,WAAW,QAAQ,QAAQ,GAAG;EAC/C,MAAM,UAAU,SAAS,SAAS,KAAK,GACnC,IACA,SAAS,SAAS,IAAI,GACpB,IACA;EACN,MAAM,cAAc,KAAK,MAAO,SAAS,SAAS,IAAK,EAAE,GAAG;AAC5D,MAAI,cAAc,4BAChB,OAAM,eAAe,gBACnB,iDAAiD,YAAY,KAAK,4BAA4B,SAC/F;EAGH,MAAM,cAAc,iBAAiB,SAAS;AAC9C,MAAI,YACF,QAAO;GACL,GAAG;GACH,QAAQ;IACN,GAAI,SAAS;IAOb,cAAc,SAAS;IACvB;IACD;GACF;AAGH,SAAO;;CAGT,MAAc,sBACZ,UACA,iBACA,QASC;EACD,MAAM,cAAc,SAAS;AAC7B,SAAO,EACL,QAAQ;GACN,cAAc;GACd,QAAQ;IACN,OAAO,SAAS,QAAQ;IACxB,OAAO,SAAS,QAAQ;IACzB;GACD,aAAa,iBAAiB,SAAS;GAMvC,gBAAgB,MAAM,KAAK,yBACzB,iBACA,aACA,UACA,OACD;GAID,kBAAkB,KAAK,wBACrB,iBACA,YACD;GACF,EACF;;;;;;;;;;;;;CAcH,MAAc,yBACZ,iBACA,aACA,UACA,QACyC;EACzC,MAAM,QAAQ,SAAS,QAAQ;AAC/B,MAAI,CAAC,SAAS,MAAM,WAAW,EAAG,QAAO;EAEzC,MAAM,QAA4B,CAAC,GAAG,MAAM;EAM5C,MAAM,aACJ,SAAS,UAAU,qBAAqB;EAC1C,IAAI,OAAO,KAAK,gBAAgB,MAAM;AACtC,OAAK,IAAI,UAAU,GAAG,QAAQ,QAAQ,UAAU,YAAY,WAAW;AACrE,OAAI,QAAQ,QAAS,OAAM,eAAe,UAAU;GAMpD,MAAM,cAJJ,MAAM,gBAAgB,mBAAmB,yBACvC;IAAE,cAAc;IAAa,aAAa;IAAM,EAChD,KAAK,eAAe,OAAO,CAC5B,EACsB,kBAAkB,EAAE;AAC7C,OAAI,WAAW,WAAW,EAAG;AAC7B,SAAM,KAAK,GAAG,WAAW;AACzB,UAAO,KAAK,gBAAgB,WAAW;;AAEzC,SAAO;;;CAIT,AAAQ,gBAAgB,OAA+C;AACrE,OAAK,MAAM,QAAQ,MACjB,KAAI,KAAK,oBAAoB,KAAM,QAAO,KAAK;;;;;;;;;CAYnD,AAAQ,wBACN,iBACA,aACkB;AAClB,SAAO,OAAO,YAAY,WAAW;AAMnC,WAJE,MAAM,gBAAgB,mBAAmB,yBACvC;IAAE,cAAc;IAAa,aAAa;IAAY,EACtD,KAAK,eAAe,OAAO,CAC5B,EACU,gBAAgB,MAAM,MAAM,EAAE,gBAAgB,WAAW;;;;;;;;;;CAW1E,oBACE,QACA,QACA,SAC2C;AAC3C,SAAO,KAAK,eAAe,aAAa,QAAQ,QAAQ,QAAQ;;;;;;;;CASlE,MAAM,eACJ,iBACA,OACA,QAC+B;AAK/B,SAAO,iBAJU,MAAM,gBAAgB,mBAAmB,aACxD,EAAE,cAAc,OAAO,EACvB,KAAK,eAAe,OAAO,CAC5B,CACgC;;;;;;;;;;CAWnC,AAAQ,uBAAuB,OAAuB;AACpD,MAAI,iBAAiB,SAAS,MAAM,SAAS,aAC3C,OAAM;AAER,MAAI,iBAAiB,YACnB,OAAM;EAER,MAAM,eACJ,SAAS,OAAO,UAAU,YAAY,eAAe,QAChD,MAAkC,YACnC;AACN,QAAM,eAAe,gBACnB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,EACtD,OAAO,iBAAiB,WAAW,eAAe,OACnD;;CAIH,AAAQ,eAAe,QAAsB;AAC3C,SAAO,IAAI,QAAQ,EACjB,mBAAmB;GACjB,yBAAyB,QAAQ,WAAW;GAC5C,0BAA0B,OAAmB;AAC3C,YAAQ,iBAAiB,SAAS,IAAI,EAAE,MAAM,MAAM,CAAC;;GAExD,EACF,CAAC"}

package/dist/core/appkit.d.ts

@@ -46,7 +46,7 @@ declare function createApp<T extends PluginData<PluginConstructor, unknown, stri
   plugins?: T;
   telemetry?: TelemetryConfig;
   cache?: CacheConfig;
-  client?: WorkspaceClient;
+  client?: WorkspaceClient; /** Runs after plugin setup but **before** the server starts. */
   onPluginsReady?: (appkit: PluginMap<T>) => void | Promise<void>;
   disableInternalTelemetry?: boolean;
 }): Promise<PluginMap<T>>;

package/dist/core/appkit.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"appkit.d.ts","names":[],"sources":["../../src/core/appkit.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA8XsB,SAAA,WACV,UAAA,CAAW,iBAAA,qBAAA,CAErB,MAAA;EACE,OAAA,GAAU,CAAA;EACV,SAAA,GAAY,eAAA;EACZ,KAAA,GAAQ,WAAA;EACR,MAAA,GAAS,eAAA;EACT,cAAA,IAAkB,MAAA,EAAQ,SAAA,CAAU,CAAA,aAAc,OAAA;EAClD,wBAAA;AAAA,IAED,OAAA,CAAQ,SAAA,CAAU,CAAA"}
+{"version":3,"file":"appkit.d.ts","names":[],"sources":["../../src/core/appkit.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA8ZsB,SAAA,WACV,UAAA,CAAW,iBAAA,qBAAA,CAErB,MAAA;EACE,OAAA,GAAU,CAAA;EACV,SAAA,GAAY,eAAA;EACZ,KAAA,GAAQ,WAAA;EACR,MAAA,GAAS,eAAA;EAET,cAAA,IAAkB,MAAA,EAAQ,SAAA,CAAU,CAAA,aAAc,OAAA;EAClD,wBAAA;AAAA,IAED,OAAA,CAAQ,SAAA,CAAU,CAAA"}

package/dist/core/appkit.js

@@ -18,10 +18,19 @@ import { PluginContext, isToolProvider } from "./plugin-context.js";
 
 //#region src/core/appkit.ts
 const logger = createLogger("appkit");
+/**
+* Internal teardown entry for the test harness (see `createTestApp`).
+* Symbol-keyed so it cannot collide with — or be shadowed by — a plugin
+* manifest name, and so it stays off the public `PluginMap` surface.
+* @internal
+*/
+const disposeApp = Symbol("appkit.internal.dispose");
 var AppKit = class AppKit {
 	#pluginInstances = {};
 	#setupPromises = [];
 	#context;
+	/** Owns the shutdown sequence; assigned once every plugin has started. */
+	#lifecycle;
 	constructor(config) {
 		const { plugins, ...globalConfig } = config;
 		this.#context = new PluginContext();
@@ -112,17 +121,29 @@ var AppKit = class AppKit {
 		await Promise.all(instance.#setupPromises);
 		TelemetryManager.start();
 		await instance.#context.emitLifecycle("setup:complete");
-		const handle = instance;
+		const app = instance;
 		if (config.onPluginsReady) {
 			logger.debug("Running onPluginsReady hook");
-			await config.onPluginsReady(handle);
+			await config.onPluginsReady(app);
 			logger.debug("onPluginsReady hook completed");
 		}
 		if (isInternalTelemetryEnabled(config)) AppKit.bootstrapInternalTelemetry();
 		const serverPlugin = instance.#pluginInstances.server;
 		if (serverPlugin && typeof serverPlugin.start === "function") await serverPlugin.start();
-		new LifecycleManager(instance.#context).installSignalHandlers();
-		return handle;
+		instance.#lifecycle = new LifecycleManager(instance.#context);
+		if (config.installSignalHandlers !== false) instance.#lifecycle.installSignalHandlers();
+		return app;
+	}
+	/**
+	* Internal teardown entry point for the test harness: delegates to the
+	* lifecycle's non-exiting `shutdown({ exit: false })` (the phases are
+	* canonical there). Not public API — AppKit does not support re-booting or
+	* embedding, so real apps tear down only through the signal path. The harness
+	* drops the core singletons and restores env after this resolves.
+	* @internal
+	*/
+	async [disposeApp]() {
+		await this.#lifecycle?.shutdown({ exit: false });
 	}
 	static bootstrapInternalTelemetry() {
 		const serviceCtx = ServiceContext.get();
@@ -221,5 +242,5 @@ async function createApp(config = {}) {
 }
 
 //#endregion
-export { createApp };
+export { AppKit, createApp, disposeApp };
 //# sourceMappingURL=appkit.js.map