index.ts

import express from 'express'
import { ParamsDictionary, RequestHandler } from 'express-serve-static-core'
import path from 'path'
import { BullBoardQueues, QueueAdapter } from './@types/app'
import { cleanAll } from './routes/cleanAll'
import { cleanJob } from './routes/cleanJob'
import { entryPoint } from './routes/index'
import { promoteJob } from './routes/promoteJob'

import { queuesHandler } from './routes/queues'
import { retryAll } from './routes/retryAll'
import { retryJob } from './routes/retryJob'

export { BullMQAdapter } from './queueAdapters/bullMQ'
export { BullAdapter } from './queueAdapters/bull'

const bullBoardQueues: BullBoardQueues = {}

const wrapAsync = <Params extends ParamsDictionary>(
  fn: RequestHandler<Params>,
): RequestHandler<Params> => async (req, res, next) =>
  Promise.resolve(fn(req, res, next)).catch(next)

const router = express()
router.locals.bullBoardQueues = bullBoardQueues

router.set('view engine', 'ejs')
router.set('views', path.resolve(__dirname, '../dist/ui'))

router.use('/static', express.static(path.resolve(__dirname, '../static')))

router.get(['/', '/queue/:queueName'], entryPoint)
router.get('/api/queues', wrapAsync(queuesHandler))
router.put('/api/queues/:queueName/retry', wrapAsync(retryAll))
router.put('/api/queues/:queueName/:id/retry', wrapAsync(retryJob))
router.put('/api/queues/:queueName/:id/clean', wrapAsync(cleanJob))
router.put('/api/queues/:queueName/:id/promote', wrapAsync(promoteJob))
router.put('/api/queues/:queueName/clean/:queueStatus', wrapAsync(cleanAll))

export const setQueues = (bullQueues: ReadonlyArray<QueueAdapter>): void => {
  bullQueues.forEach((queue) => {
    const name = queue.getName()

    bullBoardQueues[name] = { queue }
  })
}

export const replaceQueues = (
  bullQueues: ReadonlyArray<QueueAdapter>,
): void => {
  const queuesToPersist: string[] = bullQueues.map((queue) => queue.getName())

  Object.keys(bullBoardQueues).forEach((name) => {
    if (queuesToPersist.indexOf(name) === -1) {
      delete bullBoardQueues[name]
    }
  })

  return setQueues(bullQueues)
}

export { router }