use crate::{ApiResponse, error::ApiError};
use actix_web::{HttpResponse, Result, web};
use service::AppService;
use service::agent::model_version::{CreateModelVersionRequest, UpdateModelVersionRequest};
use session::Session;
use uuid::Uuid;

#[derive(serde::Deserialize, utoipa::IntoParams)]
pub struct ListQuery {
    pub model_id: Option<String>,
}

#[utoipa::path(
    get,
    path = "/api/agents/versions",
    params(ListQuery),
    responses(
        (status = 200, body = Vec<service::agent::model_version::ModelVersionResponse>),
        (status = 401, description = "Unauthorized"),
    ),
    tag = "Agent"
)]
pub async fn model_version_list(
    service: web::Data<AppService>,
    session: Session,
    query: web::Query<ListQuery>,
) -> Result<HttpResponse, ApiError> {
    let model_id =
        if let Some(ref s) = query.model_id {
            Some(Uuid::parse_str(s).map_err(|_| {
                service::error::AppError::BadRequest("Invalid model UUID".to_string())
            })?)
        } else {
            None
        };
    let resp = service.agent_model_version_list(model_id, &session).await?;
    Ok(ApiResponse::ok(resp).to_response())
}

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

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

#[utoipa::path(
    patch,
    path = "/api/agents/versions/{id}",
    params(("id" = String, Path)),
    request_body = UpdateModelVersionRequest,
    responses(
        (status = 200, body = service::agent::model_version::ModelVersionResponse),
        (status = 401),
        (status = 403),
        (status = 404),
    ),
    tag = "Agent"
)]
pub async fn model_version_update(
    service: web::Data<AppService>,
    session: Session,
    path: web::Path<String>,
    body: web::Json<UpdateModelVersionRequest>,
) -> Result<HttpResponse, ApiError> {
    let id = Uuid::parse_str(&path.into_inner())
        .map_err(|_| service::error::AppError::BadRequest("Invalid UUID".to_string()))?;
    let resp = service
        .agent_model_version_update(id, body.into_inner(), &session)
        .await?;
    Ok(ApiResponse::ok(resp).to_response())
}

#[utoipa::path(
    delete,
    path = "/api/agents/versions/{id}",
    params(("id" = String, Path)),
    responses(
        (status = 200),
        (status = 401),
        (status = 403),
        (status = 404),
    ),
    tag = "Agent"
)]
pub async fn model_version_delete(
    service: web::Data<AppService>,
    session: Session,
    path: web::Path<String>,
) -> Result<HttpResponse, ApiError> {
    let id = Uuid::parse_str(&path.into_inner())
        .map_err(|_| service::error::AppError::BadRequest("Invalid UUID".to_string()))?;
    service.agent_model_version_delete(id, &session).await?;
    Ok(HttpResponse::Ok().json(serde_json::json!({ "success": true })))
}
