navidrome/persistence/album_repository.go

84 lines
1.8 KiB
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)