eslint-plugin-effector/utils/replace-by-sample.js

Enforcing best practices for Effector

const { buildObjectInText } = require("./builders");

function* replaceGuardBySample(
  guardConfig,
  { fixer, node, context, importNodes }
) {
  let mapperFunctionNode = null;

  let clockNode = guardConfig.clock?.value;
  let targetNode = guardConfig.target?.value;
  let sourceNode = guardConfig.source?.value;
  let filterNode = guardConfig.filter?.value;

  if (
    targetNode?.type === "CallExpression" &&
    targetNode?.callee?.property?.name === "prepend"
  ) {
    mapperFunctionNode = targetNode?.arguments?.[0];
    targetNode = targetNode.callee.object;
    targetMapperUsed = true;
  }

  yield* replaceBySample(
    { clockNode, sourceNode, filterNode, mapperFunctionNode, targetNode },
    { node, fixer, context, importNodes, methodName: "guard" }
  );
}

function* replaceForwardBySample(
  forwardConfig,
  { fixer, node, context, importNodes }
) {
  let mapperFunctionNode = null;

  let clockMapperUsed = false;
  let targetMapperUsed = false;

  let clockNode = forwardConfig.from.value;
  let targetNode = forwardConfig.to.value;

  if (
    clockNode?.type === "CallExpression" &&
    clockNode?.callee?.property?.name === "map"
  ) {
    mapperFunctionNode = clockNode?.arguments?.[0];
    clockNode = clockNode.callee.object;
    clockMapperUsed = true;
  }

  if (
    targetNode?.type === "CallExpression" &&
    targetNode?.callee?.property?.name === "prepend"
  ) {
    mapperFunctionNode = targetNode?.arguments?.[0];
    targetNode = targetNode.callee.object;
    targetMapperUsed = true;
  }

  // We cannot apply two mappers in one sample
  // Let's revert mappers and use .map + .prepend
  if (clockMapperUsed && targetMapperUsed) {
    mapperFunctionNode = null;
    clockNode = forwardConfig.from.value;
    targetNode = forwardConfig.to.value;
  }

  yield* replaceBySample(
    { clockNode, mapperFunctionNode, targetNode },
    { node, fixer, context, importNodes, methodName: "forward" }
  );
}

function* replaceBySample(
  { clockNode, sourceNode, filterNode, mapperFunctionNode, targetNode },
  { node, fixer, context, importNodes, methodName }
) {
  yield fixer.replaceText(
    node,
    `sample(${buildObjectInText.fromMapOfNodes({
      properties: {
        clock: clockNode,
        source: sourceNode,
        filter: filterNode,
        fn: mapperFunctionNode,
        target: targetNode,
      },
      context,
    })})`
  );

  const importNode = importNodes.get(methodName);

  if (!importNodes.has("sample")) {
    yield fixer.insertTextAfter(importNode, ", sample");
  }
}

module.exports = { replaceForwardBySample, replaceGuardBySample };