import logging
from typing import Any, Dict, Optional

import requests

logging.basicConfig(
    level=logging.DEBUG,
    format="%(asctime)s[%(levelname)-5s] %(message)s",
    datefmt="%Y-%m-%d %H:%M:%S",
)
log = logging.getLogger(__file__)


class Endpoint:
    """
    Utility class for handling endpoint operations.
    """

    @staticmethod
    def get_endpoint_api_key(
        endpoint_name: str, account_api_key: str, instance: str
    ) -> Optional[str]:
        """
        Fetch endpoint API key from VastAI console following the healthcheck pattern.

        Args:
            endpoint_name: Name of the endpoint
            account_api_key: Account API key for authentication

        Returns:
            Endpoint API key if successful, None otherwise
        """
        vast_console_url = "https://console.vast.ai/api/v0/endptjobs/"
        headers = {"Authorization": f"Bearer {account_api_key}"}

        try:
            log.debug(f"Fetching endpoint API key for endpoint: {endpoint_name}")
            response = requests.get(
                f"{vast_console_url}?autoscaler_instance={instance}", headers=headers
            )

            if response.status_code != 200:
                error_msg = f"Failed to fetch endpoint API key: {response.status_code} - {response.text}"
                log.debug(error_msg)
                return None

            try:
                data = response.json()
            except requests.exceptions.JSONDecodeError as e:
                log.debug(f"Failed to parse JSON response: {e}")
                return None

            result = data.get("results", [])

            endpoint: Optional[Dict[str, Any]] = next(
                (item for item in result if item["endpoint_name"] == endpoint_name),
                None,
            )
            if not endpoint:
                error_msg = f"Endpoint '{endpoint_name}' not found."
                log.debug(error_msg)
                return None

            endpoint_api_key = endpoint.get("api_key")
            if not endpoint_api_key:
                error_msg = f"API key for endpoint '{endpoint_name}' not found."
                log.debug(error_msg)
                return None

            log.debug(f"Successfully retrieved API key for endpoint: {endpoint_name}")
            return endpoint_api_key

        except requests.exceptions.RequestException as e:
            error_msg = f"Request error while fetching endpoint API key: {e}"
            log.debug(error_msg)
            return None
        except Exception as e:
            error_msg = f"Unexpected error while fetching endpoint API key: {e}"
            log.debug(error_msg)
            return None