zitadel/console/src/app/modules/features/features.component.ts

import { Component, Injector, OnDestroy, Type } from '@angular/core';
import { ActivatedRoute } from '@angular/router';
import { Subscription } from 'rxjs';
import { switchMap } from 'rxjs/operators';
import {
    GetOrgFeaturesResponse,
    SetDefaultFeaturesRequest,
    SetOrgFeaturesRequest,
} from 'src/app/proto/generated/zitadel/admin_pb';
import { Features } from 'src/app/proto/generated/zitadel/features_pb';
import { GetFeaturesResponse } from 'src/app/proto/generated/zitadel/management_pb';
import { Org } from 'src/app/proto/generated/zitadel/org_pb';
import { AdminService } from 'src/app/services/admin.service';
import { ManagementService } from 'src/app/services/mgmt.service';
import { StorageService } from 'src/app/services/storage.service';
import { ToastService } from 'src/app/services/toast.service';

export enum FeatureServiceType {
    MGMT,
    ADMIN,
}

@Component({
    selector: 'app-features',
    templateUrl: './features.component.html',
    styleUrls: ['./features.component.scss'],
})
export class FeaturesComponent implements OnDestroy {
    private managementService!: ManagementService;
    public serviceType!: FeatureServiceType;

    public features!: Features.AsObject;

    private sub: Subscription = new Subscription();
    private org!: Org.AsObject;

    public FeatureServiceType: any = FeatureServiceType;
    constructor(
        private route: ActivatedRoute,
        private toast: ToastService,
        private sessionStorage: StorageService,
        private injector: Injector,
        private adminService: AdminService,
    ) {
        const temporg = this.sessionStorage.getItem('organization') as Org.AsObject;
        if (temporg) {
            this.org = temporg;
        }
        this.sub = this.route.data.pipe(switchMap(data => {
            this.serviceType = data.serviceType;
            if (this.serviceType === FeatureServiceType.MGMT) {
                this.managementService = this.injector.get(ManagementService as Type<ManagementService>);
            }
            return this.route.params;
        })).subscribe(_ => {
            this.fetchData();
        });
    }

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

    public fetchData(): void {
        this.getData().then(resp => {
            if (resp?.features) {
                this.features = resp.features;
            }
        });
    }

    private async getData(): Promise<GetFeaturesResponse.AsObject | GetOrgFeaturesResponse.AsObject | undefined> {
        switch (this.serviceType) {
            case FeatureServiceType.MGMT:
                return this.managementService.getFeatures();
            case FeatureServiceType.ADMIN:
                if (this.org?.id) {
                    return this.adminService.getDefaultFeatures();
                }
                break;
        }
    }

    public savePolicy(): void {
        switch (this.serviceType) {
            case FeatureServiceType.MGMT:
                const req = new SetOrgFeaturesRequest();
                req.setOrgId(this.org.id);

                req.setLoginPolicyUsernameLogin(this.features.loginPolicyUsernameLogin);
                req.setLoginPolicyRegistration(this.features.loginPolicyRegistration);
                req.setLoginPolicyIdp(this.features.loginPolicyIdp);
                req.setLoginPolicyFactors(this.features.loginPolicyFactors);
                req.setLoginPolicyPasswordless(this.features.loginPolicyPasswordless);
                req.setPasswordComplexityPolicy(this.features.passwordComplexityPolicy);
                req.setLabelPolicy(this.features.labelPolicy);

                this.adminService.setOrgFeatures(req).then(() => {
                    this.toast.showInfo('POLICY.TOAST.SET', true);
                }).catch(error => {
                    this.toast.showError(error);
                });
                break;
            case FeatureServiceType.ADMIN:
                // update Default org iam policy?
                const dreq = new SetDefaultFeaturesRequest();
                dreq.setLoginPolicyUsernameLogin(this.features.loginPolicyUsernameLogin);
                dreq.setLoginPolicyRegistration(this.features.loginPolicyRegistration);
                dreq.setLoginPolicyIdp(this.features.loginPolicyIdp);
                dreq.setLoginPolicyFactors(this.features.loginPolicyFactors);
                dreq.setLoginPolicyPasswordless(this.features.loginPolicyPasswordless);
                dreq.setPasswordComplexityPolicy(this.features.passwordComplexityPolicy);
                dreq.setLabelPolicy(this.features.labelPolicy);

                this.adminService.setDefaultFeatures(dreq).then(() => {
                    this.toast.showInfo('POLICY.TOAST.SET', true);
                }).catch(error => {
                    this.toast.showError(error);
                });
                break;
        }
    }

    public resetFeatures(): void {
        if (this.serviceType === FeatureServiceType.MGMT) {
            this.adminService.resetOrgFeatures(this.org.id).then(() => {
                this.toast.showInfo('POLICY.TOAST.RESETSUCCESS', true);
                setTimeout(() => {
                    this.fetchData();
                }, 1000);
            }).catch(error => {
                this.toast.showError(error);
            });
        }
    }

    public get isDefault(): boolean {
        if (this.features && this.serviceType === FeatureServiceType.MGMT) {
            return this.features.isDefault;
        } else {
            return false;
        }
    }
}