109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"gitea.mrixs.me/Mrixs/yandex-music-api-go/internal/models"
|
|
"gitea.mrixs.me/Mrixs/yandex-music-api-go/internal/utils"
|
|
)
|
|
|
|
// GetTracks gets tracks by IDs
|
|
func (c *Client) GetTracks(ctx context.Context, trackIDs []string) ([]*models.Track, error) {
|
|
if len(trackIDs) == 0 {
|
|
return nil, nil
|
|
}
|
|
|
|
params := map[string]string{
|
|
"track-ids": joinIDs(trackIDs),
|
|
}
|
|
|
|
var result []models.Track
|
|
if err := c.request.Get(ctx, "/tracks", params, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
tracks := make([]*models.Track, len(result))
|
|
for i := range result {
|
|
result[i].SetClient(c)
|
|
tracks[i] = &result[i]
|
|
}
|
|
return tracks, nil
|
|
}
|
|
|
|
// GetTrack gets a single track by ID
|
|
func (c *Client) GetTrack(ctx context.Context, trackID string) (*models.Track, error) {
|
|
tracks, err := c.GetTracks(ctx, []string{trackID})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(tracks) == 0 {
|
|
return nil, utils.NewYandexMusicError("track not found", utils.ErrNotFound)
|
|
}
|
|
return tracks[0], nil
|
|
}
|
|
|
|
// GetTrackDownloadInfo gets download info for a track
|
|
func (c *Client) GetTrackDownloadInfo(ctx context.Context, trackID, quality string) ([]models.DownloadInfo, error) {
|
|
// This requires signing - use the sign utility
|
|
// For now, return basic implementation
|
|
params := map[string]string{
|
|
"track_id": trackID,
|
|
}
|
|
|
|
var result []models.DownloadInfo
|
|
if err := c.request.Get(ctx, fmt.Sprintf("/tracks/%s/download-info", trackID), params, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
// DownloadTrack downloads a track
|
|
func (c *Client) DownloadTrack(ctx context.Context, trackID, format string, bitrate int) ([]byte, error) {
|
|
info, err := c.GetTrackDownloadInfo(ctx, trackID, format)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Find matching download info
|
|
var url string
|
|
for _, di := range info {
|
|
if di.Codec == format && (bitrate == 0 || di.Bitrate == bitrate) {
|
|
url = di.URL
|
|
break
|
|
}
|
|
}
|
|
|
|
if url == "" && len(info) > 0 {
|
|
url = info[0].URL
|
|
}
|
|
|
|
if url == "" {
|
|
return nil, utils.NewYandexMusicError("no download URL available", utils.ErrNotFound)
|
|
}
|
|
|
|
return c.request.Download(ctx, url)
|
|
}
|
|
|
|
// SearchResult represents search results
|
|
type SearchResult struct {
|
|
Tracks *SearchTracksResult `json:"tracks"`
|
|
}
|
|
|
|
// SearchTracksResult represents track search results
|
|
type SearchTracksResult struct {
|
|
Total int `json:"total"`
|
|
Results []models.Track `json:"results"`
|
|
}
|
|
|
|
func joinIDs(ids []string) string {
|
|
result := ""
|
|
for i, id := range ids {
|
|
if i > 0 {
|
|
result += ","
|
|
}
|
|
result += id
|
|
}
|
|
return result
|
|
}
|