From 86a0e47d15afde8be46ccdadb5eaebe751b7556d Mon Sep 17 00:00:00 2001
From: Itay Mizeretz <itay.mizeretz@guardicore.com>
Date: Wed, 14 Feb 2018 15:51:22 +0200
Subject: [PATCH] Add log downloading from map

---
 monkey_island/cc/ui/package.json              |  1 +
 .../map/preview-pane/PreviewPane.js           | 30 +++++++++++++++++++
 2 files changed, 31 insertions(+)

diff --git a/monkey_island/cc/ui/package.json b/monkey_island/cc/ui/package.json
index 5ee2e5389..27b536365 100644
--- a/monkey_island/cc/ui/package.json
+++ b/monkey_island/cc/ui/package.json
@@ -63,6 +63,7 @@
   "dependencies": {
     "bootstrap": "^3.3.7",
     "core-js": "^2.5.1",
+    "downloadjs": "^1.4.7",
     "fetch": "^1.1.0",
     "js-file-download": "^0.4.1",
     "normalize.css": "^4.0.0",
diff --git a/monkey_island/cc/ui/src/components/map/preview-pane/PreviewPane.js b/monkey_island/cc/ui/src/components/map/preview-pane/PreviewPane.js
index 842440149..56c6d0e75 100644
--- a/monkey_island/cc/ui/src/components/map/preview-pane/PreviewPane.js
+++ b/monkey_island/cc/ui/src/components/map/preview-pane/PreviewPane.js
@@ -2,6 +2,7 @@ import React from 'react';
 import {Icon} from 'react-fa';
 import Toggle from 'react-toggle';
 import {OverlayTrigger, Tooltip} from 'react-bootstrap';
+import download from 'downloadjs'
 
 class PreviewPaneComponent extends React.Component {
 
@@ -88,6 +89,34 @@ class PreviewPaneComponent extends React.Component {
     );
   }
 
+  downloadLog(asset) {
+
+    fetch('/api/log?id=' + asset.id)
+      .then(res => res.json())
+      .then(res => {
+        let timestamp = res['timestamp'];
+        timestamp = timestamp.substr(0, timestamp.indexOf('.'));
+        let filename = res['monkey_label'].split(':').join('-') + ' - ' + timestamp + '.log';
+        download(atob(res['log']), filename, 'text/plain');
+      });
+
+  }
+
+  downloadLogRow(asset) {
+    return (
+      <tr>
+        <th>
+          Download Log
+        </th>
+        <td>
+          <a type="button" className="btn btn-primary"
+             disabled={!asset.has_log}
+             onClick={() => this.downloadLog(asset)}>Download</a>
+        </td>
+      </tr>
+    );
+  }
+
   exploitsTimeline(asset) {
     if (asset.exploits.length === 0) {
       return (<div />);
@@ -140,6 +169,7 @@ class PreviewPaneComponent extends React.Component {
             {this.servicesRow(asset)}
             {this.accessibleRow(asset)}
             {this.forceKillRow(asset)}
+            {this.downloadLogRow(asset)}
           </tbody>
         </table>
         {this.exploitsTimeline(asset)}