mirror of
https://github.com/BobbyWibowo/lolisafe.git
synced 2025-01-31 07:11:33 +00:00
WIP
This commit is contained in:
parent
a7201c4b96
commit
0258c290ff
@ -25,15 +25,26 @@ const upload = multer({
|
||||
|
||||
uploadsController.upload = function(req, res, next){
|
||||
|
||||
// Get the token
|
||||
let token = req.headers.token
|
||||
|
||||
// If we're running in private and there's no token, error
|
||||
if(config.private === true)
|
||||
if(req.headers.auth !== config.clientToken)
|
||||
return res.status(401).json({ success: false, description: 'not-authorized'})
|
||||
if(token === undefined) return res.status(401).json({ success: false, description: 'No token provided' })
|
||||
|
||||
let album = req.params.albumid
|
||||
// Let's see if it's a valid token
|
||||
db.table('users').where('token', token).then((user) => {
|
||||
let userid
|
||||
if(user.length > 0)
|
||||
userid = user.id
|
||||
|
||||
if(album !== undefined)
|
||||
if(req.headers.adminauth !== config.adminToken)
|
||||
return res.status(401).json({ success: false, description: 'not-authorized'})
|
||||
// Check if user is trying to upload to an album
|
||||
let album = undefined
|
||||
if(userid !== undefined){
|
||||
album = req.headers.albumid
|
||||
if(album === undefined)
|
||||
album = req.params.albumid
|
||||
}
|
||||
|
||||
upload(req, res, function (err) {
|
||||
if (err) {
|
||||
@ -80,6 +91,7 @@ uploadsController.upload = function(req, res, next){
|
||||
hash: fileHash,
|
||||
ip: req.ip,
|
||||
albumid: album,
|
||||
userid: userid,
|
||||
timestamp: Math.floor(Date.now() / 1000)
|
||||
})
|
||||
}
|
||||
@ -87,19 +99,16 @@ uploadsController.upload = function(req, res, next){
|
||||
if(iteration === req.files.length)
|
||||
return uploadsController.processFilesForDisplay(req, res, files, existingFiles)
|
||||
iteration++
|
||||
})
|
||||
|
||||
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
||||
})
|
||||
|
||||
})
|
||||
|
||||
})
|
||||
|
||||
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
||||
}
|
||||
|
||||
uploadsController.processFilesForDisplay = function(req, res, files, existingFiles){
|
||||
|
||||
|
||||
let basedomain = req.get('host')
|
||||
for(let domain of config.domains)
|
||||
if(domain.host === req.get('host'))
|
||||
@ -139,14 +148,23 @@ uploadsController.processFilesForDisplay = function(req, res, files, existingFil
|
||||
|
||||
uploadsController.delete = function(req, res){
|
||||
|
||||
if(req.headers.auth !== config.adminToken)
|
||||
return res.status(401).json({ success: false, description: 'not-authorized'})
|
||||
let token = req.headers.token
|
||||
if(token === undefined) return res.status(401).json({ success: false, description: 'No token provided' })
|
||||
|
||||
let id = req.body.id
|
||||
if(id === undefined || id === '')
|
||||
return res.json({ success: false, description: 'No file specified' })
|
||||
|
||||
db.table('files').where('id', id).then((file) => {
|
||||
db.table('users').where('token', token).then((user) => {
|
||||
if(user.length === 0) return res.status(401).json({ success: false, description: 'Invalid token'})
|
||||
|
||||
db.table('files')
|
||||
.where('id', id)
|
||||
.where(function(){
|
||||
if(user.username !== 'root')
|
||||
this.where('userid', user.id)
|
||||
})
|
||||
.then((file) => {
|
||||
|
||||
uploadsController.deleteFile(file[0].name).then(() => {
|
||||
db.table('files').where('id', id).del().then(() =>{
|
||||
@ -160,6 +178,7 @@ uploadsController.delete = function(req, res){
|
||||
})
|
||||
|
||||
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
||||
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
||||
|
||||
}
|
||||
|
||||
@ -179,8 +198,11 @@ uploadsController.deleteFile = function(file){
|
||||
|
||||
uploadsController.list = function(req, res){
|
||||
|
||||
if(req.headers.auth !== config.adminToken)
|
||||
return res.status(401).json({ success: false, description: 'not-authorized'})
|
||||
let token = req.headers.token
|
||||
if(token === undefined) return res.status(401).json({ success: false, description: 'No token provided' })
|
||||
|
||||
db.table('users').where('token', token).then((user) => {
|
||||
if(user.length === 0) return res.status(401).json({ success: false, description: 'Invalid token'})
|
||||
|
||||
let offset = req.params.page
|
||||
if(offset === undefined) offset = 0
|
||||
@ -192,6 +214,10 @@ uploadsController.list = function(req, res){
|
||||
else
|
||||
this.where('albumid', req.params.id)
|
||||
})
|
||||
.where(function(){
|
||||
if(user.username !== 'root')
|
||||
this.where('userid', user.id)
|
||||
})
|
||||
.orderBy('id', 'DESC')
|
||||
.limit(25)
|
||||
.offset(25 * offset)
|
||||
@ -256,6 +282,8 @@ uploadsController.list = function(req, res){
|
||||
|
||||
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
||||
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
||||
|
||||
})
|
||||
}
|
||||
|
||||
module.exports = uploadsController
|
Loading…
Reference in New Issue
Block a user