import React, { ReactElement } from "react";
import { createRoot } from "react-dom/client";
import { Controller } from "@hotwired/stimulus";
import { ReactModule } from "./index";

export default class extends Controller {
  declare readonly componentValue?: string;
  declare readonly propsValue?: object;

  static values = {
    component: String,
    props: Object,
  };

  connect() {
    const props = this.propsValue ? this.propsValue : null;

    this.dispatchEvent("connect", { component: this.componentValue, props: props });

    if (!this.componentValue) {
      throw new Error("No component specified.");
    }

    const importedReactModule = window.resolveReactComponent(this.componentValue);

    const onload = (reactModule: ReactModule) => {
      const component = reactModule.default;
      this._renderReactElement(React.createElement(component, props, null));

      this.dispatchEvent("mount", {
        componentName: this.componentValue,
        component: component,
        props: props,
      });
    };

    if (typeof importedReactModule === "function") {
      importedReactModule().then(onload);
    } else {
      onload(importedReactModule);
    }
  }

  disconnect() {
    (this.element as any).root.unmount();
    this.dispatchEvent("unmount", {
      component: this.componentValue,
      props: this.propsValue ? this.propsValue : null,
    });
  }

  _renderReactElement(reactElement: ReactElement) {
    const element: any = this.element as any;

    // If a root has already been created for this element, reuse it
    if (!element.root) {
      element.root = createRoot(this.element);
    }
    element.root.render(reactElement);
  }

  private dispatchEvent(name: string, payload: any) {
    this.dispatch(name, { detail: payload, prefix: "react" });
  }
}