#![allow(unused_mut)]
#![allow(unused_variables)]
#![allow(unused_imports)]
#![allow(clippy::redundant_clone)]
pub mod models;
#[derive(Clone)]
pub struct Client {
endpoint: String,
credential: std::sync::Arc<dyn azure_core::auth::TokenCredential>,
scopes: Vec<String>,
pipeline: azure_core::Pipeline,
}
#[derive(Clone)]
pub struct ClientBuilder {
credential: std::sync::Arc<dyn azure_core::auth::TokenCredential>,
endpoint: Option<String>,
scopes: Option<Vec<String>>,
options: azure_core::ClientOptions,
}
pub const DEFAULT_ENDPOINT: &str = "https://169.254.169.254/metadata";
impl ClientBuilder {
#[doc = "Create a new instance of `ClientBuilder`."]
#[must_use]
pub fn new(credential: std::sync::Arc<dyn azure_core::auth::TokenCredential>) -> Self {
Self {
credential,
endpoint: None,
scopes: None,
options: azure_core::ClientOptions::default(),
}
}
#[doc = "Set the endpoint."]
#[must_use]
pub fn endpoint(mut self, endpoint: impl Into<String>) -> Self {
self.endpoint = Some(endpoint.into());
self
}
#[doc = "Set the scopes."]
#[must_use]
pub fn scopes(mut self, scopes: &[&str]) -> Self {
self.scopes = Some(scopes.iter().map(|scope| (*scope).to_owned()).collect());
self
}
#[doc = "Set the retry options."]
#[must_use]
pub fn retry(mut self, retry: impl Into<azure_core::RetryOptions>) -> Self {
self.options = self.options.retry(retry);
self
}
#[doc = "Set the transport options."]
#[must_use]
pub fn transport(mut self, transport: impl Into<azure_core::TransportOptions>) -> Self {
self.options = self.options.transport(transport);
self
}
#[doc = "Convert the builder into a `Client` instance."]
#[must_use]
pub fn build(self) -> Client {
let endpoint = self.endpoint.unwrap_or_else(|| DEFAULT_ENDPOINT.to_owned());
let scopes = self.scopes.unwrap_or_else(|| vec![format!("{}/", endpoint)]);
Client::new(endpoint, self.credential, scopes, self.options)
}
}
impl Client {
pub(crate) fn endpoint(&self) -> &str {
self.endpoint.as_str()
}
pub(crate) fn token_credential(&self) -> &dyn azure_core::auth::TokenCredential {
self.credential.as_ref()
}
pub(crate) fn scopes(&self) -> Vec<&str> {
self.scopes.iter().map(String::as_str).collect()
}
pub(crate) async fn send(&self, request: &mut azure_core::Request) -> azure_core::Result<azure_core::Response> {
let mut context = azure_core::Context::default();
self.pipeline.send(&mut context, request).await
}
#[doc = "Create a new `ClientBuilder`."]
#[must_use]
pub fn builder(credential: std::sync::Arc<dyn azure_core::auth::TokenCredential>) -> ClientBuilder {
ClientBuilder::new(credential)
}
#[doc = "Create a new `Client`."]
#[must_use]
pub fn new(
endpoint: impl Into<String>,
credential: std::sync::Arc<dyn azure_core::auth::TokenCredential>,
scopes: Vec<String>,
options: azure_core::ClientOptions,
) -> Self {
let endpoint = endpoint.into();
let pipeline = azure_core::Pipeline::new(
option_env!("CARGO_PKG_NAME"),
option_env!("CARGO_PKG_VERSION"),
options,
Vec::new(),
Vec::new(),
);
Self {
endpoint,
credential,
scopes,
pipeline,
}
}
pub fn attested_client(&self) -> attested::Client {
attested::Client(self.clone())
}
pub fn identity_client(&self) -> identity::Client {
identity::Client(self.clone())
}
pub fn instances_client(&self) -> instances::Client {
instances::Client(self.clone())
}
}
pub mod instances {
use super::models;
pub struct Client(pub(crate) super::Client);
impl Client {
#[doc = "Get Instance Metadata for the Virtual Machine."]
#[doc = ""]
#[doc = "Arguments:"]
#[doc = "* `metadata`: This must be set to 'true'."]
pub fn get_metadata(&self, metadata: impl Into<String>) -> get_metadata::RequestBuilder {
get_metadata::RequestBuilder {
client: self.0.clone(),
metadata: metadata.into(),
}
}
}
pub mod get_metadata {
use super::models;
pub struct Response(azure_core::Response);
impl Response {
pub async fn into_body(self) -> azure_core::Result<models::Instance> {
let bytes = self.0.into_body().collect().await?;
let body: models::Instance = serde_json::from_slice(&bytes)?;
Ok(body)
}
pub fn into_raw_response(self) -> azure_core::Response {
self.0
}
pub fn as_raw_response(&self) -> &azure_core::Response {
&self.0
}
}
impl From<Response> for azure_core::Response {
fn from(rsp: Response) -> Self {
rsp.into_raw_response()
}
}
impl AsRef<azure_core::Response> for Response {
fn as_ref(&self) -> &azure_core::Response {
self.as_raw_response()
}
}
#[derive(Clone)]
pub struct RequestBuilder {
pub(crate) client: super::super::Client,
pub(crate) metadata: String,
}
impl RequestBuilder {
#[doc = "Send the request and returns the response."]
pub fn send(self) -> futures::future::BoxFuture<'static, azure_core::Result<Response>> {
Box::pin({
let this = self.clone();
async move {
let url = azure_core::Url::parse(&format!("{}/instance", this.client.endpoint(),))?;
let mut req = azure_core::Request::new(url, azure_core::Method::Get);
let credential = this.client.token_credential();
let token_response = credential.get_token(&this.client.scopes().join(" ")).await?;
req.insert_header(
azure_core::headers::AUTHORIZATION,
format!("Bearer {}", token_response.token.secret()),
);
req.url_mut()
.query_pairs_mut()
.append_pair(azure_core::query_param::API_VERSION, "2021-11-01");
req.insert_header("metadata", &this.metadata);
let req_body = azure_core::EMPTY_BODY;
req.set_body(req_body);
Ok(Response(this.client.send(&mut req).await?))
}
})
}
#[doc = "Send the request and return the response body."]
pub fn into_future(self) -> futures::future::BoxFuture<'static, azure_core::Result<models::Instance>> {
Box::pin(async move { self.send().await?.into_body().await })
}
}
}
}
pub mod attested {
use super::models;
pub struct Client(pub(crate) super::Client);
impl Client {
#[doc = "Get Attested Data for the Virtual Machine."]
#[doc = ""]
#[doc = "Arguments:"]
#[doc = "* `metadata`: This must be set to 'true'."]
pub fn get_document(&self, metadata: impl Into<String>) -> get_document::RequestBuilder {
get_document::RequestBuilder {
client: self.0.clone(),
metadata: metadata.into(),
nonce: None,
}
}
}
pub mod get_document {
use super::models;
pub struct Response(azure_core::Response);
impl Response {
pub async fn into_body(self) -> azure_core::Result<models::AttestedData> {
let bytes = self.0.into_body().collect().await?;
let body: models::AttestedData = serde_json::from_slice(&bytes)?;
Ok(body)
}
pub fn into_raw_response(self) -> azure_core::Response {
self.0
}
pub fn as_raw_response(&self) -> &azure_core::Response {
&self.0
}
}
impl From<Response> for azure_core::Response {
fn from(rsp: Response) -> Self {
rsp.into_raw_response()
}
}
impl AsRef<azure_core::Response> for Response {
fn as_ref(&self) -> &azure_core::Response {
self.as_raw_response()
}
}
#[derive(Clone)]
pub struct RequestBuilder {
pub(crate) client: super::super::Client,
pub(crate) metadata: String,
pub(crate) nonce: Option<String>,
}
impl RequestBuilder {
#[doc = "This is a string of up to 32 random alphanumeric characters."]
pub fn nonce(mut self, nonce: impl Into<String>) -> Self {
self.nonce = Some(nonce.into());
self
}
#[doc = "Send the request and returns the response."]
pub fn send(self) -> futures::future::BoxFuture<'static, azure_core::Result<Response>> {
Box::pin({
let this = self.clone();
async move {
let url = azure_core::Url::parse(&format!("{}/attested/document", this.client.endpoint(),))?;
let mut req = azure_core::Request::new(url, azure_core::Method::Get);
let credential = this.client.token_credential();
let token_response = credential.get_token(&this.client.scopes().join(" ")).await?;
req.insert_header(
azure_core::headers::AUTHORIZATION,
format!("Bearer {}", token_response.token.secret()),
);
req.url_mut()
.query_pairs_mut()
.append_pair(azure_core::query_param::API_VERSION, "2021-11-01");
if let Some(nonce) = &this.nonce {
req.url_mut().query_pairs_mut().append_pair("nonce", nonce);
}
req.insert_header("metadata", &this.metadata);
let req_body = azure_core::EMPTY_BODY;
req.set_body(req_body);
Ok(Response(this.client.send(&mut req).await?))
}
})
}
#[doc = "Send the request and return the response body."]
pub fn into_future(self) -> futures::future::BoxFuture<'static, azure_core::Result<models::AttestedData>> {
Box::pin(async move { self.send().await?.into_body().await })
}
}
}
}
pub mod identity {
use super::models;
pub struct Client(pub(crate) super::Client);
impl Client {
#[doc = "Get a Token from Azure AD"]
#[doc = ""]
#[doc = "Arguments:"]
#[doc = "* `metadata`: This must be set to 'true'."]
#[doc = "* `resource`: This is the urlencoded identifier URI of the sink resource for the requested Azure AD token. The resulting token contains the corresponding aud for this resource."]
pub fn get_token(&self, metadata: impl Into<String>, resource: impl Into<String>) -> get_token::RequestBuilder {
get_token::RequestBuilder {
client: self.0.clone(),
metadata: metadata.into(),
resource: resource.into(),
client_id: None,
object_id: None,
msi_res_id: None,
authority: None,
bypass_cache: None,
}
}
#[doc = "Get information about AAD Metadata"]
#[doc = ""]
#[doc = "Arguments:"]
#[doc = "* `metadata`: This must be set to 'true'."]
pub fn get_info(&self, metadata: impl Into<String>) -> get_info::RequestBuilder {
get_info::RequestBuilder {
client: self.0.clone(),
metadata: metadata.into(),
}
}
}
pub mod get_token {
use super::models;
pub struct Response(azure_core::Response);
impl Response {
pub async fn into_body(self) -> azure_core::Result<models::IdentityTokenResponse> {
let bytes = self.0.into_body().collect().await?;
let body: models::IdentityTokenResponse = serde_json::from_slice(&bytes)?;
Ok(body)
}
pub fn into_raw_response(self) -> azure_core::Response {
self.0
}
pub fn as_raw_response(&self) -> &azure_core::Response {
&self.0
}
}
impl From<Response> for azure_core::Response {
fn from(rsp: Response) -> Self {
rsp.into_raw_response()
}
}
impl AsRef<azure_core::Response> for Response {
fn as_ref(&self) -> &azure_core::Response {
self.as_raw_response()
}
}
#[derive(Clone)]
pub struct RequestBuilder {
pub(crate) client: super::super::Client,
pub(crate) metadata: String,
pub(crate) resource: String,
pub(crate) client_id: Option<String>,
pub(crate) object_id: Option<String>,
pub(crate) msi_res_id: Option<String>,
pub(crate) authority: Option<String>,
pub(crate) bypass_cache: Option<String>,
}
impl RequestBuilder {
#[doc = "This identifies, by Azure AD client id, a specific explicit identity to use when authenticating to Azure AD. Mutually exclusive with object_id and msi_res_id."]
pub fn client_id(mut self, client_id: impl Into<String>) -> Self {
self.client_id = Some(client_id.into());
self
}
#[doc = "This identifies, by Azure AD object id, a specific explicit identity to use when authenticating to Azure AD. Mutually exclusive with client_id and msi_res_id."]
pub fn object_id(mut self, object_id: impl Into<String>) -> Self {
self.object_id = Some(object_id.into());
self
}
#[doc = "This identifies, by urlencoded ARM resource id, a specific explicit identity to use when authenticating to Azure AD. Mutually exclusive with client_id and object_id."]
pub fn msi_res_id(mut self, msi_res_id: impl Into<String>) -> Self {
self.msi_res_id = Some(msi_res_id.into());
self
}
#[doc = "This indicates the authority to request AAD tokens from. Defaults to the known authority of the identity to be used."]
pub fn authority(mut self, authority: impl Into<String>) -> Self {
self.authority = Some(authority.into());
self
}
#[doc = "If provided, the value must be 'true'. This indicates to the server that the token must be retrieved from Azure AD and cannot be retrieved from an internal cache."]
pub fn bypass_cache(mut self, bypass_cache: impl Into<String>) -> Self {
self.bypass_cache = Some(bypass_cache.into());
self
}
#[doc = "Send the request and returns the response."]
pub fn send(self) -> futures::future::BoxFuture<'static, azure_core::Result<Response>> {
Box::pin({
let this = self.clone();
async move {
let url = azure_core::Url::parse(&format!("{}/identity/oauth2/token", this.client.endpoint(),))?;
let mut req = azure_core::Request::new(url, azure_core::Method::Get);
let credential = this.client.token_credential();
let token_response = credential.get_token(&this.client.scopes().join(" ")).await?;
req.insert_header(
azure_core::headers::AUTHORIZATION,
format!("Bearer {}", token_response.token.secret()),
);
req.url_mut()
.query_pairs_mut()
.append_pair(azure_core::query_param::API_VERSION, "2021-11-01");
req.insert_header("metadata", &this.metadata);
let resource = &this.resource;
req.url_mut().query_pairs_mut().append_pair("resource", resource);
if let Some(client_id) = &this.client_id {
req.url_mut().query_pairs_mut().append_pair("client_id", client_id);
}
if let Some(object_id) = &this.object_id {
req.url_mut().query_pairs_mut().append_pair("object_id", object_id);
}
if let Some(msi_res_id) = &this.msi_res_id {
req.url_mut().query_pairs_mut().append_pair("msi_res_id", msi_res_id);
}
if let Some(authority) = &this.authority {
req.url_mut().query_pairs_mut().append_pair("authority", authority);
}
if let Some(bypass_cache) = &this.bypass_cache {
req.url_mut().query_pairs_mut().append_pair("bypass_cache", bypass_cache);
}
let req_body = azure_core::EMPTY_BODY;
req.set_body(req_body);
Ok(Response(this.client.send(&mut req).await?))
}
})
}
#[doc = "Send the request and return the response body."]
pub fn into_future(self) -> futures::future::BoxFuture<'static, azure_core::Result<models::IdentityTokenResponse>> {
Box::pin(async move { self.send().await?.into_body().await })
}
}
}
pub mod get_info {
use super::models;
pub struct Response(azure_core::Response);
impl Response {
pub async fn into_body(self) -> azure_core::Result<models::IdentityInfoResponse> {
let bytes = self.0.into_body().collect().await?;
let body: models::IdentityInfoResponse = serde_json::from_slice(&bytes)?;
Ok(body)
}
pub fn into_raw_response(self) -> azure_core::Response {
self.0
}
pub fn as_raw_response(&self) -> &azure_core::Response {
&self.0
}
}
impl From<Response> for azure_core::Response {
fn from(rsp: Response) -> Self {
rsp.into_raw_response()
}
}
impl AsRef<azure_core::Response> for Response {
fn as_ref(&self) -> &azure_core::Response {
self.as_raw_response()
}
}
#[derive(Clone)]
pub struct RequestBuilder {
pub(crate) client: super::super::Client,
pub(crate) metadata: String,
}
impl RequestBuilder {
#[doc = "Send the request and returns the response."]
pub fn send(self) -> futures::future::BoxFuture<'static, azure_core::Result<Response>> {
Box::pin({
let this = self.clone();
async move {
let url = azure_core::Url::parse(&format!("{}/identity/info", this.client.endpoint(),))?;
let mut req = azure_core::Request::new(url, azure_core::Method::Get);
let credential = this.client.token_credential();
let token_response = credential.get_token(&this.client.scopes().join(" ")).await?;
req.insert_header(
azure_core::headers::AUTHORIZATION,
format!("Bearer {}", token_response.token.secret()),
);
req.url_mut()
.query_pairs_mut()
.append_pair(azure_core::query_param::API_VERSION, "2021-11-01");
req.insert_header("metadata", &this.metadata);
let req_body = azure_core::EMPTY_BODY;
req.set_body(req_body);
Ok(Response(this.client.send(&mut req).await?))
}
})
}
#[doc = "Send the request and return the response body."]
pub fn into_future(self) -> futures::future::BoxFuture<'static, azure_core::Result<models::IdentityInfoResponse>> {
Box::pin(async move { self.send().await?.into_body().await })
}
}
}
}