index-background.js

'use strict'

require('dotenv').config()

const Blipp = require('blipp')
const Hapi = require('@hapi/hapi')
const HapiAuthJwt2 = require('hapi-auth-jwt2')
const moment = require('moment')

const config = require('./config')
const db = require('./src/lib/connectors/db')
const HapiPinoPlugin = require('./src/plugins/hapi-pino.plugin.js')
const { JobRegistrationService } = require('./src/lib/queue-manager/job-registration-service')
const { logger } = require('./src/logger')
const routes = require('./src/routes/background.js')
const { StartUpJobsService } = require('./src/lib/queue-manager/start-up-jobs-service')
const { validate } = require('./src/lib/validate')

// This changes the locale for moment globally to English (United Kingdom). Any new instances of moment will use this
// rather than the default 'en-US'
moment.locale('en-gb')

// Define server
const server = Hapi.server({
  ...config.serverBackground
})

const plugins = [
  require('./src/lib/queue-manager').plugin
]

// Register plugins
const _registerServerPlugins = async (server) => {
  // Service plugins
  await server.register(HapiPinoPlugin())
  await server.register(plugins)

  await server.register({
    plugin: Blipp,
    options: config.blipp
  })

  // JWT token auth
  await server.register(HapiAuthJwt2)
}

const _configureServerAuthStrategy = (server) => {
  server.auth.strategy('jwt', 'jwt', {
    ...config.jwt,
    validate
  })
  server.auth.default('jwt')
}

const start = async function () {
  try {
    await _registerServerPlugins(server)
    _configureServerAuthStrategy(server)
    server.route(routes)

    // TODO: Ideally we wouldn't bother registering all the Queues and workers when running unit tests. So, we would
    // move this into the `if (!module.parent)` block. But when we do the tests only run the first few and then exit.
    // We should look into what's happening and why we can't move it there.
    JobRegistrationService.go(server.queueManager)

    if (!module.parent) {
      StartUpJobsService.go(server.queueManager)
      await server.start()
      const name = `${process.env.SERVICE_NAME}-background`
      const uri = server.info.uri
      server.log('info', `Service ${name} running at: ${uri}`)
    }
  } catch (err) {
    logger.error('Failed to start server', err.stack)
  }
}

const _processError = message => err => {
  logger.error(message, err.stack)
  process.exit(1)
}

process
  .on('unhandledRejection', _processError('unhandledRejection'))
  .on('uncaughtException', _processError('uncaughtException'))
  .on('SIGINT', async () => {
    logger.info('Stopping hapi server: existing requests have 25 seconds to complete')
    await server.stop({ timeout: 25 * 1000 })

    logger.info('Stopping BullMQ workers')
    await server.queueManager.closeAll()

    logger.info('Closing connection pool')
    await db.pool.end()

    logger.info("That's all folks!")
    return process.exit(0)
  })

if (!module.parent) {
  start()
}

module.exports = {
  server,
  start
}