Websocket

.gitignore

@@ -104,6 +104,7 @@ fabric.properties
 # auto-generated tag files
 tags
 =======
-
-.idea/
-manual_test/.serverless
+
+.idea/
+.serverless
+.dynamodb

README.md

@@ -28,6 +28,7 @@ This plugin is updated by its users, I just do maintenance and ensure that PRs a
 * [Custom headers](#custom-headers)
 * [Environment variables](#environment-variables)
 * [AWS API Gateway features](#aws-api-gateway-features)
+* [WebSocket](#websocket)
 * [Usage with Webpack](#usage-with-webpack)
 * [Velocity nuances](#velocity-nuances)
 * [Debug process](#debug-process)
@@ -75,6 +76,7 @@ All CLI options are optional:
 --location              -l  The root location of the handlers' files. Defaults to the current directory
 --host                  -o  Host name to listen on. Default: localhost
 --port                  -P  Port to listen on. Default: 3000
+--wsPort                    WebSocket port to listen on. Default: 3001
 --stage                 -s  The stage used to populate your templates. Default: the first stage found in your project.
 --region                -r  The region used to populate your templates. Default: the first region for the first stage found.
 --noTimeout             -t  Disables the timeout feature.
@@ -358,6 +360,30 @@ resources:
 
 To disable the model validation you can use `--disableModelValidation`.
 
+## WebSocket
+`This has experimental functionality. Please report any bugs or missing features.`
+
+serverless-offline suports running a WebSocket local endpoint.
+
+Usage in order to send messages back to clients:
+
+`POST http://localhost:{port+1}/@connections/{connectionId}`
+
+Or,
+
+```
+let endpoint=event.apiGatewayUrl;
+if (!endpoint) endpoint = event.requestContext.domainName+'/'+event.requestContext.stage;
+const apiVersion='2018-11-29';
+const apiGM=new API.ApiGatewayManagementApi({ apiVersion, endpoint });
+apiGM.postToConnection({ConnectionId, Data});
+```
+Where the `event` is received in the lambda handler function.
+
+There's support for `websocketsApiRouteSelectionExpression` in it's basic form: `$request.body.x.y.z`, where the default value is `$request.body.action`.
+
+Authorizers and wss:// are currectly not supported in serverless-offline.
+
 ## Usage with Webpack
 
 Use [serverless-webpack](https://github.com/serverless-heaven/serverless-webpack) to compile and bundle your ES-next code

manual_test_websocket/README.md

@@ -8,6 +8,7 @@ Set AWS credentials, e.g.: `export AWS_PROFILE=...`
 
 To start AWS DynamoDB locally (can run only after first deploying locally): `sls dynamodb install` `sls dynamodb start`
 
+### In either main or RouteSelection folder the following:
 
 ## Deploying locally
 
@@ -26,28 +27,24 @@ To start AWS DynamoDB locally (can run only after first deploying locally): `sls
 
 ## Testing on AWS
 
-`npm --endpoint={WebSocket endpoint URL on AWS} run test`
+`npm --endpoint={WebSocket endpoint URL on AWS} --timeout={timeout in ms} run test`
 
 
-## Usage Assumption - In order to send messages back to clients
-`const newAWSApiGatewayManagementApi=(event, context)=>{`
+## Usage in order to send messages back to clients
+
+`POST http://localhost:3001/@connections/{connectionId}`
 
-`  const endpoint=event.requestContext.domainName+'/'+event.requestContext.stage;`
+Or,
 
-`  const apiVersion='2018-11-29';`
+`let endpoint=event.apiGatewayUrl;`
 
-`  let API=context.API;`
+`if (!endpoint) endpoint = event.requestContext.domainName+'/'+event.requestContext.stage;`
 
-`  if (!process.env.IS_OFFLINE) {`
+`const apiVersion='2018-11-29';`
 
-`    API = require('aws-sdk');`
+`const apiGM=new API.ApiGatewayManagementApi({ apiVersion, endpoint });`
 
-`    require('aws-sdk/clients/apigatewaymanagementapi');`
+`apiGM.postToConnection({ConnectionId, Data});`
 
-`  }`
-
-`  return new API.ApiGatewayManagementApi({ apiVersion, endpoint });`
-
-`};`
 
 

manual_test_websocket/manual_test_websocket_RouteSelection/handler.js

@@ -0,0 +1,37 @@
+'use strict';
+
+const AWS = require('aws-sdk');
+
+
+const successfullResponse = {
+  statusCode: 200,
+  body: 'Request is OK.'
+};
+
+const errorResponse = {
+  statusCode: 400,
+  body: 'Request is not OK.'
+};
+
+module.exports.echo = async (event, context) => {
+  const action = JSON.parse(event.body);
+
+  await sendToClient(action.message, event.requestContext.connectionId, newAWSApiGatewayManagementApi(event, context));
+  return successfullResponse; 
+};
+
+const newAWSApiGatewayManagementApi=(event, context)=>{
+  let endpoint=event.apiGatewayUrl;
+
+  if (!endpoint) endpoint = event.requestContext.domainName+'/'+event.requestContext.stage;
+  const apiVersion='2018-11-29';
+  return new AWS.ApiGatewayManagementApi({ apiVersion, endpoint });
+};
+
+const sendToClient = (data, connectionId, apigwManagementApi) => {
+  // console.log(`sendToClient:${connectionId}`);
+  let sendee=data;
+  if ('object'==typeof data) sendee=JSON.stringify(data);
+
+  return apigwManagementApi.postToConnection({ConnectionId: connectionId, Data: sendee}).promise();
+};