mirrors/vivo
1
0
Fork 0
mirror of https://github.com/bytedream/vivo.git synced 2025-05-09 20:25:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

General regex improvements; renamed GetVideoWithProxy to GetVideoWithClient

Browse Source
This commit is contained in:
ByteDream 2021-07-19 01:16:38 +02:00
parent 7213c78bf5
commit b334506528
1 changed files with 50 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

91
vivo.go
View File

@ -25,23 +25,24 @@ type Vivo struct {
// GetVideo extracts the video url and some other nice information from a vivo.sx page // GetVideo extracts the video url and some other nice information from a vivo.sx page
func GetVideo(URL string) (*Vivo, error) { func GetVideo(URL string) (*Vivo, error) {
return GetVideoWithProxy(URL, http.DefaultClient) return GetVideoWithClient(URL, http.DefaultClient)
} }
// GetVideoWithProxy extracts the video url and some other nice information from a vivo.sx page with a pre defined proxy // GetVideoWithClient extracts the video url and some other nice information from a vivo.sx page with a pre defined proxy
func GetVideoWithProxy(URL string, proxy *http.Client) (*Vivo, error) { func GetVideoWithClient(URL string, client *http.Client) (*Vivo, error) {
var scheme string var scheme string
re := regexp.MustCompile("^(?P<scheme>http(s?)://)?vivo\\.(sx|st)/(|embed/).{10}(/|$)") URL = strings.TrimSuffix(URL, "/")
groupNames := re.SubexpNames()
reMatch := re.FindAllStringSubmatch(URL, 1)
if len(reMatch) == 0 { urlPattern := regexp.MustCompile(`(?m)^(?P<scheme>https?://)?vivo\.(sx|st)/(embed/)?.{10}`)
return &Vivo{}, errors.New("Not a valid vivo.sx url") urlGroups := urlPattern.SubexpNames()
if urlMatch := urlPattern.FindAllStringSubmatch(URL, -1); len(urlMatch) == 0 {
return &Vivo{}, errors.New("Not a valid url")
} else { } else {
for _, match := range reMatch { for _, match := range urlMatch {
for i, content := range match { for i, content := range match {
if groupNames[i] == "scheme" { if urlGroups[i] == "scheme" {
scheme = content scheme = content
break break
} }
@ -52,13 +53,17 @@ func GetVideoWithProxy(URL string, proxy *http.Client) (*Vivo, error) {
URL = "https://" + URL URL = "https://" + URL
} }
//return &Vivo{}, errors.New("Not a valid vivo.sx url")
if strings.Contains(URL, "/embed/") { if strings.Contains(URL, "/embed/") {
URL = strings.ReplaceAll(URL, "/embed/", "/") URL = strings.ReplaceAll(URL, "/embed/", "/")
} }
response, err := proxy.Get(URL) req, err := http.NewRequest(http.MethodGet, URL, nil)
if err != nil {
return &Vivo{}, err
}
req.Header.Add("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:83.0) Gecko/20100101 Firefox/83.0")
response, err := client.Do(req)
if err != nil { if err != nil {
return &Vivo{}, err return &Vivo{}, err
} }
@ -68,44 +73,48 @@ func GetVideoWithProxy(URL string, proxy *http.Client) (*Vivo, error) {
if err != nil { if err != nil {
return &Vivo{}, err return &Vivo{}, err
} }
bodyAsString := string(bodyAsBytes)
parameter := regexp.MustCompile("(?s)InitializeStream\\s*\\(\\s*({.+?})\\s*\\)\\s*;").FindString(bodyAsString) streamPattern := regexp.MustCompile(`(?m)<h1>Watch (?P<title>[\S\s]+)&nbsp|quality:\s*(?P<quality>\S+),|source:\s*'(?P<source>\S+)'`)
parameter = strings.NewReplacer("\n", "", "\t", "", "InitializeStream ({", "", "});", "", "'", "\"").Replace(strings.TrimSpace(parameter)) streamGroups := streamPattern.SubexpNames()
vivo := &Vivo{client: proxy, vivo := &Vivo{
client: client,
VivoURL: URL, VivoURL: URL,
ID: URL[strings.LastIndex(URL, "/")+1:], ID: URL[strings.LastIndex(URL, "/")+1:],
Title: strings.TrimPrefix(strings.TrimSuffix(regexp.MustCompile(`<h1>(.*?)<strong>`).FindString(bodyAsString), "&nbsp;<strong>"), "<h1>Watch ")} }
for _, info := range strings.Split(parameter, ",") { for _, match := range streamPattern.FindAllSubmatch(bodyAsBytes, -1) {
keyValue := strings.Split(info, ": ") for i, content := range match {
if len(keyValue) <= 1 { contentAsString := string(content)
continue if contentAsString != "" {
} switch streamGroups[i] {
key := keyValue[0] case "title":
value := strings.ReplaceAll(keyValue[1], "\"", "") vivo.Title = contentAsString
case "quality":
vivo.Quality = contentAsString + "p"
case "source":
decodedURL, err := url.QueryUnescape(contentAsString)
if err != nil {
return &Vivo{}, err
}
videoURL := rot47(decodedURL)
vivo.VideoURL = videoURL
switch key { video, err := client.Get(videoURL)
case "quality": if err != nil {
vivo.Quality = value + "p" return &Vivo{}, err
case "source": }
decodedURL, err := url.QueryUnescape(value) vivo.Mime = video.Header.Get("content-type")
if err != nil { vivo.Length = video.ContentLength
return &Vivo{}, err }
} }
videoURL := rot47(decodedURL)
vivo.VideoURL = videoURL
video, err := proxy.Get(videoURL)
if err != nil {
return vivo, nil
}
vivo.Mime = video.Header.Get("content-type")
vivo.Length = video.ContentLength
} }
} }
if vivo.VideoURL == "" {
return &Vivo{}, errors.New("Could not find video")
}
return vivo, nil return vivo, nil
} }