41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
|
package controller
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"github.com/PuerkitoBio/goquery"
|
||
|
|
"net/http"
|
||
|
|
"regexp"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
var downloadGameUrlPattern, _ = regexp.Compile("^https://game\\.eroge\\.xyz/hhh\\.php\\?id=\\d+$")
|
||
|
|
|
||
|
|
func downloadGame(url string) (string, error) {
|
||
|
|
if downloadGameUrlPattern.MatchString(url) {
|
||
|
|
return cacheable(fmt.Sprintf("GAME_PAGE$%s", url), time.Hour*1000, func() (interface{}, error) {
|
||
|
|
req, err := http.NewRequest("GET", url, nil)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
req.Header.Set("user-agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.131 Safari/537.36")
|
||
|
|
rsp, err := client.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer rsp.Body.Close()
|
||
|
|
doc, err := goquery.NewDocumentFromReader(rsp.Body)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
find := doc.Find("a.layui-btn")
|
||
|
|
list := make([]string, find.Size())
|
||
|
|
find.Each(func(i int, selection *goquery.Selection) {
|
||
|
|
list[i] = selection.AttrOr("href", "")
|
||
|
|
})
|
||
|
|
return list, nil
|
||
|
|
})
|
||
|
|
}
|
||
|
|
return "", errors.New("not support url")
|
||
|
|
}
|