From 704e8d770e9c2308dd5fc9544207969740710a81 Mon Sep 17 00:00:00 2001 From: GenzPN <39546317+GenzPN@users.noreply.github.com> Date: Tue, 15 Apr 2025 22:03:15 +0700 Subject: [PATCH 1/6] Update Support Download Classical From Apple Music https://classical.music.apple.com/vn/album/1780442290 --- main.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.go b/main.go index 362375c..72417ec 100644 --- a/main.go +++ b/main.go @@ -92,7 +92,7 @@ func fileExists(path string) (bool, error) { } func checkUrl(url string) (string, string) { - pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music)\.apple\.com\/(\w{2})(?:\/album|\/album\/.+))\/(?:id)?(\d[^\D]+)(?:$|\?)`) + pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music|classical\.music)\.apple\.com\/(\w{2})(?:\/album|\/album\/.+))\/(?:id)?(\d[^\D]+)(?:$|\?)`) matches := pat.FindAllStringSubmatch(url, -1) if matches == nil { From 4f7a11246a714d2f15e6bf55a93dff8593bd05ed Mon Sep 17 00:00:00 2001 From: Mrykerflorendo Date: Wed, 7 May 2025 12:13:24 -0700 Subject: [PATCH 2/6] Update README.md --- README.md | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index 2029cf9..a398dfe 100644 --- a/README.md +++ b/README.md @@ -1,16 +1,18 @@ -### ！！必须先安装[MP4Box](https://gpac.io/downloads/gpac-nightly-builds/)，并确认[MP4Box](https://gpac.io/downloads/gpac-nightly-builds/)已正确添加到环境变量 +English / [简体中文](./README-CN.md) -### 添加功能 +### ！！Must be installed first [MP4Box](https://gpac.io/downloads/gpac-nightly-builds/)，And confirm [MP4Box](https://gpac.io/downloads/gpac-nightly-builds/) Correctly added to environment variables ！！ -1. 支持内嵌封面和LRC歌词（需要`media-user-token`，获取方式看最后的说明） -2. 支持获取逐词与未同步歌词 -3. 支持下载歌手 `go run main.go https://music.apple.com/us/artist/taylor-swift/159260351` `--all-album` 自动选择歌手的所有专辑 -4. 下载解密部分更换为Sendy McSenderson的代码，实现边下载边解密,解决大文件解密时内存不足 -5. MV下载，需要安装[mp4decrypt](https://www.bento4.com/downloads/) +### Add features + +1. Supports inline covers and LRC lyrics（Demand`media-user-token`，See the instructions at the end for how to get it） +2. Added support for getting word-by-word and out-of-sync lyrics +3. Support downloading singers `go run main.go https://music.apple.com/us/artist/taylor-swift/159260351` `--all-album` Automatically select all albums of the artist +4. The download decryption part is replaced with Sendy McSenderson to decrypt while downloading, and solve the lack of memory when decrypting large files +5. MV Download, installation required[mp4decrypt](https://www.bento4.com/downloads/) ### Special thanks to `chocomint` for creating `agent-arm64.js` -对于获取`aac-lc` `MV` `歌词` 必须填入有订阅的`media-user-token` +For acquisition`aac-lc` `MV` `lyrics` You must fill in the information with a subscription`media-user-token` - `alac (audio-alac-stereo)` - `ec3 (audio-atmos / audio-ec3)` @@ -34,7 +36,7 @@ Original script by Sorrow. Modified by me to include some fixes and improvements 7. For aac: `go run main.go --aac https://music.apple.com/us/album/1989-taylors-version-deluxe/1713845538`. 8. For see quality: `go run main.go --debug https://music.apple.com/us/album/1989-taylors-version-deluxe/1713845538`. -[中文教程-详见方法三](https://telegra.ph/Apple-Music-Alac高解析度无损音乐下载教程-04-02-2) +[Chinese tutorial - see Method 3 for details](https://telegra.ph/Apple-Music-Alac高解析度无损音乐下载教程-04-02-2) ## Downloading lyrics From 63705266b6584dee7fdfffe26721074c0b6fcb5a Mon Sep 17 00:00:00 2001 From: Mrykerflorendo Date: Wed, 7 May 2025 12:14:12 -0700 Subject: [PATCH 3/6] Create README-CN.md --- README-CN.md | 46 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 46 insertions(+) create mode 100644 README-CN.md diff --git a/README-CN.md b/README-CN.md new file mode 100644 index 0000000..bc2d91b --- /dev/null +++ b/README-CN.md @@ -0,0 +1,46 @@ +[English](./README.md) / 简体中文 + +### ！！必须先安装[MP4Box](https://gpac.io/downloads/gpac-nightly-builds/)，并确认[MP4Box](https://gpac.io/downloads/gpac-nightly-builds/)已正确添加到环境变量 + +### 添加功能 + +1. 支持内嵌封面和LRC歌词（需要`media-user-token`，获取方式看最后的说明） +2. 支持获取逐词与未同步歌词 +3. 支持下载歌手 `go run main.go https://music.apple.com/us/artist/taylor-swift/159260351` `--all-album` 自动选择歌手的所有专辑 +4. 下载解密部分更换为Sendy McSenderson的代码，实现边下载边解密,解决大文件解密时内存不足 +5. MV下载，需要安装[mp4decrypt](https://www.bento4.com/downloads/) + +### 特别感谢 `chocomint` 创建 `agent-arm64.js` +对于获取`aac-lc` `MV` `歌词` 必须填入有订阅的`media-user-token` + +- `alac (audio-alac-stereo)` +- `ec3 (audio-atmos / audio-ec3)` +- `aac (audio-stereo)` +- `aac-lc (audio-stereo)` +- `aac-binaural (audio-stereo-binaural)` +- `aac-downmix (audio-stereo-downmix)` +- `MV` + +# Apple Music ALAC/杜比全景声下载器 + +原脚本由 Sorrow 编写。本人已修改，包含一些修复和改进。 + +## 使用方法 +1. 确保解密程序 [wrapper](https://github.com/zhaarey/wrapper) 正在运行 +2. 开始下载部分专辑：`go run main.go https://music.apple.com/us/album/whenever-you-need-somebody-2022-remaster/1624945511`。 +3. 开始下载单曲：`go run main.go --song https://music.apple.com/us/album/never-gonna-give-you-up-2022-remaster/1624945511?i=1624945512` 或 `go run main.go https://music.apple.com/us/song/you-move-me-2022-remaster/1624945520`。 +4. 开始下载所选曲目：`go run main.go --select https://music.apple.com/us/album/whenever-you-need-somebody-2022-remaster/1624945511` 输入以空格分隔的数字。 +5. 开始下载部分播放列表：`go run main.go https://music.apple.com/us/playlist/taylor-swift-essentials/pl.3950454ced8c45a3b0cc693c2a7db97b` 或 `go run main.go https://music.apple.com/us/playlist/hi-res-lossless-24-bit-192khz/pl.u-MDAWvpjt38370N`。 +6. 对于杜比全景声 (Dolby Atmos)：`go run main.go --atmos https://music.apple.com/us/album/1989-taylors-version-deluxe/1713845538`。 +7. 对于 AAC (AAC)：`go run main.go --aac https://music.apple.com/us/album/1989-taylors-version-deluxe/1713845538`。 +8. 要查看音质：`go run main.go --debug https://music.apple.com/us/album/1989-taylors-version-deluxe/1713845538`。 + +[中文教程-详见方法三](https://telegra.ph/Apple-Music-Alac高解析度无损音乐下载教程-04-02-2) + +## 下载歌词 + +1. 打开 [Apple Music](https://music.apple.com) 并登录 +2. 打开开发者工具，点击“应用程序 -> 存储 -> Cookies -> https://music.apple.com” +3. 找到名为“media-user-token”的 Cookie 并复制其值 +4. 将步骤 3 中获取的 Cookie 值粘贴到 config.yaml 文件中并保存 +5. 正常启动脚本 From 12dfca3ee84438e0f37d95bdd23fe07aaf4747e8 Mon Sep 17 00:00:00 2001 From: Mrykerflorendo Date: Wed, 7 May 2025 12:14:45 -0700 Subject: [PATCH 4/6] Translation Process Complete --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index a398dfe..a0a4de5 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,6 @@ English / [简体中文](./README-CN.md) -### ！！Must be installed first [MP4Box](https://gpac.io/downloads/gpac-nightly-builds/)，And confirm [MP4Box](https://gpac.io/downloads/gpac-nightly-builds/) Correctly added to environment variables ！！ +### ！！Must be installed first [MP4Box](https://gpac.io/downloads/gpac-nightly-builds/)，And confirm [MP4Box](https://gpac.io/downloads/gpac-nightly-builds/) Correctly added to environment variables ### Add features From 6940c82d707ed1e5a1d6f8b0ea1bd76cef2dba85 Mon Sep 17 00:00:00 2001 From: MDSVJ Date: Sun, 15 Jun 2025 23:46:26 -0700 Subject: [PATCH 5/6] Enhanced LRC support by @AAGaming00 --- utils/lyrics/lyrics.go | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/utils/lyrics/lyrics.go b/utils/lyrics/lyrics.go index af9fde9..7473857 100644 --- a/utils/lyrics/lyrics.go +++ b/utils/lyrics/lyrics.go @@ -26,7 +26,6 @@ type SongLyrics struct { } `json:"data"` } - func Get(storefront, songId, lrcType, language, lrcFormat, token, mediaUserToken string) (string, error) { if len(mediaUserToken) < 50 { return "", errors.New("MediaUserToken not set") @@ -166,7 +165,7 @@ func conventSyllableTTMLToLRC(ttml string) (string, error) { return "", err } var lrcLines []string - parseTime := func(timeValue string) (string, error) { + parseTime := func(timeValue string, newLine bool) (string, error) { var h, m, s, ms int if strings.Contains(timeValue, ":") { _, err = fmt.Sscanf(timeValue, "%d:%d:%d.%d", &h, &m, &s, &ms) @@ -183,7 +182,11 @@ func conventSyllableTTMLToLRC(ttml string) (string, error) { } m += h * 60 ms = ms / 10 - return fmt.Sprintf("[%02d:%02d.%02d]", m, s, ms), nil + if newLine { + return fmt.Sprintf("[%02d:%02d.%02d]<%02d:%02d.%02d>", m, s, ms, m, s, ms), nil + } else { + return fmt.Sprintf("<%02d:%02d.%02d>", m, s, ms), nil + } } divs := parsedTTML.FindElement("tt").FindElement("body").FindElements("div") //get trans @@ -217,11 +220,11 @@ func conventSyllableTTMLToLRC(ttml string) (string, error) { if lyric.SelectAttr("begin") == nil { continue } - beginTime, err := parseTime(lyric.SelectAttr("begin").Value) + beginTime, err := parseTime(lyric.SelectAttr("begin").Value, i == 0) if err != nil { return "", err } - endTime, err = parseTime(lyric.SelectAttr("end").Value) + endTime, err = parseTime(lyric.SelectAttr("end").Value, false) if err != nil { return "", err } From 5d252104327509de93aa258a897cfba2a884a3cb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=96=E7=95=8C=E8=A7=82=E5=AF=9F=E6=97=A5=E5=BF=97?= Date: Tue, 24 Jun 2025 18:43:41 +0800 Subject: [PATCH 6/6] fix: unable to collect information about albums with more than 300 tracks --- main.go | 3942 +++++++++++++++++++++++++++---------------------------- 1 file changed, 1971 insertions(+), 1971 deletions(-) diff --git a/main.go b/main.go index 72417ec..3bac7e0 100644 --- a/main.go +++ b/main.go @@ -1,1971 +1,1971 @@ -package main - -import ( - "bufio" - "bytes" - "encoding/json" - "errors" - "fmt" - "io" - "log" - "net" - "net/http" - "net/url" - "os" - "os/exec" - "path/filepath" - "regexp" - "sort" - "strconv" - "strings" - "time" - - "main/utils/lyrics" - "main/utils/runv2" - "main/utils/runv3" - "main/utils/structs" - - "github.com/fatih/color" - "github.com/grafov/m3u8" - "github.com/olekukonko/tablewriter" - "github.com/spf13/pflag" - "github.com/zhaarey/go-mp4tag" - "gopkg.in/yaml.v2" -) - -var ( - forbiddenNames = regexp.MustCompile(`[/\\<>:"|?*]`) - dl_atmos bool - dl_aac bool - dl_select bool - dl_song bool - artist_select bool - debug_mode bool - alac_max *int - atmos_max *int - mv_max *int - mv_audio_type *string - aac_type *string - Config structs.ConfigSet - counter structs.Counter - okDict = make(map[string][]int) -) - -func loadConfig() error { - // 读取config.yaml文件内容 - data, err := os.ReadFile("config.yaml") - if err != nil { - return err - } - // 将yaml解析到config变量中 - err = yaml.Unmarshal(data, &Config) - if err != nil { - return err - } - return nil -} - -func LimitString(s string) string { - if len([]rune(s)) > Config.LimitMax { - return string([]rune(s)[:Config.LimitMax]) - } - return s -} - -func isInArray(arr []int, target int) bool { - for _, num := range arr { - if num == target { - return true - } - } - return false -} - -func fileExists(path string) (bool, error) { - f, err := os.Stat(path) - if err == nil { - return !f.IsDir(), nil - } else if os.IsNotExist(err) { - return false, nil - } - return false, err -} - -func checkUrl(url string) (string, string) { - pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music|classical\.music)\.apple\.com\/(\w{2})(?:\/album|\/album\/.+))\/(?:id)?(\d[^\D]+)(?:$|\?)`) - matches := pat.FindAllStringSubmatch(url, -1) - - if matches == nil { - return "", "" - } else { - return matches[0][1], matches[0][2] - } -} -func checkUrlMv(url string) (string, string) { - pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music)\.apple\.com\/(\w{2})(?:\/music-video|\/music-video\/.+))\/(?:id)?(\d[^\D]+)(?:$|\?)`) - matches := pat.FindAllStringSubmatch(url, -1) - - if matches == nil { - return "", "" - } else { - return matches[0][1], matches[0][2] - } -} -func checkUrlSong(url string) (string, string) { - pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music)\.apple\.com\/(\w{2})(?:\/song|\/song\/.+))\/(?:id)?(\d[^\D]+)(?:$|\?)`) - matches := pat.FindAllStringSubmatch(url, -1) - - if matches == nil { - return "", "" - } else { - return matches[0][1], matches[0][2] - } -} -func checkUrlPlaylist(url string) (string, string) { - pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music)\.apple\.com\/(\w{2})(?:\/playlist|\/playlist\/.+))\/(?:id)?(pl\.[\w-]+)(?:$|\?)`) - matches := pat.FindAllStringSubmatch(url, -1) - - if matches == nil { - return "", "" - } else { - return matches[0][1], matches[0][2] - } -} - -func checkUrlArtist(url string) (string, string) { - pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music)\.apple\.com\/(\w{2})(?:\/artist|\/artist\/.+))\/(?:id)?(\d[^\D]+)(?:$|\?)`) - matches := pat.FindAllStringSubmatch(url, -1) - - if matches == nil { - return "", "" - } else { - return matches[0][1], matches[0][2] - } -} -func getUrlSong(songUrl string, token string) (string, error) { - storefront, songId := checkUrlSong(songUrl) - manifest, err := getInfoFromAdam(songId, token, storefront) - if err != nil { - fmt.Println("\u26A0 Failed to get manifest:", err) - counter.NotSong++ - return "", err - } - albumId := manifest.Relationships.Albums.Data[0].ID - songAlbumUrl := fmt.Sprintf("https://music.apple.com/%s/album/1/%s?i=%s", storefront, albumId, songId) - return songAlbumUrl, nil -} -func getUrlArtistName(artistUrl string, token string) (string, string, error) { - storefront, artistId := checkUrlArtist(artistUrl) - req, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/artists/%s", storefront, artistId), 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") - query := url.Values{} - query.Set("l", Config.Language) - 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(structs.AutoGeneratedArtist) - err = json.NewDecoder(do.Body).Decode(&obj) - if err != nil { - return "", "", err - } - return obj.Data[0].Attributes.Name, obj.Data[0].ID, nil -} - -func checkArtist(artistUrl string, token string, relationship string) ([]string, error) { - storefront, artistId := checkUrlArtist(artistUrl) - Num := 0 - //id := 1 - var args []string - var urls []string - var options [][]string - for { - req, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/artists/%s/%s?limit=100&offset=%d&l=%s", storefront, artistId, relationship, Num, Config.Language), nil) - if err != nil { - return nil, 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") - do, err := http.DefaultClient.Do(req) - if err != nil { - return nil, err - } - defer do.Body.Close() - if do.StatusCode != http.StatusOK { - return nil, errors.New(do.Status) - } - obj := new(structs.AutoGeneratedArtist) - err = json.NewDecoder(do.Body).Decode(&obj) - if err != nil { - return nil, err - } - for _, album := range obj.Data { - options = append(options, []string{album.Attributes.Name, album.Attributes.ReleaseDate, album.ID, album.Attributes.URL}) - } - Num = Num + 100 - if len(obj.Next) == 0 { - break - } - } - sort.Slice(options, func(i, j int) bool { - // 将日期字符串解析为 time.Time 类型进行比较 - dateI, _ := time.Parse("2006-01-02", options[i][1]) - dateJ, _ := time.Parse("2006-01-02", options[j][1]) - return dateI.Before(dateJ) // 返回 true 表示 i 在 j 前面 - }) - - table := tablewriter.NewWriter(os.Stdout) - if relationship == "albums" { - table.SetHeader([]string{"", "Album Name", "Date", "Album ID"}) - } else if relationship == "music-videos" { - table.SetHeader([]string{"", "MV Name", "Date", "MV ID"}) - } - //table.SetFooter([]string{"", "", "Total", "$146.93"}) - //table.SetAutoMergeCells(true) - //table.SetAutoMergeCellsByColumnIndex([]int{1,2,3}) - table.SetRowLine(false) - //table.AppendBulk(options) - table.SetHeaderColor(tablewriter.Colors{}, - tablewriter.Colors{tablewriter.FgRedColor, tablewriter.Bold}, - tablewriter.Colors{tablewriter.Bold, tablewriter.FgBlackColor}, - tablewriter.Colors{tablewriter.Bold, tablewriter.FgBlackColor}) - - 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 i, v := range options { - urls = append(urls, v[3]) - options[i] = append([]string{fmt.Sprint(i + 1)}, v[:3]...) - table.Append(options[i]) - } - table.Render() - if artist_select { - fmt.Println("You have selected all options:") - return urls, nil - } - reader := bufio.NewReader(os.Stdin) - fmt.Println("Please select from the " + relationship + " options above (multiple options separated by commas, ranges supported, or type 'all' to select all)") - cyanColor := color.New(color.FgCyan) - cyanColor.Print("Enter your choice: ") - input, _ := reader.ReadString('\n') - - // Remove newline and whitespace - input = strings.TrimSpace(input) - if input == "all" { - fmt.Println("You have selected all options:") - return urls, nil - } - - // Split input into string slices - 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}) - } - } - - // Print selected options - fmt.Println("You have selected the following options:") - 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(options) { - fmt.Println(options[num-1]) - 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(options) || start > end { - fmt.Println("Range out of range:", opt) - continue - } - for i := start; i <= end; i++ { - fmt.Println(options[i-1]) - args = append(args, urls[i-1]) - } - } else { - fmt.Println("Invalid option:", opt) - } - } - return args, nil -} - -func getMeta(albumId string, token string, storefront string) (*structs.AutoGenerated, error) { - var mtype string - var next string - if strings.Contains(albumId, "pl.") { - mtype = "playlists" - } else { - mtype = "albums" - } - req, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/%s/%s", storefront, mtype, albumId), nil) - if err != nil { - return nil, 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") - query := url.Values{} - query.Set("omit[resource]", "autos") - query.Set("include", "tracks,artists,record-labels") - query.Set("include[songs]", "artists,albums") - query.Set("fields[artists]", "name,artwork") - query.Set("fields[albums:albums]", "artistName,artwork,name,releaseDate,url") - query.Set("fields[record-labels]", "name") - query.Set("extend", "editorialVideo") - query.Set("l", Config.Language) - req.URL.RawQuery = query.Encode() - do, err := http.DefaultClient.Do(req) - if err != nil { - return nil, err - } - defer do.Body.Close() - if do.StatusCode != http.StatusOK { - return nil, errors.New(do.Status) - } - obj := new(structs.AutoGenerated) - err = json.NewDecoder(do.Body).Decode(&obj) - if err != nil { - return nil, err - } - if strings.Contains(albumId, "pl.") { - obj.Data[0].Attributes.ArtistName = "Apple Music" - if len(obj.Data[0].Relationships.Tracks.Next) > 0 { - next = obj.Data[0].Relationships.Tracks.Next - for { - req, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/%s&l=%s&include=albums", next, Config.Language), nil) - if err != nil { - return nil, 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") - do, err := http.DefaultClient.Do(req) - if err != nil { - return nil, err - } - defer do.Body.Close() - if do.StatusCode != http.StatusOK { - return nil, errors.New(do.Status) - } - obj2 := new(structs.AutoGeneratedTrack) - err = json.NewDecoder(do.Body).Decode(&obj2) - if err != nil { - return nil, err - } - for _, value := range obj2.Data { - obj.Data[0].Relationships.Tracks.Data = append(obj.Data[0].Relationships.Tracks.Data, value) - } - next = obj2.Next - if len(next) == 0 { - break - } - } - } - } - return obj, nil -} - -func writeCover(sanAlbumFolder, name string, url string) (string, error) { - covPath := filepath.Join(sanAlbumFolder, name+"."+Config.CoverFormat) - if Config.CoverFormat == "original" { - ext := strings.Split(url, "/")[len(strings.Split(url, "/"))-2] - ext = ext[strings.LastIndex(ext, ".")+1:] - covPath = filepath.Join(sanAlbumFolder, name+"."+ext) - } - exists, err := fileExists(covPath) - if err != nil { - fmt.Println("Failed to check if cover exists.") - return "", err - } - if exists { - _ = os.Remove(covPath) - } - if Config.CoverFormat == "png" { - re := regexp.MustCompile(`\{w\}x\{h\}`) - parts := re.Split(url, 2) - url = parts[0] + "{w}x{h}" + strings.Replace(parts[1], ".jpg", ".png", 1) - } - url = strings.Replace(url, "{w}x{h}", Config.CoverSize, 1) - if Config.CoverFormat == "original" { - url = strings.Replace(url, "is1-ssl.mzstatic.com/image/thumb", "a5.mzstatic.com/us/r1000/0", 1) - url = url[:strings.LastIndex(url, "/")] - } - req, err := http.NewRequest("GET", url, nil) - if err != nil { - return "", err - } - 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") - 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) - } - f, err := os.Create(covPath) - if err != nil { - return "", err - } - defer f.Close() - _, err = io.Copy(f, do.Body) - if err != nil { - return "", err - } - return covPath, nil -} - -func writeLyrics(sanAlbumFolder, filename string, lrc string) error { - lyricspath := filepath.Join(sanAlbumFolder, filename) - f, err := os.Create(lyricspath) - if err != nil { - return err - } - defer f.Close() - _, err = f.WriteString(lrc) - if err != nil { - return err - } - return nil -} - -func contains(slice []string, item string) bool { - for _, v := range slice { - if v == item { - return true - } - } - return false -} - -// 下载单曲逻辑 -func downloadTrack(trackNum int, trackTotal int, meta *structs.AutoGenerated, track structs.TrackData, albumId, token, storefront, mediaUserToken, sanAlbumFolder, Codec string, covPath string) { - counter.Total++ - fmt.Printf("Track %d of %d:\n", trackNum, trackTotal) - - //mv dl dev - if track.Type == "music-videos" { - if len(mediaUserToken) <= 50 { - fmt.Println("meida-user-token is not set, skip MV dl") - counter.Success++ - return - } - if _, err := exec.LookPath("mp4decrypt"); err != nil { - fmt.Println("mp4decrypt is not found, skip MV dl") - counter.Success++ - return - } - err := mvDownloader(track.ID, sanAlbumFolder, token, storefront, mediaUserToken, meta) - if err != nil { - fmt.Println("\u26A0 Failed to dl MV:", err) - counter.Error++ - return - } - counter.Success++ - return - } - - manifest, err := getInfoFromAdam(track.ID, token, storefront) - if err != nil { - fmt.Println("\u26A0 Failed to get manifest:", err) - counter.NotSong++ - return - } - needDlAacLc := false - if dl_aac && Config.AacType == "aac-lc" { - needDlAacLc = true - } - if manifest.Attributes.ExtendedAssetUrls.EnhancedHls == "" { - if dl_atmos { - fmt.Println("Unavailable") - counter.Unavailable++ - return - } - fmt.Println("Unavailable, Try DL AAC-LC") - needDlAacLc = true - } - needCheck := false - - if Config.GetM3u8Mode == "all" { - needCheck = true - } else if Config.GetM3u8Mode == "hires" && contains(track.Attributes.AudioTraits, "hi-res-lossless") { - needCheck = true - } - var EnhancedHls_m3u8 string - if needCheck && !needDlAacLc { - EnhancedHls_m3u8, _ = checkM3u8(track.ID, "song") - if strings.HasSuffix(EnhancedHls_m3u8, ".m3u8") { - manifest.Attributes.ExtendedAssetUrls.EnhancedHls = EnhancedHls_m3u8 - } - } - var Quality string - if strings.Contains(Config.SongFileFormat, "Quality") { - if dl_atmos { - Quality = fmt.Sprintf("%dkbps", Config.AtmosMax-2000) - } else if needDlAacLc { - Quality = "256kbps" - } else { - _, Quality, err = extractMedia(manifest.Attributes.ExtendedAssetUrls.EnhancedHls, true) - if err != nil { - fmt.Println("Failed to extract quality from manifest.\n", err) - counter.Error++ - return - } - } - } - stringsToJoin := []string{} - if track.Attributes.IsAppleDigitalMaster { - if Config.AppleMasterChoice != "" { - stringsToJoin = append(stringsToJoin, Config.AppleMasterChoice) - } - } - if track.Attributes.ContentRating == "explicit" { - if Config.ExplicitChoice != "" { - stringsToJoin = append(stringsToJoin, Config.ExplicitChoice) - } - } - if track.Attributes.ContentRating == "clean" { - if Config.CleanChoice != "" { - stringsToJoin = append(stringsToJoin, Config.CleanChoice) - } - } - Tag_string := strings.Join(stringsToJoin, " ") - - songName := strings.NewReplacer( - "{SongId}", track.ID, - "{SongNumer}", fmt.Sprintf("%02d", trackNum), - "{SongName}", LimitString(track.Attributes.Name), - "{DiscNumber}", fmt.Sprintf("%0d", track.Attributes.DiscNumber), - "{TrackNumber}", fmt.Sprintf("%0d", track.Attributes.TrackNumber), - "{Quality}", Quality, - "{Tag}", Tag_string, - "{Codec}", Codec, - ).Replace(Config.SongFileFormat) - fmt.Println(songName) - filename := fmt.Sprintf("%s.m4a", forbiddenNames.ReplaceAllString(songName, "_")) - lrcFilename := fmt.Sprintf("%s.%s", forbiddenNames.ReplaceAllString(songName, "_"), Config.LrcFormat) - trackPath := filepath.Join(sanAlbumFolder, filename) - - //get lrc - var lrc string = "" - if Config.EmbedLrc || Config.SaveLrcFile { - lrcStr, err := lyrics.Get(storefront, track.ID, Config.LrcType, Config.Language, Config.LrcFormat, token, mediaUserToken) - if err != nil { - fmt.Println(err) - } else { - if Config.SaveLrcFile { - err := writeLyrics(sanAlbumFolder, lrcFilename, lrcStr) - if err != nil { - fmt.Printf("Failed to write lyrics") - } - } - if Config.EmbedLrc { - lrc = lrcStr - } - } - } - - exists, err := fileExists(trackPath) - if err != nil { - fmt.Println("Failed to check if track exists.") - } - if exists { - fmt.Println("Track already exists locally.") - counter.Success++ - okDict[albumId] = append(okDict[albumId], trackNum) - return - } - if needDlAacLc { - if len(mediaUserToken) <= 50 { - fmt.Println("Invalid media-user-token") - counter.Error++ - return - } - _, err := runv3.Run(track.ID, trackPath, token, mediaUserToken, false) - if err != nil { - fmt.Println("Failed to dl aac-lc:", err) - counter.Error++ - return - } - } else { - trackM3u8Url, _, err := extractMedia(manifest.Attributes.ExtendedAssetUrls.EnhancedHls, false) - if err != nil { - fmt.Println("\u26A0 Failed to extract info from manifest:", err) - counter.Unavailable++ - return - } - //边下载边解密 - err = runv2.Run(track.ID, trackM3u8Url, trackPath, Config) - if err != nil { - fmt.Println("Failed to run v2:", err) - counter.Error++ - return - } - } - tags := []string{ - "tool=", - fmt.Sprintf("artist=%s", meta.Data[0].Attributes.ArtistName), - //fmt.Sprintf("lyrics=%s", lrc), - } - var trackCovPath string - if Config.EmbedCover { - if strings.Contains(albumId, "pl.") && Config.DlAlbumcoverForPlaylist { - trackCovPath, err = writeCover(sanAlbumFolder, track.ID, track.Attributes.Artwork.URL) - if err != nil { - fmt.Println("Failed to write cover.") - } - tags = append(tags, fmt.Sprintf("cover=%s", trackCovPath)) - } else { - tags = append(tags, fmt.Sprintf("cover=%s", covPath)) - } - } - tagsString := strings.Join(tags, ":") - cmd := exec.Command("MP4Box", "-itags", tagsString, trackPath) - if err := cmd.Run(); err != nil { - fmt.Printf("Embed failed: %v\n", err) - counter.Error++ - return - } - if strings.Contains(albumId, "pl.") && Config.DlAlbumcoverForPlaylist && trackCovPath != "" { - if err := os.Remove(trackCovPath); err != nil { - fmt.Printf("Error deleting file: %s\n", trackCovPath) - counter.Error++ - return - } - } - err = writeMP4Tags(trackPath, lrc, meta, trackNum, trackTotal) - if err != nil { - fmt.Println("\u26A0 Failed to write tags in media:", err) - counter.Unavailable++ - return - } - counter.Success++ - okDict[albumId] = append(okDict[albumId], trackNum) -} - -func rip(albumId string, token string, storefront string, mediaUserToken string, urlArg_i string) error { - meta, err := getMeta(albumId, token, storefront) - if err != nil { - return err - } - - if debug_mode { - // Print album info - fmt.Println(meta.Data[0].Attributes.ArtistName) - fmt.Println(meta.Data[0].Attributes.Name) - - for trackNum, track := range meta.Data[0].Relationships.Tracks.Data { - trackNum++ - fmt.Printf("\nTrack %d of %d:\n", trackNum, len(meta.Data[0].Relationships.Tracks.Data)) - fmt.Printf("%02d. %s\n", trackNum, track.Attributes.Name) - - manifest, err := getInfoFromAdam(track.ID, token, storefront) - if err != nil { - fmt.Printf("Failed to get manifest for track %d: %v\n", trackNum, err) - continue - } - - var m3u8Url string - //Web端m3u8 - if manifest.Attributes.ExtendedAssetUrls.EnhancedHls != "" { - m3u8Url = manifest.Attributes.ExtendedAssetUrls.EnhancedHls - } - //设备端满血m3u8 - needCheck := false - if Config.GetM3u8Mode == "all" { - needCheck = true - } else if Config.GetM3u8Mode == "hires" && contains(track.Attributes.AudioTraits, "hi-res-lossless") { - needCheck = true - } - if needCheck { - fullM3u8Url, err := checkM3u8(track.ID, "song") - if err == nil && strings.HasSuffix(fullM3u8Url, ".m3u8") { - m3u8Url = fullM3u8Url - } else { - fmt.Println("Failed to get best quality m3u8 from device m3u8 port, will use m3u8 from Web API") - } - } - - _, _, err = extractMedia(m3u8Url, true) - if err != nil { - fmt.Printf("Failed to extract quality info for track %d: %v\n", trackNum, err) - continue - } - } - return nil // Return directly without showing statistics - } - var Codec string - if dl_atmos { - Codec = "ATMOS" - } else if dl_aac { - Codec = "AAC" - } else { - Codec = "ALAC" - } - var singerFoldername string - if Config.ArtistFolderFormat != "" { - if strings.Contains(albumId, "pl.") { - singerFoldername = strings.NewReplacer( - "{ArtistName}", "Apple Music", - "{ArtistId}", "", - "{UrlArtistName}", "Apple Music", - ).Replace(Config.ArtistFolderFormat) - } else if len(meta.Data[0].Relationships.Artists.Data) > 0 { - singerFoldername = strings.NewReplacer( - "{UrlArtistName}", LimitString(meta.Data[0].Attributes.ArtistName), - "{ArtistName}", LimitString(meta.Data[0].Attributes.ArtistName), - "{ArtistId}", meta.Data[0].Relationships.Artists.Data[0].ID, - ).Replace(Config.ArtistFolderFormat) - } else { - singerFoldername = strings.NewReplacer( - "{UrlArtistName}", LimitString(meta.Data[0].Attributes.ArtistName), - "{ArtistName}", LimitString(meta.Data[0].Attributes.ArtistName), - "{ArtistId}", "", - ).Replace(Config.ArtistFolderFormat) - } - if strings.HasSuffix(singerFoldername, ".") { - singerFoldername = strings.ReplaceAll(singerFoldername, ".", "") - } - singerFoldername = strings.TrimSpace(singerFoldername) - fmt.Println(singerFoldername) - } - singerFolder := filepath.Join(Config.AlacSaveFolder, forbiddenNames.ReplaceAllString(singerFoldername, "_")) - if dl_atmos { - singerFolder = filepath.Join(Config.AtmosSaveFolder, forbiddenNames.ReplaceAllString(singerFoldername, "_")) - } - var Quality string - if strings.Contains(Config.AlbumFolderFormat, "Quality") { - if dl_atmos { - Quality = fmt.Sprintf("%dkbps", Config.AtmosMax-2000) - } else if dl_aac && Config.AacType == "aac-lc" { - Quality = "256kbps" - } else { - manifest1, err := getInfoFromAdam(meta.Data[0].Relationships.Tracks.Data[0].ID, token, storefront) - if err != nil { - fmt.Println("Failed to get manifest.\n", err) - } else { - if manifest1.Attributes.ExtendedAssetUrls.EnhancedHls == "" { - Codec = "AAC" - Quality = "256kbps" - //fmt.Println("Unavailable.\n") - } else { - needCheck := false - - if Config.GetM3u8Mode == "all" { - needCheck = true - } else if Config.GetM3u8Mode == "hires" && contains(meta.Data[0].Relationships.Tracks.Data[0].Attributes.AudioTraits, "hi-res-lossless") { - needCheck = true - } - var EnhancedHls_m3u8 string - if needCheck { - EnhancedHls_m3u8, _ = checkM3u8(meta.Data[0].Relationships.Tracks.Data[0].ID, "album") - if strings.HasSuffix(EnhancedHls_m3u8, ".m3u8") { - manifest1.Attributes.ExtendedAssetUrls.EnhancedHls = EnhancedHls_m3u8 - } - } - _, Quality, err = extractMedia(manifest1.Attributes.ExtendedAssetUrls.EnhancedHls, true) - if err != nil { - fmt.Println("Failed to extract quality from manifest.\n", err) - } - } - } - } - } - stringsToJoin := []string{} - if meta.Data[0].Attributes.IsAppleDigitalMaster || meta.Data[0].Attributes.IsMasteredForItunes { - if Config.AppleMasterChoice != "" { - stringsToJoin = append(stringsToJoin, Config.AppleMasterChoice) - } - } - if meta.Data[0].Attributes.ContentRating == "explicit" { - if Config.ExplicitChoice != "" { - stringsToJoin = append(stringsToJoin, Config.ExplicitChoice) - } - } - if meta.Data[0].Attributes.ContentRating == "clean" { - if Config.CleanChoice != "" { - stringsToJoin = append(stringsToJoin, Config.CleanChoice) - } - } - Tag_string := strings.Join(stringsToJoin, " ") - var albumFolder string - if strings.Contains(albumId, "pl.") { - albumFolder = strings.NewReplacer( - "{ArtistName}", "Apple Music", - "{PlaylistName}", LimitString(meta.Data[0].Attributes.Name), - "{PlaylistId}", albumId, - "{Quality}", Quality, - "{Codec}", Codec, - "{Tag}", Tag_string, - ).Replace(Config.PlaylistFolderFormat) - } else { - albumFolder = strings.NewReplacer( - "{ReleaseDate}", meta.Data[0].Attributes.ReleaseDate, - "{ReleaseYear}", meta.Data[0].Attributes.ReleaseDate[:4], - "{ArtistName}", LimitString(meta.Data[0].Attributes.ArtistName), - "{AlbumName}", LimitString(meta.Data[0].Attributes.Name), - "{UPC}", meta.Data[0].Attributes.Upc, - "{RecordLabel}", meta.Data[0].Attributes.RecordLabel, - "{Copyright}", meta.Data[0].Attributes.Copyright, - "{AlbumId}", albumId, - "{Quality}", Quality, - "{Codec}", Codec, - "{Tag}", Tag_string, - ).Replace(Config.AlbumFolderFormat) - } - if strings.HasSuffix(albumFolder, ".") { - albumFolder = strings.ReplaceAll(albumFolder, ".", "") - } - albumFolder = strings.TrimSpace(albumFolder) - sanAlbumFolder := filepath.Join(singerFolder, forbiddenNames.ReplaceAllString(albumFolder, "_")) - os.MkdirAll(sanAlbumFolder, os.ModePerm) - fmt.Println(albumFolder) - //get artist cover - if Config.SaveArtistCover && !(strings.Contains(albumId, "pl.")) { - if len(meta.Data[0].Relationships.Artists.Data) > 0 { - _, err = writeCover(singerFolder, "folder", meta.Data[0].Relationships.Artists.Data[0].Attributes.Artwork.Url) - if err != nil { - fmt.Println("Failed to write artist cover.") - } - } - } - //get album cover - covPath, err := writeCover(sanAlbumFolder, "cover", meta.Data[0].Attributes.Artwork.URL) - if err != nil { - fmt.Println("Failed to write cover.") - } - //get animated artwork - if Config.SaveAnimatedArtwork && meta.Data[0].Attributes.EditorialVideo.MotionDetailSquare.Video != "" { - fmt.Println("Found Animation Artwork.") - - // Download square version - motionvideoUrlSquare, err := extractVideo(meta.Data[0].Attributes.EditorialVideo.MotionDetailSquare.Video) - if err != nil { - fmt.Println("no motion video square.\n", err) - } else { - exists, err := fileExists(filepath.Join(sanAlbumFolder, "square_animated_artwork.mp4")) - if err != nil { - fmt.Println("Failed to check if animated artwork square exists.") - } - if exists { - fmt.Println("Animated artwork square already exists locally.") - } else { - fmt.Println("Animation Artwork Square Downloading...") - cmd := exec.Command("ffmpeg", "-loglevel", "quiet", "-y", "-i", motionvideoUrlSquare, "-c", "copy", filepath.Join(sanAlbumFolder, "square_animated_artwork.mp4")) - if err := cmd.Run(); err != nil { - fmt.Printf("animated artwork square dl err: %v\n", err) - } else { - fmt.Println("Animation Artwork Square Downloaded") - } - } - } - - if Config.EmbyAnimatedArtwork { - // Convert square version to gif - cmd3 := exec.Command("ffmpeg", "-i", filepath.Join(sanAlbumFolder, "square_animated_artwork.mp4"), "-vf", "scale=440:-1", "-r", "24", "-f", "gif", filepath.Join(sanAlbumFolder, "folder.jpg")) - if err := cmd3.Run(); err != nil { - fmt.Printf("animated artwork square to gif err: %v\n", err) - } - } - - // Download tall version - motionvideoUrlTall, err := extractVideo(meta.Data[0].Attributes.EditorialVideo.MotionDetailTall.Video) - if err != nil { - fmt.Println("no motion video tall.\n", err) - } else { - exists, err := fileExists(filepath.Join(sanAlbumFolder, "tall_animated_artwork.mp4")) - if err != nil { - fmt.Println("Failed to check if animated artwork tall exists.") - } - if exists { - fmt.Println("Animated artwork tall already exists locally.") - } else { - fmt.Println("Animation Artwork Tall Downloading...") - cmd := exec.Command("ffmpeg", "-loglevel", "quiet", "-y", "-i", motionvideoUrlTall, "-c", "copy", filepath.Join(sanAlbumFolder, "tall_animated_artwork.mp4")) - if err := cmd.Run(); err != nil { - fmt.Printf("animated artwork tall dl err: %v\n", err) - } else { - fmt.Println("Animation Artwork Tall Downloaded") - } - } - } - } - trackTotal := len(meta.Data[0].Relationships.Tracks.Data) - arr := make([]int, trackTotal) - for i := 0; i < trackTotal; i++ { - arr[i] = i + 1 - } - selected := []int{} - - if dl_song { - if urlArg_i == "" { - //fmt.Println("URL does not contain parameter 'i'. Please ensure the URL includes 'i' or use another mode.") - //return nil - } else { - for trackNum, track := range meta.Data[0].Relationships.Tracks.Data { - trackNum++ - if urlArg_i == track.ID { - downloadTrack(trackNum, trackTotal, meta, track, albumId, token, storefront, mediaUserToken, sanAlbumFolder, Codec, covPath) - return nil - } - } - } - return nil - } - - if !dl_select { - selected = arr - } else { - var data [][]string - for trackNum, track := range meta.Data[0].Relationships.Tracks.Data { - trackNum++ - var trackName string - if meta.Data[0].Type == "albums" { - trackName = fmt.Sprintf("%02d. %s", track.Attributes.TrackNumber, track.Attributes.Name) - } else { - trackName = fmt.Sprintf("%s - %s", track.Attributes.Name, track.Attributes.ArtistName) - } - 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(meta.Data[0].Type == "albums", fmt.Sprintf("Storefront: %s, %d tracks missing", strings.ToUpper(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) - } - } - } - fmt.Println("Selected options:", selected) - } - for trackNum, track := range meta.Data[0].Relationships.Tracks.Data { - trackNum++ - if isInArray(okDict[albumId], trackNum) { - //fmt.Println("已完成直接跳过.\n") - counter.Total++ - counter.Success++ - continue - } - if isInArray(selected, trackNum) { - downloadTrack(trackNum, trackTotal, meta, track, albumId, token, storefront, mediaUserToken, sanAlbumFolder, Codec, covPath) - } - } - return nil -} - -func writeMP4Tags(trackPath, lrc string, meta *structs.AutoGenerated, trackNum, trackTotal int) error { - index := trackNum - 1 - - t := &mp4tag.MP4Tags{ - Title: meta.Data[0].Relationships.Tracks.Data[index].Attributes.Name, - TitleSort: meta.Data[0].Relationships.Tracks.Data[index].Attributes.Name, - Artist: meta.Data[0].Relationships.Tracks.Data[index].Attributes.ArtistName, - ArtistSort: meta.Data[0].Relationships.Tracks.Data[index].Attributes.ArtistName, - Custom: map[string]string{ - "PERFORMER": meta.Data[0].Relationships.Tracks.Data[index].Attributes.ArtistName, - "RELEASETIME": meta.Data[0].Relationships.Tracks.Data[index].Attributes.ReleaseDate, - "ISRC": meta.Data[0].Relationships.Tracks.Data[index].Attributes.Isrc, - "LABEL": meta.Data[0].Attributes.RecordLabel, - "UPC": meta.Data[0].Attributes.Upc, - }, - Composer: meta.Data[0].Relationships.Tracks.Data[index].Attributes.ComposerName, - ComposerSort: meta.Data[0].Relationships.Tracks.Data[index].Attributes.ComposerName, - Date: meta.Data[0].Attributes.ReleaseDate, - CustomGenre: meta.Data[0].Relationships.Tracks.Data[index].Attributes.GenreNames[0], - Copyright: meta.Data[0].Attributes.Copyright, - Publisher: meta.Data[0].Attributes.RecordLabel, - Lyrics: lrc, - } - - if !strings.Contains(meta.Data[0].ID, "pl.") { - albumID, err := strconv.ParseUint(meta.Data[0].ID, 10, 32) - if err != nil { - return err - } - t.ItunesAlbumID = int32(albumID) - } - - if len(meta.Data[0].Relationships.Artists.Data) > 0 { - if len(meta.Data[0].Relationships.Tracks.Data[index].Relationships.Artists.Data) > 0 { - artistID, err := strconv.ParseUint(meta.Data[0].Relationships.Tracks.Data[index].Relationships.Artists.Data[0].ID, 10, 32) - if err != nil { - return err - } - t.ItunesArtistID = int32(artistID) - } - } - - if strings.Contains(meta.Data[0].ID, "pl.") && !Config.UseSongInfoForPlaylist { - t.DiscNumber = 1 - t.DiscTotal = 1 - t.TrackNumber = int16(trackNum) - t.TrackTotal = int16(trackTotal) - t.Album = meta.Data[0].Attributes.Name - t.AlbumSort = meta.Data[0].Attributes.Name - t.AlbumArtist = meta.Data[0].Attributes.ArtistName - t.AlbumArtistSort = meta.Data[0].Attributes.ArtistName - } else if strings.Contains(meta.Data[0].ID, "pl.") && Config.UseSongInfoForPlaylist { - t.DiscNumber = int16(meta.Data[0].Relationships.Tracks.Data[index].Attributes.DiscNumber) - t.DiscTotal = int16(meta.Data[0].Relationships.Tracks.Data[trackTotal-1].Attributes.DiscNumber) - t.TrackNumber = int16(meta.Data[0].Relationships.Tracks.Data[index].Attributes.TrackNumber) - t.TrackTotal = int16(trackTotal) - t.Album = meta.Data[0].Relationships.Tracks.Data[index].Attributes.AlbumName - t.AlbumSort = meta.Data[0].Relationships.Tracks.Data[index].Attributes.AlbumName - t.AlbumArtist = meta.Data[0].Relationships.Tracks.Data[index].Relationships.Albums.Data[0].Attributes.ArtistName - t.AlbumArtistSort = meta.Data[0].Relationships.Tracks.Data[index].Relationships.Albums.Data[0].Attributes.ArtistName - } else { - t.DiscNumber = int16(meta.Data[0].Relationships.Tracks.Data[index].Attributes.DiscNumber) - t.DiscTotal = int16(meta.Data[0].Relationships.Tracks.Data[trackTotal-1].Attributes.DiscNumber) - t.TrackNumber = int16(meta.Data[0].Relationships.Tracks.Data[index].Attributes.TrackNumber) - t.TrackTotal = int16(trackTotal) - t.Album = meta.Data[0].Relationships.Tracks.Data[index].Attributes.AlbumName - t.AlbumSort = meta.Data[0].Relationships.Tracks.Data[index].Attributes.AlbumName - t.AlbumArtist = meta.Data[0].Attributes.ArtistName - t.AlbumArtistSort = meta.Data[0].Attributes.ArtistName - } - - if meta.Data[0].Relationships.Tracks.Data[index].Attributes.ContentRating == "explicit" { - t.ItunesAdvisory = mp4tag.ItunesAdvisoryExplicit - } else if meta.Data[0].Relationships.Tracks.Data[index].Attributes.ContentRating == "clean" { - t.ItunesAdvisory = mp4tag.ItunesAdvisoryClean - } else { - t.ItunesAdvisory = mp4tag.ItunesAdvisoryNone - } - - mp4, err := mp4tag.Open(trackPath) - if err != nil { - return err - } - defer mp4.Close() - err = mp4.Write(t, []string{}) - if err != nil { - return err - } - return nil -} - -func main() { - err := loadConfig() - if err != nil { - fmt.Printf("load Config failed: %v", err) - return - } - token, err := getToken() - if err != nil { - if Config.AuthorizationToken != "" && Config.AuthorizationToken != "your-authorization-token" { - token = strings.Replace(Config.AuthorizationToken, "Bearer ", "", -1) - } else { - fmt.Println("Failed to get token.") - return - } - } - // Define command-line flags - pflag.BoolVar(&dl_atmos, "atmos", false, "Enable atmos download mode") - pflag.BoolVar(&dl_aac, "aac", false, "Enable adm-aac download mode") - pflag.BoolVar(&dl_select, "select", false, "Enable selective download") - pflag.BoolVar(&dl_song, "song", false, "Enable single song download mode") - pflag.BoolVar(&artist_select, "all-album", false, "Download all artist albums") - pflag.BoolVar(&debug_mode, "debug", false, "Enable debug mode to show audio quality information") - alac_max = pflag.Int("alac-max", Config.AlacMax, "Specify the max quality for download alac") - atmos_max = pflag.Int("atmos-max", Config.AtmosMax, "Specify the max quality for download atmos") - aac_type = pflag.String("aac-type", Config.AacType, "Select AAC type, aac aac-binaural aac-downmix") - mv_audio_type = pflag.String("mv-audio-type", Config.MVAudioType, "Select MV audio type, atmos ac3 aac") - mv_max = pflag.Int("mv-max", Config.MVMax, "Specify the max quality for download MV") - - // Custom usage message for help - pflag.Usage = func() { - fmt.Fprintf(os.Stderr, "Usage: %s [options] url1 url2 ...\n", "[main | main.exe | go run main.go]") - fmt.Println("Options:") - pflag.PrintDefaults() - } - - // Parse the flag arguments - pflag.Parse() - Config.AlacMax = *alac_max - Config.AtmosMax = *atmos_max - Config.AacType = *aac_type - Config.MVAudioType = *mv_audio_type - Config.MVMax = *mv_max - - args := pflag.Args() - if len(args) == 0 { - fmt.Println("No URLs provided. Please provide at least one URL.") - pflag.Usage() - return - } - os.Args = args - if strings.Contains(os.Args[0], "/artist/") { - urlArtistName, urlArtistID, err := getUrlArtistName(os.Args[0], token) - if err != nil { - fmt.Println("Failed to get artistname.") - return - } - //fmt.Println("get artistname:", urlArtistName) - Config.ArtistFolderFormat = strings.NewReplacer( - "{UrlArtistName}", LimitString(urlArtistName), - "{ArtistId}", urlArtistID, - ).Replace(Config.ArtistFolderFormat) - albumArgs, err := checkArtist(os.Args[0], token, "albums") - if err != nil { - fmt.Println("Failed to get artist albums.") - return - } - mvArgs, err := checkArtist(os.Args[0], token, "music-videos") - if err != nil { - fmt.Println("Failed to get artist music-videos.") - //return - } - os.Args = append(albumArgs, mvArgs...) - } - albumTotal := len(os.Args) - for { - for albumNum, urlRaw := range os.Args { - fmt.Printf("Album %d of %d:\n", albumNum+1, albumTotal) - var storefront, albumId string - //mv dl dev - if strings.Contains(urlRaw, "/music-video/") { - if debug_mode { - continue - } - counter.Total++ - if len(Config.MediaUserToken) <= 50 { - fmt.Println("meida-user-token is not set, skip MV dl") - counter.Success++ - continue - } - if _, err := exec.LookPath("mp4decrypt"); err != nil { - fmt.Println("mp4decrypt is not found, skip MV dl") - counter.Success++ - continue - } - mvSaveDir := strings.NewReplacer( - "{ArtistName}", "", - "{UrlArtistName}", "", - "{ArtistId}", "", - ).Replace(Config.ArtistFolderFormat) - if mvSaveDir != "" { - mvSaveDir = filepath.Join(Config.AlacSaveFolder, forbiddenNames.ReplaceAllString(mvSaveDir, "_")) - } else { - mvSaveDir = Config.AlacSaveFolder - } - storefront, albumId = checkUrlMv(urlRaw) - err := mvDownloader(albumId, mvSaveDir, token, storefront, Config.MediaUserToken, nil) - if err != nil { - fmt.Println("\u26A0 Failed to dl MV:", err) - counter.Error++ - continue - } - counter.Success++ - continue - } - if strings.Contains(urlRaw, "/song/") { - urlRaw, err = getUrlSong(urlRaw, token) - dl_song = true - if err != nil { - fmt.Println("Failed to get Song info.") - } - } - if strings.Contains(urlRaw, "/playlist/") { - storefront, albumId = checkUrlPlaylist(urlRaw) - } else { - storefront, albumId = checkUrl(urlRaw) - } - if albumId == "" { - fmt.Printf("Invalid URL: %s\n", urlRaw) - continue - } - parse, err := url.Parse(urlRaw) - if err != nil { - log.Fatalf("Invalid URL: %v", err) - } - var urlArg_i = parse.Query().Get("i") - err = rip(albumId, token, storefront, Config.MediaUserToken, urlArg_i) - if err != nil { - fmt.Println("Album failed.") - fmt.Println(err) - } - } - fmt.Printf("======= [\u2714 ] Completed: %d/%d | [\u26A0 ] Warnings: %d | [\u2716 ] Errors: %d =======\n", counter.Success, counter.Total, counter.Unavailable+counter.NotSong, counter.Error) - if counter.Error == 0 { - break - } - fmt.Println("Error detected, press Enter to try again...") - fmt.Scanln() - fmt.Println("Start trying again...") - counter = structs.Counter{} - } -} -func mvDownloader(adamID string, saveDir string, token string, storefront string, mediaUserToken string, meta *structs.AutoGenerated) error { - MVInfo, err := getMVInfoFromAdam(adamID, token, storefront) - if err != nil { - fmt.Println("\u26A0 Failed to get MV manifest:", err) - return nil - } - - //获取传入的专辑信息当中该mv所在的位置 - var trackTotal int - var trackNum int - var index int - if meta != nil { - trackTotal = len(meta.Data[0].Relationships.Tracks.Data) - for i, track := range meta.Data[0].Relationships.Tracks.Data { - if adamID == track.ID { - index = i - trackNum = i + 1 - } - } - } - - if strings.HasSuffix(saveDir, ".") { - saveDir = strings.ReplaceAll(saveDir, ".", "") - } - saveDir = strings.TrimSpace(saveDir) - - vidPath := filepath.Join(saveDir, fmt.Sprintf("%s_vid.mp4", adamID)) - audPath := filepath.Join(saveDir, fmt.Sprintf("%s_aud.mp4", adamID)) - mvSaveName := fmt.Sprintf("%s (%s)", MVInfo.Data[0].Attributes.Name, adamID) - if meta != nil { - mvSaveName = fmt.Sprintf("%02d. %s", trackNum, MVInfo.Data[0].Attributes.Name) - } - - mvOutPath := filepath.Join(saveDir, fmt.Sprintf("%s.mp4", forbiddenNames.ReplaceAllString(mvSaveName, "_"))) - - fmt.Println(MVInfo.Data[0].Attributes.Name) - - exists, _ := fileExists(mvOutPath) - if exists { - fmt.Println("MV already exists locally.") - return nil - } - - mvm3u8url, _, _ := runv3.GetWebplayback(adamID, token, mediaUserToken, true) - if mvm3u8url == "" { - return errors.New("media-user-token may wrong or expired") - } - - os.MkdirAll(saveDir, os.ModePerm) - //video - videom3u8url, _ := extractVideo(mvm3u8url) - videokeyAndUrls, _ := runv3.Run(adamID, videom3u8url, token, mediaUserToken, true) - _ = runv3.ExtMvData(videokeyAndUrls, vidPath) - //audio - audiom3u8url, _ := extractMvAudio(mvm3u8url) - audiokeyAndUrls, _ := runv3.Run(adamID, audiom3u8url, token, mediaUserToken, true) - _ = runv3.ExtMvData(audiokeyAndUrls, audPath) - - //tags - tags := []string{ - "tool=", - fmt.Sprintf("artist=%s", MVInfo.Data[0].Attributes.ArtistName), - fmt.Sprintf("title=%s", MVInfo.Data[0].Attributes.Name), - fmt.Sprintf("genre=%s", MVInfo.Data[0].Attributes.GenreNames[0]), - fmt.Sprintf("created=%s", MVInfo.Data[0].Attributes.ReleaseDate), - fmt.Sprintf("ISRC=%s", MVInfo.Data[0].Attributes.Isrc), - } - - // ContentRating tag - if MVInfo.Data[0].Attributes.ContentRating == "explicit" { - tags = append(tags, "rating=1") - } else if MVInfo.Data[0].Attributes.ContentRating == "clean" { - tags = append(tags, "rating=2") - } else { - tags = append(tags, "rating=0") - } - - //根据情况额外添加可使用的tags - if meta != nil { - if meta.Data[0].Type == "playlists" && !Config.UseSongInfoForPlaylist { - tags = append(tags, "disk=1/1") - tags = append(tags, fmt.Sprintf("album=%s", meta.Data[0].Attributes.Name)) - tags = append(tags, fmt.Sprintf("track=%d", trackNum)) - tags = append(tags, fmt.Sprintf("tracknum=%d/%d", trackNum, trackTotal)) - tags = append(tags, fmt.Sprintf("album_artist=%s", meta.Data[0].Attributes.ArtistName)) - tags = append(tags, fmt.Sprintf("performer=%s", meta.Data[0].Relationships.Tracks.Data[index].Attributes.ArtistName)) - tags = append(tags, fmt.Sprintf("copyright=%s", meta.Data[0].Attributes.Copyright)) - tags = append(tags, fmt.Sprintf("UPC=%s", meta.Data[0].Attributes.Upc)) - } else { - tags = append(tags, fmt.Sprintf("album=%s", meta.Data[0].Relationships.Tracks.Data[index].Attributes.AlbumName)) - tags = append(tags, fmt.Sprintf("disk=%d/%d", meta.Data[0].Relationships.Tracks.Data[index].Attributes.DiscNumber, meta.Data[0].Relationships.Tracks.Data[trackTotal-1].Attributes.DiscNumber)) - tags = append(tags, fmt.Sprintf("track=%d", meta.Data[0].Relationships.Tracks.Data[index].Attributes.TrackNumber)) - tags = append(tags, fmt.Sprintf("tracknum=%d/%d", meta.Data[0].Relationships.Tracks.Data[index].Attributes.TrackNumber, meta.Data[0].Attributes.TrackCount)) - tags = append(tags, fmt.Sprintf("album_artist=%s", meta.Data[0].Attributes.ArtistName)) - tags = append(tags, fmt.Sprintf("performer=%s", meta.Data[0].Relationships.Tracks.Data[index].Attributes.ArtistName)) - tags = append(tags, fmt.Sprintf("copyright=%s", meta.Data[0].Attributes.Copyright)) - tags = append(tags, fmt.Sprintf("UPC=%s", meta.Data[0].Attributes.Upc)) - } - } else { - tags = append(tags, fmt.Sprintf("album=%s", MVInfo.Data[0].Attributes.AlbumName)) - tags = append(tags, fmt.Sprintf("disk=%d", MVInfo.Data[0].Attributes.DiscNumber)) - tags = append(tags, fmt.Sprintf("track=%d", MVInfo.Data[0].Attributes.TrackNumber)) - tags = append(tags, fmt.Sprintf("tracknum=%d", MVInfo.Data[0].Attributes.TrackNumber)) - //tags = append(tags, fmt.Sprintf("album_artist=%s", MVInfo.Data[0].Attributes.ArtistName)) - tags = append(tags, fmt.Sprintf("performer=%s", MVInfo.Data[0].Attributes.ArtistName)) - } - - // Extract and save thumbnail if enabled - var covPath string - //强制嵌入封面 - if true { - // Get the highest quality thumbnail URL from the MV info - thumbURL := MVInfo.Data[0].Attributes.Artwork.URL - - // Generate base name without extension - baseThumbName := forbiddenNames.ReplaceAllString(mvSaveName, "_") + "_thumbnail" - - // Download and save thumbnail - covPath, err = writeCover(saveDir, baseThumbName, thumbURL) - if err != nil { - fmt.Println("Failed to save MV thumbnail:", err) - } else { - //fmt.Println("MV thumbnail saved successfully") - tags = append(tags, fmt.Sprintf("cover=%s", covPath)) - } - } - - //mux and add tag - tagsString := strings.Join(tags, ":") - muxCmd := exec.Command("MP4Box", "-itags", tagsString, "-quiet", "-add", vidPath, "-add", audPath, "-keep-utc", "-new", mvOutPath) - fmt.Printf("MV Remuxing...") - if err := muxCmd.Run(); err != nil { - fmt.Printf("MV mux failed: %v\n", err) - return err - } - fmt.Printf("\rMV Remuxed. \n") - defer os.Remove(vidPath) - defer os.Remove(audPath) - defer os.Remove(covPath) - - return nil -} - -func extractMvAudio(c string) (string, error) { - MediaUrl, err := url.Parse(c) - if err != nil { - return "", err - } - - resp, err := http.Get(c) - if err != nil { - return "", err - } - defer resp.Body.Close() - - if resp.StatusCode != http.StatusOK { - return "", errors.New(resp.Status) - } - - body, err := io.ReadAll(resp.Body) - if err != nil { - return "", err - } - - audioString := string(body) - from, listType, err := m3u8.DecodeFrom(strings.NewReader(audioString), true) - if err != nil || listType != m3u8.MASTER { - return "", errors.New("m3u8 not of media type") - } - - audio := from.(*m3u8.MasterPlaylist) - - var audioPriority = []string{"audio-atmos", "audio-ac3", "audio-stereo-256"} - if Config.MVAudioType == "ac3" { - audioPriority = []string{"audio-ac3", "audio-stereo-256"} - } else if Config.MVAudioType == "aac" { - audioPriority = []string{"audio-stereo-256"} - } - - re := regexp.MustCompile(`_gr(\d+)_`) - - type AudioStream struct { - URL string - Rank int - GroupID string - } - var audioStreams []AudioStream - - for _, variant := range audio.Variants { - for _, audiov := range variant.Alternatives { - if audiov.URI != "" { - for _, priority := range audioPriority { - if audiov.GroupId == priority { - matches := re.FindStringSubmatch(audiov.URI) - if len(matches) == 2 { - var rank int - fmt.Sscanf(matches[1], "%d", &rank) - streamUrl, _ := MediaUrl.Parse(audiov.URI) - audioStreams = append(audioStreams, AudioStream{ - URL: streamUrl.String(), - Rank: rank, - GroupID: audiov.GroupId, - }) - } - } - } - } - } - } - - if len(audioStreams) == 0 { - return "", errors.New("no suitable audio stream found") - } - - sort.Slice(audioStreams, func(i, j int) bool { - return audioStreams[i].Rank > audioStreams[j].Rank - }) - fmt.Println("Audio: " + audioStreams[0].GroupID) - return audioStreams[0].URL, nil -} - -func checkM3u8(b string, f string) (string, error) { - var EnhancedHls string - if Config.GetM3u8FromDevice { - adamID := b - conn, err := net.Dial("tcp", Config.GetM3u8Port) - if err != nil { - fmt.Println("Error connecting to device:", err) - return "none", err - } - defer conn.Close() - if f == "song" { - fmt.Println("Connected to device") - } - - // Send the length of adamID and the adamID itself - adamIDBuffer := []byte(adamID) - lengthBuffer := []byte{byte(len(adamIDBuffer))} - - // Write length and adamID to the connection - _, err = conn.Write(lengthBuffer) - if err != nil { - fmt.Println("Error writing length to device:", err) - return "none", err - } - - _, err = conn.Write(adamIDBuffer) - if err != nil { - fmt.Println("Error writing adamID to device:", err) - return "none", err - } - - // Read the response (URL) from the device - response, err := bufio.NewReader(conn).ReadBytes('\n') - if err != nil { - fmt.Println("Error reading response from device:", err) - return "none", err - } - - // Trim any newline characters from the response - - response = bytes.TrimSpace(response) - if len(response) > 0 { - if f == "song" { - fmt.Println("Received URL:", string(response)) - } - EnhancedHls = string(response) - } else { - fmt.Println("Received an empty response") - } - } - return EnhancedHls, nil -} - -func formatAvailability(available bool, quality string) string { - if !available { - return "Not Available" - } - return quality -} - -func extractMedia(b string, more_mode bool) (string, string, error) { - masterUrl, err := url.Parse(b) - if err != nil { - return "", "", err - } - resp, err := http.Get(b) - if err != nil { - return "", "", err - } - defer resp.Body.Close() - if resp.StatusCode != http.StatusOK { - return "", "", errors.New(resp.Status) - } - body, err := io.ReadAll(resp.Body) - if err != nil { - return "", "", err - } - masterString := string(body) - from, listType, err := m3u8.DecodeFrom(strings.NewReader(masterString), true) - if err != nil || listType != m3u8.MASTER { - return "", "", errors.New("m3u8 not of master type") - } - master := from.(*m3u8.MasterPlaylist) - var streamUrl *url.URL - sort.Slice(master.Variants, func(i, j int) bool { - return master.Variants[i].AverageBandwidth > master.Variants[j].AverageBandwidth - }) - if debug_mode && more_mode { - fmt.Println("\nDebug: All Available Variants:") - var data [][]string - for _, variant := range master.Variants { - data = append(data, []string{variant.Codecs, variant.Audio, fmt.Sprint(variant.Bandwidth)}) - //fmt.Printf("Codec: %s, Audio: %s, Bandwidth: %d\n", - //variant.Codecs, variant.Audio, variant.Bandwidth) - } - table := tablewriter.NewWriter(os.Stdout) - table.SetHeader([]string{"Codec", "Audio", "Bandwidth"}) - //table.SetFooter([]string{"", "", "Total", "$146.93"}) - table.SetAutoMergeCells(true) - //table.SetAutoMergeCellsByColumnIndex([]int{1,2,3}) - table.SetRowLine(true) - table.AppendBulk(data) - table.Render() - - var hasAAC, hasLossless, hasHiRes, hasAtmos, hasDolbyAudio bool - var aacQuality, losslessQuality, hiResQuality, atmosQuality, dolbyAudioQuality string - - // Check for all formats - for _, variant := range master.Variants { - if variant.Codecs == "mp4a.40.2" { // AAC - hasAAC = true - split := strings.Split(variant.Audio, "-") - if len(split) >= 3 { - bitrate, _ := strconv.Atoi(split[2]) - currentBitrate := 0 - if aacQuality != "" { - current := strings.Split(aacQuality, " | ")[2] - current = strings.Split(current, " ")[0] - currentBitrate, _ = strconv.Atoi(current) - } - if bitrate > currentBitrate { - aacQuality = fmt.Sprintf("AAC | 2 Channel | %d kbps", bitrate) - } - } - } else if variant.Codecs == "ec-3" && strings.Contains(variant.Audio, "atmos") { // Dolby Atmos - hasAtmos = true - split := strings.Split(variant.Audio, "-") - if len(split) > 0 { - bitrateStr := split[len(split)-1] - // Remove leading "2" if present in "2768" - if len(bitrateStr) == 4 && bitrateStr[0] == '2' { - bitrateStr = bitrateStr[1:] - } - bitrate, _ := strconv.Atoi(bitrateStr) - currentBitrate := 0 - if atmosQuality != "" { - current := strings.Split(strings.Split(atmosQuality, " | ")[2], " ")[0] - currentBitrate, _ = strconv.Atoi(current) - } - if bitrate > currentBitrate { - atmosQuality = fmt.Sprintf("E-AC-3 | 16 Channel | %d kbps", bitrate) - } - } - } else if variant.Codecs == "alac" { // ALAC (Lossless or Hi-Res) - split := strings.Split(variant.Audio, "-") - if len(split) >= 3 { - bitDepth := split[len(split)-1] - sampleRate := split[len(split)-2] - sampleRateInt, _ := strconv.Atoi(sampleRate) - if sampleRateInt > 48000 { // Hi-Res - hasHiRes = true - hiResQuality = fmt.Sprintf("ALAC | 2 Channel | %s-bit/%d kHz", bitDepth, sampleRateInt/1000) - } else { // Standard Lossless - hasLossless = true - losslessQuality = fmt.Sprintf("ALAC | 2 Channel | %s-bit/%d kHz", bitDepth, sampleRateInt/1000) - } - } - } else if variant.Codecs == "ac-3" { // Dolby Audio - hasDolbyAudio = true - split := strings.Split(variant.Audio, "-") - if len(split) > 0 { - bitrate, _ := strconv.Atoi(split[len(split)-1]) - dolbyAudioQuality = fmt.Sprintf("AC-3 | 16 Channel | %d kbps", bitrate) - } - } - } - - fmt.Println("Available Audio Formats:") - fmt.Println("------------------------") - fmt.Printf("AAC : %s\n", formatAvailability(hasAAC, aacQuality)) - fmt.Printf("Lossless : %s\n", formatAvailability(hasLossless, losslessQuality)) - fmt.Printf("Hi-Res Lossless : %s\n", formatAvailability(hasHiRes, hiResQuality)) - fmt.Printf("Dolby Atmos : %s\n", formatAvailability(hasAtmos, atmosQuality)) - fmt.Printf("Dolby Audio : %s\n", formatAvailability(hasDolbyAudio, dolbyAudioQuality)) - fmt.Println("------------------------") - - return "", "", nil - } - var Quality string - for _, variant := range master.Variants { - if dl_atmos { - if variant.Codecs == "ec-3" && strings.Contains(variant.Audio, "atmos") { - if debug_mode && !more_mode { - fmt.Printf("Debug: Found Dolby Atmos variant - %s (Bitrate: %d kbps)\n", - variant.Audio, variant.Bandwidth/1000) - } - split := strings.Split(variant.Audio, "-") - length := len(split) - length_int, err := strconv.Atoi(split[length-1]) - if err != nil { - return "", "", err - } - if length_int <= Config.AtmosMax { - if !debug_mode && !more_mode { - fmt.Printf("%s\n", variant.Audio) - } - streamUrlTemp, err := masterUrl.Parse(variant.URI) - if err != nil { - return "", "", err - } - streamUrl = streamUrlTemp - Quality = fmt.Sprintf("%s kbps", split[len(split)-1]) - break - } - } else if variant.Codecs == "ac-3" { // Add Dolby Audio support - if debug_mode && !more_mode { - fmt.Printf("Debug: Found Dolby Audio variant - %s (Bitrate: %d kbps)\n", - variant.Audio, variant.Bandwidth/1000) - } - streamUrlTemp, err := masterUrl.Parse(variant.URI) - if err != nil { - return "", "", err - } - streamUrl = streamUrlTemp - split := strings.Split(variant.Audio, "-") - Quality = fmt.Sprintf("%s kbps", split[len(split)-1]) - break - } - } else if dl_aac { - if variant.Codecs == "mp4a.40.2" { - if debug_mode && !more_mode { - fmt.Printf("Debug: Found AAC variant - %s (Bitrate: %d)\n", variant.Audio, variant.Bandwidth) - } - aacregex := regexp.MustCompile(`audio-stereo-\d+`) - replaced := aacregex.ReplaceAllString(variant.Audio, "aac") - if replaced == Config.AacType { - if !debug_mode && !more_mode { - fmt.Printf("%s\n", variant.Audio) - } - streamUrlTemp, err := masterUrl.Parse(variant.URI) - if err != nil { - panic(err) - } - streamUrl = streamUrlTemp - split := strings.Split(variant.Audio, "-") - Quality = fmt.Sprintf("%s kbps", split[2]) - break - } - } - } else { - if variant.Codecs == "alac" { - split := strings.Split(variant.Audio, "-") - length := len(split) - length_int, err := strconv.Atoi(split[length-2]) - if err != nil { - return "", "", err - } - if length_int <= Config.AlacMax { - if !debug_mode && !more_mode { - fmt.Printf("%s-bit / %s Hz\n", split[length-1], split[length-2]) - } - streamUrlTemp, err := masterUrl.Parse(variant.URI) - if err != nil { - panic(err) - } - streamUrl = streamUrlTemp - KHZ := float64(length_int) / 1000.0 - Quality = fmt.Sprintf("%sB-%.1fkHz", split[length-1], KHZ) - break - } - } - } - } - if streamUrl == nil { - return "", "", errors.New("no codec found") - } - return streamUrl.String(), Quality, nil -} -func extractVideo(c string) (string, error) { - MediaUrl, err := url.Parse(c) - if err != nil { - return "", err - } - - resp, err := http.Get(c) - if err != nil { - return "", err - } - defer resp.Body.Close() - - if resp.StatusCode != http.StatusOK { - return "", errors.New(resp.Status) - } - - body, err := io.ReadAll(resp.Body) - if err != nil { - return "", err - } - videoString := string(body) - - from, listType, err := m3u8.DecodeFrom(strings.NewReader(videoString), true) - if err != nil || listType != m3u8.MASTER { - return "", errors.New("m3u8 not of media type") - } - - video := from.(*m3u8.MasterPlaylist) - - re := regexp.MustCompile(`_(\d+)x(\d+)`) - - var streamUrl *url.URL - sort.Slice(video.Variants, func(i, j int) bool { - return video.Variants[i].AverageBandwidth > video.Variants[j].AverageBandwidth - }) - - maxHeight := Config.MVMax - - for _, variant := range video.Variants { - matches := re.FindStringSubmatch(variant.URI) - if len(matches) == 3 { - height := matches[2] - var h int - _, err := fmt.Sscanf(height, "%d", &h) - if err != nil { - continue - } - if h <= maxHeight { - streamUrl, err = MediaUrl.Parse(variant.URI) - if err != nil { - return "", err - } - fmt.Println("Video: " + variant.Resolution + "-" + variant.VideoRange) - break - } - } - } - - if streamUrl == nil { - return "", errors.New("no suitable video stream found") - } - - return streamUrl.String(), nil -} - -func getInfoFromAdam(adamId string, token string, storefront string) (*structs.SongData, error) { - request, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/songs/%s", storefront, adamId), nil) - if err != nil { - return nil, err - } - query := url.Values{} - query.Set("extend", "extendedAssetUrls") - query.Set("include", "albums") - query.Set("l", Config.Language) - request.URL.RawQuery = query.Encode() - - request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", token)) - request.Header.Set("User-Agent", "iTunes/12.11.3 (Windows; Microsoft Windows 10 x64 Professional Edition (Build 19041); x64) AppleWebKit/7611.1022.4001.1 (dt:2)") - request.Header.Set("Origin", "https://music.apple.com") - - do, err := http.DefaultClient.Do(request) - if err != nil { - return nil, err - } - defer do.Body.Close() - if do.StatusCode != http.StatusOK { - return nil, errors.New(do.Status) - } - - obj := new(structs.ApiResult) - err = json.NewDecoder(do.Body).Decode(&obj) - if err != nil { - return nil, err - } - - for _, d := range obj.Data { - if d.ID == adamId { - return &d, nil - } - } - return nil, nil -} - -func getMVInfoFromAdam(adamId string, token string, storefront string) (*structs.AutoGeneratedMusicVideo, error) { - request, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/music-videos/%s", storefront, adamId), nil) - if err != nil { - return nil, err - } - query := url.Values{} - query.Set("l", Config.Language) - request.URL.RawQuery = query.Encode() - request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", token)) - request.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") - request.Header.Set("Origin", "https://music.apple.com") - - do, err := http.DefaultClient.Do(request) - if err != nil { - return nil, err - } - defer do.Body.Close() - if do.StatusCode != http.StatusOK { - return nil, errors.New(do.Status) - } - - obj := new(structs.AutoGeneratedMusicVideo) - err = json.NewDecoder(do.Body).Decode(&obj) - if err != nil { - return nil, err - } - - return obj, nil -} - -func getToken() (string, error) { - req, err := http.NewRequest("GET", "https://beta.music.apple.com", nil) - if err != nil { - return "", err - } - - resp, err := http.DefaultClient.Do(req) - if err != nil { - return "", err - } - defer resp.Body.Close() - - body, err := io.ReadAll(resp.Body) - if err != nil { - return "", err - } - - regex := regexp.MustCompile(`/assets/index-legacy-[^/]+\.js`) - indexJsUri := regex.FindString(string(body)) - - req, err = http.NewRequest("GET", "https://beta.music.apple.com"+indexJsUri, nil) - if err != nil { - return "", err - } - - resp, err = http.DefaultClient.Do(req) - if err != nil { - return "", err - } - defer resp.Body.Close() - - body, err = io.ReadAll(resp.Body) - if err != nil { - return "", err - } - - regex = regexp.MustCompile(`eyJh([^"]*)`) - token := regex.FindString(string(body)) - - return token, nil -} +package main + +import ( + "bufio" + "bytes" + "encoding/json" + "errors" + "fmt" + "io" + "log" + "net" + "net/http" + "net/url" + "os" + "os/exec" + "path/filepath" + "regexp" + "sort" + "strconv" + "strings" + "time" + + "main/utils/lyrics" + "main/utils/runv2" + "main/utils/runv3" + "main/utils/structs" + + "github.com/fatih/color" + "github.com/grafov/m3u8" + "github.com/olekukonko/tablewriter" + "github.com/spf13/pflag" + "github.com/zhaarey/go-mp4tag" + "gopkg.in/yaml.v2" +) + +var ( + forbiddenNames = regexp.MustCompile(`[/\\<>:"|?*]`) + dl_atmos bool + dl_aac bool + dl_select bool + dl_song bool + artist_select bool + debug_mode bool + alac_max *int + atmos_max *int + mv_max *int + mv_audio_type *string + aac_type *string + Config structs.ConfigSet + counter structs.Counter + okDict = make(map[string][]int) +) + +func loadConfig() error { + // 读取config.yaml文件内容 + data, err := os.ReadFile("config.yaml") + if err != nil { + return err + } + // 将yaml解析到config变量中 + err = yaml.Unmarshal(data, &Config) + if err != nil { + return err + } + return nil +} + +func LimitString(s string) string { + if len([]rune(s)) > Config.LimitMax { + return string([]rune(s)[:Config.LimitMax]) + } + return s +} + +func isInArray(arr []int, target int) bool { + for _, num := range arr { + if num == target { + return true + } + } + return false +} + +func fileExists(path string) (bool, error) { + f, err := os.Stat(path) + if err == nil { + return !f.IsDir(), nil + } else if os.IsNotExist(err) { + return false, nil + } + return false, err +} + +func checkUrl(url string) (string, string) { + pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music|classical\.music)\.apple\.com\/(\w{2})(?:\/album|\/album\/.+))\/(?:id)?(\d[^\D]+)(?:$|\?)`) + matches := pat.FindAllStringSubmatch(url, -1) + + if matches == nil { + return "", "" + } else { + return matches[0][1], matches[0][2] + } +} +func checkUrlMv(url string) (string, string) { + pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music)\.apple\.com\/(\w{2})(?:\/music-video|\/music-video\/.+))\/(?:id)?(\d[^\D]+)(?:$|\?)`) + matches := pat.FindAllStringSubmatch(url, -1) + + if matches == nil { + return "", "" + } else { + return matches[0][1], matches[0][2] + } +} +func checkUrlSong(url string) (string, string) { + pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music)\.apple\.com\/(\w{2})(?:\/song|\/song\/.+))\/(?:id)?(\d[^\D]+)(?:$|\?)`) + matches := pat.FindAllStringSubmatch(url, -1) + + if matches == nil { + return "", "" + } else { + return matches[0][1], matches[0][2] + } +} +func checkUrlPlaylist(url string) (string, string) { + pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music)\.apple\.com\/(\w{2})(?:\/playlist|\/playlist\/.+))\/(?:id)?(pl\.[\w-]+)(?:$|\?)`) + matches := pat.FindAllStringSubmatch(url, -1) + + if matches == nil { + return "", "" + } else { + return matches[0][1], matches[0][2] + } +} + +func checkUrlArtist(url string) (string, string) { + pat := regexp.MustCompile(`^(?:https:\/\/(?:beta\.music|music)\.apple\.com\/(\w{2})(?:\/artist|\/artist\/.+))\/(?:id)?(\d[^\D]+)(?:$|\?)`) + matches := pat.FindAllStringSubmatch(url, -1) + + if matches == nil { + return "", "" + } else { + return matches[0][1], matches[0][2] + } +} +func getUrlSong(songUrl string, token string) (string, error) { + storefront, songId := checkUrlSong(songUrl) + manifest, err := getInfoFromAdam(songId, token, storefront) + if err != nil { + fmt.Println("\u26A0 Failed to get manifest:", err) + counter.NotSong++ + return "", err + } + albumId := manifest.Relationships.Albums.Data[0].ID + songAlbumUrl := fmt.Sprintf("https://music.apple.com/%s/album/1/%s?i=%s", storefront, albumId, songId) + return songAlbumUrl, nil +} +func getUrlArtistName(artistUrl string, token string) (string, string, error) { + storefront, artistId := checkUrlArtist(artistUrl) + req, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/artists/%s", storefront, artistId), 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") + query := url.Values{} + query.Set("l", Config.Language) + 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(structs.AutoGeneratedArtist) + err = json.NewDecoder(do.Body).Decode(&obj) + if err != nil { + return "", "", err + } + return obj.Data[0].Attributes.Name, obj.Data[0].ID, nil +} + +func checkArtist(artistUrl string, token string, relationship string) ([]string, error) { + storefront, artistId := checkUrlArtist(artistUrl) + Num := 0 + //id := 1 + var args []string + var urls []string + var options [][]string + for { + req, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/artists/%s/%s?limit=100&offset=%d&l=%s", storefront, artistId, relationship, Num, Config.Language), nil) + if err != nil { + return nil, 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") + do, err := http.DefaultClient.Do(req) + if err != nil { + return nil, err + } + defer do.Body.Close() + if do.StatusCode != http.StatusOK { + return nil, errors.New(do.Status) + } + obj := new(structs.AutoGeneratedArtist) + err = json.NewDecoder(do.Body).Decode(&obj) + if err != nil { + return nil, err + } + for _, album := range obj.Data { + options = append(options, []string{album.Attributes.Name, album.Attributes.ReleaseDate, album.ID, album.Attributes.URL}) + } + Num = Num + 100 + if len(obj.Next) == 0 { + break + } + } + sort.Slice(options, func(i, j int) bool { + // 将日期字符串解析为 time.Time 类型进行比较 + dateI, _ := time.Parse("2006-01-02", options[i][1]) + dateJ, _ := time.Parse("2006-01-02", options[j][1]) + return dateI.Before(dateJ) // 返回 true 表示 i 在 j 前面 + }) + + table := tablewriter.NewWriter(os.Stdout) + if relationship == "albums" { + table.SetHeader([]string{"", "Album Name", "Date", "Album ID"}) + } else if relationship == "music-videos" { + table.SetHeader([]string{"", "MV Name", "Date", "MV ID"}) + } + //table.SetFooter([]string{"", "", "Total", "$146.93"}) + //table.SetAutoMergeCells(true) + //table.SetAutoMergeCellsByColumnIndex([]int{1,2,3}) + table.SetRowLine(false) + //table.AppendBulk(options) + table.SetHeaderColor(tablewriter.Colors{}, + tablewriter.Colors{tablewriter.FgRedColor, tablewriter.Bold}, + tablewriter.Colors{tablewriter.Bold, tablewriter.FgBlackColor}, + tablewriter.Colors{tablewriter.Bold, tablewriter.FgBlackColor}) + + 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 i, v := range options { + urls = append(urls, v[3]) + options[i] = append([]string{fmt.Sprint(i + 1)}, v[:3]...) + table.Append(options[i]) + } + table.Render() + if artist_select { + fmt.Println("You have selected all options:") + return urls, nil + } + reader := bufio.NewReader(os.Stdin) + fmt.Println("Please select from the " + relationship + " options above (multiple options separated by commas, ranges supported, or type 'all' to select all)") + cyanColor := color.New(color.FgCyan) + cyanColor.Print("Enter your choice: ") + input, _ := reader.ReadString('\n') + + // Remove newline and whitespace + input = strings.TrimSpace(input) + if input == "all" { + fmt.Println("You have selected all options:") + return urls, nil + } + + // Split input into string slices + 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}) + } + } + + // Print selected options + fmt.Println("You have selected the following options:") + 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(options) { + fmt.Println(options[num-1]) + 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(options) || start > end { + fmt.Println("Range out of range:", opt) + continue + } + for i := start; i <= end; i++ { + fmt.Println(options[i-1]) + args = append(args, urls[i-1]) + } + } else { + fmt.Println("Invalid option:", opt) + } + } + return args, nil +} + +func getMeta(albumId string, token string, storefront string) (*structs.AutoGenerated, error) { + var mtype string + var next string + if strings.Contains(albumId, "pl.") { + mtype = "playlists" + } else { + mtype = "albums" + } + req, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/%s/%s", storefront, mtype, albumId), nil) + if err != nil { + return nil, 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") + query := url.Values{} + query.Set("omit[resource]", "autos") + query.Set("include", "tracks,artists,record-labels") + query.Set("include[songs]", "artists,albums") + query.Set("fields[artists]", "name,artwork") + query.Set("fields[albums:albums]", "artistName,artwork,name,releaseDate,url") + query.Set("fields[record-labels]", "name") + query.Set("extend", "editorialVideo") + query.Set("l", Config.Language) + req.URL.RawQuery = query.Encode() + do, err := http.DefaultClient.Do(req) + if err != nil { + return nil, err + } + defer do.Body.Close() + if do.StatusCode != http.StatusOK { + return nil, errors.New(do.Status) + } + obj := new(structs.AutoGenerated) + err = json.NewDecoder(do.Body).Decode(&obj) + if err != nil { + return nil, err + } + if strings.Contains(albumId, "pl.") { + obj.Data[0].Attributes.ArtistName = "Apple Music" + } + if len(obj.Data[0].Relationships.Tracks.Next) > 0 { + next = obj.Data[0].Relationships.Tracks.Next + for { + req, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/%s&l=%s&include=albums", next, Config.Language), nil) + if err != nil { + return nil, 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") + do, err := http.DefaultClient.Do(req) + if err != nil { + return nil, err + } + defer do.Body.Close() + if do.StatusCode != http.StatusOK { + return nil, errors.New(do.Status) + } + obj2 := new(structs.AutoGeneratedTrack) + err = json.NewDecoder(do.Body).Decode(&obj2) + if err != nil { + return nil, err + } + for _, value := range obj2.Data { + obj.Data[0].Relationships.Tracks.Data = append(obj.Data[0].Relationships.Tracks.Data, value) + } + next = obj2.Next + if len(next) == 0 { + break + } + } + } + return obj, nil +} + +func writeCover(sanAlbumFolder, name string, url string) (string, error) { + covPath := filepath.Join(sanAlbumFolder, name+"."+Config.CoverFormat) + if Config.CoverFormat == "original" { + ext := strings.Split(url, "/")[len(strings.Split(url, "/"))-2] + ext = ext[strings.LastIndex(ext, ".")+1:] + covPath = filepath.Join(sanAlbumFolder, name+"."+ext) + } + exists, err := fileExists(covPath) + if err != nil { + fmt.Println("Failed to check if cover exists.") + return "", err + } + if exists { + _ = os.Remove(covPath) + } + if Config.CoverFormat == "png" { + re := regexp.MustCompile(`\{w\}x\{h\}`) + parts := re.Split(url, 2) + url = parts[0] + "{w}x{h}" + strings.Replace(parts[1], ".jpg", ".png", 1) + } + url = strings.Replace(url, "{w}x{h}", Config.CoverSize, 1) + if Config.CoverFormat == "original" { + url = strings.Replace(url, "is1-ssl.mzstatic.com/image/thumb", "a5.mzstatic.com/us/r1000/0", 1) + url = url[:strings.LastIndex(url, "/")] + } + req, err := http.NewRequest("GET", url, nil) + if err != nil { + return "", err + } + 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") + 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) + } + f, err := os.Create(covPath) + if err != nil { + return "", err + } + defer f.Close() + _, err = io.Copy(f, do.Body) + if err != nil { + return "", err + } + return covPath, nil +} + +func writeLyrics(sanAlbumFolder, filename string, lrc string) error { + lyricspath := filepath.Join(sanAlbumFolder, filename) + f, err := os.Create(lyricspath) + if err != nil { + return err + } + defer f.Close() + _, err = f.WriteString(lrc) + if err != nil { + return err + } + return nil +} + +func contains(slice []string, item string) bool { + for _, v := range slice { + if v == item { + return true + } + } + return false +} + +// 下载单曲逻辑 +func downloadTrack(trackNum int, trackTotal int, meta *structs.AutoGenerated, track structs.TrackData, albumId, token, storefront, mediaUserToken, sanAlbumFolder, Codec string, covPath string) { + counter.Total++ + fmt.Printf("Track %d of %d:\n", trackNum, trackTotal) + + //mv dl dev + if track.Type == "music-videos" { + if len(mediaUserToken) <= 50 { + fmt.Println("meida-user-token is not set, skip MV dl") + counter.Success++ + return + } + if _, err := exec.LookPath("mp4decrypt"); err != nil { + fmt.Println("mp4decrypt is not found, skip MV dl") + counter.Success++ + return + } + err := mvDownloader(track.ID, sanAlbumFolder, token, storefront, mediaUserToken, meta) + if err != nil { + fmt.Println("\u26A0 Failed to dl MV:", err) + counter.Error++ + return + } + counter.Success++ + return + } + + manifest, err := getInfoFromAdam(track.ID, token, storefront) + if err != nil { + fmt.Println("\u26A0 Failed to get manifest:", err) + counter.NotSong++ + return + } + needDlAacLc := false + if dl_aac && Config.AacType == "aac-lc" { + needDlAacLc = true + } + if manifest.Attributes.ExtendedAssetUrls.EnhancedHls == "" { + if dl_atmos { + fmt.Println("Unavailable") + counter.Unavailable++ + return + } + fmt.Println("Unavailable, Try DL AAC-LC") + needDlAacLc = true + } + needCheck := false + + if Config.GetM3u8Mode == "all" { + needCheck = true + } else if Config.GetM3u8Mode == "hires" && contains(track.Attributes.AudioTraits, "hi-res-lossless") { + needCheck = true + } + var EnhancedHls_m3u8 string + if needCheck && !needDlAacLc { + EnhancedHls_m3u8, _ = checkM3u8(track.ID, "song") + if strings.HasSuffix(EnhancedHls_m3u8, ".m3u8") { + manifest.Attributes.ExtendedAssetUrls.EnhancedHls = EnhancedHls_m3u8 + } + } + var Quality string + if strings.Contains(Config.SongFileFormat, "Quality") { + if dl_atmos { + Quality = fmt.Sprintf("%dkbps", Config.AtmosMax-2000) + } else if needDlAacLc { + Quality = "256kbps" + } else { + _, Quality, err = extractMedia(manifest.Attributes.ExtendedAssetUrls.EnhancedHls, true) + if err != nil { + fmt.Println("Failed to extract quality from manifest.\n", err) + counter.Error++ + return + } + } + } + stringsToJoin := []string{} + if track.Attributes.IsAppleDigitalMaster { + if Config.AppleMasterChoice != "" { + stringsToJoin = append(stringsToJoin, Config.AppleMasterChoice) + } + } + if track.Attributes.ContentRating == "explicit" { + if Config.ExplicitChoice != "" { + stringsToJoin = append(stringsToJoin, Config.ExplicitChoice) + } + } + if track.Attributes.ContentRating == "clean" { + if Config.CleanChoice != "" { + stringsToJoin = append(stringsToJoin, Config.CleanChoice) + } + } + Tag_string := strings.Join(stringsToJoin, " ") + + songName := strings.NewReplacer( + "{SongId}", track.ID, + "{SongNumer}", fmt.Sprintf("%02d", trackNum), + "{SongName}", LimitString(track.Attributes.Name), + "{DiscNumber}", fmt.Sprintf("%0d", track.Attributes.DiscNumber), + "{TrackNumber}", fmt.Sprintf("%0d", track.Attributes.TrackNumber), + "{Quality}", Quality, + "{Tag}", Tag_string, + "{Codec}", Codec, + ).Replace(Config.SongFileFormat) + fmt.Println(songName) + filename := fmt.Sprintf("%s.m4a", forbiddenNames.ReplaceAllString(songName, "_")) + lrcFilename := fmt.Sprintf("%s.%s", forbiddenNames.ReplaceAllString(songName, "_"), Config.LrcFormat) + trackPath := filepath.Join(sanAlbumFolder, filename) + + //get lrc + var lrc string = "" + if Config.EmbedLrc || Config.SaveLrcFile { + lrcStr, err := lyrics.Get(storefront, track.ID, Config.LrcType, Config.Language, Config.LrcFormat, token, mediaUserToken) + if err != nil { + fmt.Println(err) + } else { + if Config.SaveLrcFile { + err := writeLyrics(sanAlbumFolder, lrcFilename, lrcStr) + if err != nil { + fmt.Printf("Failed to write lyrics") + } + } + if Config.EmbedLrc { + lrc = lrcStr + } + } + } + + exists, err := fileExists(trackPath) + if err != nil { + fmt.Println("Failed to check if track exists.") + } + if exists { + fmt.Println("Track already exists locally.") + counter.Success++ + okDict[albumId] = append(okDict[albumId], trackNum) + return + } + if needDlAacLc { + if len(mediaUserToken) <= 50 { + fmt.Println("Invalid media-user-token") + counter.Error++ + return + } + _, err := runv3.Run(track.ID, trackPath, token, mediaUserToken, false) + if err != nil { + fmt.Println("Failed to dl aac-lc:", err) + counter.Error++ + return + } + } else { + trackM3u8Url, _, err := extractMedia(manifest.Attributes.ExtendedAssetUrls.EnhancedHls, false) + if err != nil { + fmt.Println("\u26A0 Failed to extract info from manifest:", err) + counter.Unavailable++ + return + } + //边下载边解密 + err = runv2.Run(track.ID, trackM3u8Url, trackPath, Config) + if err != nil { + fmt.Println("Failed to run v2:", err) + counter.Error++ + return + } + } + tags := []string{ + "tool=", + fmt.Sprintf("artist=%s", meta.Data[0].Attributes.ArtistName), + //fmt.Sprintf("lyrics=%s", lrc), + } + var trackCovPath string + if Config.EmbedCover { + if strings.Contains(albumId, "pl.") && Config.DlAlbumcoverForPlaylist { + trackCovPath, err = writeCover(sanAlbumFolder, track.ID, track.Attributes.Artwork.URL) + if err != nil { + fmt.Println("Failed to write cover.") + } + tags = append(tags, fmt.Sprintf("cover=%s", trackCovPath)) + } else { + tags = append(tags, fmt.Sprintf("cover=%s", covPath)) + } + } + tagsString := strings.Join(tags, ":") + cmd := exec.Command("MP4Box", "-itags", tagsString, trackPath) + if err := cmd.Run(); err != nil { + fmt.Printf("Embed failed: %v\n", err) + counter.Error++ + return + } + if strings.Contains(albumId, "pl.") && Config.DlAlbumcoverForPlaylist && trackCovPath != "" { + if err := os.Remove(trackCovPath); err != nil { + fmt.Printf("Error deleting file: %s\n", trackCovPath) + counter.Error++ + return + } + } + err = writeMP4Tags(trackPath, lrc, meta, trackNum, trackTotal) + if err != nil { + fmt.Println("\u26A0 Failed to write tags in media:", err) + counter.Unavailable++ + return + } + counter.Success++ + okDict[albumId] = append(okDict[albumId], trackNum) +} + +func rip(albumId string, token string, storefront string, mediaUserToken string, urlArg_i string) error { + meta, err := getMeta(albumId, token, storefront) + if err != nil { + return err + } + + if debug_mode { + // Print album info + fmt.Println(meta.Data[0].Attributes.ArtistName) + fmt.Println(meta.Data[0].Attributes.Name) + + for trackNum, track := range meta.Data[0].Relationships.Tracks.Data { + trackNum++ + fmt.Printf("\nTrack %d of %d:\n", trackNum, len(meta.Data[0].Relationships.Tracks.Data)) + fmt.Printf("%02d. %s\n", trackNum, track.Attributes.Name) + + manifest, err := getInfoFromAdam(track.ID, token, storefront) + if err != nil { + fmt.Printf("Failed to get manifest for track %d: %v\n", trackNum, err) + continue + } + + var m3u8Url string + //Web端m3u8 + if manifest.Attributes.ExtendedAssetUrls.EnhancedHls != "" { + m3u8Url = manifest.Attributes.ExtendedAssetUrls.EnhancedHls + } + //设备端满血m3u8 + needCheck := false + if Config.GetM3u8Mode == "all" { + needCheck = true + } else if Config.GetM3u8Mode == "hires" && contains(track.Attributes.AudioTraits, "hi-res-lossless") { + needCheck = true + } + if needCheck { + fullM3u8Url, err := checkM3u8(track.ID, "song") + if err == nil && strings.HasSuffix(fullM3u8Url, ".m3u8") { + m3u8Url = fullM3u8Url + } else { + fmt.Println("Failed to get best quality m3u8 from device m3u8 port, will use m3u8 from Web API") + } + } + + _, _, err = extractMedia(m3u8Url, true) + if err != nil { + fmt.Printf("Failed to extract quality info for track %d: %v\n", trackNum, err) + continue + } + } + return nil // Return directly without showing statistics + } + var Codec string + if dl_atmos { + Codec = "ATMOS" + } else if dl_aac { + Codec = "AAC" + } else { + Codec = "ALAC" + } + var singerFoldername string + if Config.ArtistFolderFormat != "" { + if strings.Contains(albumId, "pl.") { + singerFoldername = strings.NewReplacer( + "{ArtistName}", "Apple Music", + "{ArtistId}", "", + "{UrlArtistName}", "Apple Music", + ).Replace(Config.ArtistFolderFormat) + } else if len(meta.Data[0].Relationships.Artists.Data) > 0 { + singerFoldername = strings.NewReplacer( + "{UrlArtistName}", LimitString(meta.Data[0].Attributes.ArtistName), + "{ArtistName}", LimitString(meta.Data[0].Attributes.ArtistName), + "{ArtistId}", meta.Data[0].Relationships.Artists.Data[0].ID, + ).Replace(Config.ArtistFolderFormat) + } else { + singerFoldername = strings.NewReplacer( + "{UrlArtistName}", LimitString(meta.Data[0].Attributes.ArtistName), + "{ArtistName}", LimitString(meta.Data[0].Attributes.ArtistName), + "{ArtistId}", "", + ).Replace(Config.ArtistFolderFormat) + } + if strings.HasSuffix(singerFoldername, ".") { + singerFoldername = strings.ReplaceAll(singerFoldername, ".", "") + } + singerFoldername = strings.TrimSpace(singerFoldername) + fmt.Println(singerFoldername) + } + singerFolder := filepath.Join(Config.AlacSaveFolder, forbiddenNames.ReplaceAllString(singerFoldername, "_")) + if dl_atmos { + singerFolder = filepath.Join(Config.AtmosSaveFolder, forbiddenNames.ReplaceAllString(singerFoldername, "_")) + } + var Quality string + if strings.Contains(Config.AlbumFolderFormat, "Quality") { + if dl_atmos { + Quality = fmt.Sprintf("%dkbps", Config.AtmosMax-2000) + } else if dl_aac && Config.AacType == "aac-lc" { + Quality = "256kbps" + } else { + manifest1, err := getInfoFromAdam(meta.Data[0].Relationships.Tracks.Data[0].ID, token, storefront) + if err != nil { + fmt.Println("Failed to get manifest.\n", err) + } else { + if manifest1.Attributes.ExtendedAssetUrls.EnhancedHls == "" { + Codec = "AAC" + Quality = "256kbps" + //fmt.Println("Unavailable.\n") + } else { + needCheck := false + + if Config.GetM3u8Mode == "all" { + needCheck = true + } else if Config.GetM3u8Mode == "hires" && contains(meta.Data[0].Relationships.Tracks.Data[0].Attributes.AudioTraits, "hi-res-lossless") { + needCheck = true + } + var EnhancedHls_m3u8 string + if needCheck { + EnhancedHls_m3u8, _ = checkM3u8(meta.Data[0].Relationships.Tracks.Data[0].ID, "album") + if strings.HasSuffix(EnhancedHls_m3u8, ".m3u8") { + manifest1.Attributes.ExtendedAssetUrls.EnhancedHls = EnhancedHls_m3u8 + } + } + _, Quality, err = extractMedia(manifest1.Attributes.ExtendedAssetUrls.EnhancedHls, true) + if err != nil { + fmt.Println("Failed to extract quality from manifest.\n", err) + } + } + } + } + } + stringsToJoin := []string{} + if meta.Data[0].Attributes.IsAppleDigitalMaster || meta.Data[0].Attributes.IsMasteredForItunes { + if Config.AppleMasterChoice != "" { + stringsToJoin = append(stringsToJoin, Config.AppleMasterChoice) + } + } + if meta.Data[0].Attributes.ContentRating == "explicit" { + if Config.ExplicitChoice != "" { + stringsToJoin = append(stringsToJoin, Config.ExplicitChoice) + } + } + if meta.Data[0].Attributes.ContentRating == "clean" { + if Config.CleanChoice != "" { + stringsToJoin = append(stringsToJoin, Config.CleanChoice) + } + } + Tag_string := strings.Join(stringsToJoin, " ") + var albumFolder string + if strings.Contains(albumId, "pl.") { + albumFolder = strings.NewReplacer( + "{ArtistName}", "Apple Music", + "{PlaylistName}", LimitString(meta.Data[0].Attributes.Name), + "{PlaylistId}", albumId, + "{Quality}", Quality, + "{Codec}", Codec, + "{Tag}", Tag_string, + ).Replace(Config.PlaylistFolderFormat) + } else { + albumFolder = strings.NewReplacer( + "{ReleaseDate}", meta.Data[0].Attributes.ReleaseDate, + "{ReleaseYear}", meta.Data[0].Attributes.ReleaseDate[:4], + "{ArtistName}", LimitString(meta.Data[0].Attributes.ArtistName), + "{AlbumName}", LimitString(meta.Data[0].Attributes.Name), + "{UPC}", meta.Data[0].Attributes.Upc, + "{RecordLabel}", meta.Data[0].Attributes.RecordLabel, + "{Copyright}", meta.Data[0].Attributes.Copyright, + "{AlbumId}", albumId, + "{Quality}", Quality, + "{Codec}", Codec, + "{Tag}", Tag_string, + ).Replace(Config.AlbumFolderFormat) + } + if strings.HasSuffix(albumFolder, ".") { + albumFolder = strings.ReplaceAll(albumFolder, ".", "") + } + albumFolder = strings.TrimSpace(albumFolder) + sanAlbumFolder := filepath.Join(singerFolder, forbiddenNames.ReplaceAllString(albumFolder, "_")) + os.MkdirAll(sanAlbumFolder, os.ModePerm) + fmt.Println(albumFolder) + //get artist cover + if Config.SaveArtistCover && !(strings.Contains(albumId, "pl.")) { + if len(meta.Data[0].Relationships.Artists.Data) > 0 { + _, err = writeCover(singerFolder, "folder", meta.Data[0].Relationships.Artists.Data[0].Attributes.Artwork.Url) + if err != nil { + fmt.Println("Failed to write artist cover.") + } + } + } + //get album cover + covPath, err := writeCover(sanAlbumFolder, "cover", meta.Data[0].Attributes.Artwork.URL) + if err != nil { + fmt.Println("Failed to write cover.") + } + //get animated artwork + if Config.SaveAnimatedArtwork && meta.Data[0].Attributes.EditorialVideo.MotionDetailSquare.Video != "" { + fmt.Println("Found Animation Artwork.") + + // Download square version + motionvideoUrlSquare, err := extractVideo(meta.Data[0].Attributes.EditorialVideo.MotionDetailSquare.Video) + if err != nil { + fmt.Println("no motion video square.\n", err) + } else { + exists, err := fileExists(filepath.Join(sanAlbumFolder, "square_animated_artwork.mp4")) + if err != nil { + fmt.Println("Failed to check if animated artwork square exists.") + } + if exists { + fmt.Println("Animated artwork square already exists locally.") + } else { + fmt.Println("Animation Artwork Square Downloading...") + cmd := exec.Command("ffmpeg", "-loglevel", "quiet", "-y", "-i", motionvideoUrlSquare, "-c", "copy", filepath.Join(sanAlbumFolder, "square_animated_artwork.mp4")) + if err := cmd.Run(); err != nil { + fmt.Printf("animated artwork square dl err: %v\n", err) + } else { + fmt.Println("Animation Artwork Square Downloaded") + } + } + } + + if Config.EmbyAnimatedArtwork { + // Convert square version to gif + cmd3 := exec.Command("ffmpeg", "-i", filepath.Join(sanAlbumFolder, "square_animated_artwork.mp4"), "-vf", "scale=440:-1", "-r", "24", "-f", "gif", filepath.Join(sanAlbumFolder, "folder.jpg")) + if err := cmd3.Run(); err != nil { + fmt.Printf("animated artwork square to gif err: %v\n", err) + } + } + + // Download tall version + motionvideoUrlTall, err := extractVideo(meta.Data[0].Attributes.EditorialVideo.MotionDetailTall.Video) + if err != nil { + fmt.Println("no motion video tall.\n", err) + } else { + exists, err := fileExists(filepath.Join(sanAlbumFolder, "tall_animated_artwork.mp4")) + if err != nil { + fmt.Println("Failed to check if animated artwork tall exists.") + } + if exists { + fmt.Println("Animated artwork tall already exists locally.") + } else { + fmt.Println("Animation Artwork Tall Downloading...") + cmd := exec.Command("ffmpeg", "-loglevel", "quiet", "-y", "-i", motionvideoUrlTall, "-c", "copy", filepath.Join(sanAlbumFolder, "tall_animated_artwork.mp4")) + if err := cmd.Run(); err != nil { + fmt.Printf("animated artwork tall dl err: %v\n", err) + } else { + fmt.Println("Animation Artwork Tall Downloaded") + } + } + } + } + trackTotal := len(meta.Data[0].Relationships.Tracks.Data) + arr := make([]int, trackTotal) + for i := 0; i < trackTotal; i++ { + arr[i] = i + 1 + } + selected := []int{} + + if dl_song { + if urlArg_i == "" { + //fmt.Println("URL does not contain parameter 'i'. Please ensure the URL includes 'i' or use another mode.") + //return nil + } else { + for trackNum, track := range meta.Data[0].Relationships.Tracks.Data { + trackNum++ + if urlArg_i == track.ID { + downloadTrack(trackNum, trackTotal, meta, track, albumId, token, storefront, mediaUserToken, sanAlbumFolder, Codec, covPath) + return nil + } + } + } + return nil + } + + if !dl_select { + selected = arr + } else { + var data [][]string + for trackNum, track := range meta.Data[0].Relationships.Tracks.Data { + trackNum++ + var trackName string + if meta.Data[0].Type == "albums" { + trackName = fmt.Sprintf("%02d. %s", track.Attributes.TrackNumber, track.Attributes.Name) + } else { + trackName = fmt.Sprintf("%s - %s", track.Attributes.Name, track.Attributes.ArtistName) + } + 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(meta.Data[0].Type == "albums", fmt.Sprintf("Storefront: %s, %d tracks missing", strings.ToUpper(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) + } + } + } + fmt.Println("Selected options:", selected) + } + for trackNum, track := range meta.Data[0].Relationships.Tracks.Data { + trackNum++ + if isInArray(okDict[albumId], trackNum) { + //fmt.Println("已完成直接跳过.\n") + counter.Total++ + counter.Success++ + continue + } + if isInArray(selected, trackNum) { + downloadTrack(trackNum, trackTotal, meta, track, albumId, token, storefront, mediaUserToken, sanAlbumFolder, Codec, covPath) + } + } + return nil +} + +func writeMP4Tags(trackPath, lrc string, meta *structs.AutoGenerated, trackNum, trackTotal int) error { + index := trackNum - 1 + + t := &mp4tag.MP4Tags{ + Title: meta.Data[0].Relationships.Tracks.Data[index].Attributes.Name, + TitleSort: meta.Data[0].Relationships.Tracks.Data[index].Attributes.Name, + Artist: meta.Data[0].Relationships.Tracks.Data[index].Attributes.ArtistName, + ArtistSort: meta.Data[0].Relationships.Tracks.Data[index].Attributes.ArtistName, + Custom: map[string]string{ + "PERFORMER": meta.Data[0].Relationships.Tracks.Data[index].Attributes.ArtistName, + "RELEASETIME": meta.Data[0].Relationships.Tracks.Data[index].Attributes.ReleaseDate, + "ISRC": meta.Data[0].Relationships.Tracks.Data[index].Attributes.Isrc, + "LABEL": meta.Data[0].Attributes.RecordLabel, + "UPC": meta.Data[0].Attributes.Upc, + }, + Composer: meta.Data[0].Relationships.Tracks.Data[index].Attributes.ComposerName, + ComposerSort: meta.Data[0].Relationships.Tracks.Data[index].Attributes.ComposerName, + Date: meta.Data[0].Attributes.ReleaseDate, + CustomGenre: meta.Data[0].Relationships.Tracks.Data[index].Attributes.GenreNames[0], + Copyright: meta.Data[0].Attributes.Copyright, + Publisher: meta.Data[0].Attributes.RecordLabel, + Lyrics: lrc, + } + + if !strings.Contains(meta.Data[0].ID, "pl.") { + albumID, err := strconv.ParseUint(meta.Data[0].ID, 10, 32) + if err != nil { + return err + } + t.ItunesAlbumID = int32(albumID) + } + + if len(meta.Data[0].Relationships.Artists.Data) > 0 { + if len(meta.Data[0].Relationships.Tracks.Data[index].Relationships.Artists.Data) > 0 { + artistID, err := strconv.ParseUint(meta.Data[0].Relationships.Tracks.Data[index].Relationships.Artists.Data[0].ID, 10, 32) + if err != nil { + return err + } + t.ItunesArtistID = int32(artistID) + } + } + + if strings.Contains(meta.Data[0].ID, "pl.") && !Config.UseSongInfoForPlaylist { + t.DiscNumber = 1 + t.DiscTotal = 1 + t.TrackNumber = int16(trackNum) + t.TrackTotal = int16(trackTotal) + t.Album = meta.Data[0].Attributes.Name + t.AlbumSort = meta.Data[0].Attributes.Name + t.AlbumArtist = meta.Data[0].Attributes.ArtistName + t.AlbumArtistSort = meta.Data[0].Attributes.ArtistName + } else if strings.Contains(meta.Data[0].ID, "pl.") && Config.UseSongInfoForPlaylist { + t.DiscNumber = int16(meta.Data[0].Relationships.Tracks.Data[index].Attributes.DiscNumber) + t.DiscTotal = int16(meta.Data[0].Relationships.Tracks.Data[trackTotal-1].Attributes.DiscNumber) + t.TrackNumber = int16(meta.Data[0].Relationships.Tracks.Data[index].Attributes.TrackNumber) + t.TrackTotal = int16(trackTotal) + t.Album = meta.Data[0].Relationships.Tracks.Data[index].Attributes.AlbumName + t.AlbumSort = meta.Data[0].Relationships.Tracks.Data[index].Attributes.AlbumName + t.AlbumArtist = meta.Data[0].Relationships.Tracks.Data[index].Relationships.Albums.Data[0].Attributes.ArtistName + t.AlbumArtistSort = meta.Data[0].Relationships.Tracks.Data[index].Relationships.Albums.Data[0].Attributes.ArtistName + } else { + t.DiscNumber = int16(meta.Data[0].Relationships.Tracks.Data[index].Attributes.DiscNumber) + t.DiscTotal = int16(meta.Data[0].Relationships.Tracks.Data[trackTotal-1].Attributes.DiscNumber) + t.TrackNumber = int16(meta.Data[0].Relationships.Tracks.Data[index].Attributes.TrackNumber) + t.TrackTotal = int16(trackTotal) + t.Album = meta.Data[0].Relationships.Tracks.Data[index].Attributes.AlbumName + t.AlbumSort = meta.Data[0].Relationships.Tracks.Data[index].Attributes.AlbumName + t.AlbumArtist = meta.Data[0].Attributes.ArtistName + t.AlbumArtistSort = meta.Data[0].Attributes.ArtistName + } + + if meta.Data[0].Relationships.Tracks.Data[index].Attributes.ContentRating == "explicit" { + t.ItunesAdvisory = mp4tag.ItunesAdvisoryExplicit + } else if meta.Data[0].Relationships.Tracks.Data[index].Attributes.ContentRating == "clean" { + t.ItunesAdvisory = mp4tag.ItunesAdvisoryClean + } else { + t.ItunesAdvisory = mp4tag.ItunesAdvisoryNone + } + + mp4, err := mp4tag.Open(trackPath) + if err != nil { + return err + } + defer mp4.Close() + err = mp4.Write(t, []string{}) + if err != nil { + return err + } + return nil +} + +func main() { + err := loadConfig() + if err != nil { + fmt.Printf("load Config failed: %v", err) + return + } + token, err := getToken() + if err != nil { + if Config.AuthorizationToken != "" && Config.AuthorizationToken != "your-authorization-token" { + token = strings.Replace(Config.AuthorizationToken, "Bearer ", "", -1) + } else { + fmt.Println("Failed to get token.") + return + } + } + // Define command-line flags + pflag.BoolVar(&dl_atmos, "atmos", false, "Enable atmos download mode") + pflag.BoolVar(&dl_aac, "aac", false, "Enable adm-aac download mode") + pflag.BoolVar(&dl_select, "select", false, "Enable selective download") + pflag.BoolVar(&dl_song, "song", false, "Enable single song download mode") + pflag.BoolVar(&artist_select, "all-album", false, "Download all artist albums") + pflag.BoolVar(&debug_mode, "debug", false, "Enable debug mode to show audio quality information") + alac_max = pflag.Int("alac-max", Config.AlacMax, "Specify the max quality for download alac") + atmos_max = pflag.Int("atmos-max", Config.AtmosMax, "Specify the max quality for download atmos") + aac_type = pflag.String("aac-type", Config.AacType, "Select AAC type, aac aac-binaural aac-downmix") + mv_audio_type = pflag.String("mv-audio-type", Config.MVAudioType, "Select MV audio type, atmos ac3 aac") + mv_max = pflag.Int("mv-max", Config.MVMax, "Specify the max quality for download MV") + + // Custom usage message for help + pflag.Usage = func() { + fmt.Fprintf(os.Stderr, "Usage: %s [options] url1 url2 ...\n", "[main | main.exe | go run main.go]") + fmt.Println("Options:") + pflag.PrintDefaults() + } + + // Parse the flag arguments + pflag.Parse() + Config.AlacMax = *alac_max + Config.AtmosMax = *atmos_max + Config.AacType = *aac_type + Config.MVAudioType = *mv_audio_type + Config.MVMax = *mv_max + + args := pflag.Args() + if len(args) == 0 { + fmt.Println("No URLs provided. Please provide at least one URL.") + pflag.Usage() + return + } + os.Args = args + if strings.Contains(os.Args[0], "/artist/") { + urlArtistName, urlArtistID, err := getUrlArtistName(os.Args[0], token) + if err != nil { + fmt.Println("Failed to get artistname.") + return + } + //fmt.Println("get artistname:", urlArtistName) + Config.ArtistFolderFormat = strings.NewReplacer( + "{UrlArtistName}", LimitString(urlArtistName), + "{ArtistId}", urlArtistID, + ).Replace(Config.ArtistFolderFormat) + albumArgs, err := checkArtist(os.Args[0], token, "albums") + if err != nil { + fmt.Println("Failed to get artist albums.") + return + } + mvArgs, err := checkArtist(os.Args[0], token, "music-videos") + if err != nil { + fmt.Println("Failed to get artist music-videos.") + //return + } + os.Args = append(albumArgs, mvArgs...) + } + albumTotal := len(os.Args) + for { + for albumNum, urlRaw := range os.Args { + fmt.Printf("Album %d of %d:\n", albumNum+1, albumTotal) + var storefront, albumId string + //mv dl dev + if strings.Contains(urlRaw, "/music-video/") { + if debug_mode { + continue + } + counter.Total++ + if len(Config.MediaUserToken) <= 50 { + fmt.Println("meida-user-token is not set, skip MV dl") + counter.Success++ + continue + } + if _, err := exec.LookPath("mp4decrypt"); err != nil { + fmt.Println("mp4decrypt is not found, skip MV dl") + counter.Success++ + continue + } + mvSaveDir := strings.NewReplacer( + "{ArtistName}", "", + "{UrlArtistName}", "", + "{ArtistId}", "", + ).Replace(Config.ArtistFolderFormat) + if mvSaveDir != "" { + mvSaveDir = filepath.Join(Config.AlacSaveFolder, forbiddenNames.ReplaceAllString(mvSaveDir, "_")) + } else { + mvSaveDir = Config.AlacSaveFolder + } + storefront, albumId = checkUrlMv(urlRaw) + err := mvDownloader(albumId, mvSaveDir, token, storefront, Config.MediaUserToken, nil) + if err != nil { + fmt.Println("\u26A0 Failed to dl MV:", err) + counter.Error++ + continue + } + counter.Success++ + continue + } + if strings.Contains(urlRaw, "/song/") { + urlRaw, err = getUrlSong(urlRaw, token) + dl_song = true + if err != nil { + fmt.Println("Failed to get Song info.") + } + } + if strings.Contains(urlRaw, "/playlist/") { + storefront, albumId = checkUrlPlaylist(urlRaw) + } else { + storefront, albumId = checkUrl(urlRaw) + } + if albumId == "" { + fmt.Printf("Invalid URL: %s\n", urlRaw) + continue + } + parse, err := url.Parse(urlRaw) + if err != nil { + log.Fatalf("Invalid URL: %v", err) + } + var urlArg_i = parse.Query().Get("i") + err = rip(albumId, token, storefront, Config.MediaUserToken, urlArg_i) + if err != nil { + fmt.Println("Album failed.") + fmt.Println(err) + } + } + fmt.Printf("======= [\u2714 ] Completed: %d/%d | [\u26A0 ] Warnings: %d | [\u2716 ] Errors: %d =======\n", counter.Success, counter.Total, counter.Unavailable+counter.NotSong, counter.Error) + if counter.Error == 0 { + break + } + fmt.Println("Error detected, press Enter to try again...") + fmt.Scanln() + fmt.Println("Start trying again...") + counter = structs.Counter{} + } +} +func mvDownloader(adamID string, saveDir string, token string, storefront string, mediaUserToken string, meta *structs.AutoGenerated) error { + MVInfo, err := getMVInfoFromAdam(adamID, token, storefront) + if err != nil { + fmt.Println("\u26A0 Failed to get MV manifest:", err) + return nil + } + + //获取传入的专辑信息当中该mv所在的位置 + var trackTotal int + var trackNum int + var index int + if meta != nil { + trackTotal = len(meta.Data[0].Relationships.Tracks.Data) + for i, track := range meta.Data[0].Relationships.Tracks.Data { + if adamID == track.ID { + index = i + trackNum = i + 1 + } + } + } + + if strings.HasSuffix(saveDir, ".") { + saveDir = strings.ReplaceAll(saveDir, ".", "") + } + saveDir = strings.TrimSpace(saveDir) + + vidPath := filepath.Join(saveDir, fmt.Sprintf("%s_vid.mp4", adamID)) + audPath := filepath.Join(saveDir, fmt.Sprintf("%s_aud.mp4", adamID)) + mvSaveName := fmt.Sprintf("%s (%s)", MVInfo.Data[0].Attributes.Name, adamID) + if meta != nil { + mvSaveName = fmt.Sprintf("%02d. %s", trackNum, MVInfo.Data[0].Attributes.Name) + } + + mvOutPath := filepath.Join(saveDir, fmt.Sprintf("%s.mp4", forbiddenNames.ReplaceAllString(mvSaveName, "_"))) + + fmt.Println(MVInfo.Data[0].Attributes.Name) + + exists, _ := fileExists(mvOutPath) + if exists { + fmt.Println("MV already exists locally.") + return nil + } + + mvm3u8url, _, _ := runv3.GetWebplayback(adamID, token, mediaUserToken, true) + if mvm3u8url == "" { + return errors.New("media-user-token may wrong or expired") + } + + os.MkdirAll(saveDir, os.ModePerm) + //video + videom3u8url, _ := extractVideo(mvm3u8url) + videokeyAndUrls, _ := runv3.Run(adamID, videom3u8url, token, mediaUserToken, true) + _ = runv3.ExtMvData(videokeyAndUrls, vidPath) + //audio + audiom3u8url, _ := extractMvAudio(mvm3u8url) + audiokeyAndUrls, _ := runv3.Run(adamID, audiom3u8url, token, mediaUserToken, true) + _ = runv3.ExtMvData(audiokeyAndUrls, audPath) + + //tags + tags := []string{ + "tool=", + fmt.Sprintf("artist=%s", MVInfo.Data[0].Attributes.ArtistName), + fmt.Sprintf("title=%s", MVInfo.Data[0].Attributes.Name), + fmt.Sprintf("genre=%s", MVInfo.Data[0].Attributes.GenreNames[0]), + fmt.Sprintf("created=%s", MVInfo.Data[0].Attributes.ReleaseDate), + fmt.Sprintf("ISRC=%s", MVInfo.Data[0].Attributes.Isrc), + } + + // ContentRating tag + if MVInfo.Data[0].Attributes.ContentRating == "explicit" { + tags = append(tags, "rating=1") + } else if MVInfo.Data[0].Attributes.ContentRating == "clean" { + tags = append(tags, "rating=2") + } else { + tags = append(tags, "rating=0") + } + + //根据情况额外添加可使用的tags + if meta != nil { + if meta.Data[0].Type == "playlists" && !Config.UseSongInfoForPlaylist { + tags = append(tags, "disk=1/1") + tags = append(tags, fmt.Sprintf("album=%s", meta.Data[0].Attributes.Name)) + tags = append(tags, fmt.Sprintf("track=%d", trackNum)) + tags = append(tags, fmt.Sprintf("tracknum=%d/%d", trackNum, trackTotal)) + tags = append(tags, fmt.Sprintf("album_artist=%s", meta.Data[0].Attributes.ArtistName)) + tags = append(tags, fmt.Sprintf("performer=%s", meta.Data[0].Relationships.Tracks.Data[index].Attributes.ArtistName)) + tags = append(tags, fmt.Sprintf("copyright=%s", meta.Data[0].Attributes.Copyright)) + tags = append(tags, fmt.Sprintf("UPC=%s", meta.Data[0].Attributes.Upc)) + } else { + tags = append(tags, fmt.Sprintf("album=%s", meta.Data[0].Relationships.Tracks.Data[index].Attributes.AlbumName)) + tags = append(tags, fmt.Sprintf("disk=%d/%d", meta.Data[0].Relationships.Tracks.Data[index].Attributes.DiscNumber, meta.Data[0].Relationships.Tracks.Data[trackTotal-1].Attributes.DiscNumber)) + tags = append(tags, fmt.Sprintf("track=%d", meta.Data[0].Relationships.Tracks.Data[index].Attributes.TrackNumber)) + tags = append(tags, fmt.Sprintf("tracknum=%d/%d", meta.Data[0].Relationships.Tracks.Data[index].Attributes.TrackNumber, meta.Data[0].Attributes.TrackCount)) + tags = append(tags, fmt.Sprintf("album_artist=%s", meta.Data[0].Attributes.ArtistName)) + tags = append(tags, fmt.Sprintf("performer=%s", meta.Data[0].Relationships.Tracks.Data[index].Attributes.ArtistName)) + tags = append(tags, fmt.Sprintf("copyright=%s", meta.Data[0].Attributes.Copyright)) + tags = append(tags, fmt.Sprintf("UPC=%s", meta.Data[0].Attributes.Upc)) + } + } else { + tags = append(tags, fmt.Sprintf("album=%s", MVInfo.Data[0].Attributes.AlbumName)) + tags = append(tags, fmt.Sprintf("disk=%d", MVInfo.Data[0].Attributes.DiscNumber)) + tags = append(tags, fmt.Sprintf("track=%d", MVInfo.Data[0].Attributes.TrackNumber)) + tags = append(tags, fmt.Sprintf("tracknum=%d", MVInfo.Data[0].Attributes.TrackNumber)) + //tags = append(tags, fmt.Sprintf("album_artist=%s", MVInfo.Data[0].Attributes.ArtistName)) + tags = append(tags, fmt.Sprintf("performer=%s", MVInfo.Data[0].Attributes.ArtistName)) + } + + // Extract and save thumbnail if enabled + var covPath string + //强制嵌入封面 + if true { + // Get the highest quality thumbnail URL from the MV info + thumbURL := MVInfo.Data[0].Attributes.Artwork.URL + + // Generate base name without extension + baseThumbName := forbiddenNames.ReplaceAllString(mvSaveName, "_") + "_thumbnail" + + // Download and save thumbnail + covPath, err = writeCover(saveDir, baseThumbName, thumbURL) + if err != nil { + fmt.Println("Failed to save MV thumbnail:", err) + } else { + //fmt.Println("MV thumbnail saved successfully") + tags = append(tags, fmt.Sprintf("cover=%s", covPath)) + } + } + + //mux and add tag + tagsString := strings.Join(tags, ":") + muxCmd := exec.Command("MP4Box", "-itags", tagsString, "-quiet", "-add", vidPath, "-add", audPath, "-keep-utc", "-new", mvOutPath) + fmt.Printf("MV Remuxing...") + if err := muxCmd.Run(); err != nil { + fmt.Printf("MV mux failed: %v\n", err) + return err + } + fmt.Printf("\rMV Remuxed. \n") + defer os.Remove(vidPath) + defer os.Remove(audPath) + defer os.Remove(covPath) + + return nil +} + +func extractMvAudio(c string) (string, error) { + MediaUrl, err := url.Parse(c) + if err != nil { + return "", err + } + + resp, err := http.Get(c) + if err != nil { + return "", err + } + defer resp.Body.Close() + + if resp.StatusCode != http.StatusOK { + return "", errors.New(resp.Status) + } + + body, err := io.ReadAll(resp.Body) + if err != nil { + return "", err + } + + audioString := string(body) + from, listType, err := m3u8.DecodeFrom(strings.NewReader(audioString), true) + if err != nil || listType != m3u8.MASTER { + return "", errors.New("m3u8 not of media type") + } + + audio := from.(*m3u8.MasterPlaylist) + + var audioPriority = []string{"audio-atmos", "audio-ac3", "audio-stereo-256"} + if Config.MVAudioType == "ac3" { + audioPriority = []string{"audio-ac3", "audio-stereo-256"} + } else if Config.MVAudioType == "aac" { + audioPriority = []string{"audio-stereo-256"} + } + + re := regexp.MustCompile(`_gr(\d+)_`) + + type AudioStream struct { + URL string + Rank int + GroupID string + } + var audioStreams []AudioStream + + for _, variant := range audio.Variants { + for _, audiov := range variant.Alternatives { + if audiov.URI != "" { + for _, priority := range audioPriority { + if audiov.GroupId == priority { + matches := re.FindStringSubmatch(audiov.URI) + if len(matches) == 2 { + var rank int + fmt.Sscanf(matches[1], "%d", &rank) + streamUrl, _ := MediaUrl.Parse(audiov.URI) + audioStreams = append(audioStreams, AudioStream{ + URL: streamUrl.String(), + Rank: rank, + GroupID: audiov.GroupId, + }) + } + } + } + } + } + } + + if len(audioStreams) == 0 { + return "", errors.New("no suitable audio stream found") + } + + sort.Slice(audioStreams, func(i, j int) bool { + return audioStreams[i].Rank > audioStreams[j].Rank + }) + fmt.Println("Audio: " + audioStreams[0].GroupID) + return audioStreams[0].URL, nil +} + +func checkM3u8(b string, f string) (string, error) { + var EnhancedHls string + if Config.GetM3u8FromDevice { + adamID := b + conn, err := net.Dial("tcp", Config.GetM3u8Port) + if err != nil { + fmt.Println("Error connecting to device:", err) + return "none", err + } + defer conn.Close() + if f == "song" { + fmt.Println("Connected to device") + } + + // Send the length of adamID and the adamID itself + adamIDBuffer := []byte(adamID) + lengthBuffer := []byte{byte(len(adamIDBuffer))} + + // Write length and adamID to the connection + _, err = conn.Write(lengthBuffer) + if err != nil { + fmt.Println("Error writing length to device:", err) + return "none", err + } + + _, err = conn.Write(adamIDBuffer) + if err != nil { + fmt.Println("Error writing adamID to device:", err) + return "none", err + } + + // Read the response (URL) from the device + response, err := bufio.NewReader(conn).ReadBytes('\n') + if err != nil { + fmt.Println("Error reading response from device:", err) + return "none", err + } + + // Trim any newline characters from the response + + response = bytes.TrimSpace(response) + if len(response) > 0 { + if f == "song" { + fmt.Println("Received URL:", string(response)) + } + EnhancedHls = string(response) + } else { + fmt.Println("Received an empty response") + } + } + return EnhancedHls, nil +} + +func formatAvailability(available bool, quality string) string { + if !available { + return "Not Available" + } + return quality +} + +func extractMedia(b string, more_mode bool) (string, string, error) { + masterUrl, err := url.Parse(b) + if err != nil { + return "", "", err + } + resp, err := http.Get(b) + if err != nil { + return "", "", err + } + defer resp.Body.Close() + if resp.StatusCode != http.StatusOK { + return "", "", errors.New(resp.Status) + } + body, err := io.ReadAll(resp.Body) + if err != nil { + return "", "", err + } + masterString := string(body) + from, listType, err := m3u8.DecodeFrom(strings.NewReader(masterString), true) + if err != nil || listType != m3u8.MASTER { + return "", "", errors.New("m3u8 not of master type") + } + master := from.(*m3u8.MasterPlaylist) + var streamUrl *url.URL + sort.Slice(master.Variants, func(i, j int) bool { + return master.Variants[i].AverageBandwidth > master.Variants[j].AverageBandwidth + }) + if debug_mode && more_mode { + fmt.Println("\nDebug: All Available Variants:") + var data [][]string + for _, variant := range master.Variants { + data = append(data, []string{variant.Codecs, variant.Audio, fmt.Sprint(variant.Bandwidth)}) + //fmt.Printf("Codec: %s, Audio: %s, Bandwidth: %d\n", + //variant.Codecs, variant.Audio, variant.Bandwidth) + } + table := tablewriter.NewWriter(os.Stdout) + table.SetHeader([]string{"Codec", "Audio", "Bandwidth"}) + //table.SetFooter([]string{"", "", "Total", "$146.93"}) + table.SetAutoMergeCells(true) + //table.SetAutoMergeCellsByColumnIndex([]int{1,2,3}) + table.SetRowLine(true) + table.AppendBulk(data) + table.Render() + + var hasAAC, hasLossless, hasHiRes, hasAtmos, hasDolbyAudio bool + var aacQuality, losslessQuality, hiResQuality, atmosQuality, dolbyAudioQuality string + + // Check for all formats + for _, variant := range master.Variants { + if variant.Codecs == "mp4a.40.2" { // AAC + hasAAC = true + split := strings.Split(variant.Audio, "-") + if len(split) >= 3 { + bitrate, _ := strconv.Atoi(split[2]) + currentBitrate := 0 + if aacQuality != "" { + current := strings.Split(aacQuality, " | ")[2] + current = strings.Split(current, " ")[0] + currentBitrate, _ = strconv.Atoi(current) + } + if bitrate > currentBitrate { + aacQuality = fmt.Sprintf("AAC | 2 Channel | %d kbps", bitrate) + } + } + } else if variant.Codecs == "ec-3" && strings.Contains(variant.Audio, "atmos") { // Dolby Atmos + hasAtmos = true + split := strings.Split(variant.Audio, "-") + if len(split) > 0 { + bitrateStr := split[len(split)-1] + // Remove leading "2" if present in "2768" + if len(bitrateStr) == 4 && bitrateStr[0] == '2' { + bitrateStr = bitrateStr[1:] + } + bitrate, _ := strconv.Atoi(bitrateStr) + currentBitrate := 0 + if atmosQuality != "" { + current := strings.Split(strings.Split(atmosQuality, " | ")[2], " ")[0] + currentBitrate, _ = strconv.Atoi(current) + } + if bitrate > currentBitrate { + atmosQuality = fmt.Sprintf("E-AC-3 | 16 Channel | %d kbps", bitrate) + } + } + } else if variant.Codecs == "alac" { // ALAC (Lossless or Hi-Res) + split := strings.Split(variant.Audio, "-") + if len(split) >= 3 { + bitDepth := split[len(split)-1] + sampleRate := split[len(split)-2] + sampleRateInt, _ := strconv.Atoi(sampleRate) + if sampleRateInt > 48000 { // Hi-Res + hasHiRes = true + hiResQuality = fmt.Sprintf("ALAC | 2 Channel | %s-bit/%d kHz", bitDepth, sampleRateInt/1000) + } else { // Standard Lossless + hasLossless = true + losslessQuality = fmt.Sprintf("ALAC | 2 Channel | %s-bit/%d kHz", bitDepth, sampleRateInt/1000) + } + } + } else if variant.Codecs == "ac-3" { // Dolby Audio + hasDolbyAudio = true + split := strings.Split(variant.Audio, "-") + if len(split) > 0 { + bitrate, _ := strconv.Atoi(split[len(split)-1]) + dolbyAudioQuality = fmt.Sprintf("AC-3 | 16 Channel | %d kbps", bitrate) + } + } + } + + fmt.Println("Available Audio Formats:") + fmt.Println("------------------------") + fmt.Printf("AAC : %s\n", formatAvailability(hasAAC, aacQuality)) + fmt.Printf("Lossless : %s\n", formatAvailability(hasLossless, losslessQuality)) + fmt.Printf("Hi-Res Lossless : %s\n", formatAvailability(hasHiRes, hiResQuality)) + fmt.Printf("Dolby Atmos : %s\n", formatAvailability(hasAtmos, atmosQuality)) + fmt.Printf("Dolby Audio : %s\n", formatAvailability(hasDolbyAudio, dolbyAudioQuality)) + fmt.Println("------------------------") + + return "", "", nil + } + var Quality string + for _, variant := range master.Variants { + if dl_atmos { + if variant.Codecs == "ec-3" && strings.Contains(variant.Audio, "atmos") { + if debug_mode && !more_mode { + fmt.Printf("Debug: Found Dolby Atmos variant - %s (Bitrate: %d kbps)\n", + variant.Audio, variant.Bandwidth/1000) + } + split := strings.Split(variant.Audio, "-") + length := len(split) + length_int, err := strconv.Atoi(split[length-1]) + if err != nil { + return "", "", err + } + if length_int <= Config.AtmosMax { + if !debug_mode && !more_mode { + fmt.Printf("%s\n", variant.Audio) + } + streamUrlTemp, err := masterUrl.Parse(variant.URI) + if err != nil { + return "", "", err + } + streamUrl = streamUrlTemp + Quality = fmt.Sprintf("%s kbps", split[len(split)-1]) + break + } + } else if variant.Codecs == "ac-3" { // Add Dolby Audio support + if debug_mode && !more_mode { + fmt.Printf("Debug: Found Dolby Audio variant - %s (Bitrate: %d kbps)\n", + variant.Audio, variant.Bandwidth/1000) + } + streamUrlTemp, err := masterUrl.Parse(variant.URI) + if err != nil { + return "", "", err + } + streamUrl = streamUrlTemp + split := strings.Split(variant.Audio, "-") + Quality = fmt.Sprintf("%s kbps", split[len(split)-1]) + break + } + } else if dl_aac { + if variant.Codecs == "mp4a.40.2" { + if debug_mode && !more_mode { + fmt.Printf("Debug: Found AAC variant - %s (Bitrate: %d)\n", variant.Audio, variant.Bandwidth) + } + aacregex := regexp.MustCompile(`audio-stereo-\d+`) + replaced := aacregex.ReplaceAllString(variant.Audio, "aac") + if replaced == Config.AacType { + if !debug_mode && !more_mode { + fmt.Printf("%s\n", variant.Audio) + } + streamUrlTemp, err := masterUrl.Parse(variant.URI) + if err != nil { + panic(err) + } + streamUrl = streamUrlTemp + split := strings.Split(variant.Audio, "-") + Quality = fmt.Sprintf("%s kbps", split[2]) + break + } + } + } else { + if variant.Codecs == "alac" { + split := strings.Split(variant.Audio, "-") + length := len(split) + length_int, err := strconv.Atoi(split[length-2]) + if err != nil { + return "", "", err + } + if length_int <= Config.AlacMax { + if !debug_mode && !more_mode { + fmt.Printf("%s-bit / %s Hz\n", split[length-1], split[length-2]) + } + streamUrlTemp, err := masterUrl.Parse(variant.URI) + if err != nil { + panic(err) + } + streamUrl = streamUrlTemp + KHZ := float64(length_int) / 1000.0 + Quality = fmt.Sprintf("%sB-%.1fkHz", split[length-1], KHZ) + break + } + } + } + } + if streamUrl == nil { + return "", "", errors.New("no codec found") + } + return streamUrl.String(), Quality, nil +} +func extractVideo(c string) (string, error) { + MediaUrl, err := url.Parse(c) + if err != nil { + return "", err + } + + resp, err := http.Get(c) + if err != nil { + return "", err + } + defer resp.Body.Close() + + if resp.StatusCode != http.StatusOK { + return "", errors.New(resp.Status) + } + + body, err := io.ReadAll(resp.Body) + if err != nil { + return "", err + } + videoString := string(body) + + from, listType, err := m3u8.DecodeFrom(strings.NewReader(videoString), true) + if err != nil || listType != m3u8.MASTER { + return "", errors.New("m3u8 not of media type") + } + + video := from.(*m3u8.MasterPlaylist) + + re := regexp.MustCompile(`_(\d+)x(\d+)`) + + var streamUrl *url.URL + sort.Slice(video.Variants, func(i, j int) bool { + return video.Variants[i].AverageBandwidth > video.Variants[j].AverageBandwidth + }) + + maxHeight := Config.MVMax + + for _, variant := range video.Variants { + matches := re.FindStringSubmatch(variant.URI) + if len(matches) == 3 { + height := matches[2] + var h int + _, err := fmt.Sscanf(height, "%d", &h) + if err != nil { + continue + } + if h <= maxHeight { + streamUrl, err = MediaUrl.Parse(variant.URI) + if err != nil { + return "", err + } + fmt.Println("Video: " + variant.Resolution + "-" + variant.VideoRange) + break + } + } + } + + if streamUrl == nil { + return "", errors.New("no suitable video stream found") + } + + return streamUrl.String(), nil +} + +func getInfoFromAdam(adamId string, token string, storefront string) (*structs.SongData, error) { + request, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/songs/%s", storefront, adamId), nil) + if err != nil { + return nil, err + } + query := url.Values{} + query.Set("extend", "extendedAssetUrls") + query.Set("include", "albums") + query.Set("l", Config.Language) + request.URL.RawQuery = query.Encode() + + request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", token)) + request.Header.Set("User-Agent", "iTunes/12.11.3 (Windows; Microsoft Windows 10 x64 Professional Edition (Build 19041); x64) AppleWebKit/7611.1022.4001.1 (dt:2)") + request.Header.Set("Origin", "https://music.apple.com") + + do, err := http.DefaultClient.Do(request) + if err != nil { + return nil, err + } + defer do.Body.Close() + if do.StatusCode != http.StatusOK { + return nil, errors.New(do.Status) + } + + obj := new(structs.ApiResult) + err = json.NewDecoder(do.Body).Decode(&obj) + if err != nil { + return nil, err + } + + for _, d := range obj.Data { + if d.ID == adamId { + return &d, nil + } + } + return nil, nil +} + +func getMVInfoFromAdam(adamId string, token string, storefront string) (*structs.AutoGeneratedMusicVideo, error) { + request, err := http.NewRequest("GET", fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/music-videos/%s", storefront, adamId), nil) + if err != nil { + return nil, err + } + query := url.Values{} + query.Set("l", Config.Language) + request.URL.RawQuery = query.Encode() + request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", token)) + request.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") + request.Header.Set("Origin", "https://music.apple.com") + + do, err := http.DefaultClient.Do(request) + if err != nil { + return nil, err + } + defer do.Body.Close() + if do.StatusCode != http.StatusOK { + return nil, errors.New(do.Status) + } + + obj := new(structs.AutoGeneratedMusicVideo) + err = json.NewDecoder(do.Body).Decode(&obj) + if err != nil { + return nil, err + } + + return obj, nil +} + +func getToken() (string, error) { + req, err := http.NewRequest("GET", "https://beta.music.apple.com", nil) + if err != nil { + return "", err + } + + resp, err := http.DefaultClient.Do(req) + if err != nil { + return "", err + } + defer resp.Body.Close() + + body, err := io.ReadAll(resp.Body) + if err != nil { + return "", err + } + + regex := regexp.MustCompile(`/assets/index-legacy-[^/]+\.js`) + indexJsUri := regex.FindString(string(body)) + + req, err = http.NewRequest("GET", "https://beta.music.apple.com"+indexJsUri, nil) + if err != nil { + return "", err + } + + resp, err = http.DefaultClient.Do(req) + if err != nil { + return "", err + } + defer resp.Body.Close() + + body, err = io.ReadAll(resp.Body) + if err != nil { + return "", err + } + + regex = regexp.MustCompile(`eyJh([^"]*)`) + token := regex.FindString(string(body)) + + return token, nil +}