130 lines
4.3 KiB
JavaScript
130 lines
4.3 KiB
JavaScript
const express = require('express');
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
const router = express.Router();
|
|
const fileUpload = require('express-fileupload');
|
|
const { loggers } = require('winston');
|
|
const ncp = require('ncp');
|
|
const util = require('util');
|
|
const configFile = fs.readFileSync(path.join(__dirname, '../../../data', 'setup.json'), 'utf-8')
|
|
const config = JSON.parse(configFile);
|
|
const bodyParser = require('body-parser');
|
|
const jwt = require('jsonwebtoken');
|
|
const authMiddleware = require('../../../Middlewares/authMiddleware');
|
|
const { getUserData, getSetupData } = require('../../../Middlewares/watcherMiddleware');
|
|
const { logger, logRequestInfo, ErrorLogger, authLogger } = require('../../../config/logs');
|
|
|
|
let setupData = getSetupData();
|
|
let userData = getUserData();
|
|
router.use(bodyParser.json());
|
|
|
|
function authenticateToken(req, res, next) {
|
|
let token = null;
|
|
const authHeader = req.headers['authorization'];
|
|
|
|
if (authHeader) {
|
|
token = authHeader.split(' ')[1];
|
|
} else if (req.query.token) {
|
|
token = req.query.token;
|
|
}
|
|
|
|
if (token == null) {
|
|
if (req.user) {
|
|
return next();
|
|
} else {
|
|
return res.status(401).json({ message: 'Unauthorized' });
|
|
}
|
|
}
|
|
|
|
fs.readFile(path.join(__dirname, '../../../data', 'user.json'), 'utf8', (err, data) => {
|
|
if (err) {
|
|
console.error('Error reading user.json:', err);
|
|
return res.status(401).json({ message: 'Unauthorized' });
|
|
}
|
|
|
|
const users = JSON.parse(data);
|
|
|
|
const user = users.find(u => u.token === token);
|
|
|
|
if (user) {
|
|
req.user = user;
|
|
next();
|
|
} else {
|
|
return res.status(401).json({ message: 'Unauthorized' });
|
|
}
|
|
});
|
|
}
|
|
|
|
router.get('/', (req, res) => {
|
|
res.status(400).json({ error: 'Bad Request. The request cannot be fulfilled due to bad syntax or missing parameters.' });
|
|
});
|
|
|
|
router.use(fileUpload({
|
|
limits: { fileSize: 15 * 1024 * 1024 * 1024 },
|
|
}));
|
|
|
|
router.post('/', authenticateToken, async (req, res) => {
|
|
try {
|
|
if (!req.files || Object.keys(req.files).length === 0) {
|
|
return res.status(400).send('5410 - Download error, please try again later.');
|
|
}
|
|
|
|
const file = req.files.file;
|
|
const userId = req.user.name;
|
|
const Id = req.user.id;
|
|
const uploadDir = path.join('cdn-files', userId);
|
|
const originalFileName = file.name;
|
|
const domain = config.domain || 'mydomain.com';
|
|
let expiryDate = req.body.expiryDate;
|
|
let password = req.body.password;
|
|
|
|
if (!fs.existsSync(uploadDir)) {
|
|
fs.mkdirSync(uploadDir, { recursive: true });
|
|
}
|
|
|
|
file.mv(path.join(uploadDir, originalFileName), async (err) => {
|
|
if (err) {
|
|
console.error(err);
|
|
return res.status(500).send({ message: 'Error downloading file.' });
|
|
}
|
|
|
|
const fileExtension = path.extname(originalFileName).toLowerCase();
|
|
|
|
const bcrypt = require('bcrypt');
|
|
const saltRounds = 10;
|
|
|
|
let hashedPassword = '';
|
|
if (password) {
|
|
hashedPassword = bcrypt.hashSync(password, saltRounds);
|
|
}
|
|
|
|
const fileInfo = {
|
|
fileName: originalFileName,
|
|
expiryDate: expiryDate || '',
|
|
password: hashedPassword,
|
|
Id: Id,
|
|
path: path.join(uploadDir, originalFileName)
|
|
};
|
|
|
|
if (expiryDate || password) {
|
|
let data = [];
|
|
if (fs.existsSync(path.join(__dirname, '../../../data', 'file_info.json'))) {
|
|
const existingData = await fs.promises.readFile(path.join(__dirname, '../../../data', 'file_info.json'), 'utf8');
|
|
data = JSON.parse(existingData);
|
|
if (!Array.isArray(data)) {
|
|
data = [];
|
|
}
|
|
}
|
|
data.push(fileInfo);
|
|
await fs.promises.writeFile(path.join(__dirname, '../../../data', 'file_info.json'), JSON.stringify(data, null, 2));
|
|
}
|
|
|
|
res.status(200).send({ message: 'Your file has been successfully uploaded.' });
|
|
});
|
|
} catch (error) {
|
|
console.error(error);
|
|
return res.status(500).send({ message: 'Error downloading file.' });
|
|
}
|
|
});
|
|
|
|
module.exports = router; |