heroku-debug/tmp/node_modules/opbeat/lib/instrumentation/modules/http.js

debugging plugin for the CLI

'use strict'

var debug = require('debug')('opbeat')
var shimmer = require('../shimmer')
var shared = require('../http-shared')

module.exports = function (http, agent) {
  debug('shimming http.Server.prototype.emit function')
  shimmer.wrap(http && http.Server && http.Server.prototype, 'emit', shared.instrumentRequest(agent, 'http'))

  debug('shimming http.request function')
  shimmer.wrap(http, 'request', shared.traceOutgoingRequest(agent, 'http'))

  debug('shimming http.ServerResponse.prototype.writeHead function')
  shimmer.wrap(http && http.ServerResponse && http.ServerResponse.prototype, 'writeHead', wrapWriteHead)

  return http

  function wrapWriteHead (original) {
    return function wrappedWriteHead () {
      var headers = arguments.length === 1
        ? this._headers // might be because of implicit headers
        : arguments[arguments.length - 1]

      if (headers && typeof headers === 'object' && !Array.isArray(headers)) {
        Object.keys(headers).some(function (key) {
          if (key.toLowerCase() !== 'content-type') return false
          if (String(headers[key]).toLowerCase().indexOf('text/event-stream') !== 0) return false
          var trans = agent._instrumentation.currentTransaction
          var uuid = trans && trans.uuid
          debug('detected SSE response - ending transaction %o', { uuid: uuid })
          agent.endTransaction()
          return true
        })
      }

      return original.apply(this, arguments)
    }
  }
}