import urwid
import yaml
from concurrent.futures import ThreadPoolExecutor
from API import Fetcher


class LeftPane(urwid.ListBox):
    def __init__(self, categories):
        super().__init__(self)
        items = [urwid.AttrMap(urwid.Columns([(5, urwid.Text(str(category["count"]))), urwid.Text(category["name"])]),
                               category["id"], "reveal focus") for category in categories]
        walker = urwid.SimpleListWalker(items)
        self.body = walker
        self.categoryPosition = 0

    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()
            self.categoryPosition = idx
            itemId = focus_widget.attr_map[None]
            feeds = tui.fetcher.feedsFromCategory(itemId)
            feedItems = [urwid.AttrMap(urwid.Columns([(5, urwid.Text(str(tui.fetcher.unreadCounts[feed["id"]][0]))),
                                       urwid.Text(feed["title"])]), feed["id"], "reveal focus") for feed in feeds]
            walker = urwid.SimpleListWalker(feedItems)
            self.body = walker
            tui.leftBox.set_title(itemId[13:])
            return
        elif key in ("h", "left"):
            items = [urwid.AttrMap(urwid.Columns([(5, urwid.Text(str(category["count"]))), urwid.Text(category["name"])]),
                                   category["id"], "reveal focus") for category in tui.fetcher.categories]
            walker = urwid.SimpleFocusListWalker(items)
            self.body = walker
            tui.leftBox.set_title("Categories")
            tui.feeds.set_focus(self.categoryPosition)
            return

        return super().keypress(size, key)


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):

        with open('config.yml', 'r') as file:
            config = yaml.safe_load(file)
            URL = config["server"]["URL"]
            token = config["server"]["token"]

        self.fetcher = Fetcher(URL, token)

        self.leftPaneItems = {}
        self.activePane = False
        self.executor = ThreadPoolExecutor(max_workers=1)
        self.loop = None
        self.articleItems = [urwid.Text("Article# " + str(i)) for i in range(0, 10)]
        self.feeds = LeftPane(self.fetcher.categories)
        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()