navidrome/persistence/db_ledis/mediafile_repository.go

package db_ledis

import (
	"errors"
	"sort"
	"time"

	"github.com/cloudsonic/sonic-server/domain"
)

type mediaFileRepository struct {
	ledisRepository
}

func NewMediaFileRepository() domain.MediaFileRepository {
	r := &mediaFileRepository{}
	r.init("mediafile", &domain.MediaFile{})
	return r
}

func (r *mediaFileRepository) Put(m *domain.MediaFile) error {
	if m.ID == "" {
		return errors.New("mediaFile ID is not set")
	}
	return r.saveOrUpdate(m.ID, m)
}

func (r *mediaFileRepository) Get(id string) (*domain.MediaFile, error) {
	m, err := r.readEntity(id)
	if err != nil {
		return nil, err
	}
	mf := m.(*domain.MediaFile)
	if mf.ID != id {
		return nil, nil
	}
	return mf, nil
}

func (r *mediaFileRepository) FindByAlbum(albumId string) (domain.MediaFiles, error) {
	var mfs = make(domain.MediaFiles, 0)
	err := r.loadChildren("album", albumId, &mfs, domain.QueryOptions{SortBy: "TrackNumber"})
	sort.Sort(mfs)
	return mfs, err
}

func (r *mediaFileRepository) GetStarred(options domain.QueryOptions) (domain.MediaFiles, error) {
	var mfs = make(domain.MediaFiles, 0)
	start := time.Time{}.Add(1 * time.Hour)
	err := r.loadRange("Starred", start, time.Now(), &mfs, options)
	return mfs, err
}

func (r *mediaFileRepository) GetAllIds() ([]string, error) {
	idMap, err := r.getAllIds()
	if err != nil {
		return nil, err
	}
	ids := make([]string, len(idMap))

	i := 0
	for id := range idMap {
		ids[i] = id
		i++
	}

	return ids, nil
}

func (r *mediaFileRepository) PurgeInactive(active domain.MediaFiles) ([]string, error) {
	return r.purgeInactive(active, func(e interface{}) string {
		return e.(domain.MediaFile).ID
	})
}

var _ domain.MediaFileRepository = (*mediaFileRepository)(nil)