// GetTracks uses the passed URL to find and return // tracks associated with the URL. An error is returned // if any error occurs during the API call. func (yt *YouTube) GetTracks(url string, submitter *gumble.User) ([]interfaces.Track, error) { var ( playlistURL string playlistItemsURL string id string err error resp *http.Response v *jason.Object track bot.Track tracks []interfaces.Track ) playlistURL = "https://www.googleapis.com/youtube/v3/playlists?part=snippet&id=%s&key=%s" playlistItemsURL = "https://www.googleapis.com/youtube/v3/playlistItems?part=snippet,contentDetails&playlistId=%s&maxResults=%d&key=%s&pageToken=%s" id, err = yt.getID(url) if err != nil { return nil, err } if yt.isPlaylist(url) { resp, err = http.Get(fmt.Sprintf(playlistURL, id, viper.GetString("api_keys.youtube"))) defer resp.Body.Close() if err != nil { return nil, err } v, err = jason.NewObjectFromReader(resp.Body) if err != nil { return nil, err } items, _ := v.GetObjectArray("items") item := items[0] title, _ := item.GetString("snippet", "title") playlist := &bot.Playlist{ ID: id, Title: title, Submitter: submitter.Name, Service: yt.ReadableName, } maxItems := math.MaxInt32 if viper.GetInt("queue.max_tracks_per_playlist") > 0 { maxItems = viper.GetInt("queue.max_tracks_per_playlist") } pageToken := "" for len(tracks) < maxItems { curResp, curErr := http.Get(fmt.Sprintf(playlistItemsURL, id, maxItems, viper.GetString("api_keys.youtube"), pageToken)) defer curResp.Body.Close() if curErr != nil { // An error occurred, simply skip this track. continue } v, err = jason.NewObjectFromReader(curResp.Body) if err != nil { // An error occurred, simply skip this track. continue } curTracks, _ := v.GetObjectArray("items") for _, track := range curTracks { videoID, _ := track.GetString("snippet", "resourceId", "videoId") // Unfortunately we have to execute another API call for each video as the YouTube API does not // return video durations from the playlistItems endpoint... newTrack, _ := yt.getTrack(videoID, submitter) newTrack.Playlist = playlist tracks = append(tracks, newTrack) if len(tracks) >= maxItems { break } } } if len(tracks) == 0 { return nil, errors.New("Invalid playlist. No tracks were added") } return tracks, nil } track, err = yt.getTrack(id, submitter) if err != nil { return nil, err } tracks = append(tracks, track) return tracks, nil }