From 6224e712cf66a6c7f0bc1642a18ad720e8529767 Mon Sep 17 00:00:00 2001
From: Pavithra Ramesh <pavithrar@google.com>
Date: Thu, 26 Jan 2023 06:44:12 -0800
Subject: [PATCH] clear TOTP QR code display in the demo app once enrollment is
 complete.

---
 packages/auth/demo/src/index.js | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/packages/auth/demo/src/index.js b/packages/auth/demo/src/index.js
index 72c783d293f..5159d1169d7 100644
--- a/packages/auth/demo/src/index.js
+++ b/packages/auth/demo/src/index.js
@@ -100,6 +100,7 @@ let selectedMultiFactorHint = null;
 let recaptchaSize = 'normal';
 let webWorker = null;
 let totpSecret = null;
+let totpDeadlineId = null;
 
 // The corresponding Font Awesome icons for each provider.
 const providersIcons = {
@@ -715,11 +716,11 @@ async function onStartEnrollWithTotpMultiFactor() {
     );
     // display the numbr of seconds left to enroll.
     $('p.totp-deadline').show();
-    var id = setInterval(function () {
+    totpDeadlineId = setInterval(function () {
       var deadline = new Date(totpSecret.enrollmentCompletionDeadline);
       var t = deadline - new Date().getTime();
       if (t < 0) {
-        clearInterval(id);
+        clearInterval(totpDeadlineId);
         document.getElementById('totp-deadline').innerText =
           'TOTP enrollment expired!';
       } else {
@@ -756,12 +757,21 @@ async function onFinalizeEnrollWithTotpMultiFactor() {
   try {
     await multiFactor(activeUser()).enroll(multiFactorAssertion, displayName);
     refreshUserData();
+    clearTOTPUIState();
     alertSuccess('TOTP MFA enrolled!');
   } catch (e) {
     onAuthError(e);
   }
 }
 
+function clearTOTPUIState() {
+  $('p.totp-deadline').hide();
+  $('img.totp-qr-image').hide();
+  $('p.totp-text').hide();
+  $('enroll-mfa-totp-verification-code').hide();
+  $('enroll-mfa-totp-display-name').hide();
+  clearInterval(totpDeadlineId);
+}
 /**
  * Signs in or links a provider's credential, based on current tab opened.
  * @param {!AuthCredential} credential The provider's credential.