From 72767dec748fac597c5ebbcb9b0574df03e63939 Mon Sep 17 00:00:00 2001
From: Jason Ng <jng@bitwarden.com>
Date: Thu, 15 Aug 2024 10:03:14 -0400
Subject: [PATCH] [PM-10514] New TOTP Counter Component (#10486)

* new vault totp countdown component
---
 .../cipher-view/cipher-view.component.html    |  6 +-
 .../login-credentials-view.component.html     | 35 +++++---
 .../login-credentials-view.component.ts       | 14 +++-
 .../totp-countdown.component.html             | 34 ++++++++
 .../totp-countdown.component.ts               | 80 +++++++++++++++++++
 5 files changed, 149 insertions(+), 20 deletions(-)
 create mode 100644 libs/vault/src/components/totp-countdown/totp-countdown.component.html
 create mode 100644 libs/vault/src/components/totp-countdown/totp-countdown.component.ts

diff --git a/libs/vault/src/cipher-view/cipher-view.component.html b/libs/vault/src/cipher-view/cipher-view.component.html
index 2a2a4ded05..68c80a7bd5 100644
--- a/libs/vault/src/cipher-view/cipher-view.component.html
+++ b/libs/vault/src/cipher-view/cipher-view.component.html
@@ -9,11 +9,7 @@
   </app-item-details-v2>
 
   <!-- LOGIN CREDENTIALS -->
-  <app-login-credentials-view
-    *ngIf="hasLogin"
-    [login]="cipher.login"
-    [viewPassword]="cipher.viewPassword"
-  ></app-login-credentials-view>
+  <app-login-credentials-view *ngIf="hasLogin" [cipher]="cipher"></app-login-credentials-view>
 
   <!-- AUTOFILL OPTIONS -->
   <app-autofill-options-view *ngIf="hasAutofill" [loginUris]="cipher.login.uris">
diff --git a/libs/vault/src/cipher-view/login-credentials/login-credentials-view.component.html b/libs/vault/src/cipher-view/login-credentials/login-credentials-view.component.html
index 9ca983a5c9..3cd8431519 100644
--- a/libs/vault/src/cipher-view/login-credentials/login-credentials-view.component.html
+++ b/libs/vault/src/cipher-view/login-credentials/login-credentials-view.component.html
@@ -3,7 +3,7 @@
     <h2 bitTypography="h6">{{ "loginCredentials" | i18n }}</h2>
   </bit-section-header>
   <bit-card>
-    <bit-form-field [disableMargin]="!login.password && !login.totp">
+    <bit-form-field [disableMargin]="!cipher.login.password && !cipher.login.totp">
       <bit-label>
         {{ "username" | i18n }}
       </bit-label>
@@ -11,7 +11,7 @@
         readonly
         bitInput
         type="text"
-        [value]="login.username"
+        [value]="cipher.login.username"
         aria-readonly="true"
         data-testid="login-username"
       />
@@ -19,20 +19,20 @@
         bitIconButton="bwi-clone"
         bitSuffix
         type="button"
-        [appCopyClick]="login.username"
+        [appCopyClick]="cipher.login.username"
         [valueLabel]="'username' | i18n"
         showToast
         [appA11yTitle]="'copyValue' | i18n"
         data-testid="toggle-username"
       ></button>
     </bit-form-field>
-    <bit-form-field [disableMargin]="!login.totp">
+    <bit-form-field [disableMargin]="!cipher.login.totp">
       <bit-label>{{ "password" | i18n }}</bit-label>
       <input
         readonly
         bitInput
         type="password"
-        [value]="login.password"
+        [value]="cipher.login.password"
         aria-readonly="true"
         data-testid="login-password"
       />
@@ -45,7 +45,7 @@
         (toggledChange)="pwToggleValue($event)"
       ></button>
       <button
-        *ngIf="viewPassword && passwordRevealed"
+        *ngIf="cipher.viewPassword && passwordRevealed"
         bitIconButton="bwi-numbered-list"
         bitSuffix
         type="button"
@@ -58,7 +58,7 @@
         bitIconButton="bwi-clone"
         bitSuffix
         type="button"
-        [appCopyClick]="login.password"
+        [appCopyClick]="cipher.login.password"
         [valueLabel]="'password' | i18n"
         showToast
         [appA11yTitle]="'copyValue' | i18n"
@@ -66,9 +66,12 @@
       ></button>
     </bit-form-field>
     <ng-container *ngIf="showPasswordCount && passwordRevealed">
-      <bit-color-password [password]="login.password" [showCount]="true"></bit-color-password>
+      <bit-color-password
+        [password]="cipher.login.password"
+        [showCount]="true"
+      ></bit-color-password>
     </ng-container>
-    <bit-form-field disableMargin *ngIf="login.totp">
+    <bit-form-field disableMargin *ngIf="cipher.login.totp">
       <bit-label
         >{{ "verificationCodeTotp" | i18n }}
         <span
@@ -84,17 +87,25 @@
       <input
         readonly
         bitInput
-        type="text"
-        [value]="login.totp"
+        [type]="!(isPremium$ | async) ? 'password' : 'text'"
+        [value]="totpCopyCode || '*** ***'"
         aria-readonly="true"
         data-testid="login-totp"
         [disabled]="!(isPremium$ | async)"
       />
+      <button
+        *ngIf="isPremium$ | async"
+        bitTotpCountdown
+        [cipher]="cipher"
+        bitSuffix
+        type="button"
+        (sendCopyCode)="setTotpCopyCode($event)"
+      ></button>
       <button
         bitIconButton="bwi-clone"
         bitSuffix
         type="button"
-        [appCopyClick]="login.totp"
+        [appCopyClick]="totpCopyCode"
         [valueLabel]="'verificationCodeTotp' | i18n"
         showToast
         [appA11yTitle]="'copyValue' | i18n"
diff --git a/libs/vault/src/cipher-view/login-credentials/login-credentials-view.component.ts b/libs/vault/src/cipher-view/login-credentials/login-credentials-view.component.ts
index 15234556e9..880fbffc7b 100644
--- a/libs/vault/src/cipher-view/login-credentials/login-credentials-view.component.ts
+++ b/libs/vault/src/cipher-view/login-credentials/login-credentials-view.component.ts
@@ -5,7 +5,7 @@ import { Observable, shareReplay } from "rxjs";
 
 import { JslibModule } from "@bitwarden/angular/jslib.module";
 import { BillingAccountProfileStateService } from "@bitwarden/common/billing/abstractions";
-import { LoginView } from "@bitwarden/common/vault/models/view/login.view";
+import { CipherView } from "@bitwarden/common/vault/models/view/cipher.view";
 import {
   CardComponent,
   FormFieldModule,
@@ -17,6 +17,8 @@ import {
   ColorPasswordModule,
 } from "@bitwarden/components";
 
+import { BitTotpCountdownComponent } from "../../components/totp-countdown/totp-countdown.component";
+
 @Component({
   selector: "app-login-credentials-view",
   templateUrl: "login-credentials-view.component.html",
@@ -32,17 +34,19 @@ import {
     IconButtonModule,
     BadgeModule,
     ColorPasswordModule,
+    BitTotpCountdownComponent,
   ],
 })
 export class LoginCredentialsViewComponent {
-  @Input() login: LoginView;
-  @Input() viewPassword: boolean;
+  @Input() cipher: CipherView;
+
   isPremium$: Observable<boolean> =
     this.billingAccountProfileStateService.hasPremiumFromAnySource$.pipe(
       shareReplay({ refCount: true, bufferSize: 1 }),
     );
   showPasswordCount: boolean = false;
   passwordRevealed: boolean = false;
+  totpCopyCode: string;
 
   constructor(
     private billingAccountProfileStateService: BillingAccountProfileStateService,
@@ -60,4 +64,8 @@ export class LoginCredentialsViewComponent {
   togglePasswordCount() {
     this.showPasswordCount = !this.showPasswordCount;
   }
+
+  setTotpCopyCode(e: any) {
+    this.totpCopyCode = e;
+  }
 }
diff --git a/libs/vault/src/components/totp-countdown/totp-countdown.component.html b/libs/vault/src/components/totp-countdown/totp-countdown.component.html
new file mode 100644
index 0000000000..ba4ff04a09
--- /dev/null
+++ b/libs/vault/src/components/totp-countdown/totp-countdown.component.html
@@ -0,0 +1,34 @@
+<div class="tw-flex tw-items-center tw-justify-center totp-v2">
+  <span class="tw-relative tw-flex tw-justify-center tw-items-center" aria-hidden="true">
+    <span
+      class="tw-absolute"
+      [ngClass]="{ 'tw-text-main': !totpLow, 'tw-text-danger': totpLow }"
+      bitTypography="helper"
+      >{{ totpSec }}</span
+    >
+    <svg class="tw-w-7 tw-h-7" transform="rotate(-90)">
+      <g>
+        <circle
+          class="tw-fill-none"
+          [ngClass]="{ 'tw-stroke-text-main': !totpLow, 'tw-stroke-danger-600': totpLow }"
+          r="9.5"
+          cy="11.8"
+          cx="11.8"
+          stroke-width="2"
+          stroke-dasharray="60"
+          [ngStyle]="{ 'stroke-dashoffset.px': totpDash }"
+        ></circle>
+        <circle
+          class="tw-fill-none"
+          [ngClass]="{ 'tw-stroke-text-main': !totpLow, 'tw-stroke-danger-600': totpLow }"
+          r="11"
+          cy="11.8"
+          cx="11.8"
+          stroke-width="1"
+          stroke-dasharray="71"
+          stroke-dashoffset="0"
+        ></circle>
+      </g>
+    </svg>
+  </span>
+</div>
diff --git a/libs/vault/src/components/totp-countdown/totp-countdown.component.ts b/libs/vault/src/components/totp-countdown/totp-countdown.component.ts
new file mode 100644
index 0000000000..aeb7d30654
--- /dev/null
+++ b/libs/vault/src/components/totp-countdown/totp-countdown.component.ts
@@ -0,0 +1,80 @@
+import { CommonModule } from "@angular/common";
+import { Component, EventEmitter, Input, OnInit, Output } from "@angular/core";
+
+import { TotpService } from "@bitwarden/common/vault/abstractions/totp.service";
+import { CipherView } from "@bitwarden/common/vault/models/view/cipher.view";
+import { TypographyModule } from "@bitwarden/components";
+
+@Component({
+  selector: "button[bitTotpCountdown]:not(button[bitButton])",
+  templateUrl: "totp-countdown.component.html",
+  standalone: true,
+  imports: [CommonModule, TypographyModule],
+})
+export class BitTotpCountdownComponent implements OnInit {
+  @Input() cipher: CipherView;
+  @Output() sendCopyCode = new EventEmitter();
+
+  totpCode: string;
+  totpCodeFormatted: string;
+  totpDash: number;
+  totpSec: number;
+  totpLow: boolean;
+  private totpInterval: any;
+
+  constructor(protected totpService: TotpService) {}
+
+  async ngOnInit() {
+    await this.totpUpdateCode();
+    const interval = this.totpService.getTimeInterval(this.cipher.login.totp);
+    await this.totpTick(interval);
+
+    this.totpInterval = setInterval(async () => {
+      await this.totpTick(interval);
+    }, 1000);
+  }
+
+  private async totpUpdateCode() {
+    if (this.cipher.login.totp == null) {
+      this.clearTotp();
+      return;
+    }
+
+    this.totpCode = await this.totpService.getCode(this.cipher.login.totp);
+    if (this.totpCode != null) {
+      if (this.totpCode.length > 4) {
+        this.totpCodeFormatted = this.formatTotpCode();
+        this.sendCopyCode.emit(this.totpCodeFormatted);
+      } else {
+        this.totpCodeFormatted = this.totpCode;
+      }
+    } else {
+      this.totpCodeFormatted = null;
+      this.sendCopyCode.emit(this.totpCodeFormatted);
+      this.clearTotp();
+    }
+  }
+
+  private async totpTick(intervalSeconds: number) {
+    const epoch = Math.round(new Date().getTime() / 1000.0);
+    const mod = epoch % intervalSeconds;
+
+    this.totpSec = intervalSeconds - mod;
+    this.totpDash = +(Math.round(((60 / intervalSeconds) * mod + "e+2") as any) + "e-2");
+    this.totpLow = this.totpSec <= 7;
+    if (mod === 0) {
+      await this.totpUpdateCode();
+    }
+  }
+
+  private formatTotpCode(): string {
+    const half = Math.floor(this.totpCode.length / 2);
+    return this.totpCode.substring(0, half) + " " + this.totpCode.substring(half);
+  }
+
+  private clearTotp() {
+    if (this.totpInterval) {
+      clearInterval(this.totpInterval);
+    }
+  }
+}