WIP
This commit is contained in:
Pitu
parent
a7201c4b96
commit
0258c290ff
|
|
@ -25,81 +25,90 @@ const upload = multer({
|
||||||
|
|
||||||
uploadsController.upload = function(req, res, next){
|
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(config.private === true)
|
||||||
if(req.headers.auth !== config.clientToken)
|
if(token === undefined) return res.status(401).json({ success: false, description: 'No token provided' })
|
||||||
return res.status(401).json({ success: false, description: 'not-authorized'})
|
|
||||||
|
|
||||||
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)
|
// Check if user is trying to upload to an album
|
||||||
if(req.headers.adminauth !== config.adminToken)
|
let album = undefined
|
||||||
return res.status(401).json({ success: false, description: 'not-authorized'})
|
if(userid !== undefined){
|
||||||
|
album = req.headers.albumid
|
||||||
upload(req, res, function (err) {
|
if(album === undefined)
|
||||||
if (err) {
|
album = req.params.albumid
|
||||||
console.error(err)
|
|
||||||
return res.json({
|
|
||||||
success: false,
|
|
||||||
description: err
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if(req.files.length === 0) return res.json({ success: false, description: 'no-files' })
|
upload(req, res, function (err) {
|
||||||
|
if (err) {
|
||||||
|
console.error(err)
|
||||||
|
return res.json({
|
||||||
|
success: false,
|
||||||
|
description: err
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
let files = []
|
if(req.files.length === 0) return res.json({ success: false, description: 'no-files' })
|
||||||
let existingFiles = []
|
|
||||||
let iteration = 1
|
|
||||||
|
|
||||||
req.files.forEach(function(file) {
|
let files = []
|
||||||
|
let existingFiles = []
|
||||||
|
let iteration = 1
|
||||||
|
|
||||||
// Check if the file exists by checking hash and size
|
req.files.forEach(function(file) {
|
||||||
let hash = crypto.createHash('md5')
|
|
||||||
let stream = fs.createReadStream('./' + config.uploads.folder + '/' + file.filename)
|
|
||||||
|
|
||||||
stream.on('data', function (data) {
|
// Check if the file exists by checking hash and size
|
||||||
hash.update(data, 'utf8')
|
let hash = crypto.createHash('md5')
|
||||||
})
|
let stream = fs.createReadStream('./' + config.uploads.folder + '/' + file.filename)
|
||||||
|
|
||||||
stream.on('end', function () {
|
stream.on('data', function (data) {
|
||||||
let fileHash = hash.digest('hex') // 34f7a3113803f8ed3b8fd7ce5656ebec
|
hash.update(data, 'utf8')
|
||||||
|
|
||||||
db.table('files').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,
|
|
||||||
timestamp: Math.floor(Date.now() / 1000)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
if(iteration === req.files.length)
|
|
||||||
return uploadsController.processFilesForDisplay(req, res, files, existingFiles)
|
|
||||||
iteration++
|
|
||||||
})
|
})
|
||||||
|
|
||||||
|
stream.on('end', function () {
|
||||||
|
let fileHash = hash.digest('hex') // 34f7a3113803f8ed3b8fd7ce5656ebec
|
||||||
|
|
||||||
|
db.table('files').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'}) })
|
||||||
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
})
|
})
|
||||||
|
}).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')
|
||||||
for(let domain of config.domains)
|
for(let domain of config.domains)
|
||||||
if(domain.host === req.get('host'))
|
if(domain.host === req.get('host'))
|
||||||
|
|
@ -139,26 +148,36 @@ uploadsController.processFilesForDisplay = function(req, res, files, existingFil
|
||||||
|
|
||||||
uploadsController.delete = function(req, res){
|
uploadsController.delete = function(req, res){
|
||||||
|
|
||||||
if(req.headers.auth !== config.adminToken)
|
let token = req.headers.token
|
||||||
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 id = req.body.id
|
let id = req.body.id
|
||||||
if(id === undefined || id === '')
|
if(id === undefined || id === '')
|
||||||
return res.json({ success: false, description: 'No file specified' })
|
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'})
|
||||||
|
|
||||||
uploadsController.deleteFile(file[0].name).then(() => {
|
db.table('files')
|
||||||
db.table('files').where('id', id).del().then(() =>{
|
.where('id', id)
|
||||||
return res.json({ success: true })
|
.where(function(){
|
||||||
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
if(user.username !== 'root')
|
||||||
}).catch((e) => {
|
this.where('userid', user.id)
|
||||||
console.log(e.toString())
|
|
||||||
db.table('files').where('id', id).del().then(() =>{
|
|
||||||
return res.json({ success: true })
|
|
||||||
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
|
||||||
})
|
})
|
||||||
|
.then((file) => {
|
||||||
|
|
||||||
|
uploadsController.deleteFile(file[0].name).then(() => {
|
||||||
|
db.table('files').where('id', id).del().then(() =>{
|
||||||
|
return res.json({ success: true })
|
||||||
|
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
||||||
|
}).catch((e) => {
|
||||||
|
console.log(e.toString())
|
||||||
|
db.table('files').where('id', id).del().then(() =>{
|
||||||
|
return res.json({ success: true })
|
||||||
|
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
||||||
|
})
|
||||||
|
|
||||||
|
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
||||||
}).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,83 +198,92 @@ uploadsController.deleteFile = function(file){
|
||||||
|
|
||||||
uploadsController.list = function(req, res){
|
uploadsController.list = function(req, res){
|
||||||
|
|
||||||
if(req.headers.auth !== config.adminToken)
|
let token = req.headers.token
|
||||||
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 offset = req.params.page
|
db.table('users').where('token', token).then((user) => {
|
||||||
if(offset === undefined) offset = 0
|
if(user.length === 0) return res.status(401).json({ success: false, description: 'Invalid token'})
|
||||||
|
|
||||||
db.table('files')
|
let offset = req.params.page
|
||||||
.where(function(){
|
if(offset === undefined) offset = 0
|
||||||
if(req.params.id === undefined)
|
|
||||||
this.where('id', '<>', '')
|
|
||||||
else
|
|
||||||
this.where('albumid', req.params.id)
|
|
||||||
})
|
|
||||||
.orderBy('id', 'DESC')
|
|
||||||
.limit(25)
|
|
||||||
.offset(25 * offset)
|
|
||||||
.then((files) => {
|
|
||||||
db.table('albums').then((albums) => {
|
|
||||||
|
|
||||||
let basedomain = req.get('host')
|
db.table('files')
|
||||||
for(let domain of config.domains)
|
.where(function(){
|
||||||
if(domain.host === req.get('host'))
|
if(req.params.id === undefined)
|
||||||
if(domain.hasOwnProperty('resolve'))
|
this.where('id', '<>', '')
|
||||||
basedomain = domain.resolve
|
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)
|
||||||
|
.then((files) => {
|
||||||
|
db.table('albums').then((albums) => {
|
||||||
|
|
||||||
for(let file of files){
|
let basedomain = req.get('host')
|
||||||
file.file = basedomain + '/' + file.name
|
for(let domain of config.domains)
|
||||||
file.date = new Date(file.timestamp * 1000)
|
if(domain.host === req.get('host'))
|
||||||
file.date = file.date.getFullYear() + '-' + (file.date.getMonth() + 1) + '-' + file.date.getDate() + ' ' + (file.date.getHours() < 10 ? '0' : '') + file.date.getHours() + ':' + (file.date.getMinutes() < 10 ? '0' : '') + file.date.getMinutes() + ':' + (file.date.getSeconds() < 10 ? '0' : '') + file.date.getSeconds()
|
if(domain.hasOwnProperty('resolve'))
|
||||||
|
basedomain = domain.resolve
|
||||||
|
|
||||||
file.album = ''
|
for(let file of files){
|
||||||
|
file.file = basedomain + '/' + file.name
|
||||||
|
file.date = new Date(file.timestamp * 1000)
|
||||||
|
file.date = file.date.getFullYear() + '-' + (file.date.getMonth() + 1) + '-' + file.date.getDate() + ' ' + (file.date.getHours() < 10 ? '0' : '') + file.date.getHours() + ':' + (file.date.getMinutes() < 10 ? '0' : '') + file.date.getMinutes() + ':' + (file.date.getSeconds() < 10 ? '0' : '') + file.date.getSeconds()
|
||||||
|
|
||||||
if(file.albumid !== undefined)
|
file.album = ''
|
||||||
for(let album of albums)
|
|
||||||
if(file.albumid === album.id)
|
|
||||||
file.album = album.name
|
|
||||||
|
|
||||||
if(config.uploads.generateThumbnails === true){
|
if(file.albumid !== undefined)
|
||||||
|
for(let album of albums)
|
||||||
|
if(file.albumid === album.id)
|
||||||
|
file.album = album.name
|
||||||
|
|
||||||
let extensions = ['.jpg', '.jpeg', '.bmp', '.gif', '.png']
|
if(config.uploads.generateThumbnails === true){
|
||||||
for(let ext of extensions){
|
|
||||||
if(path.extname(file.name) === ext){
|
|
||||||
|
|
||||||
file.thumb = basedomain + '/thumbs/' + file.name.slice(0, -4) + '.png'
|
let extensions = ['.jpg', '.jpeg', '.bmp', '.gif', '.png']
|
||||||
|
for(let ext of extensions){
|
||||||
|
if(path.extname(file.name) === ext){
|
||||||
|
|
||||||
let thumbname = path.join(__dirname, '..', 'uploads', 'thumbs') + '/' + file.name.slice(0, -4) + '.png'
|
file.thumb = basedomain + '/thumbs/' + file.name.slice(0, -4) + '.png'
|
||||||
fs.access(thumbname, function(err) {
|
|
||||||
if (err && err.code === 'ENOENT') {
|
|
||||||
// File doesnt exist
|
|
||||||
|
|
||||||
let size = {
|
let thumbname = path.join(__dirname, '..', 'uploads', 'thumbs') + '/' + file.name.slice(0, -4) + '.png'
|
||||||
width: 200,
|
fs.access(thumbname, function(err) {
|
||||||
height: 200
|
if (err && err.code === 'ENOENT') {
|
||||||
|
// File doesnt exist
|
||||||
|
|
||||||
|
let size = {
|
||||||
|
width: 200,
|
||||||
|
height: 200
|
||||||
|
}
|
||||||
|
|
||||||
|
gm('./' + config.uploads.folder + '/' + file.name)
|
||||||
|
.resize(size.width, size.height + '>')
|
||||||
|
.gravity('Center')
|
||||||
|
.extent(size.width, size.height)
|
||||||
|
.background('transparent')
|
||||||
|
.write(thumbname, function (error) {
|
||||||
|
if (error) console.log('Error - ', error)
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
})
|
||||||
gm('./' + config.uploads.folder + '/' + file.name)
|
}
|
||||||
.resize(size.width, size.height + '>')
|
|
||||||
.gravity('Center')
|
|
||||||
.extent(size.width, size.height)
|
|
||||||
.background('transparent')
|
|
||||||
.write(thumbname, function (error) {
|
|
||||||
if (error) console.log('Error - ', error)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
return res.json({
|
return res.json({
|
||||||
success: true,
|
success: true,
|
||||||
files
|
files
|
||||||
})
|
})
|
||||||
|
|
||||||
|
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
||||||
}).catch(function(error) { console.log(error); res.json({success: false, description: 'error'}) })
|
}).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
|
module.exports = uploadsController
|
||||||
Loading…
Reference in New Issue