diff --git a/main.go b/main.go index cb9deee..635a32c 100644 --- a/main.go +++ b/main.go @@ -41,6 +41,7 @@ var ( type Config struct { MediaUserToken string `yaml:"media-user-token"` SaveLrcFile bool `yaml:"save-lrc-file"` + SaveAnimatedArtwork bool `yaml:"save-animated-artwork"` EmbedLrc bool `yaml:"embed-lrc"` EmbedCover bool `yaml:"embed-cover"` CoverSize string `yaml:"cover-size"` @@ -1153,6 +1154,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) @@ -1389,6 +1391,25 @@ 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.MotionSquareVideo1x1.Video != "" { + motionvideoUrl, err := extractVideo(meta.Data[0].Attributes.EditorialVideo.MotionSquareVideo1x1.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 { + 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) + } + } + } trackTotal := len(meta.Data[0].Relationships.Tracks.Data) for trackNum, track := range meta.Data[0].Relationships.Tracks.Data { trackNum++ @@ -1857,7 +1878,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...") track, err := http.Get(url) @@ -2307,6 +2366,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 {