systemprompt-client 0.1.22

use crate::error::{ClientError, ClientResult};
use reqwest::{Client, Response};
use serde::de::DeserializeOwned;
use systemprompt_identifiers::JwtToken;

async fn extract_error(response: Response) -> ClientError {
    let status = response.status().as_u16();
    let body = response.text().await.unwrap_or_else(|e| {
        tracing::warn!(error = %e, status = %status, "Failed to read error response body");
        format!("(body unreadable: {})", e)
    });
    ClientError::from_response(status, body)
}

fn apply_auth(
    request: reqwest::RequestBuilder,
    token: Option<&JwtToken>,
) -> reqwest::RequestBuilder {
    match token {
        Some(t) => request.header("Authorization", format!("Bearer {}", t.as_str())),
        None => request,
    }
}

pub async fn get<T: DeserializeOwned>(
    client: &Client,
    url: &str,
    token: Option<&JwtToken>,
) -> ClientResult<T> {
    let request = apply_auth(client.get(url), token);
    let response = request.send().await?;

    if !response.status().is_success() {
        return Err(extract_error(response).await);
    }

    Ok(response.json().await?)
}

pub async fn post<T: DeserializeOwned, B: serde::Serialize + Sync>(
    client: &Client,
    url: &str,
    body: &B,
    token: Option<&JwtToken>,
) -> ClientResult<T> {
    let request = apply_auth(client.post(url), token).json(body);
    let response = request.send().await?;

    if !response.status().is_success() {
        return Err(extract_error(response).await);
    }

    Ok(response.json().await?)
}

pub async fn put<B: serde::Serialize + Sync>(
    client: &Client,
    url: &str,
    body: &B,
    token: Option<&JwtToken>,
) -> ClientResult<()> {
    let request = apply_auth(client.put(url), token).json(body);
    let response = request.send().await?;

    if !response.status().is_success() {
        return Err(extract_error(response).await);
    }

    Ok(())
}

pub async fn delete(client: &Client, url: &str, token: Option<&JwtToken>) -> ClientResult<()> {
    let request = apply_auth(client.delete(url), token);
    let response = request.send().await?;

    if !response.status().is_success() {
        return Err(extract_error(response).await);
    }

    Ok(())
}