Publish current app and server

server/internal/api/images.go

@@ -12,6 +12,7 @@ import (
 	"strings"
 	"syscall"
 
+	"github.com/ponzischeme89/memby/server/internal/radarr"
 	"github.com/ponzischeme89/memby/server/internal/sonarr"
 	"github.com/ponzischeme89/memby/server/internal/store"
 )
@@ -40,6 +41,10 @@ func (s *Server) handleImage(w http.ResponseWriter, r *http.Request, sess store.
 		s.handleSonarrImage(w, r, itemID, imageType)
 		return
 	}
+	if strings.HasPrefix(itemID, "radarr:") {
+		s.handleRadarrImage(w, r, itemID, imageType)
+		return
+	}
 
 	params := url.Values{}
 	for _, key := range []string{"tag", "maxWidth", "maxHeight", "quality"} {
@@ -78,6 +83,50 @@ func (s *Server) handleImage(w http.ResponseWriter, r *http.Request, sess store.
 		"source", "emby", "item_id", itemID, "image_type", imageType)
 }
 
+func (s *Server) handleRadarrImage(w http.ResponseWriter, r *http.Request, itemID, imageType string) {
+	if s.radarr == nil {
+		writeError(w, http.StatusNotFound, "unknown image")
+		return
+	}
+	parts := strings.Split(itemID, ":")
+	if len(parts) != 2 {
+		writeError(w, http.StatusNotFound, "unknown image")
+		return
+	}
+	movieID, err := strconv.Atoi(parts[1])
+	if err != nil || movieID <= 0 {
+		writeError(w, http.StatusNotFound, "unknown image")
+		return
+	}
+	coverType := map[string]string{"Primary": "poster", "Backdrop": "fanart"}[imageType]
+	if coverType == "" {
+		writeError(w, http.StatusNotFound, "unknown image")
+		return
+	}
+	resp, err := s.radarr.MediaCover(r.Context(), movieID, coverType)
+	if err != nil {
+		var apiErr *radarr.APIError
+		if errors.As(err, &apiErr) && apiErr.StatusCode == http.StatusNotFound {
+			writeError(w, http.StatusNotFound, "image not found")
+			return
+		}
+		s.log.Warn("radarr image failed", "movie_id", movieID, "type", coverType, "error", err)
+		writeError(w, http.StatusBadGateway, "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)
+	}
+	w.Header().Set("Cache-Control", "private, max-age=3600")
+	w.WriteHeader(http.StatusOK)
+	copyImage(w, r, resp.Body, s.log,
+		"source", "radarr", "item_id", itemID, "image_type", imageType)
+}
+
 func (s *Server) handleSonarrImage(w http.ResponseWriter, r *http.Request, itemID, imageType string) {
 	if s.sonarr == nil {
 		writeError(w, http.StatusNotFound, "unknown image")