dkforest

commit 58c1e84d6540f87bdfd28fdf9d8e13474d821b28
parent b0577b489341e3b3dc60d907c8954062b8844145
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Fri, 23 Dec 2022 00:37:51 -0800

cleanup

Diffstat:
M
pkg/web/handlers/handlers.go
 | 
11
+++++++++--

1 file changed, 9 insertions(+), 2 deletions(-)
diff --git a/pkg/web/handlers/handlers.go b/pkg/web/handlers/handlers.go
@@ -3370,6 +3370,13 @@ func SettingsAgeHandler(c echo.Context) error {
 	return c.Render(http.StatusOK, "settings.age", data)
 }
 
+func generateTokenMsg(token string) string {
+	msg := "The required code is below the line.\n"
+	msg += "----------------------------------------------------------------------------------\n"
+	msg += token + "\n"
+	return msg
+}
+
 func generateAgeEncryptedTokenMessage(userID database.UserID, pkey string) (string, error) {
 	token := utils.GenerateToken32()
 	ageTokenCache.SetD(userID, ValueTokenCache{Value: token, PKey: pkey})
@@ -3382,7 +3389,7 @@ func generateAgeEncryptedTokenMessage(userID database.UserID, pkey string) (stri
 	out := &bytes.Buffer{}
 	aw := armor1.NewWriter(out)
 	w, err := age.Encrypt(aw, recipient)
-	msg := "The required code is below the line.\n----------------------------------------------------------------------------------\n" + token + "\n"
+	msg := generateTokenMsg(token)
 	if _, err := io.WriteString(w, msg); err != nil {
 		logrus.Error(err)
 		w.Close()
@@ -3398,7 +3405,7 @@ func generateAgeEncryptedTokenMessage(userID database.UserID, pkey string) (stri
 func generatePgpEncryptedTokenMessage(userID database.UserID, pkey string) (string, error) {
 	token := utils.GenerateToken32()
 	pgpTokenCache.SetD(userID, ValueTokenCache{Value: token, PKey: pkey})
-	msg := "The required code is below the line.\n----------------------------------------------------------------------------------\n" + token + "\n"
+	msg := generateTokenMsg(token)
 	return utils.GeneratePgpEncryptedMessage(pkey, msg)
 }