import { FloatingArrow } from "@floating-ui/react";
import { clsx } from "clsx";
import React, { createElement, useEffect } from "react";

import Portal from "./portal";
import TabLoop from "./tab_loop";
import withFloating from "./with_floating";

import type { FloatingProps } from "./with_floating";
import type { ReactNode } from "react";

interface PortalProps extends Omit<
  React.ComponentPropsWithoutRef<typeof Portal>,
  "children"
> {}
interface TabLoopProps extends Omit<
  React.ComponentPropsWithoutRef<typeof TabLoop>,
  "children"
> {}

interface PopperComponentProps
  extends Omit<PortalProps, "portalId">, TabLoopProps, FloatingProps {
  className?: string;
  wrapperClassName?: string;
  popperComponent: React.ReactNode;
  popperContainer?: React.FC<{ children?: ReactNode | undefined }>;
  targetComponent: React.ReactNode;
  popperOnKeyDown: React.KeyboardEventHandler<HTMLDivElement>;
  showArrow?: boolean;
  portalId?: PortalProps["portalId"];
  popperTargetRef?: React.RefObject<HTMLElement | null>;
  monthHeaderPosition?: "top" | "middle" | "bottom";
}

// Exported for testing purposes
export const PopperComponent: React.FC<PopperComponentProps> = (props) => {
  const {
    className,
    wrapperClassName,
    hidePopper = true,
    popperComponent,
    targetComponent,
    enableTabLoop,
    popperOnKeyDown,
    portalId,
    portalHost,
    popperProps,
    showArrow,
    popperTargetRef,
    monthHeaderPosition,
  } = props;

  // When a custom popperTargetRef is provided, use it as the position reference
  // This allows the popper to be positioned relative to a specific element
  // within the custom input, rather than the wrapper div
  useEffect(() => {
    if (popperTargetRef?.current) {
      popperProps.refs.setPositionReference(popperTargetRef.current);
    }
  }, [popperTargetRef, popperProps.refs]);

  let popper: React.ReactElement | undefined = undefined;

  if (!hidePopper) {
    const classes = clsx(
      "react-datepicker-popper",
      !showArrow && "react-datepicker-popper-offset",
      monthHeaderPosition === "middle" &&
        "react-datepicker-popper--header-middle",
      monthHeaderPosition === "bottom" &&
        "react-datepicker-popper--header-bottom",
      className,
    );
    popper = (
      <TabLoop enableTabLoop={enableTabLoop}>
        {/* eslint-disable react-hooks/refs -- Floating UI values are designed to be used during render */}
        <div
          ref={popperProps.refs.setFloating}
          style={popperProps.floatingStyles}
          className={classes}
          data-placement={popperProps.placement}
          onKeyDown={popperOnKeyDown}
        >
          {popperComponent}
          {showArrow && (
            <FloatingArrow
              ref={popperProps.arrowRef}
              context={popperProps.context}
              fill="currentColor"
              strokeWidth={1}
              height={8}
              width={16}
              style={{ transform: "translateY(-1px)" }}
              className="react-datepicker__triangle"
            />
          )}
        </div>
        {/* eslint-enable react-hooks/refs */}
      </TabLoop>
    );
  }

  if (props.popperContainer) {
    popper = createElement(props.popperContainer, {}, popper);
  }

  if (portalId && !hidePopper) {
    popper = (
      <Portal portalId={portalId} portalHost={portalHost}>
        {popper}
      </Portal>
    );
  }

  const wrapperClasses = clsx("react-datepicker-wrapper", wrapperClassName);

  return (
    <>
      {/* eslint-disable-next-line react-hooks/refs -- Floating UI refs are designed to be used during render */}
      <div ref={popperProps.refs.setReference} className={wrapperClasses}>
        {targetComponent}
      </div>
      {popper}
    </>
  );
};

export default withFloating<PopperComponentProps>(PopperComponent);