"use client"

import * as React from "react"
import { type OcrRow } from "@/db/schema"
import { type Row } from "@tanstack/react-table"
import { Loader, Trash } from "lucide-react"
import { toast } from "sonner"

import { useMediaQuery } from "@/hooks/use-media-query"
import { Button } from "@/components/ui/button"
import {
  Dialog,
  DialogClose,
  DialogContent,
  DialogDescription,
  DialogFooter,
  DialogHeader,
  DialogTitle,
  DialogTrigger,
} from "@/components/ui/dialog"
import {
  Drawer,
  DrawerClose,
  DrawerContent,
  DrawerDescription,
  DrawerFooter,
  DrawerHeader,
  DrawerTitle,
  DrawerTrigger,
} from "@/components/ui/drawer"

import { removeOcrRows } from "../service"

interface DeleteOcrRowsDialogProps
  extends React.ComponentPropsWithoutRef<typeof Dialog> {
  ocrRows: Row<OcrRow>["original"][]
  showTrigger?: boolean
  onSuccess?: () => void
}

export function DeleteOcrRowsDialog({
  ocrRows,
  showTrigger = true,
  onSuccess,
  ...props
}: DeleteOcrRowsDialogProps) {
  const [isDeletePending, startDeleteTransition] = React.useTransition()
  const isDesktop = useMediaQuery("(min-width: 640px)")

  function onDelete() {
    startDeleteTransition(async () => {
      const { error } = await removeOcrRows({
        ids: ocrRows.map((row) => row.id),
      })

      if (error) {
        toast.error(error)
        return
      }

      props.onOpenChange?.(false)
      toast.success(
        `${ocrRows.length}개의 OCR 데이터가 성공적으로 삭제되었습니다.`
      )
      onSuccess?.()
    })
  }

  if (isDesktop) {
    return (
      <Dialog {...props}>
        {showTrigger ? (
          <DialogTrigger asChild>
            <Button variant="destructive" size="sm">
              <Trash className="mr-2 size-4" aria-hidden="true" />
              삭제 ({ocrRows.length})
            </Button>
          </DialogTrigger>
        ) : null}
        <DialogContent>
          <DialogHeader>
            <DialogTitle>정말로 삭제하시겠습니까?</DialogTitle>
            <DialogDescription>
              이 작업은 되돌릴 수 없습니다. 선택한{" "}
              <span className="font-medium">{ocrRows.length}</span>개의 OCR 데이터가
              서버에서 영구적으로 삭제됩니다.
            </DialogDescription>
          </DialogHeader>
          <DialogFooter className="gap-2 sm:space-x-0">
            <DialogClose asChild>
              <Button variant="outline">취소</Button>
            </DialogClose>
            <Button
              aria-label="선택된 행 삭제"
              variant="destructive"
              onClick={onDelete}
              disabled={isDeletePending}
            >
              {isDeletePending && (
                <Loader
                  className="mr-2 size-4 animate-spin"
                  aria-hidden="true"
                />
              )}
              삭제
            </Button>
          </DialogFooter>
        </DialogContent>
      </Dialog>
    )
  }

  return (
    <Drawer {...props}>
      {showTrigger ? (
        <DrawerTrigger asChild>
          <Button variant="destructive" size="sm">
            <Trash className="mr-2 size-4" aria-hidden="true" />
            삭제 ({ocrRows.length})
          </Button>
        </DrawerTrigger>
      ) : null}
      <DrawerContent>
        <DrawerHeader>
          <DrawerTitle>정말로 삭제하시겠습니까?</DrawerTitle>
          <DrawerDescription>
            이 작업은 되돌릴 수 없습니다. 선택한{" "}
            <span className="font-medium">{ocrRows.length}</span>개의 OCR 데이터가
            서버에서 영구적으로 삭제됩니다.
          </DrawerDescription>
        </DrawerHeader>
        <DrawerFooter className="gap-2 sm:space-x-0">
          <DrawerClose asChild>
            <Button variant="outline">취소</Button>
          </DrawerClose>
          <Button
            aria-label="선택된 행 삭제"
            variant="destructive"
            onClick={onDelete}
            disabled={isDeletePending}
          >
            {isDeletePending && (
              <Loader className="mr-2 size-4 animate-spin" aria-hidden="true" />
            )}
            삭제
          </Button>
        </DrawerFooter>
      </DrawerContent>
    </Drawer>
  )
}