index.js

'use strict'

const async = require('async')
const Base = require('@bitfinex/bfx-facs-base')
const Fastify = require('fastify')

const DEFAULT_ROUTES = [
  {
    method: 'GET',
    url: '/echo',
    schema: {
      querystring: {
        type: 'object',
        properties: {
          value: { type: 'string' }
        },
        required: ['value']
      },
      response: {
        200: {
          type: 'object',
          properties: {
            value: { type: 'string' }
          }
        }
      }
    },
    handler: async (request, reply) => {
      return { value: request.query.value }
    }
  }
]

class HttpdFacility extends Base {
  constructor (caller, opts, ctx) {
    super(caller, opts, ctx)

    this.name = 'httpd'
    this._hasConf = true

    this.init()

    this.mem = { plugins: [], routes: [], decorators: [], hooks: [] }
    this.errorHandler = opts.errorHandler ?? null
  }

  addRoute (r) {
    if (this.server) {
      throw new Error('ERR_FACS_SERVER_HTTP_ALREADY_INITED')
    }

    this.mem.routes.push(r)
  }

  addDecorator (d) {
    if (this.server) {
      throw new Error('ERR_FACS_SERVER_HTTP_ALREADY_INITED')
    }

    this.mem.decorators.push(d)
  }

  addPlugin (p) {
    if (this.server) {
      throw new Error('ERR_FACS_SERVER_HTTP_ALREADY_INITED')
    }

    this.mem.plugins.push(p)
  }

  addHook (hookName, handler) {
    if (this.server) {
      throw new Error('ERR_FACS_SERVER_HTTP_ALREADY_INITED')
    }

    this.mem.hooks.push({ name: hookName, handler })
  }

  addErrorHandler (handler) {
    if (this.server) {
      throw new Error('ERR_FACS_SERVER_HTTP_ALREADY_INITED')
    }

    this.errorHandler = handler
  }

  async startServer () {
    if (this.server) {
      throw new Error('ERR_FACS_SERVER_HTTP_CREATE_DUP')
    }

    const fastify = Fastify({
      logger: this.opts.logger,
      trustProxy: this.opts.trustProxy,
      bodyLimit: this.opts.bodyLimit
    })

    this.server = fastify

    if (this.opts.staticRootPath) {
      this.server.register(require('@fastify/static'), {
        root: this.opts.staticRootPath,
        prefix: this.opts.staticPrefix,
        constraints: this.opts.staticConstraints
      })

      if (this.opts.staticOn404File) {
        this.server.setNotFoundHandler((_, reply) => {
          return reply.code(404).type('text/html').sendFile(this.opts.staticOn404File)
        })
      }
    }

    this.mem.plugins.forEach(p => {
      this.server.register(p[0], p[1])
    })

    this.mem.decorators.forEach(d => {
      this.server.decorate(d[0], d[1], d[2])
    })

    // Register hooks
    this.mem.hooks.forEach(hook => {
      this.server.addHook(hook.name, hook.handler)
    })

    if (this.opts.addDefaultRoutes) {
      DEFAULT_ROUTES.forEach(r => {
        this.server.route(r)
      })
    }

    this.server.register(async () => {
      this.mem.routes.forEach(r => {
        this.server.route(r)
      })
    })

    if (this.errorHandler) {
      this.server.setErrorHandler(this.errorHandler)
    }

    return await this.server.listen({
      port: this.opts.port || this.conf.port
    })
  }

  _stop (cb) {
    async.series([
      next => { super._stop(next) },
      async () => {
        if (this.server) {
          await this.server.close()
        }
      }
    ], cb)
  }
}

module.exports = HttpdFacility