From f377aa45519135bc04962611fe0291ac6cf9fb83 Mon Sep 17 00:00:00 2001
From: Abhinav Adduri <abhinadduri@gmail.com>
Date: Tue, 6 Jun 2017 10:23:37 -0700
Subject: [PATCH] working s3 integration

---
 frontend/src/fileReceiver.js |   2 +-
 server/portal_server.js      | 108 +++++++++++++++++------------------
 2 files changed, 54 insertions(+), 56 deletions(-)

diff --git a/frontend/src/fileReceiver.js b/frontend/src/fileReceiver.js
index 81311648..14a539ea 100644
--- a/frontend/src/fileReceiver.js
+++ b/frontend/src/fileReceiver.js
@@ -37,7 +37,7 @@ class FileReceiver extends EventEmitter {
               data: this.result,
               fname: xhr
                 .getResponseHeader('Content-Disposition')
-                .match(/="(.+)"/)[1]
+                .match(/=(.+)/)[1]
             });
           };
 
diff --git a/server/portal_server.js b/server/portal_server.js
index c1577fd7..ca1a1bbf 100644
--- a/server/portal_server.js
+++ b/server/portal_server.js
@@ -37,36 +37,48 @@ app.get('/assets/download/:id', (req, res) => {
     return;
   }
 
-  let params = {
-    Bucket: aws_credentials.bucketName,
-    Key: req.params.id
-  }
+  redis_client.hget(id, "filename", (err, reply) => { // maybe some expiration logic too
+    if (!reply) {
+      res.sendStatus(404);
+    } else {
+
+      let params = {
+        Bucket: aws_credentials.bucketName,
+        Key: id
+      }
+
+      s3.headObject(params, function(err, data) {
+        res.writeHead(200, {"Content-Disposition": "attachment; filename=" + reply,
+                            "Content-Type": "application/octet-stream",
+                            "Content-Length": data.ContentLength});
+        let file_stream = s3.getObject(params).createReadStream();
+
+        file_stream.on('finish', () => {
+          redis_client.del(id);
+          s3.deleteObject(params, function(err, data) {
+            if (!err) {
+              console.log('Deleted off s3.');
+            }
+          })
+        });
+
+        file_stream.pipe(res);
+      });
+
+      // s3.getObject(params, function(err, data) {
+      //   if (err) {
+      //     console.log(err, err.stack); // an error occurred
+      //     res.sendStatus(404);
+      //   }
+      //   else {
+
+
+
+      //   }
+      // })
 
-  s3.getObject(params, function(err, data) {
-    if (err) {
-      console.log(err, err.stack); // an error occurred
-    }
-    else {
-      res.writeHead(200, {"Content-Disposition": "attachment; filename=response", //+ reply,
-                          "Content-Type": "application/octet-stream"});
-      // res.setHeader("Content-Type", "application/octet-stream");
-      res.end(new Buffer(data.Body))
     }
   })
-
-  // redis_client.hget(id, "filename", (err, reply) => { // maybe some expiration logic too
-  //   if (!reply) {
-  //     res.sendStatus(404);
-  //   } else {
-
-      // res.download(__dirname + "/../static/" + id, reply, (err) => {
-      //   if (!err) {
-      //     redis_client.del(id);
-      //     fs.unlinkSync(__dirname + "/../static/" + id);
-      //   }
-      // });
-  //   }
-  // })
 });
 
 app.post('/delete/:id', (req, res) => {
@@ -83,21 +95,22 @@ app.post('/delete/:id', (req, res) => {
     res.sendStatus(404);
   }
 
-<<<<<<< HEAD
-  client.hget(id, 'delete', (err, reply) => {
-    if (!reply) {
-      res.sendStatus(404);
-    } else {
-      client.del(id);
-      fs.unlinkSync(__dirname + '/../static/' + id);
-=======
   redis_client.hget(id, "delete", (err, reply) => {
-    if (!reply) {
+    if (!reply || (delete_token !== reply)) {
       res.sendStatus(404);
     } else {
       redis_client.del(id);
-      fs.unlinkSync(__dirname + "/../static/" + id);
->>>>>>> currently not working, decryption seems to fail
+      let params = {
+        Bucket: aws_credentials.bucketName,
+        Key: id
+      }
+
+      s3.deleteObject(params, function(err, data) {
+        if (!err) {
+          console.log('Deleted off s3.');
+        }
+      })
+
       res.sendStatus(200);
     }
   });
@@ -110,12 +123,10 @@ app.post("/upload/:id", (req, res, next) => {
       return;
     }
 
-    let fstream;
     req.pipe(req.busboy);
     req.busboy.on("file", (fieldname, file, filename) => {
         console.log("Uploading: " + filename);
 
-
         let params = {
           Bucket: aws_credentials.bucketName,
           Key: req.params.id,
@@ -126,29 +137,16 @@ app.post("/upload/:id", (req, res, next) => {
           if (err) {
             console.log(err, err.stack); // an error occurred
           } else {
-            console.log(data);
-          }
-        })
-
-        return;
-
-        fstream = fs.createWriteStream(__dirname + "/../static/" + req.params.id);
-        file.pipe(fstream);
-        fstream.on("close", () => {
             let id = req.params.id;
             let uuid = crypto.randomBytes(10).toString('hex');
 
             redis_client.hmset([id, "filename", filename, "delete", uuid]);
 
-            // delete the file off the server in 24 hours
-            // setTimeout(() => {
-            //   fs.unlinkSync(__dirname + "/static/" + id);
-            // }, 86400000);
-
             redis_client.expire(id, 86400000);
             console.log("Upload Finished of " + filename);
             res.send(uuid);
-        });
+          }
+        })
     });
   });
 });