fix(middleware-sdk-s3-control): place host prefix dedupe after serial…

…izer
aws · Dec 14, 2022 · 7dff8ee · 7dff8ee
1 parent c5c8cd1
commit 7dff8ee
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/...dleware-sdk-s3-control/src/host-prefix-deduplication/hostPrefixDeduplicationMiddleware.ts b/...dleware-sdk-s3-control/src/host-prefix-deduplication/hostPrefixDeduplicationMiddleware.ts
@@ -1,6 +1,6 @@
 import {
-  EndpointV2,
   HandlerExecutionContext,
+  HttpRequest,
   Pluggable,
   RelativeMiddlewareOptions,
   SerializeHandler,
@@ -21,11 +21,9 @@ import { deduplicateHostPrefix } from "./deduplicateHostPrefix";
 export const hostPrefixDeduplicationMiddleware = (): SerializeMiddleware<any, any> => {
   return (next: SerializeHandler<any, any>, context: HandlerExecutionContext): SerializeHandler<any, any> =>
     async (args: SerializeHandlerArguments<any>): Promise<SerializeHandlerOutput<any>> => {
-      const endpoint: EndpointV2 | undefined = context.endpointV2;
-      if (endpoint?.url?.hostname) {
-        endpoint.url.hostname = deduplicateHostPrefix(endpoint.url.hostname);
-      } else {
-        throw new Error("Endpoint w/ url.hostname not found in hostPrefixDeduplicationMiddleware.");
+      const httpRequest: HttpRequest = (args.request ?? {}) as HttpRequest;
+      if (httpRequest?.hostname) {
+        httpRequest.hostname = deduplicateHostPrefix(httpRequest.hostname);
       }
       return next(args);
     };
@@ -36,7 +34,7 @@ export const hostPrefixDeduplicationMiddleware = (): SerializeMiddleware<any, an
  */
 export const hostPrefixDeduplicationMiddlewareOptions: RelativeMiddlewareOptions = {
   tags: ["HOST_PREFIX_DEDUPLICATION", "ENDPOINT_V2", "ENDPOINT"],
-  toMiddleware: "endpointV2Middleware",
+  toMiddleware: "serializerMiddleware",
   relation: "after",
   name: "hostPrefixDeduplicationMiddleware",
   override: true,
@@ -45,8 +43,8 @@ export const hostPrefixDeduplicationMiddlewareOptions: RelativeMiddlewareOptions
 /**
  * @internal
  */
-export const getHostPrefixDeduplicationPlugin = <T>(config?: T): Pluggable<any, any> => ({
+export const getHostPrefixDeduplicationPlugin = <T>(config: T): Pluggable<any, any> => ({
   applyToStack: (clientStack) => {
-    clientStack.add(hostPrefixDeduplicationMiddleware(), hostPrefixDeduplicationMiddlewareOptions);
+    clientStack.addRelativeTo(hostPrefixDeduplicationMiddleware(), hostPrefixDeduplicationMiddlewareOptions);
   },
 });