import urwid
from concurrent.futures import ThreadPoolExecutor


class Items(urwid.ListBox):
    def __init__(self, items):
        super().__init__(self)
        walker = urwid.SimpleListWalker([urwid.AttrMap(item, None, 'reveal focus') for item in items])
        self.body = walker

    def keypress(self, size, key):
        if key in ("j", "down"):
            item_size = len(self.body)
            focus_widget, idx = self.get_focus()
            if idx < item_size - 1:
                idx = idx + 1
                self.set_focus(idx)
            return
        elif key in ("k", "up"):
            focus_widget, idx = self.get_focus()
            if idx > 0:
                idx = idx - 1
                self.set_focus(idx)
            return
        elif key in ("l", "right"):
            focus_widget, idx = self.get_focus()
            value = next(focus_widget.render((100,), focus=False).content())[1][2].decode().strip()
            feeds = tui.leftPaneItems[value]["feeds"]
            feedItems = [urwid.Columns([(4, urwid.Text(item["count"])), urwid.Text(item["name"])])
                         for item in feeds]
            walker = urwid.SimpleListWalker([urwid.AttrMap(item, None, 'reveal focus') for item in feedItems])
            self.body = walker
            tui.leftBox.set_title(value)
            return
        elif key in ("h", "left"):
            feedItems = tui.feedItems
            walker = urwid.SimpleListWalker([urwid.AttrMap(item, None, 'reveal focus') for item in feedItems])
            self.body = walker
            tui.leftBox.set_title("Categories")
            return

        return super().keypress(size, key)


class TUI(urwid.Frame):

    @classmethod
    def create(cls):

        tui = cls()
        palette = [('header', 'white', 'black'), ('reveal focus', 'black', 'dark cyan', 'standout')]
        loop = urwid.MainLoop(
            tui,
            palette,
            event_loop=urwid.AsyncioEventLoop(),
            unhandled_input=tui.unhandled_input,
        )
        tui.loop = loop

        return tui

    def __init__(self):

        self.leftPaneItems = {}
        self.activePane = False
        for i in range(0, 15):
            self.leftPaneItems["Category# " + str(i)] = {"count": str(i),
                                                         "feeds": [{"count": str(k), "name": "Feed# " + str(k)} for k in range(0, 10)]}

        self.executor = ThreadPoolExecutor(max_workers=1)
        self.loop = None
        self.feedItems = [urwid.Columns([(4, urwid.Text(self.leftPaneItems[item]["count"])), urwid.Text(item)])
                          for item in self.leftPaneItems.keys()]
        self.articleItems = [urwid.Text("Article# " + str(i)) for i in range(0, 10)]
        self.feeds = Items(self.feedItems)
        self.articles = Items(self.articleItems)
        self.leftBox = urwid.LineBox(self.feeds, title="Categories")
        self.container = urwid.Columns(
            [(40, self.leftBox),
             urwid.LineBox(self.articles, title="Articles")])
        self.body = self.container

        super().__init__(self.body)

    def run(self):
        self.loop.run()
        self.executor.shutdown(wait=False)

    def unhandled_input(self, key):
        if key == "tab":
            self.activePane = not self.activePane
            self.container.set_focus(int(self.activePane))
        elif key == "q":
            raise urwid.ExitMainLoop()


tui = TUI.create()
tui.run()