import vm from "vm";
import React, { ComponentType } from "react";
import { nodeResolve } from "@rollup/plugin-node-resolve";
import commonjs from "@rollup/plugin-commonjs";
import json from "@rollup/plugin-json";
import replace from "@rollup/plugin-replace";
import sucrase from "@rollup/plugin-sucrase";
import alias from "@rollup/plugin-alias";
import externalGlobals from "rollup-plugin-external-globals";
import { createScript } from "../script";
import orgStyled from "styled-components";
import * as styledExports from "styled-components";
import ReactHelmetAsync from "react-helmet-async";
import { resolve } from "path";

const styled = orgStyled.bind(null);
for (let key of Object.keys(orgStyled)) {
  if (key === "default") {
    continue;
  }
  (styled as any)[key] = (orgStyled as any)[key];
}
for (let key of Object.keys(styledExports)) {
  if (key === "default") {
    continue;
  }
  (styled as any)[key] = (styledExports as any)[key];
}

const createReact = <TProps = any>(path: string) => {
  const script = createScript({
    path,
    format: "cjs",
    plugins: [
      replace({
        preventAssignment: true,
        "process.env.NODE_ENV": JSON.stringify("production"),
      }),
      alias({
        entries: [
          { find: "@", replacement: resolve("content/templates/react") },
        ],
      }),
      sucrase({
        exclude: ["node_modules/**"],
        transforms: ["jsx", "typescript"],
      }),
      nodeResolve({
        browser: true,
        preferBuiltins: false,
        extensions: [".js", ".ts", ".tsx"],
      }),
      json(),
      commonjs({
        include: /node_modules/,
      }),
      externalGlobals({
        react: "React",
        "styled-components": "StyledComponents",
        "react-helmet-async": "ReactHelmetAsync",
      }),
    ],
  });
  const template = script.pipe<ComponentType<TProps>>(async () => {
    const scriptContent = await script.data;
    const exports: any = {};
    const module = { exports };
    const globals = {
      module,
      exports,
      React,
      StyledComponents: styled,
      ReactHelmetAsync,
    };
    vm.runInNewContext(scriptContent, globals);
    return module.exports;
  });

  return template;
};

export { createReact };