import { Directive, EventEmitter, Output } from "@angular/core"; import { ApiService } from "@bitwarden/common/abstractions/api.service"; import { I18nService } from "@bitwarden/common/abstractions/i18n.service"; import { LogService } from "@bitwarden/common/abstractions/log.service"; import { PlatformUtilsService } from "@bitwarden/common/abstractions/platformUtils.service"; import { UserVerificationService } from "@bitwarden/common/abstractions/userVerification.service"; import { TwoFactorProviderType } from "@bitwarden/common/enums/twoFactorProviderType"; import { VerificationType } from "@bitwarden/common/enums/verificationType"; import { SecretVerificationRequest } from "@bitwarden/common/models/request/secretVerificationRequest"; import { TwoFactorProviderRequest } from "@bitwarden/common/models/request/twoFactorProviderRequest"; @Directive() export abstract class TwoFactorBaseComponent { @Output() onUpdated = new EventEmitter(); type: TwoFactorProviderType; organizationId: string; twoFactorProviderType = TwoFactorProviderType; enabled = false; authed = false; protected hashedSecret: string; protected verificationType: VerificationType; constructor( protected apiService: ApiService, protected i18nService: I18nService, protected platformUtilsService: PlatformUtilsService, protected logService: LogService, protected userVerificationService: UserVerificationService ) {} protected auth(authResponse: any) { this.hashedSecret = authResponse.secret; this.verificationType = authResponse.verificationType; this.authed = true; } protected async enable(enableFunction: () => Promise) { try { await enableFunction(); this.onUpdated.emit(true); } catch (e) { this.logService.error(e); } } protected async disable(promise: Promise) { const confirmed = await this.platformUtilsService.showDialog( this.i18nService.t("twoStepDisableDesc"), this.i18nService.t("disable"), this.i18nService.t("yes"), this.i18nService.t("no"), "warning" ); if (!confirmed) { return; } try { const request = await this.buildRequestModel(TwoFactorProviderRequest); request.type = this.type; if (this.organizationId != null) { promise = this.apiService.putTwoFactorOrganizationDisable(this.organizationId, request); } else { promise = this.apiService.putTwoFactorDisable(request); } await promise; this.enabled = false; this.platformUtilsService.showToast("success", null, this.i18nService.t("twoStepDisabled")); this.onUpdated.emit(false); } catch (e) { this.logService.error(e); } } protected async buildRequestModel( requestClass: new () => T ) { return this.userVerificationService.buildRequest( { secret: this.hashedSecret, type: this.verificationType, }, requestClass, true ); } }