import timeout from 'express-timeout-handler'

import statsd from '../lib/statsd.js'

// Heroku router requests timeout after 30 seconds. We should stop them earlier!

const maxRequestTimeout = parseInt(process.env.REQUEST_TIMEOUT, 10) || 10000

export default timeout.handler({

// Default timeout for all endpoints

// To override for a given router/endpoint, use `xExpressTimeoutHandler.set(...)`

timeout: maxRequestTimeout,

// IMPORTANT:

// We cannot allow the middleware to disable the `res` object's methods like

// it does by default if we want to use `next` in the `onTimeout` handler!

disable: [],

onTimeout: function (req, res, next) {

const incrementTags = []

// Be careful with depending on attributes set on the `req` because

// under certain conditions the contextualizers might not yet have

// had a chance to run.

if (req.pagePath) {

incrementTags.push(`path:${req.pagePath}`)

}

if (req.context?.currentCategory) {

incrementTags.push(`product:${req.context.currentCategory}`)

}

statsd.increment('middleware.timeout', 1, incrementTags)

// Create a custom timeout error

const timeoutError = new Error('Request timed out')

timeoutError.statusCode = 503

timeoutError.code = 'ETIMEDOUT'

// Pass the error to our Express error handler for consolidated processing

return next(timeoutError)

},

// Can also set an `onDelayedResponse` property IF AND ONLY IF you allow for disabling methods

})