import { Dialog, DialogBackdrop, DialogPanel, DialogTitle } from '@headlessui/react'
import StyledButton, { StyledButtonProps } from './StyledButton'
import React from 'react'
import classNames from '~/lib/classNames'
import { useTranslation } from 'react-i18next'

export type StyledModalProps = {
  onClose?: () => void
  title: string
  cancelText?: string
  cancelIcon?: StyledButtonProps['icon']
  cancelLoading?: boolean
  confirmText?: string
  confirmIcon?: StyledButtonProps['icon']
  confirmVariant?: StyledButtonProps['variant']
  confirmLoading?: boolean
  open: boolean
  onCancel?: () => void
  onConfirm?: () => void
  children: React.ReactNode
  icon?: React.ReactNode
  large?: boolean
}

const StyledModal: React.FC<StyledModalProps> = ({
  children,
  title,
  open,
  onClose,
  cancelText,
  cancelIcon,
  cancelLoading = false,
  confirmText,
  confirmIcon,
  confirmVariant = 'action',
  confirmLoading = false,
  onCancel,
  onConfirm,
  icon,
  large = false,
}) => {
  const { t } = useTranslation()
  const resolvedCancelText = cancelText ?? t('common.cancel')
  const resolvedConfirmText = confirmText ?? t('common.confirm')
  return (
    <Dialog
      open={open}
      onClose={() => {
        if (onClose) onClose()
      }}
      className="relative z-50"
    >
      <DialogBackdrop
        transition
        className="fixed inset-0 bg-black/50 transition-opacity data-[closed]:opacity-0 data-[enter]:duration-300 data-[leave]:duration-200 data-[enter]:ease-out data-[leave]:ease-in"
      />
      <div className="fixed inset-0 z-10 w-screen overflow-y-auto">
        <div
          className={classNames(
            'flex min-h-full items-end justify-center p-4 text-center sm:items-center !w-screen',
            large ? 'sm:px-4' : 'sm:p-0'
          )}
        >
          <DialogPanel
            transition
            className={classNames(
              'relative transform overflow-hidden rounded-lg bg-surface-primary px-4 pb-4 pt-5 text-left shadow-xl transition-all data-[closed]:translate-y-4 data-[closed]:opacity-0 data-[enter]:duration-300 data-[leave]:duration-200 data-[enter]:ease-out data-[leave]:ease-in sm:my-8  sm:p-6 data-[closed]:sm:translate-y-0 data-[closed]:sm:scale-95',
              large ? 'sm:max-w-7xl !w-full' : 'sm:max-w-lg'
            )}
          >
            <div>
              {icon && <div className="flex items-center justify-center">{icon}</div>}
              <div className="mt-3 text-center sm:mt-5">
                <DialogTitle as="h3" className="text-base font-semibold text-text-primary">
                  {title}
                </DialogTitle>
                <div className="mt-2 !h-fit">{children}</div>
              </div>
            </div>
            <div className="mt-5 sm:mt-6 sm:grid sm:grid-flow-row-dense sm:grid-cols-2 sm:gap-3">
              {resolvedCancelText && onCancel && (
                <StyledButton
                  variant="outline"
                  onClick={() => {
                    if (onCancel) onCancel()
                  }}
                  icon={cancelIcon}
                  loading={cancelLoading}
                >
                  {resolvedCancelText}
                </StyledButton>
              )}
              {resolvedConfirmText && onConfirm && (
                <StyledButton
                  variant={confirmVariant}
                  onClick={() => {
                    if (onConfirm) onConfirm()
                  }}
                  icon={confirmIcon}
                  loading={confirmLoading}
                >
                  {resolvedConfirmText}
                </StyledButton>
              )}
            </div>
          </DialogPanel>
        </div>
      </div>
    </Dialog>
  )
}

export default StyledModal