use derive_builder::Builder;
use http::{HeaderMap, HeaderName, HeaderValue};
use crate::api::rest_endpoint_prelude::*;
use crate::api::common::serialize_sensitive_optional_string;
use crate::api::common::serialize_sensitive_string;
use secrecy::SecretString;
use serde::Deserialize;
use serde::Serialize;
use std::borrow::Cow;
#[derive(Debug, Deserialize, Clone, Serialize)]
pub enum Methods {
#[serde(rename = "application_credential")]
ApplicationCredential,
#[serde(rename = "password")]
Password,
#[serde(rename = "token")]
Token,
#[serde(rename = "totp")]
Totp,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Domain<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) name: Option<Cow<'a, str>>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct PasswordUser<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) domain: Option<Domain<'a>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) name: Option<Cow<'a, str>>,
#[serde(
serialize_with = "serialize_sensitive_optional_string",
skip_serializing_if = "Option::is_none"
)]
#[builder(default, setter(into))]
pub(crate) password: Option<SecretString>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Password<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) user: Option<PasswordUser<'a>>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Token {
#[serde(serialize_with = "serialize_sensitive_string")]
#[builder(setter(into))]
pub(crate) id: SecretString,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct TotpUser<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) domain: Option<Domain<'a>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) name: Option<Cow<'a, str>>,
#[serde(serialize_with = "serialize_sensitive_string")]
#[builder(setter(into))]
pub(crate) passcode: SecretString,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Totp<'a> {
#[serde()]
#[builder(setter(into))]
pub(crate) user: TotpUser<'a>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct ApplicationCredentialUser<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) domain: Option<Domain<'a>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) name: Option<Cow<'a, str>>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct ApplicationCredential<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) name: Option<Cow<'a, str>>,
#[serde(serialize_with = "serialize_sensitive_string")]
#[builder(setter(into))]
pub(crate) secret: SecretString,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) user: Option<ApplicationCredentialUser<'a>>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Identity<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) application_credential: Option<ApplicationCredential<'a>>,
#[serde()]
#[builder(setter(into))]
pub(crate) methods: Vec<Methods>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) password: Option<Password<'a>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) token: Option<Token>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) totp: Option<Totp<'a>>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Project<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) domain: Option<Domain<'a>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) name: Option<Cow<'a, str>>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct OsTrustTrust<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) id: Option<Cow<'a, str>>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct System {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) all: Option<bool>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Scope<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) domain: Option<Domain<'a>>,
#[serde(rename = "OS-TRUST:trust", skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) os_trust_trust: Option<OsTrustTrust<'a>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) project: Option<Project<'a>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) system: Option<System>,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Auth<'a> {
#[serde()]
#[builder(setter(into))]
pub(crate) identity: Identity<'a>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) scope: Option<Scope<'a>>,
}
#[derive(Builder, Debug, Clone)]
#[builder(setter(strip_option))]
pub struct Request<'a> {
#[builder(setter(into))]
pub(crate) auth: Auth<'a>,
#[builder(setter(name = "_headers"), default, private)]
_headers: Option<HeaderMap>,
}
impl RequestBuilder<'_> {
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::POST
}
fn endpoint(&self) -> Cow<'static, str> {
"auth/tokens".to_string().into()
}
fn parameters(&self) -> QueryParams<'_> {
QueryParams::default()
}
fn body(&self) -> Result<Option<(&'static str, Vec<u8>)>, BodyError> {
let mut params = JsonBodyParams::default();
params.push("auth", serde_json::to_value(&self.auth)?);
params.into_body()
}
fn service_type(&self) -> ServiceType {
ServiceType::Identity
}
fn response_key(&self) -> Option<Cow<'static, str>> {
Some("token".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 {
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!(
RequestBuilder::default()
.auth(
AuthBuilder::default()
.identity(
IdentityBuilder::default()
.methods(Vec::from([Methods::ApplicationCredential]))
.build()
.unwrap()
)
.build()
.unwrap()
)
.build()
.unwrap()
.service_type(),
ServiceType::Identity
);
}
#[test]
fn test_response_key() {
assert_eq!(
RequestBuilder::default()
.auth(
AuthBuilder::default()
.identity(
IdentityBuilder::default()
.methods(Vec::from([Methods::ApplicationCredential]))
.build()
.unwrap()
)
.build()
.unwrap()
)
.build()
.unwrap()
.response_key()
.unwrap(),
"token"
);
}
#[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::POST)
.path("/auth/tokens".to_string());
then.status(200)
.header("content-type", "application/json")
.json_body(json!({ "token": {} }));
});
let endpoint = RequestBuilder::default()
.auth(
AuthBuilder::default()
.identity(
IdentityBuilder::default()
.methods(Vec::from([Methods::ApplicationCredential]))
.build()
.unwrap(),
)
.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::POST)
.path("/auth/tokens".to_string())
.header("foo", "bar");
then.status(200)
.header("content-type", "application/json")
.json_body(json!({ "token": {} }));
});
let endpoint = RequestBuilder::default()
.auth(
AuthBuilder::default()
.identity(
IdentityBuilder::default()
.methods(Vec::from([Methods::ApplicationCredential]))
.build()
.unwrap(),
)
.build()
.unwrap(),
)
.headers(
[(
Some(HeaderName::from_static("foo")),
HeaderValue::from_static("bar"),
)]
.into_iter(),
)
.build()
.unwrap();
let _: serde_json::Value = endpoint.query(&client).unwrap();
mock.assert();
}
}