Remove .promise() from API calls in variable declarator (#44)

Author: trivikr

.changeset/curvy-emus-occur.md

@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": patch
+---
+
+Remove .promise() from API calls in variable declarator

src/transforms/v2-to-v3/__fixtures__/api-promise-variable-declerator.input.ts

@@ -0,0 +1,9 @@
+import AWS from "aws-sdk";
+
+const client = new AWS.DynamoDB();
+
+const listTablesPromise = client.listTables().promise();
+listTablesPromise
+  .then((data) => console.log(data))
+  .catch((err) => console.log(err, err.stack));
+

src/transforms/v2-to-v3/__fixtures__/api-promise-variable-declerator.output.ts

@@ -0,0 +1,8 @@
+import { DynamoDB } from "@aws-sdk/client-dynamodb";
+
+const client = new DynamoDB();
+
+const listTablesPromise = client.listTables();
+listTablesPromise
+  .then((data) => console.log(data))
+  .catch((err) => console.log(err, err.stack));

src/transforms/v2-to-v3/utils/removePromiseCalls.ts

@@ -43,13 +43,18 @@ export const removePromiseCalls = (
         })
         .forEach((callExpressionPath) => {
           switch (callExpressionPath.parentPath.value.type) {
+            case "AwaitExpression":
+              callExpressionPath.parentPath.value.argument = (
+                callExpressionPath.value.callee as MemberExpression
+              ).object;
+              break;
             case "MemberExpression":
               callExpressionPath.parentPath.value.object = (
                 callExpressionPath.value.callee as MemberExpression
               ).object;
               break;
-            case "AwaitExpression":
-              callExpressionPath.parentPath.value.argument = (
+            case "VariableDeclarator":
+              callExpressionPath.parentPath.value.init = (
                 callExpressionPath.value.callee as MemberExpression
               ).object;
               break;