From 69420c38cb377abcf5aa793561564f7fe6d83758 Mon Sep 17 00:00:00 2001
From: Sunil Mohan Adapa <sunil@medhas.org>
Date: Fri, 21 Aug 2015 20:05:28 +0530
Subject: [PATCH] privoxy: Implement diagnostics

- Mostly moved from freedombox-setup.
---
 plinth/modules/privoxy/__init__.py            | 38 +++++++++++++++++++
 plinth/modules/privoxy/templates/privoxy.html |  2 +
 2 files changed, 40 insertions(+)

diff --git a/plinth/modules/privoxy/__init__.py b/plinth/modules/privoxy/__init__.py
index 473175166..5eeab52e2 100644
--- a/plinth/modules/privoxy/__init__.py
+++ b/plinth/modules/privoxy/__init__.py
@@ -20,6 +20,7 @@ Plinth module to configure Privoxy.
 """
 
 from gettext import gettext as _
+import json
 
 from plinth import actions
 from plinth import action_utils
@@ -52,3 +53,40 @@ def is_enabled():
 def is_running():
     """Return whether the service is running."""
     return action_utils.service_is_running('privoxy')
+
+
+def diagnose():
+    """Run diagnostics and return the results."""
+    results = []
+
+    results.append(action_utils.diagnose_port_listening(8118, 'tcp4'))
+    results.append(action_utils.diagnose_port_listening(8118, 'tcp6'))
+    results.append(action_utils.diagnose_url('https://www.debian.org'))
+    results.extend(diagnose_url_with_proxy())
+
+    return results
+
+
+def diagnose_url_with_proxy():
+    """Run a diagnostic on a URL with a proxy."""
+    url = 'https://debian.org/'
+
+    results = []
+    for address in action_utils.get_addresses():
+        if address['kind'] == '6':
+            address['address'] = '[{0}]'.format(address['address'])
+
+        proxy = 'http://{host}:8118/'.format(host=address['address'])
+        if address['kind'] == '4':
+            env = {'http_proxy': proxy}
+        else:
+            env = {'https_proxy': proxy}
+
+        result = action_utils.diagnose_url(url, kind=address['kind'], env=env)
+        result[0] = _('Access {url} with proxy {proxy} on tcp{kind}') \
+                    .format(url=url, proxy=proxy, kind=address['kind'])
+        results.append(result)
+
+    return results
+
+
diff --git a/plinth/modules/privoxy/templates/privoxy.html b/plinth/modules/privoxy/templates/privoxy.html
index c8ab5758c..9d36cceaf 100644
--- a/plinth/modules/privoxy/templates/privoxy.html
+++ b/plinth/modules/privoxy/templates/privoxy.html
@@ -44,6 +44,8 @@
   {% else %}
   <span class='running-status inactive'></span> Privoxy is not running
   {% endif %}
+  <a href="{% url 'diagnostics:module' 'privoxy' %}"
+     class="btn btn-default">Run Diagnostics</a>
 </p>
 
 <h3>Configuration</h3>