mirror of
https://github.com/bpatrik/pigallery2.git
synced 2024-12-14 11:23:17 +02:00
174 lines
5.7 KiB
TypeScript
174 lines
5.7 KiB
TypeScript
///<reference path="../customtypings/jimp.d.ts"/>
|
|
import * as path from "path";
|
|
import * as crypto from "crypto";
|
|
import * as fs from "fs";
|
|
import * as os from "os";
|
|
import {NextFunction, Request, Response} from "express";
|
|
import {ErrorCodes, ErrorDTO} from "../../../common/entities/Error";
|
|
import {ContentWrapper} from "../../../common/entities/ConentWrapper";
|
|
import {DirectoryDTO} from "../../../common/entities/DirectoryDTO";
|
|
import {ProjectPath} from "../../ProjectPath";
|
|
import {PhotoDTO} from "../../../common/entities/PhotoDTO";
|
|
import {Config} from "../../../common/config/private/Config";
|
|
import {ThumbnailProcessingLib} from "../../../common/config/private/IPrivateConfig";
|
|
import {ThumbnailTH} from "../../model/threading/ThreadPool";
|
|
import {RendererFactory, RendererInput} from "../../model/threading/ThumbnailWoker";
|
|
|
|
|
|
export class ThumbnailGeneratorMWs {
|
|
private static initDone = false;
|
|
private static ThumbnailFunction: (input: RendererInput) => Promise<void> = null;
|
|
private static threadPool: ThumbnailTH = null;
|
|
|
|
public static init() {
|
|
if (this.initDone == true) {
|
|
return
|
|
}
|
|
|
|
|
|
if (Config.Server.enableThreading == true ||
|
|
Config.Server.thumbnail.processingLibrary != ThumbnailProcessingLib.Jimp) {
|
|
Config.Client.concurrentThumbnailGenerations = Math.max(1, os.cpus().length - 1);
|
|
} else {
|
|
Config.Client.concurrentThumbnailGenerations = 1;
|
|
}
|
|
|
|
this.ThumbnailFunction = RendererFactory.build(Config.Server.thumbnail.processingLibrary);
|
|
|
|
if (Config.Server.enableThreading == true &&
|
|
Config.Server.thumbnail.processingLibrary == ThumbnailProcessingLib.Jimp) {
|
|
this.threadPool = new ThumbnailTH(Config.Client.concurrentThumbnailGenerations);
|
|
}
|
|
|
|
this.initDone = true;
|
|
}
|
|
|
|
private static addThInfoTODir(directory: DirectoryDTO) {
|
|
if (typeof directory.photos == "undefined") {
|
|
directory.photos = [];
|
|
}
|
|
if (typeof directory.directories == "undefined") {
|
|
directory.directories = [];
|
|
}
|
|
ThumbnailGeneratorMWs.addThInfoToPhotos(directory.photos);
|
|
|
|
for (let i = 0; i < directory.directories.length; i++) {
|
|
ThumbnailGeneratorMWs.addThInfoTODir(directory.directories[i]);
|
|
}
|
|
|
|
}
|
|
|
|
private static addThInfoToPhotos(photos: Array<PhotoDTO>) {
|
|
let thumbnailFolder = ProjectPath.ThumbnailFolder;
|
|
for (let i = 0; i < photos.length; i++) {
|
|
let fullImagePath = path.join(ProjectPath.ImageFolder, photos[i].directory.path, photos[i].directory.name, photos[i].name);
|
|
for (let j = 0; j < Config.Client.Thumbnail.thumbnailSizes.length; j++) {
|
|
let size = Config.Client.Thumbnail.thumbnailSizes[j];
|
|
let thPath = path.join(thumbnailFolder, ThumbnailGeneratorMWs.generateThumbnailName(fullImagePath, size));
|
|
if (fs.existsSync(thPath) === true) {
|
|
if (typeof photos[i].readyThumbnails == "undefined") {
|
|
photos[i].readyThumbnails = [];
|
|
}
|
|
photos[i].readyThumbnails.push(size);
|
|
}
|
|
}
|
|
let iconPath = path.join(thumbnailFolder, ThumbnailGeneratorMWs.generateThumbnailName(fullImagePath, Config.Client.Thumbnail.iconSize));
|
|
if (fs.existsSync(iconPath) === true) {
|
|
photos[i].readyIcon = true;
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
public static addThumbnailInformation(req: Request, res: Response, next: NextFunction) {
|
|
if (!req.resultPipe)
|
|
return next();
|
|
|
|
let cw: ContentWrapper = req.resultPipe;
|
|
if (cw.directory) {
|
|
ThumbnailGeneratorMWs.addThInfoTODir(cw.directory);
|
|
}
|
|
if (cw.searchResult) {
|
|
ThumbnailGeneratorMWs.addThInfoToPhotos(cw.searchResult.photos);
|
|
}
|
|
|
|
|
|
return next();
|
|
|
|
}
|
|
|
|
public static generateThumbnail(req: Request, res: Response, next: NextFunction) {
|
|
if (!req.resultPipe)
|
|
return next();
|
|
|
|
//load parameters
|
|
let imagePath = req.resultPipe;
|
|
let size: number = parseInt(req.params.size) || Config.Client.Thumbnail.thumbnailSizes[0];
|
|
|
|
//validate size
|
|
if (Config.Client.Thumbnail.thumbnailSizes.indexOf(size) === -1) {
|
|
size = Config.Client.Thumbnail.thumbnailSizes[0];
|
|
}
|
|
|
|
ThumbnailGeneratorMWs.generateImage(imagePath, size, false, req, res, next);
|
|
|
|
|
|
}
|
|
|
|
public static generateIcon(req: Request, res: Response, next: NextFunction) {
|
|
if (!req.resultPipe)
|
|
return next();
|
|
|
|
//load parameters
|
|
let imagePath = req.resultPipe;
|
|
let size: number = Config.Client.Thumbnail.iconSize;
|
|
ThumbnailGeneratorMWs.generateImage(imagePath, size, true, req, res, next);
|
|
|
|
|
|
}
|
|
|
|
|
|
private static async generateImage(imagePath: string, size: number, makeSquare: boolean, req: Request, res: Response, next: NextFunction) {
|
|
//generate thumbnail path
|
|
let thPath = path.join(ProjectPath.ThumbnailFolder, ThumbnailGeneratorMWs.generateThumbnailName(imagePath, size));
|
|
|
|
|
|
req.resultPipe = thPath;
|
|
|
|
//check if thumbnail already exist
|
|
if (fs.existsSync(thPath) === true) {
|
|
return next();
|
|
}
|
|
|
|
//create thumbnail folder if not exist
|
|
if (!fs.existsSync(ProjectPath.ThumbnailFolder)) {
|
|
fs.mkdirSync(ProjectPath.ThumbnailFolder);
|
|
}
|
|
|
|
//run on other thread
|
|
let input = <RendererInput>{
|
|
imagePath: imagePath,
|
|
size: size,
|
|
thPath: thPath,
|
|
makeSquare: makeSquare,
|
|
qualityPriority: Config.Server.thumbnail.qualityPriority
|
|
};
|
|
try {
|
|
if (this.threadPool !== null) {
|
|
await this.threadPool.execute(input);
|
|
return next();
|
|
} else {
|
|
await ThumbnailGeneratorMWs.ThumbnailFunction(input);
|
|
return next();
|
|
}
|
|
} catch (error) {
|
|
return next(new ErrorDTO(ErrorCodes.THUMBNAIL_GENERATION_ERROR, "ErrorDTO during generating thumbnail", error));
|
|
}
|
|
}
|
|
|
|
private static generateThumbnailName(imagePath: string, size: number): string {
|
|
return crypto.createHash('md5').update(imagePath).digest('hex') + "_" + size + ".jpg";
|
|
}
|
|
}
|
|
|