82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package engine
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/deluan/navidrome/model"
|
|
"github.com/kennygrant/sanitize"
|
|
)
|
|
|
|
type Search interface {
|
|
SearchArtist(ctx context.Context, q string, offset int, size int) (Entries, error)
|
|
SearchAlbum(ctx context.Context, q string, offset int, size int) (Entries, error)
|
|
SearchSong(ctx context.Context, q string, offset int, size int) (Entries, error)
|
|
}
|
|
|
|
type search struct {
|
|
ds model.DataStore
|
|
}
|
|
|
|
func NewSearch(ds model.DataStore) Search {
|
|
s := &search{ds}
|
|
return s
|
|
}
|
|
|
|
func (s *search) SearchArtist(ctx context.Context, q string, offset int, size int) (Entries, error) {
|
|
q = sanitize.Accents(strings.ToLower(strings.TrimSuffix(q, "*")))
|
|
artists, err := s.ds.Artist(ctx).Search(q, offset, size)
|
|
if len(artists) == 0 || err != nil {
|
|
return nil, nil
|
|
}
|
|
|
|
artistIds := make([]string, len(artists))
|
|
for i, al := range artists {
|
|
artistIds[i] = al.ID
|
|
}
|
|
annMap, err := s.ds.Annotation(ctx).GetMap(getUserID(ctx), model.ArtistItemType, artistIds)
|
|
if err != nil {
|
|
return nil, nil
|
|
}
|
|
|
|
return FromArtists(artists, annMap), nil
|
|
}
|
|
|
|
func (s *search) SearchAlbum(ctx context.Context, q string, offset int, size int) (Entries, error) {
|
|
q = sanitize.Accents(strings.ToLower(strings.TrimSuffix(q, "*")))
|
|
albums, err := s.ds.Album(ctx).Search(q, offset, size)
|
|
if len(albums) == 0 || err != nil {
|
|
return nil, nil
|
|
}
|
|
|
|
albumIds := make([]string, len(albums))
|
|
for i, al := range albums {
|
|
albumIds[i] = al.ID
|
|
}
|
|
annMap, err := s.ds.Annotation(ctx).GetMap(getUserID(ctx), model.AlbumItemType, albumIds)
|
|
if err != nil {
|
|
return nil, nil
|
|
}
|
|
|
|
return FromAlbums(albums, annMap), nil
|
|
}
|
|
|
|
func (s *search) SearchSong(ctx context.Context, q string, offset int, size int) (Entries, error) {
|
|
q = sanitize.Accents(strings.ToLower(strings.TrimSuffix(q, "*")))
|
|
mediaFiles, err := s.ds.MediaFile(ctx).Search(q, offset, size)
|
|
if len(mediaFiles) == 0 || err != nil {
|
|
return nil, nil
|
|
}
|
|
|
|
trackIds := make([]string, len(mediaFiles))
|
|
for i, mf := range mediaFiles {
|
|
trackIds[i] = mf.ID
|
|
}
|
|
annMap, err := s.ds.Annotation(ctx).GetMap(getUserID(ctx), model.MediaItemType, trackIds)
|
|
if err != nil {
|
|
return nil, nil
|
|
}
|
|
|
|
return FromMediaFiles(mediaFiles, annMap), nil
|
|
}
|