From ff55cf085e4cf298de1e3bbeffd541e165fa7189 Mon Sep 17 00:00:00 2001
From: Mcat12 <newtoncat12@yahoo.com>
Date: Sat, 10 Dec 2016 14:16:49 -0500
Subject: [PATCH] Add button to Gravity page

This makes sure that the user doesn't accidentally cause the lists
to update because they were curious and clicked the Update Lists
link on the sidebar. Also, they can update the lists multiple
times, if they so desire.
---
 gravity.php          |  3 ++-
 js/pihole/gravity.js | 14 +++++++++-----
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/gravity.php b/gravity.php
index ee606d56..3cb41228 100644
--- a/gravity.php
+++ b/gravity.php
@@ -16,7 +16,8 @@
     Success!
 </div>
 
-<pre id="output" style="width: 100%; height: 100%;"></pre>
+<button class="btn btn-lg btn-primary btn-block" id="gravityBtn">Update Lists</button>
+<pre id="output" style="width: 100%; height: 100%;" hidden="true"></pre>
 
 <?php
     require "footer.php";
diff --git a/js/pihole/gravity.js b/js/pihole/gravity.js
index 04691c1c..cd9bac3a 100644
--- a/js/pihole/gravity.js
+++ b/js/pihole/gravity.js
@@ -1,9 +1,11 @@
-function eventsourcetest() {
+function eventsource() {
     var alInfo = $("#alInfo");
     var alSuccess = $("#alSuccess");
-    var ta = document.getElementById("output");
+    var ta = $("#output");
     var source = new EventSource("php/gravity.sh.php");
 
+    ta.html("");
+    ta.show();
     alInfo.show();
     alSuccess.hide();
 
@@ -13,7 +15,7 @@ function eventsourcetest() {
             alSuccess.show();
         }
 
-        ta.innerHTML += e.data;
+        ta.append(e.data);
 
     }, false);
 
@@ -21,11 +23,13 @@ function eventsourcetest() {
     source.addEventListener("error", function(e) {
         alInfo.delay(1000).fadeOut(2000, function() { alInfo.hide(); });
         source.close();
+        $("#gravityBtn").removeAttr("disabled");
     }, false);
 }
 
-$(function(){
-   eventsourcetest();
+$("#gravityBtn").on("click", () => {
+    $("#gravityBtn").attr("disabled", true);
+    eventsource();
 });
 
 // Handle hiding of alerts