extractors/youku: ⬆️ ccode, 🔧 title, close #137

iawia002 · Jun 2, 2018 · f1be07e · f1be07e
1 parent c329a53
commit f1be07e
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 11 deletions.
diff --git a/extractors/youku.go b/extractors/youku.go
@@ -57,7 +57,7 @@ const youkuReferer = "https://v.youku.com"
 
 // var ccodes = []string{"0510", "0502", "0507", "0508", "0512", "0513", "0514", "0503", "0590"}
 
-var ccodes = []string{"0510"}
+var ccodes = []string{"0519"}
 
 func youkuUps(vid string) youkuData {
 	var url string
@@ -134,7 +134,9 @@ func Youku(url string) downloader.VideoData {
 	}
 	format := genData(youkuData.Data)
 	var title string
-	if youkuData.Data.Show.Title == "" {
+	if youkuData.Data.Show.Title == "" || strings.Contains(
+		youkuData.Data.Video.Title, youkuData.Data.Show.Title,
+	) {
 		title = youkuData.Data.Video.Title
 	} else {
 		title = fmt.Sprintf("%s %s", youkuData.Data.Show.Title, youkuData.Data.Video.Title)

diff --git a/extractors/youku_test.go b/extractors/youku_test.go
@@ -22,15 +22,24 @@ func TestYouku(t *testing.T) {
 				Quality: "mp4hd3v2 1920x1080",
 			},
 		},
-		// {
-		// 	name: "normal test",
-		// 	args: test.Args{
-		// 		URL:     "http://v.youku.com/v_show/id_XMzQ1MTAzNjQwNA==.html",
-		// 		Title:   "这！就是街舞 第一季 第3期：百强“互杀”队长不忍直视",
-		// 		Size:    750911635,
-		// 		Quality: "mp4hd2v2 1280x720",
-		// 	},
-		// },
+		{
+			name: "normal test",
+			args: test.Args{
+				URL:     "http://v.youku.com/v_show/id_XMzQ1MTAzNjQwNA==.html",
+				Title:   "这！就是街舞 第一季 第3期：百强“互杀”队长不忍直视",
+				Size:    1419459808,
+				Quality: "mp4hd3v2 1920x1080",
+			},
+		},
+		{
+			name: "title test",
+			args: test.Args{
+				URL:     "http://v.youku.com/v_show/id_XMzYzMDgxMTAyNA.html",
+				Title:   "想看你微笑 01",
+				Size:    568769655,
+				Quality: "mp4hd3v2 1920x1080",
+			},
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {