use crate::{ApiResponse, error::ApiError};
use actix_web::{HttpResponse, Result, web};
use service::AppService;
use service::agent::model_pricing::{CreateModelPricingRequest, UpdateModelPricingRequest};
use session::Session;
use uuid::Uuid;

#[utoipa::path(
    get,
    path = "/api/agents/versions/{model_version_id}/pricing",
    params(("model_version_id" = String, Path)),
    responses(
        (status = 200, body = Vec<service::agent::model_pricing::ModelPricingResponse>),
        (status = 401, description = "Unauthorized"),
    ),
    tag = "Agent"
)]
pub async fn model_pricing_list(
    service: web::Data<AppService>,
    session: Session,
    path: web::Path<String>,
) -> Result<HttpResponse, ApiError> {
    let model_version_id = Uuid::parse_str(&path.into_inner())
        .map_err(|_| service::error::AppError::BadRequest("Invalid UUID".to_string()))?;
    let resp = service
        .agent_model_pricing_list(model_version_id, &session)
        .await?;
    Ok(ApiResponse::ok(resp).to_response())
}

#[utoipa::path(
    get,
    path = "/api/agents/pricing/{id}",
    params(("id" = i64, Path)),
    responses(
        (status = 200, body = service::agent::model_pricing::ModelPricingResponse),
        (status = 401, description = "Unauthorized"),
        (status = 404),
    ),
    tag = "Agent"
)]
pub async fn model_pricing_get(
    service: web::Data<AppService>,
    session: Session,
    path: web::Path<i64>,
) -> Result<HttpResponse, ApiError> {
    let id = path.into_inner();
    let resp = service.agent_model_pricing_get(id, &session).await?;
    Ok(ApiResponse::ok(resp).to_response())
}

#[utoipa::path(
    post,
    path = "/api/agents/pricing",
    request_body = CreateModelPricingRequest,
    responses(
        (status = 200, body = service::agent::model_pricing::ModelPricingResponse),
        (status = 401),
        (status = 403),
    ),
    tag = "Agent"
)]
pub async fn model_pricing_create(
    service: web::Data<AppService>,
    session: Session,
    body: web::Json<CreateModelPricingRequest>,
) -> Result<HttpResponse, ApiError> {
    let resp = service
        .agent_model_pricing_create(body.into_inner(), &session)
        .await?;
    Ok(ApiResponse::ok(resp).to_response())
}

#[utoipa::path(
    patch,
    path = "/api/agents/pricing/{id}",
    params(("id" = i64, Path)),
    request_body = UpdateModelPricingRequest,
    responses(
        (status = 200, body = service::agent::model_pricing::ModelPricingResponse),
        (status = 401),
        (status = 403),
        (status = 404),
    ),
    tag = "Agent"
)]
pub async fn model_pricing_update(
    service: web::Data<AppService>,
    session: Session,
    path: web::Path<i64>,
    body: web::Json<UpdateModelPricingRequest>,
) -> Result<HttpResponse, ApiError> {
    let id = path.into_inner();
    let resp = service
        .agent_model_pricing_update(id, body.into_inner(), &session)
        .await?;
    Ok(ApiResponse::ok(resp).to_response())
}

#[utoipa::path(
    delete,
    path = "/api/agents/pricing/{id}",
    params(("id" = i64, Path)),
    responses(
        (status = 200),
        (status = 401),
        (status = 403),
        (status = 404),
    ),
    tag = "Agent"
)]
pub async fn model_pricing_delete(
    service: web::Data<AppService>,
    session: Session,
    path: web::Path<i64>,
) -> Result<HttpResponse, ApiError> {
    let id = path.into_inner();
    service.agent_model_pricing_delete(id, &session).await?;
    Ok(HttpResponse::Ok().json(serde_json::json!({ "success": true })))
}
