filesafe/controllers/uploadController.js
2017-01-14 03:01:23 -03:00

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(() => {
return res.json({
'filename': req.file.filename
})
})
})
}
module.exports = uploadsController