120 lines
2.2 KiB
Go
120 lines
2.2 KiB
Go
|
package xvideos
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/iawia002/annie/downloader"
|
||
|
|
"github.com/iawia002/annie/request"
|
||
|
|
"github.com/iawia002/annie/utils"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
lowFlag = "html5player.setVideoUrlLow('"
|
||
|
|
lowFinalFlag = `');
|
||
|
|
html5player.setVideoUrlHigh(`
|
||
|
|
highFlag = "html5player.setVideoUrlHigh('"
|
||
|
|
highFinalFlag = `');
|
||
|
|
html5player.setVideoHLS(`
|
||
|
|
qualityLow = "low"
|
||
|
|
qualityHigh = "high"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
lowFlagLength = len(lowFlag)
|
||
|
|
highFlagLength = len(highFlag)
|
||
|
|
)
|
||
|
|
|
||
|
|
type src struct {
|
||
|
|
url string
|
||
|
|
quality string
|
||
|
|
}
|
||
|
|
|
||
|
|
func getSrc(html string) []*src {
|
||
|
|
var wg sync.WaitGroup
|
||
|
|
wg.Add(4)
|
||
|
|
|
||
|
|
startIndexLow := 0
|
||
|
|
go func() {
|
||
|
|
startIndexLow = strings.Index(html, lowFlag)
|
||
|
|
startIndexLow += lowFlagLength
|
||
|
|
wg.Done()
|
||
|
|
}()
|
||
|
|
endIndexLow := 0
|
||
|
|
go func() {
|
||
|
|
endIndexLow = strings.Index(html, lowFinalFlag)
|
||
|
|
wg.Done()
|
||
|
|
}()
|
||
|
|
|
||
|
|
startIndexHigh := 0
|
||
|
|
go func() {
|
||
|
|
startIndexHigh = strings.Index(html, highFlag)
|
||
|
|
startIndexHigh += highFlagLength
|
||
|
|
wg.Done()
|
||
|
|
}()
|
||
|
|
endIndexHigh := 0
|
||
|
|
go func() {
|
||
|
|
endIndexHigh = strings.Index(html, highFinalFlag)
|
||
|
|
wg.Done()
|
||
|
|
}()
|
||
|
|
wg.Wait()
|
||
|
|
|
||
|
|
var srcs []*src
|
||
|
|
if startIndexLow != -1 {
|
||
|
|
srcs = append(srcs, &src{
|
||
|
|
url: html[startIndexLow:endIndexLow],
|
||
|
|
quality: qualityLow,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
if startIndexHigh != -1 {
|
||
|
|
srcs = append(srcs, &src{
|
||
|
|
url: html[startIndexHigh:endIndexHigh],
|
||
|
|
quality: qualityHigh,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
return srcs
|
||
|
|
}
|
||
|
|
|
||
|
|
// Extract is the main function for extracting data
|
||
|
|
func Extract(url string) ([]downloader.Data, error) {
|
||
|
|
html, err := request.Get(url, url, nil)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var title string
|
||
|
|
desc := utils.MatchOneOf(html, `<title>(.+?)</title>`)
|
||
|
|
if desc != nil && len(desc) > 1 {
|
||
|
|
title = desc[1]
|
||
|
|
} else {
|
||
|
|
title = "xvideos"
|
||
|
|
}
|
||
|
|
|
||
|
|
streams := make(map[string]downloader.Stream)
|
||
|
|
for _, src := range getSrc(html) {
|
||
|
|
size, err := request.Size(src.url, url)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
urlData := downloader.URL{
|
||
|
|
URL: src.url,
|
||
|
|
Size: size,
|
||
|
|
Ext: "mp4",
|
||
|
|
}
|
||
|
|
streams[src.quality] = downloader.Stream{
|
||
|
|
URLs: []downloader.URL{urlData},
|
||
|
|
Size: size,
|
||
|
|
Quality: fmt.Sprintf("%s", src.quality),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return []downloader.Data{
|
||
|
|
{
|
||
|
|
Site: "XVIDEOS xvideos.com",
|
||
|
|
Title: title,
|
||
|
|
Type: "video",
|
||
|
|
Streams: streams,
|
||
|
|
URL: url,
|
||
|
|
},
|
||
|
|
}, nil
|
||
|
|
}
|