Remove .promise() from API calls which use await (#41)

Author: trivikr

.changeset/blue-experts-decide.md

@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": patch
+---
+
+Remove .promise() from API calls which use await

src/transforms/v2-to-v3/__fixtures__/api-await.input.ts

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

src/transforms/v2-to-v3/__fixtures__/api-await.output.ts

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

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

@@ -42,9 +42,22 @@ export const removePromiseCalls = (
           },
         })
         .forEach((callExpressionPath) => {
-          callExpressionPath.parentPath.value.object = (
-            callExpressionPath.value.callee as MemberExpression
-          ).object;
+          switch (callExpressionPath.parentPath.value.type) {
+            case "MemberExpression":
+              callExpressionPath.parentPath.value.object = (
+                callExpressionPath.value.callee as MemberExpression
+              ).object;
+              break;
+            case "AwaitExpression":
+              callExpressionPath.parentPath.value.argument = (
+                callExpressionPath.value.callee as MemberExpression
+              ).object;
+              break;
+            default:
+              throw new Error(
+                `Removal of .promise() not implemented for ${callExpressionPath.parentPath.value.type}`
+              );
+          }
         });
     });
 };