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

Added support for embedded videos

Browse Source
This commit is contained in:
ByteDream 2020-11-27 22:29:31 +01:00
parent 45335ba958
commit ca0d96bab0
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
vivo.go
View File

@ -10,10 +10,11 @@ import (
"strings" "strings"
) )
//Vivo is the base struct where all information about a vivo.x video are saved //Vivo is the base struct where all information about a vivo.sx video are saved
type Vivo struct { type Vivo struct {
VivoURL string VivoURL string
VideoURL string VideoURL string
ID string
Title string Title string
Mime string Mime string
Quality string Quality string
@ -22,10 +23,14 @@ 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/") {
URL = strings.ReplaceAll(URL, "/embed/", "/")
}
response, err := http.Get(URL) response, err := http.Get(URL)
if err != nil { if err != nil {
return Vivo{}, err return Vivo{}, err
@ -42,6 +47,7 @@ func GetVideo(URL string) (Vivo, error) {
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:],
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 ")}
for _, info := range strings.Split(parameter, ",") { for _, info := range strings.Split(parameter, ",") {