const visited = 'visitedByInvariantWarningTransformer';

const warningDevCheckTemplate = `
  if (process.env.NODE_ENV !== 'production') {
    NODE;
  }
`.trim();

const plugin = ({ template, types: t }) => {
  const wrapWithDevCheck = template(warningDevCheckTemplate, {
    placeholderPattern: /^NODE$/,
  });

  return {
    visitor: {
      CallExpression(path) {
        const { name } = path.node.callee;
        if (
          (name === 'warn' || name === 'deprecationWarning') &&
          !path.node[visited]
        ) {
          path.node[visited] = true;

          // The production-check may be hoisted if the parent
          // is already an if-statement only containing the
          // warn call
          let p = path;
          while (t.isExpressionStatement(p.parentPath.node)) {
            if (
              t.isBlockStatement(p.parentPath.parentPath.node) &&
              p.parentPath.parentPath.node.body.length === 1 &&
              p.parentPath.parentPath.node.body[0] === path.parentPath.node &&
              t.isIfStatement(p.parentPath.parentPath.parentPath.node) &&
              p.parentPath.parentPath.parentPath.node.consequent ===
                p.parentPath.parentPath.node &&
              !p.parentPath.parentPath.node.alternate
            ) {
              p = p.parentPath.parentPath.parentPath;
            } else if (
              t.isIfStatement(p.parentPath.parentPath.node) &&
              p.parentPath.parentPath.node.consequent === p.parentPath.node &&
              !p.parentPath.parentPath.node.alternate
            ) {
              p = path.parentPath.parentPath;
            } else {
              break;
            }
          }

          p.replaceWith(wrapWithDevCheck({ NODE: p.node }));
        } else if (name === 'invariant' && !path.node[visited]) {
          path.node[visited] = true;
          const formerNode = path.node.arguments[1];
          path.node.arguments[1] = t.conditionalExpression(
            t.binaryExpression(
              '!==',
              t.memberExpression(
                t.memberExpression(
                  t.identifier('process'),
                  t.identifier('env')
                ),
                t.identifier('NODE_ENV')
              ),
              t.stringLiteral('production')
            ),
            formerNode,
            t.stringLiteral('')
          );
        }
      },
    },
  };
};

export default plugin;