isolated-function/src/compile/transform-dependencies.js

Run JavaScript from AI agents, plugins, and workflows in a separate Node.js process with strict safety limits.

'use strict'

const walk = require('acorn-walk')
const acorn = require('acorn')

/**
 * Transforms dependency module names by removing version specifiers.
 *
 * Parses JavaScript code and walks through the AST to find all require() calls
 * and import declarations. Extracts module names from strings that include
 * version information (e.g., 'is-emoji@1.0.0' becomes 'is-emoji').
 *
 * Handles both:
 * - Scoped packages: '@scope/package@1.0.0' → '@scope/package'
 * - Regular packages: 'package@1.0.0' → 'package'
 *
 * This transformation is necessary because the dependency strings include
 * version specifiers for installation tracking, but require/import statements
 * should only reference the base module name.
 *
 * @param {string} code - JavaScript code containing require() or import statements
 * @returns {string} Transformed code with version specifiers removed from dependencies
 */
module.exports = code => {
  const ast = acorn.parse(code, { ecmaVersion: 2023, sourceType: 'module' })

  let newCode = ''
  let lastIndex = 0

  // Helper function to process and transform nodes
  const processNode = node => {
    if (node.type === 'Literal' && node.value.includes('@')) {
      // Check if it's a scoped module
      if (node.value.startsWith('@')) {
        // Handle scoped packages
        const slashIndex = node.value.indexOf('/')
        if (slashIndex !== -1) {
          const atVersionIndex = node.value.indexOf('@', slashIndex)
          const moduleName =
            atVersionIndex !== -1 ? node.value.substring(0, atVersionIndex) : node.value
          // Append code before this node
          newCode += code.substring(lastIndex, node.start)
          // Append transformed dependency
          newCode += `'${moduleName}'`
        }
      } else {
        // Handle non-scoped packages
        const [moduleName] = node.value.split('@')
        // Append code before this node
        newCode += code.substring(lastIndex, node.start)
        // Append transformed dependency
        newCode += `'${moduleName}'`
      }
      // Update lastIndex to end of current node
      lastIndex = node.end
    }
  }

  // Traverse the AST to find require and import declarations
  walk.simple(ast, {
    CallExpression (node) {
      if (node.callee.name === 'require' && node.arguments.length === 1) {
        processNode(node.arguments[0])
      }
    },
    ImportDeclaration (node) {
      processNode(node.source)
    }
  })

  // Append remaining code after last modified dependency
  newCode += code.substring(lastIndex)

  return newCode
}