app/control/proxy/config.py

"""Proxy clearance config helpers shared by control and dataplane code."""

from dataclasses import dataclass
from typing import Any

from app.platform.config.snapshot import get_config


@dataclass(frozen=True)
class ClearanceConfig:
    cf_cookies: str = ""
    user_agent: str = ""
    cf_clearance: str = ""
    browser: str = ""


def _cfg_str(cfg: Any, key: str) -> str:
    value = cfg.get_str(key, "")
    return value if value.strip() else ""


def first_config_str(cfg: Any, *keys: str) -> str:
    for key in keys:
        value = _cfg_str(cfg, key)
        if value:
            return value
    return ""


def resolve_clearance_config(cfg: Any | None = None) -> ClearanceConfig:
    cfg = cfg or get_config()
    return ClearanceConfig(
        cf_cookies=first_config_str(
            cfg,
            "proxy.cf_cookies",
            "proxy.clearance.cf_cookies",
        ),
        user_agent=first_config_str(
            cfg,
            "proxy.user_agent",
            "proxy.clearance.user_agent",
        ),
        cf_clearance=first_config_str(
            cfg,
            "proxy.cf_clearance",
            "proxy.clearance.cf_clearance",
        ),
        browser=first_config_str(
            cfg,
            "proxy.browser",
            "proxy.clearance.browser",
        ),
    )


__all__ = ["ClearanceConfig", "first_config_str", "resolve_clearance_config"]