image.variation.processor.ts

/*
 * VPDB - Virtual Pinball Database
 * Copyright (C) 2019 freezy <freezy@vpdb.io>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

import { createWriteStream } from 'fs';
import gm from 'gm';

import { logger } from '../../common/logger';
import { RequestState } from '../../common/typings/context';
import { FileDocument } from '../file.document';
import { FileUtil } from '../file.util';
import { FileVariation, ImageFileVariation } from '../file.variations';
import { CreationProcessor } from './processor';

require('bluebird').promisifyAll(gm.prototype);

export class ImageVariationProcessor implements CreationProcessor<ImageFileVariation> {

	public name: string = 'image.variation';

	public canProcess(file: FileDocument, srcVariation: FileVariation, destVariation: FileVariation): boolean {
		return file.getMimeTypePrimary(srcVariation) === 'image' && file.getMimeTypePrimary(srcVariation) === 'image';
	}

	public getOrder(variation?: FileVariation): number {
		return 100 + (variation && variation.priority ? variation.priority : 0);
	}

	public async process(requestState: RequestState, file: FileDocument, src: string, dest: string, variation?: ImageFileVariation): Promise<string> {

		logger.debug(requestState, '[ImageVariationProcessor] Start: %s from %s to %s', file.toDetailedString(variation), FileUtil.log(src), FileUtil.log(dest));

		// do the processing
		const img = gm(src);
		img.quality(variation.quality || 70);
		img.interlace('Line');

		if (variation.width && variation.height) {
			img.resize(variation.width, variation.height);
		}

		if (variation.rotate) {
			img.rotate('black', variation.rotate);
		}

		let srcSize: { width: number, height: number };
		let scale: number;
		if (variation.portraitToSquare) {
			srcSize = file.metadata.size;
			scale = srcSize.height / 1920;
			img.rotate('black', -35);
			img.crop(300 * scale, 300 * scale, 1000 * scale, 1300 * scale);
			if (variation.size) {
				img.resize(variation.size, variation.size);
			}
		}
		if (variation.wideToSquare) {
			srcSize = file.metadata.size;
			scale = srcSize.width / 1920;
			img.rotate('black', -30);
			img.crop(220 * scale, 220 * scale, 1020 * scale, 970 * scale);
			if (variation.size) {
				img.resize(variation.size, variation.size);
			}
		}

		if (variation.landscape && file.metadata.size.width < file.metadata.size.height) {
			img.rotate('black', 90);
		}
		img.setFormat(file.getMimeSubtype(variation));

		logger.debug(requestState, '[ImageVariationProcessor] Saving: %s to %s', file.toDetailedString(variation), FileUtil.log(dest));
		//await (img as any).writeAsync(dest);

		await new Promise<FileDocument>((resolve, reject) => {
			const writeStream = createWriteStream(dest);
			// setup success handler
			writeStream.on('finish', () => {
				resolve(file);
			});
			writeStream.on('error', reject);
			img.stream().on('error', reject).pipe(writeStream).on('error', reject);
		});
		logger.debug(requestState, '[ImageVariationProcessor] Done: %s at %s', file.toDetailedString(variation), FileUtil.log(dest));
		return dest;
	}
}