@databricks/appkit 0.59.0 → 0.61.0

package/dist/connectors/files/client.js.map

@@ -1 +1 @@
-{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/files/client.ts"],"sourcesContent":["import { AsyncLocalStorage } from \"node:async_hooks\";\nimport type { TelemetryOptions } from \"shared\";\nimport { createLogger } from \"../../logging/logger\";\nimport type {\n  DirectoryEntry,\n  DownloadResponse,\n  FileMetadata,\n  FilePreview,\n} from \"../../plugins/files/types\";\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 { ApiError, type WorkspaceClient } from \"../../workspace-client\";\nimport {\n  contentTypeFromPath,\n  FILES_MAX_READ_SIZE,\n  isTextContentType,\n} from \"./defaults\";\n\nconst logger = createLogger(\"connectors:files\");\n\n/**\n * Ambient span-attribute propagation for `FilesConnector.traced()`.\n *\n * Callers (e.g. the plugin's `_withAuthModeAttributes` wrapper) set extra\n * span attributes here via `runWithFilesSpanAttributes(attrs, fn)`. The\n * connector's `traced()` decorator reads them and merges them into the\n * span it creates around the SDK call. This lets the plugin tag spans with\n * `files.auth_mode` without opening a duplicate `files.<op>` span.\n *\n * AsyncLocalStorage is used so concurrent requests don't see each other's\n * attributes. Outside an active scope, `getStore()` returns `undefined` and\n * the connector falls back to the static attribute set.\n */\nconst filesSpanAttributesStorage = new AsyncLocalStorage<\n  Record<string, string>\n>();\n\n/**\n * Run `fn` with the supplied attributes attached to whatever span the\n * `FilesConnector` opens for its SDK call. Used to propagate request-scoped\n * attributes (e.g. `files.auth_mode`) onto the connector's span without\n * opening a parent span — avoids the 2x span allocation that\n * `startActiveSpan` parented otherwise.\n */\nexport function runWithFilesSpanAttributes<T>(\n  attributes: Record<string, string>,\n  fn: () => Promise<T>,\n): Promise<T> {\n  return filesSpanAttributesStorage.run(attributes, fn);\n}\n\ninterface FilesConnectorConfig {\n  defaultVolume?: string;\n  timeout?: number;\n  telemetry?: TelemetryOptions;\n  customContentTypes?: Record<string, string>;\n}\n\nexport class FilesConnector {\n  private readonly name = \"files\";\n  private defaultVolume: string | undefined;\n  private readonly customContentTypes: Record<string, string> | undefined;\n\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    operationCount: Counter;\n    operationDuration: Histogram;\n  };\n\n  constructor(config: FilesConnectorConfig) {\n    this.defaultVolume = config.defaultVolume;\n    this.customContentTypes = config.customContentTypes;\n\n    this.telemetry = TelemetryManager.getProvider(this.name, config.telemetry);\n    this.telemetryMetrics = {\n      operationCount: this.telemetry\n        .getMeter()\n        .createCounter(\"files.operation.count\", {\n          description: \"Total number of file operations\",\n          unit: \"1\",\n        }),\n      operationDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"files.operation.duration\", {\n          description: \"Duration of file operations\",\n          unit: \"ms\",\n        }),\n    };\n  }\n\n  resolvePath(filePath: string): string {\n    if (filePath.length > 4096) {\n      throw new Error(\n        `Path exceeds maximum length of 4096 characters (got ${filePath.length}).`,\n      );\n    }\n    if (filePath.includes(\"\\0\")) {\n      throw new Error(\"Path must not contain null bytes.\");\n    }\n\n    const segments = filePath.split(\"/\");\n    if (segments.some((s) => s === \"..\")) {\n      throw new Error('Path traversal (\"../\") is not allowed.');\n    }\n    if (filePath.startsWith(\"/\")) {\n      if (!filePath.startsWith(\"/Volumes/\")) {\n        throw new Error(\n          'Absolute paths must start with \"/Volumes/\". ' +\n            \"Unity Catalog volume paths follow the format: /Volumes/<catalog>/<schema>/<volume>/\",\n        );\n      }\n      return filePath;\n    }\n    if (!this.defaultVolume) {\n      throw new Error(\n        \"Cannot resolve relative path: no default volume set. Use an absolute path or set a default volume.\",\n      );\n    }\n    return `${this.defaultVolume}/${filePath}`;\n  }\n\n  private async traced<T>(\n    operation: string,\n    attributes: Record<string, string>,\n    fn: (span: Span) => Promise<T>,\n  ): Promise<T> {\n    const startTime = Date.now();\n    let success = false;\n\n    // Pull any ambient attributes set by `runWithFilesSpanAttributes` (e.g.\n    // `files.auth_mode` from the plugin layer). Static `attributes` win on\n    // collision so callers can override per-call.\n    const ambient = filesSpanAttributesStorage.getStore();\n\n    return this.telemetry.startActiveSpan(\n      `files.${operation}`,\n      {\n        kind: SpanKind.CLIENT,\n        attributes: {\n          \"files.operation\": operation,\n          ...(ambient ?? {}),\n          ...attributes,\n        },\n      },\n      async (span: Span) => {\n        try {\n          const result = await fn(span);\n          success = true;\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\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          throw error;\n        } finally {\n          span.end();\n          const duration = Date.now() - startTime;\n          const metricAttrs = {\n            \"files.operation\": operation,\n            success: String(success),\n          };\n          this.telemetryMetrics.operationCount.add(1, metricAttrs);\n          this.telemetryMetrics.operationDuration.record(duration, metricAttrs);\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  async list(\n    client: WorkspaceClient,\n    directoryPath?: string,\n  ): Promise<DirectoryEntry[]> {\n    const resolvedPath = directoryPath\n      ? this.resolvePath(directoryPath)\n      : this.defaultVolume;\n    if (!resolvedPath) {\n      throw new Error(\"No directory path provided and no default volume set.\");\n    }\n\n    return this.traced(\"list\", { \"files.path\": resolvedPath }, async () => {\n      const entries: DirectoryEntry[] = [];\n      for await (const entry of client.files.listDirectoryContents({\n        directory_path: resolvedPath,\n      })) {\n        entries.push(entry);\n      }\n      return entries;\n    });\n  }\n\n  async read(\n    client: WorkspaceClient,\n    filePath: string,\n    options?: { maxSize?: number },\n  ): Promise<string> {\n    const resolvedPath = this.resolvePath(filePath);\n    const maxSize = options?.maxSize ?? FILES_MAX_READ_SIZE;\n    return this.traced(\"read\", { \"files.path\": resolvedPath }, async () => {\n      const response = await this.download(client, filePath);\n      if (!response.contents) {\n        return \"\";\n      }\n      const reader = response.contents.getReader();\n      const decoder = new TextDecoder();\n      let result = \"\";\n      let bytesRead = 0;\n      while (true) {\n        const { done, value } = await reader.read();\n        if (done) break;\n        bytesRead += value.byteLength;\n        if (bytesRead > maxSize) {\n          await reader.cancel();\n          throw new Error(\n            `File exceeds maximum read size (${maxSize} bytes). Use download() for large files.`,\n          );\n        }\n        result += decoder.decode(value, { stream: true });\n      }\n      result += decoder.decode();\n      return result;\n    });\n  }\n\n  async download(\n    client: WorkspaceClient,\n    filePath: string,\n  ): Promise<DownloadResponse> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"download\", { \"files.path\": resolvedPath }, async () => {\n      return client.files.download({\n        file_path: resolvedPath,\n      });\n    });\n  }\n\n  async exists(client: WorkspaceClient, filePath: string): Promise<boolean> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"exists\", { \"files.path\": resolvedPath }, async () => {\n      try {\n        await this.metadata(client, filePath);\n        return true;\n      } catch (error) {\n        if (error instanceof ApiError && error.statusCode === 404) {\n          return false;\n        }\n        throw error;\n      }\n    });\n  }\n\n  async metadata(\n    client: WorkspaceClient,\n    filePath: string,\n  ): Promise<FileMetadata> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"metadata\", { \"files.path\": resolvedPath }, async () => {\n      const response = await client.files.getMetadata({\n        file_path: resolvedPath,\n      });\n      return {\n        contentLength: response[\"content-length\"],\n        contentType: contentTypeFromPath(\n          filePath,\n          response[\"content-type\"],\n          this.customContentTypes,\n        ),\n        lastModified: response[\"last-modified\"],\n      };\n    });\n  }\n\n  async upload(\n    client: WorkspaceClient,\n    filePath: string,\n    contents: ReadableStream | Buffer | string,\n    options?: { overwrite?: boolean },\n  ): Promise<void> {\n    const resolvedPath = this.resolvePath(filePath);\n\n    return this.traced(\"upload\", { \"files.path\": resolvedPath }, async () => {\n      const body = contents;\n      const overwrite = options?.overwrite ?? true;\n\n      // Workaround: The SDK's files.upload() has two bugs:\n      // 1. It ignores the `contents` field (sets body to undefined)\n      // 2. apiClient.request() checks `instanceof` against its own ReadableStream\n      //    subclass, so standard ReadableStream instances get JSON.stringified to \"{}\"\n      // Bypass both by calling the REST API directly with SDK-provided auth.\n      const hostValue = client.config.host;\n      if (!hostValue) {\n        throw new Error(\n          \"Databricks host is not configured. Set DATABRICKS_HOST or configure client.config.host.\",\n        );\n      }\n      const host = hostValue.startsWith(\"http\")\n        ? hostValue\n        : `https://${hostValue}`;\n      const url = new URL(`/api/2.0/fs/files${resolvedPath}`, host);\n      url.searchParams.set(\"overwrite\", String(overwrite));\n\n      const headers = new Headers({\n        \"Content-Type\": \"application/octet-stream\",\n        \"User-Agent\": client.apiClient.userAgent(),\n      });\n      const fetchOptions: RequestInit = { method: \"PUT\", headers, body };\n\n      if (body instanceof ReadableStream) {\n        fetchOptions.duplex = \"half\";\n      } else if (body instanceof Buffer) {\n        headers.set(\"Content-Length\", String(body.length));\n      } else if (typeof body === \"string\") {\n        headers.set(\"Content-Length\", String(Buffer.byteLength(body)));\n      }\n\n      await client.config.authenticate(headers);\n\n      const res = await fetch(url.toString(), fetchOptions);\n\n      if (!res.ok) {\n        const text = await res.text();\n        logger.error(`Upload failed (${res.status}): ${text}`);\n        const safeMessage = text.length > 200 ? `${text.slice(0, 200)}…` : text;\n        throw new ApiError(\n          `Upload failed: ${safeMessage}`,\n          \"UPLOAD_FAILED\",\n          res.status,\n          undefined,\n          [],\n        );\n      }\n    });\n  }\n\n  async createDirectory(\n    client: WorkspaceClient,\n    directoryPath: string,\n  ): Promise<void> {\n    const resolvedPath = this.resolvePath(directoryPath);\n    return this.traced(\n      \"createDirectory\",\n      { \"files.path\": resolvedPath },\n      async () => {\n        await client.files.createDirectory({\n          directory_path: resolvedPath,\n        });\n      },\n    );\n  }\n\n  async delete(client: WorkspaceClient, filePath: string): Promise<void> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"delete\", { \"files.path\": resolvedPath }, async () => {\n      await client.files.delete({\n        file_path: resolvedPath,\n      });\n    });\n  }\n\n  async preview(\n    client: WorkspaceClient,\n    filePath: string,\n    options?: { maxChars?: number },\n  ): Promise<FilePreview> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"preview\", { \"files.path\": resolvedPath }, async () => {\n      const meta = await this.metadata(client, filePath);\n      const isText = isTextContentType(meta.contentType);\n      const isImage = meta.contentType?.startsWith(\"image/\") || false;\n\n      if (!isText) {\n        return { ...meta, textPreview: null, isText: false, isImage };\n      }\n\n      const response = await client.files.download({\n        file_path: resolvedPath,\n      });\n      if (!response.contents) {\n        return { ...meta, textPreview: \"\", isText: true, isImage: false };\n      }\n\n      const reader = response.contents.getReader();\n      const decoder = new TextDecoder();\n      let preview = \"\";\n      const maxChars = options?.maxChars ?? 1024;\n\n      while (preview.length < maxChars) {\n        const { done, value } = await reader.read();\n        if (done) break;\n        preview += decoder.decode(value, { stream: true });\n      }\n      preview += decoder.decode();\n      await reader.cancel();\n\n      if (preview.length > maxChars) {\n        preview = preview.slice(0, maxChars);\n      }\n\n      return { ...meta, textPreview: preview, isText: true, isImage: false };\n    });\n  }\n}\n"],"mappings":";;;;;;;;;AAyBA,MAAM,SAAS,aAAa,mBAAmB;;;;;;;;;;;;;;AAe/C,MAAM,6BAA6B,IAAI,mBAEpC;;;;;;;;AASH,SAAgB,2BACd,YACA,IACY;AACZ,QAAO,2BAA2B,IAAI,YAAY,GAAG;;AAUvD,IAAa,iBAAb,MAA4B;CAC1B,AAAiB,OAAO;CACxB,AAAQ;CACR,AAAiB;CAEjB,AAAiB;CACjB,AAAiB;CAKjB,YAAY,QAA8B;AACxC,OAAK,gBAAgB,OAAO;AAC5B,OAAK,qBAAqB,OAAO;AAEjC,OAAK,YAAY,iBAAiB,YAAY,KAAK,MAAM,OAAO,UAAU;AAC1E,OAAK,mBAAmB;GACtB,gBAAgB,KAAK,UAClB,UAAU,CACV,cAAc,yBAAyB;IACtC,aAAa;IACb,MAAM;IACP,CAAC;GACJ,mBAAmB,KAAK,UACrB,UAAU,CACV,gBAAgB,4BAA4B;IAC3C,aAAa;IACb,MAAM;IACP,CAAC;GACL;;CAGH,YAAY,UAA0B;AACpC,MAAI,SAAS,SAAS,KACpB,OAAM,IAAI,MACR,uDAAuD,SAAS,OAAO,IACxE;AAEH,MAAI,SAAS,SAAS,KAAK,CACzB,OAAM,IAAI,MAAM,oCAAoC;AAItD,MADiB,SAAS,MAAM,IAAI,CACvB,MAAM,MAAM,MAAM,KAAK,CAClC,OAAM,IAAI,MAAM,2CAAyC;AAE3D,MAAI,SAAS,WAAW,IAAI,EAAE;AAC5B,OAAI,CAAC,SAAS,WAAW,YAAY,CACnC,OAAM,IAAI,MACR,oIAED;AAEH,UAAO;;AAET,MAAI,CAAC,KAAK,cACR,OAAM,IAAI,MACR,qGACD;AAEH,SAAO,GAAG,KAAK,cAAc,GAAG;;CAGlC,MAAc,OACZ,WACA,YACA,IACY;EACZ,MAAM,YAAY,KAAK,KAAK;EAC5B,IAAI,UAAU;EAKd,MAAM,UAAU,2BAA2B,UAAU;AAErD,SAAO,KAAK,UAAU,gBACpB,SAAS,aACT;GACE,MAAM,SAAS;GACf,YAAY;IACV,mBAAmB;IACnB,GAAI,WAAW,EAAE;IACjB,GAAG;IACJ;GACF,EACD,OAAO,SAAe;AACpB,OAAI;IACF,MAAM,SAAS,MAAM,GAAG,KAAK;AAC7B,cAAU;AACV,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU;KACb,MAAM,eAAe;KACrB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;KAChE,CAAC;AACF,UAAM;aACE;AACR,SAAK,KAAK;IACV,MAAM,WAAW,KAAK,KAAK,GAAG;IAC9B,MAAM,cAAc;KAClB,mBAAmB;KACnB,SAAS,OAAO,QAAQ;KACzB;AACD,SAAK,iBAAiB,eAAe,IAAI,GAAG,YAAY;AACxD,SAAK,iBAAiB,kBAAkB,OAAO,UAAU,YAAY;;KAGzE;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;CAGH,MAAM,KACJ,QACA,eAC2B;EAC3B,MAAM,eAAe,gBACjB,KAAK,YAAY,cAAc,GAC/B,KAAK;AACT,MAAI,CAAC,aACH,OAAM,IAAI,MAAM,wDAAwD;AAG1E,SAAO,KAAK,OAAO,QAAQ,EAAE,cAAc,cAAc,EAAE,YAAY;GACrE,MAAM,UAA4B,EAAE;AACpC,cAAW,MAAM,SAAS,OAAO,MAAM,sBAAsB,EAC3D,gBAAgB,cACjB,CAAC,CACA,SAAQ,KAAK,MAAM;AAErB,UAAO;IACP;;CAGJ,MAAM,KACJ,QACA,UACA,SACiB;EACjB,MAAM,eAAe,KAAK,YAAY,SAAS;EAC/C,MAAM,UAAU,SAAS,WAAW;AACpC,SAAO,KAAK,OAAO,QAAQ,EAAE,cAAc,cAAc,EAAE,YAAY;GACrE,MAAM,WAAW,MAAM,KAAK,SAAS,QAAQ,SAAS;AACtD,OAAI,CAAC,SAAS,SACZ,QAAO;GAET,MAAM,SAAS,SAAS,SAAS,WAAW;GAC5C,MAAM,UAAU,IAAI,aAAa;GACjC,IAAI,SAAS;GACb,IAAI,YAAY;AAChB,UAAO,MAAM;IACX,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,MAAM;AAC3C,QAAI,KAAM;AACV,iBAAa,MAAM;AACnB,QAAI,YAAY,SAAS;AACvB,WAAM,OAAO,QAAQ;AACrB,WAAM,IAAI,MACR,mCAAmC,QAAQ,0CAC5C;;AAEH,cAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM,CAAC;;AAEnD,aAAU,QAAQ,QAAQ;AAC1B,UAAO;IACP;;CAGJ,MAAM,SACJ,QACA,UAC2B;EAC3B,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,YAAY,EAAE,cAAc,cAAc,EAAE,YAAY;AACzE,UAAO,OAAO,MAAM,SAAS,EAC3B,WAAW,cACZ,CAAC;IACF;;CAGJ,MAAM,OAAO,QAAyB,UAAoC;EACxE,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;AACvE,OAAI;AACF,UAAM,KAAK,SAAS,QAAQ,SAAS;AACrC,WAAO;YACA,OAAO;AACd,QAAI,iBAAiB,YAAY,MAAM,eAAe,IACpD,QAAO;AAET,UAAM;;IAER;;CAGJ,MAAM,SACJ,QACA,UACuB;EACvB,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,YAAY,EAAE,cAAc,cAAc,EAAE,YAAY;GACzE,MAAM,WAAW,MAAM,OAAO,MAAM,YAAY,EAC9C,WAAW,cACZ,CAAC;AACF,UAAO;IACL,eAAe,SAAS;IACxB,aAAa,oBACX,UACA,SAAS,iBACT,KAAK,mBACN;IACD,cAAc,SAAS;IACxB;IACD;;CAGJ,MAAM,OACJ,QACA,UACA,UACA,SACe;EACf,MAAM,eAAe,KAAK,YAAY,SAAS;AAE/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;GACvE,MAAM,OAAO;GACb,MAAM,YAAY,SAAS,aAAa;GAOxC,MAAM,YAAY,OAAO,OAAO;AAChC,OAAI,CAAC,UACH,OAAM,IAAI,MACR,0FACD;GAEH,MAAM,OAAO,UAAU,WAAW,OAAO,GACrC,YACA,WAAW;GACf,MAAM,MAAM,IAAI,IAAI,oBAAoB,gBAAgB,KAAK;AAC7D,OAAI,aAAa,IAAI,aAAa,OAAO,UAAU,CAAC;GAEpD,MAAM,UAAU,IAAI,QAAQ;IAC1B,gBAAgB;IAChB,cAAc,OAAO,UAAU,WAAW;IAC3C,CAAC;GACF,MAAM,eAA4B;IAAE,QAAQ;IAAO;IAAS;IAAM;AAElE,OAAI,gBAAgB,eAClB,cAAa,SAAS;YACb,gBAAgB,OACzB,SAAQ,IAAI,kBAAkB,OAAO,KAAK,OAAO,CAAC;YACzC,OAAO,SAAS,SACzB,SAAQ,IAAI,kBAAkB,OAAO,OAAO,WAAW,KAAK,CAAC,CAAC;AAGhE,SAAM,OAAO,OAAO,aAAa,QAAQ;GAEzC,MAAM,MAAM,MAAM,MAAM,IAAI,UAAU,EAAE,aAAa;AAErD,OAAI,CAAC,IAAI,IAAI;IACX,MAAM,OAAO,MAAM,IAAI,MAAM;AAC7B,WAAO,MAAM,kBAAkB,IAAI,OAAO,KAAK,OAAO;AAEtD,UAAM,IAAI,SACR,kBAFkB,KAAK,SAAS,MAAM,GAAG,KAAK,MAAM,GAAG,IAAI,CAAC,KAAK,QAGjE,iBACA,IAAI,QACJ,QACA,EAAE,CACH;;IAEH;;CAGJ,MAAM,gBACJ,QACA,eACe;EACf,MAAM,eAAe,KAAK,YAAY,cAAc;AACpD,SAAO,KAAK,OACV,mBACA,EAAE,cAAc,cAAc,EAC9B,YAAY;AACV,SAAM,OAAO,MAAM,gBAAgB,EACjC,gBAAgB,cACjB,CAAC;IAEL;;CAGH,MAAM,OAAO,QAAyB,UAAiC;EACrE,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;AACvE,SAAM,OAAO,MAAM,OAAO,EACxB,WAAW,cACZ,CAAC;IACF;;CAGJ,MAAM,QACJ,QACA,UACA,SACsB;EACtB,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,WAAW,EAAE,cAAc,cAAc,EAAE,YAAY;GACxE,MAAM,OAAO,MAAM,KAAK,SAAS,QAAQ,SAAS;GAClD,MAAM,SAAS,kBAAkB,KAAK,YAAY;GAClD,MAAM,UAAU,KAAK,aAAa,WAAW,SAAS,IAAI;AAE1D,OAAI,CAAC,OACH,QAAO;IAAE,GAAG;IAAM,aAAa;IAAM,QAAQ;IAAO;IAAS;GAG/D,MAAM,WAAW,MAAM,OAAO,MAAM,SAAS,EAC3C,WAAW,cACZ,CAAC;AACF,OAAI,CAAC,SAAS,SACZ,QAAO;IAAE,GAAG;IAAM,aAAa;IAAI,QAAQ;IAAM,SAAS;IAAO;GAGnE,MAAM,SAAS,SAAS,SAAS,WAAW;GAC5C,MAAM,UAAU,IAAI,aAAa;GACjC,IAAI,UAAU;GACd,MAAM,WAAW,SAAS,YAAY;AAEtC,UAAO,QAAQ,SAAS,UAAU;IAChC,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,MAAM;AAC3C,QAAI,KAAM;AACV,eAAW,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM,CAAC;;AAEpD,cAAW,QAAQ,QAAQ;AAC3B,SAAM,OAAO,QAAQ;AAErB,OAAI,QAAQ,SAAS,SACnB,WAAU,QAAQ,MAAM,GAAG,SAAS;AAGtC,UAAO;IAAE,GAAG;IAAM,aAAa;IAAS,QAAQ;IAAM,SAAS;IAAO;IACtE"}
+{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/files/client.ts"],"sourcesContent":["import { AsyncLocalStorage } from \"node:async_hooks\";\nimport type { TelemetryOptions } from \"shared\";\nimport { createLogger } from \"../../logging/logger\";\nimport type {\n  DirectoryEntry,\n  DownloadResponse,\n  FileMetadata,\n  FilePreview,\n} from \"../../plugins/files/types\";\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 { ApiError, type WorkspaceClient } from \"../../workspace-client\";\nimport {\n  contentTypeFromPath,\n  FILES_MAX_READ_SIZE,\n  isTextContentType,\n} from \"./defaults\";\n\nconst logger = createLogger(\"connectors:files\");\n\n/**\n * Ambient span-attribute propagation for `FilesConnector.traced()`.\n *\n * Callers (e.g. the plugin's `_withAuthModeAttributes` wrapper) set extra\n * span attributes here via `runWithFilesSpanAttributes(attrs, fn)`. The\n * connector's `traced()` decorator reads them and merges them into the\n * span it creates around the SDK call. This lets the plugin tag spans with\n * `files.auth_mode` without opening a duplicate `files.<op>` span.\n *\n * AsyncLocalStorage is used so concurrent requests don't see each other's\n * attributes. Outside an active scope, `getStore()` returns `undefined` and\n * the connector falls back to the static attribute set.\n */\nconst filesSpanAttributesStorage = new AsyncLocalStorage<\n  Record<string, string>\n>();\n\n/**\n * Run `fn` with the supplied attributes attached to whatever span the\n * `FilesConnector` opens for its SDK call. Used to propagate request-scoped\n * attributes (e.g. `files.auth_mode`) onto the connector's span without\n * opening a parent span — avoids the 2x span allocation that\n * `startActiveSpan` parented otherwise.\n */\nexport function runWithFilesSpanAttributes<T>(\n  attributes: Record<string, string>,\n  fn: () => Promise<T>,\n): Promise<T> {\n  return filesSpanAttributesStorage.run(attributes, fn);\n}\n\ninterface FilesConnectorConfig {\n  defaultVolume?: string;\n  timeout?: number;\n  telemetry?: TelemetryOptions;\n  customContentTypes?: Record<string, string>;\n}\n\nexport class FilesConnector {\n  private readonly name = \"files\";\n  private defaultVolume: string | undefined;\n  private readonly customContentTypes: Record<string, string> | undefined;\n\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    operationCount: Counter;\n    operationDuration: Histogram;\n  };\n\n  constructor(config: FilesConnectorConfig) {\n    this.defaultVolume = config.defaultVolume;\n    this.customContentTypes = config.customContentTypes;\n\n    this.telemetry = TelemetryManager.getProvider(this.name, config.telemetry);\n    this.telemetryMetrics = {\n      operationCount: this.telemetry\n        .getMeter()\n        .createCounter(\"files.operation.count\", {\n          description: \"Total number of file operations\",\n          unit: \"1\",\n        }),\n      operationDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"files.operation.duration\", {\n          description: \"Duration of file operations\",\n          unit: \"ms\",\n        }),\n    };\n  }\n\n  resolvePath(filePath: string): string {\n    if (filePath.length > 4096) {\n      throw new Error(\n        `Path exceeds maximum length of 4096 characters (got ${filePath.length}).`,\n      );\n    }\n    if (filePath.includes(\"\\0\")) {\n      throw new Error(\"Path must not contain null bytes.\");\n    }\n\n    const segments = filePath.split(\"/\");\n    if (segments.some((s) => s === \"..\")) {\n      throw new Error('Path traversal (\"../\") is not allowed.');\n    }\n    if (filePath.startsWith(\"/\")) {\n      if (!filePath.startsWith(\"/Volumes/\")) {\n        throw new Error(\n          'Absolute paths must start with \"/Volumes/\". ' +\n            \"Unity Catalog volume paths follow the format: /Volumes/<catalog>/<schema>/<volume>/\",\n        );\n      }\n      return filePath;\n    }\n    if (!this.defaultVolume) {\n      throw new Error(\n        \"Cannot resolve relative path: no default volume set. Use an absolute path or set a default volume.\",\n      );\n    }\n    return `${this.defaultVolume}/${filePath}`;\n  }\n\n  private async traced<T>(\n    operation: string,\n    attributes: Record<string, string>,\n    fn: (span: Span) => Promise<T>,\n  ): Promise<T> {\n    const startTime = Date.now();\n    let success = false;\n\n    // Pull any ambient attributes set by `runWithFilesSpanAttributes` (e.g.\n    // `files.auth_mode` from the plugin layer). Static `attributes` win on\n    // collision so callers can override per-call.\n    const ambient = filesSpanAttributesStorage.getStore();\n\n    return this.telemetry.startActiveSpan(\n      `files.${operation}`,\n      {\n        kind: SpanKind.CLIENT,\n        attributes: {\n          \"files.operation\": operation,\n          ...(ambient ?? {}),\n          ...attributes,\n        },\n      },\n      async (span: Span) => {\n        try {\n          const result = await fn(span);\n          success = true;\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\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          throw error;\n        } finally {\n          span.end();\n          const duration = Date.now() - startTime;\n          const metricAttrs = {\n            \"files.operation\": operation,\n            success: String(success),\n          };\n          this.telemetryMetrics.operationCount.add(1, metricAttrs);\n          this.telemetryMetrics.operationDuration.record(duration, metricAttrs);\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  async list(\n    client: WorkspaceClient,\n    directoryPath?: string,\n  ): Promise<DirectoryEntry[]> {\n    const resolvedPath = directoryPath\n      ? this.resolvePath(directoryPath)\n      : this.defaultVolume;\n    if (!resolvedPath) {\n      throw new Error(\"No directory path provided and no default volume set.\");\n    }\n\n    return this.traced(\"list\", { \"files.path\": resolvedPath }, async () => {\n      const entries: DirectoryEntry[] = [];\n      for await (const entry of client.files.listDirectoryContents({\n        directory_path: resolvedPath,\n      })) {\n        entries.push(entry);\n      }\n      return entries;\n    });\n  }\n\n  async read(\n    client: WorkspaceClient,\n    filePath: string,\n    options?: { maxSize?: number },\n  ): Promise<string> {\n    const resolvedPath = this.resolvePath(filePath);\n    const maxSize = options?.maxSize ?? FILES_MAX_READ_SIZE;\n    return this.traced(\"read\", { \"files.path\": resolvedPath }, async () => {\n      const response = await this.download(client, filePath);\n      if (!response.contents) {\n        return \"\";\n      }\n      const reader = response.contents.getReader();\n      const decoder = new TextDecoder();\n      let result = \"\";\n      let bytesRead = 0;\n      while (true) {\n        const { done, value } = await reader.read();\n        if (done) break;\n        bytesRead += value.byteLength;\n        if (bytesRead > maxSize) {\n          await reader.cancel();\n          throw new Error(\n            `File exceeds maximum read size (${maxSize} bytes). Use download() for large files.`,\n          );\n        }\n        result += decoder.decode(value, { stream: true });\n      }\n      result += decoder.decode();\n      return result;\n    });\n  }\n\n  async download(\n    client: WorkspaceClient,\n    filePath: string,\n  ): Promise<DownloadResponse> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"download\", { \"files.path\": resolvedPath }, async () => {\n      return client.files.download({\n        file_path: resolvedPath,\n      });\n    });\n  }\n\n  async exists(client: WorkspaceClient, filePath: string): Promise<boolean> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"exists\", { \"files.path\": resolvedPath }, async () => {\n      try {\n        await this.metadata(client, filePath);\n        return true;\n      } catch (error) {\n        if (error instanceof ApiError && error.statusCode === 404) {\n          return false;\n        }\n        throw error;\n      }\n    });\n  }\n\n  async metadata(\n    client: WorkspaceClient,\n    filePath: string,\n  ): Promise<FileMetadata> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"metadata\", { \"files.path\": resolvedPath }, async () => {\n      const response = await client.files.getMetadata({\n        file_path: resolvedPath,\n      });\n      return {\n        contentLength: response[\"content-length\"],\n        contentType: contentTypeFromPath(\n          filePath,\n          response[\"content-type\"],\n          this.customContentTypes,\n        ),\n        lastModified: response[\"last-modified\"],\n      };\n    });\n  }\n\n  async upload(\n    client: WorkspaceClient,\n    filePath: string,\n    contents: ReadableStream | Buffer | string,\n    options?: { overwrite?: boolean },\n  ): Promise<void> {\n    const resolvedPath = this.resolvePath(filePath);\n\n    return this.traced(\"upload\", { \"files.path\": resolvedPath }, async () => {\n      const body = contents;\n      const overwrite = options?.overwrite ?? true;\n\n      // Workaround: The SDK's files.upload() has two bugs:\n      // 1. It ignores the `contents` field (sets body to undefined)\n      // 2. apiClient.request() checks `instanceof` against its own ReadableStream\n      //    subclass, so standard ReadableStream instances get JSON.stringified to \"{}\"\n      // Bypass both by calling the REST API directly with SDK-provided auth.\n      const hostValue = client.config.host;\n      if (!hostValue) {\n        throw new Error(\n          \"Databricks host is not configured. Set DATABRICKS_HOST or configure client.config.host.\",\n        );\n      }\n      const host = hostValue.startsWith(\"http\")\n        ? hostValue\n        : `https://${hostValue}`;\n      const url = new URL(`/api/2.0/fs/files${resolvedPath}`, host);\n      url.searchParams.set(\"overwrite\", String(overwrite));\n\n      const headers = new Headers({\n        \"Content-Type\": \"application/octet-stream\",\n        \"User-Agent\": client.apiClient.userAgent(),\n      });\n      const fetchOptions: RequestInit = { method: \"PUT\", headers, body };\n\n      if (body instanceof ReadableStream) {\n        fetchOptions.duplex = \"half\";\n      } else if (body instanceof Buffer) {\n        headers.set(\"Content-Length\", String(body.length));\n      } else if (typeof body === \"string\") {\n        headers.set(\"Content-Length\", String(Buffer.byteLength(body)));\n      }\n\n      await client.config.authenticate(headers);\n\n      const res = await fetch(url.toString(), fetchOptions);\n\n      if (!res.ok) {\n        const text = await res.text();\n        logger.error(`Upload failed (${res.status}): ${text}`);\n        const safeMessage = text.length > 200 ? `${text.slice(0, 200)}…` : text;\n        throw new ApiError(\n          `Upload failed: ${safeMessage}`,\n          \"UPLOAD_FAILED\",\n          res.status,\n          undefined,\n          [],\n        );\n      }\n    });\n  }\n\n  async createDirectory(\n    client: WorkspaceClient,\n    directoryPath: string,\n  ): Promise<void> {\n    const resolvedPath = this.resolvePath(directoryPath);\n    return this.traced(\n      \"createDirectory\",\n      { \"files.path\": resolvedPath },\n      async () => {\n        await client.files.createDirectory({\n          directory_path: resolvedPath,\n        });\n      },\n    );\n  }\n\n  async delete(client: WorkspaceClient, filePath: string): Promise<void> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"delete\", { \"files.path\": resolvedPath }, async () => {\n      await client.files.delete({\n        file_path: resolvedPath,\n      });\n    });\n  }\n\n  async preview(\n    client: WorkspaceClient,\n    filePath: string,\n    options?: { maxChars?: number },\n  ): Promise<FilePreview> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"preview\", { \"files.path\": resolvedPath }, async () => {\n      const meta = await this.metadata(client, filePath);\n      const isText = isTextContentType(meta.contentType);\n      const isImage = meta.contentType?.startsWith(\"image/\") || false;\n\n      if (!isText) {\n        return { ...meta, textPreview: null, isText: false, isImage };\n      }\n\n      const response = await client.files.download({\n        file_path: resolvedPath,\n      });\n      if (!response.contents) {\n        return { ...meta, textPreview: \"\", isText: true, isImage: false };\n      }\n\n      const reader = response.contents.getReader();\n      const decoder = new TextDecoder();\n      let preview = \"\";\n      const maxChars = options?.maxChars ?? 1024;\n\n      while (preview.length < maxChars) {\n        const { done, value } = await reader.read();\n        if (done) break;\n        preview += decoder.decode(value, { stream: true });\n      }\n      preview += decoder.decode();\n      await reader.cancel();\n\n      if (preview.length > maxChars) {\n        preview = preview.slice(0, maxChars);\n      }\n\n      return { ...meta, textPreview: preview, isText: true, isImage: false };\n    });\n  }\n}\n"],"mappings":";;;;;;;;AAyBA,MAAM,SAAS,aAAa,mBAAmB;;;;;;;;;;;;;;AAe/C,MAAM,6BAA6B,IAAI,mBAEpC;;;;;;;;AASH,SAAgB,2BACd,YACA,IACY;AACZ,QAAO,2BAA2B,IAAI,YAAY,GAAG;;AAUvD,IAAa,iBAAb,MAA4B;CAC1B,AAAiB,OAAO;CACxB,AAAQ;CACR,AAAiB;CAEjB,AAAiB;CACjB,AAAiB;CAKjB,YAAY,QAA8B;AACxC,OAAK,gBAAgB,OAAO;AAC5B,OAAK,qBAAqB,OAAO;AAEjC,OAAK,YAAY,iBAAiB,YAAY,KAAK,MAAM,OAAO,UAAU;AAC1E,OAAK,mBAAmB;GACtB,gBAAgB,KAAK,UAClB,UAAU,CACV,cAAc,yBAAyB;IACtC,aAAa;IACb,MAAM;IACP,CAAC;GACJ,mBAAmB,KAAK,UACrB,UAAU,CACV,gBAAgB,4BAA4B;IAC3C,aAAa;IACb,MAAM;IACP,CAAC;GACL;;CAGH,YAAY,UAA0B;AACpC,MAAI,SAAS,SAAS,KACpB,OAAM,IAAI,MACR,uDAAuD,SAAS,OAAO,IACxE;AAEH,MAAI,SAAS,SAAS,KAAK,CACzB,OAAM,IAAI,MAAM,oCAAoC;AAItD,MADiB,SAAS,MAAM,IAAI,CACvB,MAAM,MAAM,MAAM,KAAK,CAClC,OAAM,IAAI,MAAM,2CAAyC;AAE3D,MAAI,SAAS,WAAW,IAAI,EAAE;AAC5B,OAAI,CAAC,SAAS,WAAW,YAAY,CACnC,OAAM,IAAI,MACR,oIAED;AAEH,UAAO;;AAET,MAAI,CAAC,KAAK,cACR,OAAM,IAAI,MACR,qGACD;AAEH,SAAO,GAAG,KAAK,cAAc,GAAG;;CAGlC,MAAc,OACZ,WACA,YACA,IACY;EACZ,MAAM,YAAY,KAAK,KAAK;EAC5B,IAAI,UAAU;EAKd,MAAM,UAAU,2BAA2B,UAAU;AAErD,SAAO,KAAK,UAAU,gBACpB,SAAS,aACT;GACE,MAAM,SAAS;GACf,YAAY;IACV,mBAAmB;IACnB,GAAI,WAAW,EAAE;IACjB,GAAG;IACJ;GACF,EACD,OAAO,SAAe;AACpB,OAAI;IACF,MAAM,SAAS,MAAM,GAAG,KAAK;AAC7B,cAAU;AACV,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU;KACb,MAAM,eAAe;KACrB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;KAChE,CAAC;AACF,UAAM;aACE;AACR,SAAK,KAAK;IACV,MAAM,WAAW,KAAK,KAAK,GAAG;IAC9B,MAAM,cAAc;KAClB,mBAAmB;KACnB,SAAS,OAAO,QAAQ;KACzB;AACD,SAAK,iBAAiB,eAAe,IAAI,GAAG,YAAY;AACxD,SAAK,iBAAiB,kBAAkB,OAAO,UAAU,YAAY;;KAGzE;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;CAGH,MAAM,KACJ,QACA,eAC2B;EAC3B,MAAM,eAAe,gBACjB,KAAK,YAAY,cAAc,GAC/B,KAAK;AACT,MAAI,CAAC,aACH,OAAM,IAAI,MAAM,wDAAwD;AAG1E,SAAO,KAAK,OAAO,QAAQ,EAAE,cAAc,cAAc,EAAE,YAAY;GACrE,MAAM,UAA4B,EAAE;AACpC,cAAW,MAAM,SAAS,OAAO,MAAM,sBAAsB,EAC3D,gBAAgB,cACjB,CAAC,CACA,SAAQ,KAAK,MAAM;AAErB,UAAO;IACP;;CAGJ,MAAM,KACJ,QACA,UACA,SACiB;EACjB,MAAM,eAAe,KAAK,YAAY,SAAS;EAC/C,MAAM,UAAU,SAAS,WAAW;AACpC,SAAO,KAAK,OAAO,QAAQ,EAAE,cAAc,cAAc,EAAE,YAAY;GACrE,MAAM,WAAW,MAAM,KAAK,SAAS,QAAQ,SAAS;AACtD,OAAI,CAAC,SAAS,SACZ,QAAO;GAET,MAAM,SAAS,SAAS,SAAS,WAAW;GAC5C,MAAM,UAAU,IAAI,aAAa;GACjC,IAAI,SAAS;GACb,IAAI,YAAY;AAChB,UAAO,MAAM;IACX,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,MAAM;AAC3C,QAAI,KAAM;AACV,iBAAa,MAAM;AACnB,QAAI,YAAY,SAAS;AACvB,WAAM,OAAO,QAAQ;AACrB,WAAM,IAAI,MACR,mCAAmC,QAAQ,0CAC5C;;AAEH,cAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM,CAAC;;AAEnD,aAAU,QAAQ,QAAQ;AAC1B,UAAO;IACP;;CAGJ,MAAM,SACJ,QACA,UAC2B;EAC3B,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,YAAY,EAAE,cAAc,cAAc,EAAE,YAAY;AACzE,UAAO,OAAO,MAAM,SAAS,EAC3B,WAAW,cACZ,CAAC;IACF;;CAGJ,MAAM,OAAO,QAAyB,UAAoC;EACxE,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;AACvE,OAAI;AACF,UAAM,KAAK,SAAS,QAAQ,SAAS;AACrC,WAAO;YACA,OAAO;AACd,QAAI,iBAAiB,YAAY,MAAM,eAAe,IACpD,QAAO;AAET,UAAM;;IAER;;CAGJ,MAAM,SACJ,QACA,UACuB;EACvB,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,YAAY,EAAE,cAAc,cAAc,EAAE,YAAY;GACzE,MAAM,WAAW,MAAM,OAAO,MAAM,YAAY,EAC9C,WAAW,cACZ,CAAC;AACF,UAAO;IACL,eAAe,SAAS;IACxB,aAAa,oBACX,UACA,SAAS,iBACT,KAAK,mBACN;IACD,cAAc,SAAS;IACxB;IACD;;CAGJ,MAAM,OACJ,QACA,UACA,UACA,SACe;EACf,MAAM,eAAe,KAAK,YAAY,SAAS;AAE/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;GACvE,MAAM,OAAO;GACb,MAAM,YAAY,SAAS,aAAa;GAOxC,MAAM,YAAY,OAAO,OAAO;AAChC,OAAI,CAAC,UACH,OAAM,IAAI,MACR,0FACD;GAEH,MAAM,OAAO,UAAU,WAAW,OAAO,GACrC,YACA,WAAW;GACf,MAAM,MAAM,IAAI,IAAI,oBAAoB,gBAAgB,KAAK;AAC7D,OAAI,aAAa,IAAI,aAAa,OAAO,UAAU,CAAC;GAEpD,MAAM,UAAU,IAAI,QAAQ;IAC1B,gBAAgB;IAChB,cAAc,OAAO,UAAU,WAAW;IAC3C,CAAC;GACF,MAAM,eAA4B;IAAE,QAAQ;IAAO;IAAS;IAAM;AAElE,OAAI,gBAAgB,eAClB,cAAa,SAAS;YACb,gBAAgB,OACzB,SAAQ,IAAI,kBAAkB,OAAO,KAAK,OAAO,CAAC;YACzC,OAAO,SAAS,SACzB,SAAQ,IAAI,kBAAkB,OAAO,OAAO,WAAW,KAAK,CAAC,CAAC;AAGhE,SAAM,OAAO,OAAO,aAAa,QAAQ;GAEzC,MAAM,MAAM,MAAM,MAAM,IAAI,UAAU,EAAE,aAAa;AAErD,OAAI,CAAC,IAAI,IAAI;IACX,MAAM,OAAO,MAAM,IAAI,MAAM;AAC7B,WAAO,MAAM,kBAAkB,IAAI,OAAO,KAAK,OAAO;AAEtD,UAAM,IAAI,SACR,kBAFkB,KAAK,SAAS,MAAM,GAAG,KAAK,MAAM,GAAG,IAAI,CAAC,KAAK,QAGjE,iBACA,IAAI,QACJ,QACA,EAAE,CACH;;IAEH;;CAGJ,MAAM,gBACJ,QACA,eACe;EACf,MAAM,eAAe,KAAK,YAAY,cAAc;AACpD,SAAO,KAAK,OACV,mBACA,EAAE,cAAc,cAAc,EAC9B,YAAY;AACV,SAAM,OAAO,MAAM,gBAAgB,EACjC,gBAAgB,cACjB,CAAC;IAEL;;CAGH,MAAM,OAAO,QAAyB,UAAiC;EACrE,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;AACvE,SAAM,OAAO,MAAM,OAAO,EACxB,WAAW,cACZ,CAAC;IACF;;CAGJ,MAAM,QACJ,QACA,UACA,SACsB;EACtB,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,WAAW,EAAE,cAAc,cAAc,EAAE,YAAY;GACxE,MAAM,OAAO,MAAM,KAAK,SAAS,QAAQ,SAAS;GAClD,MAAM,SAAS,kBAAkB,KAAK,YAAY;GAClD,MAAM,UAAU,KAAK,aAAa,WAAW,SAAS,IAAI;AAE1D,OAAI,CAAC,OACH,QAAO;IAAE,GAAG;IAAM,aAAa;IAAM,QAAQ;IAAO;IAAS;GAG/D,MAAM,WAAW,MAAM,OAAO,MAAM,SAAS,EAC3C,WAAW,cACZ,CAAC;AACF,OAAI,CAAC,SAAS,SACZ,QAAO;IAAE,GAAG;IAAM,aAAa;IAAI,QAAQ;IAAM,SAAS;IAAO;GAGnE,MAAM,SAAS,SAAS,SAAS,WAAW;GAC5C,MAAM,UAAU,IAAI,aAAa;GACjC,IAAI,UAAU;GACd,MAAM,WAAW,SAAS,YAAY;AAEtC,UAAO,QAAQ,SAAS,UAAU;IAChC,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,MAAM;AAC3C,QAAI,KAAM;AACV,eAAW,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM,CAAC;;AAEpD,cAAW,QAAQ,QAAQ;AAC3B,SAAM,OAAO,QAAQ;AAErB,OAAI,QAAQ,SAAS,SACnB,WAAU,QAAQ,MAAM,GAAG,SAAS;AAGtC,UAAO;IAAE,GAAG;IAAM,aAAa;IAAS,QAAQ;IAAM,SAAS;IAAO;IACtE"}

package/dist/connectors/genie/client.js

@@ -1,6 +1,5 @@
+import { Time, TimeUnits } from "../../shared/src/workspace-client/legacy.js";
 import { createLogger } from "../../logging/logger.js";
-import { Time, TimeUnits } from "../../workspace-client/legacy.js";
-import "../../workspace-client/index.js";
 import "../../logging/index.js";
 import { genieConnectorDefaults } from "./defaults.js";
 import { pollWaiter } from "./poll-waiter.js";

package/dist/connectors/genie/client.js.map

@@ -1 +1 @@
-{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/genie/client.ts"],"sourcesContent":["import { createLogger } from \"../../logging\";\nimport {\n  type GenieMessage,\n  Time,\n  TimeUnits,\n  type Waiter,\n  type WorkspaceClient,\n} from \"../../workspace-client\";\nimport { genieConnectorDefaults } from \"./defaults\";\nimport { pollWaiter } from \"./poll-waiter\";\nimport type {\n  GenieAttachmentResponse,\n  GenieConversationHistoryResponse,\n  GenieMessageResponse,\n  GenieStatementResponse,\n  GenieStreamEvent,\n} from \"./types\";\n\nconst logger = createLogger(\"connectors:genie\");\n\nconst GenieErrors = {\n  SPACE_ACCESS_DENIED: \"You don't have access to this Genie Space.\",\n  TABLE_PERMISSIONS:\n    \"You may not have access to the data tables. Please verify your table permissions.\",\n  REQUEST_FAILED: \"Genie request failed\",\n  QUERY_RESULT_FAILED: \"Failed to fetch query result\",\n} as const;\n\ntype CreateMessageWaiter = Waiter<GenieMessage, GenieMessage>;\n\ninterface GenieConnectorConfig {\n  timeout?: number;\n  maxMessages?: number;\n}\n\nfunction mapAttachments(message: GenieMessage): GenieAttachmentResponse[] {\n  return (\n    message.attachments?.map((att) => ({\n      attachmentId: att.attachment_id,\n      query: att.query\n        ? {\n            title: att.query.title,\n            description: att.query.description,\n            query: att.query.query,\n            statementId: att.query.statement_id,\n          }\n        : undefined,\n      text: att.text ? { content: att.text.content } : undefined,\n      suggestedQuestions: att.suggested_questions?.questions,\n    })) ?? []\n  );\n}\n\nfunction toMessageResponse(message: GenieMessage): GenieMessageResponse {\n  return {\n    messageId: message.message_id,\n    conversationId: message.conversation_id,\n    spaceId: message.space_id,\n    status: message.status ?? \"COMPLETED\",\n    content: message.content,\n    attachments: mapAttachments(message),\n    error: message.error?.error,\n  };\n}\n\nfunction classifyGenieError(error: unknown): string {\n  const message = error instanceof Error ? error.message : String(error);\n\n  if (message.includes(\"RESOURCE_DOES_NOT_EXIST\")) {\n    return GenieErrors.SPACE_ACCESS_DENIED;\n  }\n\n  if (\n    message.includes(\"failed to reach COMPLETED state\") &&\n    message.includes(\"FAILED\")\n  ) {\n    return GenieErrors.TABLE_PERMISSIONS;\n  }\n\n  return message || GenieErrors.REQUEST_FAILED;\n}\n\nexport class GenieConnector {\n  private readonly config: Required<GenieConnectorConfig>;\n\n  constructor(config: GenieConnectorConfig = {}) {\n    this.config = {\n      timeout: config.timeout ?? genieConnectorDefaults.timeout,\n      maxMessages: config.maxMessages ?? genieConnectorDefaults.maxMessages,\n    };\n  }\n\n  async startMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    content: string,\n    conversationId: string | undefined,\n  ): Promise<{\n    messageWaiter: CreateMessageWaiter;\n    conversationId: string;\n    messageId: string;\n  }> {\n    if (conversationId) {\n      const waiter = await workspaceClient.genie.createMessage({\n        space_id: spaceId,\n        conversation_id: conversationId,\n        content,\n      });\n      return {\n        messageWaiter: waiter,\n        conversationId,\n        messageId: waiter.message_id ?? \"\",\n      };\n    }\n    const start = await workspaceClient.genie.startConversation({\n      space_id: spaceId,\n      content,\n    });\n    return {\n      messageWaiter: start as unknown as CreateMessageWaiter,\n      conversationId: start.conversation_id,\n      messageId: start.message_id,\n    };\n  }\n\n  async waitForMessage(\n    messageWaiter: CreateMessageWaiter,\n    options?: { timeout?: number },\n  ): Promise<GenieMessage> {\n    const timeout = options?.timeout ?? this.config.timeout;\n    const waitOptions =\n      timeout > 0 ? { timeout: new Time(timeout, TimeUnits.milliseconds) } : {};\n    return messageWaiter.wait(waitOptions);\n  }\n\n  async listConversationMessages(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    options?: { pageSize?: number; pageToken?: string },\n  ): Promise<{\n    messages: GenieMessageResponse[];\n    nextPageToken: string | null;\n  }> {\n    const pageSize =\n      options?.pageSize ?? genieConnectorDefaults.initialPageSize;\n\n    const response = await workspaceClient.genie.listConversationMessages({\n      space_id: spaceId,\n      conversation_id: conversationId,\n      page_size: pageSize,\n      ...(options?.pageToken ? { page_token: options.pageToken } : {}),\n    });\n\n    const messages = (response.messages ?? []).reverse().map(toMessageResponse);\n\n    return {\n      messages,\n      nextPageToken: response.next_page_token ?? null,\n    };\n  }\n\n  async getMessageAttachmentQueryResult(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    messageId: string,\n    attachmentId: string,\n    _signal?: AbortSignal,\n  ): Promise<GenieStatementResponse> {\n    const response =\n      await workspaceClient.genie.getMessageAttachmentQueryResult({\n        space_id: spaceId,\n        conversation_id: conversationId,\n        message_id: messageId,\n        attachment_id: attachmentId,\n      });\n    return response.statement_response as GenieStatementResponse;\n  }\n\n  async *streamSendMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    content: string,\n    conversationId: string | undefined,\n    options?: { timeout?: number; signal?: AbortSignal },\n  ): AsyncGenerator<GenieStreamEvent> {\n    try {\n      const {\n        messageWaiter,\n        conversationId: resultConversationId,\n        messageId: resultMessageId,\n      } = await this.startMessage(\n        workspaceClient,\n        spaceId,\n        content,\n        conversationId,\n      );\n\n      yield {\n        type: \"message_start\",\n        conversationId: resultConversationId,\n        messageId: resultMessageId,\n        spaceId,\n      };\n\n      const timeout =\n        options?.timeout != null ? options.timeout : this.config.timeout;\n      const waitOptions =\n        timeout > 0\n          ? { timeout: new Time(timeout, TimeUnits.milliseconds) }\n          : {};\n\n      let completedMessage!: GenieMessage;\n      for await (const event of pollWaiter(messageWaiter, waitOptions)) {\n        if (event.type === \"progress\" && event.value.status) {\n          yield { type: \"status\", status: event.value.status };\n        } else if (event.type === \"completed\") {\n          completedMessage = event.value;\n        }\n      }\n\n      const messageResponse = toMessageResponse(completedMessage);\n      yield { type: \"message_result\", message: messageResponse };\n\n      yield* this.emitQueryResults(\n        workspaceClient,\n        spaceId,\n        resultConversationId,\n        messageResponse.messageId,\n        messageResponse,\n      );\n    } catch (error) {\n      logger.error(\n        \"Genie message error (spaceId=%s, conversationId=%s): %O\",\n        spaceId,\n        conversationId ?? \"new\",\n        error,\n      );\n      yield { type: \"error\", error: classifyGenieError(error) };\n    }\n  }\n\n  private async *emitQueryResults(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    messageId: string,\n    messageResponse: GenieMessageResponse,\n  ): AsyncGenerator<\n    Extract<GenieStreamEvent, { type: \"query_result\" } | { type: \"error\" }>\n  > {\n    const attachments = messageResponse.attachments ?? [];\n    for (const att of attachments) {\n      if (!att.query?.statementId || !att.attachmentId) continue;\n      try {\n        const data = await this.getMessageAttachmentQueryResult(\n          workspaceClient,\n          spaceId,\n          conversationId,\n          messageId,\n          att.attachmentId,\n        );\n        yield {\n          type: \"query_result\",\n          attachmentId: att.attachmentId,\n          statementId: att.query.statementId,\n          data,\n        };\n      } catch (error) {\n        logger.error(\n          \"Failed to fetch query result for attachment %s: %O\",\n          att.attachmentId,\n          error,\n        );\n        yield {\n          type: \"error\",\n          error: `${GenieErrors.QUERY_RESULT_FAILED} for attachment ${att.attachmentId}`,\n        };\n      }\n    }\n  }\n\n  async *streamConversation(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    options?: {\n      includeQueryResults?: boolean;\n      pageSize?: number;\n      pageToken?: string;\n      signal?: AbortSignal;\n    },\n  ): AsyncGenerator<GenieStreamEvent> {\n    const includeQueryResults = options?.includeQueryResults !== false;\n\n    try {\n      const { messages: messageResponses, nextPageToken } =\n        await this.listConversationMessages(\n          workspaceClient,\n          spaceId,\n          conversationId,\n          { pageSize: options?.pageSize, pageToken: options?.pageToken },\n        );\n\n      for (const messageResponse of messageResponses) {\n        yield { type: \"message_result\", message: messageResponse };\n      }\n\n      yield {\n        type: \"history_info\",\n        conversationId,\n        spaceId,\n        nextPageToken,\n        loadedCount: messageResponses.length,\n      };\n\n      if (includeQueryResults) {\n        const queryAttachments: Array<{\n          messageId: string;\n          attachmentId: string;\n          statementId: string;\n        }> = [];\n\n        for (const msg of messageResponses) {\n          for (const att of msg.attachments ?? []) {\n            if (att.query?.statementId && att.attachmentId) {\n              queryAttachments.push({\n                messageId: msg.messageId,\n                attachmentId: att.attachmentId,\n                statementId: att.query.statementId,\n              });\n            }\n          }\n        }\n\n        const results = await Promise.allSettled(\n          queryAttachments.map(async (att) => {\n            const data = await this.getMessageAttachmentQueryResult(\n              workspaceClient,\n              spaceId,\n              conversationId,\n              att.messageId,\n              att.attachmentId,\n            );\n            return {\n              attachmentId: att.attachmentId,\n              statementId: att.statementId,\n              data,\n            };\n          }),\n        );\n\n        for (const result of results) {\n          if (result.status === \"fulfilled\") {\n            yield {\n              type: \"query_result\",\n              attachmentId: result.value.attachmentId,\n              statementId: result.value.statementId,\n              data: result.value.data,\n            };\n          } else {\n            logger.error(\"Failed to fetch query result: %O\", result.reason);\n            yield {\n              type: \"error\",\n              error:\n                result.reason instanceof Error\n                  ? result.reason.message\n                  : GenieErrors.QUERY_RESULT_FAILED,\n            };\n          }\n        }\n      }\n    } catch (error) {\n      logger.error(\n        \"Genie getConversation error (spaceId=%s, conversationId=%s): %O\",\n        spaceId,\n        conversationId,\n        error,\n      );\n      yield { type: \"error\", error: classifyGenieError(error) };\n    }\n  }\n\n  /**\n   * Polls a single message via `getMessage` until it reaches a terminal\n   * state (`COMPLETED` or `FAILED`). Yields the same event types as\n   * `streamSendMessage` so callers can reuse the same SSE processing logic.\n   */\n  async *streamGetMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    messageId: string,\n    options?: { timeout?: number; pollInterval?: number; signal?: AbortSignal },\n  ): AsyncGenerator<GenieStreamEvent> {\n    const pollInterval = options?.pollInterval ?? 3_000;\n    const signal = options?.signal;\n    let lastStatus = \"\";\n\n    try {\n      while (true) {\n        if (signal?.aborted) return;\n\n        const message = await workspaceClient.genie.getMessage({\n          space_id: spaceId,\n          conversation_id: conversationId,\n          message_id: messageId,\n        });\n\n        if (message.status && message.status !== lastStatus) {\n          lastStatus = message.status;\n          yield { type: \"status\", status: message.status };\n        }\n\n        const isTerminal =\n          message.status === \"COMPLETED\" || message.status === \"FAILED\";\n        if (isTerminal) {\n          const messageResponse = toMessageResponse(message);\n          yield { type: \"message_result\", message: messageResponse };\n          yield* this.emitQueryResults(\n            workspaceClient,\n            spaceId,\n            conversationId,\n            messageId,\n            messageResponse,\n          );\n          return;\n        }\n\n        await new Promise<void>((resolve) => {\n          const timer = setTimeout(resolve, pollInterval);\n          signal?.addEventListener(\n            \"abort\",\n            () => {\n              clearTimeout(timer);\n              resolve();\n            },\n            { once: true },\n          );\n        });\n      }\n    } catch (error) {\n      if (signal?.aborted) return;\n      logger.error(\n        \"Genie getMessage poll error (spaceId=%s, conversationId=%s, messageId=%s): %O\",\n        spaceId,\n        conversationId,\n        messageId,\n        error,\n      );\n      yield { type: \"error\", error: classifyGenieError(error) };\n    }\n  }\n\n  async sendMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    content: string,\n    conversationId: string | undefined,\n  ): Promise<GenieMessageResponse> {\n    const { messageWaiter, conversationId: resultConversationId } =\n      await this.startMessage(\n        workspaceClient,\n        spaceId,\n        content,\n        conversationId,\n      );\n    const completedMessage = await this.waitForMessage(messageWaiter);\n    const messageResponse = toMessageResponse(completedMessage);\n    return {\n      ...messageResponse,\n      conversationId: resultConversationId,\n    };\n  }\n\n  async getConversation(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n  ): Promise<GenieConversationHistoryResponse> {\n    const allMessages: GenieMessageResponse[] = [];\n    let pageToken: string | undefined;\n\n    do {\n      const { messages, nextPageToken } = await this.listConversationMessages(\n        workspaceClient,\n        spaceId,\n        conversationId,\n        {\n          pageSize: genieConnectorDefaults.pageSize,\n          pageToken,\n        },\n      );\n      allMessages.push(...messages);\n      pageToken = nextPageToken ?? undefined;\n    } while (pageToken && allMessages.length < this.config.maxMessages);\n\n    return {\n      conversationId,\n      spaceId,\n      messages: allMessages.slice(0, this.config.maxMessages),\n    };\n  }\n}\n"],"mappings":";;;;;;;;AAkBA,MAAM,SAAS,aAAa,mBAAmB;AAE/C,MAAM,cAAc;CAClB,qBAAqB;CACrB,mBACE;CACF,gBAAgB;CAChB,qBAAqB;CACtB;AASD,SAAS,eAAe,SAAkD;AACxE,QACE,QAAQ,aAAa,KAAK,SAAS;EACjC,cAAc,IAAI;EAClB,OAAO,IAAI,QACP;GACE,OAAO,IAAI,MAAM;GACjB,aAAa,IAAI,MAAM;GACvB,OAAO,IAAI,MAAM;GACjB,aAAa,IAAI,MAAM;GACxB,GACD;EACJ,MAAM,IAAI,OAAO,EAAE,SAAS,IAAI,KAAK,SAAS,GAAG;EACjD,oBAAoB,IAAI,qBAAqB;EAC9C,EAAE,IAAI,EAAE;;AAIb,SAAS,kBAAkB,SAA6C;AACtE,QAAO;EACL,WAAW,QAAQ;EACnB,gBAAgB,QAAQ;EACxB,SAAS,QAAQ;EACjB,QAAQ,QAAQ,UAAU;EAC1B,SAAS,QAAQ;EACjB,aAAa,eAAe,QAAQ;EACpC,OAAO,QAAQ,OAAO;EACvB;;AAGH,SAAS,mBAAmB,OAAwB;CAClD,MAAM,UAAU,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;AAEtE,KAAI,QAAQ,SAAS,0BAA0B,CAC7C,QAAO,YAAY;AAGrB,KACE,QAAQ,SAAS,kCAAkC,IACnD,QAAQ,SAAS,SAAS,CAE1B,QAAO,YAAY;AAGrB,QAAO,WAAW,YAAY;;AAGhC,IAAa,iBAAb,MAA4B;CAC1B,AAAiB;CAEjB,YAAY,SAA+B,EAAE,EAAE;AAC7C,OAAK,SAAS;GACZ,SAAS,OAAO,WAAW,uBAAuB;GAClD,aAAa,OAAO,eAAe,uBAAuB;GAC3D;;CAGH,MAAM,aACJ,iBACA,SACA,SACA,gBAKC;AACD,MAAI,gBAAgB;GAClB,MAAM,SAAS,MAAM,gBAAgB,MAAM,cAAc;IACvD,UAAU;IACV,iBAAiB;IACjB;IACD,CAAC;AACF,UAAO;IACL,eAAe;IACf;IACA,WAAW,OAAO,cAAc;IACjC;;EAEH,MAAM,QAAQ,MAAM,gBAAgB,MAAM,kBAAkB;GAC1D,UAAU;GACV;GACD,CAAC;AACF,SAAO;GACL,eAAe;GACf,gBAAgB,MAAM;GACtB,WAAW,MAAM;GAClB;;CAGH,MAAM,eACJ,eACA,SACuB;EACvB,MAAM,UAAU,SAAS,WAAW,KAAK,OAAO;EAChD,MAAM,cACJ,UAAU,IAAI,EAAE,SAAS,IAAI,KAAK,SAAS,UAAU,aAAa,EAAE,GAAG,EAAE;AAC3E,SAAO,cAAc,KAAK,YAAY;;CAGxC,MAAM,yBACJ,iBACA,SACA,gBACA,SAIC;EACD,MAAM,WACJ,SAAS,YAAY,uBAAuB;EAE9C,MAAM,WAAW,MAAM,gBAAgB,MAAM,yBAAyB;GACpE,UAAU;GACV,iBAAiB;GACjB,WAAW;GACX,GAAI,SAAS,YAAY,EAAE,YAAY,QAAQ,WAAW,GAAG,EAAE;GAChE,CAAC;AAIF,SAAO;GACL,WAHgB,SAAS,YAAY,EAAE,EAAE,SAAS,CAAC,IAAI,kBAAkB;GAIzE,eAAe,SAAS,mBAAmB;GAC5C;;CAGH,MAAM,gCACJ,iBACA,SACA,gBACA,WACA,cACA,SACiC;AAQjC,UANE,MAAM,gBAAgB,MAAM,gCAAgC;GAC1D,UAAU;GACV,iBAAiB;GACjB,YAAY;GACZ,eAAe;GAChB,CAAC,EACY;;CAGlB,OAAO,kBACL,iBACA,SACA,SACA,gBACA,SACkC;AAClC,MAAI;GACF,MAAM,EACJ,eACA,gBAAgB,sBAChB,WAAW,oBACT,MAAM,KAAK,aACb,iBACA,SACA,SACA,eACD;AAED,SAAM;IACJ,MAAM;IACN,gBAAgB;IAChB,WAAW;IACX;IACD;GAED,MAAM,UACJ,SAAS,WAAW,OAAO,QAAQ,UAAU,KAAK,OAAO;GAC3D,MAAM,cACJ,UAAU,IACN,EAAE,SAAS,IAAI,KAAK,SAAS,UAAU,aAAa,EAAE,GACtD,EAAE;GAER,IAAI;AACJ,cAAW,MAAM,SAAS,WAAW,eAAe,YAAY,CAC9D,KAAI,MAAM,SAAS,cAAc,MAAM,MAAM,OAC3C,OAAM;IAAE,MAAM;IAAU,QAAQ,MAAM,MAAM;IAAQ;YAC3C,MAAM,SAAS,YACxB,oBAAmB,MAAM;GAI7B,MAAM,kBAAkB,kBAAkB,iBAAiB;AAC3D,SAAM;IAAE,MAAM;IAAkB,SAAS;IAAiB;AAE1D,UAAO,KAAK,iBACV,iBACA,SACA,sBACA,gBAAgB,WAChB,gBACD;WACM,OAAO;AACd,UAAO,MACL,2DACA,SACA,kBAAkB,OAClB,MACD;AACD,SAAM;IAAE,MAAM;IAAS,OAAO,mBAAmB,MAAM;IAAE;;;CAI7D,OAAe,iBACb,iBACA,SACA,gBACA,WACA,iBAGA;EACA,MAAM,cAAc,gBAAgB,eAAe,EAAE;AACrD,OAAK,MAAM,OAAO,aAAa;AAC7B,OAAI,CAAC,IAAI,OAAO,eAAe,CAAC,IAAI,aAAc;AAClD,OAAI;IACF,MAAM,OAAO,MAAM,KAAK,gCACtB,iBACA,SACA,gBACA,WACA,IAAI,aACL;AACD,UAAM;KACJ,MAAM;KACN,cAAc,IAAI;KAClB,aAAa,IAAI,MAAM;KACvB;KACD;YACM,OAAO;AACd,WAAO,MACL,sDACA,IAAI,cACJ,MACD;AACD,UAAM;KACJ,MAAM;KACN,OAAO,GAAG,YAAY,oBAAoB,kBAAkB,IAAI;KACjE;;;;CAKP,OAAO,mBACL,iBACA,SACA,gBACA,SAMkC;EAClC,MAAM,sBAAsB,SAAS,wBAAwB;AAE7D,MAAI;GACF,MAAM,EAAE,UAAU,kBAAkB,kBAClC,MAAM,KAAK,yBACT,iBACA,SACA,gBACA;IAAE,UAAU,SAAS;IAAU,WAAW,SAAS;IAAW,CAC/D;AAEH,QAAK,MAAM,mBAAmB,iBAC5B,OAAM;IAAE,MAAM;IAAkB,SAAS;IAAiB;AAG5D,SAAM;IACJ,MAAM;IACN;IACA;IACA;IACA,aAAa,iBAAiB;IAC/B;AAED,OAAI,qBAAqB;IACvB,MAAM,mBAID,EAAE;AAEP,SAAK,MAAM,OAAO,iBAChB,MAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CACrC,KAAI,IAAI,OAAO,eAAe,IAAI,aAChC,kBAAiB,KAAK;KACpB,WAAW,IAAI;KACf,cAAc,IAAI;KAClB,aAAa,IAAI,MAAM;KACxB,CAAC;IAKR,MAAM,UAAU,MAAM,QAAQ,WAC5B,iBAAiB,IAAI,OAAO,QAAQ;KAClC,MAAM,OAAO,MAAM,KAAK,gCACtB,iBACA,SACA,gBACA,IAAI,WACJ,IAAI,aACL;AACD,YAAO;MACL,cAAc,IAAI;MAClB,aAAa,IAAI;MACjB;MACD;MACD,CACH;AAED,SAAK,MAAM,UAAU,QACnB,KAAI,OAAO,WAAW,YACpB,OAAM;KACJ,MAAM;KACN,cAAc,OAAO,MAAM;KAC3B,aAAa,OAAO,MAAM;KAC1B,MAAM,OAAO,MAAM;KACpB;SACI;AACL,YAAO,MAAM,oCAAoC,OAAO,OAAO;AAC/D,WAAM;MACJ,MAAM;MACN,OACE,OAAO,kBAAkB,QACrB,OAAO,OAAO,UACd,YAAY;MACnB;;;WAIA,OAAO;AACd,UAAO,MACL,mEACA,SACA,gBACA,MACD;AACD,SAAM;IAAE,MAAM;IAAS,OAAO,mBAAmB,MAAM;IAAE;;;;;;;;CAS7D,OAAO,iBACL,iBACA,SACA,gBACA,WACA,SACkC;EAClC,MAAM,eAAe,SAAS,gBAAgB;EAC9C,MAAM,SAAS,SAAS;EACxB,IAAI,aAAa;AAEjB,MAAI;AACF,UAAO,MAAM;AACX,QAAI,QAAQ,QAAS;IAErB,MAAM,UAAU,MAAM,gBAAgB,MAAM,WAAW;KACrD,UAAU;KACV,iBAAiB;KACjB,YAAY;KACb,CAAC;AAEF,QAAI,QAAQ,UAAU,QAAQ,WAAW,YAAY;AACnD,kBAAa,QAAQ;AACrB,WAAM;MAAE,MAAM;MAAU,QAAQ,QAAQ;MAAQ;;AAKlD,QADE,QAAQ,WAAW,eAAe,QAAQ,WAAW,UACvC;KACd,MAAM,kBAAkB,kBAAkB,QAAQ;AAClD,WAAM;MAAE,MAAM;MAAkB,SAAS;MAAiB;AAC1D,YAAO,KAAK,iBACV,iBACA,SACA,gBACA,WACA,gBACD;AACD;;AAGF,UAAM,IAAI,SAAe,YAAY;KACnC,MAAM,QAAQ,WAAW,SAAS,aAAa;AAC/C,aAAQ,iBACN,eACM;AACJ,mBAAa,MAAM;AACnB,eAAS;QAEX,EAAE,MAAM,MAAM,CACf;MACD;;WAEG,OAAO;AACd,OAAI,QAAQ,QAAS;AACrB,UAAO,MACL,iFACA,SACA,gBACA,WACA,MACD;AACD,SAAM;IAAE,MAAM;IAAS,OAAO,mBAAmB,MAAM;IAAE;;;CAI7D,MAAM,YACJ,iBACA,SACA,SACA,gBAC+B;EAC/B,MAAM,EAAE,eAAe,gBAAgB,yBACrC,MAAM,KAAK,aACT,iBACA,SACA,SACA,eACD;AAGH,SAAO;GACL,GAFsB,kBADC,MAAM,KAAK,eAAe,cAAc,CACN;GAGzD,gBAAgB;GACjB;;CAGH,MAAM,gBACJ,iBACA,SACA,gBAC2C;EAC3C,MAAM,cAAsC,EAAE;EAC9C,IAAI;AAEJ,KAAG;GACD,MAAM,EAAE,UAAU,kBAAkB,MAAM,KAAK,yBAC7C,iBACA,SACA,gBACA;IACE,UAAU,uBAAuB;IACjC;IACD,CACF;AACD,eAAY,KAAK,GAAG,SAAS;AAC7B,eAAY,iBAAiB;WACtB,aAAa,YAAY,SAAS,KAAK,OAAO;AAEvD,SAAO;GACL;GACA;GACA,UAAU,YAAY,MAAM,GAAG,KAAK,OAAO,YAAY;GACxD"}
+{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/genie/client.ts"],"sourcesContent":["import { createLogger } from \"../../logging\";\nimport {\n  type GenieMessage,\n  Time,\n  TimeUnits,\n  type Waiter,\n  type WorkspaceClient,\n} from \"../../workspace-client\";\nimport { genieConnectorDefaults } from \"./defaults\";\nimport { pollWaiter } from \"./poll-waiter\";\nimport type {\n  GenieAttachmentResponse,\n  GenieConversationHistoryResponse,\n  GenieMessageResponse,\n  GenieStatementResponse,\n  GenieStreamEvent,\n} from \"./types\";\n\nconst logger = createLogger(\"connectors:genie\");\n\nconst GenieErrors = {\n  SPACE_ACCESS_DENIED: \"You don't have access to this Genie Space.\",\n  TABLE_PERMISSIONS:\n    \"You may not have access to the data tables. Please verify your table permissions.\",\n  REQUEST_FAILED: \"Genie request failed\",\n  QUERY_RESULT_FAILED: \"Failed to fetch query result\",\n} as const;\n\ntype CreateMessageWaiter = Waiter<GenieMessage, GenieMessage>;\n\ninterface GenieConnectorConfig {\n  timeout?: number;\n  maxMessages?: number;\n}\n\nfunction mapAttachments(message: GenieMessage): GenieAttachmentResponse[] {\n  return (\n    message.attachments?.map((att) => ({\n      attachmentId: att.attachment_id,\n      query: att.query\n        ? {\n            title: att.query.title,\n            description: att.query.description,\n            query: att.query.query,\n            statementId: att.query.statement_id,\n          }\n        : undefined,\n      text: att.text ? { content: att.text.content } : undefined,\n      suggestedQuestions: att.suggested_questions?.questions,\n    })) ?? []\n  );\n}\n\nfunction toMessageResponse(message: GenieMessage): GenieMessageResponse {\n  return {\n    messageId: message.message_id,\n    conversationId: message.conversation_id,\n    spaceId: message.space_id,\n    status: message.status ?? \"COMPLETED\",\n    content: message.content,\n    attachments: mapAttachments(message),\n    error: message.error?.error,\n  };\n}\n\nfunction classifyGenieError(error: unknown): string {\n  const message = error instanceof Error ? error.message : String(error);\n\n  if (message.includes(\"RESOURCE_DOES_NOT_EXIST\")) {\n    return GenieErrors.SPACE_ACCESS_DENIED;\n  }\n\n  if (\n    message.includes(\"failed to reach COMPLETED state\") &&\n    message.includes(\"FAILED\")\n  ) {\n    return GenieErrors.TABLE_PERMISSIONS;\n  }\n\n  return message || GenieErrors.REQUEST_FAILED;\n}\n\nexport class GenieConnector {\n  private readonly config: Required<GenieConnectorConfig>;\n\n  constructor(config: GenieConnectorConfig = {}) {\n    this.config = {\n      timeout: config.timeout ?? genieConnectorDefaults.timeout,\n      maxMessages: config.maxMessages ?? genieConnectorDefaults.maxMessages,\n    };\n  }\n\n  async startMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    content: string,\n    conversationId: string | undefined,\n  ): Promise<{\n    messageWaiter: CreateMessageWaiter;\n    conversationId: string;\n    messageId: string;\n  }> {\n    if (conversationId) {\n      const waiter = await workspaceClient.genie.createMessage({\n        space_id: spaceId,\n        conversation_id: conversationId,\n        content,\n      });\n      return {\n        messageWaiter: waiter,\n        conversationId,\n        messageId: waiter.message_id ?? \"\",\n      };\n    }\n    const start = await workspaceClient.genie.startConversation({\n      space_id: spaceId,\n      content,\n    });\n    return {\n      messageWaiter: start as unknown as CreateMessageWaiter,\n      conversationId: start.conversation_id,\n      messageId: start.message_id,\n    };\n  }\n\n  async waitForMessage(\n    messageWaiter: CreateMessageWaiter,\n    options?: { timeout?: number },\n  ): Promise<GenieMessage> {\n    const timeout = options?.timeout ?? this.config.timeout;\n    const waitOptions =\n      timeout > 0 ? { timeout: new Time(timeout, TimeUnits.milliseconds) } : {};\n    return messageWaiter.wait(waitOptions);\n  }\n\n  async listConversationMessages(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    options?: { pageSize?: number; pageToken?: string },\n  ): Promise<{\n    messages: GenieMessageResponse[];\n    nextPageToken: string | null;\n  }> {\n    const pageSize =\n      options?.pageSize ?? genieConnectorDefaults.initialPageSize;\n\n    const response = await workspaceClient.genie.listConversationMessages({\n      space_id: spaceId,\n      conversation_id: conversationId,\n      page_size: pageSize,\n      ...(options?.pageToken ? { page_token: options.pageToken } : {}),\n    });\n\n    const messages = (response.messages ?? []).reverse().map(toMessageResponse);\n\n    return {\n      messages,\n      nextPageToken: response.next_page_token ?? null,\n    };\n  }\n\n  async getMessageAttachmentQueryResult(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    messageId: string,\n    attachmentId: string,\n    _signal?: AbortSignal,\n  ): Promise<GenieStatementResponse> {\n    const response =\n      await workspaceClient.genie.getMessageAttachmentQueryResult({\n        space_id: spaceId,\n        conversation_id: conversationId,\n        message_id: messageId,\n        attachment_id: attachmentId,\n      });\n    return response.statement_response as GenieStatementResponse;\n  }\n\n  async *streamSendMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    content: string,\n    conversationId: string | undefined,\n    options?: { timeout?: number; signal?: AbortSignal },\n  ): AsyncGenerator<GenieStreamEvent> {\n    try {\n      const {\n        messageWaiter,\n        conversationId: resultConversationId,\n        messageId: resultMessageId,\n      } = await this.startMessage(\n        workspaceClient,\n        spaceId,\n        content,\n        conversationId,\n      );\n\n      yield {\n        type: \"message_start\",\n        conversationId: resultConversationId,\n        messageId: resultMessageId,\n        spaceId,\n      };\n\n      const timeout =\n        options?.timeout != null ? options.timeout : this.config.timeout;\n      const waitOptions =\n        timeout > 0\n          ? { timeout: new Time(timeout, TimeUnits.milliseconds) }\n          : {};\n\n      let completedMessage!: GenieMessage;\n      for await (const event of pollWaiter(messageWaiter, waitOptions)) {\n        if (event.type === \"progress\" && event.value.status) {\n          yield { type: \"status\", status: event.value.status };\n        } else if (event.type === \"completed\") {\n          completedMessage = event.value;\n        }\n      }\n\n      const messageResponse = toMessageResponse(completedMessage);\n      yield { type: \"message_result\", message: messageResponse };\n\n      yield* this.emitQueryResults(\n        workspaceClient,\n        spaceId,\n        resultConversationId,\n        messageResponse.messageId,\n        messageResponse,\n      );\n    } catch (error) {\n      logger.error(\n        \"Genie message error (spaceId=%s, conversationId=%s): %O\",\n        spaceId,\n        conversationId ?? \"new\",\n        error,\n      );\n      yield { type: \"error\", error: classifyGenieError(error) };\n    }\n  }\n\n  private async *emitQueryResults(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    messageId: string,\n    messageResponse: GenieMessageResponse,\n  ): AsyncGenerator<\n    Extract<GenieStreamEvent, { type: \"query_result\" } | { type: \"error\" }>\n  > {\n    const attachments = messageResponse.attachments ?? [];\n    for (const att of attachments) {\n      if (!att.query?.statementId || !att.attachmentId) continue;\n      try {\n        const data = await this.getMessageAttachmentQueryResult(\n          workspaceClient,\n          spaceId,\n          conversationId,\n          messageId,\n          att.attachmentId,\n        );\n        yield {\n          type: \"query_result\",\n          attachmentId: att.attachmentId,\n          statementId: att.query.statementId,\n          data,\n        };\n      } catch (error) {\n        logger.error(\n          \"Failed to fetch query result for attachment %s: %O\",\n          att.attachmentId,\n          error,\n        );\n        yield {\n          type: \"error\",\n          error: `${GenieErrors.QUERY_RESULT_FAILED} for attachment ${att.attachmentId}`,\n        };\n      }\n    }\n  }\n\n  async *streamConversation(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    options?: {\n      includeQueryResults?: boolean;\n      pageSize?: number;\n      pageToken?: string;\n      signal?: AbortSignal;\n    },\n  ): AsyncGenerator<GenieStreamEvent> {\n    const includeQueryResults = options?.includeQueryResults !== false;\n\n    try {\n      const { messages: messageResponses, nextPageToken } =\n        await this.listConversationMessages(\n          workspaceClient,\n          spaceId,\n          conversationId,\n          { pageSize: options?.pageSize, pageToken: options?.pageToken },\n        );\n\n      for (const messageResponse of messageResponses) {\n        yield { type: \"message_result\", message: messageResponse };\n      }\n\n      yield {\n        type: \"history_info\",\n        conversationId,\n        spaceId,\n        nextPageToken,\n        loadedCount: messageResponses.length,\n      };\n\n      if (includeQueryResults) {\n        const queryAttachments: Array<{\n          messageId: string;\n          attachmentId: string;\n          statementId: string;\n        }> = [];\n\n        for (const msg of messageResponses) {\n          for (const att of msg.attachments ?? []) {\n            if (att.query?.statementId && att.attachmentId) {\n              queryAttachments.push({\n                messageId: msg.messageId,\n                attachmentId: att.attachmentId,\n                statementId: att.query.statementId,\n              });\n            }\n          }\n        }\n\n        const results = await Promise.allSettled(\n          queryAttachments.map(async (att) => {\n            const data = await this.getMessageAttachmentQueryResult(\n              workspaceClient,\n              spaceId,\n              conversationId,\n              att.messageId,\n              att.attachmentId,\n            );\n            return {\n              attachmentId: att.attachmentId,\n              statementId: att.statementId,\n              data,\n            };\n          }),\n        );\n\n        for (const result of results) {\n          if (result.status === \"fulfilled\") {\n            yield {\n              type: \"query_result\",\n              attachmentId: result.value.attachmentId,\n              statementId: result.value.statementId,\n              data: result.value.data,\n            };\n          } else {\n            logger.error(\"Failed to fetch query result: %O\", result.reason);\n            yield {\n              type: \"error\",\n              error:\n                result.reason instanceof Error\n                  ? result.reason.message\n                  : GenieErrors.QUERY_RESULT_FAILED,\n            };\n          }\n        }\n      }\n    } catch (error) {\n      logger.error(\n        \"Genie getConversation error (spaceId=%s, conversationId=%s): %O\",\n        spaceId,\n        conversationId,\n        error,\n      );\n      yield { type: \"error\", error: classifyGenieError(error) };\n    }\n  }\n\n  /**\n   * Polls a single message via `getMessage` until it reaches a terminal\n   * state (`COMPLETED` or `FAILED`). Yields the same event types as\n   * `streamSendMessage` so callers can reuse the same SSE processing logic.\n   */\n  async *streamGetMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n    messageId: string,\n    options?: { timeout?: number; pollInterval?: number; signal?: AbortSignal },\n  ): AsyncGenerator<GenieStreamEvent> {\n    const pollInterval = options?.pollInterval ?? 3_000;\n    const signal = options?.signal;\n    let lastStatus = \"\";\n\n    try {\n      while (true) {\n        if (signal?.aborted) return;\n\n        const message = await workspaceClient.genie.getMessage({\n          space_id: spaceId,\n          conversation_id: conversationId,\n          message_id: messageId,\n        });\n\n        if (message.status && message.status !== lastStatus) {\n          lastStatus = message.status;\n          yield { type: \"status\", status: message.status };\n        }\n\n        const isTerminal =\n          message.status === \"COMPLETED\" || message.status === \"FAILED\";\n        if (isTerminal) {\n          const messageResponse = toMessageResponse(message);\n          yield { type: \"message_result\", message: messageResponse };\n          yield* this.emitQueryResults(\n            workspaceClient,\n            spaceId,\n            conversationId,\n            messageId,\n            messageResponse,\n          );\n          return;\n        }\n\n        await new Promise<void>((resolve) => {\n          const timer = setTimeout(resolve, pollInterval);\n          signal?.addEventListener(\n            \"abort\",\n            () => {\n              clearTimeout(timer);\n              resolve();\n            },\n            { once: true },\n          );\n        });\n      }\n    } catch (error) {\n      if (signal?.aborted) return;\n      logger.error(\n        \"Genie getMessage poll error (spaceId=%s, conversationId=%s, messageId=%s): %O\",\n        spaceId,\n        conversationId,\n        messageId,\n        error,\n      );\n      yield { type: \"error\", error: classifyGenieError(error) };\n    }\n  }\n\n  async sendMessage(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    content: string,\n    conversationId: string | undefined,\n  ): Promise<GenieMessageResponse> {\n    const { messageWaiter, conversationId: resultConversationId } =\n      await this.startMessage(\n        workspaceClient,\n        spaceId,\n        content,\n        conversationId,\n      );\n    const completedMessage = await this.waitForMessage(messageWaiter);\n    const messageResponse = toMessageResponse(completedMessage);\n    return {\n      ...messageResponse,\n      conversationId: resultConversationId,\n    };\n  }\n\n  async getConversation(\n    workspaceClient: WorkspaceClient,\n    spaceId: string,\n    conversationId: string,\n  ): Promise<GenieConversationHistoryResponse> {\n    const allMessages: GenieMessageResponse[] = [];\n    let pageToken: string | undefined;\n\n    do {\n      const { messages, nextPageToken } = await this.listConversationMessages(\n        workspaceClient,\n        spaceId,\n        conversationId,\n        {\n          pageSize: genieConnectorDefaults.pageSize,\n          pageToken,\n        },\n      );\n      allMessages.push(...messages);\n      pageToken = nextPageToken ?? undefined;\n    } while (pageToken && allMessages.length < this.config.maxMessages);\n\n    return {\n      conversationId,\n      spaceId,\n      messages: allMessages.slice(0, this.config.maxMessages),\n    };\n  }\n}\n"],"mappings":";;;;;;;AAkBA,MAAM,SAAS,aAAa,mBAAmB;AAE/C,MAAM,cAAc;CAClB,qBAAqB;CACrB,mBACE;CACF,gBAAgB;CAChB,qBAAqB;CACtB;AASD,SAAS,eAAe,SAAkD;AACxE,QACE,QAAQ,aAAa,KAAK,SAAS;EACjC,cAAc,IAAI;EAClB,OAAO,IAAI,QACP;GACE,OAAO,IAAI,MAAM;GACjB,aAAa,IAAI,MAAM;GACvB,OAAO,IAAI,MAAM;GACjB,aAAa,IAAI,MAAM;GACxB,GACD;EACJ,MAAM,IAAI,OAAO,EAAE,SAAS,IAAI,KAAK,SAAS,GAAG;EACjD,oBAAoB,IAAI,qBAAqB;EAC9C,EAAE,IAAI,EAAE;;AAIb,SAAS,kBAAkB,SAA6C;AACtE,QAAO;EACL,WAAW,QAAQ;EACnB,gBAAgB,QAAQ;EACxB,SAAS,QAAQ;EACjB,QAAQ,QAAQ,UAAU;EAC1B,SAAS,QAAQ;EACjB,aAAa,eAAe,QAAQ;EACpC,OAAO,QAAQ,OAAO;EACvB;;AAGH,SAAS,mBAAmB,OAAwB;CAClD,MAAM,UAAU,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;AAEtE,KAAI,QAAQ,SAAS,0BAA0B,CAC7C,QAAO,YAAY;AAGrB,KACE,QAAQ,SAAS,kCAAkC,IACnD,QAAQ,SAAS,SAAS,CAE1B,QAAO,YAAY;AAGrB,QAAO,WAAW,YAAY;;AAGhC,IAAa,iBAAb,MAA4B;CAC1B,AAAiB;CAEjB,YAAY,SAA+B,EAAE,EAAE;AAC7C,OAAK,SAAS;GACZ,SAAS,OAAO,WAAW,uBAAuB;GAClD,aAAa,OAAO,eAAe,uBAAuB;GAC3D;;CAGH,MAAM,aACJ,iBACA,SACA,SACA,gBAKC;AACD,MAAI,gBAAgB;GAClB,MAAM,SAAS,MAAM,gBAAgB,MAAM,cAAc;IACvD,UAAU;IACV,iBAAiB;IACjB;IACD,CAAC;AACF,UAAO;IACL,eAAe;IACf;IACA,WAAW,OAAO,cAAc;IACjC;;EAEH,MAAM,QAAQ,MAAM,gBAAgB,MAAM,kBAAkB;GAC1D,UAAU;GACV;GACD,CAAC;AACF,SAAO;GACL,eAAe;GACf,gBAAgB,MAAM;GACtB,WAAW,MAAM;GAClB;;CAGH,MAAM,eACJ,eACA,SACuB;EACvB,MAAM,UAAU,SAAS,WAAW,KAAK,OAAO;EAChD,MAAM,cACJ,UAAU,IAAI,EAAE,SAAS,IAAI,KAAK,SAAS,UAAU,aAAa,EAAE,GAAG,EAAE;AAC3E,SAAO,cAAc,KAAK,YAAY;;CAGxC,MAAM,yBACJ,iBACA,SACA,gBACA,SAIC;EACD,MAAM,WACJ,SAAS,YAAY,uBAAuB;EAE9C,MAAM,WAAW,MAAM,gBAAgB,MAAM,yBAAyB;GACpE,UAAU;GACV,iBAAiB;GACjB,WAAW;GACX,GAAI,SAAS,YAAY,EAAE,YAAY,QAAQ,WAAW,GAAG,EAAE;GAChE,CAAC;AAIF,SAAO;GACL,WAHgB,SAAS,YAAY,EAAE,EAAE,SAAS,CAAC,IAAI,kBAAkB;GAIzE,eAAe,SAAS,mBAAmB;GAC5C;;CAGH,MAAM,gCACJ,iBACA,SACA,gBACA,WACA,cACA,SACiC;AAQjC,UANE,MAAM,gBAAgB,MAAM,gCAAgC;GAC1D,UAAU;GACV,iBAAiB;GACjB,YAAY;GACZ,eAAe;GAChB,CAAC,EACY;;CAGlB,OAAO,kBACL,iBACA,SACA,SACA,gBACA,SACkC;AAClC,MAAI;GACF,MAAM,EACJ,eACA,gBAAgB,sBAChB,WAAW,oBACT,MAAM,KAAK,aACb,iBACA,SACA,SACA,eACD;AAED,SAAM;IACJ,MAAM;IACN,gBAAgB;IAChB,WAAW;IACX;IACD;GAED,MAAM,UACJ,SAAS,WAAW,OAAO,QAAQ,UAAU,KAAK,OAAO;GAC3D,MAAM,cACJ,UAAU,IACN,EAAE,SAAS,IAAI,KAAK,SAAS,UAAU,aAAa,EAAE,GACtD,EAAE;GAER,IAAI;AACJ,cAAW,MAAM,SAAS,WAAW,eAAe,YAAY,CAC9D,KAAI,MAAM,SAAS,cAAc,MAAM,MAAM,OAC3C,OAAM;IAAE,MAAM;IAAU,QAAQ,MAAM,MAAM;IAAQ;YAC3C,MAAM,SAAS,YACxB,oBAAmB,MAAM;GAI7B,MAAM,kBAAkB,kBAAkB,iBAAiB;AAC3D,SAAM;IAAE,MAAM;IAAkB,SAAS;IAAiB;AAE1D,UAAO,KAAK,iBACV,iBACA,SACA,sBACA,gBAAgB,WAChB,gBACD;WACM,OAAO;AACd,UAAO,MACL,2DACA,SACA,kBAAkB,OAClB,MACD;AACD,SAAM;IAAE,MAAM;IAAS,OAAO,mBAAmB,MAAM;IAAE;;;CAI7D,OAAe,iBACb,iBACA,SACA,gBACA,WACA,iBAGA;EACA,MAAM,cAAc,gBAAgB,eAAe,EAAE;AACrD,OAAK,MAAM,OAAO,aAAa;AAC7B,OAAI,CAAC,IAAI,OAAO,eAAe,CAAC,IAAI,aAAc;AAClD,OAAI;IACF,MAAM,OAAO,MAAM,KAAK,gCACtB,iBACA,SACA,gBACA,WACA,IAAI,aACL;AACD,UAAM;KACJ,MAAM;KACN,cAAc,IAAI;KAClB,aAAa,IAAI,MAAM;KACvB;KACD;YACM,OAAO;AACd,WAAO,MACL,sDACA,IAAI,cACJ,MACD;AACD,UAAM;KACJ,MAAM;KACN,OAAO,GAAG,YAAY,oBAAoB,kBAAkB,IAAI;KACjE;;;;CAKP,OAAO,mBACL,iBACA,SACA,gBACA,SAMkC;EAClC,MAAM,sBAAsB,SAAS,wBAAwB;AAE7D,MAAI;GACF,MAAM,EAAE,UAAU,kBAAkB,kBAClC,MAAM,KAAK,yBACT,iBACA,SACA,gBACA;IAAE,UAAU,SAAS;IAAU,WAAW,SAAS;IAAW,CAC/D;AAEH,QAAK,MAAM,mBAAmB,iBAC5B,OAAM;IAAE,MAAM;IAAkB,SAAS;IAAiB;AAG5D,SAAM;IACJ,MAAM;IACN;IACA;IACA;IACA,aAAa,iBAAiB;IAC/B;AAED,OAAI,qBAAqB;IACvB,MAAM,mBAID,EAAE;AAEP,SAAK,MAAM,OAAO,iBAChB,MAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CACrC,KAAI,IAAI,OAAO,eAAe,IAAI,aAChC,kBAAiB,KAAK;KACpB,WAAW,IAAI;KACf,cAAc,IAAI;KAClB,aAAa,IAAI,MAAM;KACxB,CAAC;IAKR,MAAM,UAAU,MAAM,QAAQ,WAC5B,iBAAiB,IAAI,OAAO,QAAQ;KAClC,MAAM,OAAO,MAAM,KAAK,gCACtB,iBACA,SACA,gBACA,IAAI,WACJ,IAAI,aACL;AACD,YAAO;MACL,cAAc,IAAI;MAClB,aAAa,IAAI;MACjB;MACD;MACD,CACH;AAED,SAAK,MAAM,UAAU,QACnB,KAAI,OAAO,WAAW,YACpB,OAAM;KACJ,MAAM;KACN,cAAc,OAAO,MAAM;KAC3B,aAAa,OAAO,MAAM;KAC1B,MAAM,OAAO,MAAM;KACpB;SACI;AACL,YAAO,MAAM,oCAAoC,OAAO,OAAO;AAC/D,WAAM;MACJ,MAAM;MACN,OACE,OAAO,kBAAkB,QACrB,OAAO,OAAO,UACd,YAAY;MACnB;;;WAIA,OAAO;AACd,UAAO,MACL,mEACA,SACA,gBACA,MACD;AACD,SAAM;IAAE,MAAM;IAAS,OAAO,mBAAmB,MAAM;IAAE;;;;;;;;CAS7D,OAAO,iBACL,iBACA,SACA,gBACA,WACA,SACkC;EAClC,MAAM,eAAe,SAAS,gBAAgB;EAC9C,MAAM,SAAS,SAAS;EACxB,IAAI,aAAa;AAEjB,MAAI;AACF,UAAO,MAAM;AACX,QAAI,QAAQ,QAAS;IAErB,MAAM,UAAU,MAAM,gBAAgB,MAAM,WAAW;KACrD,UAAU;KACV,iBAAiB;KACjB,YAAY;KACb,CAAC;AAEF,QAAI,QAAQ,UAAU,QAAQ,WAAW,YAAY;AACnD,kBAAa,QAAQ;AACrB,WAAM;MAAE,MAAM;MAAU,QAAQ,QAAQ;MAAQ;;AAKlD,QADE,QAAQ,WAAW,eAAe,QAAQ,WAAW,UACvC;KACd,MAAM,kBAAkB,kBAAkB,QAAQ;AAClD,WAAM;MAAE,MAAM;MAAkB,SAAS;MAAiB;AAC1D,YAAO,KAAK,iBACV,iBACA,SACA,gBACA,WACA,gBACD;AACD;;AAGF,UAAM,IAAI,SAAe,YAAY;KACnC,MAAM,QAAQ,WAAW,SAAS,aAAa;AAC/C,aAAQ,iBACN,eACM;AACJ,mBAAa,MAAM;AACnB,eAAS;QAEX,EAAE,MAAM,MAAM,CACf;MACD;;WAEG,OAAO;AACd,OAAI,QAAQ,QAAS;AACrB,UAAO,MACL,iFACA,SACA,gBACA,WACA,MACD;AACD,SAAM;IAAE,MAAM;IAAS,OAAO,mBAAmB,MAAM;IAAE;;;CAI7D,MAAM,YACJ,iBACA,SACA,SACA,gBAC+B;EAC/B,MAAM,EAAE,eAAe,gBAAgB,yBACrC,MAAM,KAAK,aACT,iBACA,SACA,SACA,eACD;AAGH,SAAO;GACL,GAFsB,kBADC,MAAM,KAAK,eAAe,cAAc,CACN;GAGzD,gBAAgB;GACjB;;CAGH,MAAM,gBACJ,iBACA,SACA,gBAC2C;EAC3C,MAAM,cAAsC,EAAE;EAC9C,IAAI;AAEJ,KAAG;GACD,MAAM,EAAE,UAAU,kBAAkB,MAAM,KAAK,yBAC7C,iBACA,SACA,gBACA;IACE,UAAU,uBAAuB;IACjC;IACD,CACF;AACD,eAAY,KAAK,GAAG,SAAS;AAC7B,eAAY,iBAAiB;WACtB,aAAa,YAAY,SAAS,KAAK,OAAO;AAEvD,SAAO;GACL;GACA;GACA,UAAU,YAAY,MAAM,GAAG,KAAK,OAAO,YAAY;GACxD"}

package/dist/connectors/jobs/client.js

@@ -1,9 +1,8 @@
+import { Context } from "../../shared/src/workspace-client/legacy.js";
 import { createLogger } from "../../logging/logger.js";
 import { AppKitError } from "../../errors/base.js";
 import { ExecutionError } from "../../errors/execution.js";
 import "../../errors/index.js";
-import { Context } from "../../workspace-client/legacy.js";
-import "../../workspace-client/index.js";
 import { TelemetryManager } from "../../telemetry/telemetry-manager.js";
 import { SpanKind, SpanStatusCode } from "../../telemetry/index.js";
 

package/dist/connectors/jobs/client.js.map

@@ -1 +1 @@
-{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/jobs/client.ts"],"sourcesContent":["import { AppKitError, ExecutionError } from \"../../errors\";\nimport { createLogger } from \"../../logging/logger\";\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 jobs,\n  type WorkspaceClient,\n} from \"../../workspace-client\";\nimport type { JobsConnectorConfig } from \"./types\";\n\nconst logger = createLogger(\"connectors:jobs\");\n\nexport class JobsConnector {\n  private readonly name = \"jobs\";\n  private readonly config: JobsConnectorConfig;\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    apiCallCount: Counter;\n    apiCallDuration: Histogram;\n  };\n\n  constructor(config: JobsConnectorConfig) {\n    this.config = config;\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      apiCallCount: this.telemetry\n        .getMeter()\n        .createCounter(\"jobs.api_call.count\", {\n          description: \"Total number of Jobs API calls\",\n          unit: \"1\",\n        }),\n      apiCallDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"jobs.api_call.duration\", {\n          description: \"Duration of Jobs API calls\",\n          unit: \"ms\",\n        }),\n    };\n  }\n\n  async submitRun(\n    workspaceClient: WorkspaceClient,\n    request: jobs.SubmitRun,\n    signal?: AbortSignal,\n  ): Promise<jobs.SubmitRunResponse> {\n    return this._callApi(\"submit\", async () => {\n      return workspaceClient.jobs.submit(request, this._createContext(signal));\n    });\n  }\n\n  async runNow(\n    workspaceClient: WorkspaceClient,\n    request: jobs.RunNow,\n    signal?: AbortSignal,\n  ): Promise<jobs.RunNowResponse> {\n    return this._callApi(\"runNow\", async () => {\n      return workspaceClient.jobs.runNow(request, this._createContext(signal));\n    });\n  }\n\n  async getRun(\n    workspaceClient: WorkspaceClient,\n    request: jobs.GetRunRequest,\n    signal?: AbortSignal,\n  ): Promise<jobs.Run> {\n    return this._callApi(\"getRun\", async () => {\n      return workspaceClient.jobs.getRun(request, this._createContext(signal));\n    });\n  }\n\n  async getRunOutput(\n    workspaceClient: WorkspaceClient,\n    request: jobs.GetRunOutputRequest,\n    signal?: AbortSignal,\n  ): Promise<jobs.RunOutput> {\n    return this._callApi(\"getRunOutput\", async () => {\n      return workspaceClient.jobs.getRunOutput(\n        request,\n        this._createContext(signal),\n      );\n    });\n  }\n\n  async cancelRun(\n    workspaceClient: WorkspaceClient,\n    request: jobs.CancelRun,\n    signal?: AbortSignal,\n  ): Promise<void> {\n    await this._callApi(\"cancelRun\", async () => {\n      return workspaceClient.jobs.cancelRun(\n        request,\n        this._createContext(signal),\n      );\n    });\n  }\n\n  async listRuns(\n    workspaceClient: WorkspaceClient,\n    request: jobs.ListRunsRequest,\n    signal?: AbortSignal,\n  ): Promise<jobs.BaseRun[]> {\n    return this._callApi(\"listRuns\", async () => {\n      const runs: jobs.BaseRun[] = [];\n      const limit = Math.max(1, Math.min(request.limit ?? 100, 100));\n      for await (const run of workspaceClient.jobs.listRuns(\n        { ...request, limit },\n        this._createContext(signal),\n      )) {\n        runs.push(run);\n        if (runs.length >= limit) break;\n      }\n      return runs;\n    });\n  }\n\n  async getJob(\n    workspaceClient: WorkspaceClient,\n    request: jobs.GetJobRequest,\n    signal?: AbortSignal,\n  ): Promise<jobs.Job> {\n    return this._callApi(\"getJob\", async () => {\n      return workspaceClient.jobs.get(request, this._createContext(signal));\n    });\n  }\n\n  private async _callApi<T>(\n    operation: string,\n    fn: () => Promise<T>,\n  ): Promise<T> {\n    const startTime = Date.now();\n    let success = false;\n\n    return this.telemetry.startActiveSpan(\n      `jobs.${operation}`,\n      {\n        kind: SpanKind.CLIENT,\n        attributes: {\n          \"jobs.operation\": operation,\n        },\n      },\n      async (span: Span) => {\n        try {\n          const result = await fn();\n          success = true;\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\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          if (error instanceof AppKitError) {\n            throw error;\n          }\n          // Preserve SDK ApiError (and any error with a numeric statusCode)\n          // so Plugin.execute() can map it to the correct HTTP status.\n          if (\n            error instanceof Error &&\n            \"statusCode\" in error &&\n            typeof (error as Record<string, unknown>).statusCode === \"number\"\n          ) {\n            throw error;\n          }\n          throw new ExecutionError(\n            `Jobs API call failed: ${error instanceof Error ? error.message : String(error)}`,\n          );\n        } finally {\n          const duration = Date.now() - startTime;\n          span.end();\n          this.telemetryMetrics.apiCallCount.add(1, {\n            operation,\n            success: success.toString(),\n          });\n          this.telemetryMetrics.apiCallDuration.record(duration, {\n            operation,\n            success: success.toString(),\n          });\n\n          logger.event()?.setContext(\"jobs\", {\n            operation,\n            duration_ms: duration,\n            success,\n          });\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  private _createContext(signal?: AbortSignal) {\n    return new Context({\n      cancellationToken: {\n        // Getter — evaluated on every read so SDK code paths that poll\n        // (rather than subscribe) observe cancellation live.\n        get isCancellationRequested() {\n          return signal?.aborted ?? false;\n        },\n        onCancellationRequested: (cb: () => void) => {\n          signal?.addEventListener(\"abort\", cb, { once: true });\n        },\n      },\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;AAkBA,MAAM,SAAS,aAAa,kBAAkB;AAE9C,IAAa,gBAAb,MAA2B;CACzB,AAAiB,OAAO;CACxB,AAAiB;CACjB,AAAiB;CACjB,AAAiB;CAKjB,YAAY,QAA6B;AACvC,OAAK,SAAS;AACd,OAAK,YAAY,iBAAiB,YAChC,KAAK,MACL,KAAK,OAAO,UACb;AACD,OAAK,mBAAmB;GACtB,cAAc,KAAK,UAChB,UAAU,CACV,cAAc,uBAAuB;IACpC,aAAa;IACb,MAAM;IACP,CAAC;GACJ,iBAAiB,KAAK,UACnB,UAAU,CACV,gBAAgB,0BAA0B;IACzC,aAAa;IACb,MAAM;IACP,CAAC;GACL;;CAGH,MAAM,UACJ,iBACA,SACA,QACiC;AACjC,SAAO,KAAK,SAAS,UAAU,YAAY;AACzC,UAAO,gBAAgB,KAAK,OAAO,SAAS,KAAK,eAAe,OAAO,CAAC;IACxE;;CAGJ,MAAM,OACJ,iBACA,SACA,QAC8B;AAC9B,SAAO,KAAK,SAAS,UAAU,YAAY;AACzC,UAAO,gBAAgB,KAAK,OAAO,SAAS,KAAK,eAAe,OAAO,CAAC;IACxE;;CAGJ,MAAM,OACJ,iBACA,SACA,QACmB;AACnB,SAAO,KAAK,SAAS,UAAU,YAAY;AACzC,UAAO,gBAAgB,KAAK,OAAO,SAAS,KAAK,eAAe,OAAO,CAAC;IACxE;;CAGJ,MAAM,aACJ,iBACA,SACA,QACyB;AACzB,SAAO,KAAK,SAAS,gBAAgB,YAAY;AAC/C,UAAO,gBAAgB,KAAK,aAC1B,SACA,KAAK,eAAe,OAAO,CAC5B;IACD;;CAGJ,MAAM,UACJ,iBACA,SACA,QACe;AACf,QAAM,KAAK,SAAS,aAAa,YAAY;AAC3C,UAAO,gBAAgB,KAAK,UAC1B,SACA,KAAK,eAAe,OAAO,CAC5B;IACD;;CAGJ,MAAM,SACJ,iBACA,SACA,QACyB;AACzB,SAAO,KAAK,SAAS,YAAY,YAAY;GAC3C,MAAM,OAAuB,EAAE;GAC/B,MAAM,QAAQ,KAAK,IAAI,GAAG,KAAK,IAAI,QAAQ,SAAS,KAAK,IAAI,CAAC;AAC9D,cAAW,MAAM,OAAO,gBAAgB,KAAK,SAC3C;IAAE,GAAG;IAAS;IAAO,EACrB,KAAK,eAAe,OAAO,CAC5B,EAAE;AACD,SAAK,KAAK,IAAI;AACd,QAAI,KAAK,UAAU,MAAO;;AAE5B,UAAO;IACP;;CAGJ,MAAM,OACJ,iBACA,SACA,QACmB;AACnB,SAAO,KAAK,SAAS,UAAU,YAAY;AACzC,UAAO,gBAAgB,KAAK,IAAI,SAAS,KAAK,eAAe,OAAO,CAAC;IACrE;;CAGJ,MAAc,SACZ,WACA,IACY;EACZ,MAAM,YAAY,KAAK,KAAK;EAC5B,IAAI,UAAU;AAEd,SAAO,KAAK,UAAU,gBACpB,QAAQ,aACR;GACE,MAAM,SAAS;GACf,YAAY,EACV,kBAAkB,WACnB;GACF,EACD,OAAO,SAAe;AACpB,OAAI;IACF,MAAM,SAAS,MAAM,IAAI;AACzB,cAAU;AACV,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU;KACb,MAAM,eAAe;KACrB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;KAChE,CAAC;AACF,QAAI,iBAAiB,YACnB,OAAM;AAIR,QACE,iBAAiB,SACjB,gBAAgB,SAChB,OAAQ,MAAkC,eAAe,SAEzD,OAAM;AAER,UAAM,IAAI,eACR,yBAAyB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,GAChF;aACO;IACR,MAAM,WAAW,KAAK,KAAK,GAAG;AAC9B,SAAK,KAAK;AACV,SAAK,iBAAiB,aAAa,IAAI,GAAG;KACxC;KACA,SAAS,QAAQ,UAAU;KAC5B,CAAC;AACF,SAAK,iBAAiB,gBAAgB,OAAO,UAAU;KACrD;KACA,SAAS,QAAQ,UAAU;KAC5B,CAAC;AAEF,WAAO,OAAO,EAAE,WAAW,QAAQ;KACjC;KACA,aAAa;KACb;KACD,CAAC;;KAGN;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;CAGH,AAAQ,eAAe,QAAsB;AAC3C,SAAO,IAAI,QAAQ,EACjB,mBAAmB;GAGjB,IAAI,0BAA0B;AAC5B,WAAO,QAAQ,WAAW;;GAE5B,0BAA0B,OAAmB;AAC3C,YAAQ,iBAAiB,SAAS,IAAI,EAAE,MAAM,MAAM,CAAC;;GAExD,EACF,CAAC"}
+{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/jobs/client.ts"],"sourcesContent":["import { AppKitError, ExecutionError } from \"../../errors\";\nimport { createLogger } from \"../../logging/logger\";\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 jobs,\n  type WorkspaceClient,\n} from \"../../workspace-client\";\nimport type { JobsConnectorConfig } from \"./types\";\n\nconst logger = createLogger(\"connectors:jobs\");\n\nexport class JobsConnector {\n  private readonly name = \"jobs\";\n  private readonly config: JobsConnectorConfig;\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    apiCallCount: Counter;\n    apiCallDuration: Histogram;\n  };\n\n  constructor(config: JobsConnectorConfig) {\n    this.config = config;\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      apiCallCount: this.telemetry\n        .getMeter()\n        .createCounter(\"jobs.api_call.count\", {\n          description: \"Total number of Jobs API calls\",\n          unit: \"1\",\n        }),\n      apiCallDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"jobs.api_call.duration\", {\n          description: \"Duration of Jobs API calls\",\n          unit: \"ms\",\n        }),\n    };\n  }\n\n  async submitRun(\n    workspaceClient: WorkspaceClient,\n    request: jobs.SubmitRun,\n    signal?: AbortSignal,\n  ): Promise<jobs.SubmitRunResponse> {\n    return this._callApi(\"submit\", async () => {\n      return workspaceClient.jobs.submit(request, this._createContext(signal));\n    });\n  }\n\n  async runNow(\n    workspaceClient: WorkspaceClient,\n    request: jobs.RunNow,\n    signal?: AbortSignal,\n  ): Promise<jobs.RunNowResponse> {\n    return this._callApi(\"runNow\", async () => {\n      return workspaceClient.jobs.runNow(request, this._createContext(signal));\n    });\n  }\n\n  async getRun(\n    workspaceClient: WorkspaceClient,\n    request: jobs.GetRunRequest,\n    signal?: AbortSignal,\n  ): Promise<jobs.Run> {\n    return this._callApi(\"getRun\", async () => {\n      return workspaceClient.jobs.getRun(request, this._createContext(signal));\n    });\n  }\n\n  async getRunOutput(\n    workspaceClient: WorkspaceClient,\n    request: jobs.GetRunOutputRequest,\n    signal?: AbortSignal,\n  ): Promise<jobs.RunOutput> {\n    return this._callApi(\"getRunOutput\", async () => {\n      return workspaceClient.jobs.getRunOutput(\n        request,\n        this._createContext(signal),\n      );\n    });\n  }\n\n  async cancelRun(\n    workspaceClient: WorkspaceClient,\n    request: jobs.CancelRun,\n    signal?: AbortSignal,\n  ): Promise<void> {\n    await this._callApi(\"cancelRun\", async () => {\n      return workspaceClient.jobs.cancelRun(\n        request,\n        this._createContext(signal),\n      );\n    });\n  }\n\n  async listRuns(\n    workspaceClient: WorkspaceClient,\n    request: jobs.ListRunsRequest,\n    signal?: AbortSignal,\n  ): Promise<jobs.BaseRun[]> {\n    return this._callApi(\"listRuns\", async () => {\n      const runs: jobs.BaseRun[] = [];\n      const limit = Math.max(1, Math.min(request.limit ?? 100, 100));\n      for await (const run of workspaceClient.jobs.listRuns(\n        { ...request, limit },\n        this._createContext(signal),\n      )) {\n        runs.push(run);\n        if (runs.length >= limit) break;\n      }\n      return runs;\n    });\n  }\n\n  async getJob(\n    workspaceClient: WorkspaceClient,\n    request: jobs.GetJobRequest,\n    signal?: AbortSignal,\n  ): Promise<jobs.Job> {\n    return this._callApi(\"getJob\", async () => {\n      return workspaceClient.jobs.get(request, this._createContext(signal));\n    });\n  }\n\n  private async _callApi<T>(\n    operation: string,\n    fn: () => Promise<T>,\n  ): Promise<T> {\n    const startTime = Date.now();\n    let success = false;\n\n    return this.telemetry.startActiveSpan(\n      `jobs.${operation}`,\n      {\n        kind: SpanKind.CLIENT,\n        attributes: {\n          \"jobs.operation\": operation,\n        },\n      },\n      async (span: Span) => {\n        try {\n          const result = await fn();\n          success = true;\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\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          if (error instanceof AppKitError) {\n            throw error;\n          }\n          // Preserve SDK ApiError (and any error with a numeric statusCode)\n          // so Plugin.execute() can map it to the correct HTTP status.\n          if (\n            error instanceof Error &&\n            \"statusCode\" in error &&\n            typeof (error as Record<string, unknown>).statusCode === \"number\"\n          ) {\n            throw error;\n          }\n          throw new ExecutionError(\n            `Jobs API call failed: ${error instanceof Error ? error.message : String(error)}`,\n          );\n        } finally {\n          const duration = Date.now() - startTime;\n          span.end();\n          this.telemetryMetrics.apiCallCount.add(1, {\n            operation,\n            success: success.toString(),\n          });\n          this.telemetryMetrics.apiCallDuration.record(duration, {\n            operation,\n            success: success.toString(),\n          });\n\n          logger.event()?.setContext(\"jobs\", {\n            operation,\n            duration_ms: duration,\n            success,\n          });\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  private _createContext(signal?: AbortSignal) {\n    return new Context({\n      cancellationToken: {\n        // Getter — evaluated on every read so SDK code paths that poll\n        // (rather than subscribe) observe cancellation live.\n        get isCancellationRequested() {\n          return signal?.aborted ?? false;\n        },\n        onCancellationRequested: (cb: () => void) => {\n          signal?.addEventListener(\"abort\", cb, { once: true });\n        },\n      },\n    });\n  }\n}\n"],"mappings":";;;;;;;;;AAkBA,MAAM,SAAS,aAAa,kBAAkB;AAE9C,IAAa,gBAAb,MAA2B;CACzB,AAAiB,OAAO;CACxB,AAAiB;CACjB,AAAiB;CACjB,AAAiB;CAKjB,YAAY,QAA6B;AACvC,OAAK,SAAS;AACd,OAAK,YAAY,iBAAiB,YAChC,KAAK,MACL,KAAK,OAAO,UACb;AACD,OAAK,mBAAmB;GACtB,cAAc,KAAK,UAChB,UAAU,CACV,cAAc,uBAAuB;IACpC,aAAa;IACb,MAAM;IACP,CAAC;GACJ,iBAAiB,KAAK,UACnB,UAAU,CACV,gBAAgB,0BAA0B;IACzC,aAAa;IACb,MAAM;IACP,CAAC;GACL;;CAGH,MAAM,UACJ,iBACA,SACA,QACiC;AACjC,SAAO,KAAK,SAAS,UAAU,YAAY;AACzC,UAAO,gBAAgB,KAAK,OAAO,SAAS,KAAK,eAAe,OAAO,CAAC;IACxE;;CAGJ,MAAM,OACJ,iBACA,SACA,QAC8B;AAC9B,SAAO,KAAK,SAAS,UAAU,YAAY;AACzC,UAAO,gBAAgB,KAAK,OAAO,SAAS,KAAK,eAAe,OAAO,CAAC;IACxE;;CAGJ,MAAM,OACJ,iBACA,SACA,QACmB;AACnB,SAAO,KAAK,SAAS,UAAU,YAAY;AACzC,UAAO,gBAAgB,KAAK,OAAO,SAAS,KAAK,eAAe,OAAO,CAAC;IACxE;;CAGJ,MAAM,aACJ,iBACA,SACA,QACyB;AACzB,SAAO,KAAK,SAAS,gBAAgB,YAAY;AAC/C,UAAO,gBAAgB,KAAK,aAC1B,SACA,KAAK,eAAe,OAAO,CAC5B;IACD;;CAGJ,MAAM,UACJ,iBACA,SACA,QACe;AACf,QAAM,KAAK,SAAS,aAAa,YAAY;AAC3C,UAAO,gBAAgB,KAAK,UAC1B,SACA,KAAK,eAAe,OAAO,CAC5B;IACD;;CAGJ,MAAM,SACJ,iBACA,SACA,QACyB;AACzB,SAAO,KAAK,SAAS,YAAY,YAAY;GAC3C,MAAM,OAAuB,EAAE;GAC/B,MAAM,QAAQ,KAAK,IAAI,GAAG,KAAK,IAAI,QAAQ,SAAS,KAAK,IAAI,CAAC;AAC9D,cAAW,MAAM,OAAO,gBAAgB,KAAK,SAC3C;IAAE,GAAG;IAAS;IAAO,EACrB,KAAK,eAAe,OAAO,CAC5B,EAAE;AACD,SAAK,KAAK,IAAI;AACd,QAAI,KAAK,UAAU,MAAO;;AAE5B,UAAO;IACP;;CAGJ,MAAM,OACJ,iBACA,SACA,QACmB;AACnB,SAAO,KAAK,SAAS,UAAU,YAAY;AACzC,UAAO,gBAAgB,KAAK,IAAI,SAAS,KAAK,eAAe,OAAO,CAAC;IACrE;;CAGJ,MAAc,SACZ,WACA,IACY;EACZ,MAAM,YAAY,KAAK,KAAK;EAC5B,IAAI,UAAU;AAEd,SAAO,KAAK,UAAU,gBACpB,QAAQ,aACR;GACE,MAAM,SAAS;GACf,YAAY,EACV,kBAAkB,WACnB;GACF,EACD,OAAO,SAAe;AACpB,OAAI;IACF,MAAM,SAAS,MAAM,IAAI;AACzB,cAAU;AACV,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU;KACb,MAAM,eAAe;KACrB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;KAChE,CAAC;AACF,QAAI,iBAAiB,YACnB,OAAM;AAIR,QACE,iBAAiB,SACjB,gBAAgB,SAChB,OAAQ,MAAkC,eAAe,SAEzD,OAAM;AAER,UAAM,IAAI,eACR,yBAAyB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,GAChF;aACO;IACR,MAAM,WAAW,KAAK,KAAK,GAAG;AAC9B,SAAK,KAAK;AACV,SAAK,iBAAiB,aAAa,IAAI,GAAG;KACxC;KACA,SAAS,QAAQ,UAAU;KAC5B,CAAC;AACF,SAAK,iBAAiB,gBAAgB,OAAO,UAAU;KACrD;KACA,SAAS,QAAQ,UAAU;KAC5B,CAAC;AAEF,WAAO,OAAO,EAAE,WAAW,QAAQ;KACjC;KACA,aAAa;KACb;KACD,CAAC;;KAGN;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;CAGH,AAAQ,eAAe,QAAsB;AAC3C,SAAO,IAAI,QAAQ,EACjB,mBAAmB;GAGjB,IAAI,0BAA0B;AAC5B,WAAO,QAAQ,WAAW;;GAE5B,0BAA0B,OAAmB;AAC3C,YAAQ,iBAAiB,SAAS,IAAI,EAAE,MAAM,MAAM,CAAC;;GAExD,EACF,CAAC"}

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

@@ -1,3 +1,4 @@
+import { Context } from "../../shared/src/workspace-client/legacy.js";
 import { createLogger } from "../../logging/logger.js";
 import { AppKitError } from "../../errors/base.js";
 import { ConfigurationError } from "../../errors/configuration.js";
@@ -5,8 +6,6 @@ import { ConnectionError } from "../../errors/connection.js";
 import { ExecutionError } from "../../errors/execution.js";
 import { ValidationError } from "../../errors/validation.js";
 import "../../errors/index.js";
-import { Context } from "../../workspace-client/legacy.js";
-import "../../workspace-client/index.js";
 import { TelemetryManager } from "../../telemetry/telemetry-manager.js";
 import { SpanKind, SpanStatusCode } from "../../telemetry/index.js";
 import { ArrowStreamProcessor } from "../../stream/arrow-stream-processor.js";

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\";\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":";;;;;;;;;;;;;;;;;;AAgCA,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\";\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":";;;;;;;;;;;;;;;;;AAgCA,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"}

package/dist/context/execution-context.d.ts

@@ -1,5 +1,4 @@
 import { ExecutionContext } from "./user-context.js";
-import "../index.js";
 
 //#region src/context/execution-context.d.ts
 /**

package/dist/context/execution-context.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"execution-context.d.ts","names":[],"sources":["../../src/context/execution-context.ts"],"mappings":";;;;;;;;;;;;iBAmCgB,mBAAA,CAAA,GAAuB,gBAAA"}
+{"version":3,"file":"execution-context.d.ts","names":[],"sources":["../../src/context/execution-context.ts"],"mappings":";;;;;;;;;;;iBAmCgB,mBAAA,CAAA,GAAuB,gBAAA"}

package/dist/context/service-context.d.ts

@@ -1,5 +1,4 @@
-import { WorkspaceClient } from "../workspace-client/types.js";
-import "../workspace-client/index.js";
+import { WorkspaceClient } from "../workspace-client/index.js";
 import "./user-context.js";
 
 //#region src/context/service-context.d.ts

package/dist/context/service-context.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"service-context.d.ts","names":[],"sources":["../../src/context/service-context.ts"],"mappings":";;;;;;;;AAoBA;UAAiB,mBAAA;;EAEf,MAAA,EAAQ,eAAA;EAIM;EAFd,aAAA;EAIoB;EAFpB,WAAA,GAAc,OAAA;EAJd;EAMA,WAAA,EAAa,OAAA;AAAA"}
+{"version":3,"file":"service-context.d.ts","names":[],"sources":["../../src/context/service-context.ts"],"mappings":";;;;;;AAoBA;;UAAiB,mBAAA;EAEP;EAAR,MAAA,EAAQ,eAAA;EAMK;EAJb,aAAA;EAIoB;EAFpB,WAAA,GAAc,OAAA;EAJN;EAMR,WAAA,EAAa,OAAA;AAAA"}