add AssignResourceRole method

permitio · Apr 3, 2024 · d2725ef · d2725ef
1 parent 6b01ac6
commit d2725ef
Showing 1 changed file with 20 additions and 3 deletions.
diff --git a/pkg/api/users.go b/pkg/api/users.go
@@ -168,16 +168,33 @@ func (u *Users) Delete(ctx context.Context, userKey string) error {
 // AssignRole assigns a role to a user in your context's environment, by user key, role key and tenant key.
 // Usage Example:
 // `roleAssignment, err := PermitClient.Api.Users.AssignRole(ctx, "user-key", "role-key", "default", "document:mydoc")`
-func (u *Users) AssignRole(ctx context.Context, userKey string, roleKey string, tenantKey string, resourceInstance *string) (*models.RoleAssignmentRead, error) {
+func (u *Users) AssignRole(ctx context.Context, userKey string, roleKey string, tenantKey string) (*models.RoleAssignmentRead, error) {
 	err := u.lazyLoadPermitContext(ctx)
 	if err != nil {
 		u.logger.Error("", zap.Error(err))
 		return nil, err
 	}
 	userRoleCreate := *models.NewUserRoleCreate(roleKey, tenantKey)
-	if resourceInstance != nil {
-		userRoleCreate.SetResourceInstance(*resourceInstance)
+	roleAssignmentRead, httpRes, err := u.client.UsersApi.AssignRoleToUser(ctx, u.config.Context.GetProject(), u.config.Context.GetEnvironment(), userKey).UserRoleCreate(userRoleCreate).Execute()
+	err = errors.HttpErrorHandle(err, httpRes)
+	if err != nil {
+		u.logger.Error("error assigning role:"+roleKey+" to user:"+userKey, zap.Error(err))
+		return nil, err
+	}
+	return roleAssignmentRead, nil
+}
+
+// AssignResourceRole assigns a role to a user in your context's environment, by user key, role key and tenant key.
+// Usage Example:
+// `roleAssignment, err := PermitClient.Api.Users.AssignRole(ctx, "user-key", "role-key", "default", "document:mydoc")`
+func (u *Users) AssignResourceRole(ctx context.Context, userKey string, roleKey string, tenantKey string, resourceInstance string) (*models.RoleAssignmentRead, error) {
+	err := u.lazyLoadPermitContext(ctx)
+	if err != nil {
+		u.logger.Error("", zap.Error(err))
+		return nil, err
 	}
+	userRoleCreate := *models.NewUserRoleCreate(roleKey, tenantKey)
+	userRoleCreate.SetResourceInstance(resourceInstance)
 	roleAssignmentRead, httpRes, err := u.client.UsersApi.AssignRoleToUser(ctx, u.config.Context.GetProject(), u.config.Context.GetEnvironment(), userKey).UserRoleCreate(userRoleCreate).Execute()
 	err = errors.HttpErrorHandle(err, httpRes)
 	if err != nil {