forked from urandom/readeef
/
login.go
68 lines (59 loc) · 1.56 KB
/
login.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package readeef
import (
"net/http"
"github.com/urandom/webfw"
"github.com/urandom/webfw/context"
"github.com/urandom/webfw/renderer"
)
type Login struct {
webfw.BaseController
}
func NewLogin(pattern string) Login {
return Login{webfw.NewBaseController(pattern, webfw.MethodGet|webfw.MethodPost, "auth-login")}
}
func (con Login) Handler(c context.Context) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
l := webfw.GetLogger(c)
sess := webfw.GetSession(c, r)
data := renderer.RenderData{}
if r.Method == "GET" {
if v, ok := sess.Flash("form-error"); ok {
data["form-error"] = v
}
} else {
if err := r.ParseForm(); err != nil {
l.Fatal(err)
}
username := r.Form.Get("username")
password := r.Form.Get("password")
db := GetDB(c)
formError := false
if u, err := db.GetUser(username); err != nil {
sess.SetFlash("form-error", "login-incorrect")
formError = true
} else if !u.Authenticate(password) {
sess.SetFlash("form-error", "login-incorrect")
formError = true
} else {
sess.Set(authkey, u)
sess.Set(namekey, u.Login)
}
if formError {
http.Redirect(w, r, r.URL.String(), http.StatusTemporaryRedirect)
} else {
var returnPath string
if v, ok := sess.Flash("return-to"); ok {
returnPath = v.(string)
} else {
returnPath = webfw.GetDispatcher(c).Pattern
}
http.Redirect(w, r, returnPath, http.StatusTemporaryRedirect)
}
return
}
err := webfw.GetRenderCtx(c, r)(w, data, "login.tmpl")
if err != nil {
l.Print(err)
}
}
}