mirror of
https://github.com/BobbyWibowo/lolisafe.git
synced 2024-12-14 16:36:21 +00:00
Fixed last night's fuckup
This commit is contained in:
parent
80117235f9
commit
ad4eeb5eaa
@ -62,82 +62,93 @@ uploadsController.upload = function(req, res, next) {
|
|||||||
album = req.params.albumid
|
album = req.params.albumid
|
||||||
}
|
}
|
||||||
|
|
||||||
db.table('albums').where({ id: album, userid: userid }).then((albums) => {
|
/*
|
||||||
if (albums.length === 0) {
|
A rewrite is due so might as well do awful things here and fix them later :bloblul:
|
||||||
return res.json({
|
*/
|
||||||
success: false,
|
|
||||||
description: 'Album doesn\'t exist or it doesn\'t belong to the user'
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
upload(req, res, function (err) {
|
if (album !== undefined && userid !== undefined) {
|
||||||
if (err) {
|
// If both values are present, check if the album owner is the user uploading
|
||||||
console.error(err)
|
db.table('albums').where({ id: album, userid: userid }).then((albums) => {
|
||||||
|
if (albums.length === 0) {
|
||||||
return res.json({
|
return res.json({
|
||||||
success: false,
|
success: false,
|
||||||
description: err
|
description: 'Album doesn\'t exist or it doesn\'t belong to the user'
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
uploadsController.actuallyUpload(req, res, userid, album);
|
||||||
if (req.files.length === 0) return res.json({ success: false, description: 'no-files' })
|
|
||||||
|
|
||||||
let files = []
|
|
||||||
let existingFiles = []
|
|
||||||
let iteration = 1
|
|
||||||
|
|
||||||
req.files.forEach(function(file) {
|
|
||||||
|
|
||||||
// Check if the file exists by checking hash and size
|
|
||||||
let hash = crypto.createHash('md5')
|
|
||||||
let stream = fs.createReadStream(path.join(__dirname, '..', config.uploads.folder, file.filename))
|
|
||||||
|
|
||||||
stream.on('data', function (data) {
|
|
||||||
hash.update(data, 'utf8')
|
|
||||||
})
|
|
||||||
|
|
||||||
stream.on('end', function () {
|
|
||||||
let fileHash = hash.digest('hex')
|
|
||||||
|
|
||||||
db.table('files')
|
|
||||||
.where(function() {
|
|
||||||
if (userid === undefined)
|
|
||||||
this.whereNull('userid')
|
|
||||||
else
|
|
||||||
this.where('userid', userid)
|
|
||||||
})
|
|
||||||
.where({
|
|
||||||
hash: fileHash,
|
|
||||||
size: file.size
|
|
||||||
}).then((dbfile) => {
|
|
||||||
|
|
||||||
if (dbfile.length !== 0) {
|
|
||||||
uploadsController.deleteFile(file.filename).then(() => {}).catch((e) => console.error(e))
|
|
||||||
existingFiles.push(dbfile[0])
|
|
||||||
} else {
|
|
||||||
files.push({
|
|
||||||
name: file.filename,
|
|
||||||
original: file.originalname,
|
|
||||||
type: file.mimetype,
|
|
||||||
size: file.size,
|
|
||||||
hash: fileHash,
|
|
||||||
ip: req.ip,
|
|
||||||
albumid: album,
|
|
||||||
userid: userid,
|
|
||||||
timestamp: Math.floor(Date.now() / 1000)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
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' }) })
|
|
||||||
})
|
|
||||||
})
|
|
||||||
})
|
})
|
||||||
})
|
} else {
|
||||||
|
uploadsController.actuallyUpload(req, res, userid, album);
|
||||||
|
}
|
||||||
}).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.actuallyUpload = function(req, res, userid, album) {
|
||||||
|
upload(req, res, function (err) {
|
||||||
|
if (err) {
|
||||||
|
console.error(err)
|
||||||
|
return res.json({
|
||||||
|
success: false,
|
||||||
|
description: err
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
if (req.files.length === 0) return res.json({ success: false, description: 'no-files' })
|
||||||
|
|
||||||
|
let files = []
|
||||||
|
let existingFiles = []
|
||||||
|
let iteration = 1
|
||||||
|
|
||||||
|
req.files.forEach(function(file) {
|
||||||
|
|
||||||
|
// Check if the file exists by checking hash and size
|
||||||
|
let hash = crypto.createHash('md5')
|
||||||
|
let stream = fs.createReadStream(path.join(__dirname, '..', config.uploads.folder, file.filename))
|
||||||
|
|
||||||
|
stream.on('data', function (data) {
|
||||||
|
hash.update(data, 'utf8')
|
||||||
|
})
|
||||||
|
|
||||||
|
stream.on('end', function () {
|
||||||
|
let fileHash = hash.digest('hex')
|
||||||
|
|
||||||
|
db.table('files')
|
||||||
|
.where(function() {
|
||||||
|
if (userid === undefined)
|
||||||
|
this.whereNull('userid')
|
||||||
|
else
|
||||||
|
this.where('userid', userid)
|
||||||
|
})
|
||||||
|
.where({
|
||||||
|
hash: fileHash,
|
||||||
|
size: file.size
|
||||||
|
}).then((dbfile) => {
|
||||||
|
|
||||||
|
if (dbfile.length !== 0) {
|
||||||
|
uploadsController.deleteFile(file.filename).then(() => {}).catch((e) => console.error(e))
|
||||||
|
existingFiles.push(dbfile[0])
|
||||||
|
} else {
|
||||||
|
files.push({
|
||||||
|
name: file.filename,
|
||||||
|
original: file.originalname,
|
||||||
|
type: file.mimetype,
|
||||||
|
size: file.size,
|
||||||
|
hash: fileHash,
|
||||||
|
ip: req.ip,
|
||||||
|
albumid: album,
|
||||||
|
userid: userid,
|
||||||
|
timestamp: Math.floor(Date.now() / 1000)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
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' }) })
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
||||||
uploadsController.processFilesForDisplay = function(req, res, files, existingFiles) {
|
uploadsController.processFilesForDisplay = function(req, res, files, existingFiles) {
|
||||||
|
|
||||||
let basedomain = req.get('host')
|
let basedomain = req.get('host')
|
||||||
|
Loading…
Reference in New Issue
Block a user