reasoninglayer 0.1.2

//! Bayesian preference learning + UI preference CRUD.

use crate::error::Error;
use crate::http::HttpClient;
use crate::types::common::RequestOptions;
use crate::types::preferences::{
    ListPreferencesResponse, PredictPreferencesRequest, PredictPreferencesResponse, PreferenceDto,
    RecordSelectionRequest, RecordSelectionResponse, SetPreferenceRequest,
};

#[derive(Debug, Clone)]
pub struct PreferencesClient {
    http: HttpClient,
}

impl PreferencesClient {
    pub(crate) fn new(http: HttpClient) -> Self {
        Self { http }
    }

    /// Record a user's term selection.
    pub async fn record_selection(
        &self,
        request: RecordSelectionRequest,
        options: Option<&RequestOptions>,
    ) -> Result<RecordSelectionResponse, Error> {
        self.http
            .post("/preferences/selection", &request, options)
            .await
    }

    /// Predict user preferences for candidate terms.
    pub async fn predict(
        &self,
        request: PredictPreferencesRequest,
        options: Option<&RequestOptions>,
    ) -> Result<PredictPreferencesResponse, Error> {
        self.http
            .post("/preferences/predict", &request, options)
            .await
    }

    /// List all learned UI preferences.
    pub async fn list_preferences(
        &self,
        options: Option<&RequestOptions>,
    ) -> Result<ListPreferencesResponse, Error> {
        self.http.get("/preferences", None, options).await
    }

    /// Set or override a user preference by key.
    pub async fn set_preference(
        &self,
        key: &str,
        request: SetPreferenceRequest,
        options: Option<&RequestOptions>,
    ) -> Result<PreferenceDto, Error> {
        let path = format!("/preferences/{}", encode(key));
        self.http.put(&path, &request, options).await
    }

    /// Delete a user preference.
    pub async fn delete_preference(
        &self,
        key: &str,
        options: Option<&RequestOptions>,
    ) -> Result<(), Error> {
        let path = format!("/preferences/{}", encode(key));
        let _: serde_json::Value = self.http.delete(&path, None, options).await?;
        Ok(())
    }
}

fn encode(s: &str) -> String {
    url::form_urlencoded::byte_serialize(s.as_bytes()).collect()
}