Jason Rasmussen
61 lines
1.9 KiB
TypeScript
61 lines
1.9 KiB
TypeScript
import { APP_UPLOAD_LOCATION } from '@app/common/constants';
|
|
import { BadRequestException, UnauthorizedException } from '@nestjs/common';
|
|
import { MulterOptions } from '@nestjs/platform-express/multer/interfaces/multer-options.interface';
|
|
import { randomUUID } from 'crypto';
|
|
import { Request } from 'express';
|
|
import { existsSync, mkdirSync } from 'fs';
|
|
import { diskStorage } from 'multer';
|
|
import { extname, join } from 'path';
|
|
import sanitize from 'sanitize-filename';
|
|
|
|
export const assetUploadOption: MulterOptions = {
|
|
fileFilter,
|
|
storage: diskStorage({
|
|
destination,
|
|
filename,
|
|
}),
|
|
};
|
|
|
|
export const multerUtils = { fileFilter, filename, destination };
|
|
|
|
function fileFilter(req: Request, file: any, cb: any) {
|
|
if (!req.user) {
|
|
return cb(new UnauthorizedException());
|
|
}
|
|
if (
|
|
file.mimetype.match(/\/(jpg|jpeg|png|gif|mp4|x-msvideo|quicktime|heic|heif|dng|x-adobe-dng|webp|tiff|3gpp|nef)$/)
|
|
) {
|
|
cb(null, true);
|
|
} else {
|
|
cb(new BadRequestException(`Unsupported file type ${extname(file.originalname)}`), false);
|
|
}
|
|
}
|
|
|
|
function destination(req: Request, file: Express.Multer.File, cb: any) {
|
|
if (!req.user) {
|
|
return cb(new UnauthorizedException());
|
|
}
|
|
|
|
const basePath = APP_UPLOAD_LOCATION;
|
|
const sanitizedDeviceId = sanitize(String(req.body['deviceId']));
|
|
const originalUploadFolder = join(basePath, req.user.id, 'original', sanitizedDeviceId);
|
|
|
|
if (!existsSync(originalUploadFolder)) {
|
|
mkdirSync(originalUploadFolder, { recursive: true });
|
|
}
|
|
|
|
// Save original to disk
|
|
cb(null, originalUploadFolder);
|
|
}
|
|
|
|
function filename(req: Request, file: Express.Multer.File, cb: any) {
|
|
if (!req.user) {
|
|
return cb(new UnauthorizedException());
|
|
}
|
|
|
|
const fileNameUUID = randomUUID();
|
|
const fileName = `${fileNameUUID}${req.body['fileExtension'].toLowerCase()}`;
|
|
const sanitizedFileName = sanitize(fileName);
|
|
cb(null, sanitizedFileName);
|
|
}
|