Skip to content

Commit

Permalink
Rename 'Cover' to the more generic term 'Artwork'
Browse files Browse the repository at this point in the history
  • Loading branch information
deluan committed Jul 31, 2020
1 parent cefdeee commit 2b59d4b
Show file tree
Hide file tree
Showing 9 changed files with 81 additions and 81 deletions.
6 changes: 3 additions & 3 deletions cmd/wire_gen.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

38 changes: 19 additions & 19 deletions core/cover.go → core/artwork.go
Original file line number Diff line number Diff line change
Expand Up @@ -24,39 +24,39 @@ import (
"github.com/disintegration/imaging"
)

type Cover interface {
type Artwork interface {
Get(ctx context.Context, id string, size int, out io.Writer) error
}

type CoverCache FileCache
type ArtworkCache FileCache

func NewCover(ds model.DataStore, cache CoverCache) Cover {
return &cover{ds: ds, cache: cache}
func NewArtwork(ds model.DataStore, cache ArtworkCache) Artwork {
return &artwork{ds: ds, cache: cache}
}

type cover struct {
type artwork struct {
ds model.DataStore
cache FileCache
}

type coverInfo struct {
c *cover
type imageInfo struct {
c *artwork
path string
size int
lastUpdate time.Time
}

func (ci *coverInfo) String() string {
func (ci *imageInfo) String() string {
return fmt.Sprintf("%s.%d.%s.%d", ci.path, ci.size, ci.lastUpdate.Format(time.RFC3339Nano), conf.Server.CoverJpegQuality)
}

func (c *cover) Get(ctx context.Context, id string, size int, out io.Writer) error {
path, lastUpdate, err := c.getCoverPath(ctx, id)
func (c *artwork) Get(ctx context.Context, id string, size int, out io.Writer) error {
path, lastUpdate, err := c.getImagePath(ctx, id)
if err != nil && err != model.ErrNotFound {
return err
}

info := &coverInfo{
info := &imageInfo{
c: c,
path: path,
size: size,
Expand All @@ -73,7 +73,7 @@ func (c *cover) Get(ctx context.Context, id string, size int, out io.Writer) err
return err
}

func (c *cover) getCoverPath(ctx context.Context, id string) (path string, lastUpdated time.Time, err error) {
func (c *artwork) getImagePath(ctx context.Context, id string) (path string, lastUpdated time.Time, err error) {
// If id is an album cover ID
if strings.HasPrefix(id, "al-") {
log.Trace(ctx, "Looking for album art", "id", id)
Expand Down Expand Up @@ -102,10 +102,10 @@ func (c *cover) getCoverPath(ctx context.Context, id string) (path string, lastU

// if the mediafile does not have a coverArt, fallback to the album cover
log.Trace(ctx, "Media file does not contain art. Falling back to album art", "id", id, "albumId", "al-"+mf.AlbumID)
return c.getCoverPath(ctx, "al-"+mf.AlbumID)
return c.getImagePath(ctx, "al-"+mf.AlbumID)
}

func (c *cover) getCover(ctx context.Context, path string, size int) (reader io.Reader, err error) {
func (c *artwork) getArtwork(ctx context.Context, path string, size int) (reader io.Reader, err error) {
defer func() {
if err != nil {
log.Warn(ctx, "Error extracting image", "path", path, "size", size, err)
Expand All @@ -114,7 +114,7 @@ func (c *cover) getCover(ctx context.Context, path string, size int) (reader io.
}()

if path == "" {
return nil, errors.New("empty path given for cover")
return nil, errors.New("empty path given for artwork")
}

var data []byte
Expand Down Expand Up @@ -184,13 +184,13 @@ func readFromFile(path string) ([]byte, error) {
return buf.Bytes(), nil
}

func NewImageCache() CoverCache {
func NewImageCache() ArtworkCache {
return NewFileCache("Image", conf.Server.ImageCacheSize, consts.ImageCacheDir, consts.DefaultImageCacheMaxItems,
func(ctx context.Context, arg fmt.Stringer) (io.Reader, error) {
info := arg.(*coverInfo)
reader, err := info.c.getCover(ctx, info.path, info.size)
info := arg.(*imageInfo)
reader, err := info.c.getArtwork(ctx, info.path, info.size)
if err != nil {
log.Error(ctx, "Error loading cover art", "path", info.path, "size", info.size, err)
log.Error(ctx, "Error loading artwork art", "path", info.path, "size", info.size, err)
return nil, err
}
return reader, nil
Expand Down
38 changes: 19 additions & 19 deletions core/cover_test.go → core/artwork_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,8 @@ import (
. "github.com/onsi/gomega"
)

var _ = Describe("Cover", func() {
var cover Cover
var _ = Describe("Artwork", func() {
var artwork Artwork
var ds model.DataStore
ctx := log.NewContext(context.TODO())

Expand All @@ -32,57 +32,57 @@ var _ = Describe("Cover", func() {
conf.Server.ImageCacheSize = "100MB"
cache := NewImageCache()
Eventually(func() bool { return cache.Ready() }).Should(BeTrue())
cover = NewCover(ds, cache)
artwork = NewArtwork(ds, cache)
})

AfterEach(func() {
os.RemoveAll(conf.Server.DataFolder)
})

It("retrieves the external cover art for an album", func() {
It("retrieves the external artwork art for an album", func() {
buf := new(bytes.Buffer)

Expect(cover.Get(ctx, "al-444", 0, buf)).To(BeNil())
Expect(artwork.Get(ctx, "al-444", 0, buf)).To(BeNil())

_, format, err := image.Decode(bytes.NewReader(buf.Bytes()))
Expect(err).To(BeNil())
Expect(format).To(Equal("jpeg"))
})

It("retrieves the embedded cover art for an album", func() {
It("retrieves the embedded artwork art for an album", func() {
buf := new(bytes.Buffer)

Expect(cover.Get(ctx, "al-222", 0, buf)).To(BeNil())
Expect(artwork.Get(ctx, "al-222", 0, buf)).To(BeNil())

_, format, err := image.Decode(bytes.NewReader(buf.Bytes()))
Expect(err).To(BeNil())
Expect(format).To(Equal("jpeg"))
})

It("returns the default cover if album does not have cover", func() {
It("returns the default artwork if album does not have artwork", func() {
buf := new(bytes.Buffer)

Expect(cover.Get(ctx, "al-333", 0, buf)).To(BeNil())
Expect(artwork.Get(ctx, "al-333", 0, buf)).To(BeNil())

_, format, err := image.Decode(bytes.NewReader(buf.Bytes()))
Expect(err).To(BeNil())
Expect(format).To(Equal("png"))
})

It("returns the default cover if album is not found", func() {
It("returns the default artwork if album is not found", func() {
buf := new(bytes.Buffer)

Expect(cover.Get(ctx, "al-0101", 0, buf)).To(BeNil())
Expect(artwork.Get(ctx, "al-0101", 0, buf)).To(BeNil())

_, format, err := image.Decode(bytes.NewReader(buf.Bytes()))
Expect(err).To(BeNil())
Expect(format).To(Equal("png"))
})

It("retrieves the original cover art from a media_file", func() {
It("retrieves the original artwork art from a media_file", func() {
buf := new(bytes.Buffer)

Expect(cover.Get(ctx, "123", 0, buf)).To(BeNil())
Expect(artwork.Get(ctx, "123", 0, buf)).To(BeNil())

img, format, err := image.Decode(bytes.NewReader(buf.Bytes()))
Expect(err).To(BeNil())
Expand All @@ -91,20 +91,20 @@ var _ = Describe("Cover", func() {
Expect(img.Bounds().Size().Y).To(Equal(600))
})

It("retrieves the album cover art if media_file does not have one", func() {
It("retrieves the album artwork art if media_file does not have one", func() {
buf := new(bytes.Buffer)

Expect(cover.Get(ctx, "456", 0, buf)).To(BeNil())
Expect(artwork.Get(ctx, "456", 0, buf)).To(BeNil())

_, format, err := image.Decode(bytes.NewReader(buf.Bytes()))
Expect(err).To(BeNil())
Expect(format).To(Equal("jpeg"))
})

It("resized cover art as requested", func() {
It("resized artwork art as requested", func() {
buf := new(bytes.Buffer)

Expect(cover.Get(ctx, "123", 200, buf)).To(BeNil())
Expect(artwork.Get(ctx, "123", 200, buf)).To(BeNil())

img, format, err := image.Decode(bytes.NewReader(buf.Bytes()))
Expect(err).To(BeNil())
Expand All @@ -118,14 +118,14 @@ var _ = Describe("Cover", func() {
ds.Album(ctx).(*persistence.MockAlbum).SetError(true)
buf := new(bytes.Buffer)

Expect(cover.Get(ctx, "al-222", 0, buf)).To(MatchError("Error!"))
Expect(artwork.Get(ctx, "al-222", 0, buf)).To(MatchError("Error!"))
})

It("returns err if gets error from media_file table", func() {
ds.MediaFile(ctx).(*persistence.MockMediaFile).SetError(true)
buf := new(bytes.Buffer)

Expect(cover.Get(ctx, "123", 0, buf)).To(MatchError("Error!"))
Expect(artwork.Get(ctx, "123", 0, buf)).To(MatchError("Error!"))
})
})
})
Expand Down
2 changes: 1 addition & 1 deletion core/wire_providers.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ import (
)

var Set = wire.NewSet(
NewCover,
NewArtwork,
NewMediaStreamer,
NewTranscodingCache,
NewImageCache,
Expand Down
6 changes: 3 additions & 3 deletions server/subsonic/api.go
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ type Handler = func(http.ResponseWriter, *http.Request) (*responses.Subsonic, er

type Router struct {
Browser engine.Browser
Cover core.Cover
Artwork core.Artwork
ListGenerator engine.ListGenerator
Playlists engine.Playlists
Ratings engine.Ratings
Expand All @@ -36,10 +36,10 @@ type Router struct {
mux http.Handler
}

func New(browser engine.Browser, cover core.Cover, listGenerator engine.ListGenerator, users engine.Users,
func New(browser engine.Browser, artwork core.Artwork, listGenerator engine.ListGenerator, users engine.Users,
playlists engine.Playlists, ratings engine.Ratings, scrobbler engine.Scrobbler, search engine.Search,
streamer core.MediaStreamer, players engine.Players) *Router {
r := &Router{Browser: browser, Cover: cover, ListGenerator: listGenerator, Playlists: playlists,
r := &Router{Browser: browser, Artwork: artwork, ListGenerator: listGenerator, Playlists: playlists,
Ratings: ratings, Scrobbler: scrobbler, Search: search, Users: users, Streamer: streamer, Players: players}
r.mux = r.routes()
return r
Expand Down
10 changes: 5 additions & 5 deletions server/subsonic/media_retrieval.go
Original file line number Diff line number Diff line change
Expand Up @@ -14,11 +14,11 @@ import (
)

type MediaRetrievalController struct {
cover core.Cover
artwork core.Artwork
}

func NewMediaRetrievalController(cover core.Cover) *MediaRetrievalController {
return &MediaRetrievalController{cover: cover}
func NewMediaRetrievalController(artwork core.Artwork) *MediaRetrievalController {
return &MediaRetrievalController{artwork: artwork}
}

func (c *MediaRetrievalController) GetAvatar(w http.ResponseWriter, r *http.Request) (*responses.Subsonic, error) {
Expand All @@ -41,12 +41,12 @@ func (c *MediaRetrievalController) GetCoverArt(w http.ResponseWriter, r *http.Re
size := utils.ParamInt(r, "size", 0)

w.Header().Set("cache-control", "public, max-age=315360000")
err = c.cover.Get(r.Context(), id, size, w)
err = c.artwork.Get(r.Context(), id, size, w)

switch {
case err == model.ErrNotFound:
log.Error(r, "Couldn't find coverArt", "id", id, err)
return nil, NewError(responses.ErrorDataNotFound, "Cover not found")
return nil, NewError(responses.ErrorDataNotFound, "Artwork not found")
case err != nil:
log.Error(r, "Error retrieving coverArt", "id", id, err)
return nil, NewError(responses.ErrorGeneric, "Internal Error")
Expand Down
54 changes: 27 additions & 27 deletions server/subsonic/media_retrieval_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,44 +11,27 @@ import (
. "github.com/onsi/gomega"
)

type fakeCover struct {
data string
err error
recvId string
recvSize int
}

func (c *fakeCover) Get(ctx context.Context, id string, size int, out io.Writer) error {
if c.err != nil {
return c.err
}
c.recvId = id
c.recvSize = size
_, err := out.Write([]byte(c.data))
return err
}

var _ = Describe("MediaRetrievalController", func() {
var controller *MediaRetrievalController
var cover *fakeCover
var artwork *fakeArtwork
var w *httptest.ResponseRecorder

BeforeEach(func() {
cover = &fakeCover{}
controller = NewMediaRetrievalController(cover)
artwork = &fakeArtwork{}
controller = NewMediaRetrievalController(artwork)
w = httptest.NewRecorder()
})

Describe("GetCoverArt", func() {
It("should return data for that id", func() {
cover.data = "image data"
artwork.data = "image data"
r := newGetRequest("id=34", "size=128")
_, err := controller.GetCoverArt(w, r)

Expect(err).To(BeNil())
Expect(cover.recvId).To(Equal("34"))
Expect(cover.recvSize).To(Equal(128))
Expect(w.Body.String()).To(Equal(cover.data))
Expect(artwork.recvId).To(Equal("34"))
Expect(artwork.recvSize).To(Equal(128))
Expect(w.Body.String()).To(Equal(artwork.data))
})

It("should fail if missing id parameter", func() {
Expand All @@ -59,19 +42,36 @@ var _ = Describe("MediaRetrievalController", func() {
})

It("should fail when the file is not found", func() {
cover.err = model.ErrNotFound
artwork.err = model.ErrNotFound
r := newGetRequest("id=34", "size=128")
_, err := controller.GetCoverArt(w, r)

Expect(err).To(MatchError("Cover not found"))
Expect(err).To(MatchError("Artwork not found"))
})

It("should fail when there is an unknown error", func() {
cover.err = errors.New("weird error")
artwork.err = errors.New("weird error")
r := newGetRequest("id=34", "size=128")
_, err := controller.GetCoverArt(w, r)

Expect(err).To(MatchError("Internal Error"))
})
})
})

type fakeArtwork struct {
data string
err error
recvId string
recvSize int
}

func (c *fakeArtwork) Get(ctx context.Context, id string, size int, out io.Writer) error {
if c.err != nil {
return c.err
}
c.recvId = id
c.recvSize = size
_, err := out.Write([]byte(c.data))
return err
}

0 comments on commit 2b59d4b

Please sign in to comment.