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

import validate from './lib/validate.js'
import compileApi from './lib/api.js'
import compileIntegrations from './lib/integrations.js'
import compileRouteResponses from './lib/route-responses.js'
import compilePermissions from './lib/permissions.js'
import compileRoutes from './lib/routes.js'
import compileDeployment from './lib/deployment.js'
import compileStage from './lib/stage.js'
import compileAuthorizers from './lib/authorizers.js'

class AwsCompileWebsockets {
  constructor(serverless, options) {
    this.serverless = serverless
    this.options = options
    this.provider = this.serverless.getProvider('aws')

    Object.assign(
      this,
      validate,
      compileApi,
      compileIntegrations,
      compileRouteResponses,
      compileAuthorizers,
      compilePermissions,
      compileRoutes,
      compileDeployment,
      compileStage,
    )

    this.hooks = {
      'package:compileEvents': async () => {
        this.validated = this.validate()

        if (this.validated.events.length === 0) {
          return Promise.resolve()
        }

        return Promise.resolve(this)
          .then(() => this.compileApi())
          .then(() => this.compileIntegrations())
          .then(() => this.compileRouteResponses())
          .then(() => this.compileAuthorizers())
          .then(() => this.compilePermissions())
          .then(() => this.compileRoutes())
          .then(() => this.compileStage())
          .then(() => this.compileDeployment())
      },
    }

    this.serverless.configSchemaHandler.defineFunctionEvent(
      'aws',
      'websocket',
      {
        anyOf: [
          { type: 'string' },
          {
            type: 'object',
            properties: {
              route: { type: 'string' },
              routeResponseSelectionExpression: {
                const: '$default',
              },
              authorizer: {
                anyOf: [
                  { $ref: '#/definitions/awsArnString' },
                  { $ref: '#/definitions/functionName' },
                  {
                    type: 'object',
                    properties: {
                      name: { $ref: '#/definitions/functionName' },
                      arn: { $ref: '#/definitions/awsArn' },
                      identitySource: {
                        type: 'array',
                        items: { type: 'string' },
                      },
                    },
                    anyOf: [{ required: ['name'] }, { required: ['arn'] }],
                    additionalProperties: false,
                  },
                ],
              },
            },
            required: ['route'],
            additionalProperties: false,
          },
        ],
      },
    )
  }
}

export default AwsCompileWebsockets