refactor

Author: rochdev

packages/datadog-instrumentations/src/helpers/rewriter/index.js

@@ -72,7 +72,7 @@ function rewrite (content, filename, format) {
       ast ??= parse(content.toString(), { range: true, sourceType })
 
       const query = astQuery || fromFunctionQuery(functionQuery)
-      const state = { ...inst, format, functionQuery }
+      const state = { ...inst, sourceType, functionQuery }
 
       traverse(ast, query, (...args) => transform(state, ...args))
     }

packages/datadog-instrumentations/src/helpers/rewriter/transforms.js

@@ -8,15 +8,15 @@ const tracingChannelPredicate = (node) => (
 )
 
 const transforms = module.exports = {
-  tracingChannelImport ({ format }, node) {
+  tracingChannelImport ({ sourceType }, node) {
     if (node.body.some(tracingChannelPredicate)) return
 
     const index = node.body.findIndex(child => child.directive === 'use strict')
-    const code = format === 'module'
+    const code = sourceType === 'module'
       ? 'import { tracingChannel as tr_ch_apm_tracingChannel } from "diagnostics_channel"'
       : 'const {tracingChannel: tr_ch_apm_tracingChannel} = require("diagnostics_channel")'
 
-    node.body.splice(index + 1, 0, parse(code, { sourceType: format }).body[0])
+    node.body.splice(index + 1, 0, parse(code, { sourceType }).body[0])
   },
 
   tracingChannelDeclaration (state, node) {