From 99a1564c09358515d5ab334d04c03010947956d0 Mon Sep 17 00:00:00 2001
From: dylankb <dylan.barnard@gmail.com>
Date: Fri, 30 Mar 2018 12:14:38 -0700
Subject: [PATCH] Add logic to start script

---
 start.js | 39 +++++++++++++++++++++++++++++++++++++--
 1 file changed, 37 insertions(+), 2 deletions(-)

diff --git a/start.js b/start.js
index 223942f..181b0db 100644
--- a/start.js
+++ b/start.js
@@ -12,6 +12,7 @@ const kadNodePort = require('./constants').KADNODE_PORT
 const publicIp = require('public-ip');
 const fs = require('fs');
 const fileUtils = require('./utils/file').fileSystem;
+const backoff = require('backoff');
 
 publicIp.v4().then(ip => {
  const kademliaNode = new kad.KademliaNode({
@@ -30,6 +31,7 @@ publicIp.v4().then(ip => {
       console.log('end')
     })
     serverConnection.on('data', (receivedData, error) => {
+      if (error) { throw error; }
      receivedData = JSON.parse(receivedData)
      console.log("received data: ", receivedData)
   
@@ -72,9 +74,42 @@ publicIp.v4().then(ip => {
         batNode.retrieveFile(filePath);
       } else if (receivedData.messageType === "CLI_AUDIT_FILE") {
         let filePath = receivedData.filePath;
-        
-        console.log("received path: ", filePath); 
+        let fibonacciBackoff = backoff.exponential({
+            randomisationFactor: 0,
+            initialDelay: 20,
+            maxDelay: 2000
+        });
+
+        console.log("received path: ", filePath);
         batNode.auditFile(filePath);
+
+        // post audit cleanup
+        serverConnection.on('close', () => {
+          batnode._audit.ready = false;
+          batnode._audit.data = null;
+          batnode._audit.passed = false;
+        });
+
+        fibonacciBackoff.failAfter(10);
+
+        fibonacciBackoff.on('backoff', function(number, delay) {
+            console.log(number + ' ' + delay + 'ms');
+        });
+
+        fibonacciBackoff.on('ready', function() {
+          if (!batnode._audit.ready) {
+            fibonacciBackoff.backoff();
+          } else {
+            serverConnection.write(JSON.stringify(batnode._audit.passed));
+            return;
+          }
+        });
+
+        fibonacciBackoff.on('fail', function() {
+          console.log('Timeout: failed to complete audit');
+        });
+
+        fibonacciBackoff.backoff();
       }
     });
   }