init
This commit is contained in:
zyj
83
libs/browser.go
Normal file
83
libs/browser.go
Normal file
@@ -0,0 +1,83 @@
|
||||
package libs
|
||||
|
||||
import (
|
||||
"sync"
|
||||
|
||||
"github.com/go-rod/rod"
|
||||
)
|
||||
|
||||
type Browser struct{}
|
||||
|
||||
type BrowserInfo struct {
|
||||
Id int
|
||||
Browser *rod.Browser
|
||||
Url string
|
||||
Page *rod.Page
|
||||
}
|
||||
|
||||
// 浏览器指纹数据
|
||||
type Fingerprint struct {
|
||||
Viewport []struct {
|
||||
Height int
|
||||
Width int
|
||||
}
|
||||
Proxy string
|
||||
Lang string
|
||||
}
|
||||
|
||||
var (
|
||||
BrowserPool = make(map[int]*BrowserInfo) // 初始化map
|
||||
poolMutex sync.RWMutex // 添加读写锁保证并发安全
|
||||
)
|
||||
|
||||
var ChromeExample Chrome
|
||||
|
||||
func (*Browser) GetBrowser(id int, params *Fingerprint) *BrowserInfo {
|
||||
poolMutex.RLock()
|
||||
value, ok := BrowserPool[id]
|
||||
poolMutex.RUnlock()
|
||||
|
||||
if ok {
|
||||
return value
|
||||
}
|
||||
|
||||
// 未找到时获取写锁（互斥）
|
||||
poolMutex.Lock()
|
||||
defer poolMutex.Unlock()
|
||||
|
||||
// 双检查：避免在等待锁期间其他goroutine已创建
|
||||
if value, ok := BrowserPool[id]; ok {
|
||||
return value
|
||||
}
|
||||
|
||||
// 创建新实例（补充CancelFunc）
|
||||
var ChromeExample Chrome
|
||||
browser, url, page := ChromeExample.Create(id, params)
|
||||
browserInfo := &BrowserInfo{
|
||||
Id: id,
|
||||
Browser: browser,
|
||||
Url: url,
|
||||
Page: page,
|
||||
}
|
||||
BrowserPool[id] = browserInfo
|
||||
return browserInfo
|
||||
}
|
||||
|
||||
func (*Browser) GetAllBrowser() map[int]*BrowserInfo {
|
||||
return BrowserPool
|
||||
}
|
||||
|
||||
func (*Browser) CancelAllBrowser() {
|
||||
for _, v := range BrowserPool {
|
||||
v.Browser.MustClose()
|
||||
}
|
||||
}
|
||||
|
||||
func (*Browser) CancelBrowser(id int) {
|
||||
browserInfo := BrowserPool[id]
|
||||
if browserInfo == nil {
|
||||
return
|
||||
}
|
||||
browserInfo.Browser.MustClose()
|
||||
delete(BrowserPool, id)
|
||||
}
|
||||
89
libs/chrome.go
Normal file
89
libs/chrome.go
Normal file
@@ -0,0 +1,89 @@
|
||||
package libs
|
||||
|
||||
import (
|
||||
"log"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
|
||||
"github.com/go-rod/rod"
|
||||
"github.com/go-rod/rod/lib/launcher"
|
||||
"github.com/go-rod/rod/lib/proto"
|
||||
"github.com/go-rod/stealth"
|
||||
)
|
||||
|
||||
type Chrome struct{}
|
||||
|
||||
func (c *Chrome) Create(accountId int, params *Fingerprint) (*rod.Browser, string, *rod.Page) {
|
||||
var accountIdStr = strconv.Itoa(accountId)
|
||||
userDataDir := "user-data/" + accountIdStr
|
||||
absPath, _ := filepath.Abs(userDataDir)
|
||||
log.Printf("用户数据目录: %s", absPath)
|
||||
path, _ := launcher.LookPath() // 获取当前系统的浏览器目录
|
||||
// 判断是否传递代理
|
||||
// var proxyHandle utils.ProxyHandle
|
||||
// protocol, username, password, ip, port, errProxy := proxyHandle.ParseProxy(params.Proxy)
|
||||
|
||||
l := launcher.New().
|
||||
Bin(path).
|
||||
// Proxy("210.51.27.121:50003"). // 直接传入完整认证URL
|
||||
Delete("use-mock-keychain"). // delete flag "--use-mock-keychain"
|
||||
Set("disable-blink-features", "AutomationControlled"). // 绕过自动化检测
|
||||
// Set("incognito"). // 无痕模式
|
||||
Set("user-data-dir", absPath). // 数据持久化目录
|
||||
Set("window-size", "1920,1480"). // 窗口尺寸
|
||||
Set("disable-infobars", "true"). // 隐藏自动化提示栏
|
||||
Set("no-sandbox", "true").
|
||||
Set("excludeSwitches", "enable-automation").
|
||||
Set("enable-gpu"). // 启用 GPU 加速
|
||||
Set("ignore-certificate-errors"). // 忽略证书错误
|
||||
Set("use-fake-ui-for-media-stream"). // 允许媒体流
|
||||
Set("autoplay-policy", "no-user-gesture-required"). // 自动播放
|
||||
Set("ignore-certificate-errors").
|
||||
Set("disable-application-cache").
|
||||
Set("disable-dev-shm-usage").
|
||||
// 禁用webrtc
|
||||
Set("disable-webrtc", "true"). // 核心禁用参数
|
||||
Set("disable-features", "WebRtcHideLocalIpsWithMdns"). // 隐藏本地IP
|
||||
Set("webrtc-ip-handling-policy", "disable_non_proxied_udp"). // 禁用非代理UDP
|
||||
Set("force-webrtc-ip-handling-policy").
|
||||
Headless(false) // 关闭无头模式
|
||||
|
||||
// 修改时区
|
||||
l.Set("disable-geolocation", "true") // 启用地理定位API
|
||||
l.Set("disable-web-security", "true") // 允许跨域（某些网站需要） 可以跨域访问iframe标签
|
||||
l.Set("disable-notifications", "true")
|
||||
l.Set("disable-site-isolation-trials", "true")
|
||||
// if errProxy == nil {
|
||||
// if strings.Contains(protocol, "http") {
|
||||
// log.Println(protocol + "://" + ip + ":" + port)
|
||||
// l = l.Proxy(ip + ":" + port)
|
||||
// }
|
||||
// }
|
||||
|
||||
uri := l.MustLaunch()
|
||||
|
||||
browser := rod.New().
|
||||
ControlURL(uri).
|
||||
MustConnect().NoDefaultDevice()
|
||||
browser.MustIgnoreCertErrors(true)
|
||||
// if errProxy == nil {
|
||||
// if strings.Contains(protocol, "http") {
|
||||
// log.Println(username, password)
|
||||
// go browser.MustHandleAuth(username, password)()
|
||||
// }
|
||||
// }
|
||||
page := stealth.MustPage(browser)
|
||||
// page := browser.MustPage()
|
||||
|
||||
go browser.EachEvent(func(e *proto.TargetTargetCreated) {
|
||||
if e.TargetInfo.Type != proto.TargetTargetInfoTypePage {
|
||||
return
|
||||
}
|
||||
browser.MustPageFromTargetID(e.TargetInfo.TargetID).MustEvalOnNewDocument(stealth.JS)
|
||||
})()
|
||||
|
||||
// 在页面加载前注入时区覆盖代码
|
||||
page.MustSetViewport(1920, 1480, 1.0, false)
|
||||
|
||||
return browser, uri, page
|
||||
}
|
||||
Reference in New Issue
Block a user