[WebUI] Fix graph dataset selector initialization

1 files changed, 20 insertions, 23 deletions

diff --git a/interface/js/app/graph.js b/interface/js/app/graph.js
index 86f7ca352..70a510bc3 100644
--- a/interface/js/app/graph.js
+++ b/interface/js/app/graph.js
@@ -82,12 +82,13 @@ function($, D3Evolution, unused) {
         }
     };
 
+    // Get selectors' current state
+    function getSelector(id) {
+        var e = document.getElementById(id);
+        return e.options[e.selectedIndex].value;
+    }
+
     function initGraph() {
-        // Get selectors' current state
-        function getSelector(id) {
-            var e = document.getElementById(id);
-            return e.options[e.selectedIndex].value;
-        }
         var graph = new D3Evolution("graph", $.extend({}, graph_options, {
             type:        getSelector("selType"),
             interpolate: getSelector("selInterpolate"),
@@ -148,16 +149,6 @@ function($, D3Evolution, unused) {
         });
     }
 
-    // Handling mouse events on overlapping elements
-    $("#rrd-pie").mouseover(function () {
-        $("#rrd-pie").css("z-index", "200");
-        $("#rrd-table_toggle").css("z-index", "300");
-    });
-    $("#rrd-table_toggle").mouseover(function () {
-        $("#rrd-pie").css("z-index", "0");
-        $("#rrd-table_toggle").css("z-index", "0");
-    });
-
     var interface = {};
 
     interface.draw = function(rspamd, graphs, neighbours, checked_server, type) {
@@ -180,14 +171,6 @@ function($, D3Evolution, unused) {
             graphs.graph = initGraph();
         }
 
-        if (type === undefined) {
-            function getSelector(id) {
-                var e = document.getElementById(id);
-                return e.options[e.selectedIndex].value;
-            }
-            type = getSelector("selData");
-        }
-
         if (checked_server === "All SERVERS") {
             rspamd.queryNeighbours("graph", function (req_data) {
                 var neighbours_data = req_data
@@ -257,5 +240,19 @@ function($, D3Evolution, unused) {
         });
     };
 
+    interface.setup = function() {
+        // Handling mouse events on overlapping elements
+        $("#rrd-pie").mouseover(function () {
+            $("#rrd-pie").css("z-index", "200");
+            $("#rrd-table_toggle").css("z-index", "300");
+        });
+        $("#rrd-table_toggle").mouseover(function () {
+            $("#rrd-pie").css("z-index", "0");
+            $("#rrd-table_toggle").css("z-index", "0");
+        });
+
+        return getSelector("selData");
+    };
+
     return interface;
 });