import * as fs from 'fs'; import { Utils } from '../Utils'; import * as path from 'path'; import { Opt } from '../new_fields/Doc'; import * as sharp from 'sharp'; import request = require('request-promise'); const uploadDirectory = path.join(__dirname, './public/files/'); export namespace DashUploadUtils { export interface Size { width: number; suffix: string; } export const Sizes: { [size: string]: Size } = { SMALL: { width: 100, suffix: "_s" }, MEDIUM: { width: 400, suffix: "_m" }, LARGE: { width: 900, suffix: "_l" }, }; const gifs = [".gif"]; const pngs = [".png"]; const jpgs = [".jpg", ".jpeg"]; const imageFormats = [...pngs, ...jpgs, ...gifs]; const videoFormats = [".mov", ".mp4"]; const size = "content-length"; const type = "content-type"; export interface UploadInformation { mediaPaths: string[]; fileNames: { [key: string]: string }; contentSize?: number; contentType?: string; } const generate = (prefix: string, url: string) => `${prefix}upload_${Utils.GenerateGuid()}${path.extname(url).toLowerCase()}`; const sanitize = (filename: string) => filename.replace(/\s+/g, "_"); export interface InspectionResults { isLocal: boolean; stream: any; normalizedUrl: string; contentSize?: number; contentType?: string; } export const InspectImage = async (url: string): Promise => { const { isLocal, stream, normalized: normalizedUrl } = classify(url); const results = { isLocal, stream, normalizedUrl }; if (isLocal) { return results; } const metadata = (await new Promise((resolve, reject) => { request.head(url, async (error, res) => { if (error) { return reject(error); } resolve(res); }); })).headers; return { contentSize: parseInt(metadata[size]), contentType: metadata[type], ...results }; }; export const UploadImage = async (metadata: InspectionResults, filename?: string, prefix = ""): Promise> => { const { isLocal, stream, normalizedUrl, contentSize, contentType } = metadata; const resolved = filename ? sanitize(filename) : generate(prefix, normalizedUrl); let extension = path.extname(normalizedUrl) || path.extname(resolved); extension && (extension = extension.toLowerCase()); let information: UploadInformation = { mediaPaths: [], fileNames: { clean: resolved }, contentSize, contentType, }; return new Promise(async (resolve, reject) => { const resizers = [ { resizer: sharp().rotate(), suffix: "_o" }, ...Object.values(Sizes).map(size => ({ resizer: sharp().resize(size.width, undefined, { withoutEnlargement: true }).rotate(), suffix: size.suffix })) ]; let nonVisual = false; if (pngs.includes(extension)) { resizers.forEach(element => element.resizer = element.resizer.png()); } else if (jpgs.includes(extension)) { resizers.forEach(element => element.resizer = element.resizer.jpeg()); } else if (![...imageFormats, ...videoFormats].includes(extension.toLowerCase())) { nonVisual = true; } if (imageFormats.includes(extension)) { for (let resizer of resizers) { const suffix = resizer.suffix; let mediaPath: string; await new Promise(resolve => { const filename = resolved.substring(0, resolved.length - extension.length) + suffix + extension; information.mediaPaths.push(mediaPath = uploadDirectory + filename); information.fileNames[suffix] = filename; stream(normalizedUrl).pipe(resizer.resizer).pipe(fs.createWriteStream(mediaPath)) .on('close', resolve) .on('error', reject); }); } } if (!isLocal || nonVisual) { await new Promise(resolve => { stream(normalizedUrl).pipe(fs.createWriteStream(uploadDirectory + resolved)).on('close', resolve); }); } resolve(information); }); }; const classify = (url: string) => { const isLocal = /Dash-Web(\\|\/)src(\\|\/)server(\\|\/)public(\\|\/)files/g.test(url); return { isLocal, stream: isLocal ? fs.createReadStream : request, normalized: isLocal ? path.normalize(url) : url }; }; export const createIfNotExists = async (path: string) => { if (await new Promise(resolve => fs.exists(path, resolve))) { return true; } return new Promise(resolve => fs.mkdir(path, error => resolve(error === null))); }; export const Destroy = (mediaPath: string) => new Promise(resolve => fs.unlink(mediaPath, error => resolve(error === null))); }