import { EntityMapping } from "@/app/config/entityLabels";

interface DownloadActionsProps {
  outputText: string;
  entityMappings?: EntityMapping[];
  anonymizedText?: string; // Nouveau paramètre pour le texte déjà anonymisé par Presidio
}

export const useDownloadActions = ({
  outputText,
  anonymizedText, // Texte déjà anonymisé par Presidio
}: DownloadActionsProps) => {
  const copyToClipboard = () => {
    // Toujours utiliser le texte anonymisé de Presidio
    const textToCopy = anonymizedText || outputText;
    navigator.clipboard.writeText(textToCopy);
  };

  const downloadText = () => {
    // Utiliser le texte anonymisé de Presidio si disponible, sinon fallback sur outputText
    const textToDownload = anonymizedText || outputText;
    const blob = new Blob([textToDownload], { type: "text/plain" });
    const url = URL.createObjectURL(blob);
    const a = document.createElement("a");
    a.href = url;
    a.download = "texte-anonymise.txt";
    document.body.appendChild(a);
    a.click();
    document.body.removeChild(a);
    URL.revokeObjectURL(url);
  };

  return { copyToClipboard, downloadText };
};
