137 lines
3.1 KiB
Go
137 lines
3.1 KiB
Go
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, ", ")
|
|
}
|