68 lines
2.2 KiB
JavaScript
68 lines
2.2 KiB
JavaScript
import express from 'express';
|
|
const router = express.Router();
|
|
import { DataTypes } from 'sequelize';
|
|
import Papa from 'papaparse';
|
|
import defineCase from '../../models/cases.js';
|
|
import authMiddleware from '../../middleware/auth.js';
|
|
import visibilityMiddleware from '../../middleware/verifyVisible.js';
|
|
import { testRunStatus, priorities, testTypes, automationStatus, templates } from '../../config/enums.js';
|
|
|
|
export default function (sequelize) {
|
|
const Case = defineCase(sequelize, DataTypes);
|
|
const { verifySignedIn } = authMiddleware(sequelize);
|
|
const { verifyProjectVisibleFromFolderId } = visibilityMiddleware(sequelize);
|
|
|
|
router.get('/download', verifySignedIn, verifyProjectVisibleFromFolderId, async (req, res) => {
|
|
const { folderId, type } = req.query;
|
|
|
|
if (!folderId) {
|
|
return res.status(400).json({ error: 'folderId is required' });
|
|
}
|
|
|
|
if (!type) {
|
|
return res.status(400).json({ error: 'download type is required' });
|
|
}
|
|
|
|
try {
|
|
const cases = await Case.findAll({
|
|
where: { folderId },
|
|
raw: true,
|
|
});
|
|
|
|
if (cases.length === 0) {
|
|
return res.status(404).send('No cases found');
|
|
}
|
|
|
|
if (type === 'json') {
|
|
return res.json(cases);
|
|
} else if (type === 'csv') {
|
|
// Convert numeric values to human-readable labels
|
|
const casesWithLabels = cases.map((c) => ({
|
|
...c,
|
|
state: testRunStatus[c.state] || c.state,
|
|
priority: priorities[c.priority] || c.priority,
|
|
type: testTypes[c.type] || c.type,
|
|
automationStatus: automationStatus[c.automationStatus] || c.automationStatus,
|
|
template: templates[c.template] || c.template,
|
|
}));
|
|
|
|
const csv = Papa.unparse(casesWithLabels, {
|
|
quotes: true,
|
|
skipEmptyLines: true,
|
|
});
|
|
|
|
res.setHeader('Content-Type', 'text/csv; charset=utf-8');
|
|
res.setHeader('Content-Disposition', `attachment; filename=cases_folder_${folderId}.csv`);
|
|
return res.send(csv);
|
|
}
|
|
|
|
return res.status(400).json({ error: 'Unsupported type. Use ?type=json or ?type=csv' });
|
|
} catch (error) {
|
|
console.error(error);
|
|
res.status(500).send('Internal Server Error');
|
|
}
|
|
});
|
|
|
|
return router;
|
|
}
|