zitadel/console/src/app/pages/user-grant-create/user-grant-create.component.ts

import { Location } from '@angular/common';
import { Component, OnDestroy } from '@angular/core';
import { ActivatedRoute, Params } from '@angular/router';
import { Subscription } from 'rxjs';
import { UserGrantContext } from 'src/app/modules/user-grants/user-grants-datasource';
import { Org } from 'src/app/proto/generated/auth_pb';
import { ProjectGrantView, ProjectRole, ProjectView, User, UserGrant } from 'src/app/proto/generated/management_pb';
import { AuthService } from 'src/app/services/auth.service';
import { MgmtUserService } from 'src/app/services/mgmt-user.service';
import { ProjectService } from 'src/app/services/project.service';
import { ToastService } from 'src/app/services/toast.service';

@Component({
    selector: 'app-user-grant-create',
    templateUrl: './user-grant-create.component.html',
    styleUrls: ['./user-grant-create.component.scss'],
})
export class UserGrantCreateComponent implements OnDestroy {
    public context!: UserGrantContext;

    public org!: Org.AsObject;
    public userId: string = '';
    public projectId: string = '';
    public grantId: string = '';
    public rolesList: string[] = [];

    public STEPS: number = 2; // project, roles
    public currentCreateStep: number = 1;

    public filterValue: string = '';

    private subscription: Subscription = new Subscription();

    public UserGrantContext: any = UserGrantContext;

    public grantRolesKeyList: string[] = [];
    constructor(
        private authService: AuthService,
        private userService: MgmtUserService,
        private toast: ToastService,
        private _location: Location,
        private route: ActivatedRoute,
        private projectService: ProjectService,
    ) {
        this.subscription = this.route.params.subscribe((params: Params) => {
            const { context, projectid, grantid, userid } = params;
            this.context = context;

            this.projectId = projectid;
            this.grantId = grantid;
            this.userId = userid;

            if (this.projectId && !this.grantId) {
                this.context = UserGrantContext.OWNED_PROJECT;
            } else if (this.projectId && this.grantId) {
                this.context = UserGrantContext.GRANTED_PROJECT;
                this.projectService.GetGrantedProjectByID(this.projectId, this.grantId).then(resp => {
                    this.grantRolesKeyList = resp.toObject().roleKeysList;
                }).catch(error => {
                    this.toast.showError(error);
                });
            }
        });

        this.authService.GetActiveOrg().then(org => {
            this.org = org;
        });
    }

    public close(): void {
        this._location.back();
    }

    public addGrant(): void {
        switch (this.context) {
            case UserGrantContext.OWNED_PROJECT:
                this.userService.CreateProjectUserGrant(
                    this.projectId,
                    this.userId,
                    this.rolesList,
                ).then((data: UserGrant) => {
                    this.toast.showInfo('PROJECT.GRANT.TOAST.PROJECTGRANTADDED', true);
                    this.close();
                }).catch(error => {
                    this.toast.showError(error);
                });
                break;
            case UserGrantContext.GRANTED_PROJECT:
                this.userService.CreateProjectGrantUserGrant(
                    this.org.id,
                    this.projectId,
                    this.grantId,
                    this.userId,
                    this.rolesList,
                ).then((data: UserGrant) => {
                    this.toast.showInfo('PROJECT.GRANT.TOAST.PROJECTGRANTUSERGRANTADDED', true);
                    this.close();
                }).catch(error => {
                    this.toast.showError(error);
                });
                break;
        }

    }

    public selectProject(project: ProjectView.AsObject | ProjectGrantView.AsObject | any): void {
        this.projectId = project.projectId;
    }

    public selectUser(user: User.AsObject): void {
        this.userId = user.id;
    }

    public selectRoles(roles: ProjectRole.AsObject[]): void {
        this.rolesList = roles.map(role => role.key);
    }

    public next(): void {
        this.currentCreateStep++;
    }

    public previous(): void {
        this.currentCreateStep--;
    }

    public ngOnDestroy(): void {
        this.subscription.unsubscribe();
    }
}