import clsx from "clsx";
import { getBrowser } from "core/utils";
import type { Field } from "data/fields";
import { Switch as RadixSwitch } from "radix-ui";
import {
   type ComponentPropsWithoutRef,
   type ElementType,
   forwardRef,
   useEffect,
   useImperativeHandle,
   useRef,
   useState,
} from "react";
import { TbCalendar, TbChevronDown, TbEye, TbEyeOff, TbInfoCircle } from "react-icons/tb";
import { twMerge } from "tailwind-merge";
import { IconButton } from "ui/components/buttons/IconButton";
import { useEvent } from "ui/hooks/use-event";

export const Group = <E extends ElementType = "div">({
   error,
   as,
   ...props
}: React.ComponentProps<E> & { error?: boolean; as?: E }) => {
   const Tag = as || "div";

   return (
      <Tag
         {...props}
         className={twMerge(
            "flex flex-col gap-1.5",
            as === "fieldset" && "border border-primary/10 p-3 rounded-md",
            as === "fieldset" && error && "border-red-500",
            error && "text-red-500",
            props.className,
         )}
      />
   );
};

export const formElementFactory = (element: string, props: any) => {
   switch (element) {
      case "date":
         return DateInput;
      case "boolean":
         return BooleanInput;
      case "textarea":
         return Textarea;
      default:
         return Input;
   }
};

export const Label = <E extends ElementType = "label">({
   as,
   ...props
}: React.ComponentProps<E> & { as?: E }) => {
   const Tag = as || "label";
   return <Tag {...props} />;
};

export const Help: React.FC<React.ComponentProps<"div">> = ({ className, ...props }) => (
   <div {...props} className={twMerge("text-sm text-primary/50", className)} />
);

export const ErrorMessage: React.FC<React.ComponentProps<"div">> = ({ className, ...props }) => (
   <div {...props} className={twMerge("text-sm text-red-500", className)} />
);

export const FieldLabel: React.FC<React.ComponentProps<"label"> & { field: Field }> = ({
   field,
   ...props
}) => {
   const desc = field.getDescription();
   return (
      <Label {...props} title={desc} className="flex flex-row gap-1 items-center">
         {field.getLabel()}
         {field.isRequired() && <span className="font-medium opacity-30">*</span>}
         {desc && <TbInfoCircle className="opacity-50" />}
      </Label>
   );
};

export const Input = forwardRef<HTMLInputElement, React.ComponentProps<"input">>((props, ref) => {
   const disabledOrReadonly = props.disabled || props.readOnly;
   return (
      <input
         type="text"
         {...props}
         ref={ref}
         className={twMerge(
            "bg-muted/40 h-11 rounded-md py-2.5 px-4 outline-none w-full disabled:cursor-not-allowed",
            disabledOrReadonly && "bg-muted/50 text-primary/50",
            !disabledOrReadonly &&
               "focus:bg-muted focus:outline-none focus:ring-2 focus:ring-zinc-500 focus:border-transparent transition-all",
            props.className,
         )}
      />
   );
});

export const TypeAwareInput = forwardRef<HTMLInputElement, React.ComponentProps<"input">>(
   (props, ref) => {
      if (props.type === "password") {
         return <Password {...props} ref={ref} />;
      }

      return <Input {...props} ref={ref} />;
   },
);

export const Password = forwardRef<HTMLInputElement, React.ComponentProps<"input">>(
   (props, ref) => {
      const [visible, setVisible] = useState(false);

      function handleToggle() {
         setVisible((v) => !v);
      }

      return (
         <div className="relative w-full">
            <Input {...props} type={visible ? "text" : "password"} className="w-full" ref={ref} />
            <div className="absolute right-3 top-0 bottom-0 flex items-center">
               <IconButton
                  Icon={visible ? TbEyeOff : TbEye}
                  onClick={handleToggle}
                  variant="ghost"
                  className="opacity-70"
               />
            </div>
         </div>
      );
   },
);

export const Textarea = forwardRef<HTMLTextAreaElement, React.ComponentProps<"textarea">>(
   (props, ref) => {
      return (
         <textarea
            rows={3}
            {...props}
            ref={ref}
            className={twMerge(
               "bg-muted/40 min-h-11 rounded-md py-2.5 px-4 focus:bg-muted outline-none focus:outline-none focus:ring-2 focus:ring-zinc-500 focus:border-transparent transition-all disabled:bg-muted/50 disabled:text-primary/50",
               props.className,
            )}
         />
      );
   },
);

export const DateInput = forwardRef<HTMLInputElement, React.ComponentProps<"input">>(
   (props, ref) => {
      const innerRef = useRef<HTMLInputElement>(null);
      const browser = getBrowser();
      useImperativeHandle(ref, () => innerRef.current!);

      const handleClick = useEvent(() => {
         if (innerRef?.current) {
            innerRef.current.focus();
            if (["Safari"].includes(browser)) {
               innerRef.current.click();
            } else {
               innerRef.current.showPicker();
            }
         }
      });

      return (
         <div className="relative w-full">
            <div className="absolute h-full right-3 top-0 bottom-0 flex items-center">
               <IconButton Icon={TbCalendar} onClick={handleClick} />
            </div>
            <Input
               {...props}
               type={props.type ?? "date"}
               ref={innerRef}
               className="w-full appearance-none"
            />
         </div>
      );
   },
);

export const BooleanInput = forwardRef<HTMLInputElement, React.ComponentProps<"input">>(
   (props, ref) => {
      const [checked, setChecked] = useState(Boolean(props.value));

      useEffect(() => {
         setChecked(Boolean(props.value));
      }, [props.value]);

      function handleCheck(e) {
         setChecked(e.target.checked);
         props.onChange?.(e.target.checked);
      }

      return (
         <div className="h-11 flex items-center">
            <input
               {...props}
               type="checkbox"
               ref={ref}
               className="outline-none focus:outline-none focus:ring-2 focus:ring-zinc-500  transition-all disabled:opacity-70 scale-150 ml-1"
               checked={checked}
               onChange={handleCheck}
               disabled={props.disabled}
            />
         </div>
      );
   },
);

export type SwitchValue = boolean | 1 | 0 | "true" | "false" | "on" | "off";
const SwitchSizes = {
   xs: {
      root: "h-5 w-8",
      thumb: "data-[state=checked]:left-[calc(100%-1rem)]",
   },
   sm: {
      root: "h-6 w-10",
      thumb: "data-[state=checked]:left-[calc(100%-1.25rem)]",
   },
   md: {
      root: "h-7 w-12",
      thumb: "data-[state=checked]:left-[calc(100%-1.5rem)]",
   },
};

export const Switch = forwardRef<
   HTMLButtonElement,
   Pick<
      ComponentPropsWithoutRef<"input">,
      "name" | "required" | "disabled" | "checked" | "defaultChecked" | "id" | "type"
   > & {
      value?: SwitchValue;
      size?: keyof typeof SwitchSizes;
      onChange?: (e: { target: { value: boolean } }) => void;
      onCheckedChange?: (checked: boolean) => void;
   }
>(({ type, required, ...props }, ref) => {
   return (
      <RadixSwitch.Root
         className={clsx(
            "relative cursor-pointer rounded-full bg-muted border-2 border-transparent outline-none ring-1 dark:ring-primary/10 ring-primary/20 data-[state=checked]:ring-primary/60 data-[state=checked]:bg-primary/60 appearance-none transition-colors hover:bg-muted/80",
            SwitchSizes[props.size ?? "md"].root,
            props.disabled && "opacity-50 !cursor-not-allowed",
         )}
         onCheckedChange={(bool) => {
            console.log("setting", bool);
            props.onChange?.({ target: { value: bool } });
         }}
         {...(props as any)}
         checked={
            typeof props.checked !== "undefined"
               ? props.checked
               : typeof props.value !== "undefined"
                 ? Boolean(props.value)
                 : undefined
         }
         ref={ref}
      >
         <RadixSwitch.Thumb
            className={clsx(
               "absolute top-0 left-0 h-full aspect-square rounded-full bg-primary/30 data-[state=checked]:bg-background transition-[left,right] duration-100 border border-muted",
               SwitchSizes[props.size ?? "md"].thumb,
            )}
         />
      </RadixSwitch.Root>
   );
});

export const Select = forwardRef<
   HTMLSelectElement,
   React.ComponentProps<"select"> & {
      options?: { value: string; label: string }[] | (string | number)[];
   }
>(({ children, options, ...props }, ref) => (
   <div className="flex w-full relative">
      <select
         {...props}
         ref={ref}
         className={twMerge(
            "bg-muted/40 focus:bg-muted rounded-md py-2.5 px-4 outline-none focus:outline-none focus:ring-2 focus:ring-zinc-500 focus:border-transparent transition-all disabled:bg-muted/50 disabled:text-primary/50",
            "appearance-none h-11 w-full",
            !props.multiple && "border-r-8 border-r-transparent",
            props.className,
         )}
      >
         {options ? (
            <>
               {!props.required && <option value="" />}
               {options
                  .map((o) => {
                     if (typeof o !== "object") {
                        return { value: o, label: String(o) };
                     }
                     return o;
                  })
                  .map((opt) => (
                     <option key={opt.value} value={opt.value}>
                        {opt.label}
                     </option>
                  ))}
            </>
         ) : (
            children
         )}
      </select>
      {!props.multiple && (
         <TbChevronDown
            className="absolute right-3 top-0 bottom-0 h-full opacity-70 pointer-events-none"
            size={18}
         />
      )}
   </div>
));
