Merge pull request #52 from itouakirai/main

fix: lrc trans panic
zhaarey · Feb 12, 2025 · 75dcf30 · 75dcf30
2 parents 1e9cc05 + d6a32c0
commit 75dcf30
Showing 1 changed file with 22 additions and 15 deletions.
diff --git a/main.go b/main.go
@@ -1437,16 +1437,19 @@ func conventSyllableTTMLToLRC(ttml string) (string, error) {
 	}
 	divs := parsedTTML.FindElement("tt").FindElement("body").FindElements("div")
 	//get trans
-	Metadata := parsedTTML.FindElement("tt").FindElement("head").FindElement("metadata")
-	if len(Metadata.FindElements("iTunesMetadata")) > 0 {
-		iTunesMetadata := Metadata.FindElement("iTunesMetadata")
-		if len(iTunesMetadata.FindElements("translations")) > 0 {
-			if len(iTunesMetadata.FindElement("translations").FindElements("translation")) > 0 {
-				divs = iTunesMetadata.FindElement("translations").FindElements("translation")
+	if len(parsedTTML.FindElement("tt").FindElements("head")) > 0 {
+		if len(parsedTTML.FindElement("tt").FindElement("head").FindElements("metadata")) > 0 {
+			Metadata := parsedTTML.FindElement("tt").FindElement("head").FindElement("metadata")
+			if len(Metadata.FindElements("iTunesMetadata")) > 0 {
+				iTunesMetadata := Metadata.FindElement("iTunesMetadata")
+				if len(iTunesMetadata.FindElements("translations")) > 0 {
+					if len(iTunesMetadata.FindElement("translations").FindElements("translation")) > 0 {
+						divs = iTunesMetadata.FindElement("translations").FindElements("translation")
+					}
+				}
 			}
 		}
 	}
-
 	for _, div := range divs {
 		for _, item := range div.ChildElements() {
 			var lrcSyllables []string
@@ -1549,14 +1552,18 @@ func conventTTMLToLRC(ttml string) (string, error) {
 			}
 			var text string
 			//GET trans
-			Metadata := parsedTTML.FindElement("tt").FindElement("head").FindElement("metadata")
-			if len(Metadata.FindElements("iTunesMetadata")) > 0 {
-				iTunesMetadata := Metadata.FindElement("iTunesMetadata")
-				if len(iTunesMetadata.FindElements("translations")) > 0 {
-					if len(iTunesMetadata.FindElement("translations").FindElements("translation")) > 0 {
-						xpath := fmt.Sprintf("//text[@for='%s']", lyric.SelectAttr("itunes:key").Value)
-						trans := iTunesMetadata.FindElement("translations").FindElement("translation").FindElement(xpath)
-						lyric = trans
+			if len(parsedTTML.FindElement("tt").FindElements("head")) > 0 {
+				if len(parsedTTML.FindElement("tt").FindElement("head").FindElements("metadata")) > 0 {
+					Metadata := parsedTTML.FindElement("tt").FindElement("head").FindElement("metadata")
+					if len(Metadata.FindElements("iTunesMetadata")) > 0 {
+						iTunesMetadata := Metadata.FindElement("iTunesMetadata")
+						if len(iTunesMetadata.FindElements("translations")) > 0 {
+							if len(iTunesMetadata.FindElement("translations").FindElements("translation")) > 0 {
+								xpath := fmt.Sprintf("//text[@for='%s']", lyric.SelectAttr("itunes:key").Value)
+								trans := iTunesMetadata.FindElement("translations").FindElement("translation").FindElement(xpath)
+								lyric = trans
+							}
+						}
 					}
 				}
 			}