apple-music-downloader/utils/task/station.go

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
	a.Name = a.Resp.Data[0].Attributes.Name
	if a.Type != "tracks" {
		return nil
	}
	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],
		})
		a.Tracks[i].PlaylistData.Attributes.Name = a.Name
		a.Tracks[i].PlaylistData.Attributes.ArtistName = "Apple Music Station"
	}
	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
// }