66 lines
1.9 KiB
Go
66 lines
1.9 KiB
Go
|
package api
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/ponzischeme89/memby/server/internal/store"
|
||
|
|
)
|
||
|
|
|
||
|
|
// allowedImageTypes guards the path segment we forward to Emby.
|
||
|
|
var allowedImageTypes = map[string]string{
|
||
|
|
"backdrop": "Backdrop",
|
||
|
|
"primary": "Primary",
|
||
|
|
"logo": "Logo",
|
||
|
|
"thumb": "Thumb",
|
||
|
|
}
|
||
|
|
|
||
|
|
// handleImage proxies artwork.
|
||
|
|
//
|
||
|
|
// Going through the gateway means the TV's image URLs carry a gateway token instead of a
|
||
|
|
// live Emby api_key, and it gives Emby's resized output a cacheable home. Images are
|
||
|
|
// tag-addressed, so a hit can be cached hard by any layer in front of this.
|
||
|
|
func (s *Server) handleImage(w http.ResponseWriter, r *http.Request, sess store.Session) {
|
||
|
|
itemID := r.PathValue("itemId")
|
||
|
|
imageType, ok := allowedImageTypes[strings.ToLower(r.PathValue("imageType"))]
|
||
|
|
if !ok || itemID == "" {
|
||
|
|
writeError(w, http.StatusNotFound, "unknown image")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
params := url.Values{}
|
||
|
|
for _, key := range []string{"tag", "maxWidth", "maxHeight", "quality"} {
|
||
|
|
if v := r.URL.Query().Get(key); v != "" {
|
||
|
|
params.Set(key, v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
resp, err := s.emby.ImageResponse(r.Context(), credentials(sess), itemID, imageType, params)
|
||
|
|
if err != nil {
|
||
|
|
s.writeUpstreamError(w, err, "could not load the image")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
if ct := resp.Header.Get("Content-Type"); ct != "" {
|
||
|
|
w.Header().Set("Content-Type", ct)
|
||
|
|
}
|
||
|
|
if cl := resp.Header.Get("Content-Length"); cl != "" {
|
||
|
|
w.Header().Set("Content-Length", cl)
|
||
|
|
}
|
||
|
|
// A tag identifies exact image bytes, so it can be cached indefinitely. Without one,
|
||
|
|
// stay conservative.
|
||
|
|
if params.Get("tag") != "" {
|
||
|
|
w.Header().Set("Cache-Control", "private, max-age=31536000, immutable")
|
||
|
|
} else {
|
||
|
|
w.Header().Set("Cache-Control", "private, max-age=3600")
|
||
|
|
}
|
||
|
|
|
||
|
|
w.WriteHeader(http.StatusOK)
|
||
|
|
if _, err := io.Copy(w, resp.Body); err != nil {
|
||
|
|
s.log.Warn("image copy failed", "error", err)
|
||
|
|
}
|
||
|
|
}
|