From 8981ac7f42307d3b56e76a3aae9a99d4c6f83200 Mon Sep 17 00:00:00 2001
From: Matthias <ilovemilk@wusa.io>
Date: Fri, 17 Apr 2020 13:58:54 +0200
Subject: add recovery of detected file sequences

---
 src/components/FileOperationsTable.vue |  8 ++++--
 src/views/History.vue                  |  1 -
 src/views/Recover.vue                  | 46 ++++++++++++++++++----------------
 3 files changed, 31 insertions(+), 24 deletions(-)

diff --git a/src/components/FileOperationsTable.vue b/src/components/FileOperationsTable.vue
index d8b3efb..2f4e60f 100644
--- a/src/components/FileOperationsTable.vue
+++ b/src/components/FileOperationsTable.vue
@@ -1,6 +1,6 @@
 <template>
-    <vaadin-grid theme="row-dividers" height-by-rows column-reordering-allowed multi-sort :items.prop="fileOperations">
-        <vaadin-grid-selection-column auto-select frozen></vaadin-grid-selection-column>
+    <vaadin-grid ref="grid" theme="row-dividers" height-by-rows column-reordering-allowed multi-sort :items.prop="fileOperations">
+        <vaadin-grid-selection-column v-if="selectable" auto-select frozen></vaadin-grid-selection-column>
         <vaadin-grid-column width="5em" flex-grow="0" header="Status" ref="status"></vaadin-grid-column>
         <vaadin-grid-column width="9em" flex-grow="0" ref="operation" header="" class="operation"></vaadin-grid-column>
         <vaadin-grid-sort-column width="9em" path="originalName" header="Name" ref="name"></vaadin-grid-sort-column>
@@ -31,6 +31,10 @@ export default {
         data: {
             type: Array,
             required: true
+        },
+        selectable: {
+            type: Boolean,
+            default: true
         }
     },
     watch: {
diff --git a/src/views/History.vue b/src/views/History.vue
index 521b2b6..b973b57 100644
--- a/src/views/History.vue
+++ b/src/views/History.vue
@@ -95,7 +95,6 @@ export default {
         },
         onRecover() {
             var itemsToRecover = [];
-            const items = this.$refs.ransomware-table.items;
             const selected = this.$refs.ransomware-table.selectedItems;
             for (var i = 0; i < selected.length; i++) {
                 itemsToRecover.push(selected[i].id);
diff --git a/src/views/Recover.vue b/src/views/Recover.vue
index 9bb51c4..75b0599 100644
--- a/src/views/Recover.vue
+++ b/src/views/Recover.vue
@@ -8,11 +8,14 @@
                 <div class="notification-wrapper">
                     <Notification :text.sync="notificationText" @on-close="closeNotification" :visible.sync="visible"></Notification>
                 </div>
-                <Header header="Recover">
-                    <RecoverAction id="recover" label="Recover selected files" v-on:recover="onRecover" primary></RecoverAction>
-                </Header>
-                <div id="tables" v-if="detected">
-                    <FileOperationsTable v-for="(detection, index) in detections" :key="index" class="ransomware-table" :data="detection.fileOperations" v-on:table-state-changed="tableStateChanged"></FileOperationsTable>
+                <Header header="Recover"></Header>
+                <div id="detections" v-if="detected">
+                    <div class="detection" v-for="detection in detections" :key="detection.id">
+                        <Header :header="'Detection ' + detection.id">
+                            <RecoverAction id="recover" label="Recover detected files" v-on:recover="onRecover(detection.id)" primary></RecoverAction>
+                        </Header>
+                        <FileOperationsTable :selectable="false" :ref="'detection' + detection.id" class="ransomware-table" :data="detection.fileOperations" v-on:table-state-changed="tableStateChanged"></FileOperationsTable>
+                    </div>
                 </div>
                 <span id="message" v-if="!detected">
                     <iron-icon icon="verified-user"></iron-icon>
@@ -109,25 +112,23 @@ export default {
             })
             .catch( error => { console.error(error); });
         },
-        onRecover() {
+        onRecover(id) {
             var itemsToRecover = [];
-            const items = document.querySelector('#ransomware-table').items;
-            const selected = document.querySelector('#ransomware-table').selectedItems;
-            for (var i = 0; i < selected.length; i++) {
-                itemsToRecover.push(selected[i].id);
+            const detectionTable = this.$refs['detection' + id];
+            var items = detectionTable[0].$refs.grid.items;
+            for (var i = 0; i < items.length; i++) {
+                itemsToRecover.push(items[i].id);
             }
-            this.recover(itemsToRecover);
+            this.recover(id, itemsToRecover);
         },
-        remove(ids) {
-            ids.forEach(id => {
-                for (var i = 0; i < this.fileOperations.length; i++) {
-                    if (this.fileOperations[i].id === id) {
-                        this.fileOperations.splice(i, 1);
-                    }
+        remove(id) {
+            for (var i = 0; i < this.detections.length; i++) {
+                if (this.detections[i].id === id) {
+                    this.detections.splice(i, 1);
                 }
-            });
+            }
         },
-        async recover(ids) {
+        async recover(id, ids) {
             await this.$axios({
 					method: 'PUT',
                     url: this.recoverUrl + '/recover',
@@ -140,12 +141,12 @@ export default {
 							case 200:
                                 this.buildNotification(response.data.deleted, response.data.recovered);
                                 if(response.data.filesRecovered.length > 0)
-								    this.remove(response.data.filesRecovered);
+								    this.remove(id);
 								break;
 							default:
 								this.buildNotification(response.data.deleted, response.data.recovered);
 								if(response.data.filesRecovered.length > 0)
-								    this.remove(response.data.filesRecovered);
+								    this.remove(id);
 								break;
 						}
 					})
@@ -182,6 +183,9 @@ export default {
         font-size: 1.5em;
         font-weight: bold;
     }
+    .detection {
+        margin: 10px 0px 0px 0px;
+    }
     iron-pages {
         height: 100%;
     }
-- 
cgit v1.2.3