2020-11-14 13:03:06 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2021-04-12 17:40:43 +00:00
|
|
|
"io"
|
2020-11-14 13:03:06 +00:00
|
|
|
"log"
|
|
|
|
"net/http"
|
|
|
|
|
|
|
|
"github.com/bouncepaw/mycorrhiza/user"
|
2021-02-17 18:41:35 +00:00
|
|
|
"github.com/bouncepaw/mycorrhiza/util"
|
2021-02-23 14:25:07 +00:00
|
|
|
"github.com/bouncepaw/mycorrhiza/views"
|
2020-11-14 13:03:06 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
2021-04-12 17:40:43 +00:00
|
|
|
http.HandleFunc("/register", handlerRegister)
|
2020-11-14 13:03:06 +00:00
|
|
|
http.HandleFunc("/login", handlerLogin)
|
|
|
|
http.HandleFunc("/login-data", handlerLoginData)
|
|
|
|
http.HandleFunc("/logout", handlerLogout)
|
|
|
|
http.HandleFunc("/logout-confirm", handlerLogoutConfirm)
|
|
|
|
}
|
|
|
|
|
2021-04-12 17:40:43 +00:00
|
|
|
func handlerRegister(w http.ResponseWriter, rq *http.Request) {
|
|
|
|
log.Println(rq.URL)
|
2021-04-19 16:39:25 +00:00
|
|
|
if !util.UseRegistration {
|
2021-04-12 17:40:43 +00:00
|
|
|
w.WriteHeader(http.StatusForbidden)
|
|
|
|
}
|
|
|
|
if rq.Method == http.MethodGet {
|
|
|
|
io.WriteString(
|
|
|
|
w,
|
|
|
|
base(
|
|
|
|
"Register",
|
|
|
|
views.RegisterHTML(rq),
|
|
|
|
user.FromRequest(rq),
|
|
|
|
),
|
|
|
|
)
|
|
|
|
} else if rq.Method == http.MethodPost {
|
2021-04-19 16:39:25 +00:00
|
|
|
var (
|
|
|
|
username = rq.PostFormValue("username")
|
|
|
|
password = rq.PostFormValue("password")
|
|
|
|
err = user.Register(username, password)
|
|
|
|
)
|
|
|
|
if err != nil {
|
|
|
|
io.WriteString(w, err.Error())
|
|
|
|
} else {
|
|
|
|
user.LoginDataHTTP(w, rq, username, password)
|
|
|
|
http.Redirect(w, rq, "/"+rq.URL.RawQuery, http.StatusSeeOther)
|
|
|
|
}
|
2021-04-12 17:40:43 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-11-14 13:03:06 +00:00
|
|
|
func handlerLogout(w http.ResponseWriter, rq *http.Request) {
|
|
|
|
var (
|
|
|
|
u = user.FromRequest(rq)
|
|
|
|
can = u != nil
|
|
|
|
)
|
|
|
|
w.Header().Set("Content-Type", "text/html;charset=utf-8")
|
|
|
|
if can {
|
|
|
|
log.Println("User", u.Name, "tries to log out")
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
|
|
} else {
|
|
|
|
log.Println("Unknown user tries to log out")
|
|
|
|
w.WriteHeader(http.StatusForbidden)
|
|
|
|
}
|
2021-02-23 14:25:07 +00:00
|
|
|
w.Write([]byte(base("Logout?", views.LogoutHTML(can), u)))
|
2020-11-14 13:03:06 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func handlerLogoutConfirm(w http.ResponseWriter, rq *http.Request) {
|
|
|
|
user.LogoutFromRequest(w, rq)
|
|
|
|
http.Redirect(w, rq, "/", http.StatusSeeOther)
|
|
|
|
}
|
|
|
|
|
|
|
|
func handlerLoginData(w http.ResponseWriter, rq *http.Request) {
|
|
|
|
log.Println(rq.URL)
|
|
|
|
var (
|
2021-02-17 18:41:35 +00:00
|
|
|
username = util.CanonicalName(rq.PostFormValue("username"))
|
2020-11-14 13:03:06 +00:00
|
|
|
password = rq.PostFormValue("password")
|
|
|
|
err = user.LoginDataHTTP(w, rq, username, password)
|
|
|
|
)
|
|
|
|
if err != "" {
|
2021-02-23 14:25:07 +00:00
|
|
|
w.Write([]byte(base(err, views.LoginErrorHTML(err), user.EmptyUser())))
|
2020-11-14 13:03:06 +00:00
|
|
|
} else {
|
|
|
|
http.Redirect(w, rq, "/", http.StatusSeeOther)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func handlerLogin(w http.ResponseWriter, rq *http.Request) {
|
|
|
|
log.Println(rq.URL)
|
|
|
|
w.Header().Set("Content-Type", "text/html;charset=utf-8")
|
|
|
|
if user.AuthUsed {
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
|
|
} else {
|
|
|
|
w.WriteHeader(http.StatusForbidden)
|
|
|
|
}
|
2021-02-23 14:25:07 +00:00
|
|
|
w.Write([]byte(base("Login", views.LoginHTML(), user.EmptyUser())))
|
2020-11-14 13:03:06 +00:00
|
|
|
}
|