From 3b2b9bfc0cdcad0918e50ee6e0b9278a14e12448 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Beno=C3=AEt=20Dardenne?= <benoit.dardenne@gmail.com>
Date: Sat, 11 Apr 2020 11:48:05 +0200
Subject: [PATCH] Added module watson

---
 bumblebee/modules/watson.py     | 57 +++++++++++++++++++++++++++++++++
 themes/icons/awesome-fonts.json |  3 +-
 themes/icons/ionicons.json      |  4 +--
 3 files changed, 61 insertions(+), 3 deletions(-)
 create mode 100644 bumblebee/modules/watson.py

diff --git a/bumblebee/modules/watson.py b/bumblebee/modules/watson.py
new file mode 100644
index 0000000..a359bf4
--- /dev/null
+++ b/bumblebee/modules/watson.py
@@ -0,0 +1,57 @@
+# pylint: disable=C0111,R0903
+
+"""Displays the status of watson (time-tracking tool)
+
+Requires the following executable:
+    * watson
+"""
+
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+import bumblebee.util
+import bumblebee.popup_v2
+
+import logging
+import re
+import functools
+
+class Module(bumblebee.engine.Module):
+    def __init__(self, engine, config):
+        super(Module, self).__init__(engine, config,
+            bumblebee.output.Widget(full_text=self.text))
+        self._tracking = False
+        self._project = ""
+        engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
+                                        cmd=self.toggle)
+
+    def toggle(self, widget):
+        self._project = "hit"
+        if self._tracking:
+            bumblebee.util.execute("watson stop")
+        else:
+            bumblebee.util.execute("watson restart")
+        self._tracking = not self._tracking
+
+    def text(self, widget):
+        if self._tracking:
+            return self._project
+        else:
+            return "Paused"
+
+    def update(self, widgets):
+        output = bumblebee.util.execute("watson status")
+        if re.match('No project started', output):
+            self._tracking = False
+            return
+
+        self._tracking = True
+        m = re.search(r'Project (.+) started', output)
+        self._project = m.group(1)
+
+    #
+    def state(self, widget):
+        return "on" if self._tracking else "off"
+    #     return [widget.get("status", None), widget.get("period", None)]
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
diff --git a/themes/icons/awesome-fonts.json b/themes/icons/awesome-fonts.json
index 8c6688c..929d742 100644
--- a/themes/icons/awesome-fonts.json
+++ b/themes/icons/awesome-fonts.json
@@ -260,5 +260,6 @@
       "work": { "prefix": "" },
       "break": { "prefix": "" }
   },
-  "hddtemp": { "prefix": "" }
+  "hddtemp": { "prefix": "" },
+  "watson": { "prefix": "" }
 }
diff --git a/themes/icons/ionicons.json b/themes/icons/ionicons.json
index cf7813a..170b07c 100644
--- a/themes/icons/ionicons.json
+++ b/themes/icons/ionicons.json
@@ -193,6 +193,6 @@
       "off": { "prefix": "\uf24f" },
       "paused": { "prefix": "\uf210" },
       "on": { "prefix": "\uf488" }
-  }
-
+  },
+  "watson": { "prefix": "\uf2d1" }
 }