@fastify/accepts/index.js

Add accept parser to fastify

'use strict'

const accepts = require('accepts')

const fp = require('fastify-plugin')

const acceptsObjectSymbol = Symbol('acceptsObject')

const methodNames = [
  'charset', 'charsets',
  'encoding', 'encodings',
  'language', 'languages',
  'type', 'types'
]

function acceptsMethod () {
  if (!this.raw[acceptsObjectSymbol]) {
    this.raw[acceptsObjectSymbol] = accepts(this.raw)
  }
  return this.raw[acceptsObjectSymbol]
}

function replyAcceptMethod () {
  if (!this.request[acceptsObjectSymbol]) {
    this.request[acceptsObjectSymbol] = accepts(this.request.raw)
  }
  return this.request[acceptsObjectSymbol]
}

function fastifyAccepts (fastify, options, done) {
  fastify.decorateRequest('accepts', acceptsMethod)

  methodNames.forEach(methodName => {
    fastify.decorateRequest(methodName, function (arr) {
      const acceptsObject = this.accepts()
      if (arguments.length === 0) return acceptsObject[methodName]()
      return acceptsObject[methodName](arr)
    })
  })

  if (options.decorateReply === true) {
    fastify.decorateReply('requestAccepts', replyAcceptMethod)

    methodNames.forEach(methodName => {
      const capitalizedMethodName = methodName.replace(/(?:^|\s)\S/gu, a => a.toUpperCase())
      fastify.decorateReply('request' + capitalizedMethodName, function (arr) {
        const acceptsObject = this.requestAccepts()
        if (arguments.length === 0) return acceptsObject[methodName]()
        return acceptsObject[methodName](arr)
      })
    })
  }

  done()
}

module.exports = fp(fastifyAccepts, {
  fastify: '5.x',
  name: '@fastify/accepts'
})
module.exports.default = fastifyAccepts
module.exports.fastifyAccepts = fastifyAccepts