aws_sdk_imagebuilder/config/
auth.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2/// Auth scheme resolver trait specific to EC2 Image Builder
3pub trait ResolveAuthScheme: ::std::marker::Send + ::std::marker::Sync + ::std::fmt::Debug {
4    /// Resolve a priority list of auth scheme options with the given parameters
5    fn resolve_auth_scheme<'a>(
6        &'a self,
7        params: &'a crate::config::auth::Params,
8        cfg: &'a ::aws_smithy_types::config_bag::ConfigBag,
9        runtime_components: &'a ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponents,
10    ) -> ::aws_smithy_runtime_api::client::auth::AuthSchemeOptionsFuture<'a>;
11
12    /// Convert this service-specific resolver into a `SharedAuthSchemeOptionResolver`
13    fn into_shared_resolver(self) -> ::aws_smithy_runtime_api::client::auth::SharedAuthSchemeOptionResolver
14    where
15        Self: ::std::marker::Sized + 'static,
16    {
17        ::aws_smithy_runtime_api::client::auth::SharedAuthSchemeOptionResolver::new(DowncastParams(self))
18    }
19}
20
21#[derive(Debug)]
22struct DowncastParams<T>(T);
23impl<T> ::aws_smithy_runtime_api::client::auth::ResolveAuthSchemeOptions for DowncastParams<T>
24where
25    T: ResolveAuthScheme,
26{
27    fn resolve_auth_scheme_options_v2<'a>(
28        &'a self,
29        params: &'a ::aws_smithy_runtime_api::client::auth::AuthSchemeOptionResolverParams,
30        cfg: &'a ::aws_smithy_types::config_bag::ConfigBag,
31        runtime_components: &'a ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponents,
32    ) -> ::aws_smithy_runtime_api::client::auth::AuthSchemeOptionsFuture<'a> {
33        match params.get::<crate::config::auth::Params>() {
34            ::std::option::Option::Some(concrete_params) => self.0.resolve_auth_scheme(concrete_params, cfg, runtime_components),
35            ::std::option::Option::None => ::aws_smithy_runtime_api::client::auth::AuthSchemeOptionsFuture::ready(::std::result::Result::Err(
36                "params of expected type was not present".into(),
37            )),
38        }
39    }
40}
41
42/// The default auth scheme resolver
43#[derive(Debug)]
44#[allow(dead_code)]
45pub struct DefaultAuthSchemeResolver {
46    service_defaults: Vec<::aws_smithy_runtime_api::client::auth::AuthSchemeOption>,
47    operation_overrides: ::std::collections::HashMap<&'static str, Vec<::aws_smithy_runtime_api::client::auth::AuthSchemeOption>>,
48}
49
50// TODO(https://github.com/smithy-lang/smithy-rs/issues/4177): Remove `allow(...)` once the issue is addressed.
51// When generating code for tests (e.g., `codegen-client-test`), this manual implementation
52// of the `Default` trait may appear as if it could be derived automatically.
53// However, that is not the case in production.
54#[allow(clippy::derivable_impls)]
55impl Default for DefaultAuthSchemeResolver {
56    fn default() -> Self {
57        Self {
58            service_defaults: vec![::aws_smithy_runtime_api::client::auth::AuthSchemeOption::builder()
59                .scheme_id(::aws_runtime::auth::sigv4::SCHEME_ID)
60                .build()
61                .expect("required fields set")],
62            operation_overrides: ::std::collections::HashMap::new(),
63        }
64    }
65}
66
67impl crate::config::auth::ResolveAuthScheme for DefaultAuthSchemeResolver {
68    fn resolve_auth_scheme<'a>(
69        &'a self,
70        params: &'a crate::config::auth::Params,
71        _cfg: &'a ::aws_smithy_types::config_bag::ConfigBag,
72        _runtime_components: &'a ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponents,
73    ) -> ::aws_smithy_runtime_api::client::auth::AuthSchemeOptionsFuture<'a> {
74        let operation_name = params.operation_name();
75
76        let modeled_auth_options = match self.operation_overrides.get(operation_name) {
77            Some(overrides) => overrides,
78            None => &self.service_defaults,
79        };
80
81        let _fut = ::aws_smithy_runtime_api::client::auth::AuthSchemeOptionsFuture::ready(Ok(modeled_auth_options.clone()));
82
83        _fut
84    }
85}
86
87/// Configuration parameters for resolving the correct auth scheme
88#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::fmt::Debug)]
89pub struct Params {
90    operation_name: ::std::borrow::Cow<'static, str>,
91}
92impl Params {
93    /// Create a builder for [`Params`]
94    pub fn builder() -> crate::config::auth::ParamsBuilder {
95        crate::config::auth::ParamsBuilder::default()
96    }
97
98    /// Return the operation name for [`Params`]
99    pub fn operation_name(&self) -> &str {
100        self.operation_name.as_ref()
101    }
102}
103
104#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
105/// Builder for [`Params`]
106pub struct ParamsBuilder {
107    operation_name: ::std::option::Option<::std::borrow::Cow<'static, str>>,
108}
109impl ParamsBuilder {
110    /// Set the operation name for the builder
111    pub fn operation_name(self, operation_name: impl Into<::std::borrow::Cow<'static, str>>) -> Self {
112        self.set_operation_name(::std::option::Option::Some(operation_name.into()))
113    }
114
115    /// Set the operation name for the builder
116    pub fn set_operation_name(mut self, operation_name: ::std::option::Option<::std::borrow::Cow<'static, str>>) -> Self {
117        self.operation_name = operation_name;
118        self
119    }
120    /// Consume this builder, create [`Params`]."
121    ///
122    /// Return [`BuildError`] if any of the required fields are unset.
123    ///
124    pub fn build(self) -> ::std::result::Result<crate::config::auth::Params, crate::config::auth::BuildError> {
125        ::std::result::Result::Ok(crate::config::auth::Params {
126            operation_name: self.operation_name.ok_or_else(|| BuildError::missing("operation_name"))?,
127        })
128    }
129}
130
131/// An error that occurred while constructing `config::auth::Params`
132#[derive(Debug)]
133pub struct BuildError {
134    field: ::std::borrow::Cow<'static, str>,
135}
136
137impl BuildError {
138    fn missing(field: &'static str) -> Self {
139        Self { field: field.into() }
140    }
141}
142
143impl std::fmt::Display for BuildError {
144    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
145        write!(f, "a required field was missing: `{}`", self.field)
146    }
147}
148
149impl std::error::Error for BuildError {}