aws_sdk_ssooidc/config/
auth.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2/// Auth scheme resolver trait specific to AWS SSO OIDC
3                pub 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)]
22                struct DowncastParams<T>(T);
23                impl<T> ::aws_smithy_runtime_api::client::auth::ResolveAuthSchemeOptions for DowncastParams<T>
24                where
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("params of expected type was not present".into())),
36                        }
37                    }
38                }
39
40/// The default auth scheme resolver
41                #[derive(Debug)]
42                #[allow(dead_code)]
43                pub struct DefaultAuthSchemeResolver {
44                    service_defaults: Vec<::aws_smithy_runtime_api::client::auth::AuthSchemeOption>,
45                    operation_overrides: ::std::collections::HashMap<&'static str, Vec<::aws_smithy_runtime_api::client::auth::AuthSchemeOption>>,
46                    preference: ::std::option::Option<::aws_smithy_runtime_api::client::auth::AuthSchemePreference>,
47                }
48
49                // TODO(https://github.com/smithy-lang/smithy-rs/issues/4177): Remove `allow(...)` once the issue is addressed.
50                // When generating code for tests (e.g., `codegen-client-test`), this manual implementation
51                // of the `Default` trait may appear as if it could be derived automatically.
52                // However, that is not the case in production.
53                #[allow(clippy::derivable_impls)]
54                impl Default for DefaultAuthSchemeResolver {
55                    fn default() -> Self {
56                        Self {
57                            service_defaults: vec![::aws_smithy_runtime_api::client::auth::AuthSchemeOption::builder()
58            .scheme_id(::aws_runtime::auth::sigv4::SCHEME_ID)
59            
60            .build()
61            .expect("required fields set")],
62                            operation_overrides: [
63    ("CreateToken", vec![::aws_smithy_runtime_api::client::auth::AuthSchemeOption::from(::aws_smithy_runtime::client::auth::no_auth::NO_AUTH_SCHEME_ID)])
64    ,
65    ("RegisterClient", vec![::aws_smithy_runtime_api::client::auth::AuthSchemeOption::from(::aws_smithy_runtime::client::auth::no_auth::NO_AUTH_SCHEME_ID)])
66    ,
67    ("StartDeviceAuthorization", vec![::aws_smithy_runtime_api::client::auth::AuthSchemeOption::from(::aws_smithy_runtime::client::auth::no_auth::NO_AUTH_SCHEME_ID)])
68    ,
69]
70.into(),
71                            preference: ::std::option::Option::None,
72                        }
73                    }
74                }
75
76                impl crate::config::auth::ResolveAuthScheme for DefaultAuthSchemeResolver {
77                    fn resolve_auth_scheme<'a>(
78                        &'a self,
79                        params: &'a crate::config::auth::Params,
80                        _cfg: &'a ::aws_smithy_types::config_bag::ConfigBag,
81                        _runtime_components: &'a ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponents,
82                    ) -> ::aws_smithy_runtime_api::client::auth::AuthSchemeOptionsFuture<'a> {
83                        let operation_name = params.operation_name();
84
85                        let modeled_auth_options = match self.operation_overrides.get(operation_name) {
86                            Some(overrides) => overrides,
87                            None => &self.service_defaults,
88                        };
89
90                        let _fut = ::aws_smithy_runtime_api::client::auth::AuthSchemeOptionsFuture::ready(Ok(modeled_auth_options.clone()));
91
92                        
93
94                        match &self.preference {
95                            ::std::option::Option::Some(preference) => {
96                                _fut.map_ok({
97                                    // maps auth scheme ID to the index in the preference list
98                                    let preference_map: ::std::collections::HashMap<_, _> = preference
99                                        .clone()
100                                        .into_iter()
101                                        .enumerate()
102                                        .map(|(i, s)| (s, i))
103                                        .collect();
104                                    move |auth_scheme_options| {
105                                        let (mut preferred, non_preferred): (::std::vec::Vec<_>, ::std::vec::Vec<_>) = auth_scheme_options
106                                            .into_iter()
107                                            .partition(|auth_scheme_option| {
108                                                preference_map.contains_key(auth_scheme_option.scheme_id())
109                                            });
110
111                                        preferred.sort_by_key(|opt| {
112                                            preference_map
113                                                .get(opt.scheme_id())
114                                                .expect("guaranteed by `partition`")
115                                        });
116                                        preferred.extend(non_preferred);
117                                        preferred
118                                    }
119                                })
120                            },
121                            ::std::option::Option::None => _fut,
122                        }
123                    }
124                }
125
126                impl DefaultAuthSchemeResolver {
127                    /// Set auth scheme preference to the default auth scheme resolver
128                    pub fn with_auth_scheme_preference(mut self, preference: impl ::std::convert::Into<::aws_smithy_runtime_api::client::auth::AuthSchemePreference>) -> Self {
129                        self.preference = ::std::option::Option::Some(preference.into());
130                        self
131                    }
132                }
133
134/// Configuration parameters for resolving the correct auth scheme
135#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::fmt::Debug)]
136pub struct Params {
137    operation_name: ::std::borrow::Cow<'static, str>
138}
139impl Params {
140    /// Create a builder for [`Params`]
141                    pub fn builder() -> crate::config::auth::ParamsBuilder {
142                        crate::config::auth::ParamsBuilder::default()
143                    }
144    
145                    /// Return the operation name for [`Params`]
146                    pub fn operation_name(&self) -> &str {
147                        self.operation_name.as_ref()
148                    }
149}
150
151#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
152/// Builder for [`Params`]
153pub struct ParamsBuilder {
154    operation_name: ::std::option::Option<::std::borrow::Cow<'static, str>>,
155}
156impl ParamsBuilder {
157    /// Set the operation name for the builder
158                    pub fn operation_name(self, operation_name: impl Into<::std::borrow::Cow<'static, str>>) -> Self {
159                        self.set_operation_name(::std::option::Option::Some(operation_name.into()))
160                    }
161    
162                    /// Set the operation name for the builder
163                    pub fn set_operation_name(mut self, operation_name: ::std::option::Option<::std::borrow::Cow<'static, str>>) -> Self {
164                        self.operation_name = operation_name;
165                        self
166                    }
167    /// Consume this builder, create [`Params`]."
168    /// 
169    /// Return [`BuildError`] if any of the required fields are unset.
170    /// 
171    pub fn build(self) -> ::std::result::Result<crate::config::auth::Params, crate::config::auth::BuildError> {
172        ::std::result::Result::Ok(crate::config::auth::Params {
173                                operation_name: self.operation_name
174                                    .ok_or_else(||BuildError::missing("operation_name"))?
175                            })
176    }
177}
178
179/// An error that occurred while constructing `config::auth::Params`
180                #[derive(Debug)]
181                pub struct BuildError {
182                    field: ::std::borrow::Cow<'static, str>
183                }
184
185                impl BuildError {
186                    fn missing(field: &'static str) -> Self {
187                        Self { field: field.into() }
188                    }
189                }
190
191                impl std::fmt::Display for BuildError {
192                    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
193                        write!(f, "a required field was missing: `{}`", self.field)
194                    }
195                }
196
197                impl std::error::Error for BuildError { }
198
aws_sdk_ssooidc/config/auth.rs

aws_sdk_ssooidc/config/
auth.rs
