handle 2fa code submit via JS (#6)

internal/misc/error.go

@@ -1,7 +1,10 @@
 package misc
 
 import (
+	"encoding/json"
 	"errors"
+
+	"code.lila.network/adoralaura/go-urlsh/models"
 	"github.com/gofiber/fiber/v2"
 )
 
@@ -30,6 +33,11 @@ func New400Error() error {
 	return fiber.NewError(fiber.StatusBadRequest, "400 Bad Request")
 }
 
+func New400WithMessageError(msg string) error {
+	body, _ := json.Marshal(models.HttpErrorBody{Message: msg})
+	return fiber.NewError(fiber.StatusBadRequest, string(body[:]))
+}
+
 func New401Error() error {
 	return fiber.NewError(fiber.StatusUnauthorized, "401 Unauthorized")
 }

internal/web/multifactor-login.go

@@ -21,14 +21,13 @@ func HandleAdminLoginMFAPost(c *fiber.Ctx) error {
 		return misc.New400Error()
 	}
 
-	var token models.TokenRequest
+	token := new(models.TokenRequest)
 	var istotp bool
 
-	if err := c.BodyParser(&token); err != nil {
+	if err := c.BodyParser(token); err != nil {
 		// TODO: Debug logging
 		return misc.New400Error()
 	}
-
 	if len(token.Token) == 6 {
 		istotp = true
 	} else if len(token.Token) == 8 {
@@ -42,7 +41,7 @@ func HandleAdminLoginMFAPost(c *fiber.Ctx) error {
 	if err != nil {
 		log.Println(err)
 		// TODO: Debug logging
-		return misc.New500Error()
+		return misc.New400Error()
 	}
 
 	// check token/scratch validity
@@ -57,11 +56,10 @@ func HandleAdminLoginMFAPost(c *fiber.Ctx) error {
 		err = misc.SetLoginCookie(c, user, constants.LoginCookieExpiryDuration)
 		if err != nil {
 			log.Printf("[HandleAdminLoginPost] Error setting cookie: %q\n", err)
-			return fiber.NewError(fiber.StatusInternalServerError, "500 Internal Server Error")
+			return misc.New500Error()
 		}
 
-		c.Status(fiber.StatusSeeOther)
+		c.Status(fiber.StatusOK)
-		c.Location("/admin/")
 		return nil
 	}
 

models/error.go

@@ -0,0 +1,5 @@
+package models
+
+type HttpErrorBody struct {
+	Message string `json:"message"`
+}

views/login-multifactor.tmpl

@@ -7,8 +7,11 @@
   <title>Multi Factor Authentication - go-urlsh</title>
   <meta name='viewport' content='width=device-width, initial-scale=1'>
   <link rel="stylesheet" href="/admin/pico.min.css">
+  <script src="/admin/main.js" defer></script>
   <style>
-
+    .hidden {
+      display: none;
+    }
   </style>
 </head>
 
@@ -17,13 +20,16 @@
     <article class="grid">
       <div>
         <hgroup>
-          <h1 style='--color: var(--h1-color); font-weight: var(--font-weight); font-size: var(--font-size); font-family: var(--font-family);'>Multi Factor Authentication</h1>
+          <h1
+            style='--color: var(--h1-color); font-weight: var(--font-weight); font-size: var(--font-size); font-family: var(--font-family);'>
+            Multi Factor Authentication</h1>
+          
         </hgroup>
-        <form method="post" action="/admin/login/multifactor"  style="margin-bottom: 0;" autocomplete="off">
+        <form action="javascript:HandleMFALoginTokenPost()" style="margin-bottom: 0;" autocomplete="off">
           <label for="token">Authentication Code (Format: 123456 or a1b2c3d4 for recovery codes)</label>
           <input type="text" name="token" id="token" autocomplete="one-time-code">
-
+          <p id="error-message" class="hidden" style="color: red;">Error Message</p>
-          <input type="submit" value="Submit" class="contrast">
+          <input id="submit" type="submit" value="Submit" class="contrast">
         </form>
       </div>
     </article>

web/main.js

@@ -82,8 +82,7 @@ async function LinkAction(action){
 async function HandleLinkFieldChange() {
     console.log("HandleChange")
     let buttonactive = true
-    if (document.getElementById("link").value === "")
+    if (document.getElementById("link").value === "") {
-    {
         buttonactive = false
     }
     document.getElementById("submit").active = buttonactive
@@ -127,6 +126,7 @@ async function HandleMFASetupTokenSubmit() {
     let response = await fetch("/admin/account/mfasetup", {
         credentials: "include",
         body: JSON.stringify(body),
+        headers: { "Content-Type": "application/json", },
         mode: "same-origin",
         method: "POST"
     });
@@ -145,6 +145,35 @@ async function HandleMFASetupTokenSubmit() {
     }
 }
 
+async function HandleMFALoginTokenPost() {
+    document.getElementById("submit").disabled = true;
+    document.getElementById("token").disabled = true;
+    let token = document.getElementById("token").value
+    let body = {
+        "token": token
+    }
+
+    let response = await fetch("/admin/login/multifactor", {
+        credentials: "include",
+        headers: { "Content-Type": "application/json", },
+        body: JSON.stringify(body),
+        mode: "same-origin",
+        method: "POST"
+    });
+
+    if (response.ok) {
+        document.location = "/admin/"
+    } else {
+        document.getElementById("submit").disabled = false;
+        document.getElementById("token").disabled = false;
+        document.getElementById('error-message').innerHTML = "Two Factor Authentication failed. Please try again."
+        document.getElementById('error-message').setAttribute("class", "")
+        document.getElementById('token').value = ""
+        document.getElementById('token').setAttribute("aria-invalid", "true")
+
+    }
+}
+
 function HandleModalClose(redir) {
     document.getElementById('dialog-success').close();
     document.getElementById('dialog-error').close();