dd-trace/packages/datadog-instrumentations/src/vm.js

Datadog APM tracing client for JavaScript

'use strict'

const shimmer = require('../../datadog-shimmer')
const { channel, addHook } = require('./helpers/instrument')
const names = ['vm', 'node:vm']

const runScriptStartChannel = channel('datadog:vm:run-script:start')
const sourceTextModuleStartChannel = channel('datadog:vm:source-text-module:start')

addHook({ name: names }, function (vm) {
  vm.Script = class extends vm.Script {
    constructor (code) {
      super(...arguments)

      if (runScriptStartChannel.hasSubscribers && code) {
        runScriptStartChannel.publish({ code })
      }
    }
  }

  if (vm.SourceTextModule && typeof vm.SourceTextModule === 'function') {
    vm.SourceTextModule = class extends vm.SourceTextModule {
      constructor (code) {
        super(...arguments)

        if (sourceTextModuleStartChannel.hasSubscribers && code) {
          sourceTextModuleStartChannel.publish({ code })
        }
      }
    }
  }

  shimmer.wrap(vm, 'runInContext', wrapVMMethod)
  shimmer.wrap(vm, 'runInNewContext', wrapVMMethod)
  shimmer.wrap(vm, 'runInThisContext', wrapVMMethod)
  shimmer.wrap(vm, 'compileFunction', wrapVMMethod)

  return vm
})

function wrapVMMethod (original) {
  return function wrappedVMMethod (code) {
    if (runScriptStartChannel.hasSubscribers && code) {
      runScriptStartChannel.publish({ code })
    }

    return original.apply(this, arguments)
  }
}