82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package hunter
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/474420502/requests"
|
|
"github.com/tebeka/selenium"
|
|
)
|
|
|
|
// https://github.com/tebeka/selenium
|
|
|
|
// PreBaseDriverUrl Task的 curl bash 预处理组件
|
|
type PreBaseDriverUrl struct {
|
|
PreUrl string
|
|
Port int
|
|
service *selenium.Service
|
|
driver selenium.WebDriver
|
|
}
|
|
|
|
// Close 如果需要在最后执行销毁操作, 继承覆盖该方法
|
|
func (u *PreBaseDriverUrl) Close() error {
|
|
|
|
if u.service != nil {
|
|
// 直接退出, 所有销毁 直接忽略webdriver.Quit(). // Delete Session
|
|
if err := u.service.Stop(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// IResponse interface {
|
|
// Content() []byte
|
|
// GetStatus() string
|
|
// GetStatusCode() int
|
|
// GetHeader() http.Header
|
|
// GetCookie() []*http.Cookie
|
|
|
|
// // 返回不同的自定义的Response, 也可以是其他定义的结构体如WebDriver
|
|
// GetResponse() interface{}
|
|
// }
|
|
|
|
// Content 内容
|
|
func (u *PreBaseDriverUrl) Content() []byte {
|
|
content, err := u.driver.PageSource()
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
return []byte(content)
|
|
}
|
|
|
|
// GetStatusCode 暂时为空
|
|
func (u *PreBaseDriverUrl) GetStatusCode() int {
|
|
return 0
|
|
}
|
|
|
|
// GetStatus 内容 暂时为空
|
|
func (u *PreBaseDriverUrl) GetStatus() string {
|
|
return ""
|
|
}
|
|
|
|
// GetHeader 暂时为空
|
|
func (u *PreBaseDriverUrl) GetHeader() http.Header {
|
|
return nil
|
|
}
|
|
|
|
// GetCookie 暂时为空
|
|
func (u *PreBaseDriverUrl) GetCookie() []*http.Cookie {
|
|
return nil
|
|
}
|
|
|
|
// GetResponse 返回 webdriver
|
|
func (u *PreBaseDriverUrl) GetResponse() interface{} {
|
|
return u.driver
|
|
}
|
|
|
|
func (u *PreBaseDriverUrl) Hunt() (requests.IResponse, error) {
|
|
err := u.driver.Get(string(u.PreUrl))
|
|
return u, err
|
|
}
|