fix tiktok title

iawia002 · Jul 21, 2024 · 93da918 · 93da918
1 parent 0883b0f
commit 93da918
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/extractors/tiktok/tiktok.go b/extractors/tiktok/tiktok.go
@@ -43,14 +43,28 @@ func (e *extractor) Extract(url string, option extractors.Options) ([]*extractor
 
 	titleMatcherRegExp := regexp.MustCompile(`<title[^>]*>([^<]+)</title>`)
 
+	titleMatcherRegExpOpt := regexp.MustCompile(`"desc":"([^"]*)"`)
+
 	titleMatcher := titleMatcherRegExp.FindStringSubmatch(html)
 
+	titleMatcherOpt := titleMatcherRegExpOpt.FindStringSubmatch(html)
+
 	if len(titleMatcher) == 0 {
 		return nil, errors.WithStack(extractors.ErrURLParseFailed)
 	}
 
 	title := titleMatcher[1]
 
+	if title == "TikTok - Make Your Day" {
+		if len(titleMatcherOpt[1]) > 64 {
+			cutoff := titleMatcherOpt[1][:64]
+			lastSpace := strings.LastIndex(cutoff, " ")
+			title = titleMatcherOpt[1][:lastSpace]
+		} else {
+			title = titleMatcherOpt[1]
+		}
+	}
+
 	titleArr := strings.Split(title, "|")
 
 	if len(titleArr) == 1 {