@izara_project/izara-market-library-service-schemas 1.0.15 → 1.0.16

package/src/TemplateManager/src/PerActionEndpoint/FunctionYaml/HdrSqs/data.js

@@ -21,7 +21,7 @@ const fs = require('fs');
 
 const { ACTIONS, HANDLER, STORAGE_TYPES } = require('@izara_project/izara-core-library-service-schemas/src/Consts')
 
-const { firstLetterLowerCase: lowerCase, firstLetterUpperCase: upperCase, renameObjectType } = require("../../../../../MainLibs/src/Utils")
+const { firstLetterLowerCase: lowerCase, firstLetterUpperCase: upperCase } = require("../../../../../MainLibs/src/Utils")
 const { DEFAULT_HANDLER_PER_ACTION, createIamRole, resourceNames, RESOURCE_CLASSES, DEFAULT_LAMBDA_ROLE_PER_ACTION, SQS_RESOURCE, SOURCE_PATH, SAVE_FILE_NAME, SNS_RESOURCE, FUNCTION_NAME, DYNAMO_RESOURCE, defaultIamRolePerAction, awaitingMultipleStepsRole, shortNameHandler } = require("../../../libs/Consts");
 
 const templatePath = path.join(__dirname, "./template.ejs")
@@ -35,13 +35,12 @@ const templatePath = path.join(__dirname, "./template.ejs")
  */
 function data(_izContext, objectSchema, srcPath) {
   let createSourceArray = [];
-  let newObjectSchema = renameObjectType(objectSchema, srcPath)
   for (const action of Object.values(ACTIONS)) {
     if (objectSchema.overWriteHandlers?.[action]) {
       // if have overWriteHander of action check hdrApi inside
       if (objectSchema.overWriteHandlers[action].includes(HANDLER.hdrSqs)) {
         // create template data
-        const createSourceParam = createParamForCreateSource(newObjectSchema, action, srcPath)
+        const createSourceParam = createParamForCreateSource(objectSchema, action, srcPath)
         createSourceArray.push(createSourceParam)
       }
     } else {
@@ -49,7 +48,7 @@ function data(_izContext, objectSchema, srcPath) {
       const defaultHandler = DEFAULT_HANDLER_PER_ACTION[action]
       if (defaultHandler.includes(HANDLER.hdrSqs)) {
         // create template data and object for create source
-        const createSourceParam = createParamForCreateSource(newObjectSchema, action, srcPath)
+        const createSourceParam = createParamForCreateSource(objectSchema, action, srcPath)
         createSourceArray.push(createSourceParam)
       }
     }
@@ -150,6 +149,17 @@ function createParamForCreateSource(objectSchema, action, srcPath) {
     //     ]
     //   )
     // )
+  } else if (action === ACTIONS.get) {
+    additionalResourcePermission.push(
+      createIamRole(
+        {
+          [RESOURCE_CLASSES.sns]: [SNS_RESOURCE.publish]
+        },
+        [
+          resourceNames(RESOURCE_CLASSES.sns, "OutGetNodeComplete")
+        ]
+      )
+    )
   }
 
   // console.log("additionalResourcePermissionBeforeReturn", additionalResourcePermission)

package/src/TemplateManager/src/PerActionEndpoint/FunctionYaml/HdrWbs/data.js

@@ -0,0 +1,171 @@
+/*
+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 { ACTIONS, HANDLER, STORAGE_TYPES } = require('@izara_project/izara-core-library-service-schemas/src/Consts')
+
+const { firstLetterLowerCase: lowerCase, firstLetterUpperCase: upperCase } = require("../../../../../MainLibs/src/Utils")
+const { DEFAULT_HANDLER_PER_ACTION, createIamRole, resourceNames, RESOURCE_CLASSES, DEFAULT_LAMBDA_ROLE_PER_ACTION, SQS_RESOURCE, SOURCE_PATH, SAVE_FILE_NAME, SNS_RESOURCE, FUNCTION_NAME, DYNAMO_RESOURCE, defaultIamRolePerAction, awaitingMultipleStepsRole, shortNameHandler } = require("../../../libs/Consts");
+
+const templatePath = path.join(__dirname, "./template.ejs")
+
+/**
+ * receive objectSchema
+ * create data for hdrInv template
+ *
+ * @param {Object} objectSchema
+ * @return {{ templatePath, templateData, setting } }
+ */
+function data(_izContext, objectSchema, srcPath) {
+  let createSourceArray = [];
+  for (const action of Object.values(ACTIONS)) {
+    if (objectSchema.overWriteHandlers?.[action]) {
+      // if have overWriteHander of action check hdrApi inside
+      if (objectSchema.overWriteHandlers[action].includes(HANDLER.hdrWbs)) {
+        // create template data
+        const createSourceParam = createParamForCreateSource(objectSchema, action, srcPath)
+        createSourceArray.push(createSourceParam)
+      }
+    } else {
+      // if not have overWriteHander of action will use default handler
+      const defaultHandler = DEFAULT_HANDLER_PER_ACTION[action]
+      if (defaultHandler.includes(HANDLER.hdrWbs)) {
+        // create template data and object for create source
+        const createSourceParam = createParamForCreateSource(objectSchema, action, srcPath)
+        createSourceArray.push(createSourceParam)
+      }
+    }
+  }
+  return createSourceArray;
+}
+
+
+
+function createParamForCreateSource(objectSchema, action, srcPath) {
+  let objectType = objectSchema.objectType;
+  let actionHandler = action;
+  let functionName = upperCase(objectType) + upperCase(actionHandler);
+  let additionalResourcePermission = defaultIamRolePerAction();
+  let tableForGetItem = [];
+  let hasGraphStorage = false;
+  let functionNameObjectType = objectSchema?.shortNameObjectType ? objectSchema.shortNameObjectType : objectSchema.objectType
+  let functionNameConfig = upperCase(functionNameObjectType) + upperCase(action) + upperCase(shortNameHandler(HANDLER.hdrWbs))
+
+  for (let storageResource of Object.values(objectSchema.storageResources)) {
+    if (storageResource.storageType === STORAGE_TYPES.dynamoDB) {
+      // add IamRole for dynamodb depend on tableName in storageResources of objectSchema
+      tableForGetItem.push(resourceNames(RESOURCE_CLASSES.dynamoDbTable, storageResource.tableName));
+
+    } else if (storageResource.storageType === STORAGE_TYPES.graph) {
+      // find graph serviceName from graphServerTag or *GetNodeHdrInv ?
+
+      if (hasGraphStorage) {
+        continue;
+      } else {
+        hasGraphStorage = true;
+      }
+    }
+  }
+  additionalResourcePermission.push(awaitingMultipleStepsRole())
+
+  additionalResourcePermission.push(
+    createIamRole(
+      {
+        [RESOURCE_CLASSES.dynamoDbTable]: [
+          DYNAMO_RESOURCE.getItem,
+          DYNAMO_RESOURCE.putItem,
+          DYNAMO_RESOURCE.deleteItem
+        ]
+      },
+      [
+        resourceNames(RESOURCE_CLASSES.dynamoDbTable, "WebSocketTask")
+      ]
+    )
+  )
+
+  additionalResourcePermission.push(
+    createIamRole(
+      {
+        ["execute-api"]: ["ManageConnections"]
+      },
+      [
+        resourceNames(RESOURCE_CLASSES.webSocket)
+      ]
+    )
+  )
+
+
+  if (action == ACTIONS.update) {
+    additionalResourcePermission.push(
+      createIamRole(
+        {
+          [RESOURCE_CLASSES.sns]: [SNS_RESOURCE.publish]
+        },
+        [
+          resourceNames(RESOURCE_CLASSES.sns, "Out" + upperCase(FUNCTION_NAME.updateNodeComplete))
+        ]
+      )
+    )
+  } else if (action == ACTIONS.create) {
+    additionalResourcePermission.push(
+      createIamRole(
+        {
+          [RESOURCE_CLASSES.sns]: [SNS_RESOURCE.publish]
+        },
+        [
+          resourceNames(RESOURCE_CLASSES.sns, "Out" + upperCase(FUNCTION_NAME.createObjectComplete))
+        ]
+      )
+    )
+  } else if (action == ACTIONS.delete) {
+    // additionalResourcePermission.push(
+    //   createIamRole(
+    //     {
+    //       [RESOURCE_CLASSES.sns]: [SNS_RESOURCE.publish]
+    //     },
+    //     [
+    //       resourceNames(RESOURCE_CLASSES.sns, "Out" + upperCase(FUNCTION_NAME.createObjectComplete))
+    //     ]
+    //   )
+    // )
+  }
+
+  // console.log("additionalResourcePermissionBeforeReturn", additionalResourcePermission)
+  return {
+    templatePath: templatePath,
+    templateData: {
+      functionName: functionName,
+      actionHandler: actionHandler,
+      resourceLocation: SOURCE_PATH.resourceLocationPerAction,
+      handlerType: HANDLER.hdrWbs,
+      additionalResourcePermission,
+      functionNameConfig,
+      objectType
+    },
+    setting: {
+      savePath: path.join(srcPath, SOURCE_PATH.appYaml),
+      saveFileName: SAVE_FILE_NAME.functionPerActionYaml,
+      fileExtension: '.yml',
+      isAppend: true
+    }
+  }
+}
+
+module.exports = data;

package/src/TemplateManager/src/PerActionEndpoint/FunctionYaml/HdrWbs/request.json

@@ -0,0 +1,19 @@
+{
+  "functionName": "xxx",
+  "handler": "hdrSqs",
+  "event": {
+    "sqs": {
+      "arn": "queueName"
+    },
+    "iamRoleStatement": {
+      "Effect": "Allow",
+      "Action": {
+        "xxx": [
+          "yyy",
+          "xxx"
+        ]
+      },
+      "Resource": "xxxx"
+    }
+  }
+}

package/src/TemplateManager/src/PerActionEndpoint/FunctionYaml/HdrWbs/template.ejs

@@ -0,0 +1,29 @@
+<%_ 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: <%- objectType -%><%- firstLetterUpperCase(actionHandler) %>
+  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/PerActionEndpoint/FunctionYaml/webSocketConnect/data.js

@@ -0,0 +1,60 @@
+/*
+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 { SOURCE_PATH, SAVE_FILE_NAME, ACTIONS, HANDLER } = require("../../../libs/Consts");
+
+const templatePath = path.join(__dirname, "./template.ejs")
+
+/**
+ * receive objectSchema
+ * create data for websocket connect function yaml template
+ *
+ * @param {Object} objectSchema
+ * @return {{ templatePath, templateData, setting } }
+ */
+
+function data(_izContext, webSocketStatus, srcPath) {
+  let createSourceParam;
+  if (webSocketStatus) {
+    createSourceParam = [createWebSocketConnectYaml(_izContext, srcPath)]
+  } else {
+    createSourceParam = []
+  }
+  return createSourceParam
+}
+
+function createWebSocketConnectYaml(_izContext, srcPath) {
+  let functionName = "WebSocketConnect"
+  return {
+    templatePath: templatePath,
+    templateData: {
+      functionName,
+      resourceLocation: SOURCE_PATH.resourceLocationPerAction
+    },
+    setting: {
+      savePath: path.join(srcPath, SOURCE_PATH.appYaml),
+      saveFileName: SAVE_FILE_NAME.functionPerActionYaml,
+      fileExtension: ".yml",
+      isAppend: true
+    }
+  }
+}
+module.exports = data

package/src/TemplateManager/src/PerActionEndpoint/FunctionYaml/webSocketConnect/template.ejs

@@ -0,0 +1,10 @@
+WebSocketConnect:
+  handler: src/generatedCode/ObjectTypePerActionEndpoint/source/WebSocketConnect.main
+  name: ${self:custom.iz_resourcePrefix}WebSocketConnect
+  events:
+    - websocket:
+        route: $connect
+    - websocket:
+        route: $disconnect
+#<#<WebSocketConnectIamRole#>
+#<#/<WebSocketConnectIamRole#>

package/src/TemplateManager/src/PerActionEndpoint/Handler/WebSocket/data.js

@@ -0,0 +1,76 @@
+/*
+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 { 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 } = require("../../../libs/Consts");
+
+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, objectSchema, srcPath) {
+  let createSourceArray = [];
+  for (const action of Object.values(ACTIONS)) {
+    if (objectSchema.overWriteHandlers?.[action]) {
+      if (objectSchema.overWriteHandlers[action].includes(HANDLER.hdrWbs)) {
+        const createSourceParams = createParamsForCreateSource(objectSchema, action, HANDLER.hdrWbs, srcPath)
+        createSourceParams && createSourceArray.push(createSourceParams);
+      }
+    } else {
+      if (DEFAULT_HANDLER_PER_ACTION[action].includes(HANDLER.hdrWbs)) {
+        const createSourceParams = createParamsForCreateSource(objectSchema, action, HANDLER.hdrWbs, srcPath)
+        createSourceParams && createSourceArray.push(createSourceParams);
+      }
+    }
+  }
+  return createSourceArray;
+}
+
+function createParamsForCreateSource(objectSchema, action, handler, srcPath) {
+  let objectType = objectSchema.objectType;
+  let functionName = objectType + upperCase(action)
+
+  return {
+    templatePath: templatePath,
+    templateData: {
+      functionName,
+      objectType,
+      action,
+      isCreateMainFunction: checkOverWriteGenerateMainFunction(objectSchema, action)
+    },
+    setting: {
+      savePath: path.join(srcPath, SOURCE_PATH.lambdaPerAction),
+      saveFileName: `${upperCase(objectType)}${upperCase(action)}_${upperCase(handler)}`,
+      fileExtension: ".js",
+      isAppend: false
+    }
+  }
+}
+module.exports = data

package/src/TemplateManager/src/PerActionEndpoint/Handler/WebSocket/template.ejs

@@ -0,0 +1,141 @@
+/*
+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 { validateSchemaPerWebSocket } = require('@izara_project/izara-market-library-service-schemas/src/GenerateCodeLibs/src/GenerateCodeLibs')
+<%_ if (isCreateMainFunction == true) { _%>
+  // user generate main function
+  <%_ } else { %>
+ const <%- functionName %> =require('./<%- firstLetterUpperCase(objectType) %><%- firstLetterUpperCase(action) %>_Main')
+  <%_ } _%>
+
+module.exports.main = middleware.wrap(async (event, context, callback) => {
+
+  event._izContext.logger.debug('Event:', event);
+
+
+
+  try {
+
+    let requestParams = JSON.parse(event.body);
+    await validateSchemaPerWebSocket(
+      event._izContext,
+      requestParams,
+      "<%- objectType -%>",
+      "<%- action -%>",
+    )
+
+    if (event.requestContext) {
+      const route = event.requestContext.routeKey;
+      const connectionId = event.requestContext.connectionId;
+      const attribute = event.body;
+      event._izContext.logger.debug('body:', attribute);
+      event._izContext.logger.debug('route:', route);
+      event.body = JSON.parse(attribute)
+
+
+
+      switch (route) {
+        case '$default':
+          console.log('Received unknown route:', route)
+          break
+        case '<%- objectType -%><%- firstLetterUpperCase(action) %>':
+
+          // create Web socket task for respond
+          await dynamodbSharedLib.putItem(
+            event._izContext,
+            await dynamodbSharedLib.tableName(event._izContext, "WebSocketTask"),
+            {
+              taskKey: "<%- objectType -%>" + "<%- firstLetterUpperCase(action) %>", // objType + action
+              connectionid: connectionId,
+              // connectionid: event.body.connectionid,
+              //(<additionalParams>)
+              //(</additionalParams>)
+            }
+          );
+
+          let payloadBeforeInvokeFunction = requestParams
+          // delete action before send to main function
+          delete payloadBeforeInvokeFunction.action
+          console.log("payloadBeforeInvokeFunction", payloadBeforeInvokeFunction)
+          event._izContext.logger.debug("event before send to <%- functionName %>Main", requestParams)
+          //(<beforeInvokeMain>)
+          //(</beforeInvokeMain>)
+          // invoke main
+          let invokeUpdateMainFunction = await <%- functionName %>.<%- functionName %>Main(
+            event._izContext,
+            payloadBeforeInvokeFunction,
+            callingFlowSharedLib.addCallingFlowToPassOnProperties(attribute)
+          );
+          //(<afterInvokeMain>)
+          //(</afterInvokeMain>)
+
+          await postMessage(invokeUpdateMainFunction, connectionId);
+
+          break
+
+
+      }
+    } else {
+      console.log("not have event.requestContext")
+      const connId = event.connId;
+      await postMessage({ message: event.message }, connId);
+    }
+
+
+  } catch (err) {
+    console.log("event", event)
+    const connId = event.requestContext.connectionId
+    event._izContext.logger.error('Error, WebSocket: ', err);
+    await sendErrorMessage({ message: err.message }, connId);
+    return (izara.response.failure(err));
+  }
+});
+
+
+async function postMessage(data, connectionId) {
+  const params = {
+    ConnectionId: connectionId,
+    Data: JSON.stringify(data),
+    // Data: Buffer.from(JSON.stringify(data)),
+  }
+  return api.postToConnection(params).promise();
+};
+
+async function sendErrorMessage(errorMessage, connectionId) {
+  const params = {
+    ConnectionId: connectionId,
+    Data: JSON.stringify(errorMessage)
+  }
+  return api.postToConnection(params).promise();
+}
+
+<%_ 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/PerActionEndpoint/Handler/WebSocketConnect/data.js

@@ -0,0 +1,56 @@
+/*
+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 path = require('path');
+const fs = require('fs');
+
+const { SOURCE_PATH, SAVE_FILE_NAME, ACTIONS, HANDLER } = require("../../../libs/Consts");
+
+const templatePath = path.join(__dirname, "./template.ejs")
+
+/**
+ * receive objectSchema
+ * create data for websocket connect function yaml template
+ *
+ * @param {Object} objectSchema
+ * @return {{ templatePath, templateData, setting } }
+ */
+
+function data(_izContext, webSocketStatus, srcPath) {
+  let createSourceParam;
+  if (webSocketStatus) {
+    createSourceParam = [createWebSocketConnectYaml(_izContext, srcPath)]
+  } else {
+    createSourceParam = []
+  }
+  return createSourceParam
+}
+
+function createWebSocketConnectYaml(_izContext, srcPath) {
+  return {
+    templatePath: templatePath,
+    templateData: {},
+    setting: {
+      savePath: path.join(srcPath, SOURCE_PATH.lambdaPerAction),
+      saveFileName: "WebSocketConnect",
+      fileExtension: ".js",
+      isAppend: true
+    }
+  }
+}
+module.exports = data

package/src/TemplateManager/src/PerActionEndpoint/Handler/WebSocketConnect/template.ejs

@@ -0,0 +1,55 @@
+/*
+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 });
+
+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;
+
+      switch (route) {
+        case '$connect':
+          console.log('Connect websocket')
+          break
+        case '$disconnect':
+          console.log('Websocket are Disconnection')
+          break
+        case '$default':
+          console.log('Received unknown route:', route)
+          break
+      }
+    }
+
+
+    return (izara.response.webSocketSuccess());
+
+  } catch (err) {
+    event._izContext.logger.error('Error, WebSocketConnect: ', err);
+    return (izara.response.failure(err));
+  }
+});