alazyreader/alpha
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add basic login handling

Browse Source
This commit is contained in:
David 2022-05-29 19:47:13 -04:00
parent 9ed8108ed7
commit 5ae38a9fb4
3 changed files with 108 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
admin_templates/admin_templates.go Normal file
View File

@ -0,0 +1,13 @@
package admintemplates
var LoginPage string = `
<html>
<body>
<form action="/login" method="post">
<input name="username"/>
<input type="password" name="password"/>
<input type="submit"/>
</form>
</body>
<html>
`

105
main.go
View File

@ -10,6 +10,8 @@ import (
"strconv" "strconv"
"strings" "strings"
"time" "time"
admintemplates "git.yetaga.in/alazyreader/alpha/admin_templates"
) )
type SessionProvider interface { type SessionProvider interface {
@ -23,9 +25,15 @@ type PageProvider interface {
Save(key string, page *Page) error Save(key string, page *Page) error
} }
type UserProvider interface {
Authenticate(name string, password string) (User, error)
}
type RootHandler struct { type RootHandler struct {
Sessions SessionProvider Sessions SessionProvider
Pages PageProvider Pages PageProvider
Users UserProvider
Admin *AdminHandler
StaticDir string StaticDir string
} }
@ -37,13 +45,58 @@ func (h *AdminHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("admin route")) w.Write([]byte("admin route"))
} }
func (h *RootHandler) LoginHandler(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "GET":
// serve login form
w.Write([]byte(admintemplates.LoginPage))
case "POST":
// handle login request
r.ParseForm()
username := r.PostForm.Get("username")
password := r.PostForm.Get("password")
if username == "" {
h.ErrorHandle(http.StatusBadRequest, w)
return
}
user, err := h.Users.Authenticate(username, password)
if err != nil {
h.ErrorHandle(http.StatusForbidden, w)
return
}
sessionKey, err := h.Sessions.Create(user, time.Duration(time.Hour*24))
if err != nil {
h.ErrorHandle(http.StatusInternalServerError, w)
return
}
w.Header().Add("Set-Cookie", "alphasess="+sessionKey)
http.Redirect(w, r, "/admin", http.StatusFound)
default:
// anything else doesn't do anything
h.ErrorHandle(http.StatusMethodNotAllowed, w)
}
}
func (h *RootHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { func (h *RootHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if strings.HasPrefix(r.URL.Path, "/admin") { // handle login route
(&AdminHandler{ if r.URL.Path == "/login" {
Sessions: h.Sessions, h.LoginHandler(w, r)
}).ServeHTTP(w, r)
return return
} }
// hand off to admin router
if strings.HasPrefix(r.URL.Path, "/admin") {
if user, ok := h.IsLoggedIn(r); !ok {
http.Redirect(w, r, "/login", http.StatusFound)
return
} else if HasRole(user.Roles, "admin") {
h.Admin.ServeHTTP(w, r)
return
}
h.ErrorHandle(http.StatusForbidden, w)
return
}
// attempt to serve from the managed pages // attempt to serve from the managed pages
page, err := h.Pages.Page(r.URL.Path) page, err := h.Pages.Page(r.URL.Path)
if err == nil { if err == nil {
@ -53,19 +106,19 @@ func (h *RootHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
// fall back to serving out of the static directory, but: // fall back to serving out of the static directory, but:
// 1. prevent the generated indexes from rendering // 1. prevent the generated indexes from rendering
if strings.HasSuffix(r.URL.Path, "/") { if strings.HasSuffix(r.URL.Path, "/") {
h.ErrorHandle(404, w) h.ErrorHandle(http.StatusNotFound, w)
return return
} }
// 2. prevent hidden paths from rendering // 2. prevent hidden paths from rendering
for _, seg := range strings.Split(r.URL.Path, "/") { for _, seg := range strings.Split(r.URL.Path, "/") {
if strings.HasPrefix(seg, ".") { if strings.HasPrefix(seg, ".") {
h.ErrorHandle(404, w) h.ErrorHandle(http.StatusNotFound, w)
return return
} }
} }
// 3. catch files that would 404 and serve our own 404 page // 3. catch files that would 404 and serve our own 404 page
if !staticFileExists(filepath.Join(h.StaticDir, r.URL.Path)) { if !staticFileExists(filepath.Join(h.StaticDir, r.URL.Path)) {
h.ErrorHandle(404, w) h.ErrorHandle(http.StatusNotFound, w)
return return
} }
// finally, use the built-in fileserver to serve // finally, use the built-in fileserver to serve
@ -85,10 +138,28 @@ func (h *RootHandler) ErrorHandle(status int, w http.ResponseWriter) {
} }
w.Header().Set("Content-Type", "text/plain; charset=utf-8") w.Header().Set("Content-Type", "text/plain; charset=utf-8")
w.Header().Set("X-Content-Type-Options", "nosniff") w.Header().Set("X-Content-Type-Options", "nosniff")
w.WriteHeader(404) w.WriteHeader(http.StatusNotFound)
fmt.Fprintf(w, "%d\n", status) fmt.Fprintf(w, "%d\n", status)
} }
func (h *RootHandler) IsLoggedIn(r *http.Request) (User, bool) {
cookie, err := r.Cookie("alphasess")
if err != nil {
return User{}, false
}
user, err := h.Sessions.Get(cookie.Value)
return user, err == nil
}
func HasRole(roles []string, role string) bool {
for _, r := range roles {
if r == role {
return true
}
}
return false
}
func staticFileExists(name string) bool { func staticFileExists(name string) bool {
f, err := os.Open(name) f, err := os.Open(name)
if err != nil { if err != nil {
@ -100,18 +171,28 @@ func staticFileExists(name string) bool {
} }
func main() { func main() {
sessions := &Sessions{}
pages := &Index{}
users := &Userstore{}
users.Create("testuser", []string{"admin"}, "password")
handler := &RootHandler{ handler := &RootHandler{
Sessions: &Sessions{}, Sessions: sessions,
Pages: &Index{}, Pages: pages,
Users: users,
Admin: &AdminHandler{
Sessions: sessions,
},
StaticDir: "./static", StaticDir: "./static",
} }
handler.Pages.Save("foo", &Page{ pages.Save("foo", &Page{
Contents: []byte("foobar"), Contents: []byte("foobar"),
}) })
handler.Pages.Save("index", &Page{ pages.Save("index", &Page{
Contents: []byte("root"), Contents: []byte("root"),
}) })
log.Printf("serving on http://localhost:8080")
err := http.ListenAndServe(":8080", handler) err := http.ListenAndServe(":8080", handler)
log.Fatalf("server error: %v", err) log.Fatalf("server error: %v", err)
} }

4
session.go
View File

@ -40,7 +40,7 @@ func (s *Sessions) Get(key string) (User, error) {
if !ok { if !ok {
return User{}, ErrInvalidSession return User{}, ErrInvalidSession
} }
if sess.expr.After(time.Now()) { if sess.expr.Before(time.Now()) {
delete(s.sessions, key) delete(s.sessions, key)
return User{}, ErrInvalidSession return User{}, ErrInvalidSession
} }
@ -56,7 +56,7 @@ func (s *Sessions) Refresh(key string, user User, expr time.Duration) error {
if !ok { if !ok {
return ErrInvalidSession return ErrInvalidSession
} }
if sess.expr.After(time.Now()) { if sess.expr.Before(time.Now()) {
delete(s.sessions, key) delete(s.sessions, key)
return ErrInvalidSession return ErrInvalidSession
} }