52 lines
1.5 KiB
JavaScript
52 lines
1.5 KiB
JavaScript
|
const s3 = require('../middleware/Storage')
|
||
|
|
const express = require('express');
|
||
|
|
const router = express.Router();
|
||
|
|
const path = require('path')
|
||
|
|
const multer = require('multer')
|
||
|
|
const dotenv = require('dotenv');
|
||
|
|
const { uuid } = require('uuidv4');
|
||
|
|
const shortid = require('shortid');
|
||
|
|
|
||
|
|
const storage = multer.memoryStorage();
|
||
|
|
const upload = multer({ storage: storage });
|
||
|
|
dotenv.config();
|
||
|
|
|
||
|
|
const authenticateAPIKey = (req, res, next) => {
|
||
|
|
const userKey = req.headers['x-api-key'];
|
||
|
|
const uploadSec = process.env.UPLOAD_SEC;
|
||
|
|
|
||
|
|
if (userKey && userKey === uploadSec) {
|
||
|
|
return next();
|
||
|
|
} else {
|
||
|
|
return res.status(403).json({ error: 'Forbidden' });
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
router.post('/upload', authenticateAPIKey, upload.single('file'), async (req, res) => {
|
||
|
|
try {
|
||
|
|
const file = req.file;
|
||
|
|
const uuidGen = uuid()
|
||
|
|
const fileID = shortid.generate()
|
||
|
|
|
||
|
|
const fileName = `${fileID}${path.extname(file.originalname)}`;
|
||
|
|
const key = `attachments/${uuidGen}/` + fileName;
|
||
|
|
|
||
|
|
const fileUrl = `${process.env.R2_DOMAIN}/${key}`;
|
||
|
|
|
||
|
|
const params = {
|
||
|
|
Bucket: process.env.R2_BUCKET,
|
||
|
|
Key: key,
|
||
|
|
Body: file.buffer,
|
||
|
|
ContentType: file.mimetype,
|
||
|
|
ACL: 'public-read',
|
||
|
|
};
|
||
|
|
|
||
|
|
await s3.upload(params).promise();
|
||
|
|
res.json({ url: `${fileUrl}` });
|
||
|
|
} catch (error) {
|
||
|
|
console.error(error);
|
||
|
|
res.status(500).json({ error: 'Upload error.' });
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
module.exports = router;
|