chore(middleware-user-agent): add feature detection for account-id, s3-express, lib-dynamodb

Author: kuhe

lib/lib-dynamodb/src/baseCommand/DynamoDBDocumentClientCommand.ts

@@ -1,3 +1,4 @@
+import { setFeature } from "@aws-sdk/core";
 import { Command as $Command } from "@smithy/smithy-client";
 import {
   DeserializeHandler,
@@ -51,6 +52,7 @@ export abstract class DynamoDBDocumentClientCommand<
         async (
           args: InitializeHandlerArguments<Input | BaseInput>
         ): Promise<InitializeHandlerOutput<Output | BaseOutput>> => {
+          setFeature(context, "DDB_MAPPER", "d");
           args.input = marshallInput(this.input, this.inputKeyNodes, marshallOptions);
           context.dynamoDbDocumentClientOptions =
             context.dynamoDbDocumentClientOptions || DynamoDBDocumentClientCommand.defaultLogFilterOverrides;

packages/middleware-sdk-s3/src/s3-express/functions/s3ExpressMiddleware.ts

@@ -1,3 +1,4 @@
+import { setFeature } from "@aws-sdk/core";
 import { AwsCredentialIdentity } from "@aws-sdk/types";
 import { HttpRequest } from "@smithy/protocol-http";
 import {
@@ -51,6 +52,7 @@ export const s3ExpressMiddleware: (options: S3ExpressResolvedConfig) => BuildMid
           endpoint.properties?.bucketType === S3_EXPRESS_BUCKET_TYPE;
 
         if (isS3ExpressBucket) {
+          setFeature(context, "S3_EXPRESS_BUCKET", "J");
           context.isS3ExpressBucket = true;
         }
 

packages/middleware-sdk-s3/src/s3-express/middleware-s3-express.integ.spec.ts

@@ -84,5 +84,22 @@ describe("middleware-s3-express", () => {
 
       expect.hasAssertions();
     });
+
+    it("should feature-detect S3 express bucket", async () => {
+      const client = new S3({
+        region: "us-west-2",
+        s3ExpressIdentityProvider,
+      });
+
+      requireRequestsFrom(client).toMatch({
+        headers: {
+          "user-agent": /(.*?) m\/J$/,
+        },
+      });
+
+      await client.headBucket({
+        Bucket: "aws-sdk-js-v3-test--usw2-az1--x-s3",
+      });
+    });
   });
 });

packages/middleware-user-agent/package.json

@@ -22,6 +22,7 @@
   },
   "license": "Apache-2.0",
   "dependencies": {
+    "@aws-sdk/core": "*",
     "@aws-sdk/types": "*",
     "@aws-sdk/util-endpoints": "*",
     "@smithy/core": "^2.4.7",

packages/middleware-user-agent/src/check-features.ts

@@ -0,0 +1,40 @@
+import { setFeature } from "@aws-sdk/core";
+import type { AccountIdEndpointMode } from "@aws-sdk/core/account-id-endpoint";
+import type { AwsHandlerExecutionContext } from "@aws-sdk/types";
+import type { IHttpRequest } from "@smithy/protocol-http";
+import type { AwsCredentialIdentityProvider, BuildHandlerArguments, Provider } from "@smithy/types";
+
+/**
+ * @internal
+ */
+type PreviouslyResolved = Partial<{
+  credentials?: AwsCredentialIdentityProvider;
+  accountIdEndpointMode?: Provider<AccountIdEndpointMode>;
+}>;
+
+/**
+ * @internal
+ * Check for features that don't have a middleware activation site but
+ * may be detected on the client config.
+ */
+export async function checkFeatures<C extends {} | PreviouslyResolved>(
+  context: AwsHandlerExecutionContext,
+  config: C,
+  args: BuildHandlerArguments<any>
+): Promise<void> {
+  // eslint-disable-next-line
+  const request = args.request as IHttpRequest;
+  if ("accountIdEndpointMode" in config) {
+    switch (await config.accountIdEndpointMode?.()) {
+      case "disabled":
+        setFeature(context, "ACCOUNT_ID_MODE_DISABLED", "Q");
+        break;
+      case "preferred":
+        setFeature(context, "ACCOUNT_ID_MODE_PREFERRED", "P");
+        break;
+      case "required":
+        setFeature(context, "ACCOUNT_ID_MODE_REQUIRED", "R");
+        break;
+    }
+  }
+}

packages/middleware-user-agent/src/configurations.ts

@@ -1,5 +1,5 @@
-import { Logger, Provider, UserAgent } from "@smithy/types";
 import { normalizeProvider } from "@smithy/core";
+import { Logger, Provider, UserAgent } from "@smithy/types";
 
 /**
  * @internal

packages/middleware-user-agent/src/middleware-user-agent.integ.spec.ts

@@ -1,4 +1,7 @@
 import { CodeCatalyst } from "@aws-sdk/client-codecatalyst";
+import { DynamoDB } from "@aws-sdk/client-dynamodb";
+import { S3 } from "@aws-sdk/client-s3";
+import { DynamoDBDocument } from "@aws-sdk/lib-dynamodb";
 
 import { requireRequestsFrom } from "../../../private/aws-util-test/src";
 
@@ -23,4 +26,32 @@ describe("middleware-user-agent", () => {
       });
     });
   });
+
+  describe("features", () => {
+    it("should detect DDB mapper, and account id mode", async () => {
+      const client = new DynamoDB({
+        credentials: {
+          accessKeyId: "",
+          secretAccessKey: "",
+          accountId: "123",
+        },
+        accountIdEndpointMode: async () => "preferred" as const,
+      });
+
+      const doc = DynamoDBDocument.from(client);
+
+      requireRequestsFrom(doc).toMatch({
+        headers: {
+          "user-agent": /(.*?) m\/d,P$/,
+        },
+      });
+
+      await doc.get({
+        TableName: "table",
+        Key: {
+          id: "1",
+        },
+      });
+    });
+  });
 });

packages/middleware-user-agent/src/user-agent-middleware.ts

@@ -13,6 +13,7 @@ import {
   UserAgentPair,
 } from "@smithy/types";
 
+import { checkFeatures } from "./check-features";
 import { UserAgentResolvedConfig } from "./configurations";
 import {
   SPACE,
@@ -51,12 +52,15 @@ export const userAgentMiddleware =
     const { headers } = request;
     const userAgent = context?.userAgent?.map(escapeUserAgent) || [];
     const defaultUserAgent = (await options.defaultUserAgentProvider()).map(escapeUserAgent);
+
+    await checkFeatures(context, options, args);
     const awsContext = context as AwsHandlerExecutionContext;
     defaultUserAgent.push(
       `m/${encodeFeatures(
         Object.assign({}, context.__smithy_context?.features, awsContext.__aws_sdk_context?.features)
       )}`
     );
+
     const customUserAgent = options?.customUserAgent?.map(escapeUserAgent) || [];
     const appId = await options.userAgentAppId();
     if (appId) {

private/aws-util-test/src/requests/test-http-handler.ts

@@ -56,7 +56,7 @@ export class TestHttpHandler implements HttpHandler {
    */
   public watch(client: Client<any, any, any>, matcher: HttpRequestMatcher = this.matcher) {
     this.client = client;
-    this.originalRequestHandler = client.config.originalRequestHandler;
+    this.originalRequestHandler = client.config.requestHandler;
     // mock credentials to avoid default chain lookup.
     client.config.credentials = async () => MOCK_CREDENTIALS;
     client.config.credentialDefaultProvider = () => {