Transform DocumentClient AttributeValue to NativeAttributeValue (#838)

Author: trivikr

.changeset/mean-ghosts-sparkle.md

@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": patch
+---
+
+Transform DocumentClient AttributeValue to lib-dynamodb NativeAttributeValue

scripts/generateClientTypesMap/getClientTypesMap.ts

@@ -1,6 +1,6 @@
 import jscodeshift, { Identifier, TSArrayType, TSTypeLiteral, TSTypeReference } from "jscodeshift";
 
-import { CLIENT_NAMES_MAP } from "../../src/transforms/v2-to-v3/config";
+import { CLIENT_NAMES_MAP, DOCUMENT_CLIENT } from "../../src/transforms/v2-to-v3/config";
 import { getClientTypesMapWithKeysRemovedFromValues } from "./getClientTypesMapWithKeysRemovedFromValues";
 import { getTypesSource } from "./getTypesSource";
 
@@ -25,7 +25,12 @@ export const getClientTypesMap = async (clientName: string): Promise<Record<stri
       tsTypes
         .filter((tsType) => tsType.typeAnnotation.type === type)
         .forEach((tsType) => {
-          clientTypesMap[tsType.id.name] = value;
+          const { name } = tsType.id;
+          if (clientName === DOCUMENT_CLIENT && name === "AttributeValue" && value === "any") {
+            clientTypesMap[name] = "NativeAttributeValue";
+          } else {
+            clientTypesMap[name] = value;
+          }
         });
     }
 

src/transforms/v2-to-v3/__fixtures__/ddb-doc-client-basic-type/global-import-equals.input.ts

@@ -0,0 +1,3 @@
+import AWS = require("aws-sdk");
+
+const key: AWS.DynamoDB.DocumentClient.Key = {};

src/transforms/v2-to-v3/__fixtures__/ddb-doc-client-basic-type/global-import-equals.output.ts

@@ -0,0 +1,4 @@
+import AWS_lib_dynamodb = require("@aws-sdk/lib-dynamodb");
+import NativeAttributeValue = AWS_lib_dynamodb.NativeAttributeValue;
+
+const key: Record<string, NativeAttributeValue> = {};

src/transforms/v2-to-v3/__fixtures__/ddb-doc-client-basic-type/global-import.input.ts

@@ -0,0 +1,3 @@
+import AWS from "aws-sdk";
+
+const key: AWS.DynamoDB.DocumentClient.Key = {};

src/transforms/v2-to-v3/__fixtures__/ddb-doc-client-basic-type/global-import.output.ts

@@ -0,0 +1,3 @@
+import { NativeAttributeValue } from "@aws-sdk/lib-dynamodb";
+
+const key: Record<string, NativeAttributeValue> = {};

src/transforms/v2-to-v3/__fixtures__/ddb-doc-client-basic-type/service-import-deep.input.ts

@@ -0,0 +1,3 @@
+import DynamoDB from "aws-sdk/clients/dynamodb";
+
+const key: DynamoDB.DocumentClient.Key = {};

src/transforms/v2-to-v3/__fixtures__/ddb-doc-client-basic-type/service-import-deep.output.ts

@@ -0,0 +1,3 @@
+import { NativeAttributeValue } from "@aws-sdk/lib-dynamodb";
+
+const key: Record<string, NativeAttributeValue> = {};

src/transforms/v2-to-v3/__fixtures__/ddb-doc-client-basic-type/service-import-equals.input.ts

@@ -0,0 +1,3 @@
+import DynamoDB = require("aws-sdk/clients/dynamodb");
+
+const key: DynamoDB.DocumentClient.Key = {};

src/transforms/v2-to-v3/__fixtures__/ddb-doc-client-basic-type/service-import-equals.output.ts

@@ -0,0 +1,4 @@
+import AWS_lib_dynamodb = require("@aws-sdk/lib-dynamodb");
+import NativeAttributeValue = AWS_lib_dynamodb.NativeAttributeValue;
+
+const key: Record<string, NativeAttributeValue> = {};