Implemented attachment file download function

This commit is contained in:
Takeshi Kimata
2024-03-20 16:19:01 +09:00
parent 8a5ae80351
commit fdda581fb0
5 changed files with 82 additions and 2 deletions

View File

@@ -64,8 +64,10 @@ app.use("/steps", stepsDeleteRoute);
// "/attachments"
const attachmentsNewRoute = require("./routes/attachments/new")(sequelize);
const attachmentsDeleteRoute = require("./routes/attachments/delete")(sequelize);
const attachmentsDownloadRoute = require("./routes/attachments/download")(sequelize);
app.use("/attachments", attachmentsNewRoute);
app.use("/attachments", attachmentsDeleteRoute);
app.use("/attachments", attachmentsDownloadRoute);
// "/runs"
const runsIndexRoute = require("./routes/runs/index")(sequelize);

View File

@@ -0,0 +1,34 @@
const express = require("express");
const router = express.Router();
const path = require("path");
const fs = require("fs");
const defineAttachment = require("../../models/attachments");
const { DataTypes } = require("sequelize");
module.exports = function (sequelize) {
const Attachment = defineAttachment(sequelize, DataTypes);
router.get("/download/:attachmentId", async (req, res) => {
const attachmentId = req.params.attachmentId;
try {
const attachment = await Attachment.findByPk(attachmentId);
if (!attachment) {
return res.status(404).send("Attachment not found");
}
const filename = attachment.path.split("/").pop();
const filePath = path.join(__dirname, `../../public/uploads/${filename}`);
if (!fs.existsSync(filePath)) {
return res.status(404).json({ error: "File not found" });
}
res.download(filePath);
} catch (error) {
console.error(error);
res.status(500).send("Internal Server Error");
}
});
return router;
};

View File

@@ -4,11 +4,16 @@ import { Trash, ArrowDownToLine } from "lucide-react";
type Props = {
attachments: AttachmentType[];
onAttachmentDownload: (
attachmentId: number,
downloadFileName: string
) => void;
onAttachmentDelete: (attachmentId: number) => void;
};
export default function CaseAttachmentsEditor({
attachments = [],
onAttachmentDownload,
onAttachmentDelete,
}: Props) {
let images = [];
@@ -70,7 +75,7 @@ export default function CaseAttachmentsEditor({
isIconOnly
size="sm"
className="bg-transparent rounded-full"
onPress={() => console.log("download")}
onPress={() => onAttachmentDownload(file.id, file.title)}
>
<ArrowDownToLine size={16} />
</Button>

View File

@@ -22,6 +22,7 @@ import {
fetchDeleteStep,
updateCase,
fetchCreateAttachments,
fetchDownloadAttachment,
fetchDeleteAttachment,
} from "./caseControl";
@@ -370,6 +371,9 @@ export default function CaseEditor({
<h6>Attachments</h6>
<CaseAttachmentsEditor
attachments={testCase.Attachments}
onAttachmentDownload={(attachmentId: number, downloadFileName: string) =>
fetchDownloadAttachment(attachmentId, downloadFileName)
}
onAttachmentDelete={onAttachmentDelete}
/>
<div

View File

@@ -129,6 +129,40 @@ async function fetchCreateAttachments(caseId: number, files: File[]) {
}
}
/**
* download attachment
*/
async function fetchDownloadAttachment(attachmentId: number, downloadFileName: string) {
const fetchOptions = {
method: "GET",
headers: {
"Content-Type": "application/json",
},
};
const url = `${apiServer}/attachments/download/${attachmentId}`;
try {
const response = await fetch(url, fetchOptions);
if (!response.ok) {
throw new Error(`HTTP error! Status: ${response.status}`);
}
const blob = await response.blob();
const downloadUrl = window.URL.createObjectURL(blob);
const link = document.createElement("a");
link.href = downloadUrl;
link.download = downloadFileName;
document.body.appendChild(link);
link.click();
document.body.removeChild(link);
} catch (error) {
console.error("Error downloading file:", error);
throw error;
}
}
/**
* delete attachment
*/
@@ -148,7 +182,7 @@ async function fetchDeleteAttachment(attachmentId: number) {
throw new Error(`HTTP error! Status: ${response.status}`);
}
} catch (error) {
console.error("Error deleting project:", error);
console.error("Error deleting file:", error);
throw error;
}
}
@@ -159,5 +193,6 @@ export {
fetchDeleteStep,
updateCase,
fetchCreateAttachments,
fetchDownloadAttachment,
fetchDeleteAttachment,
};