@ikonintegration/ikapi 2.6.1 → 2.6.2-alpha3

package/index.js

@@ -29,6 +29,8 @@ module.exports = {
   IKSuccessNoContentResponse: require("./src/API/IKResponse.js").IKSuccessNoContentResponse,
   IKBadRequestResponseWithRollback: require("./src/API/IKResponse.js").IKBadRequestResponseWithRollback,
   IKStepFunctionResponse: require("./src/API/IKResponse.js").IKStepFunctionResponse,
+  IKSimpleResponse: require("./src/API/IKResponse.js").IKSimpleResponse,
+  
   //Helpers
   Utils: require("./src/API/IKUtils.js").default,
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ikonintegration/ikapi",
-  "version": "2.6.1",
+  "version": "2.6.2-alpha3",
   "description": "",
   "main": "index.js",
   "module": "main.js",

package/src/API/IKResponse.js

@@ -21,40 +21,38 @@ export default class IKResponse {
   getBody() { return this._body; }
   appendIntoBody(key, value) { this._body[key] = value; }
   appendHeader(key, value) { this._headers[key] = value; }
-  build(context, requestID, isBatch, transaction) {
+  async build(context, transaction, isBatch) {
     //Stream support
     if (this._streamingOut) return;
-    if (this._isStream) return this._pipe(context, requestID);
-    //Raw response support
-    if (this._contextRawBody) return this.rawContext(context, requestID, transaction);
+    if (this._isStream) return this._pipe(context);
     
     //append default fields
-    if (requestID && this._body) this.appendIntoBody('transactionID', requestID); //append transaction ID
+    if (transaction.request.getRequestID() && this._body) this.appendIntoBody('transactionID', transaction.request.getRequestID()); //append transaction ID
+    //Raw response support
+    if (this._contextRawBody) return this.rawContext(context, transaction);
+
     //build response
     let b = {
-      statusCode: this._statusCode,
+      statusCode: this._statusCode, headers: this._headers,
       ...(this._body ? { body: JSON.stringify(this._body) } : {}),
-      headers: this._headers
     };
     //log response and respond to context
     transaction.logger.debug(b)
+    //Check for transaction response proxy
+    if (transaction.responseProxy) await transaction.responseProxy(b);
+    //Batch does not succeed directly just on upper transaction (which will should be a batch)
     if (!isBatch) context.succeed(b);
   }
-  _pipe(context, requestID) {
+  _pipe(context) {
     //Check if not streaming
     if (this._streamingOut) return;
     this._streamingOut = true;
     //build response
-    let b = {
-      statusCode: this._statusCode,
-      body: this._body, headers: this._headers
-    };
+    let b = { statusCode: this._statusCode, body: this._body, headers: this._headers };
     //log response and respond to context
     context.succeed(b);
   }
-  rawContext(context, requestID, transaction) {
-    //append default fields
-    if(requestID && this._body) this.appendIntoBody('transactionID', requestID); //append transaction ID
+  rawContext(context, transaction) {
     //log response and respond to context
     transaction.logger.debug(this._body)
     if (this.getCode() <= 200 && this.getCode() <= 299) context.succeed(this._body);
@@ -114,3 +112,7 @@ export function IKStepFunctionResponse(body, optionalCode) {
   resp._throwOnErrors = true;
   return resp;
 }
+export function IKSimpleResponse(body, optionalCode) {
+  const resp = new IKResponse(optionalCode || 200, body);
+  return resp;
+}

package/src/BaseEvent/IKTransaction.js

@@ -23,8 +23,10 @@ export default class IKTransaction {
     //queue support
     this._isBatch = _isBatch;
     this._resp = null;
-    //
-    this._retrowErrors = _retrowErrors;
+    //Step function support
+    this._retrowErrors = _retrowErrors; /* retrow internal errors */
+    //When set, this will be called with the response context right before calling the context suceed/fail
+    this.responseProxy = null;
     //
     this.logger = new IKLogger({/*COFIG S3/SQS HERE*/}, Utils.logLevel(), (context.awsRequestId ? context.awsRequestId : (event.requestContext ? event.requestContext.requestId : 'unknown')));
     this.request = new IKRequest(this._event, this._context, this);
@@ -52,11 +54,11 @@ export default class IKTransaction {
       this._resp = await executionFunc(this);
       //Answer client
       if (this._resp && this._resp instanceof IKResponse) {
-        this._resp.build(this._context, this.request.getRequestID(), this._isBatch, this);
+        await this._resp.build(this._context, this, this._isBatch);
         executionFailed = !!(this._resp.getBody() && this._resp.getBody().rollback);
       } else {
         this._resp = this._getErrorResponse(IKGlobals.ErrorResponseInvalidServerResponse, IKGlobals.ErrorCode_APIError)
-        this._resp.build(this._context, this.request.getRequestID(), this._isBatch, this);
+        await this._resp.build(this._context, this, this._isBatch);
         this.logger.error("Invalid response object from main request code.");
       }
     } catch (e) { /*EXECUTION FAIL*/
@@ -67,7 +69,7 @@ export default class IKTransaction {
       //envelope exception?
       if (executionFailed) {
         this._resp = this._getErrorResponse(IKGlobals.ErrorResponseUnhandledError, IKGlobals.ErrorCode_APIError);
-        this._resp.build(this._context, this.request.getRequestID(), this._isBatch, this);
+        await this._resp.build(this._context, this, this._isBatch);
       }
     } return executionFailed;
   }