v3.0.0/src/api/routes/albums/albumZipGET.js

const path = require('path');
const jetpack = require('fs-jetpack');
const Route = require('../../structures/Route');
const Util = require('../../utils/Util');
const log = require('../../utils/Log');

class albumGET extends Route {
	constructor() {
		super('/album/:identifier/zip', 'get', { bypassAuth: true });
	}

	async run(req, res, db) {
		const { identifier } = req.params;
		if (!identifier) return res.status(400).json({ message: 'Invalid identifier supplied' });

		// TODO: Do we really want to let anyone create a zip of an album?
		/*
			Make sure it exists and it's enabled
		*/
		const link = await db.table('links')
			.where({
				identifier,
				enabled: true,
				enableDownload: true
			})
			.first();
		if (!link) return res.status(400).json({ message: 'The supplied identifier could not be found' });

		/*
			Same with the album, just to make sure is not a deleted album and a leftover link
		*/
		const album = await db.table('albums')
			.where('id', link.albumId)
			.first();
		if (!album) return res.status(400).json({ message: 'Album not found' });

		/*
			If the date when the album was zipped is greater than the album's last edit, we just send the zip to the user
		*/
		if (album.zippedAt > album.editedAt) {
			const filePath = path.join(__dirname, '../../../../', process.env.UPLOAD_FOLDER, 'zips', `${album.userId}-${album.id}.zip`);
			const exists = await jetpack.existsAsync(filePath);
			/*
				Make sure the file exists just in case, and if not, continue to it's generation.
			*/
			if (exists) {
				const fileName = `chibisafe-${identifier}.zip`;
				return res.download(filePath, fileName);
			}
		}

		/*
			Grab the files in a very unoptimized way. (This should be a join between both tables)
		*/
		const fileList = await db.table('albumsFiles')
			.where('albumId', link.albumId)
			.select('fileId');

		/*
			If there are no files, stop here
		*/
		if (!fileList || !fileList.length) return res.status(400).json({ message: 'Can\'t download an empty album' });

		/*
			Get the actual files
		*/
		const fileIds = fileList.map(el => el.fileId);
		const files = await db.table('files')
			.whereIn('id', fileIds)
			.select('name');
		const filesToZip = files.map(el => el.name);

		try {
			Util.createZip(filesToZip, album);
			await db.table('albums')
				.where('id', link.albumId)
				.update('zippedAt', db.fn.now());

			const filePath = path.join(__dirname, '../../../../', process.env.UPLOAD_FOLDER, 'zips', `${album.userId}-${album.id}.zip`);
			const fileName = `chibisafe-${identifier}.zip`;
			return res.download(filePath, fileName);
		} catch (error) {
			log.error(error);
			return res.status(500).json({ message: 'There was a problem downloading the album' });
		}
	}
}

module.exports = albumGET;