From ad7524f54410aca24018f80ed3effe0bd8d69f95 Mon Sep 17 00:00:00 2001
From: Randall Winkhart <idgr@tutanota.com>
Date: Wed, 20 Mar 2024 18:10:35 -0400
Subject: [PATCH] Split TOTP key generation into its own function

---
 src/offline/copy.go | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/src/offline/copy.go b/src/offline/copy.go
index 0019081..3bb6fe7 100644
--- a/src/offline/copy.go
+++ b/src/offline/copy.go
@@ -27,18 +27,8 @@ func CopyArgument(targetLocation string, field int, executableName string) {
 			if field != 5 { // TODO Update field after removed from notes (breaking sshyp entry compatibility)
 				copySubject = decryptedEntry[field]
 			} else { // TOTP mode
-				var err error
-				var copied bool // track whether the TOTP code has been copied to the clipboard for the first time
-				for {           // keep field copied to clipboard, refresh on 30-second intervals
-					copySubject, err = totp.GenerateCode(decryptedEntry[5], time.Now())
-					if err != nil {
-						fmt.Println(AnsiError + "Error generating TOTP code" + AnsiReset)
-						os.Exit(1)
-					}
-					copyField(copySubject, "")
-					if !copied {
-						copied = true
-					}
+				for { // keep field copied to clipboard, refresh on 30-second intervals
+					copyField(genTOTP(decryptedEntry[5]), "")
 					// sleep until next 30-second interval
 					time.Sleep(time.Duration(30-(time.Now().Second()%30)) * time.Second)
 				}
@@ -64,3 +54,13 @@ func ClipClearArgument() {
 		os.Exit(0)
 	}
 }
+
+// genTOTP generates a TOTP token from a secret
+func genTOTP(secret string) string {
+	totpToken, err := totp.GenerateCode(secret, time.Now())
+	if err != nil {
+		fmt.Println(AnsiError + "Error generating TOTP code" + AnsiReset)
+		os.Exit(1)
+	}
+	return totpToken
+}