@izara_project/izara-market-library-service-schemas 1.0.18 → 1.0.20

package/src/TemplateManager/src/{flowTag/webSocket/websocketHandler → flowSchema/UploadS3Case/createObject}/functionYaml/data.js

@@ -19,13 +19,11 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 const path = require('path');
 const fs = require('fs');
 
-const { ACTIONS, HANDLER } = require('@izara_project/izara-core-library-service-schemas/src/Consts')
-
-const { firstLetterLowerCase: lowerCase, firstLetterUpperCase: upperCase, firstLetterUpperCase, checkOverWriteGenerateMainFunction } = require("../../../../../../MainLibs/src/Utils")
-const { DEFAULT_HANDLER_PER_ACTION, SOURCE_PATH, defaultIamRolePerAction, createIamRole, RESOURCE_CLASSES, SNS_RESOURCE, resourceNames, DYNAMO_RESOURCE, SAVE_FILE_NAME } = require("../../../../libs/Consts");
-
+const { FLOW_SCHEMA_HOOK_STATE } = require('@izara_project/izara-core-library-service-schemas/src/Consts')
+const { firstLetterUpperCase: upperCase } = require('../../../../../../MainLibs/src/Utils')
+const { SOURCE_PATH, defaultIamRolePerAction, createIamRole, RESOURCE_CLASSES, SNS_RESOURCE, resourceNames, DYNAMO_RESOURCE, SAVE_FILE_NAME, FUNCTION_NAME, shortNameHandler, externalResourceYaml } = require("../../../../libs/Consts");
 const templatePath = path.join(__dirname, "template.ejs")
-
+const hookTemplate = path.join(__dirname, "./hookTemplate.ejs")
 /**
  * receive objectSchema
  * create data for WebScoket handler template
@@ -34,15 +32,16 @@ const templatePath = path.join(__dirname, "template.ejs")
  * @return {{templatePath, templateData,setting}}
  */
 
-function data(_izContext, srcPath) {
-  return [createSourceParams(_izContext, srcPath)]
+function data(_izContext, flowSchema, srcPath) {
+  return [createSourceParams(_izContext, flowSchema, srcPath)]
 }
 
-function createSourceParams(_izContext, srcPath) {
-  let functionName = "FlowSchema"
+function createSourceParams(_izContext, flowSchema, srcPath) {
+  let functionName = upperCase(FUNCTION_NAME.createObjectS3)
   let handlerType = upperCase("hdrWbs")
   let additionalResourcePermission = defaultIamRolePerAction();
   let route = "webSocket"
+
   additionalResourcePermission.push(
     createIamRole(
       {
@@ -51,7 +50,7 @@ function createSourceParams(_izContext, srcPath) {
         ]
       },
       [
-        resourceNames(RESOURCE_CLASSES.sns, "*")
+        resourceNames(RESOURCE_CLASSES.sns, "InImportBatchMainCreate")
       ]
     ),
     createIamRole(
@@ -79,6 +78,34 @@ function createSourceParams(_izContext, srcPath) {
       ]
     )
   )
+
+  if (flowSchema?.hook) {
+    for (const [hookName, hookProperties] of Object.entries(flowSchema.hook)) {
+      if (hookName === FLOW_SCHEMA_HOOK_STATE.beforeCreate) {
+        for (const hookFunction of hookProperties.functionList) {
+          if ((hookFunction.event.hasOwnProperty("initialEvent") && (hookFunction.event.initialEvent === true))) {
+            if (hookFunction?.permission) {
+              for (const permission of hookFunction.permission) {
+                additionalResourcePermission.push(
+                  createIamRole(
+                    permission.action,
+                    permission.resource.map(resource =>
+                      externalResourceYaml(
+                        resource.resource,
+                        resource.resourceName,
+                        resource.serviceTag),
+                      permission.effect
+                    )
+                  )
+                )
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+  // console.log("additionalResourcePermission in createObjectS3", JSON.stringify(additionalResourcePermission))
   return {
     templatePath: templatePath,
     templateData: {
@@ -86,14 +113,16 @@ function createSourceParams(_izContext, srcPath) {
       functionName,
       handlerType,
       additionalResourcePermission,
-      route
+      route,
+      functionNameConfig: upperCase(functionName) + upperCase(shortNameHandler(handlerType))
     },
     setting: {
       savePath: path.join(srcPath, SOURCE_PATH.appYaml),
-      saveFileName: SAVE_FILE_NAME.flowSchema,
+      saveFileName: upperCase(SAVE_FILE_NAME.flowSchema),
       fileExtension: ".yml",
       isAppend: true
     }
   }
 }
+
 module.exports = data;

package/src/TemplateManager/src/flowSchema/UploadS3Case/createObject/functionYaml/template.ejs

@@ -0,0 +1,37 @@
+<%_ const join = require('path').join; _%>
+<%- firstLetterUpperCase(functionNameConfig) %>:
+  handler: <%- join(resourceLocation, `${firstLetterUpperCase(functionName)}_${firstLetterUpperCase(handlerType)}.main`)%>
+  name: ${self:custom.iz_resourcePrefix}<%- firstLetterUpperCase(functionName) %><%- firstLetterUpperCase(handlerType) %>
+  events:
+    - websocket:
+        route: $connect
+    - websocket:
+        route: $disconnect
+    - websocket:
+        route: $default
+    - websocket:
+        route: <%- route %>
+        # authorizer: ${self:custom.iz_authorizerAppLevel}
+      # identifierResource: 'route.request.header.Auth'
+  iamRoleStatements:
+    <%_ additionalResourcePermission.forEach(resourcePermission => { _%>
+    - Effect: <%- resourcePermission.effect %>
+      Action:
+        <%_ Object.keys(resourcePermission.action).forEach(resourcePerAction => { _%>
+        <%_ resourcePermission.action[resourcePerAction].forEach(permission => { _%>
+        - <%- resourcePerAction %>:<%- permission %>
+        <%_}) _%>
+        <%_ }) _%>
+      Resource:
+        <%_ resourcePermission.resource.forEach(resource => { _%>
+        - <%- resource %>
+        <%_ }) _%>
+    <%_}) _%>
+    #<#<%- functionName %><%- handlerType %>IamRole#>
+    #<#/<%- functionName %><%- handlerType %>IamRole#>
+<%_ function firstLetterUpperCase(text){
+  return text.charAt(0).toUpperCase() + text.slice(1)
+} _%>
+<%_ function firstLetterLowerCase(str) {
+  return str.charAt(0).toLowerCase() + str.slice(1)
+} _%>

package/src/TemplateManager/src/flowSchema/UploadS3Case/createObject/handler/data.js

@@ -0,0 +1,75 @@
+/*
+Copyright (C) 2020 Sven Mason <http://izara.io>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as
+published by the Free Software Foundation, either version 3 of the
+License, or (at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+'use strict';
+const path = require('path');
+const fs = require('fs');
+const { FLOW_SCHEMA_HOOK_STATE } = require('@izara_project/izara-core-library-service-schemas/src/Consts')
+const { SOURCE_PATH, FUNCTION_NAME } = require("../../../../libs/Consts");
+const { firstLetterLowerCase: lowerCase, firstLetterUpperCase: upperCase } = require("../../../../../../MainLibs/src/Utils")
+const templatePath = path.join(__dirname, "template.ejs")
+
+/**
+ * receive objectSchema
+ * create data for WebScoket handler template
+ *
+ * @param {Object} objectSchema
+ * @return {{templatePath, templateData,setting}}
+ */
+
+
+function data(_izContext, flowSchema, srcPath) {
+  return [createParamsForCreateSource(_izContext, flowSchema, srcPath)]
+}
+
+function createParamsForCreateSource(_izContext, flowSchema, srcPath) {
+  let functionName = upperCase(FUNCTION_NAME.createObjectS3)
+  let handlerType = "HdrWbs"
+  let hook = false;
+  let hookFunctionName;
+  let isAsync = false;
+  if (flowSchema?.hook) {
+    for (const [hookName, hookProperties] of Object.entries(flowSchema.hook)) {
+      for (const hookFunction of hookProperties.functionList) {
+        if (hookName === FLOW_SCHEMA_HOOK_STATE.beforeCreate) {
+          if (hookFunction.event?.initialEvent && hookFunction.event.initialEvent === true) {
+            hook = true
+            hookFunctionName = hookFunction.name
+            if (hookFunction.event?.isAsync && hookFunction.event.isAsync === true) {
+              isAsync = true
+            }
+          }
+        }
+      }
+    }
+  }
+  return {
+    templatePath: templatePath,
+    templateData: {
+      hook: hook,
+      hookFunctionName: hookFunctionName,
+      isAsync
+    },
+    setting: {
+      savePath: path.join(srcPath, SOURCE_PATH.webSocket),
+      saveFileName: `${functionName}_${handlerType}`,
+      fileExtension: ".js",
+      isAppend: false
+    }
+  }
+}
+module.exports = data

package/src/TemplateManager/src/flowSchema/UploadS3Case/createObject/handler/template.ejs

@@ -0,0 +1,120 @@
+/*
+Copyright (C) 2021 Sven Mason <http://izara.io>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as
+published by the Free Software Foundation, either version 3 of the
+License, or (at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+'use strict';
+
+const izara = require("@izara_project/izara-middleware");
+const middleware = izara.middlewareHandler;
+// const AWS = require('aws-sdk');
+// const api = new AWS.ApiGatewayManagementApi({ endpoint: process.env.iz_webSocketEndpoint });
+const izaraShared = require('@izara_project/izara-shared')
+const dynamodbSharedLib = izaraShared.dynamodbSharedLib
+const callingFlowSharedLib = izaraShared.callingFlowSharedLib;
+const hash = require('object-hash')
+// const coreConsts = require("@izara_project/izara-core-library-core/src/Consts");
+const consts = require('@izara_project/izara-middleware/src/MiddlewareCore/Consts')
+const { postToConnection } = require("../../libs/source/ImportDataLib");
+<% if (hook) { _%>
+const <%- upperCase(hookFunctionName) %> = require("./FlowSchemaHook/HookLogicFunction/source/<%- upperCase(hookFunctionName) %>")
+<% } else { _%>
+const createObject = require('./CreateObjectS3_Main');
+<% } _%>
+module.exports.main = middleware.wrap(async (event, context, callback) => {
+
+  event._izContext.logger.debug('Event:', event);
+
+  try {
+
+    if (event.requestContext) {
+      const route = event.requestContext.routeKey;
+      const connectionId = event.requestContext.connectionId;
+      let correlationId = event._izContext.correlationIds.get(consts.X_CORRELATION_ID)
+
+      switch (route) {
+        case "$connect":
+          console.log("webSocket Connect");
+          break
+        case "$disconnect":
+          console.log("webSocket Disconnect");
+          break
+        case '$default':
+          console.log('Received unknown route:', route);
+          break
+        case 'webSocket':
+          console.log("event in route webSocket", event);
+          let eventParams = JSON.parse(event.body)
+
+          await dynamodbSharedLib.putItem(
+            event._izContext,
+            await dynamodbSharedLib.tableName(event._izContext, "WebSocketTask"),
+            {
+              taskKey: correlationId,
+              connectionId: connectionId,
+            }
+          )
+
+          let callingFlow = eventParams?.callingFlow ? eventParams.callingFlow : {};
+          <% if (hook) { _%>
+            <% if (isAsync) { %>
+              await <%- upperCase(hookFunctionName) %>(
+                event._izContext,
+                eventParams,
+                callingFlowSharedLib.addCallingFlowToPassOnProperties(callingFlow)
+                //(<additionalParams>)
+                  //(</additionalParams>)
+                  )
+                <% } else { %>
+              <%- upperCase(hookFunctionName) %>(
+                event._izContext,
+                eventParams,
+                callingFlowSharedLib.addCallingFlowToPassOnProperties(callingFlow)
+                //(<additionalParams>)
+                //(</additionalParams>)
+              )
+              <% } _%>
+          <% } else { _%>
+          await createObject.createObject(
+            event._izContext,
+            eventParams,
+            callingFlowSharedLib.addCallingFlowToPassOnProperties(callingFlow)
+            //(<additionalParams>)
+            //(</additionalParams>)
+          )
+          <% } _%>
+
+          break
+      }
+    } else {
+      const connId = event.connId;
+      // await postMessage({ message: event.message }, connId);
+      await postToConnection({ message: event.message }, connId);
+    }
+
+    return (izara.response.webSocketSuccess());
+
+  } catch (err) {
+    const connId = event.requestContext.connectionId
+    event._izContext.logger.error('Error, WebSocket: ', err);
+    // await sendErrorMessage({ message: err.message }, connId);
+    await postToConnection({ message: err.message }, connId);
+    return (izara.response.failure(err));
+  }
+});
+<%_ const join = require('path').join _%>
+<%_ function upperCase (text) {
+  return text.charAt(0).toUpperCase() + text.slice(1)
+  } _%>

package/src/TemplateManager/src/{flowTag/webSocket/websocketHandler/handler → flowSchema/UploadS3Case/createObject/mainFunction}/data.js

@@ -19,12 +19,10 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 const path = require('path');
 const fs = require('fs');
 
-const { ACTIONS, HANDLER } = require('@izara_project/izara-core-library-service-schemas/src/Consts')
-
-const { firstLetterLowerCase: lowerCase, firstLetterUpperCase: upperCase, firstLetterUpperCase } = require("../../../../../../MainLibs/src/Utils")
-const { DEFAULT_HANDLER_PER_ACTION, SOURCE_PATH } = require("../../../../libs/Consts");
-
-const templatePath = path.join(__dirname, "template.ejs")
+const { SOURCE_PATH, FUNCTION_NAME } = require("../../../../libs/Consts");
+const { firstLetterLowerCase: lowerCase, firstLetterUpperCase: upperCase } = require("../../../../../../MainLibs/src/Utils")
+const templatePath = path.join(__dirname, "template.ejs");
+const getObjectSchema = require('@izara_project/izara-core-library-service-schemas/src/GetObjectSchema')
 
 /**
  * receive objectSchema
@@ -35,22 +33,20 @@ const templatePath = path.join(__dirname, "template.ejs")
  */
 
 
-function data(_izContext, srcPath) {
-
-  return [createParamsForCreateSource(_izContext, srcPath)]
-
-
+async function data(_izContext, flowSchema, srcPath) {
+  return [await createParamsForCreateSource(_izContext, flowSchema, srcPath)]
 }
 
-function createParamsForCreateSource(_izContext, srcPath) {
-  let functionName = "FlowSchema"
-  let handlerType = "HdrWbs"
+async function createParamsForCreateSource(_izContext, flowSchema, srcPath) {
+  let functionName = upperCase(FUNCTION_NAME.createObjectS3)
+  // let objectSchema = await getObjectSchema.getObjectSchemaCombineFieldNamesWithCache(_izContext, flowSchema.objType)
+  // let fieldNames = Object.keys(objectSchema.fieldNames)
   return {
     templatePath: templatePath,
     templateData: {},
     setting: {
       savePath: path.join(srcPath, SOURCE_PATH.webSocket),
-      saveFileName: `${functionName}_${handlerType}`,
+      saveFileName: `${functionName}_Main`,
       fileExtension: ".js",
       isAppend: false
     }

package/src/TemplateManager/src/{flowTag/webSocket/websocketHandler → flowSchema/UploadS3Case/createObject}/mainFunction/template.ejs

@@ -51,7 +51,7 @@ const { TOPIC_NAME_GENERATE_CODE } = require('@izara_project/izara-market-librar
  * @returns {object} description of return value
  */
 
-module.exports.flowSchema = async (
+module.exports.createObject = async (
   _izContext,
   requestParams,
   callingFlowConfig = {},
@@ -65,7 +65,7 @@ module.exports.flowSchema = async (
     _izContext.logger.debug("WebSocketInvoke callingFlowConfig", callingFlowConfig)
 
     _izContext.correlationIds.set(coreConsts.BASE_USER_ID, "BasDev");
-    _izContext.correlationIds.set(coreConsts.RESOURCE_USE_USER_ID, "BasDevTarget");
+    _izContext.correlationIds.set(coreConsts.TARGET_ID, "BasDevTarget");
 
     let flowSchema = await getObjectSchema.getFlowSchemaS3WithCache(_izContext, {
       flowTag: requestParams.flowTag,
@@ -97,11 +97,11 @@ module.exports.flowSchema = async (
       fieldNames: {
         userId: _izContext.correlationIds.get(coreConsts.BASE_USER_ID),
         // userId: "mockUserId",
-        submittedByUserId: _izContext.correlationIds.get(coreConsts.RESOURCE_USE_USER_ID),
+        submittedByUserId: _izContext.correlationIds.get(coreConsts.TARGET_ID),
         // submittedByUserId: "mockSubmittedByUserId",
         startTime: Date.now(),
         importConfigId: requestParams.importConfigId,
-        importBatchStatus: "process",
+        importBatchStatus: "processing",
         importType: requestParams.importType,
         fileName: requestParams.fileName,
         fileSize: requestParams.fileSize

package/src/TemplateManager/src/{flowTag/webSocket/webSocketComplete → flowSchema/UploadS3Case/createObjectComplete}/functionYaml/data.js

@@ -23,7 +23,7 @@ const templatePath = path.join(__dirname, "./template.ejs")
 
 const { HANDLER, STORAGE_TYPES, ACTIONS } = require('@izara_project/izara-core-library-service-schemas/src/Consts')
 const { firstLetterLowerCase: lowerCase, firstLetterUpperCase: upperCase } = require("../../../../../../MainLibs/src/Utils")
-const { createIamRole, resourceNames, RESOURCE_CLASSES, DEFAULT_LAMBDA_ROLE_PER_ACTION, SQS_RESOURCE, SOURCE_PATH, SAVE_FILE_NAME, FUNCTION_NAME, shortNameHandler, defaultIamRolePerAction, DYNAMO_RESOURCE } = require("../../../../libs/Consts");
+const { createIamRole, resourceNames, RESOURCE_CLASSES, DEFAULT_LAMBDA_ROLE_PER_ACTION, SQS_RESOURCE, SOURCE_PATH, SAVE_FILE_NAME, FUNCTION_NAME, shortNameHandler, defaultIamRolePerAction, DYNAMO_RESOURCE, SHORT_FUNCTION_NAME } = require("../../../../libs/Consts");
 
 
 /**
@@ -38,7 +38,7 @@ function data(_izContext, srcPath) {
 }
 
 function createSourceParams(_izContext, srcPath) {
-  let functionName = "WebSocketTaskComplete";
+  let functionName = upperCase(FUNCTION_NAME.createObjectS3Complete);
   let handlerType = upperCase(HANDLER.hdrSqs);
   let additionalResourcePermission = defaultIamRolePerAction();
 
@@ -62,6 +62,7 @@ function createSourceParams(_izContext, srcPath) {
         [RESOURCE_CLASSES.dynamoDbTable]: [
           DYNAMO_RESOURCE.query,
           DYNAMO_RESOURCE.deleteItem,
+          DYNAMO_RESOURCE.getItem
         ]
       },
       [
@@ -86,11 +87,12 @@ function createSourceParams(_izContext, srcPath) {
       resourceLocation: SOURCE_PATH.resourceLocationWebSocket,
       functionName: functionName,
       handlerType: handlerType,
-      additionalResourcePermission: additionalResourcePermission
+      additionalResourcePermission: additionalResourcePermission,
+      functionNameConfig: upperCase(SHORT_FUNCTION_NAME.createObjectS3Complete) + upperCase(shortNameHandler(handlerType)),
     },
     setting: {
       savePath: path.join(srcPath, SOURCE_PATH.appYaml),
-      saveFileName: SAVE_FILE_NAME.flowSchema,
+      saveFileName: upperCase(SAVE_FILE_NAME.flowSchema),
       fileExtension: ".yml",
       isAppend: true
     }

package/src/TemplateManager/src/flowSchema/UploadS3Case/createObjectComplete/functionYaml/template.ejs

@@ -0,0 +1,30 @@
+<%_ const join = require('path').join; _%>
+<%- firstLetterUpperCase(functionNameConfig) %>:
+  handler: <%- join(resourceLocation, `${functionName}_${firstLetterUpperCase(handlerType)}.main`)%>
+  name: ${self:custom.iz_resourcePrefix}<%- functionName %><%- firstLetterUpperCase(handlerType) %>
+  events:
+    - sqs:
+        arn: arn:aws:sqs:${self:custom.iz_region}:${self:custom.iz_accountId}:${self:custom.iz_resourcePrefix}<%- functionName %><%- firstLetterUpperCase(handlerType) %>
+        batchSize: 10
+        filterPatterns: #**** need to update serverless framwork upper v.2.69.1
+        - body: {"MessageAttributes":{"callingFlow":{"Value":["${self:custom.iz_resourcePrefix}<%- functionName %>"]}}} # functionName of callingFlow
+        - body: {"MessageAttributes":{"callingFlow":{"Value":[{"exists":false}]}}}
+  iamRoleStatements:
+    <%_ additionalResourcePermission.forEach(resourcePermission => { _%>
+    - Effect: <%- resourcePermission.effect %>
+      Action:
+        <%_ Object.keys(resourcePermission.action).forEach(resourcePerAction => { _%>
+        <%_ resourcePermission.action[resourcePerAction].forEach(permission => { _%>
+        - <%- resourcePerAction %>:<%- permission %>
+        <%_ }) _%>
+        <%_ }) _%>
+      Resource:
+        <%_ resourcePermission.resource.forEach(resource => { _%>
+        - "<%- resource %>"
+        <%_ }) _%>
+    <%_}) _%>
+    #<#<%- functionName %><%- handlerType %>IamRole#>
+    #<#/<%- functionName %><%- handlerType %>IamRole#>
+<%_ function firstLetterUpperCase(text){
+  return text.charAt(0).toUpperCase() + text.slice(1)
+} _%>

package/src/TemplateManager/src/{flowTag/webSocket/webSocketComplete → flowSchema/UploadS3Case/createObjectComplete}/handler/data.js

@@ -38,7 +38,7 @@ function data(_izContext, srcPath) {
 }
 
 function createSourceParams(_izContext, srcPath) {
-  let functionName = "WebSocketTaskComplete"
+  let functionName = upperCase(FUNCTION_NAME.createObjectS3Complete)
   let handlerType = upperCase(HANDLER.hdrSqs)
   return {
     templatePath: templatePath,

package/src/TemplateManager/src/{flowTag/webSocket/webSocketComplete → flowSchema/UploadS3Case/createObjectComplete}/handler/template.ejs

@@ -62,25 +62,25 @@ module.exports.main = middleware.wrap(async (event, context, callback) => {
 
       //validate message (and MessageAttributes)
       await recordHandlerSharedLib.validateRecord(
-        record,                             // one record will send to mainFunction
-        "WebSocketTaskCompleteHdrSqs",      // queue name that need to retry or send to dlq
-        perRecordsValidatorSchema,          // schema for record.Message
+        record, // one record will send to mainFunction
+        "WebSocketTaskCompleteHdrSqs", // queue name that need to retry or send to dlq
+        perRecordsValidatorSchema, // schema for record.Message
         // messageAttributeValidatorSchema  // ----- for msgAttr default is null -> do not send this parameter if not want to validate msgAtt
       );
 
       // add argument (to invoke lambda) to passOnProperties[]
       // passOnProperties.push(record.body.Message.MessageAttributes)
-      passOnProperties.push(record.body.Message)
+      passOnProperties.push(record.body.Message.objType)
+      passOnProperties.push(record.body.Message.updateItem)
       // passOnProperties.push(record.body.Message.route)
-      passOnProperties.push(record.body.TopicArn)
       Logger.debug('passOnProperties in handler', passOnProperties);
 
       // call recordHandlerSharedLib.recordHandler with 3 parameters and return promise(resolve)
       let recordPromise = recordHandlerSharedLib.recordHandler(
-        record,                                       // one record will send to mainFunction
-        webSocketTaskComplete.webSocketTaskComplete,  // mainFunction that need to invoke.
-        "WebSocketTaskCompleteHdrSqs",                // queue name that need to retry or send to dlq
-        passOnProperties,                             // all parameters that mainFunction needed.
+        record, // one record will send to mainFunction
+        webSocketTaskComplete.webSocketTaskComplete, // mainFunction that need to invoke.
+        "WebSocketTaskCompleteHdrSqs", // queue name that need to retry or send to dlq
+        passOnProperties, // all parameters that mainFunction needed.
       );
       record._izContext.logger.debug('after recordPromise in handler');
       recordPromises.push(recordPromise); // push promise to recordPromises
@@ -100,4 +100,4 @@ module.exports.main = middleware.wrap(async (event, context, callback) => {
     Logger.error('Unhandled Error, WebSocketTaskCompleteHdrSqs: ', err);
     throw (err);
   }
-});
+});

package/src/TemplateManager/src/{flowTag/webSocket/webSocketComplete → flowSchema/UploadS3Case/createObjectComplete}/mainFunction/data.js

@@ -21,9 +21,8 @@ const path = require('path');
 
 const templatePath = path.join(__dirname, "./template.ejs")
 
-const { HANDLER, STORAGE_TYPES, ACTIONS } = require('@izara_project/izara-core-library-service-schemas/src/Consts')
-const { firstLetterLowerCase: lowerCase, firstLetterUpperCase: upperCase } = require("../../../../../../MainLibs/src/Utils")
-const { createIamRole, resourceNames, RESOURCE_CLASSES, DEFAULT_LAMBDA_ROLE_PER_ACTION, SQS_RESOURCE, SOURCE_PATH, SAVE_FILE_NAME, FUNCTION_NAME, shortNameHandler, defaultIamRolePerAction, DYNAMO_RESOURCE } = require("../../../../libs/Consts");
+const { SOURCE_PATH, FUNCTION_NAME } = require("../../../../libs/Consts");
+const { firstLetterUpperCase: upperCase } = require('../../../../../../MainLibs/src/Utils')
 
 
 /**
@@ -38,7 +37,7 @@ function data(_izContext, srcPath) {
 }
 
 function createSourceParams(_izContext, srcPath) {
-  let functionName = "WebSocketTaskComplete"
+  let functionName = upperCase(FUNCTION_NAME.createObjectS3Complete)
   return {
     templatePath: templatePath,
     templateData: {

package/src/TemplateManager/src/flowSchema/UploadS3Case/createObjectComplete/mainFunction/template.ejs

@@ -0,0 +1,86 @@
+/*
+Copyright (C) 2020 Sven Mason <http://izara.io>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as
+published by the Free Software Foundation, either version 3 of the
+License, or (at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+'use strict';
+const { NoRetryError } = require('@izara_project/izara-core-library-core');
+const izaraShared = require('@izara_project/izara-shared');
+const dynamodbSharedLib = izaraShared.dynamodbSharedLib;
+const snsSharedLib = izaraShared.snsSharedLib;
+const asyncFlowSharedLib = izaraShared.asyncFlowSharedLib;
+let coreConsts = require('@izara_project/izara-middleware/src/MiddlewareCore/Consts')
+
+const { postToConnection } = require('../../libs/source/ImportDataLib')
+module.exports.webSocketTaskComplete = async (
+  _izContext,
+  objType,
+  updateItem
+) => {
+  try {
+    _izContext.logger.debug("requestParams", {
+      objType,
+      updateItem
+    })
+
+    let connectionId = _izContext.correlationIds.get(coreConsts.CONNECTION_ID);
+    let correlationId = _izContext.correlationIds.get(coreConsts.X_CORRELATION_ID);
+
+    if (updateItem.importBatchStatus === "waitingToUpload") {
+      // get upload url
+      let webSocketTask = await dynamodbSharedLib.getItem(_izContext,
+        await dynamodbSharedLib.tableName(_izContext, "WebSocketTask"),
+        {
+          taskKey: correlationId,
+          connectionId: connectionId
+        }
+      )
+      _izContext.logger.debug("webSocketTask", webSocketTask);
+
+
+      _izContext.logger.debug("post message to client")
+      // post message to client
+      await postToConnection({
+        message: {
+          uploadUrl: webSocketTask.uploadUrl,
+          importBatchId: updateItem.importBatchId
+        }
+      }, connectionId)
+
+      await dynamodbSharedLib.deleteItem(_izContext,
+        await dynamodbSharedLib.tableName(_izContext, "WebSocketTask"),
+        {
+          taskKey: correlationId,
+          connectionId: connectionId
+        }
+      )
+    } else {
+      await postToConnection({
+        message: "status is processing can't upload file"
+      }, connectionId)
+      // delete webSocketTask
+      await dynamodbSharedLib.deleteItem(_izContext,
+        await dynamodbSharedLib.tableName(_izContext, "WebSocketTask"),
+        {
+          taskKey: correlationId,
+          connectionId: connectionId
+        }
+      )
+
+    }
+  } catch (err) {
+    _izContext.logger.error('error websocketTaskComplete: ', err)
+    throw (err);
+  }
+};