Make AwaitExpression/VariableDeclarator generic while removing .promi…

…se() API (#403)
aws · Feb 23, 2023 · 13fd08a · 13fd08a
1 parent 8653240
commit 13fd08a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 13 deletions.
diff --git a/.changeset/empty-flowers-speak.md b/.changeset/empty-flowers-speak.md
@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": patch
+---
+
+Make AwaitExpression/VariableDeclarator generic while removing .promise() API
diff --git a/src/transforms/v2-to-v3/apis/removePromiseForCallExpression.ts b/src/transforms/v2-to-v3/apis/removePromiseForCallExpression.ts
@@ -3,27 +3,17 @@ import { print } from "recast";
 
 export const removePromiseForCallExpression = (callExpression: ASTPath<CallExpression>) => {
   switch (callExpression.parentPath.value.type) {
-    case "AwaitExpression": {
-      callExpression.parentPath.value.argument = (
-        callExpression.value.callee as MemberExpression
-      ).object;
-      break;
-    }
     case "MemberExpression": {
       callExpression.parentPath.value.object = (
         callExpression.value.callee as MemberExpression
       ).object;
       break;
     }
-    case "VariableDeclarator": {
-      callExpression.parentPath.value.init = (
-        callExpression.value.callee as MemberExpression
-      ).object;
-      break;
-    }
     case "ArrowFunctionExpression":
+    case "AwaitExpression":
     case "ObjectProperty":
-    case "ReturnStatement": {
+    case "ReturnStatement":
+    case "VariableDeclarator": {
       const currentCalleeObject = (callExpression.value.callee as MemberExpression)
         .object as CallExpression;
       if (currentCalleeObject.arguments.length > 0) {