navidrome/persistence/album_repository.go

package persistence

import (
	"errors"

	"github.com/deluan/gosonic/domain"
)

type albumRepository struct {
	ledisRepository
}

func NewAlbumRepository() domain.AlbumRepository {
	r := &albumRepository{}
	r.init("album", &domain.Album{})
	return r
}

func (r *albumRepository) Put(m *domain.Album) error {
	if m.Id == "" {
		return errors.New("Album Id is not set")
	}
	return r.saveOrUpdate(m.Id, m)
}

func (r *albumRepository) Get(id string) (*domain.Album, error) {
	var rec interface{}
	rec, err := r.readEntity(id)
	return rec.(*domain.Album), err
}

func (r *albumRepository) FindByArtist(artistId string) (*domain.Albums, error) {
	var as = make(domain.Albums, 0)
	err := r.loadChildren("artist", artistId, &as, domain.QueryOptions{SortBy: "Year"})
	return &as, err
}

func (r *albumRepository) GetAll(options domain.QueryOptions) (*domain.Albums, error) {
	var as = make(domain.Albums, 0)
	err := r.loadAll(&as, options)
	return &as, err
}

func (r *albumRepository) 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 *albumRepository) PurgeInactive(active *domain.Albums) error {
	currentIds, err := r.getAllIds()
	if err != nil {
		return err
	}
	for _, a := range *active {
		currentIds[a.Id] = false
	}
	inactiveIds := make(map[string]bool)
	for id, inactive := range currentIds {
		if inactive {
			inactiveIds[id] = true
		}
	}
	return r.DeleteAll(inactiveIds)
}

func (r *albumRepository) GetStarred(options domain.QueryOptions) (*domain.Albums, error) {
	var as = make(domain.Albums, 0)
	err := r.loadRange("Starred", true, true, &as, options)
	return &as, err
}

var _ domain.AlbumRepository = (*albumRepository)(nil)