script.service.ultimate/lib/streaming_providers/providers/hrti/constants.py

# [file name]: constants.py
# [file content begin]
# streaming_providers/providers/hrti/constants.py
"""
HRTi provider constants and default configurations
"""


class HRTiDefaults:
    """Default values for HRTi provider"""

    # Provider information
    PROVIDER_LOGO = "https://upload.wikimedia.org/wikipedia/en/thumb/9/9e/Logo_of_the_HRT.svg/2560px-Logo_of_the_HRT.svg.png"
    PROVIDER_NAME = "HRTi"

    # Website and base URLs
    BASE_WEBSITE = "https://hrti.hrt.hr"
    BASE_URL = "https://hrti.hrt.hr"
    HSAPI_BASE_URL = "https://hsapi.aviion.tv/client.svc/json"

    # Configuration endpoints
    ENV_ENDPOINT = f"{BASE_URL}/assets/config/env.json"
    CONFIG_ENDPOINT = f"{BASE_URL}/assets/config/config.production.json"

    # API endpoints
    API_ENDPOINTS = {
        "get_ip": f"{BASE_URL}/api/api/ott/getIPAddress",
        "grant_access": f"{BASE_URL}/api/api/ott/GrantAccess",
        "channels": f"{BASE_URL}/api/api/ott/GetChannels",
        "programme": f"{BASE_URL}/api/api/ott/GetProgramme",
        "authorize_session": f"{BASE_URL}/api/api/ott/AuthorizeSession",
        "report_session": f"{BASE_URL}/api/api/ott/ReportSessionEvent",
        "register_device": f"{HSAPI_BASE_URL}/RegisterDevice",
        "content_ratings": f"{HSAPI_BASE_URL}/ContentRatingsGet",
        "profiles": f"{HSAPI_BASE_URL}/ProfilesGet",
    }

    # DRM and License endpoints
    LICENSE_URL = "https://lic.drmtoday.com/license-proxy-widevine/cenc/"

    # Device information
    DEVICE_REFERENCE_ID = "6"  # String '6' as required by headers
    OPERATOR_REFERENCE_ID = "hrt"
    MERCHANT = "aviion2"
    CONNECTION_TYPE = "LAN/WiFi"
    APPLICATION_VERSION = "5.97.6"
    OS_VERSION = "Linux"
    CLIENT_TYPE = "Chrome 142"

    # User Agent
    USER_AGENT = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/142.0.0.0 Safari/537.36"

    # HTTP settings
    DEFAULT_TIMEOUT = 30


class HRTiConfig:
    """Configuration class for HRTi provider"""

    def __init__(self, config_dict: dict = None):
        """Initialize with optional configuration overrides"""
        config = config_dict or {}

        self.logo = config.get("logo", HRTiDefaults.PROVIDER_LOGO)

        # Website and base URLs
        self.base_website = config.get("base_website", HRTiDefaults.BASE_WEBSITE)
        self.base_url = config.get("base_url", HRTiDefaults.BASE_URL)
        self.hsapi_base_url = config.get("hsapi_base_url", HRTiDefaults.HSAPI_BASE_URL)
        self.env_endpoint = config.get("env_endpoint", HRTiDefaults.ENV_ENDPOINT)
        self.config_endpoint = config.get("config_endpoint", HRTiDefaults.CONFIG_ENDPOINT)

        # API endpoints configuration
        self.api_endpoints = config.get("api_endpoints", HRTiDefaults.API_ENDPOINTS.copy())

        # DRM and License
        self.license_url = config.get("license_url", HRTiDefaults.LICENSE_URL)

        # Device configuration
        self.device_reference_id = config.get(
            "device_reference_id", HRTiDefaults.DEVICE_REFERENCE_ID
        )
        self.operator_reference_id = config.get(
            "operator_reference_id", HRTiDefaults.OPERATOR_REFERENCE_ID
        )
        self.merchant = config.get("merchant", HRTiDefaults.MERCHANT)
        self.connection_type = config.get("connection_type", HRTiDefaults.CONNECTION_TYPE)
        self.application_version = config.get(
            "application_version", HRTiDefaults.APPLICATION_VERSION
        )
        self.os_version = config.get("os_version", HRTiDefaults.OS_VERSION)
        self.client_type = config.get("client_type", HRTiDefaults.CLIENT_TYPE)

        # HTTP settings
        self.user_agent = config.get("user_agent", HRTiDefaults.USER_AGENT)
        self.timeout = config.get("timeout", HRTiDefaults.DEFAULT_TIMEOUT)

        # Web API URL (can be updated from config)
        self.web_api_url = config.get("web_api_url", "api/api/ott")

    def update_from_api(self, env_data: dict, config_data: dict):
        """Update configuration from API responses"""
        try:
            # Update from env data
            if "applicationVersion" in env_data:
                self.application_version = env_data["applicationVersion"]

            # Update from config data
            if "apiUrl" in config_data:
                self.hsapi_base_url = config_data["apiUrl"]

            if "webApiUrl" in config_data:
                self.web_api_url = config_data["webApiUrl"]
                # Update API endpoints with new web API URL
                base_api_url = f"{self.base_url}/{self.web_api_url}"
                self.api_endpoints.update(
                    {
                        "get_ip": f"{base_api_url}/getIPAddress",
                        "grant_access": f"{base_api_url}/GrantAccess",
                        "channels": f"{base_api_url}/GetChannels",
                        "programme": f"{base_api_url}/GetProgramme",
                        "authorize_session": f"{base_api_url}/AuthorizeSession",
                    }
                )

            if "operators" in config_data and config_data["operators"]:
                operator = config_data["operators"][0]
                if "playerMerchant" in operator:
                    self.merchant = operator["playerMerchant"]
                if "selfcareUrl" in operator:
                    # Store selfcare URL if needed
                    pass

        except Exception as e:
            # Log error but don't raise - use defaults if update fails
            import logging

            logging.debug(f"Error updating HRTi config from API: {e}")

    def get_base_headers(self) -> dict:
        """Get base HTTP headers"""
        return {
            "User-Agent": self.user_agent,
            "Accept": "application/json",
            "Content-Type": "application/json",
        }

    def get_auth_headers(
        self, device_id: str = None, ip_address: str = None, token: str = None
    ) -> dict:
        """Get authenticated headers for API requests"""
        headers = self.get_base_headers()

        if device_id:
            headers["deviceid"] = device_id
        if ip_address:
            headers["ipaddress"] = ip_address
        if token:
            headers["authorization"] = f"Client {token}"

        headers.update(
            {
                "devicetypeid": self.device_reference_id,  # Added devicetypeid
                "operatorreferenceid": self.operator_reference_id,
                "origin": self.base_website,
                "referer": self.base_website,
            }
        )

        return headers


# [file content end]