const config = require('../config.js') const db = require('knex')(config.database) const randomstring = require('randomstring') const utils = require('./utilsController.js') const path = require('path') 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') fields.push('identifier') } 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 = utils.getPrettyDate(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() let basedomain = req.get('host') for (let domain of config.domains) if (domain.host === req.get('host')) if (domain.hasOwnProperty('resolve')) basedomain = domain.resolve album.identifier = basedomain + '/a/' + album.identifier 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 let ext = path.extname(file.name).toLowerCase() if (utils.extensions.includes(ext)) { file.thumb = basedomain + '/thumbs/' + file.name.slice(0, -ext.length) + '.png' utils.generateThumbs(file) } } return res.json({ success: true, title: title, count: files.length, 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