aws_sdk_bedrockruntime/config/
auth.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2/// Auth scheme resolver trait specific to Amazon Bedrock Runtime
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![
59                ::aws_smithy_runtime_api::client::auth::AuthSchemeOption::builder()
60                    .scheme_id(::aws_runtime::auth::sigv4::SCHEME_ID)
61                    .build()
62                    .expect("required fields set"),
63                ::aws_smithy_runtime_api::client::auth::AuthSchemeOption::builder()
64                    .scheme_id(::aws_smithy_runtime_api::client::auth::http::HTTP_BEARER_AUTH_SCHEME_ID)
65                    .build()
66                    .expect("required fields set"),
67            ],
68            operation_overrides: ::std::collections::HashMap::new(),
69        }
70    }
71}
72
73impl crate::config::auth::ResolveAuthScheme for DefaultAuthSchemeResolver {
74    fn resolve_auth_scheme<'a>(
75        &'a self,
76        params: &'a crate::config::auth::Params,
77        _cfg: &'a ::aws_smithy_types::config_bag::ConfigBag,
78        _runtime_components: &'a ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponents,
79    ) -> ::aws_smithy_runtime_api::client::auth::AuthSchemeOptionsFuture<'a> {
80        let operation_name = params.operation_name();
81
82        let modeled_auth_options = match self.operation_overrides.get(operation_name) {
83            Some(overrides) => overrides,
84            None => &self.service_defaults,
85        };
86
87        let _fut = ::aws_smithy_runtime_api::client::auth::AuthSchemeOptionsFuture::ready(Ok(modeled_auth_options.clone()));
88
89        _fut
90    }
91}
92
93/// Configuration parameters for resolving the correct auth scheme
94#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::fmt::Debug)]
95pub struct Params {
96    operation_name: ::std::borrow::Cow<'static, str>,
97}
98impl Params {
99    /// Create a builder for [`Params`]
100    pub fn builder() -> crate::config::auth::ParamsBuilder {
101        crate::config::auth::ParamsBuilder::default()
102    }
103
104    /// Return the operation name for [`Params`]
105    pub fn operation_name(&self) -> &str {
106        self.operation_name.as_ref()
107    }
108}
109
110#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
111/// Builder for [`Params`]
112pub struct ParamsBuilder {
113    operation_name: ::std::option::Option<::std::borrow::Cow<'static, str>>,
114}
115impl ParamsBuilder {
116    /// Set the operation name for the builder
117    pub fn operation_name(self, operation_name: impl Into<::std::borrow::Cow<'static, str>>) -> Self {
118        self.set_operation_name(::std::option::Option::Some(operation_name.into()))
119    }
120
121    /// Set the operation name for the builder
122    pub fn set_operation_name(mut self, operation_name: ::std::option::Option<::std::borrow::Cow<'static, str>>) -> Self {
123        self.operation_name = operation_name;
124        self
125    }
126    /// Consume this builder, create [`Params`]."
127    ///
128    /// Return [`BuildError`] if any of the required fields are unset.
129    ///
130    pub fn build(self) -> ::std::result::Result<crate::config::auth::Params, crate::config::auth::BuildError> {
131        ::std::result::Result::Ok(crate::config::auth::Params {
132            operation_name: self.operation_name.ok_or_else(|| BuildError::missing("operation_name"))?,
133        })
134    }
135}
136
137/// An error that occurred while constructing `config::auth::Params`
138#[derive(Debug)]
139pub struct BuildError {
140    field: ::std::borrow::Cow<'static, str>,
141}
142
143impl BuildError {
144    fn missing(field: &'static str) -> Self {
145        Self { field: field.into() }
146    }
147}
148
149impl std::fmt::Display for BuildError {
150    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
151        write!(f, "a required field was missing: `{}`", self.field)
152    }
153}
154
155impl std::error::Error for BuildError {}
aws_sdk_bedrockruntime/config/auth.rs

aws_sdk_bedrockruntime/config/
auth.rs
