fix lambda error response payload for aws v3 (#566)

OpenZeppelin · May 17, 2024 · 41c14ef · 41c14ef
1 parent a4f42cc
commit 41c14ef
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/packages/base/src/autotask/index.ts b/packages/base/src/autotask/index.ts
@@ -1,20 +1,21 @@
-import Lambda, { _Blob } from 'aws-sdk/clients/lambda';
 import { rateLimitModule, RateLimitModule } from '../utils/rate-limit';
 import { getTimestampInSeconds } from '../utils/time';
-import { getLambdaFromCredentials, isLambdaV3, LambdaLike, PayloadResponseV2, PayloadResponseV3 } from '../utils/lambda';
+import { getLambdaFromCredentials, isLambdaV3, isV3ResponsePayload, LambdaLike, PayloadResponseV2, PayloadResponseV3 } from '../utils/lambda';
 
 // do our best to get .errorMessage, but return object by default
 function cleanError(payload?: PayloadResponseV2 | PayloadResponseV3): PayloadResponseV2 | PayloadResponseV3 {
   if (!payload) {
     return 'Error occurred, but error payload was not defined';
   }
+  const error = isV3ResponsePayload(payload) ? payload.transformToString() : payload;
   try {
-    const errMsg = JSON.parse(payload.toString()).errorMessage;
+    const errMsg = JSON.parse(error.toString()).errorMessage;
     if (errMsg) {
       return errMsg;
     }
   } catch (e) {}
-  return payload;
+
+  return error;
 }
 
 export abstract class BaseAutotaskClient {