use crate::{LanguageModelProviderId, LanguageModelProviderName};

pub const ANTHROPIC_PROVIDER_ID: LanguageModelProviderId =
    LanguageModelProviderId::new("anthropic");
pub const ANTHROPIC_PROVIDER_NAME: LanguageModelProviderName =
    LanguageModelProviderName::new("Anthropic");

pub const OPEN_AI_PROVIDER_ID: LanguageModelProviderId = LanguageModelProviderId::new("openai");
pub const OPEN_AI_PROVIDER_NAME: LanguageModelProviderName =
    LanguageModelProviderName::new("OpenAI");

pub const GOOGLE_PROVIDER_ID: LanguageModelProviderId = LanguageModelProviderId::new("google");
pub const GOOGLE_PROVIDER_NAME: LanguageModelProviderName =
    LanguageModelProviderName::new("Google AI");

pub const X_AI_PROVIDER_ID: LanguageModelProviderId = LanguageModelProviderId::new("x_ai");
pub const X_AI_PROVIDER_NAME: LanguageModelProviderName = LanguageModelProviderName::new("xAI");

pub const ZED_CLOUD_PROVIDER_ID: LanguageModelProviderId = LanguageModelProviderId::new("zed.dev");
pub const ZED_CLOUD_PROVIDER_NAME: LanguageModelProviderName =
    LanguageModelProviderName::new("Zed");

pub fn provider_name_for_id(provider_id: &LanguageModelProviderId) -> LanguageModelProviderName {
    if provider_id == &OPEN_AI_PROVIDER_ID {
        OPEN_AI_PROVIDER_NAME
    } else if provider_id == &ANTHROPIC_PROVIDER_ID {
        ANTHROPIC_PROVIDER_NAME
    } else if provider_id == &GOOGLE_PROVIDER_ID {
        GOOGLE_PROVIDER_NAME
    } else if provider_id == &X_AI_PROVIDER_ID {
        X_AI_PROVIDER_NAME
    } else if provider_id == &ZED_CLOUD_PROVIDER_ID {
        ZED_CLOUD_PROVIDER_NAME
    } else {
        LanguageModelProviderName(provider_id.0.clone())
    }
}