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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
package web
import (
"fmt"
"html/template"
"log"
"net/http"
"time"
"tildegit.org/andinus/perseus/auth"
"tildegit.org/andinus/perseus/auth/token"
"tildegit.org/andinus/perseus/core"
"tildegit.org/andinus/perseus/storage/sqlite3"
)
// HandleLogin handles /login pages.
func HandleLogin(w http.ResponseWriter, r *http.Request, db *sqlite3.DB) {
p := Page{Version: core.Version()}
error := []string{}
success := []string{}
switch r.Method {
case http.MethodGet:
t, _ := template.ParseFiles("web/login.html")
t.Execute(w, p)
case http.MethodPost:
if err := r.ParseForm(); err != nil {
log.Printf("web/login.go: 400 Bad Request :: %s", err.Error())
http.Error(w, "400 Bad Request", http.StatusBadRequest)
return
}
// Get form values
uInfo := make(map[string]string)
uInfo["username"] = r.FormValue("username")
uInfo["password"] = r.FormValue("password")
// Perform authentication
err := auth.Login(db, uInfo)
if err != nil {
log.Printf("web/login.go: %s :: %s :: %s",
"login failed",
uInfo["username"],
err.Error())
error = append(error,
fmt.Sprintf("Login failed"))
p.Error = error
} else {
success = append(success,
fmt.Sprintf("Login successful"))
p.Success = success
// Set token if login was successful.
token, err := token.AddToken(db, uInfo)
if err != nil {
log.Printf("web/login.go: %s :: %s :: %s",
"token generation failed",
uInfo["username"],
err.Error())
error = append(error,
fmt.Sprintf("Token generation failed"))
}
// If token was generated then ask browser to
// set it as cookie.
expiration := time.Now().Add(1 * 24 * time.Hour)
cookie := http.Cookie{Name: "token", Value: token, Expires: expiration}
http.SetCookie(w, &cookie)
}
t, _ := template.ParseFiles("web/login.html")
t.Execute(w, p)
default:
w.WriteHeader(http.StatusMethodNotAllowed)
log.Printf("web/login.go: %v not allowed on %v", r.Method, r.URL)
}
}
|