fastify/lib/hookRunner.js

Fast and low overhead web framework, for Node.js

'use strict'

function hookRunner (functions, runner, state, cb) {
  var i = 0

  function next (err) {
    if (err || i === functions.length) {
      cb(err, state)
      return
    }

    const result = runner(functions[i++], state, next)
    if (result && typeof result.then === 'function') {
      result.then(handleResolve, handleReject)
    }
  }

  function handleResolve () {
    next()
  }

  function handleReject (err) {
    cb(err, state)
  }

  next()
}

function onSendHookRunner (functions, reply, payload, cb) {
  var i = 0

  function next (err, newPayload) {
    if (err) {
      cb(err, reply, payload)
      return
    }

    if (newPayload !== undefined) {
      payload = newPayload
    }

    if (i === functions.length) {
      cb(null, reply, payload)
      return
    }

    const result = functions[i++](reply.request, reply, payload, next)
    if (result && typeof result.then === 'function') {
      result.then(handleResolve, handleReject)
    }
  }

  function handleResolve (newPayload) {
    next(null, newPayload)
  }

  function handleReject (err) {
    cb(err, reply, payload)
  }

  next()
}

module.exports = { hookRunner, onSendHookRunner }