X-Git-Url: https://git.rmz.io/dotfiles.git/blobdiff_plain/4264968ad117b6d08217808f72d32032a3f6b253..b76f00cb193eaba72d7515efda5098a724c05620:/awesome/widgets/mpc.lua?ds=sidebyside

diff --git a/awesome/widgets/mpc.lua b/awesome/widgets/mpc.lua
index f357f18..932a4a5 100644
--- a/awesome/widgets/mpc.lua
+++ b/awesome/widgets/mpc.lua
@@ -4,6 +4,8 @@ local lgi = require "lgi"
 local GLib = lgi.GLib
 local Gio = lgi.Gio
 
+local trace_mpc = false
+
 local mpc = {}
 
 local function parse_password(host)
@@ -28,12 +30,19 @@ function mpc.new(host, port, password, error_handler, ...)
         _error_handler = error_handler or function() end,
         _connected = false,
         _try_reconnect = false,
-        _idle_commands = { ... }
+        _idle_subsystems = { ... }
     }, { __index = mpc })
     self:_connect()
     return self
 end
 
+function mpc:_write(command)
+    if trace_mpc then
+        print("write: " .. command)
+    end
+    self._output:write(command .. "\n")
+end
+
 function mpc:_error(err)
     self._connected = false
     self._error_handler(err)
@@ -95,7 +104,6 @@ function mpc:_connect()
                 self._output, self._input = nil, nil
                 self:_error(err)
             else
-                do_read()
                 line = tostring(line)
                 if line == "OK" or line:match("^ACK ") then
                     local success = line == "OK"
@@ -109,53 +117,63 @@ function mpc:_connect()
                     table.remove(self._reply_handlers, 1)
                     self._pending_reply = {}
                     handler(success, arg)
+
+                    if next(self._reply_handlers) == nil then
+                        self:_start_idle()
+                    end
                 else
                     local _, _, key, value = string.find(line, "([^:]+):%s(.+)")
                     if key then
-                        self._pending_reply[string.lower(key)] = value
+                        local k = string.lower(key)
+                        if k == "binary" then
+                            value = tonumber(value)
+                            local data = {}
+                            while value > 0 do
+                                local b = assert(obj:read_bytes(value))
+                                table.insert(data, b.data)
+                                value = value - #b
+                            end
+                            local w = obj:read_bytes(1)  -- read newline at end of binary
+                            self._pending_reply[k] = table.concat(data)
+                        elseif k == "changed" then
+                            if not self._pending_reply[k] then
+                                self._pending_reply[k] = {}
+                            end
+
+                            self._pending_reply[k][value] = true
+                        else
+                            self._pending_reply[k] = value
+                        end
                     end
                 end
+                do_read()
             end
         end)
     end
     do_read()
 
-    -- To synchronize the state on startup, send the idle commands now. As a
-    -- side effect, this will enable idle state.
-    self:_send_idle_commands(true)
-
-    return self
-end
-
-function mpc:_send_idle_commands(skip_stop_idle)
-    -- We use a ping to unset this to make sure we never get into a busy
-    -- loop sending idle / unidle commands. Next call to
-    -- _send_idle_commands() might be ignored!
-    if self._idle_commands_pending then
-        return
-    end
-    if not skip_stop_idle then
-        self:_stop_idle()
+    -- To synchronize the state on startup, send the idle commands now.
+    for i = 1, #self._idle_subsystems, 2 do
+        self._idle_subsystems[i+1](self)
     end
 
-    self._idle_commands_pending = true
-    for i = 1, #self._idle_commands, 2 do
-        self:_send(self._idle_commands[i], self._idle_commands[i+1])
-    end
-    self:_send("ping", function()
-        self._idle_commands_pending = false
-    end)
-    self:_start_idle()
+    return self
 end
 
 function mpc:_start_idle()
     if self._idle then
-        error("Still idle?!")
+        error("start_idle but already idle")
     end
     self:_send("idle", function(success, reply)
+        self._idle = false
         if reply.changed then
             -- idle mode was disabled by mpd
-            self:_send_idle_commands()
+            for i = 1, #self._idle_subsystems, 2 do
+                local subsys = self._idle_subsystems[i]
+                if reply.changed[subsys] then
+                    self._idle_subsystems[i+1](self)
+                end
+            end
         end
     end)
     self._idle = true
@@ -163,9 +181,9 @@ end
 
 function mpc:_stop_idle()
     if not self._idle then
-        error("Not idle?!")
+        error("stop_idle but not idle")
     end
-    self._output:write("noidle\n")
+    self:_write("noidle")
     self._idle = false
 end
 
@@ -173,7 +191,7 @@ function mpc:_send(command, callback)
     if self._idle then
         error("Still idle in send()?!")
     end
-    self._output:write(command .. "\n")
+    self:_write(command)
     table.insert(self._reply_handlers, callback or function() end)
 end
 
@@ -182,15 +200,13 @@ function mpc:send(...)
     if not self._connected then
         return
     end
-    local args = { ... }
-    if not self._idle then
-        error("Something is messed up, we should be idle here...")
+    if self._idle then
+        self:_stop_idle()
     end
-    self:_stop_idle()
+    local args = { ... }
     for i = 1, #args, 2 do
         self:_send(args[i], args[i+1])
     end
-    self:_start_idle()
 end
 
 function mpc:toggle_play()
@@ -214,6 +230,39 @@ function mpc:change_volume(change)
     end)
 end
 
+function mpc:currentsong()
+    local currentsong
+    self:send("currentsong", function(err, song)
+        if err then error(err) end
+        currentsong = song
+    end)
+    return currentsong
+end
+
+local function escape(str)
+    return "\"" .. str .. "\""
+end
+
+function mpc:albumart(uri, handler)
+    local image_table = {}
+    local get_art_at
+    get_art_at = function(off)
+        self:send("albumart " .. escape(uri) .. " " .. tostring(off), function(success, data)
+            if not success then
+                handler(success, data)
+            end
+            table.insert(image_table, data.binary)
+            if data.binary and #data.binary > 0 then
+                get_art_at(off + #data.binary)
+            else
+                data.binary = table.concat(image_table)
+                handler(success, data)
+            end
+        end)
+    end
+    get_art_at(0)
+end
+
 --[[
 
 -- Example on how to use this (standalone)