navidrome/engine/search.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().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().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().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().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().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().GetMap(getUserID(ctx), model.MediaItemType, trackIds)
	if err != nil {
		return nil, nil
	}

	return FromMediaFiles(mediaFiles, annMap), nil
}