mirror of
https://github.com/navidrome/navidrome.git
synced 2024-09-11 20:12:54 +02:00
221 lines
5.3 KiB
Go
221 lines
5.3 KiB
Go
package core
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"image"
|
|
_ "image/gif"
|
|
"image/jpeg"
|
|
_ "image/png"
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/dhowden/tag"
|
|
"github.com/disintegration/imaging"
|
|
"github.com/navidrome/navidrome/conf"
|
|
"github.com/navidrome/navidrome/consts"
|
|
"github.com/navidrome/navidrome/log"
|
|
"github.com/navidrome/navidrome/model"
|
|
"github.com/navidrome/navidrome/resources"
|
|
"github.com/navidrome/navidrome/utils"
|
|
"github.com/navidrome/navidrome/utils/cache"
|
|
_ "golang.org/x/image/webp"
|
|
)
|
|
|
|
type Artwork interface {
|
|
Get(ctx context.Context, id string, size int) (io.ReadCloser, error)
|
|
}
|
|
|
|
type ArtworkCache cache.FileCache
|
|
|
|
func NewArtwork(ds model.DataStore, cache ArtworkCache) Artwork {
|
|
return &artwork{ds: ds, cache: cache}
|
|
}
|
|
|
|
type artwork struct {
|
|
ds model.DataStore
|
|
cache cache.FileCache
|
|
}
|
|
|
|
type imageInfo struct {
|
|
a *artwork
|
|
id string
|
|
path string
|
|
size int
|
|
lastUpdate time.Time
|
|
}
|
|
|
|
func (ci *imageInfo) Key() string {
|
|
return fmt.Sprintf("%s.%d.%s.%d", ci.path, ci.size, ci.lastUpdate.Format(time.RFC3339Nano), conf.Server.CoverJpegQuality)
|
|
}
|
|
|
|
func (a *artwork) Get(ctx context.Context, id string, size int) (io.ReadCloser, error) {
|
|
path, lastUpdate, err := a.getImagePath(ctx, id)
|
|
if err != nil && err != model.ErrNotFound {
|
|
return nil, err
|
|
}
|
|
|
|
if !conf.Server.DevFastAccessCoverArt {
|
|
if stat, err := os.Stat(path); err == nil {
|
|
lastUpdate = stat.ModTime()
|
|
}
|
|
}
|
|
|
|
info := &imageInfo{
|
|
a: a,
|
|
id: id,
|
|
path: path,
|
|
size: size,
|
|
lastUpdate: lastUpdate,
|
|
}
|
|
|
|
r, err := a.cache.Get(ctx, info)
|
|
if err != nil {
|
|
log.Error(ctx, "Error accessing image cache", "path", path, "size", size, err)
|
|
return nil, err
|
|
}
|
|
return r, err
|
|
}
|
|
|
|
func (a *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)
|
|
id = strings.TrimPrefix(id, "al-")
|
|
var al *model.Album
|
|
al, err = a.ds.Album(ctx).Get(id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if al.CoverArtId == "" {
|
|
err = model.ErrNotFound
|
|
}
|
|
return al.CoverArtPath, al.UpdatedAt, err
|
|
}
|
|
|
|
log.Trace(ctx, "Looking for media file art", "id", id)
|
|
|
|
// Check if id is a mediaFile id
|
|
var mf *model.MediaFile
|
|
mf, err = a.ds.MediaFile(ctx).Get(id)
|
|
|
|
// If it is not, may be an albumId
|
|
if err == model.ErrNotFound {
|
|
return a.getImagePath(ctx, "al-"+id)
|
|
}
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// If it is a mediaFile and it has cover art, return it (if feature is disabled, skip)
|
|
if !conf.Server.DevFastAccessCoverArt && mf.HasCoverArt {
|
|
return mf.Path, mf.UpdatedAt, nil
|
|
}
|
|
|
|
// 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 a.getImagePath(ctx, "al-"+mf.AlbumID)
|
|
}
|
|
|
|
func (a *artwork) getArtwork(ctx context.Context, id string, path string, size int) (reader io.ReadCloser, err error) {
|
|
defer func() {
|
|
if err != nil {
|
|
log.Warn(ctx, "Error extracting image", "path", path, "size", size, err)
|
|
reader, err = resources.FS.Open(consts.PlaceholderAlbumArt)
|
|
}
|
|
}()
|
|
|
|
if path == "" {
|
|
return nil, errors.New("empty path given for artwork")
|
|
}
|
|
|
|
if size == 0 {
|
|
// If requested original size, just read from the file
|
|
if utils.IsAudioFile(path) {
|
|
reader, err = readFromTag(path)
|
|
} else {
|
|
reader, err = readFromFile(path)
|
|
}
|
|
} else {
|
|
// If requested a resized image, get the original (possibly from cache) and resize it
|
|
var r io.ReadCloser
|
|
r, err = a.Get(ctx, id, 0)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer r.Close()
|
|
reader, err = resizeImage(r, size)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func resizeImage(reader io.Reader, size int) (io.ReadCloser, error) {
|
|
img, _, err := image.Decode(reader)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Preserve the aspect ratio of the image.
|
|
var m *image.NRGBA
|
|
bounds := img.Bounds()
|
|
if bounds.Max.X > bounds.Max.Y {
|
|
m = imaging.Resize(img, size, 0, imaging.Lanczos)
|
|
} else {
|
|
m = imaging.Resize(img, 0, size, imaging.Lanczos)
|
|
}
|
|
|
|
buf := new(bytes.Buffer)
|
|
err = jpeg.Encode(buf, m, &jpeg.Options{Quality: conf.Server.CoverJpegQuality})
|
|
return io.NopCloser(buf), err
|
|
}
|
|
|
|
func readFromTag(path string) (io.ReadCloser, error) {
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
|
|
m, err := tag.ReadFrom(f)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
picture := m.Picture()
|
|
if picture == nil {
|
|
return nil, errors.New("file does not contain embedded art")
|
|
}
|
|
return io.NopCloser(bytes.NewReader(picture.Data)), nil
|
|
}
|
|
|
|
func readFromFile(path string) (io.ReadCloser, error) {
|
|
return os.Open(path)
|
|
}
|
|
|
|
var (
|
|
onceImageCache sync.Once
|
|
instanceImageCache ArtworkCache
|
|
)
|
|
|
|
func GetImageCache() ArtworkCache {
|
|
onceImageCache.Do(func() {
|
|
instanceImageCache = cache.NewFileCache("Image", conf.Server.ImageCacheSize, consts.ImageCacheDir, consts.DefaultImageCacheMaxItems,
|
|
func(ctx context.Context, arg cache.Item) (io.Reader, error) {
|
|
info := arg.(*imageInfo)
|
|
reader, err := info.a.getArtwork(ctx, info.id, info.path, info.size)
|
|
if err != nil {
|
|
log.Error(ctx, "Error loading artwork art", "path", info.path, "size", info.size, err)
|
|
return nil, err
|
|
}
|
|
return reader, nil
|
|
})
|
|
})
|
|
return instanceImageCache
|
|
}
|