Implement Optional Secret Injection

.gitignore

@@ -1,4 +1,4 @@
 *.secret
 *.png
 *.env
-*.db
+**/*.db

handlers.go

@@ -19,7 +19,7 @@ func HandleRegisterApp(w http.ResponseWriter, r *http.Request) {
 		http.Error(w, "Some of the required parameters is missing", http.StatusBadRequest)
 	}
 
-	totp.RegisterApp(requestBody.AppName, requestBody.AccountName)
+	totp.RegisterApp(requestBody.AppName, requestBody.AccountName, requestBody.Secret)
 	response := typings.ControllerResponse{Data: "App Registered"}
 	w.Header().Set("Content-Type", "application/json")
 

totp/totp.go

@@ -17,18 +17,24 @@ func GetSecret(appName string) (string, error) {
 	return strings.TrimSpace(secret), nil
 }
 
-func RegisterApp(appName, accountName string) {
-	key, err := totp.Generate(totp.GenerateOpts{
-		Issuer:      appName,
-		AccountName: accountName,
-	})
-	if err != nil {
-		log.Fatal("Error generating key:", err)
+func RegisterApp(appName, accountName, secret string) {
+	var secretKey string
+	if secret == "" {
+		key, err := totp.Generate(totp.GenerateOpts{
+			Issuer:      appName,
+			AccountName: accountName,
+		})
+		if err != nil {
+			log.Fatal("Error generating key:", err)
+		}
+		secretKey = key.Secret()
+	} else {
+		secretKey = secret
 	}
 
 	// Save secret to database
 	db := dbhandler.InitDB()
-	dbhandler.StoreSecret(db, accountName, appName, key.Secret())
+	dbhandler.StoreSecret(db, accountName, appName, secretKey)
 
 	fmt.Printf("App %s registered.\n", appName)
 }

typings/types.go

@@ -3,6 +3,7 @@ package typings
 type RegisterRequest struct {
 	AppName     string `json:"appName"`
 	AccountName string `json:"accountName"`
+	Secret      string `json:"secret"`
 }
 
 type GenerateRequest struct {