mirror of
https://github.com/navidrome/navidrome.git
synced 2024-09-09 11:11:45 +02:00
69 lines
1.8 KiB
Go
69 lines
1.8 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
|
|
}
|
|
return FromArtists(artists), 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
|
|
}
|
|
|
|
return FromAlbums(albums), 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
|
|
}
|
|
|
|
return FromMediaFiles(mediaFiles), nil
|
|
}
|