systemprompt-agent 0.2.2

use axum::Json;
use axum::extract::State;
use axum::http::StatusCode;
use axum::response::IntoResponse;
use serde_json::json;
use std::sync::Arc;
use systemprompt_models::Config;

use super::state::AgentHandlerState;
use crate::services::registry::AgentRegistry;

pub async fn handle_agent_card(State(state): State<Arc<AgentHandlerState>>) -> impl IntoResponse {
    let config = state.config.read().await;
    let agent_name = config.name.clone();
    drop(config);

    tracing::info!(agent_name = %agent_name, "Fetching agent card");

    let base_url = Config::get().map_or_else(
        |_| "http://localhost:3000".to_string(),
        |c| c.api_external_url.clone(),
    );

    match AgentRegistry::new() {
        Ok(registry) => match registry.get_agent(&agent_name).await {
            Ok(agent_config) => {
                match registry
                    .to_agent_card(&agent_config.name, &base_url, vec![], None)
                    .await
                {
                    Ok(agent_card) => (StatusCode::OK, Json(agent_card)).into_response(),
                    Err(e) => {
                        tracing::error!(error = %e, "Failed to build agent card");
                        let error_response = json!({
                            "error": "Internal server error",
                            "message": "Failed to build agent card"
                        });
                        (StatusCode::INTERNAL_SERVER_ERROR, Json(error_response)).into_response()
                    },
                }
            },
            Err(_e) => {
                tracing::error!(agent_name = %agent_name, "Agent card not found");
                let error_response = json!({
                    "error": "Agent card not found",
                    "message": format!("No agent card available for agent: {agent_name}")
                });
                (StatusCode::NOT_FOUND, Json(error_response)).into_response()
            },
        },
        Err(e) => {
            tracing::error!(error = %e, "Failed to initialize registry");
            let error_response = json!({
                "error": "Internal server error",
                "message": "Failed to initialize agent registry"
            });
            (StatusCode::INTERNAL_SERVER_ERROR, Json(error_response)).into_response()
        },
    }
}