update

Author: parthban-db

package-lock.json

@@ -19,6 +19,7 @@ export type {
   CheckPolicyRequest,
   CheckPolicyResponse,
   ConsistencyToken,
+  CreateAccessControlRequest,
   DeleteWorkspacePermissionAssignmentRequest,
   DeleteWorkspacePermissionAssignmentResponse,
   GetObjectPermissionsRequest,
@@ -42,6 +43,7 @@ export type {
   RuleSet,
   RuleSetUpdateRequest,
   SetObjectPermissionsRequest,
+  UpdateAccessControlRequest,
   UpdateObjectPermissionsRequest,
   UpdateRuleSetRequest,
   UpdateWorkspacePermissionAssignmentRequest,

packages/accessmanagement/src/v1/index.ts

@@ -131,6 +131,27 @@ export interface ConsistencyToken {
   value?: string | undefined;
 }
 
+export interface CreateAccessControlRequest {
+  principalName?:
+    | {
+        $case: 'userName';
+        /** name of the user */
+        userName: string;
+      }
+    | {
+        $case: 'groupName';
+        /** name of the group */
+        groupName: string;
+      }
+    | {
+        $case: 'servicePrincipalName';
+        /** application ID of a service principal */
+        servicePrincipalName: string;
+      }
+    | undefined;
+  permissionLevel?: PermissionLevel | undefined;
+}
+
 /** Removes all permission assignments for a workspace given a principal. */
 export interface DeleteWorkspacePermissionAssignmentRequest {
   /** The account ID. */
@@ -350,15 +371,36 @@ export interface SetObjectPermissionsRequest {
   requestObjectType?: string | undefined;
   /** The id of the request object. */
   requestObjectId?: string | undefined;
-  accessControlList?: AccessControlRequest[] | undefined;
+  accessControlList?: CreateAccessControlRequest[] | undefined;
+}
+
+export interface UpdateAccessControlRequest {
+  principalName?:
+    | {
+        $case: 'userName';
+        /** name of the user */
+        userName: string;
+      }
+    | {
+        $case: 'groupName';
+        /** name of the group */
+        groupName: string;
+      }
+    | {
+        $case: 'servicePrincipalName';
+        /** application ID of a service principal */
+        servicePrincipalName: string;
+      }
+    | undefined;
+  permissionLevel?: PermissionLevel | undefined;
 }
 
 export interface UpdateObjectPermissionsRequest {
   /** The type of the request object. Can be one of the following: alerts, alertsv2, authorization, clusters, cluster-policies, dashboards, database-projects, dbsql-dashboards, directories, experiments, files, genie, instance-pools, jobs, knowledge-assistants, notebooks, pipelines, queries, registered-models, repos, serving-endpoints, supervisor-agents, vector-search-endpoints, or warehouses. */
   requestObjectType?: string | undefined;
   /** The id of the request object. */
   requestObjectId?: string | undefined;
-  accessControlList?: AccessControlRequest[] | undefined;
+  accessControlList?: UpdateAccessControlRequest[] | undefined;
 }
 
 export interface UpdateRuleSetRequest {
@@ -610,7 +652,27 @@ export const unmarshalWorkspacePermissionAssignmentOutputSchema: z.ZodType<Works
       error: d.error,
     }));
 
-export const marshalAccessControlRequestSchema: z.ZodType = z
+export const marshalActorSchema: z.ZodType = z
+  .object({
+    kind: z
+      .discriminatedUnion('$case', [
+        z.object({$case: z.literal('actorId'), actorId: z.bigint()}),
+      ])
+      .optional(),
+  })
+  .transform(d => ({
+    ...(d.kind?.$case === 'actorId' && {actor_id: d.kind.actorId}),
+  }));
+
+export const marshalConsistencyTokenSchema: z.ZodType = z
+  .object({
+    value: z.string().optional(),
+  })
+  .transform(d => ({
+    value: d.value,
+  }));
+
+export const marshalCreateAccessControlRequestSchema: z.ZodType = z
   .object({
     principalName: z
       .discriminatedUnion('$case', [
@@ -637,26 +699,6 @@ export const marshalAccessControlRequestSchema: z.ZodType = z
     permission_level: d.permissionLevel,
   }));
 
-export const marshalActorSchema: z.ZodType = z
-  .object({
-    kind: z
-      .discriminatedUnion('$case', [
-        z.object({$case: z.literal('actorId'), actorId: z.bigint()}),
-      ])
-      .optional(),
-  })
-  .transform(d => ({
-    ...(d.kind?.$case === 'actorId' && {actor_id: d.kind.actorId}),
-  }));
-
-export const marshalConsistencyTokenSchema: z.ZodType = z
-  .object({
-    value: z.string().optional(),
-  })
-  .transform(d => ({
-    value: d.value,
-  }));
-
 export const marshalGrantRuleSchema: z.ZodType = z
   .object({
     principals: z.array(z.string()).optional(),
@@ -696,7 +738,7 @@ export const marshalSetObjectPermissionsRequestSchema: z.ZodType = z
     requestObjectType: z.string().optional(),
     requestObjectId: z.string().optional(),
     accessControlList: z
-      .array(z.lazy(() => marshalAccessControlRequestSchema))
+      .array(z.lazy(() => marshalCreateAccessControlRequestSchema))
       .optional(),
   })
   .transform(d => ({
@@ -705,12 +747,39 @@ export const marshalSetObjectPermissionsRequestSchema: z.ZodType = z
     access_control_list: d.accessControlList,
   }));
 
+export const marshalUpdateAccessControlRequestSchema: z.ZodType = z
+  .object({
+    principalName: z
+      .discriminatedUnion('$case', [
+        z.object({$case: z.literal('userName'), userName: z.string()}),
+        z.object({$case: z.literal('groupName'), groupName: z.string()}),
+        z.object({
+          $case: z.literal('servicePrincipalName'),
+          servicePrincipalName: z.string(),
+        }),
+      ])
+      .optional(),
+    permissionLevel: z.string().optional(),
+  })
+  .transform(d => ({
+    ...(d.principalName?.$case === 'userName' && {
+      user_name: d.principalName.userName,
+    }),
+    ...(d.principalName?.$case === 'groupName' && {
+      group_name: d.principalName.groupName,
+    }),
+    ...(d.principalName?.$case === 'servicePrincipalName' && {
+      service_principal_name: d.principalName.servicePrincipalName,
+    }),
+    permission_level: d.permissionLevel,
+  }));
+
 export const marshalUpdateObjectPermissionsRequestSchema: z.ZodType = z
   .object({
     requestObjectType: z.string().optional(),
     requestObjectId: z.string().optional(),
     accessControlList: z
-      .array(z.lazy(() => marshalAccessControlRequestSchema))
+      .array(z.lazy(() => marshalUpdateAccessControlRequestSchema))
       .optional(),
   })
   .transform(d => ({

packages/accessmanagement/src/v1/model.ts

@@ -27,7 +27,8 @@ import type {
   UpdateAlertRequest,
 } from './model';
 import {
-  marshalAlertSchema,
+  marshalCreateAlertSchema,
+  marshalUpdateAlertSchema,
   unmarshalAlertSchema,
   unmarshalEmptySchema,
   unmarshalListAlertsResponseSchema,
@@ -73,7 +74,7 @@ export class AlertsClient {
   ): Promise<Alert> {
     const {host, workspaceId, httpClient} = await this.resolveConfig();
     const url = `${host}/api/2.0/alerts`;
-    const body = marshalRequest(req.alert, marshalAlertSchema);
+    const body = marshalRequest(req.alert, marshalCreateAlertSchema);
     let resp: Alert | undefined;
     const call = async (callSignal?: AbortSignal): Promise<void> => {
       const headers = new Headers({'Content-Type': 'application/json'});
@@ -225,7 +226,7 @@ export class AlertsClient {
     }
     const query = params.toString();
     const fullUrl = query !== '' ? `${url}?${query}` : url;
-    const body = marshalRequest(req.alert, marshalAlertSchema);
+    const body = marshalRequest(req.alert, marshalUpdateAlertSchema);
     let resp: Alert | undefined;
     const call = async (callSignal?: AbortSignal): Promise<void> => {
       const headers = new Headers({'Content-Type': 'application/json'});

packages/alerts/src/v2/client.ts

@@ -23,14 +23,32 @@ export type {
   AlertOperandValue,
   AlertRunAs,
   AlertSubscription,
+  CreateAlert,
+  CreateAlertEvaluation,
+  CreateAlertNotification,
+  CreateAlertOperand,
+  CreateAlertOperandColumn,
+  CreateAlertOperandValue,
   CreateAlertRequest,
+  CreateAlertRunAs,
+  CreateAlertSubscription,
+  CreateCronSchedule,
   CronSchedule,
   Empty,
   GetAlertRequest,
   ListAlertsRequest,
   ListAlertsResponse,
   TrashAlertRequest,
+  UpdateAlert,
+  UpdateAlertEvaluation,
+  UpdateAlertNotification,
+  UpdateAlertOperand,
+  UpdateAlertOperandColumn,
+  UpdateAlertOperandValue,
   UpdateAlertRequest,
+  UpdateAlertRunAs,
+  UpdateAlertSubscription,
+  UpdateCronSchedule,
 } from './model';
 
-export {alertFieldMask} from './model';
+export {updateAlertFieldMask} from './model';