mirror of
https://github.com/zhaarey/apple-music-downloader.git
synced 2025-10-23 15:11:05 +00:00
test: station dl (need media-user-token)
This commit is contained in:
itouakirai
207
utils/task/station.go
Normal file
207
utils/task/station.go
Normal file
@@ -0,0 +1,207 @@
|
||||
package task
|
||||
|
||||
import (
|
||||
//"bufio"
|
||||
"errors"
|
||||
"fmt"
|
||||
|
||||
//"os"
|
||||
//"strconv"
|
||||
//"strings"
|
||||
|
||||
//"github.com/fatih/color"
|
||||
//"github.com/olekukonko/tablewriter"
|
||||
|
||||
"main/utils/ampapi"
|
||||
)
|
||||
|
||||
type Station struct {
|
||||
Storefront string
|
||||
ID string
|
||||
|
||||
SaveDir string
|
||||
SaveName string
|
||||
Codec string
|
||||
CoverPath string
|
||||
|
||||
Language string
|
||||
Resp ampapi.StationResp
|
||||
Type string
|
||||
Name string
|
||||
Tracks []Track
|
||||
}
|
||||
|
||||
func NewStation(st string, id string) *Station {
|
||||
a := new(Station)
|
||||
a.Storefront = st
|
||||
a.ID = id
|
||||
//fmt.Println("Album created")
|
||||
return a
|
||||
|
||||
}
|
||||
|
||||
func (a *Station) GetResp(mutoken, token, l string) error {
|
||||
var err error
|
||||
a.Language = l
|
||||
resp, err := ampapi.GetStationResp(a.Storefront, a.ID, a.Language, token)
|
||||
if err != nil {
|
||||
return errors.New("error getting station response")
|
||||
}
|
||||
a.Resp = *resp
|
||||
//简化高频调用名称
|
||||
a.Type = a.Resp.Data[0].Attributes.PlayParams.Format
|
||||
if a.Type != "tracks" {
|
||||
return errors.New("stream类型暂未开发")
|
||||
}
|
||||
a.Name = a.Resp.Data[0].Attributes.Name
|
||||
tracksResp, err := ampapi.GetStationNextTracks(a.ID, mutoken, a.Language, token)
|
||||
if err != nil {
|
||||
return errors.New("error getting station tracks response")
|
||||
}
|
||||
//fmt.Println("Getting album response")
|
||||
//从resp中的Tracks数据中提取trackData信息到新的Track结构体中
|
||||
for i, trackData := range tracksResp.Data {
|
||||
albumResp, err := ampapi.GetAlbumRespByHref(trackData.Href, a.Language, token)
|
||||
if err != nil {
|
||||
fmt.Println("Error getting album response:", err)
|
||||
continue
|
||||
}
|
||||
albumLen := len(albumResp.Data[0].Relationships.Tracks.Data)
|
||||
a.Tracks = append(a.Tracks, Track{
|
||||
ID: trackData.ID,
|
||||
Type: trackData.Type,
|
||||
Name: trackData.Attributes.Name,
|
||||
Language: a.Language,
|
||||
Storefront: a.Storefront,
|
||||
|
||||
//SaveDir: filepath.Join(a.SaveDir, a.SaveName),
|
||||
//Codec: a.Codec,
|
||||
TaskNum: i + 1,
|
||||
TaskTotal: len(tracksResp.Data),
|
||||
M3u8: trackData.Attributes.ExtendedAssetUrls.EnhancedHls,
|
||||
WebM3u8: trackData.Attributes.ExtendedAssetUrls.EnhancedHls,
|
||||
//CoverPath: a.CoverPath,
|
||||
|
||||
Resp: trackData,
|
||||
PreType: "stations",
|
||||
DiscTotal: albumResp.Data[0].Relationships.Tracks.Data[albumLen-1].Attributes.DiscNumber,
|
||||
PreID: a.ID,
|
||||
AlbumData: albumResp.Data[0],
|
||||
})
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *Station) GetArtwork() string {
|
||||
return a.Resp.Data[0].Attributes.Artwork.URL
|
||||
}
|
||||
|
||||
// func (a *Album) ShowSelect() []int {
|
||||
// meta := a.Resp
|
||||
// trackTotal := len(meta.Data[0].Relationships.Tracks.Data)
|
||||
// arr := make([]int, trackTotal)
|
||||
// for i := 0; i < trackTotal; i++ {
|
||||
// arr[i] = i + 1
|
||||
// }
|
||||
// selected := []int{}
|
||||
// var data [][]string
|
||||
// for trackNum, track := range meta.Data[0].Relationships.Tracks.Data {
|
||||
// trackNum++
|
||||
// trackName := fmt.Sprintf("%02d. %s", track.Attributes.TrackNumber, track.Attributes.Name)
|
||||
// data = append(data, []string{fmt.Sprint(trackNum),
|
||||
// trackName,
|
||||
// track.Attributes.ContentRating,
|
||||
// track.Type})
|
||||
|
||||
// }
|
||||
// table := tablewriter.NewWriter(os.Stdout)
|
||||
// table.SetHeader([]string{"", "Track Name", "Rating", "Type"})
|
||||
// //table.SetFooter([]string{"", "", "Footer", "Footer4"})
|
||||
// table.SetRowLine(false)
|
||||
// //table.SetAutoMergeCells(true)
|
||||
// table.SetCaption(true, fmt.Sprintf("Storefront: %s, %d tracks missing", strings.ToUpper(a.Storefront), meta.Data[0].Attributes.TrackCount-trackTotal))
|
||||
// table.SetHeaderColor(tablewriter.Colors{},
|
||||
// tablewriter.Colors{tablewriter.FgRedColor, tablewriter.Bold},
|
||||
// tablewriter.Colors{tablewriter.FgBlackColor, tablewriter.Bold},
|
||||
// tablewriter.Colors{tablewriter.FgBlackColor, tablewriter.Bold})
|
||||
|
||||
// table.SetColumnColor(tablewriter.Colors{tablewriter.FgCyanColor},
|
||||
// tablewriter.Colors{tablewriter.Bold, tablewriter.FgRedColor},
|
||||
// tablewriter.Colors{tablewriter.Bold, tablewriter.FgBlackColor},
|
||||
// tablewriter.Colors{tablewriter.Bold, tablewriter.FgBlackColor})
|
||||
// for _, row := range data {
|
||||
// if row[2] == "explicit" {
|
||||
// row[2] = "E"
|
||||
// } else if row[2] == "clean" {
|
||||
// row[2] = "C"
|
||||
// } else {
|
||||
// row[2] = "None"
|
||||
// }
|
||||
// if row[3] == "music-videos" {
|
||||
// row[3] = "MV"
|
||||
// } else if row[3] == "songs" {
|
||||
// row[3] = "SONG"
|
||||
// }
|
||||
// table.Append(row)
|
||||
// }
|
||||
// //table.AppendBulk(data)
|
||||
// table.Render()
|
||||
// fmt.Println("Please select from the track options above (multiple options separated by commas, ranges supported, or type 'all' to select all)")
|
||||
// cyanColor := color.New(color.FgCyan)
|
||||
// cyanColor.Print("select: ")
|
||||
// reader := bufio.NewReader(os.Stdin)
|
||||
// input, err := reader.ReadString('\n')
|
||||
// if err != nil {
|
||||
// fmt.Println(err)
|
||||
// }
|
||||
// input = strings.TrimSpace(input)
|
||||
// if input == "all" {
|
||||
// fmt.Println("You have selected all options:")
|
||||
// selected = arr
|
||||
// } else {
|
||||
// selectedOptions := [][]string{}
|
||||
// parts := strings.Split(input, ",")
|
||||
// for _, part := range parts {
|
||||
// if strings.Contains(part, "-") { // Range setting
|
||||
// rangeParts := strings.Split(part, "-")
|
||||
// selectedOptions = append(selectedOptions, rangeParts)
|
||||
// } else { // Single option
|
||||
// selectedOptions = append(selectedOptions, []string{part})
|
||||
// }
|
||||
// }
|
||||
// //
|
||||
// for _, opt := range selectedOptions {
|
||||
// if len(opt) == 1 { // Single option
|
||||
// num, err := strconv.Atoi(opt[0])
|
||||
// if err != nil {
|
||||
// fmt.Println("Invalid option:", opt[0])
|
||||
// continue
|
||||
// }
|
||||
// if num > 0 && num <= len(arr) {
|
||||
// selected = append(selected, num)
|
||||
// //args = append(args, urls[num-1])
|
||||
// } else {
|
||||
// fmt.Println("Option out of range:", opt[0])
|
||||
// }
|
||||
// } else if len(opt) == 2 { // Range
|
||||
// start, err1 := strconv.Atoi(opt[0])
|
||||
// end, err2 := strconv.Atoi(opt[1])
|
||||
// if err1 != nil || err2 != nil {
|
||||
// fmt.Println("Invalid range:", opt)
|
||||
// continue
|
||||
// }
|
||||
// if start < 1 || end > len(arr) || start > end {
|
||||
// fmt.Println("Range out of range:", opt)
|
||||
// continue
|
||||
// }
|
||||
// for i := start; i <= end; i++ {
|
||||
// //fmt.Println(options[i-1])
|
||||
// selected = append(selected, i)
|
||||
// }
|
||||
// } else {
|
||||
// fmt.Println("Invalid option:", opt)
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
// return selected
|
||||
// }
|
||||
Reference in New Issue
Block a user