aran/lib/trans/query/variable.mjs

JavaScript instrumenter

import { listChildren } from "estree-sentry";
import { flatMap, includes } from "../../util/index.mjs";
import { listPatternVariable } from "./pattern.mjs";

/**
 * @type {(
 *   nodes: import("estree-sentry").Node<{}>[],
 *   variable: import("estree-sentry").VariableName,
 * ) => boolean}
 */
export const hasFreeVariable = (nodes, variable) => {
  for (const node of nodes) {
    if (
      variable === "argument" &&
      (node.type === "FunctionExpression" ||
        node.type === "FunctionDeclaration")
    ) {
      return false;
    }
    if (
      node.type === "CatchClause" &&
      node.param !== null &&
      includes(listPatternVariable(node.param), variable)
    ) {
      return false;
    }
    if (
      (node.type === "FunctionExpression" ||
        node.type === "FunctionDeclaration" ||
        node.type === "ArrowFunctionExpression") &&
      includes(flatMap(node.params, listPatternVariable), variable)
    ) {
      return false;
    }
    if (node.type === "BlockStatement" || node.type === "StaticBlock") {
      for (const child of node.body) {
        if (
          child.type === "ClassDeclaration" &&
          child.id !== null &&
          child.id.name === variable
        ) {
          return false;
        }
        if (child.type === "VariableDeclaration" && child.kind !== "var") {
          for (const declarator of child.declarations) {
            if (includes(listPatternVariable(declarator.id), variable)) {
              return false;
            }
          }
        }
      }
    }
    if (node.type === "Identifier" && node.name === variable) {
      return true;
    }
    if (
      node.type === "CallExpression" &&
      node.callee.type === "Identifier" &&
      node.callee.name === "eval"
    ) {
      return true;
    }
    if (hasFreeVariable(listChildren(node), variable)) {
      return true;
    }
  }
  return false;
};