From 22a82be4c8bf2c31621c4f814de5aab62b430589 Mon Sep 17 00:00:00 2001
From: Jared Snider <jsnider@bitwarden.com>
Date: Mon, 13 Jan 2025 17:42:37 -0500
Subject: [PATCH] PM-8113 - TwoFactorAuthComponent - some progress on strict TS
 conversion

---
 .../two-factor-auth.component.ts              | 29 ++++++++++++-------
 .../abstractions/login-strategy.service.ts    |  2 +-
 .../login-strategy.service.ts                 |  2 +-
 3 files changed, 21 insertions(+), 12 deletions(-)

diff --git a/libs/auth/src/angular/two-factor-auth/two-factor-auth.component.ts b/libs/auth/src/angular/two-factor-auth/two-factor-auth.component.ts
index d6dc53d28bb..eb53cb25bf0 100644
--- a/libs/auth/src/angular/two-factor-auth/two-factor-auth.component.ts
+++ b/libs/auth/src/angular/two-factor-auth/two-factor-auth.component.ts
@@ -79,9 +79,9 @@ import {
 export class TwoFactorAuthComponent implements OnInit, OnDestroy {
   loading = true;
 
-  token = "";
+  token: string | undefined = undefined;
   remember = false;
-  orgSsoIdentifier: string = null;
+  orgSsoIdentifier: string | undefined = undefined;
   inSsoFlow = false;
 
   providers = TwoFactorProviders;
@@ -104,7 +104,7 @@ export class TwoFactorAuthComponent implements OnInit, OnDestroy {
 
   title = "";
 
-  formPromise: Promise<any>;
+  formPromise: Promise<any> | undefined;
 
   submitForm = async () => {
     await this.submit();
@@ -137,7 +137,10 @@ export class TwoFactorAuthComponent implements OnInit, OnDestroy {
 
   async ngOnInit() {
     this.inSsoFlow = this.activatedRoute.snapshot.queryParamMap.get("sso") === "true";
-    this.orgSsoIdentifier = this.activatedRoute.snapshot.queryParamMap.get("identifier");
+
+    this.orgSsoIdentifier =
+      this.activatedRoute.snapshot.queryParamMap.get("identifier") ?? undefined;
+
     this.listenFor2faSessionTimeout();
 
     if (this.twoFactorAuthComponentService.shouldCheckForWebauthnResponseOnInit()) {
@@ -149,8 +152,13 @@ export class TwoFactorAuthComponent implements OnInit, OnDestroy {
     await this.setTitleByTwoFactorProviderType();
 
     this.form.valueChanges.pipe(takeUntilDestroyed(this.destroyRef)).subscribe((value) => {
-      this.token = value.token;
-      this.remember = value.remember;
+      if (value.token) {
+        this.token = value.token;
+      }
+
+      if (value.remember) {
+        this.remember = value.remember;
+      }
     });
 
     await this.twoFactorAuthComponentService.extendPopupWidthIfRequired?.(
@@ -215,7 +223,6 @@ export class TwoFactorAuthComponent implements OnInit, OnDestroy {
     try {
       this.formPromise = this.loginStrategyService.logInTwoFactor(
         new TokenTwoFactorRequest(this.selectedProviderType, this.token, this.remember),
-        null,
       );
       const authResult: AuthResult = await this.formPromise;
       this.logService.info("Successfully submitted two factor token");
@@ -293,7 +300,9 @@ export class TwoFactorAuthComponent implements OnInit, OnDestroy {
     // Save off the OrgSsoIdentifier for use in the TDE flows
     // - TDE login decryption options component
     // - Browser SSO on extension open
-    await this.ssoLoginService.setActiveUserOrganizationSsoIdentifier(this.orgSsoIdentifier);
+    if (this.orgSsoIdentifier !== undefined) {
+      await this.ssoLoginService.setActiveUserOrganizationSsoIdentifier(this.orgSsoIdentifier);
+    }
 
     // note: this flow affects both TDE & standard users
     if (this.isForcePasswordResetRequired(authResult)) {
@@ -387,7 +396,7 @@ export class TwoFactorAuthComponent implements OnInit, OnDestroy {
     await this.router.navigate(["login-initiated"]);
   }
 
-  private async handleChangePasswordRequired(orgIdentifier: string) {
+  private async handleChangePasswordRequired(orgIdentifier: string | undefined) {
     await this.router.navigate(["set-password"], {
       queryParams: {
         identifier: orgIdentifier,
@@ -414,7 +423,7 @@ export class TwoFactorAuthComponent implements OnInit, OnDestroy {
     return forceResetReasons.includes(authResult.forcePasswordReset);
   }
 
-  private async handleForcePasswordReset(orgIdentifier: string) {
+  private async handleForcePasswordReset(orgIdentifier: string | undefined) {
     await this.router.navigate(["update-temp-password"], {
       queryParams: {
         identifier: orgIdentifier,
diff --git a/libs/auth/src/common/abstractions/login-strategy.service.ts b/libs/auth/src/common/abstractions/login-strategy.service.ts
index 28aa2c17410..4e0a4dffde9 100644
--- a/libs/auth/src/common/abstractions/login-strategy.service.ts
+++ b/libs/auth/src/common/abstractions/login-strategy.service.ts
@@ -68,7 +68,7 @@ export abstract class LoginStrategyServiceAbstraction {
   logInTwoFactor: (
     twoFactor: TokenTwoFactorRequest,
     // TODO: PM-15162 - deprecate captchaResponse
-    captchaResponse: string,
+    captchaResponse?: string,
   ) => Promise<AuthResult>;
   /**
    * Creates a master key from the provided master password and email.
diff --git a/libs/auth/src/common/services/login-strategies/login-strategy.service.ts b/libs/auth/src/common/services/login-strategies/login-strategy.service.ts
index 57a653b205e..d3441a7f3f8 100644
--- a/libs/auth/src/common/services/login-strategies/login-strategy.service.ts
+++ b/libs/auth/src/common/services/login-strategies/login-strategy.service.ts
@@ -232,7 +232,7 @@ export class LoginStrategyService implements LoginStrategyServiceAbstraction {
 
   async logInTwoFactor(
     twoFactor: TokenTwoFactorRequest,
-    captchaResponse: string,
+    captchaResponse?: string,
   ): Promise<AuthResult> {
     if (!(await this.isSessionValid())) {
       throw new Error(this.i18nService.t("sessionTimeout"));