81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
|
package http
|
||
|
|
|
||
|
|
import (
|
||
|
|
"runtime"
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"apigo.cc/gojs"
|
||
|
|
"apigo.cc/gojs/goja"
|
||
|
|
"github.com/go-rod/rod"
|
||
|
|
"github.com/go-rod/rod/lib/launcher"
|
||
|
|
"github.com/ssgo/log"
|
||
|
|
"github.com/ssgo/u"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Chrome struct {
|
||
|
|
id string
|
||
|
|
launcher *launcher.Launcher
|
||
|
|
browser *rod.Browser
|
||
|
|
}
|
||
|
|
|
||
|
|
var chromes = map[string]*Chrome{}
|
||
|
|
var chromesLock sync.Mutex
|
||
|
|
|
||
|
|
func (ch *Chrome) Close() {
|
||
|
|
if ch.browser != nil {
|
||
|
|
ch.browser.Close()
|
||
|
|
ch.browser = nil
|
||
|
|
}
|
||
|
|
if ch.launcher != nil {
|
||
|
|
ch.launcher.Cleanup()
|
||
|
|
ch.launcher = nil
|
||
|
|
}
|
||
|
|
|
||
|
|
chromesLock.Lock()
|
||
|
|
delete(chromes, ch.id)
|
||
|
|
chromesLock.Unlock()
|
||
|
|
}
|
||
|
|
|
||
|
|
func CloseAllChrome() {
|
||
|
|
n := len(chromes)
|
||
|
|
if n > 0 {
|
||
|
|
for _, ch := range chromes {
|
||
|
|
ch.Close()
|
||
|
|
}
|
||
|
|
log.DefaultLogger.Info("关闭所有 Chrome 浏览器", "count", n)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func StartChrome(showWindow *bool, vm *goja.Runtime) (*Chrome, error) {
|
||
|
|
logger := gojs.GetLogger(vm)
|
||
|
|
ch := &Chrome{}
|
||
|
|
ch.browser = rod.New()
|
||
|
|
if localBrowserPath, hasLocalBrowser := launcher.LookPath(); hasLocalBrowser {
|
||
|
|
ch.launcher = launcher.New().Bin(localBrowserPath).Headless(!u.Bool(showWindow)).Set("no-sandbox").Set("disable-gpu").Set("disable-dev-shm-usage").Set("single-process")
|
||
|
|
switch runtime.GOOS {
|
||
|
|
case "linux":
|
||
|
|
ch.launcher.Set("disable-setuid-sandbox")
|
||
|
|
case "windows":
|
||
|
|
ch.launcher.Set("disable-features=RendererCodeIntegrity")
|
||
|
|
}
|
||
|
|
localChromeURL, err := ch.launcher.Launch()
|
||
|
|
if err != nil {
|
||
|
|
ch.Close()
|
||
|
|
return nil, gojs.Err(err)
|
||
|
|
}
|
||
|
|
logger.Info("启动本地 Chrome 浏览器", "url", localChromeURL, "path", localBrowserPath)
|
||
|
|
ch.browser.ControlURL(localChromeURL)
|
||
|
|
}
|
||
|
|
if err := ch.browser.Connect(); err != nil {
|
||
|
|
ch.Close()
|
||
|
|
return nil, gojs.Err(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
ch.id = u.UniqueId()
|
||
|
|
chromesLock.Lock()
|
||
|
|
chromes[ch.id] = ch
|
||
|
|
chromesLock.Unlock()
|
||
|
|
return ch, nil
|
||
|
|
|
||
|
|
}
|