karma/crab-webapp
1
0
Fork 0
mirror of https://github.com/karma-riuk/crab-webapp.git synced 2025-07-05 14:18:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

formatted file

Browse Source
This commit is contained in:
Karma Riuk
2025-05-07 18:21:13 +02:00
parent 522bcc7ebc
commit 4200b5d857
1 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/routes/datasets.js
View File

@ -1,34 +1,35 @@
import { Router } from 'express'; import { Router } from "express";
import { join } from 'path'; import { join } from "path";
import { getProjectPath } from '../utils/paths.js'; import { getProjectPath } from "../utils/paths.js";
const router = Router(); const router = Router();
// Environment variables for paths (all relative to project root) // Environment variables for paths (all relative to project root)
const DATA_DIR = process.env.DATA_DIR ? getProjectPath(process.env.DATA_DIR) : getProjectPath('data'); const DATA_DIR = process.env.DATA_DIR
? getProjectPath(process.env.DATA_DIR)
: getProjectPath("data");
const DATASETS = [ const DATASETS = ["comment_generation", "code_refinement"];
"comment_generation",
"code_refinement"
];
router.get('/download/:dataset', async (req, res) => { router.get("/download/:dataset", async (req, res) => {
const { dataset } = req.params; const { dataset } = req.params;
const withContext = req.query.withContext ? JSON.parse(req.query.withContext) : false; const withContext = req.query.withContext
? JSON.parse(req.query.withContext)
: false;
if (!DATASETS.includes(dataset)) { if (!DATASETS.includes(dataset)) {
return res.status(400).json({ error: 'Invalid dataset name' }); return res.status(400).json({ error: "Invalid dataset name" });
} }
const fileName = `${dataset}_${withContext ? 'with_context' : 'no_context'}.zip`; const fileName = `${dataset}_${withContext ? "with_context" : "no_context"}.zip`;
const filePath = join(DATA_DIR, fileName); const filePath = join(DATA_DIR, fileName);
try { try {
res.download(filePath); res.download(filePath);
} catch (error) { } catch (error) {
console.error('Error serving file:', error); console.error("Error serving file:", error);
res.status(500).json({ error: 'Error serving file' }); res.status(500).json({ error: "Error serving file" });
} }
}); });
export default router; export default router;