add SearchAll function to MediaProvider

backend/mediaprovider/subsonic/searchall.go

@@ -0,0 +1,136 @@
+package subsonic
+
+import (
+	"sort"
+	"strconv"
+	"strings"
+	"sync"
+
+	"github.com/dweymouth/go-subsonic/subsonic"
+	"github.com/dweymouth/supersonic/backend/mediaprovider"
+	"github.com/dweymouth/supersonic/sharedutil"
+)
+
+func (s *subsonicMediaProvider) SearchAll(searchQuery string, maxResults int) ([]*mediaprovider.SearchResult, error) {
+	var wg sync.WaitGroup
+	var err error // only set by Search3
+	var result *subsonic.SearchResult3
+	var playlists []*subsonic.Playlist
+	var genres []*subsonic.Genre
+
+	wg.Add(1)
+	go func() {
+		count := strconv.Itoa(maxResults / 3)
+		res, e := s.client.Search3(searchQuery, map[string]string{
+			"artistCount": count,
+			"albumCount":  count,
+			"songCount":   count,
+		})
+		if e != nil {
+			err = e
+		} else {
+			result = res
+		}
+		wg.Done()
+	}()
+
+	queryLowerWords := strings.Fields(strings.ToLower(searchQuery))
+
+	wg.Add(1)
+	go func() {
+		p, e := s.client.GetPlaylists(nil)
+		if e != nil {
+			playlists = sharedutil.FilterSlice(p, func(p *subsonic.Playlist) bool {
+				return allTermsMatch(strings.ToLower(p.Name), queryLowerWords)
+			})
+		}
+		wg.Done()
+	}()
+
+	wg.Add(1)
+	go func() {
+		g, e := s.client.GetGenres()
+		if e != nil {
+			genres = sharedutil.FilterSlice(g, func(g *subsonic.Genre) bool {
+				return allTermsMatch(strings.ToLower(g.Name), queryLowerWords)
+			})
+		}
+		wg.Done()
+	}()
+
+	wg.Wait()
+	if err != nil {
+		return nil, err
+	}
+
+	results := mergeResults(result, playlists, genres)
+	//rankResults(results, queryLowerWords) // TODO
+	if len(results) > maxResults {
+		results = results[:maxResults]
+	}
+	return results, nil
+}
+
+// name and terms should be pre-converted to the same case
+func allTermsMatch(name string, terms []string) bool {
+	for _, t := range terms {
+		if !strings.Contains(name, t) {
+			return false
+		}
+	}
+	return true
+}
+
+func mergeResults(
+	searchResult *subsonic.SearchResult3,
+	matchingPlaylists []*subsonic.Playlist,
+	matchingGenres []*subsonic.Genre,
+) []*mediaprovider.SearchResult {
+	var results []*mediaprovider.SearchResult
+
+	for _, al := range searchResult.Album {
+		results = append(results, &mediaprovider.SearchResult{
+			Type:       mediaprovider.ContentTypeAlbum,
+			Name:       al.Name,
+			ArtistName: getNameString(al.Artist, al.Artists),
+			Size:       al.SongCount,
+		})
+	}
+
+	for _, ar := range searchResult.Artist {
+		results = append(results, &mediaprovider.SearchResult{
+			Type: mediaprovider.ContentTypeArtist,
+			Name: ar.Name,
+			Size: ar.AlbumCount,
+		})
+	}
+
+	for _, tr := range searchResult.Song {
+		results = append(results, &mediaprovider.SearchResult{
+			Type:       mediaprovider.ContentTypeTrack,
+			Name:       tr.Title,
+			ArtistName: getNameString(tr.Artist, tr.Artists),
+			Size:       tr.Duration,
+		})
+	}
+
+	return results
+}
+
+func rankResults(results []*mediaprovider.SearchResult, queryTerms []string) {
+	// TODO
+	sort.Slice(results, func(a, b int) bool {
+		return false
+	})
+}
+
+// select Subsonic single-valued name or join OpenSubsonic multi-valued names
+func getNameString(singleName string, idNames []subsonic.IDName) string {
+	if len(idNames) == 0 {
+		return singleName
+	}
+	names := sharedutil.MapSlice(idNames, func(a subsonic.IDName) string {
+		return a.Name
+	})
+	return strings.Join(names, ", ")
+}