restify.js

/*
 * Copyright Elasticsearch B.V. and other contributors where applicable.
 * Licensed under the BSD 2-Clause License; you may not use this file except in
 * compliance with the BSD 2-Clause License.
 */

'use strict';

const semver = require('semver');

const shimmer = require('../shimmer');

module.exports = function (restify, agent, { version, enabled }) {
  if (!enabled) {
    return restify;
  }
  if (!semver.satisfies(version, '>=5.2.0 <12.0.0')) {
    agent.logger.debug('restify version %s not supported, skipping', version);
    return restify;
  }

  agent.setFramework({ name: 'restify', version, overwrite: false });

  function patchServer(server) {
    if (semver.gte(version, '7.0.0')) {
      shimmer.wrap(server, '_onHandlerError', function (orig) {
        return function _wrappedOnHandlerError(err, req, res, isUncaught) {
          if (err)
            agent.captureError(err, { request: req, handled: !isUncaught });
          return orig.apply(this, arguments);
        };
      });
    } else {
      shimmer.wrap(server, '_emitErrorEvents', function (orig) {
        return function _wrappedOnHandlerError(req, res, route, err, cb) {
          if (err) agent.captureError(err, { request: req });
          return orig.apply(this, arguments);
        };
      });
    }
  }

  shimmer.wrap(restify, 'createServer', function (fn) {
    return function wrappedCreateServer() {
      const server = fn.apply(this, arguments);
      patchServer(server);
      return server;
    };
  });

  return restify;
};