Files
myDrive/dist/services/FileService/utils/createThumbnail.js
T

66 lines
3.5 KiB
JavaScript

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
const mongoose = require("../../../db/mongoose");
const conn = mongoose.connection;
const crypto = require("crypto");
const env = require("../../../enviroment/env");
const Thumbnail = require("../../../models/thumbnail");
const ObjectID = require('mongodb').ObjectID;
const sharp = require("sharp");
const concat = require("concat-stream");
const createThumbnail = (file, filename, user) => __awaiter(void 0, void 0, void 0, function* () {
return new Promise((resolve) => {
const password = user.getEncryptionKey();
let CIPHER_KEY = crypto.createHash('sha256').update(password).digest();
let bucket = new mongoose.mongo.GridFSBucket(conn.db, {
chunkSizeBytes: 1024 * 255
});
const readStream = bucket.openDownloadStream(ObjectID(file._id));
readStream.on("error", (e) => {
console.log("File service upload thumbnail error", e);
resolve(file);
});
const decipher = crypto.createDecipheriv('aes256', CIPHER_KEY, file.metadata.IV);
decipher.on("error", (e) => {
console.log("File service upload thumbnail decipher error", e);
resolve(file);
});
try {
const concatStream = concat((bufferData) => __awaiter(void 0, void 0, void 0, function* () {
const thumbnailIV = crypto.randomBytes(16);
//CIPHER_KEY = crypto.createHash('sha256').update(process.env.newPassword).digest()
const thumbnailCipher = crypto.createCipheriv("aes256", CIPHER_KEY, thumbnailIV);
bufferData = Buffer.concat([thumbnailIV, thumbnailCipher.update(bufferData), thumbnailCipher.final()]); //thumbnailCipher.update(bufferData)
const thumbnailModel = new Thumbnail({ name: filename, owner: user._id, data: bufferData });
yield thumbnailModel.save();
let updatedFile = yield conn.db.collection("fs.files")
.findOneAndUpdate({ "_id": file._id }, { "$set": { "metadata.hasThumbnail": true, "metadata.thumbnailID": thumbnailModel._id } });
updatedFile = updatedFile.value;
updatedFile = Object.assign(Object.assign({}, updatedFile), { metadata: Object.assign(Object.assign({}, updatedFile.metadata), { hasThumbnail: true, thumbnailID: thumbnailModel._id }) });
resolve(updatedFile);
})).on("error", (e) => {
console.log("File service upload concat stream error", e);
resolve(file);
});
const imageResize = sharp().resize(300).on("error", (e) => {
console.log("resize error", e);
resolve(file);
});
readStream.pipe(decipher).pipe(imageResize).pipe(concatStream);
}
catch (e) {
console.log(e);
resolve(file);
}
});
});
module.exports = createThumbnail;