Store clientIdentifiersRecord at transformer level (#553)

Author: trivikr

.changeset/eleven-lemons-eat.md

@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": patch
+---
+
+Store clientIdentifiersRecord at transformer level

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

@@ -1,26 +1,25 @@
 import { Collection, JSCodeshift } from "jscodeshift";
 
-import { FUNCTION_TYPE_LIST } from "../config";
+import { FUNCTION_TYPE_LIST, S3 } from "../config";
+import { ClientIdentifier } from "../types";
 import { getClientApiCallExpression } from "./getClientApiCallExpression";
-import { getClientIdentifiers } from "./getClientIdentifiers";
 import { getClientWaiterCallExpression } from "./getClientWaiterCallExpression";
 import { getClientWaiterStates } from "./getClientWaiterStates";
 
 export interface CommentsForUnsupportedAPIsOptions {
   v2ClientName: string;
-  v2ClientLocalName: string;
-  v2GlobalName?: string;
+  clientIdentifiers: ClientIdentifier[];
 }
 
 export const addNotSupportedClientComments = (
   j: JSCodeshift,
   source: Collection<unknown>,
   options: CommentsForUnsupportedAPIsOptions
 ): void => {
-  const clientIdentifiers = getClientIdentifiers(j, source, options);
+  const { v2ClientName, clientIdentifiers } = options;
 
   for (const clientId of clientIdentifiers) {
-    const waiterStates = getClientWaiterStates(j, source, options);
+    const waiterStates = getClientWaiterStates(j, source, clientIdentifiers);
 
     for (const waiterState of waiterStates) {
       source
@@ -46,7 +45,7 @@ export const addNotSupportedClientComments = (
     }
   }
 
-  if (options.v2ClientName === "S3") {
+  if (v2ClientName === S3) {
     for (const clientId of clientIdentifiers) {
       const apiMetadata = [
         {

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

@@ -1,6 +1,6 @@
 import { CallExpression } from "jscodeshift";
 
-import { ClientIdentifier } from "./getClientIdentifiers";
+import { ClientIdentifier } from "../types";
 
 export const getClientApiCallExpression = (
   clientId: ClientIdentifier,

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

@@ -1,10 +1,5 @@
-import { Collection, Identifier, JSCodeshift, MemberExpression, ThisExpression } from "jscodeshift";
-
-export interface ThisMemberExpression {
-  type: "MemberExpression";
-  object: ThisExpression;
-  property: Identifier;
-}
+import { Collection, Identifier, JSCodeshift, MemberExpression } from "jscodeshift";
+import { ThisMemberExpression } from "../types";
 
 const thisMemberExpression = { type: "MemberExpression", object: { type: "ThisExpression" } };
 

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

@@ -1,17 +1,16 @@
 import { Collection, Identifier, JSCodeshift } from "jscodeshift";
 
+import { ClientIdentifier } from "../types";
 import { getClientIdNamesFromNewExpr } from "./getClientIdNamesFromNewExpr";
 import { getClientIdNamesFromTSTypeRef } from "./getClientIdNamesFromTSTypeRef";
-import { getClientIdThisExpressions, ThisMemberExpression } from "./getClientIdThisExpressions";
+import { getClientIdThisExpressions } from "./getClientIdThisExpressions";
 
 export interface GetClientIdentifiersOptions {
   v2ClientName: string;
   v2ClientLocalName: string;
   v2GlobalName?: string;
 }
 
-export type ClientIdentifier = Identifier | ThisMemberExpression;
-
 export const getClientIdentifiers = (
   j: JSCodeshift,
   source: Collection<unknown>,

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

@@ -0,0 +1,21 @@
+import { Collection, JSCodeshift } from "jscodeshift";
+
+import { ClientIdentifiersRecord } from "../types";
+import { getClientIdentifiers } from "./getClientIdentifiers";
+
+export interface GetClientIdentifiersRecordOptions {
+  v2GlobalName?: string;
+  v2ClientNamesRecord: Record<string, string>;
+}
+
+export const getClientIdentifiersRecord = (
+  j: JSCodeshift,
+  source: Collection<unknown>,
+  { v2GlobalName, v2ClientNamesRecord }: GetClientIdentifiersRecordOptions
+): ClientIdentifiersRecord =>
+  Object.fromEntries(
+    Object.entries(v2ClientNamesRecord).map(([v2ClientName, v2ClientLocalName]) => [
+      v2ClientName,
+      getClientIdentifiers(j, source, { v2ClientName, v2ClientLocalName, v2GlobalName }),
+    ])
+  );

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

@@ -1,6 +1,6 @@
 import { CallExpression } from "jscodeshift";
 
-import { ClientIdentifier } from "./getClientIdentifiers";
+import { ClientIdentifier } from "../types";
 
 export const getClientWaiterCallExpression = (
   clientId: ClientIdentifier,

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

@@ -1,22 +1,13 @@
 import { Collection, JSCodeshift } from "jscodeshift";
-
-import { getClientIdentifiers } from "./getClientIdentifiers";
-
-export interface GetClientWaiterStatesOptions {
-  v2ClientName: string;
-  v2ClientLocalName: string;
-  v2GlobalName?: string;
-}
+import { ClientIdentifier } from "../types";
 
 export const getClientWaiterStates = (
   j: JSCodeshift,
   source: Collection<unknown>,
-  options: GetClientWaiterStatesOptions
+  clientIdentifiers: ClientIdentifier[]
 ): Set<string> => {
   const waiterStates: string[] = [];
 
-  const clientIdentifiers = getClientIdentifiers(j, source, options);
-
   for (const clientId of clientIdentifiers) {
     source
       .find(j.CallExpression, {

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

@@ -1,22 +1,13 @@
 import { Collection, JSCodeshift, Literal } from "jscodeshift";
 
-import { getClientIdentifiers } from "./getClientIdentifiers";
-
-export interface GetS3SignedUrlApiNameOptions {
-  v2ClientName: string;
-  v2ClientLocalName: string;
-  v2GlobalName?: string;
-}
+import { ClientIdentifier } from "../types";
 
 export const getS3SignedUrlApiNames = (
   j: JSCodeshift,
   source: Collection<unknown>,
-  options: GetS3SignedUrlApiNameOptions
+  clientIdentifiers: ClientIdentifier[]
 ): string[] => {
-  if (options.v2ClientName !== "S3") return [];
-
   const apiNames: Set<string> = new Set();
-  const clientIdentifiers = getClientIdentifiers(j, source, options);
 
   for (const clientId of clientIdentifiers) {
     for (const apiName of ["getSignedUrl", "getSignedUrlPromise"]) {

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

@@ -10,3 +10,4 @@ export * from "./replaceS3GetSignedUrlApi";
 export * from "./replaceS3UploadApi";
 export * from "./replaceWaiterApi";
 export * from "./getCommandName";
+export * from "./getClientIdentifiersRecord";