From 474193c5a08bb3b794ecbc159ff47843064149b7 Mon Sep 17 00:00:00 2001
From: Alfredo Cardigliano <alfredo.cardigliano@gmail.com>
Date: Tue, 7 Aug 2018 19:57:26 +0200
Subject: [PATCH] live capture: added stop button, updating buttons state when
 the download terminates

---
 scripts/lua/modules/livecapture_utils.lua | 47 +++++++++++++++++++++--
 1 file changed, 43 insertions(+), 4 deletions(-)

diff --git a/scripts/lua/modules/livecapture_utils.lua b/scripts/lua/modules/livecapture_utils.lua
index cc65d9c375..7a9acfd188 100644
--- a/scripts/lua/modules/livecapture_utils.lua
+++ b/scripts/lua/modules/livecapture_utils.lua
@@ -21,7 +21,7 @@ function drawHostLiveCaptureButton(if_id, host_info)
       <div class="modal-footer">
         <!--button class="btn btn-default" data-dismiss="modal" aria-hidden="true">]] print(i18n("close")) print[[</button-->
         <button type="submit" class="btn btn-primary" name="start" onclick="hostLiveCapture()">]] print(i18n("start")) print[[</button>
-        <!--button type="submit" class="btn btn-primary" name="stop" onclick="stopHostLiveCapture()" disabled>]] print(i18n("stop")) print[[</button-->
+        <button type="submit" class="btn btn-primary" name="stop" onclick="stopHostLiveCapture()" disabled>]] print(i18n("stop")) print[[</button>
       </div>
     </div>
   </div>
@@ -30,6 +30,29 @@ function drawHostLiveCaptureButton(if_id, host_info)
 <a href='#' onclick="$('#hostLiveCaptureModal').modal('show');">pcap</a>
 
 <script>
+  var updateHostLiveCaptureStatus_started = 0;
+  function updateHostLiveCaptureStatus() {
+    $.ajax({
+      type: 'GET',
+      url: ']] print(ntop.getHttpPrefix().."/lua/live_traffic.lua") print[[',
+      data: { 
+        ifid: ']] print(tostring(if_id)) print [[',
+        host: ']] print(host_info["host"]) print [[',
+        action: 'status',
+        csrf: ']] print(ntop.getRandomCSRFValue()) print[['
+      },
+      success: function(status) {
+        if (status.status == 'running') {
+          $("#hostLiveCaptureModal button[name=start]").prop('disabled', true);
+          $("#hostLiveCaptureModal button[name=stop]").prop('disabled', false);
+        } else {
+          $("#hostLiveCaptureModal button[name=start]").prop('disabled', false);
+          $("#hostLiveCaptureModal button[name=stop]").prop('disabled', true);
+        }
+      }
+    });
+  }
+
   function hostLiveCapture() {
     var params = {};
 
@@ -42,12 +65,28 @@ function drawHostLiveCaptureButton(if_id, host_info)
  
     var form = paramsToForm('<form action="]] print(ntop.getHttpPrefix().."/lua/live_traffic.lua") print[[" method="get"></form>', params);
     form.appendTo('body').submit();
+
+    if (!updateHostLiveCaptureStatus_started) {
+      updateHostLiveCaptureStatus_started = 1;
+      setInterval(updateHostLiveCaptureStatus, 2000);
+    }
   }
 
   function stopHostLiveCapture() {
-    $("#hostLiveCaptureModal button[name=start]").prop('disabled', false);
-    $("#hostLiveCaptureModal button[name=stop]").prop('disabled', true);
-    /* todo: use an async request here */
+    $.ajax({
+      type: 'GET',
+      url: ']] print(ntop.getHttpPrefix().."/lua/live_traffic.lua") print[[',
+      data: { 
+        ifid: ']] print(tostring(if_id)) print [[',
+        host: ']] print(host_info["host"]) print [[',
+        action: 'stop',
+        csrf: ']] print(ntop.getRandomCSRFValue()) print[['
+      },
+      success: function(content) {
+        $("#hostLiveCaptureModal button[name=start]").prop('disabled', false);
+        $("#hostLiveCaptureModal button[name=stop]").prop('disabled', true);
+      }
+    });
   }
 
 </script>