85 lines
2.2 KiB
Go
85 lines
2.2 KiB
Go
// SPDX-License-Identifier: Apache-2.0
|
|
// Copyright © 2022 Roberto Hidalgo <nidito@un.rob.mx>
|
|
package auth
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
"git.rob.mx/nidito/puerta/internal/constants"
|
|
"git.rob.mx/nidito/puerta/internal/errors"
|
|
"git.rob.mx/nidito/puerta/internal/user"
|
|
"github.com/alexedwards/scs/v2"
|
|
"github.com/go-webauthn/webauthn/webauthn"
|
|
"github.com/julienschmidt/httprouter"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/upper/db/v4"
|
|
)
|
|
|
|
var _db db.Session
|
|
var _wan *webauthn.WebAuthn
|
|
var _sess *scs.SessionManager
|
|
|
|
func Initialize(wan *webauthn.WebAuthn, db db.Session) {
|
|
sessionManager := scs.New()
|
|
sessionManager.Lifetime = 5 * time.Minute
|
|
_db = db
|
|
_wan = wan
|
|
}
|
|
|
|
func Route(router http.Handler) http.Handler {
|
|
return _sess.LoadAndSave(router)
|
|
}
|
|
|
|
func requestAuth(w http.ResponseWriter, status int) {
|
|
http.Error(w, http.StatusText(status), status)
|
|
}
|
|
|
|
func LoginHandler(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
|
err := req.ParseForm()
|
|
if err != nil {
|
|
http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
username := req.FormValue("user")
|
|
password := req.FormValue("password")
|
|
|
|
user := &user.User{}
|
|
if err := _db.Get(user, db.Cond{"name": username}); err != nil {
|
|
err := &errors.InvalidCredentials{Status: http.StatusForbidden, Reason: fmt.Sprintf("User not found for name: %s (%s)", username, err)}
|
|
err.Log()
|
|
http.Error(w, err.Error(), err.Code())
|
|
return
|
|
}
|
|
|
|
if err := user.Login(password); err != nil {
|
|
code := http.StatusBadRequest
|
|
status := http.StatusText(code)
|
|
if err, ok := err.(errors.InvalidCredentials); ok {
|
|
code = err.Code()
|
|
status = err.Error()
|
|
err.Log()
|
|
}
|
|
http.Error(w, status, code)
|
|
return
|
|
}
|
|
|
|
sess, err := NewSession(user, _db.Collection("session"))
|
|
if err != nil {
|
|
http.Error(w, fmt.Sprintf("Could not create a session: %s", err), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.Header().Add("Set-Cookie", fmt.Sprintf("%s=%s; Max-Age=%d; Path=/;", constants.ContextCookieName, sess.Token, user.TTL.Seconds()))
|
|
|
|
logrus.Infof("Created session for %s", user.Name)
|
|
|
|
if req.FormValue("async") == "true" {
|
|
w.Write([]byte(user.Greeting))
|
|
} else {
|
|
http.Redirect(w, req, "/", http.StatusSeeOther)
|
|
}
|
|
}
|