browser_detect.py

"""Detect and launch Chromium-based browsers with CDP enabled."""

import os
import socket
import subprocess
from pathlib import Path

_BROWSERS = [
    {
        "name": "Brave",
        "paths": [
            r"C:\Program Files\BraveSoftware\Brave-Browser\Application\brave.exe",
            os.path.expandvars(r"%LOCALAPPDATA%\BraveSoftware\Brave-Browser\Application\brave.exe"),
        ],
    },
    {
        "name": "Chrome",
        "paths": [
            r"C:\Program Files\Google\Chrome\Application\chrome.exe",
            os.path.expandvars(r"%LOCALAPPDATA%\Google\Chrome\Application\chrome.exe"),
        ],
    },
    {
        "name": "Edge",
        "paths": [
            r"C:\Program Files (x86)\Microsoft\Edge\Application\msedge.exe",
            os.path.expandvars(r"%PROGRAMFILES%\Microsoft\Edge\Application\msedge.exe"),
        ],
    },
]

USER_DATA_DIR = Path.home() / "AIguessr_Browser"
CDP_PORT = 9222


def detect_browsers() -> list[dict]:
    found = []
    for browser in _BROWSERS:
        for p in browser["paths"]:
            if Path(p).exists():
                found.append({"name": browser["name"], "path": p})
                break
    return found


def launch_browser(path: str) -> subprocess.Popen:
    args = [
        path,
        f"--remote-debugging-port={CDP_PORT}",
        "--remote-allow-origins=*",
        f"--user-data-dir={USER_DATA_DIR}",
        "https://www.geoguessr.com",
    ]
    return subprocess.Popen(args)


def is_cdp_available() -> bool:
    try:
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        s.settimeout(1)
        s.connect(("127.0.0.1", CDP_PORT))
        s.close()
        return True
    except (ConnectionRefusedError, socket.timeout, OSError):
        return False