102 lines
2.4 KiB
Go
102 lines
2.4 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/deluan/navidrome/assets"
|
|
"github.com/deluan/navidrome/conf"
|
|
"github.com/deluan/navidrome/model"
|
|
"github.com/deluan/rest"
|
|
"github.com/go-chi/chi"
|
|
"github.com/go-chi/jwtauth"
|
|
)
|
|
|
|
var initialUser = model.User{
|
|
UserName: "admin",
|
|
Name: "Admin",
|
|
IsAdmin: true,
|
|
}
|
|
|
|
type Router struct {
|
|
ds model.DataStore
|
|
mux http.Handler
|
|
path string
|
|
}
|
|
|
|
func New(ds model.DataStore, path string) *Router {
|
|
r := &Router{ds: ds, path: path}
|
|
r.mux = r.routes()
|
|
return r
|
|
}
|
|
|
|
func (app *Router) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
app.mux.ServeHTTP(w, r)
|
|
}
|
|
|
|
func (app *Router) routes() http.Handler {
|
|
r := chi.NewRouter()
|
|
|
|
// Basic unauthenticated ping
|
|
r.Get("/ping", func(w http.ResponseWriter, r *http.Request) { w.Write([]byte(`{"response":"pong"}`)) })
|
|
|
|
r.Post("/login", Login(app.ds))
|
|
|
|
r.Route("/api", func(r chi.Router) {
|
|
if !conf.Sonic.DevDisableAuthentication {
|
|
r.Use(jwtauth.Verifier(TokenAuth))
|
|
r.Use(Authenticator)
|
|
}
|
|
app.R(r, "/user", model.User{})
|
|
app.R(r, "/song", model.MediaFile{})
|
|
app.R(r, "/album", model.Album{})
|
|
app.R(r, "/artist", model.Artist{})
|
|
})
|
|
|
|
// Serve UI app assets
|
|
r.Handle("/*", http.StripPrefix(app.path, http.FileServer(assets.AssetFile())))
|
|
|
|
return r
|
|
}
|
|
|
|
func (app *Router) R(r chi.Router, pathPrefix string, model interface{}) {
|
|
constructor := func(ctx context.Context) rest.Repository {
|
|
return app.ds.Resource(model)
|
|
}
|
|
r.Route(pathPrefix, func(r chi.Router) {
|
|
r.Get("/", rest.GetAll(constructor))
|
|
r.Post("/", rest.Post(constructor))
|
|
r.Route("/{id:[0-9a-f\\-]+}", func(r chi.Router) {
|
|
r.Use(UrlParams)
|
|
r.Get("/", rest.Get(constructor))
|
|
r.Put("/", rest.Put(constructor))
|
|
r.Delete("/", rest.Delete(constructor))
|
|
})
|
|
})
|
|
}
|
|
|
|
// Middleware to convert Chi URL params (from Context) to query params, as expected by our REST package
|
|
func UrlParams(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
ctx := chi.RouteContext(r.Context())
|
|
parts := make([]string, 0)
|
|
for i, key := range ctx.URLParams.Keys {
|
|
value := ctx.URLParams.Values[i]
|
|
if key == "*" {
|
|
continue
|
|
}
|
|
parts = append(parts, url.QueryEscape(":"+key)+"="+url.QueryEscape(value))
|
|
}
|
|
q := strings.Join(parts, "&")
|
|
if r.URL.RawQuery == "" {
|
|
r.URL.RawQuery = q
|
|
} else {
|
|
r.URL.RawQuery += "&" + q
|
|
}
|
|
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|