add: station dl(need media-user-token)

utils/ampapi/station.go

@@ -45,6 +45,46 @@ func GetStationResp(storefront string, id string, language string, token string)
 	return obj, nil
 }
 
+func GetStationAssetsUrl(id string, mutoken string, token string) (string, error) {
+	var err error
+	if token == "" {
+		token, err = GetToken()
+		if err != nil {
+			return "", err
+		}
+	}
+
+	req, err := http.NewRequest("GET", "https://amp-api.music.apple.com/v1/play/assets", nil)
+	if err != nil {
+		return "", err
+	}
+	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", token))
+	req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
+	req.Header.Set("Origin", "https://music.apple.com")
+	req.Header.Set("Media-User-Token", mutoken)
+	query := url.Values{}
+	//query.Set("omit[resource]", "autos")
+	//query.Set("extend", "editorialVideo")
+	query.Set("id", id)
+	query.Set("kind", "radioStation")
+	query.Set("keyFormat", "web")
+	req.URL.RawQuery = query.Encode()
+	do, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return "", err
+	}
+	defer do.Body.Close()
+	if do.StatusCode != http.StatusOK {
+		return "", errors.New(do.Status)
+	}
+	obj := new(StationAssets)
+	err = json.NewDecoder(do.Body).Decode(&obj)
+	if err != nil {
+		return "", err
+	}
+	return obj.Results.Assets[0].Url, nil
+}
+
 func GetStationNextTracks(id, mutoken, language, token string) (*TrackResp, error) {
 	var err error
 	if token == "" {
@@ -92,6 +132,17 @@ type StationResp struct {
 	Data []StationRespData `json:"data"`
 }
 
+type StationAssets struct {
+	Results struct {
+		Assets []struct {
+			KeyServerUrl              string `json:"keyServerUrl"`
+			Url                       string `json:"url"`
+			WidevineKeyCertificateUrl string `json:"widevineKeyCertificateUrl"`
+			FairPlayKeyCertificateUrl string `json:"fairPlayKeyCertificateUrl"`
+		} `json:"assets"`
+	} `json:"results"`
+}
+
 type StationRespData struct {
 	ID         string `json:"id"`
 	Type       string `json:"type"`

utils/runv3/runv3.go

@@ -5,6 +5,7 @@ import (
 	"encoding/base64"
 	"fmt"
 	"path/filepath"
+
 	"github.com/gospider007/requests"
 	"google.golang.org/protobuf/proto"
 
@@ -154,7 +155,7 @@ func GetWebplayback(adamId string, authtoken string, mutoken string, mvmode bool
 			return obj.List[0].HlsPlaylistUrl, "", nil
 		}
 		// 遍历 Assets
-		for i, _ := range obj.List[0].Assets {
+		for i := range obj.List[0].Assets {
 			if obj.List[0].Assets[i].Flavor == "28:ctrp256" {
 				kidBase64, fileurl, err := extractKidBase64(obj.List[0].Assets[i].URL, false)
 				if err != nil {
@@ -298,10 +299,19 @@ func Run(adamId string, trackpath string, authtoken string, mutoken string, mvmo
 		AfterRequest:  AfterRequest,
 	}
 	key.CdmInit()
-	keystr, keybt, err := key.GetKey(ctx, "https://play.itunes.apple.com/WebObjects/MZPlay.woa/wa/acquireWebPlaybackLicense", pssh, nil)
-	if err != nil {
-		fmt.Println(err)
-		return "", err
+	var keybt []byte
+	if strings.Contains(adamId, "ra.") {
+		keystr, keybt, err = key.GetKey(ctx, "https://play.itunes.apple.com/WebObjects/MZPlay.woa/web/radio/versions/1/license", pssh, nil)
+		if err != nil {
+			fmt.Println(err)
+			return "", err
+		}
+	} else {
+		keystr, keybt, err = key.GetKey(ctx, "https://play.itunes.apple.com/WebObjects/MZPlay.woa/wa/acquireWebPlaybackLicense", pssh, nil)
+		if err != nil {
+			fmt.Println(err)
+			return "", err
+		}
 	}
 	if mvmode {
 		keyAndUrls := "1:" + keystr + ";" + fileurl

utils/task/station.go

@@ -50,10 +50,10 @@ func (a *Station) GetResp(mutoken, token, l string) error {
 	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
+	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")