extractors/vimeo: Add basic support

README.md

@@ -207,6 +207,7 @@ YouTube | <https://www.youtube.com> | ✓ | | |
 爱奇艺 | <https://www.iqiyi.com> | ✓ | | |
 芒果TV | <https://www.mgtv.com> | ✓ | | |
 Tumblr | <https://www.tumblr.com> | | ✓ | |
+Vimeo | <https://vimeo.com> | ✓ | | |
 
 
 ## Contributing

extractors/vimeo.go

@@ -0,0 +1,80 @@
+package extractors
+
+import (
+	"encoding/json"
+	"strings"
+
+	"github.com/iawia002/annie/downloader"
+	"github.com/iawia002/annie/request"
+	"github.com/iawia002/annie/utils"
+)
+
+type vimeoProgressive struct {
+	Profile int    `json:"profile"`
+	Width   int    `json:"width"`
+	Height  int    `json:"height"`
+	Quality string `json:"quality"`
+	URL     string `json:"url"`
+}
+
+type vimeoFiles struct {
+	Progressive []vimeoProgressive `json:"progressive"`
+}
+
+type vimeoRequest struct {
+	Files vimeoFiles `json:"files"`
+}
+
+type vimeoVideo struct {
+	Title string `json:"title"`
+}
+
+type vimeo struct {
+	Request vimeoRequest `json:"request"`
+	Video   vimeoVideo   `json:"video"`
+}
+
+func bestQuality(progressives []vimeoProgressive) vimeoProgressive {
+	var highestProfile int
+	var data vimeoProgressive
+	for _, progressive := range progressives {
+		if progressive.Profile > highestProfile {
+			highestProfile = progressive.Profile
+			data = progressive
+		}
+	}
+	return data
+}
+
+// Vimeo download function
+func Vimeo(url string) downloader.VideoData {
+	var html string
+	var vid string
+	if strings.Contains(url, "player.vimeo.com") {
+		html = request.Get(url)
+	} else {
+		vid = utils.MatchOneOf(url, `vimeo\.com/(\d+)`)[1]
+		html = request.Get("https://player.vimeo.com/video/" + vid)
+	}
+	jsonString := utils.MatchOneOf(html, `{var a=(.+?);`)[1]
+	var vimeoData vimeo
+	json.Unmarshal([]byte(jsonString), &vimeoData)
+	video := bestQuality(vimeoData.Request.Files.Progressive)
+
+	size := request.Size(video.URL, url)
+	urlData := downloader.URLData{
+		URL:  video.URL,
+		Size: size,
+		Ext:  "mp4",
+	}
+	data := downloader.VideoData{
+		Site:    "Vimeo vimeo.com",
+		Title:   vimeoData.Video.Title,
+		Type:    "video",
+		URLs:    []downloader.URLData{urlData},
+		Size:    size,
+		Quality: video.Quality,
+	}
+	data.Download(url)
+	return data
+}

extractors/vimeo_test.go

@@ -0,0 +1,41 @@
+package extractors
+
+import (
+	"testing"
+
+	"github.com/iawia002/annie/config"
+	"github.com/iawia002/annie/test"
+)
+
+func TestVimeo(t *testing.T) {
+	config.InfoOnly = true
+	tests := []struct {
+		name string
+		args test.Args
+	}{
+		{
+			name: "normal test",
+			args: test.Args{
+				URL:     "https://player.vimeo.com/video/259325107",
+				Title:   "prfm 20180309",
+				Size:    131051118,
+				Quality: "1080p",
+			},
+		},
+		{
+			name: "normal test",
+			args: test.Args{
+				URL:     "https://vimeo.com/254865724",
+				Title:   "MAGIC DINER PT. II",
+				Size:    138966306,
+				Quality: "1080p",
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			data := Vimeo(tt.args.URL)
+			test.Check(t, tt.args, data)
+		})
+	}
+}

main.go

@@ -63,6 +63,8 @@ func main() {
 		extractors.Mgtv(videoURL)
 	case "tumblr":
 		extractors.Tumblr(videoURL)
+	case "vimeo":
+		extractors.Vimeo(videoURL)
 	default:
 		extractors.Universal(videoURL)
 	}