meidosafe/controllers/albumsController.js

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