use crate::http::context::{AppContext, OpenApiResult, bad_request};
use crate::http::payloads::ModelPayload;
use crate::io::fs;
use crate::llm::chat;
use poem_openapi::{param::Query, payload::Json};

pub async fn get_model_preference(ctx: &AppContext) -> OpenApiResult<Json<Option<String>>> {
    let result = fs::get_model_preference(ctx.store.as_ref()).map_err(bad_request)?;
    Ok(Json(result))
}

pub async fn set_model_preference(
    payload: Json<ModelPayload>,
    ctx: &AppContext,
) -> OpenApiResult<Json<bool>> {
    fs::set_model_preference(payload.0.model, ctx.store.as_ref()).map_err(bad_request)?;
    Ok(Json(true))
}

pub async fn get_ollama_models(
    base_url: Query<Option<String>>,
) -> OpenApiResult<Json<Vec<String>>> {
    let models = chat::get_ollama_models(base_url.0)
        .await
        .map_err(bad_request)?;
    Ok(Json(models))
}