quivr/frontend/lib/hooks/useDropzone.ts

85 lines
2.6 KiB
TypeScript

import { FileRejection, useDropzone } from "react-dropzone";
import { useTranslation } from "react-i18next";
import { FeedItemUploadType } from "@/app/chat/[chatId]/components/ActionsBar/types";
import { useEventTracking } from "@/services/analytics/june/useEventTracking";
import { useOnboarding } from "./useOnboarding";
import { useOnboardingTracker } from "./useOnboardingTracker";
import { useToast } from "./useToast";
import { useKnowledgeToFeedContext } from "../context/KnowledgeToFeedProvider/hooks/useKnowledgeToFeedContext";
import { acceptedFormats } from "../helpers/acceptedFormats";
import { cloneFileWithSanitizedName } from "../helpers/cloneFileWithSanitizedName";
// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
export const useCustomDropzone = () => {
const { knowledgeToFeed, addKnowledgeToFeed, setShouldDisplayFeedCard } =
useKnowledgeToFeedContext();
const { isOnboarding } = useOnboarding();
const { trackOnboardingEvent } = useOnboardingTracker();
const files: File[] = (
knowledgeToFeed.filter((c) => c.source === "upload") as FeedItemUploadType[]
).map((c) => c.file);
const { publish } = useToast();
const { track } = useEventTracking();
const { t } = useTranslation(["upload"]);
const onDrop = (acceptedFiles: File[], fileRejections: FileRejection[]) => {
setShouldDisplayFeedCard(true);
if (fileRejections.length > 0) {
const firstRejection = fileRejections[0];
if (firstRejection.errors[0].code === "file-invalid-type") {
publish({ variant: "danger", text: t("invalidFileType") });
} else {
publish({
variant: "danger",
text: t("maxSizeError", { ns: "upload" }),
});
}
return;
}
for (const file of acceptedFiles) {
const isAlreadyInFiles =
files.filter((f) => f.name === file.name && f.size === file.size)
.length > 0;
if (isAlreadyInFiles) {
publish({
variant: "warning",
text: t("alreadyAdded", { fileName: file.name, ns: "upload" }),
});
} else {
if (isOnboarding) {
void trackOnboardingEvent("FILE_UPLOADED");
} else {
void track("FILE_UPLOADED");
}
addKnowledgeToFeed({
source: "upload",
file: cloneFileWithSanitizedName(file),
});
}
}
};
const { getInputProps, getRootProps, isDragActive, open } = useDropzone({
onDrop,
noClick: true,
maxSize: 100000000, // 1 MB
accept: acceptedFormats,
});
return {
getInputProps,
getRootProps,
isDragActive,
open,
};
};