From 52b4bfac35e01b09b92bc6b6f9e1b985f23b9a7f Mon Sep 17 00:00:00 2001
From: Antonin Dach <dach@protonmail.com>
Date: Sat, 12 May 2018 19:37:19 +0200
Subject: [PATCH] [module] - adding caps/num/scrl indicator widget

---
 README.md                      |  2 +-
 bumblebee/modules/indicator.py | 51 ++++++++++++++++++++++++++++++++++
 2 files changed, 52 insertions(+), 1 deletion(-)
 create mode 100644 bumblebee/modules/indicator.py

diff --git a/README.md b/README.md
index b016f2a..53102ce 100644
--- a/README.md
+++ b/README.md
@@ -5,7 +5,7 @@
 [![Test Coverage](https://codeclimate.com/github/tobi-wan-kenobi/bumblebee-status/badges/coverage.svg)](https://codeclimate.com/github/tobi-wan-kenobi/bumblebee-status/coverage)
 [![Issue Count](https://codeclimate.com/github/tobi-wan-kenobi/bumblebee-status/badges/issue_count.svg)](https://codeclimate.com/github/tobi-wan-kenobi/bumblebee-status)
 
-**Many, many thanks to all contributors! As of now, 26 of the modules are from various contributors (!), and only 16 from myself.**
+**Many, many thanks to all contributors! As of now, 27 of the modules are from various contributors (!), and only 16 from myself.**
 
 ![Solarized Powerline](https://github.com/tobi-wan-kenobi/bumblebee-status/blob/master/screenshots/themes/powerline-solarized.png)
 
diff --git a/bumblebee/modules/indicator.py b/bumblebee/modules/indicator.py
new file mode 100644
index 0000000..a4d1712
--- /dev/null
+++ b/bumblebee/modules/indicator.py
@@ -0,0 +1,51 @@
+#pylint: disable=C0111,R0903
+
+"""Displays the indicator status, for numlock, scrolllock and capslock 
+
+Parameters:
+    * indicator.include: Comma-separated list of interface prefixes to include (defaults to "numlock,capslock")
+    * indicator.signalstype: If you want the signali type color to be "critical" or "warning" (defaults to "warning")
+"""
+
+
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+
+class Module(bumblebee.engine.Module):
+    def __init__(self, engine, config):
+        widgets = []
+        self.status = False
+        super(Module,self).__init__(engine, config, widgets)
+        self._include = tuple(filter(len, self.parameter("include", "NumLock,CapsLock").split(",")))
+        self._signalType = self.parameter("signaltype") if not self.parameter("signaltype") == None else "warning"
+
+    def update(self, widgets):
+        self._update_widgets(widgets)
+
+    def state(self, widget):
+        states = []
+        if widget.status:
+            states.append(self._signalType)
+        elif not widget.status:
+            states.append("normal")
+        return states
+
+    def _update_widgets(self, widgets):
+        status_line = "" 
+        for line in bumblebee.util.execute("xset q").replace(" ", "").split("\n"):
+            if "capslock" in line.lower():
+                status_line = line  
+                break
+            
+        for indicator in self._include:
+            widget = self.widget(indicator)
+            if not widget:
+                widget = bumblebee.output.Widget(name=indicator)
+                widgets.append(widget)
+
+            widget.status = True if indicator.lower()+":on" in status_line.lower() else False
+            widget.full_text(indicator)
+
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4