vtlab-generic-functions 1.0.38 → 1.0.40

package/lambdaClass/index.js

@@ -23,7 +23,7 @@ module.exports = class Lambda {
     totalError: 0, totalSuccess: 0, error: [], success: [],
   };
 
-  constructor({carrier, event, DEBUG = false}) {
+  constructor({ carrier, event, DEBUG = false }) {
     this.DEBUG = false;
     this.carrier = carrier;
     if (event.body) {
@@ -57,18 +57,28 @@ module.exports = class Lambda {
   }
 
   recoverFullPayload = async () => {
-    if (this.event?.s3PayloadPath) {
-      let contentResponse = await this.s3.download(this.event.s3PayloadPath);
+    if (this.event?.s3PayloadPath || this.event?.payloadS3Key) {
+      let contentResponse = await this.s3.download(this.event.s3PayloadPath || this.event.payloadS3Key);
       this.jobs = JSON.parse(contentResponse.Body.toString());
     } else {
       this.jobs = this.event?.payload || [];
     }
   };
 
+  initialize = async () => {
+    try {
+      if(this.event?.s3PayloadPath || this.event?.payloadS3Key) {
+        await this.recoverFullPayload();
+      }
+    } catch (error) {
+      throw error;
+    }
+  };
+
   s3 = {
     upload: async (data, path, options = {}) => {
       return await s3bucket.uploadToS3(path, data, {
-        ContentEncoding: options.contentEncoding || "base64", ContentType: options.ContentType ||  "image/png",
+        ContentEncoding: options.contentEncoding || "base64", ContentType: options.ContentType || "image/png",
       });
     }, download: async (path) => {
       return await s3bucket.downloadFileFromS3(path);
@@ -77,28 +87,37 @@ module.exports = class Lambda {
   http = {
     get: async (URL, params, headers = {}) => {
       return await axiosRequest.get(URL, {
-        auth: {}, logId: this.logGroupName, headers: {...headers}, ...params
+        auth: {}, logId: this.logGroupName, headers: { ...headers }, ...params
       });
-    }, delete: async (URL, params, headers) => {
+    },
+    delete: async (URL, params, headers) => {
       return await axiosRequest.deletion(URL, {
-        auth: {}, logId: this.logGroupName, headers: {...headers}, ...params
+        auth: {}, logId: this.logGroupName, headers: { ...headers }, ...params
       });
-    }, post: async (URL, params, headers) => {
+    },
+    post: async (URL, params, headers) => {
       return await axiosRequest.post(URL, {
-        auth: {}, logId: this.logGroupName, headers: {...headers}, ...params,
+        auth: {}, logId: this.logGroupName, headers: { ...headers }, ...params,
       });
-    }, put: async (URL, params, headers) => {
+    },
+    put: async (URL, params, headers) => {
       return await axiosRequest.put(URL, {
-        auth: {}, logId: this.logGroupName, headers: {...headers}, ...params
+        auth: {}, logId: this.logGroupName, headers: { ...headers }, ...params
       });
-    }, downloadFile: async (method, URL, params, s3Path) => {
+    },
+    patch: async (URL, params, headers) => {
+      return await axiosRequest.patch(URL, {
+        auth: {}, logId: this.logGroupName, headers: { ...headers }, ...params
+      });
+    },
+    downloadFile: async (method, URL, params, s3Path) => {
       let response = await axiosRequest[method](URL, {
-        auth: {}, logId: this.logGroupName, headers: {}, config: {responseType: 'arraybuffer'}, ...params
+        auth: {}, logId: this.logGroupName, headers: {}, config: { responseType: 'arraybuffer' }, ...params
       });
       let buff = Buffer.from(response.data, "binary");
       let s3PathUploaded = await this.s3.upload(buff, s3Path);
       return s3PathUploaded;
-    }, postForm: async (URL, {attachment, document}, _headers) => {
+    }, postForm: async (URL, { attachment, document }, _headers) => {
       const formData = new FormData();
       formData.append('document', JSON.stringify(document));
       formData.append('attachment', attachment, "document.pdf");
@@ -125,25 +144,25 @@ module.exports = class Lambda {
         await this.ftpConnection.connect(this.configFTP);
         this.logMessage(`Correctly Connected To FTP`, true);
       } catch (error) {
-        throw {message: "Error from FTP"};
+        throw { message: "Error from FTP" };
       }
     }, listFolder: async (folderAndRegexp) => {
-      if (!this.ftpConnection) throw {message: "No connection was stablished"};
+      if (!this.ftpConnection) throw { message: "No connection was stablished" };
       let files = await this.ftpConnection.list(folderAndRegexp);
-      if (!files || files.length < 1) throw {message: "No files were found"};
+      if (!files || files.length < 1) throw { message: "No files were found" };
       this.logMessage(`Files Found: ${files.map((f) => f.name)}`);
       return files;
-    }, download: async ({folder, file}) => {
-      if (!this.ftpConnection) throw {message: "No connection was stablished"};
+    }, download: async ({ folder, file }) => {
+      if (!this.ftpConnection) throw { message: "No connection was stablished" };
       let downloadedFile = await promiseFTPWrapper.downloadFromFTP(this.ftpConnection, `${folder}${file}`);
       return downloadedFile;
-    }, putFile: async ({path, file}) => {
+    }, putFile: async ({ path, file }) => {
       await promiseFTPWrapper.uploadFileToFTP(this.configFTP, file, path);
-    }, downloadFiles: async ({folder, _files, removeFiles = false}) => {
+    }, downloadFiles: async ({ folder, _files, removeFiles = false }) => {
       const files = [..._files];
-      if (!this.ftpConnection) throw {message: "No connection was stablished"};
+      if (!this.ftpConnection) throw { message: "No connection was stablished" };
       for (let i = 0; i < files.length; i++) {
-        let file = await this.ftp.download({folder, file: files[i].name});
+        let file = await this.ftp.download({ folder, file: files[i].name });
         if (file) {
           files[i].file = file;
           if (removeFiles) await this.ftp.deleteFile(folder, files[i]);
@@ -153,7 +172,7 @@ module.exports = class Lambda {
     }, deleteFile: async (folder, file) => {
       await promiseFTPWrapper.deleteFileFromFTP(this.ftpConnection, `${folder}/${file}`);
     }, closeConnection: async () => {
-      if (!this.ftpConnection) throw {message: "No connection was stablished"};
+      if (!this.ftpConnection) throw { message: "No connection was stablished" };
       this.ftpConnection.end();
     }, sortFiles: (filesList) => {
       return filesList.sort((a, b) => {
@@ -168,7 +187,7 @@ module.exports = class Lambda {
         return 0;
       });
     }, getJobMatchingFile: (file) => {
-      if (!file) throw {message: "No file was sent"};
+      if (!file) throw { message: "No file was sent" };
       let jobFound = [...this.jobs].find((job) => {
         return String(file.name).includes(utils.getValueOrNull(job, "externalCarrier.trackingId"));
       });
@@ -189,11 +208,11 @@ module.exports = class Lambda {
   };
 
   logSendingToCarrier = async (payload) => {
-    await this.logMessage(`Sending to ${this.carrier} ${utils.dataToString({...payload}, "data").res}`, true);
+    await this.logMessage(`Sending to ${this.carrier} ${utils.dataToString({ ...payload }, "data").res}`, true);
   };
 
   logReceivedFromCarrier = async (payload, status) => {
-    await this.logMessage(`Received from ${this.carrier} (${status}) ${utils.dataToString({...payload}, "data").res}`, true);
+    await this.logMessage(`Received from ${this.carrier} (${status}) ${utils.dataToString({ ...payload }, "data").res}`, true);
   };
 
   addSuccessResult = (payload) => {
@@ -231,7 +250,7 @@ module.exports = class Lambda {
     }));
   };
 
-  logFinishProcess = async ({error} = {}) => {
+  logFinishProcess = async ({ error } = {}) => {
     if (this.results.totalError || error) {
       error = error && error.message ? error.message : error;
       await this.logMessage("Finished Process with errors: " + (error || "") + " " + JSON.stringify(this.results));
@@ -241,36 +260,47 @@ module.exports = class Lambda {
   };
 
   validatePayload = () => {
-    if (this.jobs.length < 1) throw {message: "No Jobs were sent"};
+    if (this.jobs.length < 1) throw { message: "No Jobs were sent" };
   };
 
   getJobsByTrackingId = () => {
     let objToReturn = {};
     this.jobs.forEach((job) => {
       let valueFound = utils.getValueOrNull(job, "externalCarrier.trackingId");
-      if (valueFound) objToReturn[job._id] = {trackingId: valueFound};
+      if (valueFound) objToReturn[job._id] = { trackingId: valueFound };
       objToReturn[job._id].jobOwner = {
-        account: {_id: job.jobOwner.account._id},
+        account: { _id: job.jobOwner.account._id },
       };
     });
     return objToReturn;
   };
 
-  endSuccess = () => {
+  uploadResultToS3 = async () => {
+    return await this.s3.upload(Buffer.from(utils.dataToString(this.results, "data").res), `tmp/results/${this.carrier}-${new Date().getTime()}.json`);
+  };
+  endSuccess = async () => {
+    //check if the response is bigger than 6291400 bytes
+    if(Buffer.byteLength(utils.dataToString(this.results, "data").res) > 6291400) {
+      return {
+        isBase64Encoded: false, statusCode: 200, body: utils.dataToString({
+          pathS3Result: (await this.uploadResultToS3())
+        }, "data").res,
+      };
+    }
     return {
       isBase64Encoded: false, statusCode: 200, body: utils.dataToString(this.results, "data").res,
     };
   };
   endError = (e) => {
-    let {code, res} = utils.dataToString(e);
+    let { code, res } = utils.dataToString(e);
     let body = {
       code, message: res, carrierCode: e.carrierCode, carrierMessage: e.carrierMessage,
     };
-    return {isBase64Encoded: false, statusCode: code ? code : 417, body: res};
+    return { isBase64Encoded: false, statusCode: code ? code : 417, body: res };
   };
   getToken = () => {
     const token = `${this.auth.username}:${this.auth.password}`;
     const encodedToken = Buffer.from(token).toString("base64");
     return encodedToken;
   };
-};
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vtlab-generic-functions",
-  "version": "1.0.38",
+  "version": "1.0.40",
   "scripts": {
     "test": "jest"
   },