@databricks/sdk-postgres 0.2.0 → 0.3.0

package/dist/v1/client.js

@@ -6,7 +6,7 @@ import { resolveClientConfig } from './transport';
 import { buildHttpRequest, executeCall, executeHttpCall, marshalRequest, parseResponse, executeWait, StillRunningError, } from './utils';
 import pkgJson from '../../package.json' with { type: 'json' };
 import { z } from 'zod';
-import { marshalBranchSchema, marshalCatalogSchema, marshalDatabaseSchema, marshalEndpointSchema, marshalGenerateDatabaseCredentialRequestSchema, marshalProjectSchema, marshalRoleSchema, marshalSyncedTableSchema, marshalUndeleteBranchRequestSchema, marshalUndeleteProjectRequestSchema, unmarshalBranchOperationMetadataSchema, unmarshalBranchSchema, unmarshalCatalogOperationMetadataSchema, unmarshalCatalogSchema, unmarshalDatabaseCredentialSchema, unmarshalDatabaseOperationMetadataSchema, unmarshalDatabaseSchema, unmarshalEndpointOperationMetadataSchema, unmarshalEndpointSchema, unmarshalListBranchesResponseSchema, unmarshalListDatabasesResponseSchema, unmarshalListEndpointsResponseSchema, unmarshalListProjectsResponseSchema, unmarshalListRolesResponseSchema, unmarshalOperationSchema, unmarshalProjectOperationMetadataSchema, unmarshalProjectSchema, unmarshalRoleOperationMetadataSchema, unmarshalRoleSchema, unmarshalSyncedTableOperationMetadataSchema, unmarshalSyncedTableSchema, } from './model';
+import { marshalBranchSchema, marshalCatalogSchema, marshalDataApiSchema, marshalDatabaseSchema, marshalEndpointSchema, marshalGenerateDatabaseCredentialRequestSchema, marshalProjectSchema, marshalRoleSchema, marshalSyncedTableSchema, marshalUndeleteBranchRequestSchema, marshalUndeleteProjectRequestSchema, unmarshalBranchOperationMetadataSchema, unmarshalBranchSchema, unmarshalCatalogOperationMetadataSchema, unmarshalCatalogSchema, unmarshalDataApiOperationMetadataSchema, unmarshalDataApiSchema, unmarshalDatabaseCredentialSchema, unmarshalDatabaseOperationMetadataSchema, unmarshalDatabaseSchema, unmarshalEndpointOperationMetadataSchema, unmarshalEndpointSchema, unmarshalListBranchesResponseSchema, unmarshalListDatabasesResponseSchema, unmarshalListEndpointsResponseSchema, unmarshalListProjectsResponseSchema, unmarshalListRolesResponseSchema, unmarshalOperationSchema, unmarshalProjectOperationMetadataSchema, unmarshalProjectSchema, unmarshalRoleOperationMetadataSchema, unmarshalRoleSchema, unmarshalSyncedTableOperationMetadataSchema, unmarshalSyncedTableSchema, } 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-/, ''),
@@ -54,7 +54,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -91,7 +91,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -113,6 +113,37 @@ export class PostgresClient {
         const op = await this.createCatalogBase(req, options);
         return new CreateCatalogOperation(op, (req, options) => this.getOperation(req, options));
     }
+    /** Enable Data API for a database. */
+    async createDataApiBase(req, options) {
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/postgres/${req.parent ?? ''}/data-api`;
+        const body = marshalRequest(req.dataApi, marshalDataApiSchema);
+        let resp;
+        const call = async (callSignal) => {
+            const headers = new Headers({ 'Content-Type': 'application/json' });
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
+            }
+            headers.set('User-Agent', this.userAgent);
+            const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
+            const respBody = await executeHttpCall({
+                request: httpReq,
+                httpClient,
+                logger: this.logger,
+            });
+            resp = parseResponse(respBody, unmarshalOperationSchema);
+        };
+        await executeCall(call, options);
+        if (resp === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return resp;
+    }
+    /** Enable Data API for a database. */
+    async createDataApi(req, options) {
+        const op = await this.createDataApiBase(req, options);
+        return new CreateDataApiOperation(op, (req, options) => this.getOperation(req, options));
+    }
     /**
      * Create a Database.
      *
@@ -132,7 +163,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -176,7 +207,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -213,7 +244,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -250,7 +281,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -287,7 +318,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', fullUrl, headers, callSignal, body);
@@ -323,7 +354,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
@@ -353,7 +384,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
@@ -375,6 +406,36 @@ export class PostgresClient {
         const op = await this.deleteCatalogBase(req, options);
         return new DeleteCatalogOperation(op, (req, options) => this.getOperation(req, options));
     }
+    /** Disable Data API for a database. */
+    async deleteDataApiBase(req, options) {
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
+        let resp;
+        const call = async (callSignal) => {
+            const headers = new Headers();
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
+            }
+            headers.set('User-Agent', this.userAgent);
+            const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
+            const respBody = await executeHttpCall({
+                request: httpReq,
+                httpClient,
+                logger: this.logger,
+            });
+            resp = parseResponse(respBody, unmarshalOperationSchema);
+        };
+        await executeCall(call, options);
+        if (resp === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return resp;
+    }
+    /** Disable Data API for a database. */
+    async deleteDataApi(req, options) {
+        const op = await this.deleteDataApiBase(req, options);
+        return new DeleteDataApiOperation(op, (req, options) => this.getOperation(req, options));
+    }
     /** Delete a Database. */
     async deleteDatabaseBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -383,7 +444,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
@@ -413,7 +474,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
@@ -449,7 +510,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
@@ -485,7 +546,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', fullUrl, headers, callSignal);
@@ -515,7 +576,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
@@ -546,7 +607,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
@@ -571,7 +632,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -596,7 +657,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -613,6 +674,31 @@ export class PostgresClient {
         }
         return resp;
     }
+    /** Get Data API configuration for a database. */
+    async getDataApi(req, options) {
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/postgres/${req.name ?? ''}`;
+        let resp;
+        const call = async (callSignal) => {
+            const headers = new Headers();
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
+            }
+            headers.set('User-Agent', this.userAgent);
+            const httpReq = buildHttpRequest('GET', url, headers, callSignal);
+            const respBody = await executeHttpCall({
+                request: httpReq,
+                httpClient,
+                logger: this.logger,
+            });
+            resp = parseResponse(respBody, unmarshalDataApiSchema);
+        };
+        await executeCall(call, options);
+        if (resp === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return resp;
+    }
     /** Get a Database. */
     async getDatabase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -621,7 +707,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -646,7 +732,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -671,7 +757,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -696,7 +782,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -721,7 +807,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -746,7 +832,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', url, headers, callSignal);
@@ -783,7 +869,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
@@ -830,7 +916,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
@@ -877,7 +963,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
@@ -927,7 +1013,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
@@ -974,7 +1060,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers();
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
@@ -1013,7 +1099,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
@@ -1044,7 +1130,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
@@ -1081,7 +1167,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
@@ -1103,6 +1189,43 @@ export class PostgresClient {
         const op = await this.updateBranchBase(req, options);
         return new UpdateBranchOperation(op, (req, options) => this.getOperation(req, options));
     }
+    /** Update Data API configuration for a database. */
+    async updateDataApiBase(req, options) {
+        const { host, workspaceId, httpClient } = await this.resolveConfig();
+        const url = `${host}/api/2.0/postgres/${req.dataApi?.name ?? ''}`;
+        const params = new URLSearchParams();
+        if (req.updateMask !== undefined) {
+            params.append('update_mask', req.updateMask.toString());
+        }
+        const query = params.toString();
+        const fullUrl = query !== '' ? `${url}?${query}` : url;
+        const body = marshalRequest(req.dataApi, marshalDataApiSchema);
+        let resp;
+        const call = async (callSignal) => {
+            const headers = new Headers({ 'Content-Type': 'application/json' });
+            if (workspaceId !== undefined) {
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
+            }
+            headers.set('User-Agent', this.userAgent);
+            const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
+            const respBody = await executeHttpCall({
+                request: httpReq,
+                httpClient,
+                logger: this.logger,
+            });
+            resp = parseResponse(respBody, unmarshalOperationSchema);
+        };
+        await executeCall(call, options);
+        if (resp === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return resp;
+    }
+    /** Update Data API configuration for a database. */
+    async updateDataApi(req, options) {
+        const op = await this.updateDataApiBase(req, options);
+        return new UpdateDataApiOperation(op, (req, options) => this.getOperation(req, options));
+    }
     /** Update a Database. */
     async updateDatabaseBase(req, options) {
         const { host, workspaceId, httpClient } = await this.resolveConfig();
@@ -1118,7 +1241,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
@@ -1155,7 +1278,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
@@ -1192,7 +1315,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
@@ -1229,7 +1352,7 @@ export class PostgresClient {
         const call = async (callSignal) => {
             const headers = new Headers({ 'Content-Type': 'application/json' });
             if (workspaceId !== undefined) {
-                headers.set('X-Databricks-Org-Id', workspaceId);
+                headers.set('X-Databricks-Workspace-Id', workspaceId);
             }
             headers.set('User-Agent', this.userAgent);
             const httpReq = buildHttpRequest('PATCH', fullUrl, headers, callSignal, body);
@@ -1384,6 +1507,72 @@ export class CreateCatalogOperation {
         return op.done;
     }
 }
+export class CreateDataApiOperation {
+    operation;
+    getOperation;
+    constructor(operation, getOperation) {
+        this.operation = operation;
+        this.getOperation = getOperation;
+    }
+    /** Returns the server-assigned name of the long-running operation. */
+    name() {
+        return Promise.resolve(this.operation.name);
+    }
+    /** Returns metadata associated with the long-running operation. */
+    metadata() {
+        if (this.operation.metadata === undefined) {
+            return Promise.resolve(undefined);
+        }
+        return Promise.resolve(z
+            .lazy(() => unmarshalDataApiOperationMetadataSchema)
+            .parse(this.operation.metadata));
+    }
+    /**
+     * Polls the operation until it completes.
+     *
+     * Throws if the operation failed.
+     */
+    async wait(options) {
+        let result;
+        const call = async (callSignal) => {
+            const op = await this.getOperation({
+                name: this.operation.name,
+            }, callSignal !== undefined ? { signal: callSignal } : undefined);
+            this.operation = op;
+            if (op.done === undefined) {
+                throw new Error('operation is missing the done field');
+            }
+            if (!op.done) {
+                throw new StillRunningError();
+            }
+            if (op.result?.$case === 'error') {
+                const err = op.result.error;
+                const msg = err.message !== undefined && err.message !== ''
+                    ? err.message
+                    : 'unknown error';
+                const errorMsg = err.errorCode !== undefined ? `[${err.errorCode}] ${msg}` : msg;
+                throw new Error(`operation failed: ${errorMsg}`, {
+                    cause: err,
+                });
+            }
+            if (op.result?.$case !== 'response') {
+                throw new Error('operation completed without a response');
+            }
+            result = z.lazy(() => unmarshalDataApiSchema).parse(op.result.response);
+        };
+        await executeWait(call, options);
+        if (result === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return result;
+    }
+    /** Checks whether the operation has completed */
+    async done(options) {
+        const op = await this.getOperation({ name: this.operation.name }, options);
+        this.operation = op;
+        return op.done;
+    }
+}
 export class CreateDatabaseOperation {
     operation;
     getOperation;
@@ -1830,6 +2019,63 @@ export class DeleteCatalogOperation {
         return op.done;
     }
 }
+export class DeleteDataApiOperation {
+    operation;
+    getOperation;
+    constructor(operation, getOperation) {
+        this.operation = operation;
+        this.getOperation = getOperation;
+    }
+    /** Returns the server-assigned name of the long-running operation. */
+    name() {
+        return Promise.resolve(this.operation.name);
+    }
+    /** Returns metadata associated with the long-running operation. */
+    metadata() {
+        if (this.operation.metadata === undefined) {
+            return Promise.resolve(undefined);
+        }
+        return Promise.resolve(z
+            .lazy(() => unmarshalDataApiOperationMetadataSchema)
+            .parse(this.operation.metadata));
+    }
+    /**
+     * Polls the operation until it completes.
+     *
+     * Throws if the operation failed.
+     */
+    async wait(options) {
+        const call = async (callSignal) => {
+            const op = await this.getOperation({
+                name: this.operation.name,
+            }, callSignal !== undefined ? { signal: callSignal } : undefined);
+            this.operation = op;
+            if (op.done === undefined) {
+                throw new Error('operation is missing the done field');
+            }
+            if (!op.done) {
+                throw new StillRunningError();
+            }
+            if (op.result?.$case === 'error') {
+                const err = op.result.error;
+                const msg = err.message !== undefined && err.message !== ''
+                    ? err.message
+                    : 'unknown error';
+                const errorMsg = err.errorCode !== undefined ? `[${err.errorCode}] ${msg}` : msg;
+                throw new Error(`operation failed: ${errorMsg}`, {
+                    cause: err,
+                });
+            }
+        };
+        await executeWait(call, options);
+    }
+    /** Checks whether the operation has completed */
+    async done(options) {
+        const op = await this.getOperation({ name: this.operation.name }, options);
+        this.operation = op;
+        return op.done;
+    }
+}
 export class DeleteDatabaseOperation {
     operation;
     getOperation;
@@ -2295,6 +2541,72 @@ export class UpdateBranchOperation {
         return op.done;
     }
 }
+export class UpdateDataApiOperation {
+    operation;
+    getOperation;
+    constructor(operation, getOperation) {
+        this.operation = operation;
+        this.getOperation = getOperation;
+    }
+    /** Returns the server-assigned name of the long-running operation. */
+    name() {
+        return Promise.resolve(this.operation.name);
+    }
+    /** Returns metadata associated with the long-running operation. */
+    metadata() {
+        if (this.operation.metadata === undefined) {
+            return Promise.resolve(undefined);
+        }
+        return Promise.resolve(z
+            .lazy(() => unmarshalDataApiOperationMetadataSchema)
+            .parse(this.operation.metadata));
+    }
+    /**
+     * Polls the operation until it completes.
+     *
+     * Throws if the operation failed.
+     */
+    async wait(options) {
+        let result;
+        const call = async (callSignal) => {
+            const op = await this.getOperation({
+                name: this.operation.name,
+            }, callSignal !== undefined ? { signal: callSignal } : undefined);
+            this.operation = op;
+            if (op.done === undefined) {
+                throw new Error('operation is missing the done field');
+            }
+            if (!op.done) {
+                throw new StillRunningError();
+            }
+            if (op.result?.$case === 'error') {
+                const err = op.result.error;
+                const msg = err.message !== undefined && err.message !== ''
+                    ? err.message
+                    : 'unknown error';
+                const errorMsg = err.errorCode !== undefined ? `[${err.errorCode}] ${msg}` : msg;
+                throw new Error(`operation failed: ${errorMsg}`, {
+                    cause: err,
+                });
+            }
+            if (op.result?.$case !== 'response') {
+                throw new Error('operation completed without a response');
+            }
+            result = z.lazy(() => unmarshalDataApiSchema).parse(op.result.response);
+        };
+        await executeWait(call, options);
+        if (result === undefined) {
+            throw new Error('operation completed without a result.');
+        }
+        return result;
+    }
+    /** Checks whether the operation has completed */
+    async done(options) {
+        const op = await this.getOperation({ name: this.operation.name }, options);
+        this.operation = op;
+        return op.done;
+    }
+}
 export class UpdateDatabaseOperation {
     operation;
     getOperation;