use derive_builder::Builder;
use http::{HeaderMap, HeaderName, HeaderValue};
use crate::api::rest_endpoint_prelude::*;
use serde::Deserialize;
use serde::Serialize;
use serde_json::Value;
use std::borrow::Cow;
use std::collections::BTreeMap;
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct AllowedAddressPairs<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) ip_address: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) max_address: Option<Cow<'a, str>>,
}
#[derive(Debug, Deserialize, Clone, Serialize)]
pub enum BindingVnicType {
#[serde(rename = "accelerator-direct")]
AcceleratorDirect,
#[serde(rename = "accelerator-direct-physical")]
AcceleratorDirectPhysical,
#[serde(rename = "baremetal")]
Baremetal,
#[serde(rename = "direct")]
Direct,
#[serde(rename = "direct-physical")]
DirectPhysical,
#[serde(rename = "macvtap")]
Macvtap,
#[serde(rename = "normal")]
Normal,
#[serde(rename = "remote-managed")]
RemoteManaged,
#[serde(rename = "smart-nic")]
SmartNic,
#[serde(rename = "vdpa")]
Vdpa,
#[serde(rename = "virtio-forwarder")]
VirtioForwarder,
}
#[derive(Debug, Deserialize, Clone, Serialize)]
pub enum DataPlaneStatus {
#[serde(rename = "ACTIVE")]
Active,
#[serde(rename = "DOWN")]
Down,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct FixedIps<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) ip_address: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) subnet_id: Option<Cow<'a, str>>,
}
#[derive(Debug, Deserialize, Clone, Serialize)]
pub enum NumaAffinityPolicy {
#[serde(rename = "legacy")]
Legacy,
#[serde(rename = "preferred")]
Preferred,
#[serde(rename = "required")]
Required,
#[serde(rename = "socket")]
Socket,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Port<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) admin_state_up: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) allowed_address_pairs: Option<Vec<AllowedAddressPairs<'a>>>,
#[serde(rename = "binding:host_id", skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) binding_host_id: Option<Cow<'a, str>>,
#[serde(rename = "binding:profile", skip_serializing_if = "Option::is_none")]
#[builder(default, private, setter(into, name = "_binding_profile"))]
pub(crate) binding_profile: Option<Option<BTreeMap<Cow<'a, str>, Value>>>,
#[serde(rename = "binding:vnic_type", skip_serializing_if = "Option::is_none")]
#[builder(default)]
pub(crate) binding_vnic_type: Option<BindingVnicType>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default)]
pub(crate) data_plane_status: Option<DataPlaneStatus>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) description: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) device_id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) device_owner: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) dns_domain: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) dns_name: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) extra_dhcp_opts: Option<Vec<BTreeMap<Cow<'a, str>, Value>>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) fixed_ips: Option<Vec<FixedIps<'a>>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, private, setter(into, name = "_hints"))]
pub(crate) hints: Option<Option<BTreeMap<Cow<'a, str>, Value>>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) mac_address: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) name: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default)]
pub(crate) numa_affinity_policy: Option<NumaAffinityPolicy>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) port_security_enabled: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) qos_policy_id: Option<Option<Cow<'a, str>>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) security_groups: Option<Vec<Cow<'a, str>>>,
}
impl<'a> PortBuilder<'a> {
pub fn binding_profile<I, K, V>(&mut self, iter: I) -> &mut Self
where
I: Iterator<Item = (K, V)>,
K: Into<Cow<'a, str>>,
V: Into<Value>,
{
self.binding_profile
.get_or_insert(None)
.get_or_insert(None)
.get_or_insert_with(BTreeMap::new)
.extend(iter.map(|(k, v)| (k.into(), v.into())));
self
}
pub fn hints<I, K, V>(&mut self, iter: I) -> &mut Self
where
I: Iterator<Item = (K, V)>,
K: Into<Cow<'a, str>>,
V: Into<Value>,
{
self.hints
.get_or_insert(None)
.get_or_insert(None)
.get_or_insert_with(BTreeMap::new)
.extend(iter.map(|(k, v)| (k.into(), v.into())));
self
}
}
#[derive(Builder, Debug, Clone)]
#[builder(setter(strip_option))]
pub struct Request<'a> {
#[builder(setter(into))]
pub(crate) port: Port<'a>,
#[builder(default, setter(into))]
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 header<K, V>(&mut self, header_name: K, header_value: V) -> &mut Self
where
K: Into<HeaderName>,
V: Into<HeaderValue>,
{
self._headers
.get_or_insert(None)
.get_or_insert_with(HeaderMap::new)
.insert(header_name.into(), header_value.into());
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 RestEndpoint for Request<'_> {
fn method(&self) -> http::Method {
http::Method::PUT
}
fn endpoint(&self) -> Cow<'static, str> {
format!("ports/{id}", id = self.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("port", serde_json::to_value(&self.port)?);
params.into_body()
}
fn service_type(&self) -> ServiceType {
ServiceType::Network
}
fn response_key(&self) -> Option<Cow<'static, str>> {
Some("port".into())
}
fn request_headers(&self) -> Option<&HeaderMap> {
self._headers.as_ref()
}
fn api_version(&self) -> Option<ApiVersion> {
Some(ApiVersion::new(2, 0))
}
}
#[cfg(test)]
mod tests {
use super::*;
#[cfg(feature = "sync")]
use crate::api::Query;
use crate::test::client::FakeOpenStackClient;
use crate::types::ServiceType;
use http::{HeaderName, HeaderValue};
use httpmock::MockServer;
use serde_json::json;
#[test]
fn test_service_type() {
assert_eq!(
Request::builder()
.port(PortBuilder::default().build().unwrap())
.build()
.unwrap()
.service_type(),
ServiceType::Network
);
}
#[test]
fn test_response_key() {
assert_eq!(
Request::builder()
.port(PortBuilder::default().build().unwrap())
.build()
.unwrap()
.response_key()
.unwrap(),
"port"
);
}
#[cfg(feature = "sync")]
#[test]
fn endpoint() {
let server = MockServer::start();
let client = FakeOpenStackClient::new(server.base_url());
let mock = server.mock(|when, then| {
when.method(httpmock::Method::PUT)
.path(format!("/ports/{id}", id = "id",));
then.status(200)
.header("content-type", "application/json")
.json_body(json!({ "port": {} }));
});
let endpoint = Request::builder()
.id("id")
.port(PortBuilder::default().build().unwrap())
.build()
.unwrap();
let _: serde_json::Value = endpoint.query(&client).unwrap();
mock.assert();
}
#[cfg(feature = "sync")]
#[test]
fn endpoint_headers() {
let server = MockServer::start();
let client = FakeOpenStackClient::new(server.base_url());
let mock = server.mock(|when, then| {
when.method(httpmock::Method::PUT)
.path(format!("/ports/{id}", id = "id",))
.header("foo", "bar")
.header("not_foo", "not_bar");
then.status(200)
.header("content-type", "application/json")
.json_body(json!({ "port": {} }));
});
let endpoint = Request::builder()
.id("id")
.port(PortBuilder::default().build().unwrap())
.headers(
[(
Some(HeaderName::from_static("foo")),
HeaderValue::from_static("bar"),
)]
.into_iter(),
)
.header(
HeaderName::from_static("not_foo"),
HeaderValue::from_static("not_bar"),
)
.build()
.unwrap();
let _: serde_json::Value = endpoint.query(&client).unwrap();
mock.assert();
}
}