use derive_builder::Builder;
use http::{HeaderMap, HeaderName, HeaderValue};
use crate::api::rest_endpoint_prelude::*;
use serde_json::Value;
use std::borrow::Cow;
use std::collections::BTreeMap;
#[derive(Builder, Debug, Clone)]
#[builder(setter(strip_option))]
pub struct Request<'a> {
#[builder(private, setter(name = "_config"))]
pub(crate) config: BTreeMap<Cow<'a, str>, BTreeMap<Cow<'a, str>, Value>>,
#[builder(default, setter(into))]
domain_id: Cow<'a, str>,
#[builder(setter(name = "_headers"), default, private)]
_headers: Option<HeaderMap>,
}
impl<'a> Request<'a> {
pub fn builder() -> RequestBuilder<'a> {
RequestBuilder::default()
}
}
impl<'a> RequestBuilder<'a> {
pub fn config<I, K, V, K1, V1>(&mut self, iter: I) -> &mut Self
where
I: Iterator<Item = (K, V)>,
K: Into<Cow<'a, str>>,
V: Iterator<Item = (K1, V1)>,
K1: Into<Cow<'a, str>>,
V1: Into<Value>,
{
self.config
.get_or_insert_with(BTreeMap::new)
.extend(iter.map(|(k, v)| {
(
k.into(),
BTreeMap::from_iter(v.into_iter().map(|(k1, v1)| (k1.into(), v1.into()))),
)
}));
self
}
pub fn header(&mut self, header_name: &'static str, header_value: &'static str) -> &mut Self
where {
self._headers
.get_or_insert(None)
.get_or_insert_with(HeaderMap::new)
.insert(header_name, HeaderValue::from_static(header_value));
self
}
pub fn headers<I, T>(&mut self, iter: I) -> &mut Self
where
I: Iterator<Item = T>,
T: Into<(Option<HeaderName>, HeaderValue)>,
{
self._headers
.get_or_insert(None)
.get_or_insert_with(HeaderMap::new)
.extend(iter.map(Into::into));
self
}
}
impl<'a> RestEndpoint for Request<'a> {
fn method(&self) -> http::Method {
http::Method::PATCH
}
fn endpoint(&self) -> Cow<'static, str> {
format!(
"domains/{domain_id}/config",
domain_id = self.domain_id.as_ref(),
)
.into()
}
fn parameters(&self) -> QueryParams {
QueryParams::default()
}
fn body(&self) -> Result<Option<(&'static str, Vec<u8>)>, BodyError> {
let mut params = JsonBodyParams::default();
params.push("config", serde_json::to_value(&self.config)?);
params.into_body()
}
fn service_type(&self) -> ServiceType {
ServiceType::Identity
}
fn response_key(&self) -> Option<Cow<'static, str>> {
Some("config".into())
}
fn request_headers(&self) -> Option<&HeaderMap> {
self._headers.as_ref()
}
fn api_version(&self) -> Option<ApiVersion> {
Some(ApiVersion::new(3, 0))
}
}
#[cfg(test)]
mod tests {
#![allow(unused_imports)]
use super::*;
#[cfg(feature = "sync")]
use crate::api::Query;
#[cfg(feature = "sync")]
use crate::test::client::MockServerClient;
use crate::types::ServiceType;
use http::{HeaderName, HeaderValue};
use serde_json::json;
#[test]
fn test_service_type() {
assert_eq!(
Request::builder()
.config(
BTreeMap::<String, BTreeMap<String, Value>>::new()
.into_iter()
.map(|(k, v)| (k, v.into_iter()))
)
.build()
.unwrap()
.service_type(),
ServiceType::Identity
);
}
#[test]
fn test_response_key() {
assert_eq!(
Request::builder()
.config(
BTreeMap::<String, BTreeMap<String, Value>>::new()
.into_iter()
.map(|(k, v)| (k, v.into_iter()))
)
.build()
.unwrap()
.response_key()
.unwrap(),
"config"
);
}
#[cfg(feature = "sync")]
#[test]
fn endpoint() {
let client = MockServerClient::new();
let mock = client.server.mock(|when, then| {
when.method(httpmock::Method::PATCH).path(format!(
"/domains/{domain_id}/config",
domain_id = "domain_id",
));
then.status(200)
.header("content-type", "application/json")
.json_body(json!({ "config": {} }));
});
let endpoint = Request::builder()
.domain_id("domain_id")
.config(
BTreeMap::<String, BTreeMap<String, Value>>::new()
.into_iter()
.map(|(k, v)| (k, v.into_iter())),
)
.build()
.unwrap();
let _: serde_json::Value = endpoint.query(&client).unwrap();
mock.assert();
}
#[cfg(feature = "sync")]
#[test]
fn endpoint_headers() {
let client = MockServerClient::new();
let mock = client.server.mock(|when, then| {
when.method(httpmock::Method::PATCH)
.path(format!(
"/domains/{domain_id}/config",
domain_id = "domain_id",
))
.header("foo", "bar")
.header("not_foo", "not_bar");
then.status(200)
.header("content-type", "application/json")
.json_body(json!({ "config": {} }));
});
let endpoint = Request::builder()
.domain_id("domain_id")
.config(
BTreeMap::<String, BTreeMap<String, Value>>::new()
.into_iter()
.map(|(k, v)| (k, v.into_iter())),
)
.headers(
[(
Some(HeaderName::from_static("foo")),
HeaderValue::from_static("bar"),
)]
.into_iter(),
)
.header("not_foo", "not_bar")
.build()
.unwrap();
let _: serde_json::Value = endpoint.query(&client).unwrap();
mock.assert();
}
}