From 87ae9ac63ed4529a7696af1708bc801b586fba7b Mon Sep 17 00:00:00 2001
From: Daniel Truong <daniel@digitalspace.ca>
Date: Thu, 12 May 2022 09:55:32 -0700
Subject: [PATCH] BRS-501: setting up upload to S3 code for export data (#18)

---
 lambda/export/index.js | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/lambda/export/index.js b/lambda/export/index.js
index a6d120b..42cf533 100644
--- a/lambda/export/index.js
+++ b/lambda/export/index.js
@@ -1,3 +1,6 @@
+const AWS = require("aws-sdk");
+const s3 = new AWS.S3();
+const fs = require("fs");
 const writeXlsxFile = require("write-excel-file/node");
 const { runScan, runQuery, TABLE_NAME } = require("../dynamoUtil");
 const {
@@ -11,6 +14,7 @@ const {
 } = require("../formulaUtils");
 
 const FILE_PATH = process.env.FILE_PATH || "./";
+const FILE_NAME = process.env.FILE_NAME || "A&R_Export.xlsx";
 
 const noteKeys = {
   "Frontcountry Camping": "notes_frontcountryCamping",
@@ -521,8 +525,15 @@ exports.handler = async (event, context) => {
 
       await writeXlsxFile(rowsArray, {
         schema,
-        filePath: FILE_PATH + "data.xlsx",
+        filePath: FILE_PATH + FILE_NAME,
       });
+      console.log("Report generated");
+
+      // This means we are uploading to S3
+      if (FILE_PATH === "/tmp/" && process.env.S3_BUCKET_DATA) {
+        await uploadToS3();
+      }
+
       console.log("=== Export successful ===");
     }
   } catch (err) {
@@ -734,3 +745,17 @@ function generateRowsArray(groupedReports) {
 
   return rowsArray;
 }
+
+async function uploadToS3() {
+  // Get file as buffer
+  const buffer = fs.readFileSync(FILE_PATH + FILE_NAME);
+
+  const params = {
+    Bucket: process.env.S3_BUCKET_DATA,
+    Key: FILE_NAME,
+    Body: buffer,
+  };
+
+  await s3.putObject(params).promise();
+  console.log("File successfully uploaded to S3");
+}