94 lines
3.4 KiB
Go
94 lines
3.4 KiB
Go
package server
|
|
|
|
import (
|
|
"html/template"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/dustin/go-humanize"
|
|
"github.com/gorilla/securecookie"
|
|
"github.com/wader/gormstore"
|
|
)
|
|
|
|
var (
|
|
tmplFuncMap = template.FuncMap{
|
|
"humandate": humanize.Time,
|
|
}
|
|
)
|
|
|
|
func extendFrom(tmpl *template.Template, key string) *template.Template {
|
|
strT, ok := assets.String(key)
|
|
if !ok {
|
|
log.Fatalf("error when reading %q from assets", key)
|
|
}
|
|
if tmpl == nil {
|
|
tmpl = template.New("layout").Funcs(tmplFuncMap)
|
|
} else {
|
|
tmpl = template.Must(tmpl.Clone())
|
|
}
|
|
newT, err := tmpl.Parse(strT)
|
|
if err != nil {
|
|
log.Fatalf("error when parsing template: %v", err)
|
|
}
|
|
return newT
|
|
}
|
|
|
|
func (s *Server) SetupAdmin() {
|
|
sessionKey := []byte(s.GetSetting("session_key"))
|
|
if len(sessionKey) == 0 {
|
|
sessionKey = securecookie.GenerateRandomKey(32)
|
|
s.SetSetting("session_key", string(sessionKey))
|
|
}
|
|
s.SessDB = gormstore.New(s.DB, sessionKey)
|
|
go s.SessDB.PeriodicCleanup(time.Hour, nil)
|
|
//
|
|
layoutT := extendFrom(nil, "/templates/layout.tmpl")
|
|
userT := extendFrom(layoutT, "/templates/user.tmpl")
|
|
s.Templates = map[string]*template.Template{
|
|
"login": extendFrom(layoutT, "/templates/pages/login.tmpl"),
|
|
"home": extendFrom(userT, "/templates/pages/home.tmpl"),
|
|
"change_own_password": extendFrom(userT, "/templates/pages/change_own_password.tmpl"),
|
|
"change_password": extendFrom(userT, "/templates/pages/change_password.tmpl"),
|
|
"delete_user": extendFrom(userT, "/templates/pages/delete_user.tmpl"),
|
|
"create_user": extendFrom(userT, "/templates/pages/create_user.tmpl"),
|
|
"update_lastfm_api_key": extendFrom(userT, "/templates/pages/update_lastfm_api_key.tmpl"),
|
|
}
|
|
//
|
|
withPublicWare := newChain(
|
|
s.WithLogging,
|
|
s.WithSession,
|
|
)
|
|
withUserWare := newChain(
|
|
withPublicWare,
|
|
s.WithUserSession,
|
|
)
|
|
withAdminWare := newChain(
|
|
withUserWare,
|
|
s.WithAdminSession,
|
|
)
|
|
// begin static server
|
|
s.mux.Handle("/admin/static/", http.StripPrefix("/admin",
|
|
http.FileServer(assets),
|
|
))
|
|
// begin public routes (creates new session)
|
|
s.mux.HandleFunc("/admin/login", withPublicWare(s.ServeLogin))
|
|
s.mux.HandleFunc("/admin/login_do", withPublicWare(s.ServeLoginDo))
|
|
// begin user routes (if session is valid)
|
|
s.mux.HandleFunc("/admin/logout", withUserWare(s.ServeLogout))
|
|
s.mux.HandleFunc("/admin/home", withUserWare(s.ServeHome))
|
|
s.mux.HandleFunc("/admin/change_own_password", withUserWare(s.ServeChangeOwnPassword))
|
|
s.mux.HandleFunc("/admin/change_own_password_do", withUserWare(s.ServeChangeOwnPasswordDo))
|
|
s.mux.HandleFunc("/admin/link_lastfm_do", withUserWare(s.ServeLinkLastFMDo))
|
|
s.mux.HandleFunc("/admin/unlink_lastfm_do", withUserWare(s.ServeUnlinkLastFMDo))
|
|
// begin admin routes (if session is valid, and is admin)
|
|
s.mux.HandleFunc("/admin/change_password", withAdminWare(s.ServeChangePassword))
|
|
s.mux.HandleFunc("/admin/change_password_do", withAdminWare(s.ServeChangePasswordDo))
|
|
s.mux.HandleFunc("/admin/delete_user", withAdminWare(s.ServeDeleteUser))
|
|
s.mux.HandleFunc("/admin/delete_user_do", withAdminWare(s.ServeDeleteUserDo))
|
|
s.mux.HandleFunc("/admin/create_user", withAdminWare(s.ServeCreateUser))
|
|
s.mux.HandleFunc("/admin/create_user_do", withAdminWare(s.ServeCreateUserDo))
|
|
s.mux.HandleFunc("/admin/update_lastfm_api_key", withAdminWare(s.ServeUpdateLastFMAPIKey))
|
|
s.mux.HandleFunc("/admin/update_lastfm_api_key_do", withAdminWare(s.ServeUpdateLastFMAPIKeyDo))
|
|
}
|