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) GetNewBrowser(params *Fingerprint) *BrowserInfo {
	// 创建新实例（补充CancelFunc）
	var ChromeExample Chrome
	browser, url, page := ChromeExample.TemporaryCreate(params)
	browserInfo := &BrowserInfo{
		Browser: browser,
		Url:     url,
		Page:    page,
	}
	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)
}