from typing import Dict, Any

from . import Provider
from .. import config


class ZenProvider(Provider):
    name = "zen"
    base_url = "https://opencode.ai/zen/v1"

    def is_available(self) -> bool:
        return True

    def is_free_model(self, model: Dict[str, Any]) -> bool:
        model_id = model.get("id", "")
        if "-free" in model_id:
            return True
        if model_id == "big-pickle":
            return True
        pricing = model.get("pricing")
        if pricing:
            prompt_price = str(pricing.get("prompt", "0"))
            completion_price = str(pricing.get("completion", "0"))
            request_price = str(pricing.get("request", "0"))
            return prompt_price == "0" and completion_price == "0" and request_price == "0"
        return False

    def get_chat_url(self) -> str:
        return f"{self.base_url}/chat/completions"

    def get_headers(self, model_id: str) -> Dict[str, str]:
        headers = {"Content-Type": "application/json"}
        api_key = config.get_zen_api_key()
        if api_key:
            headers["Authorization"] = f"Bearer {api_key}"
        return headers