package core

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"net/http"
	"net/url"
	"os"
	"regexp"
	"runtime/debug"
	"strings"
	"sync"
	"time"

	"github.com/go-rod/rod"
	"github.com/go-rod/rod/lib/launcher"
	"github.com/go-rod/rod/lib/proto"
	browserprofile "github.com/karust/openserp/core/browser"
	"github.com/sirupsen/logrus"
	"github.com/ysmood/gson"
)

// BrowserOpts configures Chromium launch and navigation behavior.
type BrowserOpts struct {
	// IsHeadless runs Chromium without visible UI.
	IsHeadless bool
	// IsLeakless forces child browser process cleanup when the parent exits.
	IsLeakless bool
	// Timeout is applied to browser connect and page navigation operations.
	Timeout time.Duration
	// LanguageCode sets Accept-Language for emulated requests.
	LanguageCode string
	// WaitRequests waits for request-idle state after navigation.
	WaitRequests bool
	// LeavePageOpen keeps pages open after search operations.
	LeavePageOpen bool
	// WaitLoadTime caps the document load-event wait after navigation. Selector
	// parsing still decides whether the page is usable after this wait expires.
	WaitLoadTime time.Duration
	// CaptchaSolverApiKey enables 2Captcha integration for supported engines.
	CaptchaSolverApiKey string
	// CaptchaSolverEnabled gates solver invocation regardless of engine flags.
	CaptchaSolverEnabled bool
	// BrowserPath optionally points to a specific browser executable.
	BrowserPath string
	// ProxyURL defines the upstream proxy for browser traffic.
	ProxyURL string
	// ProxyLaneStore keeps sticky proxy lane profiles and cookies.
	ProxyLaneStore *LaneStore
	// Insecure allows invalid TLS certificates for browser requests.
	Insecure bool
	// UserAgent optionally overrides browser-reported user agent during emulation.
	UserAgent string
	// BlockResourceTypes are blocked during page navigation when non-empty.
	// Typical tokens map to these types: image, font, css(stylesheet), js(script), media.
	BlockResourceTypes []proto.NetworkResourceType
	// BlockTrackers toggles static tracker-domain blocking.
	BlockTrackers bool
}

// Check applies default option values when optional fields are unset.
func (o *BrowserOpts) Check() {
	if o.Timeout == 0 {
		o.Timeout = time.Second * 30
	}

	if o.WaitLoadTime == 0 {
		o.WaitLoadTime = time.Second * 2
	}
}

func minPositiveDuration(a, b time.Duration) time.Duration {
	if a <= 0 {
		return b
	}
	if b <= 0 {
		return a
	}
	if a < b {
		return a
	}
	return b
}

var alwaysBlockedTrackingDomains = []string{
	"google-analytics.com",
	"googletagmanager.com",
	"doubleclick.net",
	"connect.facebook.net",
}

var alwaysBlockedTrackingURLPatterns = buildTrackingDomainURLPatterns(alwaysBlockedTrackingDomains)

var blockedResourceTypeTokenMap = map[string]proto.NetworkResourceType{
	"image": proto.NetworkResourceTypeImage,
	"font":  proto.NetworkResourceTypeFont,
	"media": proto.NetworkResourceTypeMedia,
	"css":   proto.NetworkResourceTypeStylesheet,
	"js":    proto.NetworkResourceTypeScript,
}

// ParseBlockedResourceTypes parses a comma-separated config value into
// NetworkResourceType values accepted by the request blocker.
func ParseBlockedResourceTypes(raw string) ([]proto.NetworkResourceType, error) {
	raw = strings.TrimSpace(raw)
	if raw == "" {
		return nil, nil
	}

	parts := strings.Split(raw, ",")
	seen := make(map[proto.NetworkResourceType]struct{}, len(parts))
	out := make([]proto.NetworkResourceType, 0, len(parts))

	for _, part := range parts {
		token := strings.TrimSpace(strings.ToLower(part))
		if token == "" {
			continue
		}

		resourceType, ok := blockedResourceTypeTokenMap[token]
		if !ok {
			return nil, fmt.Errorf("unsupported resource type %q", token)
		}

		if _, exists := seen[resourceType]; exists {
			continue
		}
		seen[resourceType] = struct{}{}
		out = append(out, resourceType)
	}

	return out, nil
}

// MustParseBlockedResourceTypes is like ParseBlockedResourceTypes but panics on error.
// Only call this after the value has already been validated by ParseBlockedResourceTypes.
func MustParseBlockedResourceTypes(raw string) []proto.NetworkResourceType {
	types, err := ParseBlockedResourceTypes(raw)
	if err != nil {
		panic(fmt.Sprintf("MustParseBlockedResourceTypes: %v", err))
	}
	return types
}

func buildTrackingDomainURLPatterns(domains []string) []string {
	patterns := make([]string, 0, len(domains)*2)
	for _, domain := range domains {
		domain = strings.TrimSpace(strings.ToLower(domain))
		if domain == "" {
			continue
		}
		patterns = append(patterns, "*://"+domain+"/*", "*://*."+domain+"/*")
	}
	return patterns
}

func blockedResourceTypeSet(types []proto.NetworkResourceType) map[proto.NetworkResourceType]struct{} {
	out := make(map[proto.NetworkResourceType]struct{}, len(types))
	for _, t := range types {
		if t != "" {
			out[t] = struct{}{}
		}
	}
	return out
}

func proxyAuthFetchPatterns() []*proto.FetchRequestPattern {
	return []*proto.FetchRequestPattern{
		{
			URLPattern:   "http://*/*",
			ResourceType: proto.NetworkResourceTypeDocument,
			RequestStage: proto.FetchRequestStageRequest,
		},
		{
			URLPattern:   "https://*/*",
			ResourceType: proto.NetworkResourceTypeDocument,
			RequestStage: proto.FetchRequestStageRequest,
		},
	}
}

func (b *Browser) configureRequestBlocking(ctx context.Context, page *rod.Page) error {
	if !b.BlockTrackers && len(b.BlockResourceTypes) == 0 {
		return nil
	}

	if b.BlockTrackers && len(alwaysBlockedTrackingURLPatterns) > 0 {
		if err := (proto.NetworkEnable{}).Call(page); err != nil {
			return fmt.Errorf("enable network domain for tracker blocking: %w", err)
		}
		if err := (proto.NetworkSetBlockedURLs{Urls: alwaysBlockedTrackingURLPatterns}).Call(page); err != nil {
			return fmt.Errorf("set blocked tracking URLs: %w", err)
		}
	}

	if len(b.BlockResourceTypes) == 0 {
		return nil
	}

	// HijackRequests calls Fetch.enable on the page, which collides with the
	// browser-level Fetch.enable installed by the proxy-auth listener. Two
	// consumers competing for the same RequestID produce "Invalid
	// InterceptionId" errors and stall parallel navigations (e.g. mega search
	// behind an authenticated proxy). Resource-type blocking is dropped on the
	// auth-proxy path; tracker URL blocking via NetworkSetBlockedURLs above
	// still works because it is not Fetch-based.
	if b.proxyUser != "" {
		WithRequest(ctx).Debug("Skipping HijackRequests resource blocking under proxy auth listener")
		return nil
	}

	blocked := blockedResourceTypeSet(b.BlockResourceTypes)
	router := page.HijackRequests()
	router.MustAdd("*", func(h *rod.Hijack) {
		if _, ok := blocked[h.Request.Type()]; ok {
			h.Response.Fail(proto.NetworkErrorReasonBlockedByClient)
			return
		}
		h.ContinueRequest(&proto.FetchContinueRequest{})
	})

	go router.Run()

	// Stop the router when the page context is done to avoid goroutine leak.
	go func() {
		<-ctx.Done()
		if err := router.Stop(); err != nil && !errors.Is(err, context.Canceled) && !isBrowserClosedError(err) {
			WithRequest(ctx).WithError(err).Debug("Stop request-blocking router failed")
		}
	}()

	return nil
}

// Browser wraps a launched Chromium instance used by engine implementations.
type Browser struct {
	BrowserOpts
	browserAddr   string
	proxyUser     string
	proxyPass     string
	conn          *browserConnection
	CaptchaSolver *CaptchaSolver
}

type browserConnection struct {
	mu           sync.Mutex
	browser      *rod.Browser
	laneProfiles map[string]browserprofile.Profile
	authCancel   context.CancelFunc
	authStopped  chan struct{}
}

// NewBrowser launches a new Chromium process via Rod launcher and returns a
// Browser wrapper configured with proxy and captcha solver settings.
func NewBrowser(opts BrowserOpts) (*Browser, error) {
	opts.Check()
	if strings.TrimSpace(opts.UserAgent) != "" {
		logrus.Warn("custom user_agent override can reduce profile coherence; use only for diagnostics")
	}
	logrus.WithFields(browserOptsLogFields(opts)).Debug("Browser options")

	path, err := resolveBrowserBinaryPath(opts.BrowserPath, launcher.LookPath)
	if err != nil {
		return nil, err
	}

	// Create launcher.
	// headless=new uses the full Chrome renderer; legacy --headless disables the
	// GPU process entirely, making WebGL context creation fail even with SwiftShader.
	// use-angle=swiftshader-webgl (Chrome ≥112) enables a software WebGL renderer.
	l := launcher.New().Leakless(opts.IsLeakless).
		Set("disable-blink-features", "AutomationControlled").
		Delete("enable-automation").
		Set("use-angle", "swiftshader-webgl").
		Set("ignore-gpu-blocklist")
	if opts.IsHeadless {
		l = l.HeadlessNew(true)
	} else {
		l = l.Headless(false)
	}
	if path != "" {
		logrus.WithField("browser_path", path).Debug("Using browser binary")
		l = l.Bin(path)
	}

	b := Browser{
		conn: &browserConnection{},
	}

	// Configure proxy if specified. Chrome's --proxy-server flag must NOT
	// include credentials; we strip them here and reinject via a persistent
	// CDP Fetch.handleAuthRequired listener installed after each connect.
	if opts.ProxyURL != "" {
		normalizedProxyURL, err := NormalizeProxyURL(opts.ProxyURL)
		if err != nil {
			return nil, fmt.Errorf("invalid proxy URL: %v", err)
		}
		opts.ProxyURL = normalizedProxyURL

		proxyUrl, err := url.Parse(opts.ProxyURL)
		if err != nil {
			return nil, fmt.Errorf("invalid proxy URL: %v", err)
		}

		proxyStr := proxyURLForBrowserLaunch(proxyUrl)
		logrus.WithField("proxy", MaskProxyURL(proxyStr)).Debug("Setting up proxy")
		l = l.Proxy(proxyStr)

		if proxyUrl.User != nil && (proxyUrl.Scheme == "http" || proxyUrl.Scheme == "https") {
			b.proxyUser = proxyUrl.User.Username()
			b.proxyPass, _ = proxyUrl.User.Password()
			logrus.WithFields(logrus.Fields{
				"proxy_scheme":   proxyUrl.Scheme,
				"proxy_username": b.proxyUser,
			}).Debugf("Proxy credentials configured for %s proxy: %s:****", proxyUrl.Scheme, b.proxyUser)
		}
	}

	b.BrowserOpts = opts
	b.browserAddr, err = l.Launch()

	if opts.CaptchaSolverEnabled && opts.CaptchaSolverApiKey != "" {
		b.CaptchaSolver = NewSolver(opts.CaptchaSolverApiKey)
		logrus.Debug("Captcha solver initialized")
	}

	return &b, err
}

func browserOptsLogFields(opts BrowserOpts) logrus.Fields {
	return logrus.Fields{
		"headless":                opts.IsHeadless,
		"leakless":                opts.IsLeakless,
		"timeout":                 opts.Timeout.String(),
		"language_code":           opts.LanguageCode,
		"wait_requests":           opts.WaitRequests,
		"leave_page_open":         opts.LeavePageOpen,
		"captcha_solver_enabled":  opts.CaptchaSolverEnabled,
		"captcha_solver_has_key":  strings.TrimSpace(opts.CaptchaSolverApiKey) != "",
		"browser_path_configured": strings.TrimSpace(opts.BrowserPath) != "",
		"proxy":                   maskedProxyLogValue(opts.ProxyURL),
		"insecure":                opts.Insecure,
		"user_agent_override":     strings.TrimSpace(opts.UserAgent) != "",
		"block_resource_types":    len(opts.BlockResourceTypes),
		"block_trackers":          opts.BlockTrackers,
		"proxy_lanes_enabled":     opts.ProxyLaneStore != nil,
	}
}

func maskedProxyLogValue(proxyURL string) string {
	if strings.TrimSpace(proxyURL) == "" {
		return ""
	}
	return MaskProxyURL(proxyURL)
}

func proxyURLForBrowserLaunch(u *url.URL) string {
	if u == nil {
		return ""
	}
	clone := *u
	// Chrome expects socks5 scheme in --proxy-server; socks5h is not accepted.
	if clone.Scheme == "socks5h" {
		clone.Scheme = "socks5"
	}
	clone.User = nil
	clone.Path = ""
	clone.RawPath = ""
	clone.RawQuery = ""
	clone.Fragment = ""
	return clone.String()
}

func validateBrowserBinaryPath(path string) error {
	info, err := os.Stat(path)
	if err != nil {
		return err
	}
	if info.IsDir() {
		return fmt.Errorf("path points to a directory")
	}
	return nil
}

// resolveBrowserBinaryPath prefers an explicit browser path. If no explicit path is provided,
// it falls back to launcher autodiscovery and lets Rod handle auto-download when no binary is found.
func resolveBrowserBinaryPath(browserPath string, lookPath func() (string, bool)) (string, error) {
	if browserPath != "" {
		if err := validateBrowserBinaryPath(browserPath); err != nil {
			return "", fmt.Errorf("invalid browser_path %q: %w", browserPath, err)
		}
		return browserPath, nil
	}

	path, has := lookPath()
	if has {
		return path, nil
	}

	return "", nil
}

// IsInitialized reports whether the browser launcher has been created.
func (b *Browser) IsInitialized() bool {
	return b.browserAddr != ""
}

func (b *Browser) connectionState() *browserConnection {
	if b.conn == nil {
		b.conn = &browserConnection{}
	}
	return b.conn
}

func (b *Browser) newRodBrowser() *rod.Browser {
	browser := rod.New().NoDefaultDevice().ControlURL(b.browserAddr)
	if b.Timeout > 0 {
		browser = browser.Timeout(b.Timeout)
	}
	return browser
}

func (b *Browser) connectBrowser() (*rod.Browser, error) {
	browser := b.newRodBrowser()
	if err := browser.Connect(); err != nil {
		return nil, err
	}

	// Keep cert handling on the persistent browser session.
	// Proxy runtime can surface MITM certs, and insecure mode is explicit opt-in.
	if b.ProxyURL != "" || b.Insecure {
		if err := browser.IgnoreCertErrors(true); err != nil {
			return nil, err
		}
	}

	if b.proxyUser != "" {
		if err := b.startProxyAuthListener(browser); err != nil {
			return nil, fmt.Errorf("install proxy auth listener: %w", err)
		}
	}

	return browser, nil
}

// startProxyAuthListener enables the Fetch domain with HandleAuthRequests=true
// and runs a goroutine that responds to every Fetch.requestPaused (continue
// the request) and every Fetch.authRequired (provide credentials). The
// listener lives for the lifetime of the rod.Browser session and replaces
// rod's single-shot HandleAuth helper.
func (b *Browser) startProxyAuthListener(browser *rod.Browser) error {
	state := b.connectionState()

	// Stop a previous listener attached to a stale connection.
	if state.authCancel != nil {
		state.authCancel()
		if state.authStopped != nil {
			<-state.authStopped
		}
		state.authCancel = nil
		state.authStopped = nil
	}

	listenCtx, cancel := context.WithCancel(context.Background())
	stopped := make(chan struct{})
	state.authCancel = cancel
	state.authStopped = stopped

	username := b.proxyUser
	password := b.proxyPass
	scoped := browser.Context(listenCtx)
	started := make(chan struct{})

	go func() {
		defer close(stopped)
		// Subscribe via EachEvent. The wait function it returns blocks until
		// listenCtx is cancelled. We close `started` after EachEvent has
		// installed its handlers but before we wait, so the caller can safely
		// enable the Fetch domain without racing the listener install.
		//
		// Each ack runs in its own goroutine. EachEvent invokes our callbacks
		// sequentially on a single dispatch goroutine, and each ack performs a
		// synchronous CDP roundtrip. Under concurrent load (e.g. mega search
		// fanning out 5 engines through one Chrome) sequential dispatch
		// becomes the bottleneck — Chrome times out paused requests faster
		// than we can ack them, producing "Invalid InterceptionId" errors and
		// stalled navigations. Acks for distinct RequestIDs are independent,
		// so dispatching them concurrently is safe.
		wait := scoped.EachEvent(
			func(e *proto.FetchAuthRequired) bool {
				go func(requestID proto.FetchRequestID) {
					resp := proto.FetchAuthChallengeResponseResponseProvideCredentials
					err := proto.FetchContinueWithAuth{
						RequestID: requestID,
						AuthChallengeResponse: &proto.FetchAuthChallengeResponse{
							Response: resp,
							Username: username,
							Password: password,
						},
					}.Call(scoped)
					if err != nil && !errors.Is(err, context.Canceled) {
						logrus.WithError(err).Debug("Proxy auth response failed")
					}
				}(e.RequestID)
				return false
			},
			func(e *proto.FetchRequestPaused) bool {
				go func(requestID proto.FetchRequestID) {
					err := proto.FetchContinueRequest{RequestID: requestID}.Call(scoped)
					if err != nil && !errors.Is(err, context.Canceled) {
						logrus.WithError(err).Debug("Continue paused request failed")
					}
				}(e.RequestID)
				return false
			},
		)
		close(started)
		wait()
	}()

	<-started
	if err := (proto.FetchEnable{
		Patterns:           proxyAuthFetchPatterns(),
		HandleAuthRequests: true,
	}).Call(browser); err != nil {
		cancel()
		<-stopped
		state.authCancel = nil
		state.authStopped = nil
		return fmt.Errorf("enable Fetch domain: %w", err)
	}
	return nil
}

func (b *Browser) ensureConnectedBrowser(ctx context.Context, forceReconnect bool) (*rod.Browser, error) {
	if b == nil || b.browserAddr == "" {
		return nil, fmt.Errorf("browser is not initialized")
	}

	state := b.connectionState()
	state.mu.Lock()
	defer state.mu.Unlock()

	if state.browser == nil || forceReconnect {
		connected, err := b.connectBrowser()
		if err != nil {
			return nil, err
		}
		state.browser = connected
		return state.browser, nil
	}

	if _, err := state.browser.Version(); err != nil {
		WithRequest(ctx).WithError(err).Debug("Browser ping failed, reconnecting")
		connected, reconnectErr := b.connectBrowser()
		if reconnectErr != nil {
			return nil, reconnectErr
		}
		state.browser = connected
	}

	return state.browser, nil
}

func createIsolatedPage(browser *rod.Browser, proxyURL string) (*rod.Page, proto.BrowserBrowserContextID, error) {
	create := proto.TargetCreateBrowserContext{}
	if strings.TrimSpace(proxyURL) != "" {
		parsed, err := url.Parse(proxyURL)
		if err != nil {
			return nil, "", err
		}
		create.ProxyServer = proxyURLForBrowserLaunch(parsed)
	}

	browserContext, err := create.Call(browser)
	if err != nil {
		return nil, "", err
	}

	target, err := (proto.TargetCreateTarget{
		URL:              "about:blank",
		BrowserContextID: browserContext.BrowserContextID,
	}).Call(browser)
	if err != nil {
		_ = disposeBrowserContext(browser, browserContext.BrowserContextID)
		return nil, "", err
	}

	page, err := browser.PageFromTarget(target.TargetID)
	if err != nil {
		_ = disposeBrowserContext(browser, browserContext.BrowserContextID)
		return nil, "", err
	}

	return page, browserContext.BrowserContextID, nil
}

func disposeBrowserContext(browser *rod.Browser, browserContextID proto.BrowserBrowserContextID) error {
	if browser == nil || browserContextID == "" {
		return nil
	}
	return (proto.TargetDisposeBrowserContext{
		BrowserContextID: browserContextID,
	}).Call(browser)
}

var chromeVersionPattern = regexp.MustCompile(`(?:HeadlessChrome|Chrome)/([0-9]+\.[0-9]+\.[0-9]+\.[0-9]+)`)

func (b *Browser) laneProfile(ctx context.Context, browser *rod.Browser) (browserprofile.Profile, string) {
	engine := engineFromContext(ctx)
	region := profileRegionFromContext(ctx)
	if region == "" {
		region = strings.TrimSpace(b.LanguageCode)
	}

	if forcedID := forcedProfileIDFromContext(ctx); forcedID != "" {
		profile, ok := browserprofile.ProfileByID(forcedID)
		if ok {
			profile = applyRuntimeBrowserVersion(profile, browser)
			profile = applyProfileLanguageHint(profile, region)
			if overrideUA := strings.TrimSpace(b.UserAgent); overrideUA != "" {
				profile.UserAgent = overrideUA
			}
			return profile, "forced:" + forcedID
		}
	}

	if laneKey := proxyLaneKeyFromContext(ctx); !laneKey.Empty() && b.ProxyLaneStore != nil {
		profile := b.ProxyLaneStore.Profile(laneKey, func() browserprofile.Profile {
			selected := browserprofile.SelectProfileForSession(engine, region, laneKey.SessionID)
			selected = applyRuntimeBrowserVersion(selected, browser)
			selected = applyProfileLanguageHint(selected, region)
			if overrideUA := strings.TrimSpace(b.UserAgent); overrideUA != "" {
				selected.UserAgent = overrideUA
			}
			return selected
		})
		return profile, laneKey.ID()
	}

	laneKey := browserprofile.LaneKey(engine, region)

	state := b.connectionState()
	state.mu.Lock()
	if state.laneProfiles == nil {
		state.laneProfiles = make(map[string]browserprofile.Profile)
	}
	if profile, ok := state.laneProfiles[laneKey]; ok {
		state.mu.Unlock()
		return profile, laneKey
	}
	state.mu.Unlock()

	// Resolve profile outside the lock: SelectProfileForSession reads from a separate
	// RWMutex-guarded catalog, and applyRuntimeBrowserVersion makes a CDP
	// round-trip (browser.Version). Holding state.mu over network I/O would
	// serialize all concurrent Navigate calls.
	profile := browserprofile.SelectProfileForSession(engine, region, laneKey)
	profile = applyRuntimeBrowserVersion(profile, browser)
	profile = applyProfileLanguageHint(profile, region)
	if overrideUA := strings.TrimSpace(b.UserAgent); overrideUA != "" {
		profile.UserAgent = overrideUA
	}

	state.mu.Lock()
	if state.laneProfiles == nil {
		state.laneProfiles = make(map[string]browserprofile.Profile)
	}
	if _, exists := state.laneProfiles[laneKey]; !exists {
		state.laneProfiles[laneKey] = profile
	} else {
		profile = state.laneProfiles[laneKey]
	}
	state.mu.Unlock()

	return profile, laneKey
}

func applyRuntimeBrowserVersion(profile browserprofile.Profile, browser *rod.Browser) browserprofile.Profile {
	fullVersion := ""
	if browser != nil {
		version, err := browser.Version()
		if err == nil && version != nil {
			fullVersion = extractChromeVersion(version.UserAgent)
			if fullVersion == "" {
				fullVersion = extractChromeVersion(version.Product)
			}
		}
	}
	if fullVersion == "" {
		fullVersion = extractChromeVersion(profile.UserAgent)
	}
	if fullVersion == "" {
		return profile
	}

	major := chromeMajorVersion(fullVersion)
	if major == "" {
		return profile
	}

	profile.UserAgent = replaceChromeUserAgentVersion(profile.UserAgent, major+".0.0.0")
	profile.UACHBrands = patchBrandVersions(profile.UACHBrands, major, false)
	profile.UACHFullVerList = patchBrandVersions(profile.UACHFullVerList, fullVersion, true)
	return profile
}

func extractChromeVersion(value string) string {
	matches := chromeVersionPattern.FindStringSubmatch(strings.TrimSpace(value))
	if len(matches) < 2 {
		return ""
	}
	return strings.TrimSpace(matches[1])
}

func chromeMajorVersion(fullVersion string) string {
	fullVersion = strings.TrimSpace(fullVersion)
	if fullVersion == "" {
		return ""
	}
	parts := strings.Split(fullVersion, ".")
	if len(parts) == 0 {
		return ""
	}
	return strings.TrimSpace(parts[0])
}

func replaceChromeUserAgentVersion(userAgent string, replacement string) string {
	replacement = strings.TrimSpace(replacement)
	if replacement == "" {
		return strings.ReplaceAll(strings.TrimSpace(userAgent), "HeadlessChrome/", "Chrome/")
	}
	normalized := strings.ReplaceAll(strings.TrimSpace(userAgent), "HeadlessChrome/", "Chrome/")
	return chromeVersionPattern.ReplaceAllString(normalized, "Chrome/"+replacement)
}

func patchBrandVersions(values []browserprofile.BrandVersion, version string, patchNotABrand bool) []browserprofile.BrandVersion {
	version = strings.TrimSpace(version)
	if version == "" {
		return values
	}

	out := make([]browserprofile.BrandVersion, 0, len(values))
	for _, value := range values {
		item := value
		brandLower := strings.ToLower(strings.TrimSpace(item.Brand))
		if brandLower == "chromium" || brandLower == "google chrome" {
			item.Version = version
		} else if patchNotABrand && strings.Contains(brandLower, "not_a brand") && strings.Count(version, ".") == 3 {
			item.Version = "24.0.0.0"
		}
		out = append(out, item)
	}
	return out
}

func navigatorPlatformForProfile(profile browserprofile.Profile) string {
	switch strings.ToLower(strings.TrimSpace(profile.Platform)) {
	case "windows":
		return "Win32"
	case "macos":
		return "MacIntel"
	case "linux":
		return "Linux x86_64"
	default:
		return strings.TrimSpace(profile.Platform)
	}
}

type profileDisplayMetrics struct {
	ViewportWidth  int
	ViewportHeight int
	ScreenWidth    int
	ScreenHeight   int
	AvailWidth     int
	AvailHeight    int
	AvailTop       int
	OuterWidth     int
	OuterHeight    int
	PositionX      int
	PositionY      int
}

func profileDisplayMetricsFor(profile browserprofile.Profile) profileDisplayMetrics {
	screenWidth := profile.Viewport.Width
	screenHeight := profile.Viewport.Height
	if screenWidth <= 0 {
		screenWidth = 1920
	}
	if screenHeight <= 0 {
		screenHeight = 1080
	}

	chromeHeight := 85
	systemReservedHeight := 40
	availTop := 0
	if strings.EqualFold(strings.TrimSpace(profile.Platform), "macos") {
		chromeHeight = 88
		systemReservedHeight = 25
		availTop = 25
	}

	if profile.Mobile {
		chromeHeight = 0
		systemReservedHeight = 0
		availTop = 0
	}

	availHeight := max(screenHeight-systemReservedHeight, 1)
	viewportHeight := max(availHeight-chromeHeight, 1)
	return profileDisplayMetrics{
		ViewportWidth:  screenWidth,
		ViewportHeight: viewportHeight,
		ScreenWidth:    screenWidth,
		ScreenHeight:   screenHeight,
		AvailWidth:     screenWidth,
		AvailHeight:    availHeight,
		AvailTop:       availTop,
		OuterWidth:     screenWidth,
		OuterHeight:    availHeight,
		PositionX:      0,
		PositionY:      availTop,
	}
}

// applyProfileLanguageHint overrides the profile's locale-derived fields when
// the requested language differs from the cached profile's. A bare language
// hint that already matches the profile language is treated as a no-op so that
// an explicit profile region (e.g. en-GB) isn't clobbered by a default (en-US).
func applyProfileLanguageHint(profile browserprofile.Profile, langCode string) browserprofile.Profile {
	hint := ParseLocale(langCode)
	if hint.Language == "" {
		return profile
	}

	current := ParseLocale(profile.Locale)
	if current.Language == hint.Language && (hint.Country == "" || current.Country == hint.Country) {
		return profile
	}

	primary := PrimaryLanguageTag(langCode)
	if primary == "" {
		return profile
	}

	profile.AcceptLanguage = BuildAcceptLanguageHeader(langCode)
	profile.NavigatorLangs = []string{primary}
	profile.Locale = primary

	// Fill in country from the language default when the hint has no explicit country,
	// so TimezoneForLocale can resolve even bare language codes like "de".
	if hint.Country == "" {
		if country, ok := defaultLocaleCountryByLanguage[hint.Language]; ok {
			hint.Country = country
		}
	}
	if tz := TimezoneForLocale(hint); tz != "" {
		profile.Timezone = tz
	}

	return profile
}

func applyProfile(page *rod.Page, profile browserprofile.Profile, minimal bool) error {
	if page == nil {
		return fmt.Errorf("page is nil")
	}

	navigatorLangs := profileNavigatorLanguages(profile)
	acceptLanguage := strings.TrimSpace(profile.AcceptLanguage)
	if acceptLanguage == "" {
		acceptLanguage = navigatorLangs[0]
	}
	locale := strings.TrimSpace(profile.Locale)
	if locale == "" {
		locale = navigatorLangs[0]
	}

	metrics := profileDisplayMetricsFor(profile)

	metadata := &proto.EmulationUserAgentMetadata{
		Brands:          toProtoBrandVersions(profile.UACHBrands),
		FullVersionList: toProtoBrandVersions(profile.UACHFullVerList),
		Platform:        strings.TrimSpace(profile.Platform),
		PlatformVersion: strings.TrimSpace(profile.PlatformVersion),
		Architecture:    strings.TrimSpace(profile.Architecture),
		Bitness:         strings.TrimSpace(profile.Bitness),
		Mobile:          profile.Mobile,
	}

	if err := (proto.NetworkSetUserAgentOverride{
		UserAgent:         strings.TrimSpace(profile.UserAgent),
		AcceptLanguage:    acceptLanguage,
		Platform:          navigatorPlatformForProfile(profile),
		UserAgentMetadata: metadata,
	}).Call(page); err != nil {
		return fmt.Errorf("set user agent override failed: %w", err)
	}

	if err := (proto.EmulationSetLocaleOverride{
		Locale: locale,
	}).Call(page); err != nil {
		return fmt.Errorf("set locale override failed: %w", err)
	}

	if err := (proto.EmulationSetTimezoneOverride{
		TimezoneID: strings.TrimSpace(profile.Timezone),
	}).Call(page); err != nil {
		return fmt.Errorf("set timezone override failed: %w", err)
	}

	if err := (proto.EmulationSetDeviceMetricsOverride{
		Width:             metrics.ViewportWidth,
		Height:            metrics.ViewportHeight,
		DeviceScaleFactor: 1,
		Mobile:            profile.Mobile,
		ScreenWidth:       &metrics.ScreenWidth,
		ScreenHeight:      &metrics.ScreenHeight,
		PositionX:         &metrics.PositionX,
		PositionY:         &metrics.PositionY,
	}).Call(page); err != nil {
		return fmt.Errorf("set device metrics failed: %w", err)
	}

	if err := (proto.NetworkSetExtraHTTPHeaders{
		Headers: proto.NetworkHeaders{
			"Accept-Language": gson.New(acceptLanguage),
		},
	}).Call(page); err != nil {
		return fmt.Errorf("set extra headers failed: %w", err)
	}

	if minimal {
		return nil
	}

	if err := evalPatchScript(page, profile, navigatorLangs, metrics); err != nil {
		return err
	}

	return nil
}

func (b *Browser) restoreLaneCookies(ctx context.Context, page *rod.Page) error {
	if b == nil || b.ProxyLaneStore == nil {
		return nil
	}
	laneKey := proxyLaneKeyFromContext(ctx)
	if laneKey.Empty() {
		return nil
	}
	cookies := b.ProxyLaneStore.Cookies(laneKey)
	if len(cookies) == 0 {
		return nil
	}
	if err := (proto.NetworkSetCookies{Cookies: cookieParams(cookies)}).Call(page); err != nil {
		return fmt.Errorf("restore lane cookies: %w", err)
	}
	return nil
}

func (b *Browser) saveLaneCookies(ctx context.Context, page *rod.Page, pageURL string) {
	if b == nil || b.ProxyLaneStore == nil || page == nil {
		return
	}
	laneKey := proxyLaneKeyFromContext(ctx)
	if laneKey.Empty() {
		return
	}
	res, err := (proto.NetworkGetCookies{Urls: []string{pageURL}}).Call(page)
	if err != nil {
		WithRequest(ctx).WithError(err).Debug("Save lane cookies failed")
		return
	}
	b.ProxyLaneStore.SaveCookies(laneKey, res.Cookies)
}

type mainDocumentStatusWatcher struct {
	cancel context.CancelFunc
	done   chan struct{}
	mu     sync.Mutex
	status int
}

type networkUsageWatcher struct {
	cancel context.CancelFunc
	done   chan struct{}
}

type workerPatchWatcher struct {
	cancel context.CancelFunc
	done   chan struct{}
	page   *rod.Page
}

var pageNetworkUsageWatchers sync.Map
var pageWorkerPatchWatchers sync.Map

func startMainDocumentStatusWatcher(ctx context.Context, page *rod.Page) *mainDocumentStatusWatcher {
	watchCtx, cancel := context.WithCancel(EnsureContext(ctx))
	watcher := &mainDocumentStatusWatcher{
		cancel: cancel,
		done:   make(chan struct{}),
	}

	wait := page.Context(watchCtx).EachEvent(func(e *proto.NetworkResponseReceived) bool {
		if e == nil || e.Response == nil || e.Type != proto.NetworkResourceTypeDocument {
			return false
		}
		watcher.mu.Lock()
		watcher.status = e.Response.Status
		watcher.mu.Unlock()
		return false
	})

	go func() {
		defer close(watcher.done)
		wait()
	}()

	return watcher
}

func (w *mainDocumentStatusWatcher) Stop() {
	if w == nil {
		return
	}
	w.cancel()
	select {
	case <-w.done:
	case <-time.After(100 * time.Millisecond):
	}
}

func startNetworkUsageWatcher(ctx context.Context, page *rod.Page) *networkUsageWatcher {
	if networkUsageFromContext(ctx) == nil {
		return nil
	}
	if err := (proto.NetworkEnable{}).Call(page); err != nil {
		WithRequest(ctx).WithError(err).Debug("Enable network usage tracking failed")
		return nil
	}

	watchCtx, cancel := context.WithCancel(EnsureContext(ctx))
	watcher := &networkUsageWatcher{
		cancel: cancel,
		done:   make(chan struct{}),
	}

	wait := page.Context(watchCtx).EachEvent(func(e *proto.NetworkLoadingFinished) bool {
		if e != nil && e.EncodedDataLength > 0 {
			AddNetworkBytes(ctx, int64(e.EncodedDataLength))
		}
		return false
	})

	go func() {
		defer close(watcher.done)
		wait()
	}()

	return watcher
}

func startWorkerPatchWatcher(ctx context.Context, page *rod.Page, script string) (*workerPatchWatcher, error) {
	if page == nil || strings.TrimSpace(script) == "" {
		return nil, nil
	}

	watchCtx, cancel := context.WithCancel(EnsureContext(ctx))
	watcher := &workerPatchWatcher{
		cancel: cancel,
		done:   make(chan struct{}),
		page:   page,
	}

	scopedPage := page.Context(watchCtx)
	started := make(chan struct{})
	go func() {
		defer close(watcher.done)
		wait := scopedPage.EachEvent(func(e *proto.TargetAttachedToTarget) bool {
			if e == nil || e.SessionID == "" {
				return false
			}
			go injectWorkerPatch(watchCtx, scopedPage.Browser(), e, script)
			return false
		})
		close(started)
		wait()
	}()

	<-started
	if err := (proto.TargetSetAutoAttach{
		AutoAttach:             true,
		WaitForDebuggerOnStart: true,
		Flatten:                true,
		Filter:                 workerTargetFilter(),
	}).Call(scopedPage); err != nil {
		cancel()
		<-watcher.done
		return nil, fmt.Errorf("enable worker auto-attach: %w", err)
	}

	return watcher, nil
}

func workerTargetFilter() proto.TargetTargetFilter {
	return proto.TargetTargetFilter{
		{Type: "worker"},
		{Type: string(proto.TargetTargetInfoTypeSharedWorker)},
		{Type: string(proto.TargetTargetInfoTypeServiceWorker)},
	}
}

func injectWorkerPatch(ctx context.Context, browser *rod.Browser, e *proto.TargetAttachedToTarget, script string) {
	if browser == nil || e == nil || e.SessionID == "" {
		return
	}

	injectCtx, cancel := context.WithTimeout(EnsureContext(ctx), 3*time.Second)
	defer cancel()

	if isPatchableWorkerTarget(e.TargetInfo) {
		eval := proto.RuntimeEvaluate{
			Expression:                  script,
			Silent:                      true,
			AllowUnsafeEvalBlockedByCSP: true,
		}
		if _, err := browser.Call(injectCtx, string(e.SessionID), eval.ProtoReq(), eval); err != nil && !errors.Is(err, context.Canceled) {
			logrus.WithError(err).Debug("Worker profile patch failed")
		}
	}

	if e.WaitingForDebugger {
		run := proto.RuntimeRunIfWaitingForDebugger{}
		if _, err := browser.Call(injectCtx, string(e.SessionID), run.ProtoReq(), run); err != nil && !errors.Is(err, context.Canceled) {
			logrus.WithError(err).Debug("Resume worker after profile patch failed")
		}
	}
}

func isPatchableWorkerTarget(info *proto.TargetTargetInfo) bool {
	if info == nil {
		return true
	}
	switch string(info.Type) {
	case "worker", string(proto.TargetTargetInfoTypeSharedWorker), string(proto.TargetTargetInfoTypeServiceWorker):
		return true
	default:
		return false
	}
}

func rememberNetworkUsageWatcher(page *rod.Page, watcher *networkUsageWatcher) {
	if page == nil || watcher == nil {
		return
	}
	pageNetworkUsageWatchers.Store(page, watcher)
}

func rememberWorkerPatchWatcher(page *rod.Page, watcher *workerPatchWatcher) {
	if page == nil || watcher == nil {
		return
	}
	pageWorkerPatchWatchers.Store(page, watcher)
}

func stopNetworkUsageWatcher(page *rod.Page) {
	if page == nil {
		return
	}
	raw, ok := pageNetworkUsageWatchers.LoadAndDelete(page)
	if !ok {
		return
	}
	if watcher, ok := raw.(*networkUsageWatcher); ok {
		watcher.Stop()
	}
}

func stopWorkerPatchWatcher(page *rod.Page) {
	if page == nil {
		return
	}
	raw, ok := pageWorkerPatchWatchers.LoadAndDelete(page)
	if !ok {
		return
	}
	if watcher, ok := raw.(*workerPatchWatcher); ok {
		watcher.Stop()
	}
}

func (w *networkUsageWatcher) Stop() {
	if w == nil {
		return
	}
	w.cancel()
	select {
	case <-w.done:
	case <-time.After(100 * time.Millisecond):
	}
}

func (w *workerPatchWatcher) Stop() {
	if w == nil {
		return
	}
	if w.page != nil {
		disableCtx, cancel := context.WithTimeout(context.Background(), time.Second)
		_ = (proto.TargetSetAutoAttach{
			AutoAttach:             false,
			WaitForDebuggerOnStart: false,
			Flatten:                true,
			Filter:                 workerTargetFilter(),
		}).Call(w.page.Context(disableCtx))
		cancel()
	}
	w.cancel()
	select {
	case <-w.done:
	case <-time.After(100 * time.Millisecond):
	}
}

func (w *mainDocumentStatusWatcher) Status() int {
	if w == nil {
		return 0
	}
	w.mu.Lock()
	defer w.mu.Unlock()
	return w.status
}

func classifyMainDocumentStatus(status int) error {
	switch status {
	case http.StatusForbidden:
		return ErrBlocked
	case http.StatusTooManyRequests:
		return ErrRateLimited
	default:
		return nil
	}
}

func profileNavigatorLanguages(profile browserprofile.Profile) []string {
	langs := make([]string, 0, len(profile.NavigatorLangs))
	for _, language := range profile.NavigatorLangs {
		trimmed := strings.TrimSpace(language)
		if trimmed == "" {
			continue
		}
		langs = append(langs, trimmed)
	}
	if len(langs) > 0 {
		return langs
	}

	acceptLanguage := strings.TrimSpace(profile.AcceptLanguage)
	if acceptLanguage != "" {
		parts := strings.Split(acceptLanguage, ",")
		langs = make([]string, 0, len(parts))
		for _, part := range parts {
			part = strings.TrimSpace(part)
			if part == "" {
				continue
			}
			if idx := strings.Index(part, ";"); idx >= 0 {
				part = strings.TrimSpace(part[:idx])
			}
			if part != "" {
				langs = append(langs, part)
			}
		}
		if len(langs) > 0 {
			return langs
		}
	}

	if locale := strings.TrimSpace(profile.Locale); locale != "" {
		return []string{locale}
	}
	return []string{"en-US"}
}

func toProtoBrandVersions(values []browserprofile.BrandVersion) []*proto.EmulationUserAgentBrandVersion {
	out := make([]*proto.EmulationUserAgentBrandVersion, 0, len(values))
	for _, value := range values {
		brand := strings.TrimSpace(value.Brand)
		version := strings.TrimSpace(value.Version)
		if brand == "" || version == "" {
			continue
		}
		out = append(out, &proto.EmulationUserAgentBrandVersion{
			Brand:   brand,
			Version: version,
		})
	}
	return out
}

func buildProfilePatchScript(profile browserprofile.Profile, langs []string, metrics profileDisplayMetrics) (string, error) {
	langsJSON, err := json.Marshal(langs)
	if err != nil {
		return "", fmt.Errorf("marshal navigator languages: %w", err)
	}

	webGLVendor := strings.TrimSpace(profile.WebGLVendor)
	if webGLVendor == "" {
		webGLVendor = "Intel Inc."
	}
	webGLRenderer := strings.TrimSpace(profile.WebGLRenderer)
	if webGLRenderer == "" {
		webGLRenderer = "Intel Iris OpenGL Engine"
	}

	webGLVendorJSON, err := json.Marshal(webGLVendor)
	if err != nil {
		return "", fmt.Errorf("marshal webgl vendor: %w", err)
	}
	webGLRendererJSON, err := json.Marshal(webGLRenderer)
	if err != nil {
		return "", fmt.Errorf("marshal webgl renderer: %w", err)
	}

	return fmt.Sprintf("(() => {\nconst __langs = %s;\nconst __w = %d;\nconst __h = %d;\nconst __screenW = %d;\nconst __screenH = %d;\nconst __availW = %d;\nconst __availH = %d;\nconst __availTop = %d;\nconst __outerW = %d;\nconst __outerH = %d;\nconst __webglVendor = %s;\nconst __webglRenderer = %s;\n%s\n})();",
		string(langsJSON),
		metrics.ViewportWidth,
		metrics.ViewportHeight,
		metrics.ScreenWidth,
		metrics.ScreenHeight,
		metrics.AvailWidth,
		metrics.AvailHeight,
		metrics.AvailTop,
		metrics.OuterWidth,
		metrics.OuterHeight,
		string(webGLVendorJSON),
		string(webGLRendererJSON),
		string(browserprofile.PatchJS),
	), nil
}

func evalPatchScript(page *rod.Page, profile browserprofile.Profile, langs []string, metrics profileDisplayMetrics) error {
	script, err := buildProfilePatchScript(profile, langs, metrics)
	if err != nil {
		return err
	}

	_, err = page.EvalOnNewDocument(script)
	if err != nil {
		return fmt.Errorf("eval patch script: %w", err)
	}
	return nil
}

// Navigate connects to Chromium, creates a page, applies a coherent profile and
// proxy auth, then navigates to URL. It returns an initialized page ready for
// selector queries, or an error when browser setup/navigation fails.
func (b *Browser) Navigate(ctx context.Context, URL string) (*rod.Page, error) {
	ctx = EnsureContext(ctx)
	if err := ctx.Err(); err != nil {
		return nil, err
	}

	WithRequest(ctx).WithField("url", URL).Debug("Navigate")
	// Per-context proxy override is only used for unauthenticated proxies on a
	// Chrome that was launched without a process-level proxy. When this Browser
	// was launched with a proxy (b.ProxyURL set), Chrome handles routing and
	// auth natively for the whole process; per-context override is skipped to
	// avoid breaking Chrome's auth flow.
	contextProxyURL := ""
	if strings.TrimSpace(b.ProxyURL) == "" {
		contextProxyURL = requestProxyURLFromContext(ctx)
	}
	hasProxy := contextProxyURL != "" || strings.TrimSpace(b.ProxyURL) != ""

	browser, err := b.ensureConnectedBrowser(ctx, false)
	if err != nil {
		return nil, fmt.Errorf("browser connect failed: %w", err)
	}
	if hasProxy || b.Insecure {
		if err := browser.IgnoreCertErrors(true); err != nil {
			return nil, fmt.Errorf("ignore cert errors failed: %w", err)
		}
	}

	page, browserContextID, err := createIsolatedPage(browser, contextProxyURL)
	if err != nil {
		// Single-shot reconnect for stale websocket sessions.
		browser, err = b.ensureConnectedBrowser(ctx, true)
		if err != nil {
			return nil, fmt.Errorf("create isolated page failed, reconnect also failed: %w", err)
		}
		page, browserContextID, err = createIsolatedPage(browser, contextProxyURL)
		if err != nil {
			return nil, fmt.Errorf("create isolated page failed after reconnect: %w", err)
		}
	}

	// closeOnErr closes page then disposes context. Order matters: disposing the
	// context first causes Chrome to kill the page target before our Close call,
	// producing a spurious "target closed" error on the page.Close() that follows.
	closeOnErr := func() {
		stopWorkerPatchWatcher(page)
		stopNetworkUsageWatcher(page)
		if cerr := page.Close(); cerr != nil && !isBrowserClosedError(cerr) {
			WithRequest(ctx).WithError(cerr).Debug("Close page after navigate error failed")
		}
		if derr := disposeBrowserContext(browser, browserContextID); derr != nil && !isBrowserClosedError(derr) {
			WithRequest(ctx).WithError(derr).Debug("Dispose browser context after navigate error failed")
		}
	}

	profile, laneKey := b.laneProfile(ctx, browser)
	SetBrowserProfileID(ctx, profile.ID)
	minimalProfile := minimalBrowserProfileFromContext(ctx)
	WithRequest(ctx).WithFields(logrus.Fields{
		"lane_id":         laneKey,
		"minimal_profile": minimalProfile,
	}).Info("Browser profile selected")
	if err := applyProfile(page, profile, minimalProfile); err != nil {
		closeOnErr()
		return nil, fmt.Errorf("apply profile %s (%s) failed: %w", profile.ID, laneKey, err)
	}
	if err := b.restoreLaneCookies(ctx, page); err != nil {
		closeOnErr()
		return nil, err
	}

	page = page.Context(ctx)
	if !minimalProfile {
		metrics := profileDisplayMetricsFor(profile)
		patchScript, err := buildProfilePatchScript(profile, profileNavigatorLanguages(profile), metrics)
		if err != nil {
			closeOnErr()
			return nil, err
		}
		workerPatchWatcher, err := startWorkerPatchWatcher(ctx, page, patchScript)
		if err != nil {
			closeOnErr()
			return nil, err
		}
		rememberWorkerPatchWatcher(page, workerPatchWatcher)
	}
	if err := b.configureRequestBlocking(ctx, page); err != nil {
		closeOnErr()
		return nil, fmt.Errorf("configure request blocking failed: %w", err)
	}
	networkUsageWatcher := startNetworkUsageWatcher(ctx, page)
	if b.LeavePageOpen {
		defer networkUsageWatcher.Stop()
	} else {
		rememberNetworkUsageWatcher(page, networkUsageWatcher)
	}
	statusWatcher := startMainDocumentStatusWatcher(ctx, page)
	defer statusWatcher.Stop()
	timedPage := page.Timeout(b.Timeout)

	if err := timedPage.Navigate(URL); err != nil {
		closeOnErr()
		return nil, classifyProxyNetworkError(err)
	}

	// Avoid panics from MustWaitLoad when the target navigates/closes mid-wait.
	loadWaitTimeout := minPositiveDuration(b.Timeout, b.WaitLoadTime)
	loadWaitPage := page.Timeout(loadWaitTimeout)
	if werr := loadWaitPage.WaitLoad(); werr != nil {
		if errors.Is(werr, context.DeadlineExceeded) {
			// Some engines keep loading background resources while the DOM is already usable.
			// Treat load timeout as non-fatal and let engine-specific selector timeouts decide.
			WithRequest(ctx).WithField("timeout", loadWaitTimeout.String()).Debug(
				fmt.Sprintf("WaitLoad timed out after %s; continuing with partial page state", loadWaitTimeout),
			)
		} else {
			WithRequest(ctx).WithError(werr).Debug("WaitLoad returned early")
		}
	}

	// may cause bugs with google
	if b.WaitRequests {
		wait := timedPage.WaitRequestIdle(300*time.Millisecond, nil, nil, nil)
		wait()
	}

	// WaitStable internally waits for page load too, so cap it separately.
	// Selector parsing still decides whether a partially loaded page is usable.
	stableWaitTimeout := minPositiveDuration(b.Timeout, b.WaitLoadTime+time.Second)
	if err := page.Timeout(stableWaitTimeout).WaitStable(800 * time.Millisecond); err != nil {
		WithRequest(ctx).WithError(err).Debug("WaitStable returned early; continuing")
	}
	if err := classifyMainDocumentStatus(statusWatcher.Status()); err != nil {
		closeOnErr()
		return nil, err
	}
	b.saveLaneCookies(ctx, page, URL)
	return page, nil
}

// Close closes the active browser connection.
func (b *Browser) Close() error {
	if b == nil || b.browserAddr == "" {
		return nil
	}

	state := b.connectionState()
	state.mu.Lock()
	defer state.mu.Unlock()

	browser := state.browser
	if browser == nil {
		browser = b.newRodBrowser()
		if err := browser.Connect(); err != nil {
			if isBrowserClosedError(err) {
				return nil
			}
			return err
		}
	}

	if state.authCancel != nil {
		state.authCancel()
		if state.authStopped != nil {
			<-state.authStopped
		}
		state.authCancel = nil
		state.authStopped = nil
	}
	state.browser = nil
	state.laneProfiles = nil
	if err := browser.Close(); err != nil && !isBrowserClosedError(err) {
		return err
	}
	return nil
}

func (b *Browser) ClosePage(ctx context.Context, page *rod.Page, timeout time.Duration) error {
	return ClosePageWithTimeout(ctx, page, timeout)
}

func isBrowserClosedError(err error) bool {
	if err == nil {
		return false
	}
	msg := strings.ToLower(err.Error())
	return strings.Contains(msg, "connection refused") ||
		strings.Contains(msg, "closed network connection") ||
		strings.Contains(msg, "target closed") ||
		strings.Contains(msg, "eof")
}

// ClosePageWithTimeout bounds page close calls so shutdown paths don't hang.
func ClosePageWithTimeout(ctx context.Context, page *rod.Page, timeout time.Duration) error {
	if page == nil {
		return nil
	}
	stopWorkerPatchWatcher(page)
	stopNetworkUsageWatcher(page)
	if timeout <= 0 {
		timeout = time.Second
	}
	baseCtx := EnsureContext(ctx)
	if baseCtx.Err() != nil {
		baseCtx = context.Background()
	}
	closeCtx, cancel := context.WithTimeout(baseCtx, timeout)
	defer cancel()

	pageWithTimeout := page.Context(closeCtx)
	info, _ := pageWithTimeout.Info()

	if err := pageWithTimeout.Close(); err != nil && !isBrowserClosedError(err) {
		return err
	}

	if info != nil && info.BrowserContextID != "" {
		if derr := disposeBrowserContext(page.Browser().Context(closeCtx), info.BrowserContextID); derr != nil && !isBrowserClosedError(derr) {
			return derr
		}
	}
	return nil
}

// RecoverEnginePanic converts recovered panics to a typed engine error and
// logs stack trace with engine context.
func RecoverEnginePanic(engine string, recovered interface{}, logger *EngineLogger) error {
	return RecoverEnginePanicWithContext(context.TODO(), engine, recovered, logger)
}

func RecoverEnginePanicWithContext(ctx context.Context, engine string, recovered interface{}, logger *EngineLogger) error {
	stack := debug.Stack()
	if logger != nil {
		logger.Error("Recovered panic in %s Search: panic=%v\n%s", engine, recovered, string(stack))
	} else {
		WithRequestEngine(ctx, engine).Errorf("Recovered panic in %s Search: panic=%v\n%s", engine, recovered, string(stack))
	}
	return fmt.Errorf("%w: %s", ErrEngineInternal, engine)
}

// IsRodObjectNotFound reports element/object lookup misses across rod error
// variants used by selector calls.
func IsRodObjectNotFound(err error) bool {
	if err == nil {
		return false
	}
	var objectErr *rod.ObjectNotFoundError
	if errors.As(err, &objectErr) {
		return true
	}
	var elementErr *rod.ElementNotFoundError
	return errors.As(err, &elementErr)
}