mirror of
https://github.com/BobbyWibowo/lolisafe.git
synced 2024-12-13 16:06:21 +00:00
207 lines
7.6 KiB
JavaScript
207 lines
7.6 KiB
JavaScript
const config = require('../config.js')
|
|
const db = require('knex')(config.database)
|
|
const randomstring = require('randomstring')
|
|
const path = require('path')
|
|
const fs = require('fs')
|
|
const ffmpeg = require('fluent-ffmpeg')
|
|
const gm = require('gm')
|
|
|
|
let albumsController = {}
|
|
|
|
albumsController.list = function(req, res, next){
|
|
|
|
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 fields = ['id', 'name']
|
|
|
|
if(req.params.sidebar === undefined)
|
|
fields.push('timestamp')
|
|
|
|
db.table('albums').select(fields).where({enabled: 1, userid: user[0].id}).then((albums) => {
|
|
|
|
if(req.params.sidebar !== undefined)
|
|
return res.json({ success: true, albums })
|
|
|
|
let ids = []
|
|
for(let album of albums){
|
|
album.date = new Date(album.timestamp * 1000)
|
|
album.date = album.date.getFullYear() + '-' + (album.date.getMonth() + 1) + '-' + album.date.getDate() + ' ' + (album.date.getHours() < 10 ? '0' : '') + album.date.getHours() + ':' + (album.date.getMinutes() < 10 ? '0' : '') + album.date.getMinutes() + ':' + (album.date.getSeconds() < 10 ? '0' : '') + album.date.getSeconds()
|
|
|
|
ids.push(album.id)
|
|
}
|
|
|
|
db.table('files').whereIn('albumid', ids).select('albumid').then((files) => {
|
|
|
|
let albumsCount = {}
|
|
|
|
for(let id of ids) albumsCount[id] = 0
|
|
for(let file of files) albumsCount[file.albumid] += 1
|
|
for(let album of albums) album.files = albumsCount[album.id]
|
|
|
|
return res.json({ success: true, albums })
|
|
}).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'}) })
|
|
|
|
}
|
|
|
|
albumsController.create = function(req, res, next){
|
|
|
|
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 name = req.body.name
|
|
if(name === undefined || name === '')
|
|
return res.json({ success: false, description: 'No album name specified' })
|
|
|
|
db.table('albums').where({
|
|
name: name,
|
|
enabled: 1,
|
|
userid: user[0].id
|
|
}).then((album) => {
|
|
if(album.length !== 0) return res.json({ success: false, description: 'There\'s already an album with that name' })
|
|
|
|
db.table('albums').insert({
|
|
name: name,
|
|
enabled: 1,
|
|
userid: user[0].id,
|
|
identifier: randomstring.generate(8),
|
|
timestamp: Math.floor(Date.now() / 1000)
|
|
}).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'}) })
|
|
|
|
}
|
|
|
|
albumsController.delete = function(req, res, next){
|
|
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 id = req.body.id
|
|
if(id === undefined || id === '')
|
|
return res.json({ success: false, description: 'No album specified' })
|
|
|
|
db.table('albums').where({id: id, userid: user[0].id}).update({ enabled: 0 }).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'}) })
|
|
}
|
|
|
|
albumsController.rename = function(req, res, next){
|
|
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 id = req.body.id
|
|
if(id === undefined || id === '')
|
|
return res.json({ success: false, description: 'No album specified' })
|
|
|
|
let name = req.body.name
|
|
if(name === undefined || name === '')
|
|
return res.json({ success: false, description: 'No name specified' })
|
|
|
|
db.table('albums').where({name: name, userid: user[0].id}).then((results) => {
|
|
if(results.length !== 0) return res.json({ success: false, description: 'Name already in use' })
|
|
|
|
db.table('albums').where({id: id, userid: user[0].id}).update({ name: name }).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'}) })
|
|
|
|
}
|
|
|
|
albumsController.get = function(req, res, next){
|
|
let identifier = req.params.identifier
|
|
if(identifier === undefined) return res.status(401).json({ success: false, description: 'No identifier provided' })
|
|
|
|
db.table('albums')
|
|
.where('identifier', identifier)
|
|
.then((albums) => {
|
|
if(albums.length === 0) return res.json({ success: false, description: 'Album not found' })
|
|
|
|
let title = albums[0].name
|
|
db.table('files').select('name').where('albumid', albums[0].id).orderBy('id', 'DESC').then((files) => {
|
|
|
|
let basedomain = req.get('host')
|
|
for(let domain of config.domains)
|
|
if(domain.host === req.get('host'))
|
|
if(domain.hasOwnProperty('resolve'))
|
|
basedomain = domain.resolve
|
|
|
|
for(let file of files){
|
|
file.file = basedomain + '/' + file.name
|
|
|
|
if(config.uploads.generateThumbnails === true){
|
|
|
|
let extensions = ['.jpg', '.jpeg', '.bmp', '.gif', '.png', '.webm', '.mp4']
|
|
for(let ext of extensions){
|
|
if(path.extname(file.name) === ext){
|
|
|
|
file.thumb = basedomain + '/thumbs/' + file.name.slice(0, -ext.length) + '.png'
|
|
|
|
let thumbname = path.join(__dirname, '..', config.uploads.folder, 'thumbs') + '/' + file.name.slice(0, -ext.length) + '.png'
|
|
fs.access(thumbname, function(err) {
|
|
if (err && err.code === 'ENOENT') {
|
|
// File doesnt exist
|
|
|
|
if (ext === '.webm' || ext === '.mp4') {
|
|
ffmpeg('./' + config.uploads.folder + '/' + file.name)
|
|
.thumbnail({
|
|
timestamps: [0],
|
|
filename: '%b.png',
|
|
folder: './' + config.uploads.folder + '/thumbs',
|
|
size: '200x?'
|
|
})
|
|
.on('error', function(error) {
|
|
console.log('Error - ', error.message)
|
|
})
|
|
}
|
|
else {
|
|
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)
|
|
})
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return res.json({
|
|
success: true,
|
|
title: title,
|
|
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'}) })
|
|
|
|
}
|
|
|
|
module.exports = albumsController |