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

Changed vivo.GetVideo return type to pointer

Browse Source
This commit is contained in:
ByteDream 2020-11-27 22:42:02 +01:00
parent ca0d96bab0
commit da5f7ef4a2
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
vivo.go
View File

@ -22,9 +22,9 @@ 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) {
if !regexp.MustCompile("(vivo\\.sx/)(.*)(.{10}$)").MatchString(URL) { if !regexp.MustCompile("(vivo\\.sx/)(.*)(.{10}$)").MatchString(URL) {
return Vivo{}, errors.New("Not a valid vivo.sx url") return &Vivo{}, errors.New("Not a valid vivo.sx url")
} }
if strings.Contains(URL, "/embed/") { if strings.Contains(URL, "/embed/") {
@ -33,20 +33,20 @@ func GetVideo(URL string) (Vivo, error) {
response, err := http.Get(URL) response, err := http.Get(URL)
if err != nil { if err != nil {
return Vivo{}, err return &Vivo{}, err
} }
defer response.Body.Close() defer response.Body.Close()
bodyAsBytes, err := ioutil.ReadAll(response.Body) bodyAsBytes, err := ioutil.ReadAll(response.Body)
if err != nil { if err != nil {
return Vivo{}, err return &Vivo{}, err
} }
bodyAsString := string(bodyAsBytes) bodyAsString := string(bodyAsBytes)
parameter := regexp.MustCompile("(?s)InitializeStream\\s*\\(\\s*({.+?})\\s*\\)\\s*;").FindString(bodyAsString) parameter := regexp.MustCompile("(?s)InitializeStream\\s*\\(\\s*({.+?})\\s*\\)\\s*;").FindString(bodyAsString)
parameter = strings.NewReplacer("\n", "", "\t", "", "InitializeStream ({", "", "});", "", "'", "\"").Replace(strings.TrimSpace(parameter)) parameter = strings.NewReplacer("\n", "", "\t", "", "InitializeStream ({", "", "});", "", "'", "\"").Replace(strings.TrimSpace(parameter))
vivo := Vivo{VivoURL: URL, vivo := &Vivo{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 ")} Title: strings.TrimPrefix(strings.TrimSuffix(regexp.MustCompile(`<h1>(.*?)<strong>`).FindString(bodyAsString), "&nbsp;<strong>"), "<h1>Watch ")}
@ -64,7 +64,7 @@ func GetVideo(URL string) (Vivo, error) {
case "source": case "source":
decodedURL, err := url.QueryUnescape(value) decodedURL, err := url.QueryUnescape(value)
if err != nil { if err != nil {
return Vivo{}, err return &Vivo{}, err
} }
videoURL := rot47(decodedURL) videoURL := rot47(decodedURL)
vivo.VideoURL = videoURL vivo.VideoURL = videoURL