mpd.lua~

   1 #!/usr/bin/lua
   2
   3 local hasuci,uci = pcall(require,"uci")
   4
   5 require("socket")
   6 json=require("json")
   7
   8 function url_decode(str)
   9   if not str then return nil end
  10   str = string.gsub (str, "+", " ")
  11   str = string.gsub (str, "%%(%x%x)", function(h) return
  12     string.char(tonumber(h,16)) end)
  13   str = string.gsub (str, "\r\n", "\n")
  14   return str
  15 end
  16
  17 function split(s, delimiter)
  18     local result = {};
  19     for match in (s..delimiter):gmatch("(.-)"..delimiter) do
  20         result[#result+1] = match;
  21     end
  22     return result;
  23 end
  24
  25 function string.starts(String,Start)
  26    return string.sub(String,1,string.len(Start))==Start
  27 end
  28
  29 function process_playlist(playlist)
  30   local res={}
  31   for _,record in pairs(playlist) do
  32     local splitted = split(record,": ")
  33     local name = splitted[2]
  34     local splitted = split(splitted[1],":")
  35     local id = splitted[1]
  36     local rectype = splitted[2]
  37     local rec = {}
  38     if not (id == "OK") then
  39       rec["id"] = id
  40       rec["type"] = rectype
  41       rec["name"] = name
  42       res[id] = rec
  43     end
  44   end
  45   return res
  46 end
  47
  48 function process_playlists(playlists)
  49   local res={}
  50   for _,record in pairs(playlists) do
  51     local splitted = split(record,": ")
  52     if splitted[1]=="playlist" then
  53       res[#res+1] = splitted[2]
  54     end
  55   end
  56   return res
  57 end
  58
  59 function process_directory(directory)
  60   local res={}
  61   for _,record in pairs(directory) do
  62     local splitted = split(record,": ")
  63     if splitted[1]=="directory" or splitted[1]=="file" then
  64       local rec={}
  65       rec["type"] = splitted[1]
  66       rec["name"] = splitted[2]
  67       res[#res+1] = rec
  68     end
  69   end
  70   return res
  71 end
  72
  73 function mpd_new(settings)
  74     local client = {}
  75     if settings == nil then settings = {} end
  76
  77     client.hostname = settings.hostname or "localhost"
  78     client.port     = settings.port or 6600
  79     client.desc     = settings.desc or client.hostname
  80     client.password = settings.password
  81     client.timeout  = settings.timeout or 1
  82     client.retry    = settings.retry or 60
  83
  84     return client
  85 end
  86
  87 function mpd_send(mpd,action,raw)
  88
  89     local command = string.format("%s\n", action)
  90     local values = {}
  91
  92     -- connect to MPD server if not already done.
  93     if not mpd.connected then
  94         local now = os.time();
  95         if not mpd.last_try or (now - mpd.last_try) > mpd.retry then
  96             mpd.socket = socket.tcp()
  97             mpd.socket:settimeout(mpd.timeout, 't')
  98             mpd.last_try = os.time()
  99             mpd.connected = mpd.socket:connect(mpd.hostname, mpd.port)
 100             if not mpd.connected then
 101                 return { errormsg = "could not connect" }
 102             end
 103             mpd.last_error = nil
 104
 105             -- Read the server's hello message
 106             local line = mpd.socket:receive("*l")
 107             if not line:match("^OK MPD") then -- Invalid hello message?
 108                 mpd.connected = false
 109                 return { errormsg = string.format("invalid hello message: %s", line) }
 110             else
 111                 _, _, mpd.version = string.find(line, "^OK MPD ([0-9.]+)")
 112             end
 113
 114             -- send the password if needed
 115             if mpd.password then
 116                 local rsp = mpd_send(mpd,string.format("password %s", mpd.password))
 117                 if rsp.errormsg then
 118                     return rsp
 119                 end
 120             end
 121         else
 122             local retry_sec = mpd.retry - (now - mpd.last_try)
 123             return { errormsg = string.format("%s (retrying in %d sec)", mpd.last_error, retry_sec) }
 124         end
 125     end
 126
 127     mpd.socket:send(command)
 128
 129     local line = ""; err=0
 130     while not line:match("^OK$") do
 131         line, err = mpd.socket:receive("*l")
 132         if not line then -- closed,timeout (mpd killed?)
 133             mpd.last_error = err
 134             mpd.connected = false
 135             mpd.socket:close()
 136             return mpd_send(mpd,action)
 137         end
 138
 139         if line:match("^ACK") then
 140             return { errormsg = line }
 141         end
 142
 143         if not raw then
 144
 145           local pattern = string.format("(%s)", ": ")
 146           local i = string.find (line, pattern, 0)
 147
 148           if i ~= nil then
 149               local key=string.sub(line,1,i-1)
 150               local value=string.sub(line,i+2,-1)
 151               values[string.lower(key)] = value
 152           end
 153
 154         else
 155
 156           values[#values+1]=line
 157
 158         end
 159     end
 160
 161     return values
 162 end
 163
 164 if hasuci then
 165
 166   x = uci.cursor()
 167
 168   settings = {}
 169   settings['host'] = x.get("mpd","server","host") or "localhost"
 170   settings['port'] = x.get("mpd","server","port") or 6600
 171   settings['timeout'] = x.get("mpd","server","timeout") or 1
 172
 173   volstep = x.get("mpd","control","volume_step") or 3
 174
 175   password = x.get("mpd","server","password")
 176
 177 else
 178
 179   config = arg[1]
 180   if not config then
 181     config="/etc/mpd-lua.json"
 182   end
 183
 184   settings={}
 185   local open = io.open
 186   local file = open(config, "r")
 187   if file then
 188     local content = file:read "*a"
 189     file:close()
 190     settings=json.decode(content)
 191   end
 192
 193   settings['host'] = settings['host'] or "localhost"
 194   settings['port'] = settings["port"] or 6600
 195   settings['timeout'] = settings["timeout"] or 1
 196
 197   volstep = settings["volstep"] or 3
 198
 199 end
 200
 201
 202 if password then
 203   settings["password"] = password
 204 end
 205
 206 m = mpd_new(settings)
 207
 208 command = url_decode(os.getenv('QUERY_STRING'))
 209
 210 if not command or command=="" then
 211   command="idle"
 212 end
 213
 214 if command=="play" or command=="pause" or command=="stop" then
 215
 216   res=mpd_send(m,command)
 217
 218 elseif command=="previous" or command=="next" then
 219
 220   res=mpd_send(m,"play")
 221   res=mpd_send(m,command)
 222
 223 elseif command=="idle" then
 224
 225   m.timeout=30
 226   res=mpd_send(m,command)
 227
 228 elseif command=="vold" then
 229
 230   status=mpd_send(m,"status")
 231   volume=tonumber(status["volume"])
 232   res=mpd_send(m,"setvol "..(volume-volstep))
 233
 234 elseif command=="volu" then
 235
 236   status=mpd_send(m,"status")
 237   volume=tonumber(status["volume"])
 238   res=mpd_send(m,"setvol "..(volume+volstep))
 239
 240 elseif string.starts(command,"fastfwd") then
 241
 242   cmd=split(command,"|")
 243   skip=tonumber(cmd[2])
 244   if not skip then
 245     skip=15
 246   end
 247
 248   status=mpd_send(m,"status")
 249   rec_time=status["time"]
 250   song=status["song"]
 251
 252   if song then
 253
 254     if rec_time then
 255       rec_time=split(rec_time,":")
 256       cur_time=tonumber(rec_time[1])
 257
 258       track_time=tonumber(rec_time[2])
 259       cur_time=cur_time+skip
 260       if cur_time>track_time then
 261         cur_time=track_time
 262       end
 263
 264       mpd_send(m,"seek "..song.." "..cur_time)
 265
 266     else
 267
 268       mpd_send(m,"play")
 269
 270     end
 271
 272   end
 273
 274   res={}
 275
 276 elseif string.starts(command,"rewind") then
 277
 278   cmd=split(command,"|")
 279   skip=tonumber(cmd[2])
 280   if not skip then
 281     skip=15
 282   end
 283
 284   status=mpd_send(m,"status")
 285   rec_time=status["time"]
 286   song=status["song"]
 287
 288   if song then
 289
 290     if rec_time then
 291       rec_time=split(rec_time,":")
 292       cur_time=tonumber(rec_time[1])
 293
 294       track_time=tonumber(rec_time[2])
 295       cur_time=cur_time-skip
 296       if cur_time<0 then
 297         cur_time=0
 298       end
 299
 300       mpd_send(m,"seek "..song.." "..cur_time)
 301
 302     else
 303
 304       mpd_send(m,"play")
 305       mpd_send(m,"previous")
 306
 307     end
 308
 309   end
 310
 311   res={}
 312
 313 elseif command=="status" then
 314
 315   res=mpd_send(m,"status")
 316   song=res["song"]
 317   playlist=mpd_send(m,"playlist",1)
 318   pl=process_playlist(playlist)
 319
 320   if song then
 321     res['current_playing']=pl[song]['name']
 322   else
 323     res['song']="--"
 324     res['current_playing']="---"
 325   end
 326
 327 elseif command=="playlist" then
 328
 329   playlist=mpd_send(m,"playlist",1)
 330   res=process_playlist(playlist)
 331
 332 elseif command=="repeat" then
 333
 334   status=mpd_send(m,"status")
 335   rep=1-status["repeat"]
 336   res=mpd_send(m,"repeat "..rep)
 337
 338 elseif string.starts(command,"cpl") then
 339
 340   cmd=split(command,"|")
 341   id=cmd[3]
 342   command=cmd[2]
 343
 344   if command=="playitem" then
 345     command="play "..id
 346     res=mpd_send(m,command)
 347   end
 348
 349   if command=="clear" then
 350     res=mpd_send(m,"clear")
 351   end
 352
 353   if command=="remove" then
 354     command="delete "..id
 355     res=mpd_send(m,command)
 356   end
 357
 358   if command=="moveup" then
 359     command="swap "..id.." "..(id-1)
 360     res=mpd_send(m,command)
 361   end
 362
 363   if command=="movedown" then
 364     command="swap "..id.." "..(id+1)
 365     res=mpd_send(m,command)
 366   end
 367
 368 elseif string.starts(command,"lists") then
 369
 370   cmd=split(command,"|")
 371   command=cmd[2]
 372
 373   if command=="load" then
 374     if not cmd[3] then
 375       lists=mpd_send(m,"listplaylists",1)
 376       res=process_playlists(lists)
 377     else
 378       res=mpd_send(m,"load "..cmd[3],1)
 379     end
 380   end
 381
 382   if command=="save" and cmd[3] then
 383     res=mpd_send(m,"save "..cmd[3],1)
 384   end
 385
 386   if command=="delete" and cmd[3] then
 387     res=mpd_send(m,"rm "..cmd[3],1)
 388   end
 389
 390   if command=="edit" then
 391     if cmd[3] then
 392       dir=mpd_send(m,"lsinfo \""..cmd[3].."\"",1)
 393     else
 394       dir=mpd_send(m,"lsinfo",1)
 395     end
 396     res=process_directory(dir)
 397   end
 398
 399   if command=="add" then
 400     res={}
 401     if cmd[3] then
 402       res=mpd_send(m,"add \""..cmd[3].."\"")
 403     end
 404   end
 405
 406 end
 407
 408 if not res then
 409   print("Content-Type: text/plain\r\n")
 410   print("MPD server - unknown command "..command)
 411 elseif (res['error_msg']) then
 412   print("Content-Type: text/plain\r\n")
 413   print("MPD server connection error: "..res['error_msg'])
 414 else
 415   print "Content-Type: text/plain\r\n"
 416   print(json.encode(res))
 417 end