From 418dc1be86e2e5c1c6f0a7c2a94263b089f40898 Mon Sep 17 00:00:00 2001
From: Tobi-wan Kenobi <github@tobi-wan-kenobi.at>
Date: Sat, 12 Nov 2016 08:34:21 +0100
Subject: [PATCH] [modules/cmus] Make displayed string configurable

Allow the user to use all tags read by cmus (cmus-remote -Q|grep ^tag)
as part of the displayed data (plus the special 'tags' "duration" and
"position").
---
 bumblebee/modules/cmus.py | 73 ++++++++++++++++++++++-----------------
 bumblebee/util.py         |  7 ++++
 2 files changed, 48 insertions(+), 32 deletions(-)

diff --git a/bumblebee/modules/cmus.py b/bumblebee/modules/cmus.py
index 5a68a9c..70f2b9c 100644
--- a/bumblebee/modules/cmus.py
+++ b/bumblebee/modules/cmus.py
@@ -1,47 +1,56 @@
+import string
 import datetime
-import bumblebee.module
 import subprocess
+from collections import defaultdict
+
+import bumblebee.util
+import bumblebee.module
 
 def description():
     return "Displays the current song and artist playing in cmus"
 
+def parameters():
+    return [
+        "cmus.format: Format of the displayed song information, arbitrary tags (as available from cmus-remote -Q) can be used (defaults to {artist} - {title} {position}/{duration})"
+    ]
+
 class Module(bumblebee.module.Module):
     def __init__(self, output, config, alias):
         super(Module, self).__init__(output, config, alias)
-        self._title = "None"
-        self._artist = "None"
-        self._status = 0
+        self._status = "default"
+        self._fmt = self._config.parameter("format", "{artist} - {title} {position}/{duration}")
+
+    def _loadsong(self):
+        process = subprocess.Popen(["cmus-remote", "-Q"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        self._query, self._error = process.communicate()
+        self._query = self._query.decode("utf-8").split("\n")
+        self._status = "default"
+
+    def _tags(self):
+        tags = defaultdict(lambda: '')
+        for line in self._query:
+            if line.startswith("status"):
+                ignore, status = line.split(" ", 2)
+                self._status = status
+            if line.startswith("tag"):
+                ignore, key, value = line.split(" ", 2)
+                tags.update({ key: value })
+            if line.startswith("duration"):
+                ignore, sec = line.split(" ")
+                tags.update({ "duration": bumblebee.util.durationfmt(int(sec)) })
+            if line.startswith("position"):
+                ignore, sec = line.split(" ")
+                tags.update({ "position": bumblebee.util.durationfmt(int(sec)) })
+
+        return tags
 
     def widgets(self):
-        self._process = subprocess.Popen(["cmus-remote", "-Q"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-        self._query, self._error = self._process.communicate()
-        self._query = self._query.decode("utf-8").split("\n")
-        if b'cmus is not running' in self._error:
-            return bumblebee.output.Widget(self, "-")
-        for line in self._query:
-            if "status playing" in line:
-                self._status = 1
-            if "status paused" in line:
-                self._status = 2
-            if "status stopped" in line:
-                self._status = 3
-            else:
-                if "tag title" in line:
-                    self._title = line[10:]
-                if "tag artist" in line:
-                    self._artist = line[11:]
+        self._loadsong()
+        tags = self._tags()
+
+        return bumblebee.output.Widget(self, string.Formatter().vformat(self._fmt, (), tags))
 
-        return bumblebee.output.Widget(self, "{} - {}".format(
-            self._artist,
-            self._title)
-        )
     def state(self, widget):
-        if self._status == 1:
-            return "playing"
-        if self._status == 2:
-            return "paused"
-        if self._status == 3:
-            return "stopped"
-        return "default"
+        return self._status
 
 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
diff --git a/bumblebee/util.py b/bumblebee/util.py
index 01bd63e..ef74266 100644
--- a/bumblebee/util.py
+++ b/bumblebee/util.py
@@ -6,3 +6,10 @@ def bytefmt(num):
         num /= 1024.0
     return "{:05.2f%}{}GiB".format(num)
 
+def durationfmt(duration):
+    minutes, seconds = divmod(duration, 60)
+    hours, minutes = divmod(minutes, 60)
+    res = "{:02d}:{:02d}".format(minutes, seconds)
+    if hours > 0: res = "{:02d}:{}".format(hours, res)
+
+    return res