import asyncio
from operator import itemgetter
import time

from gi.repository import Gtk

from artiq.gui.tools import Window, ListSyncer, DictSyncer
from artiq.protocols.sync_struct import Subscriber


class _ParameterStoreSyncer(DictSyncer):
    def order_key(self, kv_pair):
        return kv_pair[0]

    def convert(self, name, value):
        return [name, str(value)]


class _LastChangesStoreSyncer(ListSyncer):
    def convert(self, x):
        if len(x) == 3:
            timestamp, name, value = x
        else:
            timestamp, name = x
            value = "<deleted>"
        return [time.strftime("%m/%d %H:%M:%S", time.localtime(timestamp)),
                name, str(value)]


class ParametersWindow(Window):
    def __init__(self, **kwargs):
        Window.__init__(self,
                        title="Parameters",
                        default_size=(500, 500),
                        **kwargs)

        notebook = Gtk.Notebook()
        self.add(notebook)

        self.parameters_store = Gtk.ListStore(str, str)
        tree = Gtk.TreeView(self.parameters_store)
        for i, title in enumerate(["Parameter", "Value"]):
            renderer = Gtk.CellRendererText()
            column = Gtk.TreeViewColumn(title, renderer, text=i)
            tree.append_column(column)
        scroll = Gtk.ScrolledWindow()
        scroll.add(tree)
        notebook.insert_page(scroll, Gtk.Label("Current values"), -1)

        self.lastchanges_store = Gtk.ListStore(str, str, str)
        tree = Gtk.TreeView(self.lastchanges_store)
        for i, title in enumerate(["Time", "Parameter", "Value"]):
            renderer = Gtk.CellRendererText()
            column = Gtk.TreeViewColumn(title, renderer, text=i)
            tree.append_column(column)
        scroll = Gtk.ScrolledWindow()
        scroll.add(tree)
        notebook.insert_page(scroll, Gtk.Label("Last changes"), -1)

    @asyncio.coroutine
    def sub_connect(self, host, port):
        self.lastchanges_subscriber = Subscriber(
            "parameters_simplehist", self.init_lastchanges_store)
        yield from self.lastchanges_subscriber.connect(host, port)

    @asyncio.coroutine
    def sub_close(self):
        yield from self.lastchanges_subscriber.close()

    def init_parameters_store(self, init):
        return _ParameterStoreSyncer(self.parameters_store, init)

    def init_lastchanges_store(self, init):
        return _LastChangesStoreSyncer(self.lastchanges_store, init)