@resolveio/server-lib 12.5.53 → 12.5.54

package/src/methods/aws.ts

@@ -0,0 +1,507 @@
+import { AWS, MethodManager } from '../managers/method.manager';
+import { Files } from '../collections/file.collection';
+import SimpleSchema from 'simpl-schema';
+import { ResolveIOServer } from '../index';
+import { objectIdHexString } from '../managers/mongo.manager';
+import { GetObjectCommand } from '@aws-sdk/client-s3';
+import { getSignedUrl } from '@aws-sdk/s3-request-presigner';
+
+let request = require('request').defaults({ encoding: null });
+let fs = require('fs');
+let pdfCnt = 1000000;
+
+export function loadAWSMethods(methodManager: MethodManager) {
+	methodManager.methods({
+		uploadFileAndSave: {
+			check: new SimpleSchema({
+				fileName: {
+					type: String
+				},
+				fileData: {
+					type: String
+				},
+				fileSize: {
+					type: Number
+				},
+				fileOrder: {
+					type: Number
+				},
+				fileType: {
+					type: String
+				},
+				compress: {
+					type: Boolean
+				}
+			}),
+			function: function(fileName: string, fileData: string, fileSize: number, fileOrder: number, fileType: string, compress: Boolean) {
+				return new Promise(async (resolve, reject) => {
+					if (fileName) {
+						let key = 'uploads/' + Date.now() + '-' + (pdfCnt++) + '-' + fileName;
+						let s3 = (<AWS>this.getAWS()).s3();
+						s3.putObject({
+							Bucket: process.env.AWS_S3_BUCKET,
+							Key: key,
+							Body: Buffer.from(fileData.replace('data:' + fileData.split(',')[0].split(':')[1].split(';')[0] + ';base64,', ''), 'base64'), 
+							ACL: 'private', 
+							ContentType: fileData.split(',')[0].split(':')[1].split(';')[0], 
+							ContentDisposition: 'inline'
+						}).then(res => {
+							if (res) {
+								Files.create({
+									_id: objectIdHexString(), __v: 0, name: fileName, size: fileSize, key: key, order: fileOrder, type: fileType
+								}).then(resFile => {
+									resolve(resFile);
+								}, errFile => reject(errFile));
+							}
+							else {
+								resolve(null);
+							}
+						}, err => {
+							reject(err);
+						});
+					}
+				});
+			}
+		},
+		uploadNonPDFFileAndSave: {
+			check: new SimpleSchema({	
+				fileName: {	
+					type: String	
+				},	
+				fileData: {	
+					type: String	
+				},	
+				fileSize: {	
+					type: Number	
+				},	
+				fileOrder: {	
+					type: Number	
+				},	
+				fileType: {	
+					type: String	
+				}	
+			}),	
+			function: function(fileName: string, fileData: string, fileSize: number, fileOrder: number, fileType: string) {	
+				return new Promise((resolve, reject) => {	
+					if (fileName) {	
+						let key = 'uploads/' + Date.now() + '-' + fileName;	
+
+						let s3 = (<AWS>this.getAWS()).s3();
+						s3.putObject({
+							Bucket: process.env.AWS_S3_BUCKET, 
+							Key: key, 
+							Body: Buffer.from(fileData), 
+							ACL: 'private', 
+							ContentType: 'application/gzip', 
+							ContentDisposition: 'inline'
+						}).then(res => {
+							if (res) {
+								Files.create({	
+									_id: objectIdHexString(), __v: 0, name: fileName, size: fileSize, key: key, order: fileOrder, type: fileType
+								}).then(resFile => {	
+									resolve(resFile);	
+								}, errFile => reject(errFile));	
+							}
+							else {
+								resolve(null);
+							}
+						}, err => {
+							reject(err);
+						});
+					}	
+				});	
+			}	
+		},
+		uploadNonPDFBufferAndSave: {
+			check: new SimpleSchema({	
+				fileName: {	
+					type: String	
+				},	
+				fileData: {	
+					type: Buffer	
+				},	
+				fileSize: {	
+					type: Number	
+				},	
+				fileOrder: {	
+					type: Number	
+				},	
+				fileType: {	
+					type: String	
+				}	
+			}),	
+			function: function(fileName: string, fileData: Buffer, fileSize: number, fileOrder: number, fileType: string) {	
+				return new Promise((resolve, reject) => {	
+					if (fileName) {	
+						let key = 'uploads/' + Date.now() + '-' + fileName;	
+						
+						let s3 = (<AWS>this.getAWS()).s3();
+						s3.putObject({
+							Bucket: process.env.AWS_S3_BUCKET, 
+							Key: key, 
+							Body: fileData, 
+							ACL: 'private', 
+							ContentType: 'application/gzip', 
+							ContentDisposition: 'inline'
+						}).then(res => {	
+							if (res) {
+								Files.create({	
+									_id: objectIdHexString(), __v: 0, name: fileName, size: fileSize, key: key, order: fileOrder, type: fileType
+								}).then(resFile => {	
+									resolve(resFile);	
+								}, errFile => reject(errFile));	
+							}	
+							else {
+								resolve(null);
+							}
+						}, err => {
+							reject(err);
+						});	
+					}	
+				});	
+			}	
+		},
+		uploadNonPDFStreamAndSave: {
+			check: new SimpleSchema({	
+				fileName: {	
+					type: String	
+				},	
+				fileData: {	
+					type: String	
+				},	
+				fileOrder: {	
+					type: Number	
+				},	
+				fileType: {	
+					type: String	
+				}	
+			}),	
+			function: function(fileName: string, fileData: string, fileOrder: number, fileType: string) {	
+				return new Promise((resolve, reject) => {	
+					if (fileName) {	
+						let key = 'uploads/' + Date.now() + '-' + fileName;	
+						
+						let s3 = (<AWS>this.getAWS()).s3();
+						s3.putObject({
+							Bucket: process.env.AWS_S3_BUCKET, 
+							Key: key, 
+							Body: fs.createReadStream(fileData), 
+							ACL: 'private', 
+							ContentType: 'application/gzip', 
+							ContentDisposition: 'inline'
+						}).then(res => {
+							if (res) {
+								Files.create({	
+									_id: objectIdHexString(), __v: 0, name: fileName, size: 1, key: key, order: fileOrder, type: fileType
+								}).then(resFile => {	
+									resolve(resFile);	
+								}, errFile => reject(errFile));
+							}
+							else {
+								resolve(null);
+							}
+						}, err => {
+							reject(err);
+						});	
+					}	
+				});	
+			}	
+		},
+		deleteFiles: {
+			check: new SimpleSchema({
+				keys: {
+					type: Array
+				},
+				'keys.$': {
+					type: String
+				}
+			}),
+			function: function(keys: string[]) {	
+				return new Promise(async (resolve, reject) => {
+					let files = await Files.find({key: {$in: keys}});
+		
+					let deleteCnt = files.length;
+		
+					files.forEach(file => {
+						Files.deleteOne({_id: file._id});
+						
+						let s3 = (<AWS>this.getAWS()).s3();
+						s3.deleteObject({
+							Bucket: process.env.AWS_S3_BUCKET, 
+							Key: file.key
+						}).then(res => {
+							if (res) {
+								deleteCnt--;
+								if (!deleteCnt) {
+									resolve(files.length);
+								}
+							}
+						});
+					});
+				});
+			}
+		},
+		deleteFile: {
+			check: new SimpleSchema({
+				key: {
+					type: String
+				}
+			}),
+			function: function(key: string) {
+				return new Promise(async (resolve, reject) => {
+		
+					let file = await Files.findOne({key: key});
+		
+					if (file) {
+						Files.deleteOne({_id: file._id});
+					}
+
+					let s3 = (<AWS>this.getAWS()).s3();
+					s3.deleteObject({
+						Bucket: process.env.AWS_S3_BUCKET,
+						Key: key
+					}).then(res => {
+						resolve(res);
+					}, err => {
+						reject(err);
+					});
+				});
+			}
+		},
+		deleteFileWithId: {
+			check: new SimpleSchema({
+				id_file: {
+					type: String
+				}
+			}),
+			function: function(id_file: string) {
+				return new Promise(async (resolve, reject) => {
+		
+					let file = await Files.findOne({_id: id_file});
+		
+					if (!file) {
+						reject('Could not find file');
+					}
+		
+					let s3 = (<AWS>this.getAWS()).s3();
+					s3.deleteObject({
+						Bucket: process.env.AWS_S3_BUCKET,
+						Key: file.key
+					}).then(res => {
+						resolve(res);
+					}, err => {
+						reject(err);
+					});
+
+					Files.deleteOne({_id: file._id});
+				});
+			}
+		},
+		getSignedUrlsAndFilesWithId: {
+			check: new SimpleSchema({
+				ids: {
+					type: Array
+				},
+				'ids.$': {
+					type: String
+				},
+				expires: {
+					type: Number
+				}
+			}),
+			function: function(ids: string[], expires: number) {
+				return new Promise(async (resolve, reject) => {
+					let res = [];
+					let fileArray = [];
+
+					let files = await Files.find({_id: {$in: ids}});
+					
+					for (let i = 0; i < files.length; i++) {
+						let file = files[i];
+						let s3 = (<AWS>this.getAWS()).s3();
+						let getCommand = new GetObjectCommand({
+							Bucket: process.env.AWS_S3_BUCKET, 
+							Key: file.key
+						});
+						res.push(await getSignedUrl(s3, getCommand, {expiresIn: expires}));
+					}
+					
+					resolve({res: res, fileArray: files.map(a => a.name)});
+				});
+			}
+		},
+		getSignedUrlWithId: {
+			check: new SimpleSchema({
+				id_file: {
+					type: String
+				},
+				expires: {
+					type: Number
+				}
+			}),
+			function: function(id_file: string, expires: number) {
+				return new Promise(async (resolve, reject) => {
+					let file = await Files.findOne({_id: id_file});
+
+					let s3 = (<AWS>this.getAWS()).s3();
+					let getCommand = new GetObjectCommand({
+						Bucket: process.env.AWS_S3_BUCKET, 
+						Key: file.key
+					});
+					getSignedUrl(s3, getCommand, {expiresIn: expires}).then(res => {
+						resolve(res);
+					}, err => {
+						reject(err);
+					});
+				});
+			}
+		},
+		getSignedUrlUSEast1BucketAndKey: {
+			check: new SimpleSchema({
+				bucket: {
+					type: String
+				},
+				key: {
+					type: String
+				},
+				expires: {
+					type: Number
+				}
+			}),
+			function: function(bucket: string, key: string, expires: number) {
+				return new Promise((resolve, reject) => {
+					let s3 = (<AWS>this.getAWS()).s3USEast1();
+					let getCommand = new GetObjectCommand({
+						Bucket: bucket, 
+						Key: key
+					});
+					getSignedUrl(s3, getCommand, {expiresIn: expires}).then(res => {
+						resolve(res);
+					}, err => {
+						reject(err);
+					});
+				});
+			}
+		},
+		getSignedUrl: {
+			check: new SimpleSchema({
+				key: {
+					type: String
+				},
+				expires: {
+					type: Number
+				}
+			}),
+			function: function(key: string, expires: number) {
+				return new Promise((resolve, reject) => {
+					let s3 = (<AWS>this.getAWS()).s3();
+					let getCommand = new GetObjectCommand({
+						Bucket: process.env.AWS_S3_BUCKET, 
+						Key: key
+					});
+					getSignedUrl(s3, getCommand, {expiresIn: expires}).then(res => {
+						resolve(res);
+					}, err => {
+						reject(err);
+					});
+				});
+			}
+		},
+		getSignedUrls: {
+			check: new SimpleSchema({
+				keys: {
+					type: Array
+				},
+				'keys.$': {
+					type: String
+				},
+				expires: {
+					type: Number
+				}
+			}),
+			function: function(keys: string[], expires: number) {
+				return new Promise(async (resolve, reject) => {
+					let res = [];
+
+					for (let i = 0; i < keys.length; i++) {
+						let key = keys[i];
+						let s3 = (<AWS>this.getAWS()).s3();
+						let getCommand = new GetObjectCommand({
+							Bucket: process.env.AWS_S3_BUCKET, 
+							Key: key
+						});
+						res.push(await getSignedUrl(s3, getCommand, {expiresIn: expires}));
+					}
+
+					resolve(res);
+				});
+			}
+		},
+		getDataURIfromURL: {
+			check: new SimpleSchema({
+				url: {
+					type: String
+				}
+			}),
+			function: function(url: string) {
+				return new Promise((resolve, reject) => {
+					request.get(url, function (error, response, body) {
+						if (!error && response.statusCode === 200) {
+							let data = 'data:' + response.headers['content-type'] + ';base64,' + new Buffer(body).toString('base64');
+							resolve(data);
+						}
+						else {
+							reject(error);
+						}
+					});
+				});
+			}
+		},
+		emailFile: {
+			check: new SimpleSchema({
+				id_file: {
+					type: String
+				},
+				email: {
+					type: String
+				},
+                additional_message: {
+                    type: String
+                }
+			}),
+			function: function(id_file: string, email: string, additional_message) {
+				return new Promise(async (resolve, reject) => {
+					let file = await Files.findById(id_file);
+
+					if (!file) {
+						reject('Could not find file');
+						return;
+					}
+
+					this.callMethodInternal('getSignedUrl', file.key, 259200, (errPDF, resPDF) => {
+						if (resPDF) {
+							if (file.size < 5000000) {
+								let emailMessage = 'Attached is the file requested for email.\n' + (additional_message !== '' ? '\nAdditional Message:\n' + additional_message + '\n' : '') ;
+
+								this.sendEmail(email, ResolveIOServer.getClientName() + ' File - ' + file.name, emailMessage, '', [{
+									filename: file.name,
+									path: resPDF
+								}], null, '');
+							}
+							else {
+								let emailMessage = 'Please click the link to the requested file.\n' + (additional_message !== '' ? '\nAdditional Message:\n' + additional_message + '\n' : '') ;
+								
+								this.sendEmail(email, ResolveIOServer.getClientName() + ' File - ' + file.name, emailMessage + resPDF, '', null, null, '');
+							}
+
+							resolve(true);
+						}
+						else {
+							reject('Could not get file URL');
+						}
+					});
+				});
+			}
+		}
+	});
+}