serverless-offline/src/events/websocket/HttpServer.js

Emulate AWS λ and API Gateway locally when developing your Serverless project

import { readFile } from "node:fs/promises"
import { resolve } from "node:path"
import { exit } from "node:process"
import { Server } from "@hapi/hapi"
import { log } from "../../utils/log.js"
import { catchAllRoute, connectionsRoutes } from "./http-routes/index.js"

export default class HttpServer {
  #options = null

  #server = null

  #webSocketClients = null

  constructor(options, webSocketClients) {
    this.#options = options
    this.#webSocketClients = webSocketClients
  }

  async #loadCerts(httpsProtocol) {
    const [cert, key] = await Promise.all([
      readFile(resolve(httpsProtocol, "cert.pem"), "utf8"),
      readFile(resolve(httpsProtocol, "key.pem"), "utf8"),
    ])

    return {
      cert,
      key,
    }
  }

  async createServer() {
    const { host, httpsProtocol, websocketPort } = this.#options

    const serverOptions = {
      host,
      port: websocketPort,
      router: {
        stripTrailingSlash: true,
      },
      // https support
      ...(httpsProtocol != null && {
        tls: await this.#loadCerts(httpsProtocol),
      }),
    }

    this.#server = new Server(serverOptions)
  }

  async start() {
    // add routes
    const routes = [
      ...connectionsRoutes(this.#webSocketClients),
      catchAllRoute(),
    ]
    this.#server.route(routes)

    const { host, httpsProtocol, websocketPort } = this.#options

    try {
      await this.#server.start()
    } catch (err) {
      log.error(
        `Unexpected error while starting serverless-offline websocket server on port ${websocketPort}:`,
        err,
      )
      exit(1)
    }

    log.notice(
      `Offline [http for websocket] listening on ${
        httpsProtocol ? "https" : "http"
      }://${host}:${websocketPort}`,
    )
  }

  // stops the server
  stop(timeout) {
    return this.#server.stop({
      timeout,
    })
  }

  get server() {
    return this.#server.listener
  }
}