[DRAFT] Transform deep import of DocumentClient

src/transforms/v2-to-v3/__fixtures__/ddb-doc-client/service-import-deep-named.output.js

@@ -1,6 +1,5 @@
-// Transformation of DocumentClient named import from deep path is unsupported in aws-sdk-js-codemod.
-// Please convert to a default import, and re-run aws-sdk-js-codemod.
-import { DocumentClient } from "aws-sdk/clients/dynamodb";
+import { DynamoDBDocument } from "@aws-sdk/lib-dynamodb";
+import { DynamoDB } from "@aws-sdk/client-dynamodb";
 
-const documentClient = new DocumentClient({ region: "us-west-2" });
-const response = await documentClient.scan({ TableName: "TABLE_NAME" }).promise();
+const documentClient = DynamoDBDocument.from(new DynamoDB());
+const response = await documentClient.scan({ TableName: "TABLE_NAME" });

src/transforms/v2-to-v3/__fixtures__/ddb-doc-client/service-require-deep-named.output.js

@@ -1,6 +1,5 @@
-// Transformation of DocumentClient named import from deep path is unsupported in aws-sdk-js-codemod.
-// Please convert to a default import, and re-run aws-sdk-js-codemod.
-const { DocumentClient } = require("aws-sdk/clients/dynamodb");
+const { DynamoDBDocument } = require("@aws-sdk/lib-dynamodb");
+const { DynamoDB } = require("@aws-sdk/client-dynamodb");
 
-const documentClient = new DocumentClient({ region: "us-west-2" });
+const documentClient = DynamoDBDocument.from(new DynamoDB());
 const response = await documentClient.scan({ TableName: "TABLE_NAME" }).promise();

src/transforms/v2-to-v3/apis/index.ts

@@ -1,6 +1,5 @@
 export * from "./addEmptyObjectForUndefined";
 export * from "./addNotSupportedClientComments";
-export * from "./addNotSupportedComments";
 export * from "./getClientIdentifiersRecord";
 export * from "./getClientWaiterStates";
 export * from "./getCommandName";

src/transforms/v2-to-v3/client-names/getClientNamesRecord.ts

@@ -1,6 +1,6 @@
 import { Collection, JSCodeshift } from "jscodeshift";
 
-import { CLIENT_NAMES, PACKAGE_NAME } from "../config";
+import { CLIENT_NAMES, DOCUMENT_CLIENT, DYNAMODB_DOCUMENT_CLIENT, PACKAGE_NAME } from "../config";
 import { ImportType } from "../modules";
 import * as importEqualsModule from "../modules/importEqualsModule";
 import * as importModule from "../modules/importModule";
@@ -37,11 +37,23 @@ export const getClientNamesRecord = (
   for (const clientName of clientNamesFromDeepImport) {
     const deepImportPath = getClientDeepImportPath(clientName);
 
-    const specifiersFromDeepImport = getImportSpecifiers(j, source, deepImportPath).filter(
+    const specifiersFromNamedImport = getImportSpecifiers(j, source, deepImportPath);
+
+    const defaultSpecifiersFromDeepImport = specifiersFromNamedImport.filter(
       (importSpecifier) => !importSpecifier.importedName
     );
-    if (specifiersFromDeepImport.length > 0) {
-      clientNamesRecord[clientName] = specifiersFromDeepImport[0].localName;
+    if (defaultSpecifiersFromDeepImport.length > 0) {
+      clientNamesRecord[clientName] = defaultSpecifiersFromDeepImport[0].localName;
+    }
+
+    const namedSpecifiersFromDeepImport = specifiersFromNamedImport.filter(
+      (importSpecifier) => importSpecifier.importedName
+    );
+    if (
+      namedSpecifiersFromDeepImport.length > 0 &&
+      namedSpecifiersFromDeepImport[0].importedName === DOCUMENT_CLIENT
+    ) {
+      clientNamesRecord[DYNAMODB_DOCUMENT_CLIENT] = namedSpecifiersFromDeepImport[0].localName;
     }
   }
 

src/transforms/v2-to-v3/config/CLIENT_NAMES_MAP.ts

@@ -1,4 +1,5 @@
 import { CLIENT_NAMES } from "./CLIENT_NAMES";
+import { DOCUMENT_CLIENT, DYNAMODB_DOCUMENT_CLIENT } from "./constants";
 
 // The key is the client name in v2, and value is the client name in v3.
 export const CLIENT_NAMES_MAP: Record<string, string> = {
@@ -30,4 +31,5 @@ export const CLIENT_NAMES_MAP: Record<string, string> = {
   SavingsPlans: "Savingsplans",
   StepFunctions: "SFN",
   TranscribeService: "Transcribe",
+  [DYNAMODB_DOCUMENT_CLIENT]: DOCUMENT_CLIENT,
 };

src/transforms/v2-to-v3/transformer.ts

@@ -1,7 +1,6 @@
 import { API, FileInfo } from "jscodeshift";
 
 import {
-  addNotSupportedComments,
   addNotSupportedClientComments,
   removePromiseCalls,
   replaceWaiterApi,
@@ -49,8 +48,6 @@ const transformer = async (file: FileInfo, api: API) => {
     return file.source;
   }
 
-  addNotSupportedComments(j, source, importType);
-
   const v2GlobalName = getGlobalNameFromModule(j, source);
   const v2ClientNamesRecord = getClientNamesRecord(j, source, importType);