import * as React from 'react';
import { Dialog as DialogPrimitive } from '@base-ui/react/dialog';
import { XIcon } from 'lucide-react';

import { getPortalContainer } from '../../lib/portal';
import { classNames } from '../../lib/utils';

import { Button } from './button';

function Dialog({ ...props }: DialogPrimitive.Root.Props) {
  return <DialogPrimitive.Root data-slot="dialog" {...props} />;
}

const DialogTrigger = React.forwardRef<
  HTMLButtonElement,
  DialogPrimitive.Trigger.Props
>(function DialogTrigger({ ...props }, ref) {
  return (
    <DialogPrimitive.Trigger data-slot="dialog-trigger" {...props} ref={ref} />
  );
});

const DialogPortal = React.forwardRef<
  HTMLDivElement,
  DialogPrimitive.Portal.Props
>(function DialogPortal({ ...props }, ref) {
  return (
    <DialogPrimitive.Portal
      container={getPortalContainer()}
      data-slot="dialog-portal"
      {...props}
      ref={ref}
    />
  );
});

const DialogClose = React.forwardRef<
  HTMLButtonElement,
  DialogPrimitive.Close.Props
>(function DialogClose({ ...props }, ref) {
  return (
    <DialogPrimitive.Close data-slot="dialog-close" {...props} ref={ref} />
  );
});

const DialogOverlay = React.forwardRef<
  HTMLDivElement,
  DialogPrimitive.Backdrop.Props
>(function DialogOverlay({ className, ...props }, ref) {
  return (
    <DialogPrimitive.Backdrop
      data-slot="dialog-overlay"
      className={classNames(
        'fixed inset-0 isolate z-50 bg-black/10 duration-100 supports-backdrop-filter:backdrop-blur-xs data-open:animate-in data-open:fade-in-0 data-closed:animate-out data-closed:fade-out-0',
        className
      )}
      {...props}
      ref={ref}
    />
  );
});

const DialogContent = React.forwardRef<
  HTMLDivElement,
  DialogPrimitive.Popup.Props & {
    showCloseButton?: boolean;
  }
>(function DialogContent(
  { className, children, showCloseButton = true, ...props },
  ref
) {
  return (
    <DialogPortal>
      <DialogOverlay />
      <DialogPrimitive.Popup
        data-slot="dialog-content"
        className={classNames(
          'fixed top-1/2 left-1/2 z-50 grid w-full max-w-[calc(100%-2rem)] -translate-x-1/2 -translate-y-1/2 gap-4 rounded-2xl glass-popover bg-card p-4 text-sm duration-100 outline-none sm:max-w-[680px] data-open:animate-in data-open:fade-in-0 data-open:zoom-in-95 data-closed:animate-out data-closed:fade-out-0 data-closed:zoom-out-95',
          className
        )}
        {...props}
        ref={ref}
      >
        {children}
        {showCloseButton && (
          <DialogPrimitive.Close
            data-slot="dialog-close"
            render={
              <Button
                variant="ghost"
                className="absolute top-2 right-2"
                size="icon-sm"
              />
            }
          >
            <XIcon />
            <span className="sr-only">Close</span>
          </DialogPrimitive.Close>
        )}
      </DialogPrimitive.Popup>
    </DialogPortal>
  );
});

const DialogHeader = React.forwardRef<
  HTMLDivElement,
  React.ComponentProps<'div'>
>(function DialogHeader({ className, ...props }, ref) {
  return (
    <div
      data-slot="dialog-header"
      className={classNames('flex flex-col gap-2', className)}
      {...props}
      ref={ref}
    />
  );
});

const DialogFooter = React.forwardRef<
  HTMLDivElement,
  React.ComponentProps<'div'> & {
    showCloseButton?: boolean;
  }
>(function DialogFooter(
  { className, showCloseButton = false, children, ...props },
  ref
) {
  return (
    <div
      data-slot="dialog-footer"
      className={classNames(
        '-mx-4 -mb-4 flex flex-col-reverse gap-2 rounded-b-2xl border-t bg-muted/50 p-4 sm:flex-row sm:justify-end',
        className
      )}
      {...props}
      ref={ref}
    >
      {children}
      {showCloseButton && (
        <DialogPrimitive.Close render={<Button variant="outline" />}>
          Close
        </DialogPrimitive.Close>
      )}
    </div>
  );
});

const DialogTitle = React.forwardRef<
  HTMLHeadingElement,
  DialogPrimitive.Title.Props
>(function DialogTitle({ className, ...props }, ref) {
  return (
    <DialogPrimitive.Title
      data-slot="dialog-title"
      className={classNames('text-base leading-none font-medium', className)}
      {...props}
      ref={ref}
    />
  );
});

const DialogDescription = React.forwardRef<
  HTMLParagraphElement,
  DialogPrimitive.Description.Props
>(function DialogDescription({ className, ...props }, ref) {
  return (
    <DialogPrimitive.Description
      data-slot="dialog-description"
      className={classNames(
        'text-sm text-muted-foreground *:[a]:underline *:[a]:underline-offset-3 *:[a]:hover:text-foreground',
        className
      )}
      {...props}
      ref={ref}
    />
  );
});

export {
  Dialog,
  DialogClose,
  DialogContent,
  DialogDescription,
  DialogFooter,
  DialogHeader,
  DialogOverlay,
  DialogPortal,
  DialogTitle,
  DialogTrigger,
};