dd-trace/packages/datadog-instrumentations/src/helpers/rewriter/compiler.js

Datadog APM tracing client for JavaScript

'use strict'

/**
 * This file is meant to be only thin wrappers over core
 * parsing/traversing/generating functionality with the goal to eventually move
 * them out of the project. No other code should be added to this file such as
 * helpers etc, and the API should be kept exactly as an external API would be
 * expected to be.
 */

const log = require('../../../../dd-trace/src/log')

// eslint-disable-next-line camelcase, no-undef
const runtimeRequire = typeof __webpack_require__ === 'function' ? __non_webpack_require__ : require

const compiler = {
  parse: (sourceText, options) => {
    try {
      // TODO: Figure out ESBuild `createRequire` issue and remove this hack.
      const oxc = runtimeRequire(['oxc', 'parser'].join('-'))

      compiler.parse = (sourceText, { range, isModule } = {}) => {
        const { program, errors } = oxc.parseSync('index.js', sourceText, {
          range,
          sourceType: isModule ? 'module' : 'script',
          preserveParens: false,
        })

        if (errors?.length > 0) throw errors[0]

        return program
      }
    } catch (e) {
      log.error(e)

      const meriyah = require('../../../../../vendor/dist/meriyah')

      compiler.parse = (sourceText, { range, isModule } = {}) => {
        return meriyah.parse(sourceText.toString(), {
          loc: range,
          ranges: range,
          module: isModule,
        })
      }
    }

    return compiler.parse(sourceText, options)
  },

  traverse: (ast, query, visitor) => {
    const esquery = require('../../../../../vendor/dist/esquery')

    compiler.traverse = (ast, query, visitor) => {
      return esquery.traverse(ast, esquery.parse(query), visitor)
    }

    return compiler.traverse(ast, query, visitor)
  },

  query: (ast, query) => {
    const esquery = require('../../../../../vendor/dist/esquery')

    compiler.query = esquery.query

    return compiler.query(ast, query)
  },
}

module.exports = {
  parse: (...args) => compiler.parse(...args),
  traverse: (...args) => compiler.traverse(...args),
  query: (...args) => compiler.query(...args),
}