From 5bcbf17bf9fc4568ec2a3c0c56d5a4258c3155dd Mon Sep 17 00:00:00 2001
From: itouakirai <85016486+itouakirai@users.noreply.github.com>
Date: Fri, 28 Jun 2024 23:00:01 +0800
Subject: [PATCH] add animated artwork download

---
 main_select.go | 80 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)

diff --git a/main_select.go b/main_select.go
index 2ceb885..08778eb 100644
--- a/main_select.go
+++ b/main_select.go
@@ -41,6 +41,8 @@ var (
 type Config struct {
 	MediaUserToken          string `yaml:"media-user-token"`
 	SaveLrcFile             bool   `yaml:"save-lrc-file"`
+	SaveAnimatedArtwork     bool   `yaml:"save-animated-artwork"`
+	EmbyAnimatedArtwork     bool   `yaml:"emby-animated-artwork"`
 	EmbedLrc                bool   `yaml:"embed-lrc"`
 	EmbedCover              bool   `yaml:"embed-cover"`
 	CoverSize               string `yaml:"cover-size"`
@@ -1036,6 +1038,7 @@ func getMeta(albumId string, token string, storefront string) (*AutoGenerated, e
 	query.Set("fields[artists]", "name")
 	query.Set("fields[albums:albums]", "artistName,artwork,name,releaseDate,url")
 	query.Set("fields[record-labels]", "name")
+	query.Set("extend", "editorialVideo")
 	// query.Set("l", "en-gb")
 	req.URL.RawQuery = query.Encode()
 	do, err := http.DefaultClient.Do(req)
@@ -1277,6 +1280,36 @@ func rip(albumId string, token string, storefront string, userToken string) erro
 	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.")
+		motionvideoUrl, err := extractVideo(meta.Data[0].Attributes.EditorialVideo.MotionDetailSquare.Video)
+		if err != nil {
+			fmt.Println("no motion video.\n", err)
+		}
+		exists, err := fileExists(filepath.Join(sanAlbumFolder, "animated_artwork.mp4"))
+		if err != nil {
+			fmt.Println("Failed to check if animated artwork exists.")
+		}
+		if exists {
+			fmt.Println("Animated artwork already exists locally.")
+		} else {
+			fmt.Println("Animation Artwork Downloading...")
+			cmd := exec.Command("ffmpeg", "-loglevel", "quiet", "-y", "-i", motionvideoUrl, "-c", "copy", filepath.Join(sanAlbumFolder, "animated_artwork.mp4"))
+			if err := cmd.Run(); err != nil {
+				fmt.Printf("animated artwork dl err: %v\n", err)
+			} else {
+				fmt.Println("Animation Artwork Downloaded")
+			}
+			if config.EmbyAnimatedArtwork {
+				cmd2 := exec.Command("ffmpeg", "-i", filepath.Join(sanAlbumFolder, "animated_artwork.mp4"), "-vf", "scale=440:-1", "-r", "24", "-f", "gif", filepath.Join(sanAlbumFolder, "folder.jpg"))
+				if err := cmd2.Run(); err != nil {
+					fmt.Printf("animated artwork to gif err: %v\n", err)
+				}
+			}
+
+		}
+	}
 	trackTotal := len(meta.Data[0].Relationships.Tracks.Data)
 	arr := make([]int, trackTotal)
 	for i := 0; i < trackTotal; i++ {
@@ -1687,6 +1720,45 @@ func extractMedia(b string) (string, []string, error) {
 	}
 	return streamUrl.String(), keys, 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)
+	var streamUrl *url.URL
+	sort.Slice(video.Variants, func(i, j int) bool {
+		return video.Variants[i].AverageBandwidth > video.Variants[j].AverageBandwidth
+	})
+	if len(video.Variants) > 0 {
+		highestBandwidthVariant := video.Variants[0]
+		streamUrl, err = MediaUrl.Parse(highestBandwidthVariant.URI)
+		if err != nil {
+			return "", err
+		}
+	}
+	if streamUrl == nil {
+		return "", errors.New("no video codec found")
+	}
+	return streamUrl.String(), nil
+}
 
 func extractSong(url string) (*SongInfo, error) {
 	fmt.Println("Downloading...")
@@ -2137,6 +2209,14 @@ type AutoGenerated struct {
 				Kind string `json:"kind"`
 			} `json:"playParams"`
 			IsCompilation bool `json:"isCompilation"`
+			EditorialVideo struct {
+				MotionDetailSquare struct {
+					Video string `json:"video"`
+				} `json:"motionDetailSquare"`
+				MotionSquareVideo1x1 struct {
+					Video string `json:"video"`
+				} `json:"motionSquareVideo1x1"`
+			} `json:"editorialVideo"`
 		} `json:"attributes"`
 		Relationships struct {
 			RecordLabels struct {