antrean-anjungan/examples/nuxt3-websocket-client/node_modules/unctx/dist/transform.mjs

import * as acorn from 'acorn';
import MagicString from 'magic-string';
import { walk } from 'estree-walker';

const kInjected = "__unctx_injected__";
function createTransformer(options = {}) {
  options = {
    asyncFunctions: ["withAsyncContext"],
    helperModule: "unctx",
    helperName: "executeAsync",
    objectDefinitions: {},
    ...options
  };
  const objectDefinitionFunctions = Object.keys(options.objectDefinitions);
  const matchRE = new RegExp(
    `\\b(${[...options.asyncFunctions, ...objectDefinitionFunctions].join(
      "|"
    )})\\(`
  );
  function shouldTransform(code) {
    return typeof code === "string" && matchRE.test(code);
  }
  function transform(code, options_ = {}) {
    if (!options_.force && !shouldTransform(code)) {
      return;
    }
    const ast = acorn.parse(code, {
      sourceType: "module",
      ecmaVersion: "latest",
      locations: true
    });
    const s = new MagicString(code);
    const lines = code.split("\n");
    let detected = false;
    walk(ast, {
      enter(node) {
        if (node.type === "CallExpression") {
          const functionName = _getFunctionName(node.callee);
          if (options.asyncFunctions.includes(functionName)) {
            transformFunctionArguments(node);
            if (functionName !== "callAsync") {
              const lastArgument = node.arguments[node.arguments.length - 1];
              if (lastArgument && lastArgument.loc) {
                s.appendRight(toIndex(lastArgument.loc.end), ",1");
              }
            }
          }
          if (objectDefinitionFunctions.includes(functionName)) {
            for (const argument of node.arguments) {
              if (argument.type !== "ObjectExpression") {
                continue;
              }
              for (const property of argument.properties) {
                if (property.type !== "Property" || property.key.type !== "Identifier") {
                  continue;
                }
                if (options.objectDefinitions[functionName]?.includes(
                  property.key?.name
                )) {
                  transformFunctionBody(property.value);
                }
              }
            }
          }
        }
      }
    });
    if (!detected) {
      return;
    }
    s.appendLeft(
      0,
      `import { ${options.helperName} as __executeAsync } from "${options.helperModule}";`
    );
    return {
      code: s.toString(),
      magicString: s
    };
    function toIndex(pos) {
      return lines.slice(0, pos.line - 1).join("\n").length + pos.column + 1;
    }
    function transformFunctionBody(function_) {
      if (function_.type !== "ArrowFunctionExpression" && function_.type !== "FunctionExpression") {
        return;
      }
      if (!function_.async) {
        return;
      }
      const body = function_.body;
      let injectVariable = false;
      walk(body, {
        enter(node, parent) {
          if (node.type === "AwaitExpression" && !node[kInjected]) {
            detected = true;
            injectVariable = true;
            injectForNode(node, parent);
          } else if (node.type === "IfStatement" && node.consequent.type === "ExpressionStatement" && node.consequent.expression.type === "AwaitExpression") {
            detected = true;
            injectVariable = true;
            node.consequent.expression[kInjected] = true;
            injectForNode(node.consequent.expression, node);
          }
          if (node.type === "ArrowFunctionExpression" || node.type === "FunctionExpression" || node.type === "FunctionDeclaration") {
            return this.skip();
          }
        }
      });
      if (injectVariable && body.loc) {
        s.appendLeft(toIndex(body.loc.start) + 1, "let __temp, __restore;");
      }
    }
    function transformFunctionArguments(node) {
      for (const function_ of node.arguments) {
        transformFunctionBody(function_);
      }
    }
    function injectForNode(node, parent) {
      const isStatement = parent?.type === "ExpressionStatement";
      if (!node.loc || !node.argument.loc) {
        return;
      }
      s.remove(toIndex(node.loc.start), toIndex(node.argument.loc.start));
      s.remove(toIndex(node.loc.end), toIndex(node.argument.loc.end));
      s.appendLeft(
        toIndex(node.argument.loc.start),
        isStatement ? `;(([__temp,__restore]=__executeAsync(()=>` : `(([__temp,__restore]=__executeAsync(()=>`
      );
      s.appendRight(
        toIndex(node.argument.loc.end),
        isStatement ? `)),await __temp,__restore());` : `)),__temp=await __temp,__restore(),__temp)`
      );
    }
  }
  return {
    transform,
    shouldTransform
  };
}
function _getFunctionName(node) {
  if (node.type === "Identifier") {
    return node.name;
  } else if (node.type === "MemberExpression") {
    return _getFunctionName(node.property);
  }
  return "";
}

export { createTransformer };