From 0120d3f2267d2b012800722541e371e43ebc1d99 Mon Sep 17 00:00:00 2001
From: jokob-sk <jokob.sk@gmail.com>
Date: Fri, 14 Jan 2022 14:48:42 +1100
Subject: [PATCH] maintenance task DB

---
 back/maintenance.py          | 23 ++++++++++++
 front/maintenance.php        | 60 ++++++++++++++++++++++++------
 front/php/server/devices.php | 71 ++++++++++++++++++++++++++++--------
 3 files changed, 126 insertions(+), 28 deletions(-)
 create mode 100644 back/maintenance.py

diff --git a/back/maintenance.py b/back/maintenance.py
new file mode 100644
index 00000000..07422d1e
--- /dev/null
+++ b/back/maintenance.py
@@ -0,0 +1,23 @@
+import sys
+import shutil
+
+
+def main ():    
+
+    argument = str(sys.argv[1])
+
+    ## Main Commands
+    if argument == 'backup_db':
+        res = backup_db()
+
+    if argument == 'restore_db':
+        res = restore_db()
+
+def backup_db ():
+    # Header
+    shutil.copyfile("/home/pi/pialert/db/pialert.db", "/home/pi/pialert/config/pialert.db_bak")
+
+
+def restore_db ():
+    # Header
+    shutil.copyfile("/home/pi/pialert/config/pialert.db_bak", "/home/pi/pialert/db/pialert.db")
\ No newline at end of file
diff --git a/front/maintenance.php b/front/maintenance.php
index 6fd97a09..267f087d 100644
--- a/front/maintenance.php
+++ b/front/maintenance.php
@@ -36,28 +36,28 @@
           <button type="button" class="btn btn-default pa-btn pa-btn-delete"  style="margin-left:0px;"
             id="btnDeleteMAC"   onclick="askDeleteDevicesWithEmptyMACs()">   Delete Devices with empty MACs </button>     
       </div>
-    </div>
-
-    <div class="col-xs-12">
+   
       <div class="pull-right">
-          <button type="button" class="btn btn-default pa-btn pa-btn-delete"  style="margin-left:0px;"
-            id="btnDelete"   onclick="askDeleteAllDevices()">   Delete All Devices</button>     
+          <button type="button" class="btn btn-default pa-btn pa-btn-create"  style="margin-left:0px;"
+            id="btnDelete"   onclick="askRunScan1min()">   Run 1 min scan now</button>     
+      </div>
+
+      <div class="pull-right">
+          <button type="button" class="btn btn-default pa-btn pa-btn-create"  style="margin-left:0px;"
+            id="btnDelete"   onclick="askRunScan15min()">   Run 15 min scan now</button>     
       </div>
-    </div>
 
-    <!-- <div class="col-xs-12">
       <div class="pull-right">
           <button type="button" class="btn btn-default pa-btn pa-btn-create"  style="margin-left:0px;"
             id="btnBackup"   onclick="askCreateBackupDB()"> Backup DB </button>     
       </div>
-    </div>
-
-    <div class="col-xs-12">
       <div class="pull-right">
           <button type="button" class="btn btn-default pa-btn pa-btn-delete"  style="margin-left:0px;"
             id="btnRestore"   onclick="askRestoreBackupDB()"> Restore DB </button>     
       </div>
-    </div> -->
+   
+    </div>
+
 
     <!-- ----------------------------------------------------------------------- -->
     </section>
@@ -73,6 +73,9 @@
 
 
 <script>
+
+// delete devices with emty macs
+
   function askDeleteDevicesWithEmptyMACs () {
   // Ask delete device 
 
@@ -89,7 +92,7 @@ function deleteDevicesWithEmptyMACs()
   });
 }
 
-
+// delete all devices 
 function askDeleteAllDevices () {
   // Ask delete device 
 
@@ -107,6 +110,39 @@ function deleteAllDevices()
 }
 
 
+// Run ad-hoc scans
+
+function askRunScan1min () {
+  // Ask delete device
+
+  showModalWarning('Scan 1 min now', 'This runs the 1 min scan sequence',
+    'Cancel', 'Scan', 'runScan1min');
+}
+
+
+function runScan1min()
+{ 
+  // Scan
+  $.get('php/server/devices.php?action=runScan1min', function(msg) {
+    showMessage (msg);
+  });
+}
+
+function askRunScan15min () {
+  // Ask delete device
+
+  showModalWarning('Scan 15 min now', 'This runs the 15 min scan sequence',
+    'Cancel', 'Scan', 'runScan15min');
+}
+
+
+function runScan15min()
+{ 
+  // Scan
+  $.get('php/server/devices.php?action=runScan15min', function(msg) {
+    showMessage (msg);
+  });
+}
 
 
 // DB backup
diff --git a/front/php/server/devices.php b/front/php/server/devices.php
index 958fed48..f88f950b 100644
--- a/front/php/server/devices.php
+++ b/front/php/server/devices.php
@@ -35,6 +35,8 @@
       case 'createBackupDB':          createBackupDB();                        break;
       case 'restoreBackupDB':         restoreBackupDB();                       break;
       case 'deleteAllDevices':        deleteAllDevices();                      break;
+      case 'runScan15min':            runScan15min();                          break;
+      case 'runScan1min':             runScan1min();                           break;
       
       
  
@@ -217,20 +219,51 @@ function deleteAllDevices() {
   }
 }
 
+//------------------------------------------------------------------------------
+//  Run scan 1 min now
+//------------------------------------------------------------------------------
+function runScan1min() {
+
+  $command = escapeshellcmd('/home/pi/pialert/back/pialert.py 1');
+  $output = shell_exec($command);
+
+
+  echo $output;
+}
+
+
+//------------------------------------------------------------------------------
+//  Run scan 15 min now
+//------------------------------------------------------------------------------
+function runScan15min() {
+
+  $command = escapeshellcmd('/home/pi/pialert/back/pialert.py 15');
+  $output = shell_exec($command);
+
+
+  echo $output;
+}
+
 
 //------------------------------------------------------------------------------
 //  Create a DB backup
 //------------------------------------------------------------------------------
 function createBackupDB() {
  
-  //
-  $result = copy("/home/pi/pialert/db/pialert.db","/home/pi/pialert/config/pialert.db_bak");
-  // check result
-  if ($result == TRUE) {
-    echo "Copied successfully";
-  } else {
-    echo "Error copying DB. SSH into instance and copy manually.";
-  }
+  // //
+  // $result = copy("/home/pi/pialert/db/pialert.db","/home/pi/pialert/config/pialert.db_bak");
+  // // check result
+  // if ($result == TRUE) {
+  //   echo "Copied successfully";
+  // } else {
+  //   echo "Error copying DB. SSH into instance and copy manually.";
+  // }
+
+  $command = escapeshellcmd('/home/pi/pialert/back/maintenance.py backup_DB');
+  $output = shell_exec($command);
+
+
+  echo $output;
 }
 
 //------------------------------------------------------------------------------
@@ -238,14 +271,20 @@ function createBackupDB() {
 //------------------------------------------------------------------------------
 function restoreBackupDB() {
  
-  //
-  $result = copy("/home/pi/pialert/config/pialert.db_bak", "/home/pi/pialert/db/pialert.db");
-  // check result
-  if ($result == TRUE) {
-    echo "Copied successfully";
-  } else {
-    echo "Error copying DB. SSH into instance and copy manually.";
-  }
+  // //
+  // $result = copy("/home/pi/pialert/config/pialert.db_bak", "/home/pi/pialert/db/pialert.db");
+  // // check result
+  // if ($result == TRUE) {
+  //   echo "Copied successfully";
+  // } else {
+  //   echo "Error copying DB. SSH into instance and copy manually.";
+  // }
+
+  $command = escapeshellcmd('/home/pi/pialert/back/maintenance.py restore_DB');
+  $output = shell_exec($command);
+
+
+  echo $output;
 }