[smithy-rs] Follow up on DefaultEndpointResolver in the orchestrator (#2592)

## Motivation and Context
This PR incorporates post-merge feedback left in #2577.

----

_By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice._

---------

Co-authored-by: Yuki Saito <awsaito@amazon.com>

Author: ysaito1001

sdk/aws-smithy-runtime-api/src/client.rs

@@ -21,8 +21,5 @@ pub mod retries;
 /// Runtime plugin type definitions.
 pub mod runtime_plugin;
 
-/// Smithy endpoint resolution runtime plugins
-pub mod endpoints;
-
 /// Smithy auth runtime plugins
 pub mod auth;

sdk/aws-smithy-runtime-api/src/client/endpoints.rs

@@ -169,3 +169,31 @@ impl std::error::Error for InterceptorError {
         self.source.as_ref().map(|err| err.as_ref() as _)
     }
 }
+
+/// A convenience error that allows for adding additional `context` to `source`
+#[derive(Debug)]
+pub struct ContextAttachedError {
+    context: String,
+    source: Option<BoxError>,
+}
+
+impl ContextAttachedError {
+    pub fn new(context: impl Into<String>, source: impl Into<BoxError>) -> Self {
+        Self {
+            context: context.into(),
+            source: Some(source.into()),
+        }
+    }
+}
+
+impl fmt::Display for ContextAttachedError {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(f, "{}", self.context)
+    }
+}
+
+impl std::error::Error for ContextAttachedError {
+    fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
+        self.source.as_ref().map(|err| err.as_ref() as _)
+    }
+}

sdk/aws-smithy-runtime-api/src/client/interceptors/error.rs

@@ -6,5 +6,5 @@ allowed_external_types = [
     "http::header::name::HeaderName",
     "http::request::Request",
     "http::response::Response",
-    "uri::Uri",
+    "http::uri::Uri",
 ]

sdk/aws-smithy-runtime/external-types.toml

@@ -18,7 +18,8 @@ use aws_smithy_runtime_api::config_bag::ConfigBag;
 use tracing::{debug_span, Instrument};
 
 mod auth;
-mod endpoints;
+/// Defines types that implement a trait for endpoint resolution
+pub mod endpoints;
 mod http;
 pub(self) mod phase;
 

sdk/aws-smithy-runtime/src/client/orchestrator.rs

@@ -3,12 +3,114 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-use aws_smithy_http::endpoint::EndpointPrefix;
+use aws_smithy_http::endpoint::error::ResolveEndpointError;
+use aws_smithy_http::endpoint::{
+    apply_endpoint, EndpointPrefix, ResolveEndpoint, SharedEndpointResolver,
+};
 use aws_smithy_runtime_api::client::interceptors::InterceptorContext;
 use aws_smithy_runtime_api::client::orchestrator::{
-    BoxError, ConfigBagAccessors, HttpRequest, HttpResponse,
+    BoxError, ConfigBagAccessors, EndpointResolver, EndpointResolverParams, HttpRequest,
+    HttpResponse,
 };
 use aws_smithy_runtime_api::config_bag::ConfigBag;
+use http::header::HeaderName;
+use http::{HeaderValue, Uri};
+use std::fmt::Debug;
+use std::str::FromStr;
+
+#[derive(Debug, Clone)]
+pub struct StaticUriEndpointResolver {
+    endpoint: Uri,
+}
+
+impl StaticUriEndpointResolver {
+    pub fn http_localhost(port: u16) -> Self {
+        Self {
+            endpoint: Uri::from_str(&format!("http://localhost:{port}"))
+                .expect("all u16 values are valid ports"),
+        }
+    }
+
+    pub fn uri(endpoint: Uri) -> Self {
+        Self { endpoint }
+    }
+}
+
+impl EndpointResolver for StaticUriEndpointResolver {
+    fn resolve_and_apply_endpoint(
+        &self,
+        _params: &EndpointResolverParams,
+        _endpoint_prefix: Option<&EndpointPrefix>,
+        request: &mut HttpRequest,
+    ) -> Result<(), BoxError> {
+        apply_endpoint(request.uri_mut(), &self.endpoint, None)?;
+        Ok(())
+    }
+}
+
+#[derive(Debug, Clone)]
+pub struct DefaultEndpointResolver<Params> {
+    inner: SharedEndpointResolver<Params>,
+}
+
+impl<Params> DefaultEndpointResolver<Params> {
+    pub fn new(resolve_endpoint: SharedEndpointResolver<Params>) -> Self {
+        Self {
+            inner: resolve_endpoint,
+        }
+    }
+}
+
+impl<Params> EndpointResolver for DefaultEndpointResolver<Params>
+where
+    Params: Debug + Send + Sync + 'static,
+{
+    fn resolve_and_apply_endpoint(
+        &self,
+        params: &EndpointResolverParams,
+        endpoint_prefix: Option<&EndpointPrefix>,
+        request: &mut HttpRequest,
+    ) -> Result<(), BoxError> {
+        let endpoint = match params.get::<Params>() {
+            Some(params) => self.inner.resolve_endpoint(params)?,
+            None => {
+                return Err(Box::new(ResolveEndpointError::message(
+                    "params of expected type was not present",
+                )));
+            }
+        };
+
+        let uri: Uri = endpoint.url().parse().map_err(|err| {
+            ResolveEndpointError::from_source("endpoint did not have a valid uri", err)
+        })?;
+
+        apply_endpoint(request.uri_mut(), &uri, endpoint_prefix).map_err(|err| {
+            ResolveEndpointError::message(format!(
+                "failed to apply endpoint `{:?}` to request `{:?}`",
+                uri, request,
+            ))
+            .with_source(Some(err.into()))
+        })?;
+
+        for (header_name, header_values) in endpoint.headers() {
+            request.headers_mut().remove(header_name);
+            for value in header_values {
+                request.headers_mut().insert(
+                    HeaderName::from_str(header_name).map_err(|err| {
+                        ResolveEndpointError::message("invalid header name")
+                            .with_source(Some(err.into()))
+                    })?,
+                    HeaderValue::from_str(value).map_err(|err| {
+                        ResolveEndpointError::message("invalid header value")
+                            .with_source(Some(err.into()))
+                    })?,
+                );
+            }
+        }
+
+        Ok(())
+    }
+}
 
 pub(super) fn orchestrate_endpoint(
     ctx: &mut InterceptorContext<HttpRequest, HttpResponse>,

sdk/aws-smithy-runtime/src/client/orchestrator/endpoints.rs

@@ -1,4 +1,4 @@
-smithy_rs_revision = '1b138b09a2196d67fe6bc6a7b481ba4edfe25b0e'
+smithy_rs_revision = '08642d1e891733e08f8715f2a4d81f789ac133b5'
 aws_doc_sdk_examples_revision = '1f48dbeac254cb3fa6d3eb25628ee748d42017d1'
 
 [manual_interventions]
@@ -2141,12 +2141,12 @@ source_hash = 'd23bb3777f4a48f82be9d68582b67b14e73832f04a58dd29e632f2171a2d7df5'
 [crates.aws-smithy-runtime]
 category = 'SmithyRuntime'
 version = '0.55.1'
-source_hash = '8892131bb1110fcd32cc8bca1a7995be0124bd5ba8262d340b7c083b1387b2d3'
+source_hash = '8d1bd249588e0280782e56b1871ecdcd0ea5aad41e6b432103db0d75cc7b10b9'
 
 [crates.aws-smithy-runtime-api]
 category = 'SmithyRuntime'
 version = '0.55.1'
-source_hash = 'ea4721f30fe639b63a87f2deee516096310de52127f7bb326e1510fe61a034f0'
+source_hash = '80a5a7b92ec8b4f4eb5be0966ed5a7c0202d0866ab3c34946608b2f290e0ba04'
 
 [crates.aws-smithy-types]
 category = 'SmithyRuntime'