elastic-apm-node/lib/instrumentation/modules/pug.js

The official Elastic APM agent for Node.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';

var semver = require('semver');

var shimmer = require('../shimmer');
var templateShared = require('../template-shared');

module.exports = function (pug, agent, { version, enabled }) {
  if (!enabled) return pug;

  if (!semver.satisfies(version, '>=0.1.0 <4')) {
    agent.logger.debug(
      'cannot instrument pug version %s, skipping pug instrumentation',
      version,
    );
    return pug;
  }

  agent.logger.debug('shimming pug.compile');
  shimmer.wrap(pug, 'compile', templateShared.wrapCompile(agent, 'pug'));

  return pug;
};