import React, { useLayoutEffect, useState } from "react";
import { createRoot } from "react-dom/client";
import * as AlertDialog from "@radix-ui/react-alert-dialog";
import Btn, { BtnProps } from "@/app/components/Btn";

import Warning from "@/app/icons/warning.svg";
import Close from "@/app/icons/closeIcon.svg";

export interface ModalProps {
  onOk?: () => void;
  onCancel?: () => void;
  okText?: string;
  cancelText?: string;
  okBtnProps?: BtnProps;
  cancelBtnProps?: BtnProps;
  content?:
    | React.ReactNode
    | ((handlers: { close: () => void }) => JSX.Element);
  title?: React.ReactNode;
  visible?: boolean;
  noFooter?: boolean;
  noHeader?: boolean;
  isMobile?: boolean;
  closeble?: boolean;
  type?: "modal" | "bottom-drawer";
  headerBordered?: boolean;
  modelClassName?: string;
  onOpen?: (v: boolean) => void;
}

export interface WarnProps
  extends Omit<
    ModalProps,
    | "closeble"
    | "isMobile"
    | "noHeader"
    | "noFooter"
    | "onOk"
    | "okBtnProps"
    | "cancelBtnProps"
    | "content"
  > {
  onOk?: () => Promise<void> | void;
  content?: React.ReactNode;
}

export interface TriggerProps
  extends Omit<ModalProps, "visible" | "onOk" | "onCancel"> {
  children: JSX.Element;
  className?: string;
}

const baseZIndex = 150;

const Modal = (props: ModalProps) => {
  const {
    onOk,
    onCancel,
    okText,
    cancelText,
    content,
    title,
    visible,
    noFooter,
    noHeader,
    closeble = true,
    okBtnProps,
    cancelBtnProps,
    type = "modal",
    headerBordered,
    modelClassName,
    onOpen,
  } = props;

  const [open, setOpen] = useState(!!visible);

  const mergeOpen = visible ?? open;

  const handleClose = () => {
    setOpen(false);
    onCancel?.();
  };

  const handleOk = () => {
    setOpen(false);
    onOk?.();
  };

  useLayoutEffect(() => {
    onOpen?.(mergeOpen);
  }, [mergeOpen]);

  let layoutClassName = "";
  let panelClassName = "";
  let titleClassName = "";
  let footerClassName = "";

  switch (type) {
    case "bottom-drawer":
      layoutClassName = "fixed inset-0 flex flex-col w-[100%] bottom-0";
      panelClassName =
        "rounded-t-chat-model-select overflow-y-auto overflow-x-hidden";
      titleClassName = "px-4 py-3";
      footerClassName = "absolute w-[100%]";
      break;
    case "modal":
    default:
      layoutClassName =
        "fixed inset-0 flex flex-col item-start top-0 left-[50vw] translate-x-[-50%] max-sm:w-modal-modal-type-mobile";
      panelClassName = "rounded-lg px-6 sm:w-modal-modal-type";
      titleClassName = "py-6 max-sm:pb-3";
      footerClassName = "py-6";
  }
  const btnCommonClass = "px-4 py-2.5 rounded-md max-sm:flex-1";
  const { className: okBtnClass } = okBtnProps || {};
  const { className: cancelBtnClass } = cancelBtnProps || {};

  return (
    <AlertDialog.Root open={mergeOpen} onOpenChange={setOpen}>
      <AlertDialog.Portal>
        <AlertDialog.Overlay
          className="bg-modal-mask fixed inset-0 animate-mask "
          style={{ zIndex: baseZIndex - 1 }}
        />
        <AlertDialog.Content
          className={`
            ${layoutClassName}
          `}
          style={{ zIndex: baseZIndex - 1 }}
        >
          <div className="flex-1">&nbsp;</div>
          <div
            className={`flex flex-col flex-0      
              bg-moda-panel text-modal-mask    
              ${modelClassName}
              ${panelClassName}
            `}
          >
            {!noHeader && (
              <AlertDialog.Title
                className={`
                      flex items-center justify-between gap-3 font-common
                      md:text-chat-header-title md:font-bold md:leading-5
                      ${
                        headerBordered
                          ? " border-b border-modal-header-bottom"
                          : ""
                      }
                      ${titleClassName}
                  `}
              >
                <div className="flex gap-3 justify-start flex-1 items-center">
                  {title}
                </div>
                {closeble && (
                  <div
                    className="items-center"
                    onClick={() => {
                      handleClose();
                    }}
                  >
                    <Close />
                  </div>
                )}
              </AlertDialog.Title>
            )}
            <div className="flex-1 overflow-hidden">
              {typeof content === "function"
                ? content({
                    close: () => {
                      handleClose();
                    },
                  })
                : content}
            </div>
            {!noFooter && (
              <div
                className={`
                  flex gap-3 sm:justify-end max-sm:justify-between
                  ${footerClassName}
                  `}
              >
                <AlertDialog.Cancel asChild>
                  <Btn
                    {...cancelBtnProps}
                    onClick={() => handleClose()}
                    text={cancelText}
                    className={`${btnCommonClass} ${cancelBtnClass}`}
                  />
                </AlertDialog.Cancel>
                <AlertDialog.Action asChild>
                  <Btn
                    {...okBtnProps}
                    onClick={handleOk}
                    text={okText}
                    className={`${btnCommonClass} ${okBtnClass}`}
                  />
                </AlertDialog.Action>
              </div>
            )}
          </div>
          {type === "modal" && <div className="flex-1">&nbsp;</div>}
        </AlertDialog.Content>
      </AlertDialog.Portal>
    </AlertDialog.Root>
  );
};

export const Warn = ({
  title,
  onOk,
  visible,
  content,
  ...props
}: WarnProps) => {
  const [internalVisible, setVisible] = useState(visible);

  return (
    <Modal
      {...props}
      title={
        <>
          <Warning />
          {title}
        </>
      }
      content={
        <AlertDialog.Description
          className={`
                    font-common font-normal
                    md:text-sm-title md:leading-[158%]
                `}
        >
          {content}
        </AlertDialog.Description>
      }
      closeble={false}
      onOk={() => {
        const toDo = onOk?.();
        if (toDo instanceof Promise) {
          toDo.then(() => {
            setVisible(false);
          });
        } else {
          setVisible(false);
        }
      }}
      visible={internalVisible}
      okBtnProps={{
        className: `bg-delete-chat-ok-btn text-text-delete-chat-ok-btn `,
      }}
      cancelBtnProps={{
        className: `bg-delete-chat-cancel-btn  border border-delete-chat-cancel-btn text-text-delete-chat-cancel-btn`,
      }}
    />
  );
};

const div = document.createElement("div");
div.id = "confirm-root";
div.style.height = "0px";
document.body.appendChild(div);

Modal.warn = (props: Omit<WarnProps, "visible" | "onCancel" | "onOk">) => {
  const root = createRoot(div);
  const closeModal = () => {
    root.unmount();
  };

  return new Promise<boolean>((resolve) => {
    root.render(
      <Warn
        {...props}
        visible={true}
        onCancel={() => {
          closeModal();
          resolve(false);
        }}
        onOk={() => {
          closeModal();
          resolve(true);
        }}
      />,
    );
  });
};

export const Trigger = (props: TriggerProps) => {
  const { children, className, content, ...rest } = props;

  const [internalVisible, setVisible] = useState(false);

  return (
    <>
      <div
        className={className}
        onClick={() => {
          setVisible(true);
        }}
      >
        {children}
      </div>
      <Modal
        {...rest}
        visible={internalVisible}
        onCancel={() => {
          setVisible(false);
        }}
        content={
          typeof content === "function"
            ? content({
                close: () => {
                  setVisible(false);
                },
              })
            : content
        }
      />
    </>
  );
};

Modal.Trigger = Trigger;

export default Modal;