mirror of
https://github.com/BobbyWibowo/lolisafe.git
synced 2024-12-13 16:06:21 +00:00
49 lines
1.1 KiB
JavaScript
49 lines
1.1 KiB
JavaScript
|
const path = require('path')
|
||
|
const config = require('../config.js')
|
||
|
const multer = require('multer')
|
||
|
const randomstring = require('randomstring')
|
||
|
const db = require('knex')(config.database)
|
||
|
|
||
|
let uploadsController = {}
|
||
|
|
||
|
const storage = multer.diskStorage({
|
||
|
destination: function (req, file, cb) {
|
||
|
cb(null, './' + config.uploads.folder + '/')
|
||
|
},
|
||
|
filename: function (req, file, cb) {
|
||
|
cb(null, randomstring.generate(config.fileLength) + path.extname(file.originalname))
|
||
|
}
|
||
|
})
|
||
|
|
||
|
const upload = multer({
|
||
|
storage: storage,
|
||
|
limits: { fileSize: config.uploads.maxsize }
|
||
|
}).single('file')
|
||
|
|
||
|
uploadsController.upload = function(req, res, next){
|
||
|
|
||
|
let gallery = req.headers.gallery
|
||
|
|
||
|
if(!config.privacy.public)
|
||
|
if(!config.privacy.IPs.includes(req.ip)) return res.status(401).send('Not Authorized!')
|
||
|
|
||
|
upload(req, res, function (err) {
|
||
|
if (err) {
|
||
|
console.error(err)
|
||
|
return res.json({ error: err })
|
||
|
}
|
||
|
|
||
|
db.table('files').insert({
|
||
|
file: req.file.filename,
|
||
|
galleryid: gallery
|
||
|
}).then(() => {
|
||
|
res.json({
|
||
|
'filename': req.file.filename
|
||
|
})
|
||
|
})
|
||
|
|
||
|
})
|
||
|
|
||
|
}
|
||
|
|
||
|
module.exports = uploadsController
|