Support choose an album from a list (#2)

Author: LinuxSuRen

main.go

@@ -20,16 +20,26 @@ func main() {
 	var albumID int
 	if result, err := broadcast.Search(keyword); err != nil {
 		panic(err)
+	} else if len(result.Data.AlbumViews.Albums) == 0 {
+		fmt.Println("not found any albums by keyword:", keyword)
+		return
 	} else {
-		if len(result.Data.AlbumViews.Albums) > 0 {
-			album := result.Data.AlbumViews.Albums[0].AlbumInfo
-			albumID = album.ID
+		albumTitle := make([]string, 0)
+		albumMap := make(map[string]int, 0)
+		for _, v := range result.Data.AlbumViews.Albums {
+			albumTitle = append(albumTitle, v.AlbumInfo.Title)
+			albumMap[v.AlbumInfo.Title] = v.AlbumInfo.ID
+		}
 
-			fmt.Println("find album", album.Title)
-		} else {
-			fmt.Println("not found any albums by keyword:", keyword)
+		selector := &survey.Select{
+			Message: "Select a album",
+			Options: albumTitle,
+		}
+		var choose string
+		if err = survey.AskOne(selector, &choose); err != nil {
 			return
 		}
+		albumID = albumMap[choose]
 	}
 
 	tracks, err := broadcast.GetTrackList(albumID, 1, false)
@@ -54,15 +64,6 @@ func main() {
 		fmt.Println("start to play", choose)
 		trackInfo := trackMap[choose]
 		play(trackInfo)
-
-		//if resp, err := http.Get(playURL); err != nil {
-		//	fmt.Println(err)
-		//} else {
-		//	data, _ := io.ReadAll(resp.Body)
-		//	buffer := bytes.NewReader(data)
-		//
-		//	play(playio.SeekerWithoutCloser(buffer))
-		//}
 	}
 }
 

pkg/broadcast/core.go

@@ -11,17 +11,17 @@ import (
 
 func Search(keyword string) (result *AlbumSearchResult, err error) {
 	keyword = url.QueryEscape(keyword)
-	apiurl := fmt.Sprintf("https://m.ximalaya.com/m-revision/page/search?kw=%s&core=album&page=1&rows=10", keyword)
+	apiurl := fmt.Sprintf("https://m.ximalaya.com/m-revision/page/search?kw=%s&core=album&page=1&rows=1000", keyword)
 	resp, err := HttpGet(apiurl, Android)
 	if err != nil {
-		return nil, fmt.Errorf("无法获取专辑信息: %v", err)
+		return nil, fmt.Errorf("cannot found albums by keyword: %s, error %v", keyword, err)
 	}
 	defer resp.Body.Close()
 
 	result = &AlbumSearchResult{}
 	err = jsoniter.NewDecoder(resp.Body).Decode(result)
 	if err != nil {
-		return nil, fmt.Errorf("无法获取专辑信息: 无法解析Json: %v", err)
+		return nil, fmt.Errorf("failed to parse json: %v", err)
 	}
 	return
 }
@@ -151,7 +151,7 @@ func GetAudioInfoListByPageID(albumID, pageID int) (playlist *Playlist, err erro
 
 func GetTrackList(albumID, pageID int, isAsc bool) (tracks *TrackList, err error) {
 	url := fmt.Sprintf(
-		"https://mobile.ximalaya.com/mobile/v1/album/track/ts-%d?ac=WIFI&albumId=%d&device=android&isAsc=%t&pageId=%d&pageSize=200",
+		"https://mobile.ximalaya.com/mobile/v1/album/track/ts-%d?ac=WIFI&albumId=%d&device=android&isAsc=%t&pageId=%d&pageSize=1000",
 		time.Now().Unix(), albumID, isAsc, pageID)
 	resp, err := HttpGet(url, Android)
 	if err != nil {