From 8955c9c7f77d5cf4b3411f43f0d1b862396e4e39 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= <raphael@vinot.info>
Date: Tue, 30 Jan 2018 15:41:35 +0100
Subject: [PATCH] chg: Slight refactoring

---
 lookyloo/__init__.py    | 6 +++---
 lookyloo/static/tree.js | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/lookyloo/__init__.py b/lookyloo/__init__.py
index 3d666a92..99057adc 100644
--- a/lookyloo/__init__.py
+++ b/lookyloo/__init__.py
@@ -50,7 +50,7 @@ def load_tree(report_dir):
     pickle.dump(ct, temp)
     temp.close()
     session["tree"] = temp.name
-    return ct.jsonify(), ct.start_time.isoformat(), ct.user_agent, ct.root_url
+    return ct.to_json(), ct.start_time.isoformat(), ct.user_agent, ct.root_url
 
 
 @app.route('/scrape', methods=['GET', 'POST'])
@@ -94,7 +94,7 @@ def hostnode_details(node_uuid):
     hostnode = ct.root_hartree.get_host_node_by_uuid(node_uuid)
     urls = []
     for url in hostnode.urls:
-        urls.append(url.jsonify())
+        urls.append(url.to_json())
     return json.dumps(urls)
 
 
@@ -103,7 +103,7 @@ def urlnode_details(node_uuid):
     with open(session["tree"], 'rb') as f:
         ct = pickle.load(f)
     urlnode = ct.root_hartree.get_url_node_by_uuid(node_uuid)
-    return urlnode.jsonify()
+    return urlnode.to_json()
 
 
 @app.route('/tree/<int:tree_id>', methods=['GET'])
diff --git a/lookyloo/static/tree.js b/lookyloo/static/tree.js
index 5c524c7a..8b73bc08 100644
--- a/lookyloo/static/tree.js
+++ b/lookyloo/static/tree.js
@@ -240,7 +240,7 @@ function update(source) {
   var icons = nodeEnter
         .append('svg')
         .attr('x', 10)
-        .attr('y', 10);
+        .attr('y', 12);
 
   // Add JavaScript information
   var jsContent = icons
@@ -340,7 +340,7 @@ function update(source) {
     }).append('text')
       .attr("dy", 8)
       .attr('x', function(d) { return d.data.total_width ? d.data.total_width + 1 : 0 })
-      .text(function(d) { return d.data.redirect_to_nohing; }).call(getBB);
+      .text(function(d) { return d.data.redirect_to_nothing; }).call(getBB);
 
   // UPDATE
   var nodeUpdate = nodeEnter.merge(node);