mirror of
https://github.com/navidrome/navidrome.git
synced 2025-04-03 04:27:37 +03:00
85 lines
2.6 KiB
Go
85 lines
2.6 KiB
Go
package public
|
|
|
|
import (
|
|
"net/http"
|
|
"net/url"
|
|
"path"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
"github.com/navidrome/navidrome/conf"
|
|
"github.com/navidrome/navidrome/consts"
|
|
"github.com/navidrome/navidrome/core"
|
|
"github.com/navidrome/navidrome/core/artwork"
|
|
"github.com/navidrome/navidrome/log"
|
|
"github.com/navidrome/navidrome/model"
|
|
"github.com/navidrome/navidrome/server"
|
|
"github.com/navidrome/navidrome/ui"
|
|
)
|
|
|
|
type Router struct {
|
|
http.Handler
|
|
artwork artwork.Artwork
|
|
streamer core.MediaStreamer
|
|
archiver core.Archiver
|
|
share core.Share
|
|
assetsHandler http.Handler
|
|
ds model.DataStore
|
|
}
|
|
|
|
func New(ds model.DataStore, artwork artwork.Artwork, streamer core.MediaStreamer, share core.Share, archiver core.Archiver) *Router {
|
|
p := &Router{ds: ds, artwork: artwork, streamer: streamer, share: share, archiver: archiver}
|
|
shareRoot := path.Join(conf.Server.BasePath, consts.URLPathPublic)
|
|
p.assetsHandler = http.StripPrefix(shareRoot, http.FileServer(http.FS(ui.BuildAssets())))
|
|
p.Handler = p.routes()
|
|
|
|
return p
|
|
}
|
|
|
|
func (pub *Router) routes() http.Handler {
|
|
r := chi.NewRouter()
|
|
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(server.URLParamsMiddleware)
|
|
r.Group(func(r chi.Router) {
|
|
if conf.Server.DevArtworkMaxRequests > 0 {
|
|
log.Debug("Throttling public images endpoint", "maxRequests", conf.Server.DevArtworkMaxRequests,
|
|
"backlogLimit", conf.Server.DevArtworkThrottleBacklogLimit, "backlogTimeout",
|
|
conf.Server.DevArtworkThrottleBacklogTimeout)
|
|
r.Use(middleware.ThrottleBacklog(conf.Server.DevArtworkMaxRequests, conf.Server.DevArtworkThrottleBacklogLimit,
|
|
conf.Server.DevArtworkThrottleBacklogTimeout))
|
|
}
|
|
r.HandleFunc("/img/{id}", pub.handleImages)
|
|
})
|
|
if conf.Server.EnableSharing {
|
|
r.HandleFunc("/s/{id}", pub.handleStream)
|
|
if conf.Server.EnableDownloads {
|
|
r.HandleFunc("/d/{id}", pub.handleDownloads)
|
|
}
|
|
r.HandleFunc("/{id}/m3u", pub.handleM3U)
|
|
r.HandleFunc("/{id}", pub.handleShares)
|
|
r.HandleFunc("/", pub.handleShares)
|
|
r.Handle("/*", pub.assetsHandler)
|
|
}
|
|
})
|
|
return r
|
|
}
|
|
|
|
func ShareURL(r *http.Request, id string) string {
|
|
uri := path.Join(consts.URLPathPublic, id)
|
|
return publicURL(r, uri, nil)
|
|
}
|
|
|
|
func publicURL(r *http.Request, u string, params url.Values) string {
|
|
if conf.Server.ShareURL != "" {
|
|
shareUrl, _ := url.Parse(conf.Server.ShareURL)
|
|
buildUrl, _ := url.Parse(u)
|
|
buildUrl.Scheme = shareUrl.Scheme
|
|
buildUrl.Host = shareUrl.Host
|
|
if len(params) > 0 {
|
|
buildUrl.RawQuery = params.Encode()
|
|
}
|
|
return buildUrl.String()
|
|
}
|
|
return server.AbsoluteURL(r, u, params)
|
|
}
|