reserve/handlers/url.js

'use strict'

const http = require('http')
const https = require('https')
const headersFactory = require('../mock/headers')

const http2ForbiddenResponseHeaders = [
  'transfer-encoding',
  'connection',
  'keep-alive'
]

function protocol (url) {
  if (url.startsWith('https')) {
    return https
  }
  return http
}

function unsecureCookies (headers) {
  const setCookie = headers['set-cookie']
  if (setCookie) {
    headers['set-cookie'] = setCookie.map(cookie => cookie.replace(/\s*secure;/i, ''))
  }
}

function noop () {}

function validateHook (mapping, hookName) {
  if (typeof mapping[hookName] === 'string') {
    mapping[hookName] = require(mapping[hookName])
  }
}

module.exports = {
  schema: {
    'unsecure-cookies': {
      type: 'boolean',
      defaultValue: false
    },
    'forward-request': {
      types: ['function', 'string'],
      defaultValue: noop
    },
    'forward-response': {
      types: ['function', 'string'],
      defaultValue: noop
    },
    'ignore-unverifiable-certificate': {
      type: 'boolean',
      defaultValue: false
    }
  },
  validate: async mapping => {
    validateHook(mapping, 'forward-request')
    validateHook(mapping, 'forward-response')
  },
  redirect: async ({ configuration, mapping, match, redirect: url, request, response }) => {
    let done
    let fail
    const promise = new Promise((resolve, reject) => {
      done = resolve
      fail = reject
    })
    const { method, headers } = request
    const options = {
      method,
      url,
      headers: headersFactory(headers)
    }
    delete options.headers.host // Some websites rely on the host header
    if (mapping['ignore-unverifiable-certificate']) {
      options.rejectUnauthorized = false
    }
    const context = {}
    const hookParams = {
      configuration,
      context,
      mapping,
      match,
      request: options,
      incoming: request
    }
    await mapping['forward-request'](hookParams)
    Object.keys(options.headers)
      .filter(header => header.startsWith(':'))
      .forEach(header => delete options.headers[header])
    const redirectedRequest = protocol(options.url).request(options.url, options, async redirectedResponse => {
      if (mapping['unsecure-cookies']) {
        unsecureCookies(redirectedResponse.headers)
      }
      const { headers: responseHeaders } = redirectedResponse
      const result = await mapping['forward-response']({
        ...hookParams,
        statusCode: redirectedResponse.statusCode,
        headers: responseHeaders
      })
      if (result !== undefined) {
        if (!['GET', 'HEAD'].includes(request.method)) {
          return fail(new Error('Internal redirection impossible because the body is already consumed'))
        }
        return done(result)
      }
      if (configuration.http2) {
        http2ForbiddenResponseHeaders.forEach(header => delete responseHeaders[header])
      }
      response.writeHead(redirectedResponse.statusCode, responseHeaders)
      if (request.aborted) {
        response.end()
        return done()
      }
      response.on('finish', () => done(result))
      redirectedResponse
        .on('error', fail)
        .pipe(response)
    })
    redirectedRequest.on('error', fail)
    request
      .on('error', fail)
      .pipe(redirectedRequest)
    return promise
  }
}