From b0608c7054d6b6c5a5bf7beecff4493ea1ef2541 Mon Sep 17 00:00:00 2001
From: Tobias Witek <tobi@tobi-wan-kenobi.at>
Date: Mon, 31 Oct 2016 14:27:51 +0100
Subject: [PATCH] [modules] Add pulseaudio control modules

Add a module that retrieve mute status and volume (left, right, mono)
from pulseaudio. Unfortunately, this module is really, really hacky. It
invokes "pactl" multiple times to get the status and does some ugly
parsing on the output.

Overall, this is pretty brittle and prone to failure, but as I was not
able to find a decent pulseaudio library for Python... Probably, I
haven't searched hard enough, cannot believe such a thing does not
exist.
---
 bumblebee/modules/pasink.py               |  1 +
 bumblebee/modules/pasource.py             |  1 +
 bumblebee/modules/pulseaudio.py           | 70 +++++++++++++++++++++++
 bumblebee/modules/time.py                 |  3 +-
 bumblebee/themes/solarized-powerline.json | 20 +++++++
 5 files changed, 94 insertions(+), 1 deletion(-)
 create mode 120000 bumblebee/modules/pasink.py
 create mode 120000 bumblebee/modules/pasource.py
 create mode 100644 bumblebee/modules/pulseaudio.py

diff --git a/bumblebee/modules/pasink.py b/bumblebee/modules/pasink.py
new file mode 120000
index 0000000..85ea20d
--- /dev/null
+++ b/bumblebee/modules/pasink.py
@@ -0,0 +1 @@
+pulseaudio.py
\ No newline at end of file
diff --git a/bumblebee/modules/pasource.py b/bumblebee/modules/pasource.py
new file mode 120000
index 0000000..85ea20d
--- /dev/null
+++ b/bumblebee/modules/pasource.py
@@ -0,0 +1 @@
+pulseaudio.py
\ No newline at end of file
diff --git a/bumblebee/modules/pulseaudio.py b/bumblebee/modules/pulseaudio.py
new file mode 100644
index 0000000..bdcc173
--- /dev/null
+++ b/bumblebee/modules/pulseaudio.py
@@ -0,0 +1,70 @@
+import re
+import shlex
+import subprocess
+
+import bumblebee.module
+import bumblebee.util
+
+class Module(bumblebee.module.Module):
+    def __init__(self, args):
+        super(Module, self).__init__(args)
+
+        self._module = self.__module__.split(".")[-1]
+        self._left = 0
+        self._right = 0
+        self._mono = 0
+        self._mute = False
+
+    def data(self):
+        res = subprocess.check_output(shlex.split("pactl info"))
+        channel = "sinks" if self._module == "pasink" else "sources"
+        name = None
+        for line in res.split("\n"):
+            if line.startswith("Default Sink: ") and channel == "sinks":
+                name = line[14:]
+            if line.startswith("Default Source: ") and channel == "sources":
+                name = line[16:]
+        
+        res = subprocess.check_output(shlex.split("pactl list {}".format(channel)))
+
+        found = False
+        for line in res.split("\n"):
+            if "Name:" in line and found == True:
+                break
+            if name in line:
+                found = True
+            if "Mute:" in line and found == True:
+                self._mute = False if " no" in line.lower() else True
+
+            if "Volume:" in line and found == True:
+                m = None
+                if "mono" in line:
+                    m = re.search(r'mono:.*\s*\/\s*(\d+)%', line)
+                else:
+                    m = re.search(r'left:.*\s*\/\s*(\d+)%.*right:.*\s*\/\s*(\d+)%', line)
+                if not m: continue
+
+                if "mono" in line:
+                    self._mono = m.group(1)
+                else:
+                    self._left = m.group(1)
+                    self._right = m.group(2)
+        result = ""
+        if self._mono > 0:
+            result = "{}%".format(self._mono)
+        elif self._left == self._right:
+            result = "{}%".format(self._left)
+        else:
+            result="{}%/{}%".format(self._left, self._right)
+        return result
+
+    def state(self):
+        return "muted" if self._mute is True else "unmuted"
+
+    def warning(self):
+        return self._mute
+
+    def critical(self):
+        return False
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
diff --git a/bumblebee/modules/time.py b/bumblebee/modules/time.py
index 9076a94..5254506 100644
--- a/bumblebee/modules/time.py
+++ b/bumblebee/modules/time.py
@@ -3,12 +3,13 @@ import bumblebee.module
 
 class Module(bumblebee.module.Module):
     def __init__(self, args):
+        super(Module, self).__init__(args)
+
         module = self.__module__.split(".")[-1]
 
         default = "%x" if module == "date" else "%X"
         self._fmt = args[0] if args else default
 
-        super(Module, self).__init__(args)
 
     def data(self):
         return datetime.datetime.now().strftime(self._fmt)
diff --git a/bumblebee/themes/solarized-powerline.json b/bumblebee/themes/solarized-powerline.json
index aa5ca06..0a2ffbe 100644
--- a/bumblebee/themes/solarized-powerline.json
+++ b/bumblebee/themes/solarized-powerline.json
@@ -35,6 +35,26 @@
 	"disk": {
 		"prefix": "  "
 	},
+	"pasink": {
+		"states": {
+			"muted": {
+				"prefix": "  "
+			},
+			"unmuted": {
+				"prefix": "  "
+			}
+		}
+	},
+	"pasource": {
+		"states": {
+			"muted": {
+				"prefix": "  "
+			},
+			"unmuted": {
+				"prefix": "  "
+			}
+		}
+	},
 	"nic": {
 		"states": {
 			"wireless-up": {