eg
This commit is contained in:
sentriz
@@ -1,162 +1,59 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"html/template"
|
||||
"flag"
|
||||
"log"
|
||||
"net/http"
|
||||
"time"
|
||||
"os"
|
||||
|
||||
"github.com/gobuffalo/packr/v2"
|
||||
"github.com/gorilla/securecookie"
|
||||
"github.com/jinzhu/gorm"
|
||||
_ "github.com/jinzhu/gorm/dialects/sqlite"
|
||||
"github.com/wader/gormstore"
|
||||
"github.com/peterbourgon/ff"
|
||||
|
||||
"github.com/sentriz/gonic/db"
|
||||
"github.com/sentriz/gonic/handler"
|
||||
"github.com/sentriz/gonic/server"
|
||||
)
|
||||
|
||||
type middleware func(next http.HandlerFunc) http.HandlerFunc
|
||||
|
||||
func newChain(wares ...middleware) middleware {
|
||||
return func(final http.HandlerFunc) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
last := final
|
||||
for i := len(wares) - 1; i >= 0; i-- {
|
||||
last = wares[i](last)
|
||||
}
|
||||
last(w, r)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func extendFromBox(tmpl *template.Template, box *packr.Box, key string) *template.Template {
|
||||
strT, err := box.FindString(key)
|
||||
if err != nil {
|
||||
log.Fatalf("error when reading template from box: %v", err)
|
||||
}
|
||||
if tmpl == nil {
|
||||
tmpl = template.New("layout")
|
||||
} else {
|
||||
tmpl = template.Must(tmpl.Clone())
|
||||
}
|
||||
newT, err := tmpl.Parse(strT)
|
||||
if err != nil {
|
||||
log.Fatalf("error when parsing template template: %v", err)
|
||||
}
|
||||
return newT
|
||||
}
|
||||
|
||||
func setSubsonicRoutes(cont handler.Controller, mux *http.ServeMux) {
|
||||
withWare := newChain(
|
||||
cont.WithLogging,
|
||||
cont.WithCORS,
|
||||
cont.WithValidSubsonicArgs,
|
||||
)
|
||||
// common
|
||||
mux.HandleFunc("/rest/download", withWare(cont.Stream))
|
||||
mux.HandleFunc("/rest/download.view", withWare(cont.Stream))
|
||||
mux.HandleFunc("/rest/stream", withWare(cont.Stream))
|
||||
mux.HandleFunc("/rest/stream.view", withWare(cont.Stream))
|
||||
mux.HandleFunc("/rest/getCoverArt", withWare(cont.GetCoverArt))
|
||||
mux.HandleFunc("/rest/getCoverArt.view", withWare(cont.GetCoverArt))
|
||||
mux.HandleFunc("/rest/getLicense", withWare(cont.GetLicence))
|
||||
mux.HandleFunc("/rest/getLicense.view", withWare(cont.GetLicence))
|
||||
mux.HandleFunc("/rest/ping", withWare(cont.Ping))
|
||||
mux.HandleFunc("/rest/ping.view", withWare(cont.Ping))
|
||||
mux.HandleFunc("/rest/scrobble", withWare(cont.Scrobble))
|
||||
mux.HandleFunc("/rest/scrobble.view", withWare(cont.Scrobble))
|
||||
mux.HandleFunc("/rest/getMusicFolders", withWare(cont.GetMusicFolders))
|
||||
mux.HandleFunc("/rest/getMusicFolders.view", withWare(cont.GetMusicFolders))
|
||||
// browse by tag
|
||||
mux.HandleFunc("/rest/getAlbum", withWare(cont.GetAlbum))
|
||||
mux.HandleFunc("/rest/getAlbum.view", withWare(cont.GetAlbum))
|
||||
mux.HandleFunc("/rest/getAlbumList2", withWare(cont.GetAlbumListTwo))
|
||||
mux.HandleFunc("/rest/getAlbumList2.view", withWare(cont.GetAlbumListTwo))
|
||||
mux.HandleFunc("/rest/getArtist", withWare(cont.GetArtist))
|
||||
mux.HandleFunc("/rest/getArtist.view", withWare(cont.GetArtist))
|
||||
mux.HandleFunc("/rest/getArtists", withWare(cont.GetArtists))
|
||||
mux.HandleFunc("/rest/getArtists.view", withWare(cont.GetArtists))
|
||||
// browse by folder
|
||||
mux.HandleFunc("/rest/getIndexes", withWare(cont.GetIndexes))
|
||||
mux.HandleFunc("/rest/getIndexes.view", withWare(cont.GetIndexes))
|
||||
mux.HandleFunc("/rest/getMusicDirectory", withWare(cont.GetMusicDirectory))
|
||||
mux.HandleFunc("/rest/getMusicDirectory.view", withWare(cont.GetMusicDirectory))
|
||||
mux.HandleFunc("/rest/getAlbumList", withWare(cont.GetAlbumList))
|
||||
mux.HandleFunc("/rest/getAlbumList.view", withWare(cont.GetAlbumList))
|
||||
}
|
||||
|
||||
func setAdminRoutes(cont handler.Controller, mux *http.ServeMux) {
|
||||
sessionKey := []byte(cont.GetSetting("session_key"))
|
||||
if len(sessionKey) == 0 {
|
||||
sessionKey = securecookie.GenerateRandomKey(32)
|
||||
cont.SetSetting("session_key", string(sessionKey))
|
||||
}
|
||||
// create gormstore (and cleanup) for backend sessions
|
||||
cont.SStore = gormstore.New(cont.DB, []byte(sessionKey))
|
||||
go cont.SStore.PeriodicCleanup(1*time.Hour, nil)
|
||||
// using packr to bundle templates and static files
|
||||
box := packr.New("templates", "../../templates")
|
||||
layoutT := extendFromBox(nil, box, "layout.tmpl")
|
||||
userT := extendFromBox(layoutT, box, "user.tmpl")
|
||||
cont.Templates = map[string]*template.Template{
|
||||
"login": extendFromBox(layoutT, box, "pages/login.tmpl"),
|
||||
"home": extendFromBox(userT, box, "pages/home.tmpl"),
|
||||
"change_own_password": extendFromBox(userT, box, "pages/change_own_password.tmpl"),
|
||||
"change_password": extendFromBox(userT, box, "pages/change_password.tmpl"),
|
||||
"delete_user": extendFromBox(userT, box, "pages/delete_user.tmpl"),
|
||||
"create_user": extendFromBox(userT, box, "pages/create_user.tmpl"),
|
||||
"update_lastfm_api_key": extendFromBox(userT, box, "pages/update_lastfm_api_key.tmpl"),
|
||||
}
|
||||
withPublicWare := newChain(
|
||||
cont.WithLogging,
|
||||
cont.WithSession,
|
||||
)
|
||||
withUserWare := newChain(
|
||||
withPublicWare,
|
||||
cont.WithUserSession,
|
||||
)
|
||||
withAdminWare := newChain(
|
||||
withUserWare,
|
||||
cont.WithAdminSession,
|
||||
)
|
||||
server := http.FileServer(packr.New("static", "../../static"))
|
||||
mux.Handle("/admin/static/", http.StripPrefix("/admin/static/", server))
|
||||
mux.HandleFunc("/admin/login", withPublicWare(cont.ServeLogin))
|
||||
mux.HandleFunc("/admin/login_do", withPublicWare(cont.ServeLoginDo))
|
||||
mux.HandleFunc("/admin/logout", withUserWare(cont.ServeLogout))
|
||||
mux.HandleFunc("/admin/home", withUserWare(cont.ServeHome))
|
||||
mux.HandleFunc("/admin/change_own_password", withUserWare(cont.ServeChangeOwnPassword))
|
||||
mux.HandleFunc("/admin/change_own_password_do", withUserWare(cont.ServeChangeOwnPasswordDo))
|
||||
mux.HandleFunc("/admin/link_lastfm_do", withUserWare(cont.ServeLinkLastFMDo))
|
||||
mux.HandleFunc("/admin/unlink_lastfm_do", withUserWare(cont.ServeUnlinkLastFMDo))
|
||||
mux.HandleFunc("/admin/change_password", withAdminWare(cont.ServeChangePassword))
|
||||
mux.HandleFunc("/admin/change_password_do", withAdminWare(cont.ServeChangePasswordDo))
|
||||
mux.HandleFunc("/admin/delete_user", withAdminWare(cont.ServeDeleteUser))
|
||||
mux.HandleFunc("/admin/delete_user_do", withAdminWare(cont.ServeDeleteUserDo))
|
||||
mux.HandleFunc("/admin/create_user", withAdminWare(cont.ServeCreateUser))
|
||||
mux.HandleFunc("/admin/create_user_do", withAdminWare(cont.ServeCreateUserDo))
|
||||
mux.HandleFunc("/admin/update_lastfm_api_key", withAdminWare(cont.ServeUpdateLastFMAPIKey))
|
||||
mux.HandleFunc("/admin/update_lastfm_api_key_do", withAdminWare(cont.ServeUpdateLastFMAPIKeyDo))
|
||||
}
|
||||
const (
|
||||
programName = "gonic"
|
||||
programVar = "GONIC"
|
||||
)
|
||||
|
||||
func main() {
|
||||
address := "0.0.0.0:6969"
|
||||
mux := http.NewServeMux()
|
||||
// create a new controller and pass a copy to both routes.
|
||||
// they will add more fields to their copy if they need them
|
||||
baseController := handler.Controller{DB: db.New()}
|
||||
setSubsonicRoutes(baseController, mux)
|
||||
setAdminRoutes(baseController, mux)
|
||||
server := &http.Server{
|
||||
Addr: address,
|
||||
Handler: mux,
|
||||
ReadTimeout: 5 * time.Second,
|
||||
WriteTimeout: 10 * time.Second,
|
||||
IdleTimeout: 15 * time.Second,
|
||||
}
|
||||
log.Printf("starting server at `%s`\n", address)
|
||||
err := server.ListenAndServe()
|
||||
set := flag.NewFlagSet(programName, flag.ExitOnError)
|
||||
listenAddr := set.String(
|
||||
"listen-addr", "0.0.0.0:6969",
|
||||
"listen address (optional)")
|
||||
musicPath := set.String(
|
||||
"music-path", "",
|
||||
"path to music")
|
||||
dbPath := set.String(
|
||||
"db-path", "gonic.db",
|
||||
"path to database (optional)")
|
||||
_ = set.String(
|
||||
"config-path", "",
|
||||
"path to config (optional)")
|
||||
err := ff.Parse(set, os.Args[1:],
|
||||
ff.WithConfigFileFlag("config-path"),
|
||||
ff.WithConfigFileParser(ff.PlainParser),
|
||||
ff.WithEnvVarPrefix(programVar),
|
||||
)
|
||||
if err != nil {
|
||||
log.Printf("when starting server: %v\n", err)
|
||||
log.Fatalf("error parsing args: %v\n", err)
|
||||
}
|
||||
if _, err := os.Stat(*musicPath); os.IsNotExist(err) {
|
||||
log.Fatal("please provide a valid music directory")
|
||||
}
|
||||
db, err := gorm.Open("sqlite3", *dbPath)
|
||||
if err != nil {
|
||||
log.Fatalf("error opening database: %v\n", err)
|
||||
}
|
||||
s := server.New(
|
||||
db,
|
||||
*musicPath,
|
||||
*listenAddr,
|
||||
)
|
||||
log.Printf("starting server at %s", *listenAddr)
|
||||
err = s.ListenAndServe()
|
||||
if err != nil {
|
||||
log.Fatalf("error starting server: %v\n", err)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user