import jsx from "@babel/plugin-syntax-jsx";
import { helper } from "@babel/helper-builder-react-jsx-experimental";
import { declare } from "@babel/helper-plugin-utils";
import { types as t } from "@babel/core";

export default declare((api, options) => {
  const { runtime = "classic" } = options;
  const PURE_ANNOTATION = options.pure;

  const visitor = helper(api, {
    pre(state) {
      const tagName = state.tagName;
      const args = state.args;
      if (t.react.isCompatTag(tagName)) {
        args.push(t.stringLiteral(tagName));
      } else {
        args.push(state.tagExpr);
      }
    },

    post(state, pass) {
      if (pass.get("@babel/plugin-react-jsx/runtime") === "classic") {
        state.createElementCallee = pass.get(
          "@babel/plugin-react-jsx/createElementIdentifier",
        )();

        state.pure =
          PURE_ANNOTATION ?? !pass.get("@babel/plugin-react-jsx/pragmaSet");
      } else {
        const getter = get => ({ enumerable: true, configurable: true, get });

        // TODO(Babel 8): helper-builder-react-jsx expects those properties to be AST nodes, but we want to
        // generate them lazily so that we only inject imports when needed.
        // These should actually be functions.
        Object.defineProperties(state, {
          jsxCallee: getter(pass.get("@babel/plugin-react-jsx/jsxIdentifier")),
          jsxStaticCallee: getter(
            pass.get("@babel/plugin-react-jsx/jsxStaticIdentifier"),
          ),
          createElementCallee: getter(
            pass.get("@babel/plugin-react-jsx/createElementIdentifier"),
          ),
        });

        state.pure =
          PURE_ANNOTATION ??
          !pass.get("@babel/plugin-react-jsx/importSourceSet");
      }
    },

    ...options,
    development: false,
    runtime,
  });

  return {
    name: "transform-react-jsx",
    inherits: jsx,
    visitor,
  };
});