@databricks/sdk-oauth 0.0.0-dev → 0.1.0-dev.2

package/dist/v1/utils.js

@@ -0,0 +1,113 @@
+// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
+import { execute } from '@databricks/sdk-core/ops';
+import { ApiError } from '@databricks/sdk-core/apierror';
+import JSONBig from 'json-bigint';
+// JSON codec that preserves int64 precision. On the way in, large integer
+// literals come back as bigint instead of being rounded to JS Number. On the
+// way out, bigint values are emitted as raw JSON number digits.
+const jsonBigint = JSONBig({ useNativeBigInt: true });
+/**
+ * Translates public CallOptions to the internal Options shape accepted by
+ * execute(). Even though the shapes match today, this isolates the public
+ * API from the executor's internal type so they can diverge.
+ */
+export async function executeCall(call, options) {
+    const opts = {
+        ...(options?.retrier !== undefined && { retrier: options.retrier }),
+        ...(options?.rateLimiter !== undefined && {
+            rateLimiter: options.rateLimiter,
+        }),
+        ...(options?.timeout !== undefined && { timeout: options.timeout }),
+    };
+    return execute(options?.signal, call, opts);
+}
+async function readAll(body) {
+    if (body === null) {
+        return new Uint8Array(0);
+    }
+    const reader = body.getReader();
+    const chunks = [];
+    for (;;) {
+        const { done, value } = await reader.read();
+        if (done) {
+            break;
+        }
+        chunks.push(value);
+    }
+    const totalLength = chunks.reduce((acc, chunk) => acc + chunk.length, 0);
+    const result = new Uint8Array(totalLength);
+    let offset = 0;
+    for (const chunk of chunks) {
+        result.set(chunk, offset);
+        offset += chunk.length;
+    }
+    return result;
+}
+export async function executeHttpCall(opts) {
+    opts.logger.debug('HTTP request', {
+        method: opts.request.method,
+        url: opts.request.url,
+    });
+    let resp;
+    try {
+        resp = await opts.httpClient.send(opts.request);
+    }
+    catch (e) {
+        opts.logger.debug('HTTP request failed');
+        throw e;
+    }
+    const body = await readAll(resp.body);
+    opts.logger.debug('HTTP response', {
+        statusCode: resp.statusCode,
+        body: new TextDecoder().decode(body),
+    });
+    const apiErr = ApiError.fromHttpError(resp.statusCode, resp.headers, body);
+    if (apiErr !== undefined) {
+        throw apiErr;
+    }
+    return body;
+}
+export function buildHttpRequest(method, url, headers, signal, body) {
+    const req = { url, method, headers };
+    if (body !== undefined) {
+        req.body = body;
+    }
+    if (signal !== undefined) {
+        req.signal = signal;
+    }
+    return req;
+}
+export function parseResponse(body, schema) {
+    const text = new TextDecoder().decode(body);
+    const parsed = jsonBigint.parse(text);
+    return schema.parse(parsed);
+}
+export function marshalRequest(data, schema) {
+    return jsonBigint.stringify(schema.parse(data));
+}
+export function flattenQueryParams(prefix, value, params) {
+    if (value === null || value === undefined) {
+        return;
+    }
+    if (Array.isArray(value)) {
+        // arrays of objects are not yet supported
+        for (const item of value) {
+            params.append(prefix, String(item));
+        }
+    }
+    else if (typeof value === 'object') {
+        for (const [key, val] of Object.entries(value)) {
+            flattenQueryParams(`${prefix}.${key}`, val, params);
+        }
+    }
+    else if (typeof value === 'string' ||
+        typeof value === 'number' ||
+        typeof value === 'boolean' ||
+        typeof value === 'bigint') {
+        params.append(prefix, String(value));
+    }
+    else {
+        throw new Error(`Unsupported query parameter type: ${typeof value}`);
+    }
+}
+//# sourceMappingURL=utils.js.map

package/dist/v1/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,OAAO,EAAC,MAAM,0BAA0B,CAAC;AACjD,OAAO,EAAC,QAAQ,EAAC,MAAM,+BAA+B,CAAC;AAQvD,OAAO,OAAO,MAAM,aAAa,CAAC;AAGlC,0EAA0E;AAC1E,6EAA6E;AAC7E,gEAAgE;AAChE,MAAM,UAAU,GAAG,OAAO,CAAC,EAAC,eAAe,EAAE,IAAI,EAAC,CAAC,CAAC;AAQpD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,IAA6C,EAC7C,OAAqB;IAErB,MAAM,IAAI,GAAY;QACpB,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;QACjE,GAAG,CAAC,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI;YACxC,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC;QACF,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;KAClE,CAAC;IACF,OAAO,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAC9C,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,IAAuC;IAEvC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IAC3B,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAChC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,SAAS,CAAC;QACR,MAAM,EAAC,IAAI,EAAE,KAAK,EAAC,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,IAAI,EAAE,CAAC;YACT,MAAM;QACR,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IACD,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;IAC3C,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,IAAqB;IAErB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE;QAChC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;QAC3B,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG;KACtB,CAAC,CAAC;IAEH,IAAI,IAAkB,CAAC;IACvB,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,MAAM,CAAC,CAAC;IACV,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE;QACjC,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,IAAI,EAAE,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;KACrC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC3E,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,MAAM,MAAM,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,MAAc,EACd,GAAW,EACX,OAAgB,EAChB,MAAoB,EACpB,IAA0C;IAE1C,MAAM,GAAG,GAAgB,EAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAC,CAAC;IAChD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;IAClB,CAAC;IACD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,aAAa,CAAI,IAAgB,EAAE,MAAoB;IACrE,MAAM,IAAI,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAY,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/C,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAa,EAAE,MAAiB;IAC7D,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,MAAc,EACd,KAAc,EACd,MAAuB;IAEvB,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO;IACT,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,0CAA0C;QAC1C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;YAC1E,kBAAkB,CAAC,GAAG,MAAM,IAAI,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;SAAM,IACL,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,SAAS;QAC1B,OAAO,KAAK,KAAK,QAAQ,EACzB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,qCAAqC,OAAO,KAAK,EAAE,CAAC,CAAC;IACvE,CAAC;AACH,CAAC"}

package/package.json

@@ -1,7 +1,41 @@
 {
   "name": "@databricks/sdk-oauth",
-  "version": "0.0.0-dev",
-  "description": "Bootstrap placeholder; real contents in a later release.",
-  "main": "index.js",
-  "license": "Apache-2.0"
+  "version": "0.1.0-dev.2",
+  "description": "",
+  "type": "module",
+  "exports": {
+    "./v1": {
+      "types": "./dist/v1/index.d.ts",
+      "import": "./dist/v1/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "src",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsc -b",
+    "lint": "eslint src --ext .ts",
+    "lint:fix": "eslint src --ext .ts --fix",
+    "format": "prettier --write \"src/**/*.ts\"",
+    "format:check": "prettier --check \"src/**/*.ts\"",
+    "typecheck": "tsc --noEmit",
+    "clean": "rm -rf dist tsconfig.tsbuildinfo",
+    "test": "echo 'no tests'",
+    "test:browser": "echo 'no tests'"
+  },
+  "author": "Databricks",
+  "license": "Apache-2.0",
+  "dependencies": {
+    "@databricks/sdk-auth": ">=0.1.0-dev.3 <1.0.0",
+    "@databricks/sdk-core": ">=0.1.0-dev.4 <1.0.0",
+    "@databricks/sdk-options": ">=0.1.0-dev.3 <1.0.0",
+    "@js-temporal/polyfill": "^0.5.0",
+    "json-bigint": "^1.0.0",
+    "zod": "^4.3.6"
+  },
+  "engines": {
+    "node": ">=22.0.0"
+  }
 }

package/src/v1/client.ts

@@ -0,0 +1,522 @@
+// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
+
+import {VERSION as AUTH_VERSION} from '@databricks/sdk-auth';
+import {createDefault} from '@databricks/sdk-core/clientinfo';
+import type {Logger} from '@databricks/sdk-core/logger';
+import {NoOpLogger} from '@databricks/sdk-core/logger';
+import type {CallOptions} from '@databricks/sdk-options/call';
+import type {ClientOptions} from '@databricks/sdk-options/client';
+import type {HttpClient} from '@databricks/sdk-core/http';
+import {newHttpClient} from './transport';
+import {
+  buildHttpRequest,
+  executeCall,
+  executeHttpCall,
+  marshalRequest,
+  parseResponse,
+} from './utils';
+import pkgJson from '../../package.json' with {type: 'json'};
+import type {
+  CreateCustomOAuthAppIntegrationRequest,
+  CreatePublishedOAuthAppIntegrationRequest,
+  CreatePublishedOAuthAppIntegrationRequest_Response,
+  CustomOAuthAppIntegration,
+  CustomOAuthAppIntegrationSecret,
+  DeleteCustomOAuthAppIntegrationRequest,
+  DeleteCustomOAuthAppIntegrationRequest_Response,
+  DeletePublishedOAuthAppIntegrationRequest,
+  DeletePublishedOAuthAppIntegrationRequest_Response,
+  GetCustomOAuthAppIntegrationRequest,
+  GetPublishedOAuthAppIntegrationRequest,
+  ListCustomOAuthAppIntegrationsRequest,
+  ListCustomOAuthAppIntegrationsRequest_Response,
+  ListPublishedOAuthAppIntegrationsRequest,
+  ListPublishedOAuthAppIntegrationsRequest_Response,
+  ListPublishedOAuthAppsRequest,
+  ListPublishedOAuthAppsRequest_Response,
+  PublishedOAuthApp,
+  PublishedOAuthAppIntegration,
+  UpdateCustomOAuthAppIntegrationRequest,
+  UpdateCustomOAuthAppIntegrationRequest_Response,
+  UpdatePublishedOAuthAppIntegrationRequest,
+  UpdatePublishedOAuthAppIntegrationRequest_Response,
+} from './model';
+import {
+  marshalCreateCustomOAuthAppIntegrationRequestSchema,
+  marshalCreatePublishedOAuthAppIntegrationRequestSchema,
+  marshalUpdateCustomOAuthAppIntegrationRequestSchema,
+  marshalUpdatePublishedOAuthAppIntegrationRequestSchema,
+  unmarshalCreatePublishedOAuthAppIntegrationRequest_ResponseSchema,
+  unmarshalCustomOAuthAppIntegrationSchema,
+  unmarshalCustomOAuthAppIntegrationSecretSchema,
+  unmarshalDeleteCustomOAuthAppIntegrationRequest_ResponseSchema,
+  unmarshalDeletePublishedOAuthAppIntegrationRequest_ResponseSchema,
+  unmarshalListCustomOAuthAppIntegrationsRequest_ResponseSchema,
+  unmarshalListPublishedOAuthAppIntegrationsRequest_ResponseSchema,
+  unmarshalListPublishedOAuthAppsRequest_ResponseSchema,
+  unmarshalPublishedOAuthAppIntegrationSchema,
+  unmarshalUpdateCustomOAuthAppIntegrationRequest_ResponseSchema,
+  unmarshalUpdatePublishedOAuthAppIntegrationRequest_ResponseSchema,
+} from './model';
+
+// Package identity segment for this client to be used in the User-Agent header.
+const PACKAGE_SEGMENT = {
+  key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
+  value: pkgJson.version,
+};
+
+export class OAuthClient {
+  private readonly host: string;
+  // Fallback for endpoints whose path contains {account_id}. If the request
+  // already carries an accountId, that value wins.
+  private readonly accountId: string | undefined;
+  private readonly httpClient: HttpClient;
+  private readonly logger: Logger;
+  // User-Agent header value. Composed once at construction from
+  // createDefault() merged with this package's identity and the active
+  // credential's name.
+  private readonly userAgent: string;
+
+  constructor(options: ClientOptions) {
+    if (options.host === undefined) {
+      throw new Error('Host is required.');
+    }
+    this.host = options.host.replace(/\/$/, '');
+    this.accountId = options.accountId;
+    this.logger = options.logger ?? new NoOpLogger();
+    const info = createDefault()
+      .with(PACKAGE_SEGMENT)
+      .with({key: 'sdk-js-auth', value: AUTH_VERSION})
+      .with({key: 'auth', value: options.credentials?.name() ?? 'default'});
+    this.userAgent = info.toString();
+    this.httpClient = newHttpClient(options);
+  }
+
+  /**
+   * Create Custom OAuth App Integration.
+   *
+   * You can retrieve the custom OAuth app integration via :method:CustomAppIntegration/get.
+   */
+  async createCustomOAuthAppIntegration(
+    req: CreateCustomOAuthAppIntegrationRequest,
+    options?: CallOptions
+  ): Promise<CustomOAuthAppIntegrationSecret> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/custom-app-integrations`;
+    const body = marshalRequest(
+      req,
+      marshalCreateCustomOAuthAppIntegrationRequestSchema
+    );
+    let resp: CustomOAuthAppIntegrationSecret | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers({'Content-Type': 'application/json'});
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalCustomOAuthAppIntegrationSecretSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Create Published OAuth App Integration.
+   *
+   * You can retrieve the published OAuth app integration via :method:PublishedAppIntegration/get.
+   */
+  async createPublishedOAuthAppIntegration(
+    req: CreatePublishedOAuthAppIntegrationRequest,
+    options?: CallOptions
+  ): Promise<CreatePublishedOAuthAppIntegrationRequest_Response> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/published-app-integrations`;
+    const body = marshalRequest(
+      req,
+      marshalCreatePublishedOAuthAppIntegrationRequestSchema
+    );
+    let resp: CreatePublishedOAuthAppIntegrationRequest_Response | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers({'Content-Type': 'application/json'});
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalCreatePublishedOAuthAppIntegrationRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Delete an existing Custom OAuth App Integration.
+   * You can retrieve the custom OAuth app integration via :method:CustomAppIntegration/get.
+   */
+  async deleteCustomOAuthAppIntegration(
+    req: DeleteCustomOAuthAppIntegrationRequest,
+    options?: CallOptions
+  ): Promise<DeleteCustomOAuthAppIntegrationRequest_Response> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/custom-app-integrations/${req.integrationId ?? ''}`;
+    let resp: DeleteCustomOAuthAppIntegrationRequest_Response | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalDeleteCustomOAuthAppIntegrationRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Delete an existing Published OAuth App Integration.
+   * You can retrieve the published OAuth app integration via :method:PublishedAppIntegration/get.
+   */
+  async deletePublishedOAuthAppIntegration(
+    req: DeletePublishedOAuthAppIntegrationRequest,
+    options?: CallOptions
+  ): Promise<DeletePublishedOAuthAppIntegrationRequest_Response> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/published-app-integrations/${req.integrationId ?? ''}`;
+    let resp: DeletePublishedOAuthAppIntegrationRequest_Response | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalDeletePublishedOAuthAppIntegrationRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+
+  /** Gets the Custom OAuth App Integration for the given integration id. */
+  async getCustomOAuthAppIntegration(
+    req: GetCustomOAuthAppIntegrationRequest,
+    options?: CallOptions
+  ): Promise<CustomOAuthAppIntegration> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/custom-app-integrations/${req.integrationId ?? ''}`;
+    let resp: CustomOAuthAppIntegration | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', url, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(respBody, unmarshalCustomOAuthAppIntegrationSchema);
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+
+  /** Gets the Published OAuth App Integration for the given integration id. */
+  async getPublishedOAuthAppIntegration(
+    req: GetPublishedOAuthAppIntegrationRequest,
+    options?: CallOptions
+  ): Promise<PublishedOAuthAppIntegration> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/published-app-integrations/${req.integrationId ?? ''}`;
+    let resp: PublishedOAuthAppIntegration | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', url, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalPublishedOAuthAppIntegrationSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+
+  /** Get the list of custom OAuth app integrations for the specified <Account> */
+  async listCustomOAuthAppIntegrations(
+    req: ListCustomOAuthAppIntegrationsRequest,
+    options?: CallOptions
+  ): Promise<ListCustomOAuthAppIntegrationsRequest_Response> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/custom-app-integrations`;
+    const params = new URLSearchParams();
+    if (req.pageToken !== undefined) {
+      params.append('page_token', req.pageToken);
+    }
+    if (req.pageSize !== undefined) {
+      params.append('page_size', String(req.pageSize));
+    }
+    if (req.includeCreatorUsername !== undefined) {
+      params.append(
+        'include_creator_username',
+        String(req.includeCreatorUsername)
+      );
+    }
+    const query = params.toString();
+    const fullUrl = query !== '' ? `${url}?${query}` : url;
+    let resp: ListCustomOAuthAppIntegrationsRequest_Response | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalListCustomOAuthAppIntegrationsRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+
+  async *listCustomOAuthAppIntegrationsIter(
+    req: ListCustomOAuthAppIntegrationsRequest,
+    options?: CallOptions
+  ): AsyncGenerator<CustomOAuthAppIntegration> {
+    const pageReq: ListCustomOAuthAppIntegrationsRequest = {...req};
+    for (;;) {
+      const resp = await this.listCustomOAuthAppIntegrations(pageReq, options);
+      for (const item of resp.apps ?? []) {
+        yield item;
+      }
+      if (resp.nextPageToken === undefined || resp.nextPageToken === '') {
+        return;
+      }
+      pageReq.pageToken = resp.nextPageToken;
+    }
+  }
+
+  /** Get the list of published OAuth app integrations for the specified <Account> */
+  async listPublishedOAuthAppIntegrations(
+    req: ListPublishedOAuthAppIntegrationsRequest,
+    options?: CallOptions
+  ): Promise<ListPublishedOAuthAppIntegrationsRequest_Response> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/published-app-integrations`;
+    const params = new URLSearchParams();
+    if (req.pageToken !== undefined) {
+      params.append('page_token', req.pageToken);
+    }
+    if (req.pageSize !== undefined) {
+      params.append('page_size', String(req.pageSize));
+    }
+    const query = params.toString();
+    const fullUrl = query !== '' ? `${url}?${query}` : url;
+    let resp: ListPublishedOAuthAppIntegrationsRequest_Response | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalListPublishedOAuthAppIntegrationsRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+
+  async *listPublishedOAuthAppIntegrationsIter(
+    req: ListPublishedOAuthAppIntegrationsRequest,
+    options?: CallOptions
+  ): AsyncGenerator<PublishedOAuthAppIntegration> {
+    const pageReq: ListPublishedOAuthAppIntegrationsRequest = {...req};
+    for (;;) {
+      const resp = await this.listPublishedOAuthAppIntegrations(
+        pageReq,
+        options
+      );
+      for (const item of resp.apps ?? []) {
+        yield item;
+      }
+      if (resp.nextPageToken === undefined || resp.nextPageToken === '') {
+        return;
+      }
+      pageReq.pageToken = resp.nextPageToken;
+    }
+  }
+
+  /** Get all the available published OAuth apps in <Databricks>. */
+  async listPublishedOAuthApps(
+    req: ListPublishedOAuthAppsRequest,
+    options?: CallOptions
+  ): Promise<ListPublishedOAuthAppsRequest_Response> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/published-apps`;
+    const params = new URLSearchParams();
+    if (req.pageToken !== undefined) {
+      params.append('page_token', req.pageToken);
+    }
+    if (req.pageSize !== undefined) {
+      params.append('page_size', String(req.pageSize));
+    }
+    const query = params.toString();
+    const fullUrl = query !== '' ? `${url}?${query}` : url;
+    let resp: ListPublishedOAuthAppsRequest_Response | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalListPublishedOAuthAppsRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+
+  async *listPublishedOAuthAppsIter(
+    req: ListPublishedOAuthAppsRequest,
+    options?: CallOptions
+  ): AsyncGenerator<PublishedOAuthApp> {
+    const pageReq: ListPublishedOAuthAppsRequest = {...req};
+    for (;;) {
+      const resp = await this.listPublishedOAuthApps(pageReq, options);
+      for (const item of resp.apps ?? []) {
+        yield item;
+      }
+      if (resp.nextPageToken === undefined || resp.nextPageToken === '') {
+        return;
+      }
+      pageReq.pageToken = resp.nextPageToken;
+    }
+  }
+
+  /**
+   * Updates an existing custom OAuth App Integration.
+   * You can retrieve the custom OAuth app integration via :method:CustomAppIntegration/get.
+   */
+  async updateCustomOAuthAppIntegration(
+    req: UpdateCustomOAuthAppIntegrationRequest,
+    options?: CallOptions
+  ): Promise<UpdateCustomOAuthAppIntegrationRequest_Response> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/custom-app-integrations/${req.integrationId ?? ''}`;
+    const body = marshalRequest(
+      req,
+      marshalUpdateCustomOAuthAppIntegrationRequestSchema
+    );
+    let resp: UpdateCustomOAuthAppIntegrationRequest_Response | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers({'Content-Type': 'application/json'});
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalUpdateCustomOAuthAppIntegrationRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+
+  /**
+   * Updates an existing published OAuth App Integration.
+   * You can retrieve the published OAuth app integration via :method:PublishedAppIntegration/get.
+   */
+  async updatePublishedOAuthAppIntegration(
+    req: UpdatePublishedOAuthAppIntegrationRequest,
+    options?: CallOptions
+  ): Promise<UpdatePublishedOAuthAppIntegrationRequest_Response> {
+    const url = `${this.host}/api/2.0/accounts/${req.accountId ?? this.accountId ?? ''}/oauth2/published-app-integrations/${req.integrationId ?? ''}`;
+    const body = marshalRequest(
+      req,
+      marshalUpdatePublishedOAuthAppIntegrationRequestSchema
+    );
+    let resp: UpdatePublishedOAuthAppIntegrationRequest_Response | undefined;
+    const call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers({'Content-Type': 'application/json'});
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalUpdatePublishedOAuthAppIntegrationRequest_ResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('operation completed without a result.');
+    }
+    return resp;
+  }
+}