fix enable two fa

Author: irfansaf

internal/database/app/user.go

@@ -264,3 +264,8 @@ func UploadStudentID(userID uuid.UUID, studentID string) error {
 	_, err := database.DB.Exec(context.Background(), "UPDATE users SET student_id_verification = $1 WHERE id = $2", studentID, userID)
 	return err
 }
+
+func SaveTwoFAInfo(userID uuid.UUID, secret string, image string) error {
+	_, err := database.DB.Exec(context.Background(), "UPDATE users SET twofa_image = $1, twofa_secret = $2 WHERE id = $3", image, secret, userID)
+	return err
+}

internal/services/user_services.go

@@ -153,7 +153,7 @@ func (us *UserService) EnableTwoFA(userID uuid.UUID) (string, string, error) {
 	user.TwoFASecret = &secretStr
 	user.TwoFAImage = &qr
 
-	err = app.UpdateUser(userID, user)
+	err = app.SaveTwoFAInfo(userID, secretStr, qr)
 	if err != nil {
 		return "", "", err
 	}