import type { ReactElement } from 'react';
import { Trash } from 'lucide-react';
import { useDeleteDocument } from '../../../hooks/use-delete-document.js';
import { Button } from '../../ui/button.js';
import { ConfirmationModal } from '../index.js';

interface Props {
  documentId: string;
  onChange: () => void;
  onDone?: () => void;
  /**
   * Called when deleting the document left its charge empty, so the server deleted the charge too.
   * The consumer should drop the charge instead of refetching it — `Query.charge` throws for a
   * deleted id, which would leave a stale "shadow charge" on screen.
   */
  onChargeDeleted?: (chargeId: string) => void;
}

export function DeleteDocumentButton({
  documentId,
  onChange,
  onDone,
  onChargeDeleted,
}: Props): ReactElement {
  const { deleteDocument } = useDeleteDocument();

  function onDelete(): void {
    deleteDocument({
      documentId,
    }).then(result => {
      if (!result) {
        // failed: the hook already reported it, keep the modal open
        return;
      }
      if (result.deletedChargeId && onChargeDeleted) {
        onChargeDeleted(result.deletedChargeId);
      } else {
        // no charge was deleted, or the consumer refreshes from a list that isn't charge-scoped
        onChange();
      }
      onDone?.();
    });
  }

  return (
    <ConfirmationModal onConfirm={onDelete} title="Are you sure you want to delete this document?">
      <Button variant="ghost" size="icon" className="size-7.5 text-red-500">
        <Trash className="size-5" />
      </Button>
    </ConfirmationModal>
  );
}