@stemy/backend 2.8.4 → 2.9.1

package/fesm2015/stemy-backend.js

@@ -2,7 +2,7 @@ import { dirname, basename, join, resolve } from 'path';
 import { json } from 'body-parser';
 import { sign, verify } from 'jsonwebtoken';
 import { injectable, scoped, Lifecycle, injectAll, singleton, inject, isFactoryProvider, container } from 'tsyringe';
-import { createParamDecorator, BadRequestError, HttpError, getMetadataArgsStorage, Authorized, Post, UploadedFile, Body, Get, Param, QueryParams, QueryParam, Res, Controller, CurrentUser, Middleware, useContainer, useExpressServer } from 'routing-controllers';
+import { createParamDecorator, BadRequestError, HttpError, getMetadataArgsStorage, Authorized, Post, UploadedFile, Body, Get, Param, QueryParams, Res, QueryParam, Controller, CurrentUser, Middleware, useContainer, useExpressServer } from 'routing-controllers';
 import { OnMessage, ConnectedSocket, MessageBody, SocketController, Middleware as Middleware$1, useContainer as useContainer$1, useSocketServer } from 'socket-controllers';
 import { routingControllersToSpec } from 'routing-controllers-openapi';
 import { defaultMetadataStorage } from 'class-transformer/storage';
@@ -18,7 +18,7 @@ import { getValue as getValue$1, setValue } from 'mongoose/lib/utils';
 import { PassThrough, Readable } from 'stream';
 import { ObjectId } from 'bson';
 import fontKit_ from 'fontkit';
-import { fromBuffer, fromStream } from 'file-type';
+import { fromBuffer } from 'file-type';
 import sharp_ from 'sharp';
 import axios from 'axios';
 import { GridFSBucket } from 'mongodb';
@@ -774,18 +774,24 @@ const fontProps = [
     "xHeight", "numGlyphs", "characterSet", "availableFeatures"
 ];
 let AssetProcessor = AssetProcessor_1 = class AssetProcessor {
-    static getMimeType(buffer, mimeType) {
+    static checkTextFileType(type) {
+        return type.mime.indexOf("text") >= 0 || type.mime.indexOf("xml") >= 0;
+    }
+    static fixTextFileType(type, buffer) {
+        const text = buffer.toString("utf8");
+        if (text.indexOf("<svg") >= 0) {
+            return { ext: "svg", mime: "image/svg+xml" };
+        }
+        return type;
+    }
+    static fileTypeFromBuffer(buffer) {
+        var _a;
         return __awaiter$u(this, void 0, void 0, function* () {
-            try {
-                mimeType = (yield fromBuffer(buffer)).mime;
+            const type = ((_a = yield fromBuffer(buffer)) !== null && _a !== void 0 ? _a : { ext: "txt", mime: "text/plain" });
+            if (AssetProcessor_1.checkTextFileType(type)) {
+                return AssetProcessor_1.fixTextFileType(type, buffer);
             }
-            catch (e) {
-                if (!mimeType) {
-                    throw `Can't determine mime type`;
-                }
-                console.log(`Can't determine mime type`, e);
-            }
-            return mimeType;
+            return type;
         });
     }
     static extractFontFormat(font) {
@@ -823,12 +829,12 @@ let AssetProcessor = AssetProcessor_1 = class AssetProcessor {
             metadata[prop] = font[prop];
         });
     }
-    process(buffer, metadata, contentType) {
+    process(buffer, metadata, fileType) {
         return __awaiter$u(this, void 0, void 0, function* () {
-            if (AssetProcessor_1.isImage(contentType)) {
+            if (AssetProcessor_1.isImage(fileType.mime)) {
                 buffer = yield AssetProcessor_1.copyImageMeta(buffer, metadata);
             }
-            if (AssetProcessor_1.isFont(contentType)) {
+            if (AssetProcessor_1.isFont(fileType.mime)) {
                 AssetProcessor_1.copyFontMeta(buffer, metadata);
             }
             return buffer;
@@ -1171,56 +1177,37 @@ let Assets = class Assets {
     }
     write(stream, contentType = null, metadata = null) {
         return __awaiter$q(this, void 0, void 0, function* () {
-            let extension = null;
-            const fileTypeStream = copyStream(stream);
             const uploadStream = copyStream(stream);
+            const buffer = yield streamToBuffer(stream);
+            let fileType = { ext: "", mime: contentType };
             try {
-                const fileType = yield fromStream(fileTypeStream);
-                contentType = fileType.mime;
-                extension = fileType.ext;
+                fileType = yield AssetProcessor.fileTypeFromBuffer(buffer);
             }
             catch (e) {
-                if (!contentType) {
-                    throw `Can't determine content type`;
+                if (!fileType.mime) {
+                    throw `Can't determine mime type`;
                 }
-                console.log(`Can't determine content type`, e);
+                console.log(`Can't determine mime type`, e);
             }
-            contentType = contentType.trim();
-            extension = (extension || "").trim();
-            metadata = Object.assign({
-                extension,
-                downloadCount: 0,
-                firstDownload: null,
-                lastDownload: null
-            }, metadata || {});
-            metadata.filename = metadata.filename || new ObjectId().toHexString();
-            return new Promise(((resolve, reject) => {
-                const uploaderStream = this.bucket.openUploadStream(metadata.filename);
-                uploadStream.pipe(uploaderStream)
-                    .on("error", error => {
-                    reject(error.message || error);
-                })
-                    .on("finish", () => {
-                    const asset = new Asset(uploaderStream.id, {
-                        filename: metadata.filename,
-                        contentType,
-                        metadata
-                    }, this.collection, this.bucket);
-                    asset.save().then(() => {
-                        resolve(asset);
-                    }, error => {
-                        reject(error.message || error);
-                    });
-                });
-            }));
+            metadata = metadata || {};
+            return this.upload(uploadStream, fileType, metadata);
         });
     }
     writeBuffer(buffer, metadata = null, contentType = null) {
         return __awaiter$q(this, void 0, void 0, function* () {
-            contentType = yield AssetProcessor.getMimeType(buffer, contentType);
+            let fileType = { ext: "", mime: contentType };
+            try {
+                fileType = yield AssetProcessor.fileTypeFromBuffer(buffer);
+            }
+            catch (e) {
+                if (!fileType.mime) {
+                    throw `Can't determine mime type`;
+                }
+                console.log(`Can't determine mime type`, e);
+            }
             metadata = metadata || {};
-            buffer = yield this.assetProcessor.process(buffer, metadata, contentType);
-            return this.write(bufferToStream(buffer), contentType, metadata);
+            buffer = yield this.assetProcessor.process(buffer, metadata, fileType);
+            return this.upload(bufferToStream(buffer), fileType, metadata);
         });
     }
     writeUrl(url, metadata = null) {
@@ -1231,7 +1218,7 @@ let Assets = class Assets {
     }
     read(id) {
         return __awaiter$q(this, void 0, void 0, function* () {
-            return this.find({ _id: new ObjectId(id) });
+            return !id ? null : this.find({ _id: new ObjectId(id) });
         });
     }
     find(where) {
@@ -1261,6 +1248,38 @@ let Assets = class Assets {
             return asset.unlink();
         });
     }
+    upload(stream, fileType, metadata) {
+        return __awaiter$q(this, void 0, void 0, function* () {
+            const contentType = fileType.mime.trim();
+            const extension = (fileType.ext || "").trim();
+            metadata = Object.assign({
+                extension,
+                downloadCount: 0,
+                firstDownload: null,
+                lastDownload: null
+            }, metadata || {});
+            metadata.filename = metadata.filename || new ObjectId().toHexString();
+            return new Promise(((resolve, reject) => {
+                const uploaderStream = this.bucket.openUploadStream(metadata.filename);
+                stream.pipe(uploaderStream)
+                    .on("error", error => {
+                    reject(error.message || error);
+                })
+                    .on("finish", () => {
+                    const asset = new Asset(uploaderStream.id, {
+                        filename: metadata.filename,
+                        contentType,
+                        metadata
+                    }, this.collection, this.bucket);
+                    asset.save().then(() => {
+                        resolve(asset);
+                    }, error => {
+                        reject(error.message || error);
+                    });
+                });
+            }));
+        });
+    }
 };
 Assets = __decorate$t([
     injectable(),
@@ -1312,13 +1331,13 @@ class LazyAsset extends BaseEntity {
         this.load().then(() => {
             if (this.deleted)
                 return;
-            const progressPromise = !this.progressId ? Promise.resolve(null) : this.progresses.get(this.progressId).then(p => p.cancel());
-            progressPromise.then(() => {
-                this.startWorkingOnAsset().then(() => {
-                    console.log(`Started working on lazy asset: ${this.id}`);
-                }).catch(reason => {
-                    console.log(`Can't start working on lazy asset: ${this.id}\nReason: ${reason}`);
-                });
+            this.progresses.get(this.progressId).then(p => {
+                p === null || p === void 0 ? void 0 : p.cancel();
+            });
+            this.startWorkingOnAsset().then(() => {
+                console.log(`Started working on lazy asset: ${this.id}`);
+            }).catch(reason => {
+                console.log(`Can't start working on lazy asset: ${this.id}\nReason: ${reason}`);
             });
         });
     }
@@ -1797,7 +1816,7 @@ let Progresses = class Progresses {
     }
     get(id) {
         return __awaiter$m(this, void 0, void 0, function* () {
-            return this.find({ _id: new ObjectId(id) });
+            return !id ? null : this.find({ _id: new ObjectId(id) });
         });
     }
     find(where) {
@@ -1878,7 +1897,7 @@ let LazyAssets = class LazyAssets {
     }
     read(id) {
         return __awaiter$l(this, void 0, void 0, function* () {
-            return this.find({ _id: new ObjectId(id) });
+            return !id ? null : this.find({ _id: new ObjectId(id) });
         });
     }
     find(where) {
@@ -2860,40 +2879,42 @@ let AssetsController = class AssetsController {
             }
         });
     }
-    getImageRotation(id, params, rotation = 0) {
-        var _a;
+    getImageRotation(id, params, res, rotation = 0) {
         return __awaiter$7(this, void 0, void 0, function* () {
-            const asset = yield this.assetResolver.resolve(id, params.lazy);
-            if (!asset) {
-                throw new HttpError(404, `Image with id: '${id}' not found.`);
-            }
-            if ((_a = asset.metadata) === null || _a === void 0 ? void 0 : _a.classified) {
-                throw new HttpError(403, `Image is classified, and can be only downloaded from a custom url.`);
-            }
+            const asset = yield this.getAsset("Asset", id, params.lazy, res);
             params.rotation = params.rotation || rotation;
             return asset.downloadImage(params);
         });
     }
-    getImage(id, params) {
+    getImage(id, params, res) {
         return __awaiter$7(this, void 0, void 0, function* () {
-            return this.getImageRotation(id, params);
+            return this.getImageRotation(id, params, res);
         });
     }
     getFile(id, lazy, res) {
+        return __awaiter$7(this, void 0, void 0, function* () {
+            const asset = yield this.getAsset("Asset", id, lazy, res);
+            return asset.download();
+        });
+    }
+    getAsset(type, id, lazy, res) {
         var _a, _b;
         return __awaiter$7(this, void 0, void 0, function* () {
             const asset = yield this.assetResolver.resolve(id, lazy);
             if (!asset) {
-                throw new HttpError(404, `File with id: '${id}' not found.`);
+                throw new HttpError(404, `${type} with id: '${id}' not found.`);
             }
             if ((_a = asset.metadata) === null || _a === void 0 ? void 0 : _a.classified) {
-                throw new HttpError(403, `Asset is classified, and can be only downloaded from a custom url.`);
+                throw new HttpError(403, `${type} is classified, and can be only downloaded from a custom url.`);
             }
             const ext = (_b = asset.metadata) === null || _b === void 0 ? void 0 : _b.extension;
             if (ext) {
                 res.header("content-disposition", `inline; filename=${asset.filename}.${ext}`);
             }
-            return asset.download();
+            if (asset.contentType) {
+                res.header("content-type", asset.contentType);
+            }
+            return asset;
         });
     }
 };
@@ -2915,16 +2936,16 @@ __decorate$a([
 ], AssetsController.prototype, "uploadUrl", null);
 __decorate$a([
     Get("/image/:id/:rotation"),
-    __param$5(0, Param("id")), __param$5(1, QueryParams()), __param$5(2, Param("rotation")),
+    __param$5(0, Param("id")), __param$5(1, QueryParams()), __param$5(2, Res()), __param$5(3, Param("rotation")),
     __metadata$7("design:type", Function),
-    __metadata$7("design:paramtypes", [String, Object, Number]),
+    __metadata$7("design:paramtypes", [String, Object, Object, Number]),
     __metadata$7("design:returntype", Promise)
 ], AssetsController.prototype, "getImageRotation", null);
 __decorate$a([
     Get("/image/:id"),
-    __param$5(0, Param("id")), __param$5(1, QueryParams()),
+    __param$5(0, Param("id")), __param$5(1, QueryParams()), __param$5(2, Res()),
     __metadata$7("design:type", Function),
-    __metadata$7("design:paramtypes", [String, Object]),
+    __metadata$7("design:paramtypes", [String, Object, Object]),
     __metadata$7("design:returntype", Promise)
 ], AssetsController.prototype, "getImage", null);
 __decorate$a([