Websocket event schema

lib/plugins/aws/package/compile/events/websockets/index.js

@@ -12,6 +12,9 @@ const compileDeployment = require('./lib/deployment');
 const compileStage = require('./lib/stage');
 const compileAuthorizers = require('./lib/authorizers');
 
+const standardRoutes = new Set(['connect', 'disconnect', 'default']);
+const routePattern = new RegExp(`^(?:[^$].*|\\$(${Array.from(standardRoutes).join('|')}))$`, 'i');
+
 class AwsCompileWebsockets {
   constructor(serverless, options) {
     this.serverless = serverless;
@@ -51,9 +54,63 @@ class AwsCompileWebsockets {
       },
     };
 
-    // TODO: Complete schema, see https://github.com/serverless/serverless/issues/8019
     this.serverless.configSchemaHandler.defineFunctionEvent('aws', 'websocket', {
-      anyOf: [{ type: 'string' }, { type: 'object' }],
+      anyOf: [
+        { type: 'string', regexp: routePattern.toString() },
+        {
+          type: 'object',
+          if: {
+            properties: {
+              route: { type: 'string', regexp: /\$connect/.toString() },
+            },
+          },
+          then: {
+            properties: {
+              route: { type: 'string' },
+              routeResponseSelectionExpression: {
+                type: 'string',
+                regexp: /^(?:\$default)$/.toString(),
+              },
+              authorizer: {
+                anyOf: [
+                  { type: 'string' },
+                  {
+                    type: 'object',
+                    properties: {
+                      name: { type: 'string' },
+                      identitySource: { type: 'array', items: { type: 'string' } },
+                    },
+                    required: ['name'],
+                    additionalProperties: false,
+                  },
+                  {
+                    type: 'object',
+                    properties: {
+                      arn: { type: 'string' },
+                      identitySource: { type: 'array', items: { type: 'string' } },
+                    },
+                    required: ['arn'],
+                    additionalProperties: false,
+                  },
+                ],
+              },
+            },
+            required: ['route'],
+            additionalProperties: false,
+          },
+          else: {
+            properties: {
+              route: { type: 'string', regexp: routePattern.toString() },
+              routeResponseSelectionExpression: {
+                type: 'string',
+                regexp: /^(?:\$default)$/.toString(),
+              },
+            },
+            required: ['route'],
+            additionalProperties: false,
+          },
+        },
+      ],
     });
   }
 }

lib/plugins/aws/provider/awsProvider.js

@@ -236,6 +236,7 @@ class AwsProvider {
         },
         provider: {
           properties: {
+            // httpApi properties
             httpApi: {
               type: 'object',
               properties: {
@@ -313,6 +314,17 @@ class AwsProvider {
                     },
                   ],
                 },
+                websocket: {
+                  oneOf: [
+                    { type: 'boolean' },
+                    {
+                      type: 'object',
+                      properties: {
+                        level: { type: 'string', regexp: /^(?:(INFO|ERROR))$/.toString() },
+                      },
+                    },
+                  ],
+                },
               },
             },
             resourcePolicy: {
@@ -321,6 +333,9 @@ class AwsProvider {
                 type: 'object',
               },
             },
+            // websockets properties
+            websocketsApiName: { type: 'string' },
+            websocketsApiRouteSelectionExpression: { type: 'string' },
           },
         },
         function: {