From c7f979a735d0b8834125a007ecb9d2d59043834a Mon Sep 17 00:00:00 2001
From: Bernhard B <schluchti@gmail.com>
Date: Sun, 12 May 2019 20:45:08 +0200
Subject: [PATCH] added new pi-hole module

* allows to enable/disable pi-hole
* shows today's blocked ads
---
 bumblebee/modules/pihole.py     | 68 +++++++++++++++++++++++++++++++++
 themes/icons/awesome-fonts.json |  4 ++
 2 files changed, 72 insertions(+)
 create mode 100644 bumblebee/modules/pihole.py

diff --git a/bumblebee/modules/pihole.py b/bumblebee/modules/pihole.py
new file mode 100644
index 0000000..3ec6c9a
--- /dev/null
+++ b/bumblebee/modules/pihole.py
@@ -0,0 +1,68 @@
+# pylint: disable=C0111,R0903
+
+"""Displays the pi-hole status (up/down) together with the number of ads that were blocked today
+Parameters:
+    * pihole.address     : pi-hole address (e.q: http://192.168.1.3)
+    * pihole.pwhash      : pi-hole webinterface password hash (can be obtained from the /etc/pihole/SetupVars.conf file)
+"""
+
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+import requests
+
+class Module(bumblebee.engine.Module):
+    def __init__(self, engine, config):
+        super(Module, self).__init__(engine, config,
+            bumblebee.output.Widget(full_text=self.pihole_status)
+        )
+
+        buttons = {"LEFT_CLICK":bumblebee.input.LEFT_MOUSE}
+        self._pihole_address = self.parameter("address", "")
+
+        self._pihole_pw_hash = self.parameter("pwhash", "")
+        self._pihole_status = None
+        self._ads_blocked_today = "-"
+        self.update_pihole_status()
+
+        engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
+                                       cmd=self.toggle_pihole_status)
+
+    def pihole_status(self, widget):
+        if self._pihole_status is None:
+            return "unknown"
+        return "pi-hole " + ("up/" + self._ads_blocked_today if self._pihole_status else "down")
+
+    def update_pihole_status(self):
+        try:
+            data = requests.get(self._pihole_address + "/admin/api.php?summary").json()
+            self._pihole_status = True if data["status"] == "enabled" else False
+            self._ads_blocked_today = data["ads_blocked_today"]
+        except:
+            self._pihole_status = None
+
+    def toggle_pihole_status(self, widget):
+        if self._pihole_status is not None:
+            try:
+                req = None
+                if self._pihole_status:
+                    req = requests.get(self._pihole_address + "/admin/api.php?disable&auth=" + self._pihole_pw_hash)
+                else:
+                    req = requests.get(self._pihole_address + "/admin/api.php?enable&auth=" + self._pihole_pw_hash)
+                if req is not None:
+                    if req.status_code == 200:
+                        status = req.json()["status"]
+                        self._pihole_status = False if status == "disabled" else True
+            except:
+                pass
+
+
+    def update(self, widgets):
+        self.update_pihole_status()
+
+    def state(self, widget):
+        if self._pihole_status:
+            return ["enabled"]
+        elif not self._pihole_status:
+            return ["disabled", "warning"]
+        return []
diff --git a/themes/icons/awesome-fonts.json b/themes/icons/awesome-fonts.json
index 5513466..daa370c 100644
--- a/themes/icons/awesome-fonts.json
+++ b/themes/icons/awesome-fonts.json
@@ -192,5 +192,9 @@
     "dunst": {
         "muted": { "prefix": ""},
         "unmuted": { "prefix": "" }
+    },
+    "pihole": { 
+      "enabled": { "prefix": "" },
+      "disabled": { "prefix": "" }
     }
 }