harn-hostlib 0.8.21

//! Helpers for assembling [`VmValue::Dict`] response bodies that match the
//! `schemas/tools/<method>.response.json` contracts.

use std::collections::BTreeMap;
use std::rc::Rc;

use harn_vm::VmValue;

/// Type-erased builder for the dict that becomes a tool's response.
#[derive(Default)]
pub(crate) struct ResponseBuilder {
    inner: BTreeMap<String, VmValue>,
}

impl ResponseBuilder {
    pub(crate) fn new() -> Self {
        Self {
            inner: BTreeMap::new(),
        }
    }

    pub(crate) fn str(mut self, key: &str, value: impl Into<String>) -> Self {
        self.inner
            .insert(key.to_string(), VmValue::String(Rc::from(value.into())));
        self
    }

    pub(crate) fn int(mut self, key: &str, value: i64) -> Self {
        self.inner.insert(key.to_string(), VmValue::Int(value));
        self
    }

    pub(crate) fn bool(mut self, key: &str, value: bool) -> Self {
        self.inner.insert(key.to_string(), VmValue::Bool(value));
        self
    }

    pub(crate) fn opt_str(mut self, key: &str, value: Option<impl Into<String>>) -> Self {
        match value {
            Some(v) => {
                self.inner
                    .insert(key.to_string(), VmValue::String(Rc::from(v.into())));
            }
            None => {
                self.inner.insert(key.to_string(), VmValue::Nil);
            }
        }
        self
    }

    pub(crate) fn nil(mut self, key: &str) -> Self {
        self.inner.insert(key.to_string(), VmValue::Nil);
        self
    }

    pub(crate) fn dict(mut self, key: &str, value: BTreeMap<String, VmValue>) -> Self {
        self.inner
            .insert(key.to_string(), VmValue::Dict(Rc::new(value)));
        self
    }

    pub(crate) fn list(mut self, key: &str, value: Vec<VmValue>) -> Self {
        self.inner
            .insert(key.to_string(), VmValue::List(Rc::new(value)));
        self
    }

    pub(crate) fn build(self) -> VmValue {
        VmValue::Dict(Rc::new(self.inner))
    }
}