use axum::{
    Json,
    extract::{Extension, State},
    http::StatusCode,
    response::{IntoResponse, Response},
};
use serde_json::{Value, json};

use crate::{
    api_response::json_success_body, http_error::AppError, request_context::RequestContext,
    state::AppState,
};

pub async fn health_check(Extension(request_context): Extension<RequestContext>) -> Json<Value> {
    json_success_body(
        Some(&request_context),
        json!({
            "ok": true,
            "service": "genarrative-api-server",
        }),
    )
}

pub async fn readiness_check(
    State(state): State<AppState>,
    Extension(request_context): Extension<RequestContext>,
) -> Response {
    if state.is_ready() {
        return json_success_body(
            Some(&request_context),
            json!({
                "ok": true,
                "ready": true,
                "service": "genarrative-api-server",
            }),
        )
        .into_response();
    }

    AppError::from_status(StatusCode::SERVICE_UNAVAILABLE)
        .with_message("api-server 正在退出，不再接收新流量")
        .with_details(json!({
            "reason": "api_server_draining",
            "ready": false,
        }))
        .into_response_with_context(Some(&request_context))
}