mirror of
https://github.com/BobbyWibowo/lolisafe.git
synced 2024-12-13 16:06:21 +00:00
329 lines
10 KiB
JavaScript
329 lines
10 KiB
JavaScript
const config = require('../config.js')
|
|
const path = require('path')
|
|
const multer = require('multer')
|
|
const randomstring = require('randomstring')
|
|
const db = require('knex')(config.database)
|
|
const crypto = require('crypto')
|
|
const fs = require('fs')
|
|
const utils = require('./utilsController.js')
|
|
|
|
let uploadsController = {}
|
|
|
|
const storage = multer.diskStorage({
|
|
destination: function(req, file, cb) {
|
|
cb(null, path.join(__dirname, '..', config.uploads.folder))
|
|
},
|
|
filename: function(req, file, cb) {
|
|
cb(null, randomstring.generate(config.uploads.fileLength) + path.extname(file.originalname))
|
|
}
|
|
})
|
|
|
|
const upload = multer({
|
|
storage: storage,
|
|
limits: { fileSize: config.uploads.maxSize },
|
|
fileFilter: function(req, file, cb) {
|
|
if (config.blockedExtensions !== undefined) {
|
|
if (config.blockedExtensions.some(extension => path.extname(file.originalname).toLowerCase() === extension)) {
|
|
return cb('This file extension is not allowed');
|
|
}
|
|
return cb(null, true);
|
|
}
|
|
return cb(null, true);
|
|
}
|
|
}).array('files[]')
|
|
|
|
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 (token === undefined) return res.status(401).json({ success: false, description: 'No token provided' })
|
|
|
|
// If there is no token then just leave it blank so the query fails
|
|
if (token === undefined) token = ''
|
|
|
|
db.table('users').where('token', token).then((user) => {
|
|
|
|
if(user.length === 0)
|
|
if(config.private === true)
|
|
return res.status(401).json({ success: false, description: 'Invalid token provided' })
|
|
|
|
let userid
|
|
if(user.length > 0)
|
|
userid = user[0].id
|
|
|
|
// Check if user is trying to upload to an album
|
|
let album
|
|
if (userid !== undefined) {
|
|
album = req.headers.albumid
|
|
if (album === undefined)
|
|
album = req.params.albumid
|
|
}
|
|
|
|
/*
|
|
A rewrite is due so might as well do awful things here and fix them later :bloblul:
|
|
*/
|
|
|
|
if (album !== undefined && userid !== undefined) {
|
|
// If both values are present, check if the album owner is the user uploading
|
|
db.table('albums').where({ id: album, userid: userid }).then((albums) => {
|
|
if (albums.length === 0) {
|
|
return res.json({
|
|
success: false,
|
|
description: 'Album doesn\'t exist or it doesn\'t belong to the user'
|
|
})
|
|
}
|
|
uploadsController.actuallyUpload(req, res, userid, album);
|
|
})
|
|
} else {
|
|
uploadsController.actuallyUpload(req, res, userid, album);
|
|
}
|
|
}).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) {
|
|
|
|
let basedomain = req.get('host')
|
|
for (let domain of config.domains)
|
|
if (domain.host === req.get('host'))
|
|
if (domain.hasOwnProperty('resolve'))
|
|
basedomain = domain.resolve
|
|
|
|
if (files.length === 0) {
|
|
return res.json({
|
|
success: true,
|
|
files: existingFiles.map(file => {
|
|
return {
|
|
name: file.name,
|
|
size: file.size,
|
|
url: basedomain + '/' + file.name
|
|
}
|
|
})
|
|
})
|
|
}
|
|
|
|
db.table('files').insert(files).then(() => {
|
|
|
|
for (let efile of existingFiles) files.push(efile)
|
|
|
|
res.json({
|
|
success: true,
|
|
files: files.map(file => {
|
|
return {
|
|
name: file.name,
|
|
size: file.size,
|
|
url: basedomain + '/' + file.name
|
|
}
|
|
})
|
|
})
|
|
|
|
for (let file of files) {
|
|
let ext = path.extname(file.name).toLowerCase()
|
|
if (utils.imageExtensions.includes(ext) || utils.videoExtensions.includes(ext)) {
|
|
file.thumb = basedomain + '/thumbs/' + file.name.slice(0, -ext.length) + '.png'
|
|
utils.generateThumbs(file)
|
|
}
|
|
}
|
|
|
|
}).catch(function(error) { console.log(error); res.json({ success: false, description: 'error' }) })
|
|
}
|
|
|
|
uploadsController.delete = function(req, res) {
|
|
|
|
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('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[0].username !== 'root')
|
|
this.where('userid', user[0].id)
|
|
})
|
|
.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' }) })
|
|
}
|
|
|
|
uploadsController.deleteFile = function(file) {
|
|
|
|
return new Promise(function(resolve, reject) {
|
|
fs.stat(path.join(__dirname, '..', config.uploads.folder, file), function(err, stats) {
|
|
if (err) { return reject(err) }
|
|
fs.unlink(path.join(__dirname, '..', config.uploads.folder, file), function(err) {
|
|
if (err) { return reject(err) }
|
|
return resolve()
|
|
})
|
|
})
|
|
})
|
|
|
|
}
|
|
|
|
uploadsController.list = function(req, res) {
|
|
|
|
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
|
|
|
|
db.table('files')
|
|
.where(function() {
|
|
if (req.params.id === undefined)
|
|
this.where('id', '<>', '')
|
|
else
|
|
this.where('albumid', req.params.id)
|
|
})
|
|
.where(function() {
|
|
if (user[0].username !== 'root')
|
|
this.where('userid', user[0].id)
|
|
})
|
|
.orderBy('id', 'DESC')
|
|
.limit(25)
|
|
.offset(25 * offset)
|
|
.select('id', 'albumid', 'timestamp', 'name', 'userid')
|
|
.then((files) => {
|
|
db.table('albums').then((albums) => {
|
|
|
|
let basedomain = req.get('host')
|
|
for (let domain of config.domains)
|
|
if (domain.host === req.get('host'))
|
|
if (domain.hasOwnProperty('resolve'))
|
|
basedomain = domain.resolve
|
|
|
|
let userids = []
|
|
|
|
for (let file of files) {
|
|
file.file = basedomain + '/' + file.name
|
|
file.date = new Date(file.timestamp * 1000)
|
|
file.date = utils.getPrettyDate(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()
|
|
|
|
file.album = ''
|
|
|
|
if (file.albumid !== undefined)
|
|
for (let album of albums)
|
|
if (file.albumid === album.id)
|
|
file.album = album.name
|
|
|
|
// Only push usernames if we are root
|
|
if (user[0].username === 'root')
|
|
if (file.userid !== undefined && file.userid !== null && file.userid !== '')
|
|
userids.push(file.userid)
|
|
|
|
let ext = path.extname(file.name).toLowerCase()
|
|
if (utils.imageExtensions.includes(ext) || utils.videoExtensions.includes(ext)) {
|
|
file.thumb = basedomain + '/thumbs/' + file.name.slice(0, -ext.length) + '.png'
|
|
utils.generateThumbs(file)
|
|
}
|
|
}
|
|
|
|
// If we are a normal user, send response
|
|
if (user[0].username !== 'root') return res.json({ success: true, files })
|
|
|
|
// If we are root but there are no uploads attached to a user, send response
|
|
if (userids.length === 0) return res.json({ success: true, files })
|
|
|
|
db.table('users').whereIn('id', userids).then((users) => {
|
|
for (let user of users)
|
|
for (let file of files)
|
|
if (file.userid === user.id)
|
|
file.username = user.username
|
|
|
|
return res.json({ success: true, 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' }) })
|
|
})
|
|
}
|
|
|
|
module.exports = uploadsController
|