From 460255c88d66ba601ec2a351280f26c56f141ad2 Mon Sep 17 00:00:00 2001
From: itouakirai <85016486+itouakirai@users.noreply.github.com>
Date: Fri, 27 Sep 2024 04:27:18 +0800
Subject: [PATCH] Update main.go

---
 main.go | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index 24af3d5..0440692 100644
--- a/main.go
+++ b/main.go
@@ -52,6 +52,7 @@ type Config struct {
 	MediaUserToken          string `yaml:"media-user-token"`
 	AuthorizationToken      string `yaml:"authorization-token"`
 	SaveLrcFile             bool   `yaml:"save-lrc-file"`
+	LrcType              string `yaml:"lrc-type"`
 	LrcFormat               string `yaml:"lrc-format"`
 	SaveAnimatedArtwork     bool   `yaml:"save-animated-artwork"`
 	EmbyAnimatedArtwork     bool   `yaml:"emby-animated-artwork"`
@@ -1318,7 +1319,7 @@ func getMeta(albumId string, token string, storefront string) (*AutoGenerated, e
 
 func getSongLyrics(songId string, storefront string, token string, userToken string) (string, error) {
 	req, err := http.NewRequest("GET",
-		fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/songs/%s/%s", storefront, songId, config.LrcFormat), nil)
+		fmt.Sprintf("https://amp-api.music.apple.com/v1/catalog/%s/songs/%s/%s", storefront, songId, config.LrcType), nil)
 	if err != nil {
 		return "", err
 	}
@@ -1710,7 +1711,7 @@ func rip(albumId string, token string, storefront string, userToken string) erro
 				filename = fmt.Sprintf("%s.ec3", forbiddenNames.ReplaceAllString(songName, "_"))
 			}
 			m4afilename := fmt.Sprintf("%s.m4a", forbiddenNames.ReplaceAllString(songName, "_"))
-			lrcFilename := fmt.Sprintf("%s.lrc", forbiddenNames.ReplaceAllString(songName, "_"))
+			lrcFilename := fmt.Sprintf("%s.%s", forbiddenNames.ReplaceAllString(songName, "_"), config.LrcFormat)
 			trackPath := filepath.Join(sanAlbumFolder, filename)
 			m4atrackPath := filepath.Join(sanAlbumFolder, m4afilename)
 			var lrc string = ""
@@ -1718,6 +1719,17 @@ func rip(albumId string, token string, storefront string, userToken string) erro
 				ttml, err := getSongLyrics(track.ID, storefront, token, userToken)
 				if err != nil {
 					fmt.Println("Failed to get lyrics")
+				} else if config.LrcFormat == "ttml"{
+					if config.SaveLrcFile {
+						lrc = ttml
+						err := writeLyrics(sanAlbumFolder, lrcFilename, lrc)
+						if err != nil {
+							fmt.Printf("Failed to write lyrics")
+						}
+						if !config.EmbedLrc {
+							lrc = ""
+						}
+					}
 				} else {
 					lrc, err = conventTTMLToLRC(ttml)
 					if err != nil {