Files
pp-tcms/backend/routes/attachments/download.js

34 lines
989 B
JavaScript

const fs = require('fs');
const path = require('path');
const express = require('express');
const router = express.Router();
const { DataTypes } = require('sequelize');
const defineAttachment = require('../../models/attachments');
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 filePath = path.join(__dirname, `../../public/uploads/${attachment.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;
};