import mpdprotocol
import responsedict
import urwid
import itertools
from twisted.internet import defer

class SelectableText(urwid.Text):
    def keypress(self, __, key): return key
    def selectable(self): return True

class Playlist(urwid.WidgetWrap):
    def __init__(self, columns, plylst):
        self.playlist = plylst
        self.columns = columns
        self.texts = self.get_texts(self.playlist.as_table(), cols)

        self.titles = urwid.Columns([('weight',w,SelectableText(c)) for w,c,_ in cols],1)
        self.header = urwid.Pile([self.titles, urwid.Divider('-')])
        widget = self.make_playlist()
        widget = urwid.Frame(widget, self.header)
        urwid.WidgetWrap.__init__(self, widget)

    def get_texts(self, choices, columns):
        texts = []
        for weight,col,transform in columns:
            texts.append([(weight,transform(x[0])) for x in choices.get(col)])
        return texts

    def make_playlist(self):
        body = []
        for x in itertools.izip_longest(*self.texts):
            new_widg = [('weight',wgt,SelectableText(it,wrap='clip')) for wgt,it in x]
            new_widg = urwid.Columns(new_widg, 1)
            new_widg = urwid.AttrMap(new_widg, 'unselected', 'selected')
            body.append(new_widg)
        list_walker = urwid.SimpleFocusListWalker(body)
        return urwid.ListBox(urwid.SimpleFocusListWalker(body[:]))

def exit_on_q(key):
    if key in ('q', 'Q'):
        try: raise urwid.ExitMainLoop()
        finally: reactor.stop()

cols = [
    (1,'Pos', lambda x: str(int(x)+1)),
    (4,'Album', lambda x:x),
    (3,'Artist', lambda x:x),
    (5,'Title', lambda x:x)
]

@defer.inlineCallbacks
def get_playlist():
    client = yield mpdprotocol.get_client()
    res = yield client.sendCommand('playlistinfo')
    method, data = res
    loop.widget = Playlist(cols, data)
    loop.draw_screen()

from twisted.internet import reactor
palette = [
    ('selected', 'white', 'black'),
    ('selected', 'black', 'white'),
]
loop = urwid.MainLoop(urwid.SolidFill(), palette, event_loop=urwid.TwistedEventLoop(), unhandled_input=exit_on_q)
reactor.callWhenRunning(get_playlist)
loop.run()