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

Refactor: Create specific folder for handleDownload function

Browse source
This commit is contained in:
igorrCarvalho 2024-05-30 17:05:20 -03:00
commit 7ea7d9a407
2 changed files with 38 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -8,6 +8,7 @@ import useFlowsManagerStore from "../../../../../stores/flowsManagerStore";
import { BACKEND_URL, BASE_URL_API } from "../../../../../constants/constants";
import formatFileName from "../filePreviewChat/utils/format-file-name";
import DownloadButton from "./components/downloadButton/downloadButton";
import handleDownload from "./utils/handle-download";
const imgTypes = new Set(["png", "jpg"]);
@ -17,26 +18,6 @@ export default function FileCard({
fileType,
showFile = true,
}: fileCardPropsType): JSX.Element | undefined {
const handleDownload = async (): Promise<void> => {
try {
const response = await fetch(
`${BACKEND_URL.slice(0, BACKEND_URL.length - 1)}${BASE_URL_API}files/download/${content}`,
);
const blob = await response.blob();
const url = URL.createObjectURL(blob);
const link = document.createElement("a");
link.href = url;
link.setAttribute("download", fileName); // Set the filename
document.body.appendChild(link);
link.click();
document.body.removeChild(link);
URL.revokeObjectURL(url); // Clean up the URL object
} catch (error) {
console.error("Failed to download file:", error);
}
};
const [isHovered, setIsHovered] = useState(false);
const currentFlowId = useFlowsManagerStore((state) => state.currentFlowId);
function handleMouseEnter(): void {
@ -65,7 +46,7 @@ export default function FileCard({
/>
<DownloadButton
isHovered={isHovered}
handleDownload={handleDownload}
handleDownload={() => handleDownload({ fileName, content })}
/>
</div>
</div>
@ -77,7 +58,7 @@ export default function FileCard({
className={`relative ${false ? "h-20 w-20" : "h-20 w-80"} cursor-pointer rounded-lg border border-ring bg-muted shadow transition duration-300 hover:drop-shadow-lg ${
isHovered ? "shadow-md" : ""
}`}
onClick={handleDownload}
onClick={() => handleDownload({ fileName, content })}
onMouseEnter={handleMouseEnter}
onMouseLeave={handleMouseLeave}
>
@ -88,7 +69,10 @@ export default function FileCard({
<span>File</span>
</div>
</div>
<DownloadButton isHovered={isHovered} handleDownload={handleDownload} />
<DownloadButton
isHovered={isHovered}
handleDownload={() => handleDownload({ fileName, content })}
/>
</div>
);
}

31
src/frontend/src/modals/IOModal/components/chatView/fileComponent/utils/handle-download.tsx Normal file
View file

@ -0,0 +1,31 @@
import {
BACKEND_URL,
BASE_URL_API,
} from "../../../../../../constants/constants";
export default async function handleDownload({
fileName,
content,
}: {
fileName: string;
content: string;
}): Promise<void> {
try {
const response = await fetch(
`${BACKEND_URL.slice(0, BACKEND_URL.length - 1)}${BASE_URL_API}files/download/${content}`,
);
const blob = await response.blob();
const url = URL.createObjectURL(blob);
const link = document.createElement("a");
link.href = url;
link.setAttribute("download", fileName); // Set the filename
document.body.appendChild(link);
link.click();
document.body.removeChild(link);
URL.revokeObjectURL(url); // Clean up the URL object
} catch (error) {
console.error("Failed to download file:", error);
}
}