jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor: block non supported file types

Browse source
This commit is contained in:
igorrCarvalho 2024-06-10 11:35:40 -03:00
commit 92b398d29b
4 changed files with 8 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/frontend/src/modals/IOModal/components/chatView/chatInput/hooks/use-drag-and-drop.tsx
View file

@ -4,7 +4,7 @@ import useAlertStore from "../../../../../../stores/alertStore";
const fsErrorText = const fsErrorText =
"Please ensure your file has one of the following extensions:"; "Please ensure your file has one of the following extensions:";
const snErrorTxt = "png, jpg, jpeg, gif, bmp, webp"; const snErrorTxt = "png, jpg, jpeg";
const useDragAndDrop = ( const useDragAndDrop = (
setIsDragging, setIsDragging,
@ -49,7 +49,7 @@ const useDragAndDrop = (
const handleFiles = (files, setFiles, currentFlowId, setErrorData) => { const handleFiles = (files, setFiles, currentFlowId, setErrorData) => {
if (files) { if (files) {
const allowedExtensions = ["png", "jpg", "jpeg", "gif", "bmp", "webp"]; const allowedExtensions = ["png", "jpg", "jpeg"];
const file = files?.[0]; const file = files?.[0];
const fileExtension = file.name.split(".").pop()?.toLowerCase(); const fileExtension = file.name.split(".").pop()?.toLowerCase();
if (!fileExtension || !allowedExtensions.includes(fileExtension)) { if (!fileExtension || !allowedExtensions.includes(fileExtension)) {

4
src/frontend/src/modals/IOModal/components/chatView/chatInput/hooks/use-handle-file-change.tsx
View file

@ -4,7 +4,7 @@ import useAlertStore from "../../../../../../stores/alertStore";
const fsErrorText = const fsErrorText =
"Please ensure your file has one of the following extensions:"; "Please ensure your file has one of the following extensions:";
const snErrorTxt = "png, jpg, jpeg, gif, bmp, webp"; const snErrorTxt = "png, jpg, jpeg";
export const useHandleFileChange = (setFiles, currentFlowId) => { export const useHandleFileChange = (setFiles, currentFlowId) => {
const setErrorData = useAlertStore((state) => state.setErrorData); const setErrorData = useAlertStore((state) => state.setErrorData);
@ -14,7 +14,7 @@ export const useHandleFileChange = (setFiles, currentFlowId) => {
const fileInput = event.target; const fileInput = event.target;
const file = fileInput.files?.[0]; const file = fileInput.files?.[0];
if (file) { if (file) {
const allowedExtensions = ["png", "jpg", "jpeg", "gif", "bmp", "webp"]; const allowedExtensions = ["png", "jpg", "jpeg"];
const fileExtension = file.name.split(".").pop()?.toLowerCase(); const fileExtension = file.name.split(".").pop()?.toLowerCase();
if (!fileExtension || !allowedExtensions.includes(fileExtension)) { if (!fileExtension || !allowedExtensions.includes(fileExtension)) {

11
src/frontend/src/modals/IOModal/components/chatView/chatInput/hooks/use-upload.tsx
View file

@ -5,7 +5,7 @@ import useAlertStore from "../../../../../../stores/alertStore";
const fsErrorText = const fsErrorText =
"Please ensure your file has one of the following extensions:"; "Please ensure your file has one of the following extensions:";
const snErrorTxt = "png, jpg, jpeg, gif, bmp, webp"; const snErrorTxt = "png, jpg, jpeg";
const useUpload = (uploadFile, currentFlowId, setFiles, lockChat) => { const useUpload = (uploadFile, currentFlowId, setFiles, lockChat) => {
const setErrorData = useAlertStore((state) => state.setErrorData); const setErrorData = useAlertStore((state) => state.setErrorData);
@ -21,14 +21,7 @@ const useUpload = (uploadFile, currentFlowId, setFiles, lockChat) => {
const uid = new ShortUniqueId({ length: 3 }); const uid = new ShortUniqueId({ length: 3 });
const blob = items[i].getAsFile(); const blob = items[i].getAsFile();
if (blob) { if (blob) {
const allowedExtensions = [ const allowedExtensions = ["png", "jpg", "jpeg"];
"png",
"jpg",
"jpeg",
"gif",
"bmp",
"webp",
];
const fileExtension = blob.name.split(".").pop()?.toLowerCase(); const fileExtension = blob.name.split(".").pop()?.toLowerCase();
if (!fileExtension || !allowedExtensions.includes(fileExtension)) { if (!fileExtension || !allowedExtensions.includes(fileExtension)) {

2
src/frontend/src/modals/IOModal/components/chatView/fileComponent/index.tsx
View file

@ -32,6 +32,8 @@ export default function FileCard({
BACKEND_URL.length - 1, BACKEND_URL.length - 1,
)}${BASE_URL_API}files/images/${content}`; )}${BASE_URL_API}files/images/${content}`;
console.log(imgSrc);
if (showFile) { if (showFile) {
if (imgTypes.has(fileType)) { if (imgTypes.has(fileType)) {
return ( return (