from functools import partial
from tkinter import *

from src.entity.config import Config
from src.entity.session.drawing import DrawingElement
from src.util import Util


class SessionWindow:
    def __init__(self, app):
        self.total_row_time_value_label = None
        self.down_button = None
        self.up_button = None
        self.timer_input = None
        self.number_of_drawings_input = None
        self.selected_element_index = -1
        self.delete_button = None
        self.add_button = None
        self.app = app

        self.window = Toplevel(app.root)
        self.window.title("Custom session")
        self.window.geometry("600x600")
        self.window.protocol("WM_DELETE_WINDOW", self.save_on_closing)

        self.left_column = Frame(self.window, width=300, height=600)
        self.right_column = Frame(self.window, width=300, height=600)

        self.left_column.grid(row=0, column=0, sticky="ns")
        self.right_column.grid(row=0, column=1, sticky="ns")

        self.list_in_session = []

    def open(self):
        list = Config._CONFIG['session']
        if list:
            self.list_in_session = [DrawingElement(e) for e in list]
        self.update_session_list()

    def update_session_list(self):
        self.reset_list()
        for i, session in enumerate(self.list_in_session):
            bg = "blue" if i == self.selected_element_index else "grey85"
            new_session = Button(self.left_column,
                                 text=session.number_of_drawings + " drawings of " + session.timer + "s each",
                                 command=partial(self.edit_element, i),
                                 bg=bg
                                 )
            new_session.grid(row=i, column=0)

        button_frame = Frame(self.left_column)
        self.add_button = Button(button_frame, text="+", command=self.add_element)
        self.add_button.pack(side=LEFT, padx=0, pady=0)

        self.delete_button = Button(button_frame, text="-", command=self.remove_element)
        self.delete_button.pack(side=LEFT, padx=0, pady=0)

        self.up_button = Button(button_frame, text="^", command=self.up_element)
        self.up_button.pack(side=LEFT, padx=0, pady=0)

        self.down_button = Button(button_frame, text="v", command=self.down_element)
        self.down_button.pack(side=LEFT, padx=0, pady=0)

        save_session_button = Button(button_frame, text="v", command=self.save_session)
        save_session_button.pack(side=LEFT, padx=0, pady=0)
        button_frame.grid()

    def reset_element(self):
        for widget in self.right_column.winfo_children():
            widget.destroy()

    def reset_list(self):
        for widget in self.left_column.winfo_children():
            widget.destroy()

    def update_element(self, index=None):
        self.reset_element()
        if index is not None:
            self.selected_element_index = index
            element = self.list_in_session[self.selected_element_index]
            self.update_session_list()
        else:
            element = DrawingElement()

        number_of_drawings_label = Label(self.right_column, text="Number of drawings")
        number_of_drawings_label.pack()

        self.number_of_drawings_input = Entry(self.right_column)
        self.number_of_drawings_input.insert(INSERT, element.number_of_drawings)
        self.number_of_drawings_input.bind("<KeyRelease>", lambda e: self.update_local_timer(e))
        self.number_of_drawings_input.pack()
        self.number_of_drawings_input.focus_set()

        timer_label = Label(self.right_column, text="Time per drawing in seconds")
        timer_label.pack()

        self.timer_input = Entry(self.right_column)
        self.timer_input.insert(INSERT, element.timer)
        self.timer_input.bind("<KeyRelease>", lambda e: self.update_local_timer(e))
        self.timer_input.pack()

        total_row_time_label = Label(self.right_column, text="Total row time")
        total_row_time_label.pack()
        self.total_row_time_value_label = Label(self.right_column, text="")
        self.total_row_time_value_label.pack()
        self.update_local_timer()

        save_button = Button(self.right_column, text="Save", command=self.save)
        save_button.pack()

    def update_local_timer(self, event=None):
        self.total_row_time_value_label.config(
            text=Util.format_seconds(int(self.timer_input.get()) * int(self.number_of_drawings_input.get()))
        )

    def remove_element(self):
        self.list_in_session.pop(self.selected_element_index)
        self.selected_element_index = -1
        self.update_session_list()

    def edit_element(self, i):
        self.update_element(i)

    def add_element(self):
        self.selected_element_index = -1
        self.update_session_list()
        self.update_element()

    def up_element(self):
        self.list_in_session.insert(self.selected_element_index - 1,
                                    self.list_in_session.pop(self.selected_element_index))
        self.selected_element_index -= 1
        self.update_session_list()

    def down_element(self):
        self.list_in_session.insert(self.selected_element_index + 1,
                                    self.list_in_session.pop(self.selected_element_index))
        self.selected_element_index += 1
        self.update_session_list()

    def save(self):
        element = DrawingElement() if self.selected_element_index < 0 \
            else self.list_in_session[self.selected_element_index]
        element.timer = self.timer_input.get()
        element.number_of_drawings = self.number_of_drawings_input.get()

        if self.selected_element_index < 0:
            self.list_in_session.append(element)
        else:
            self.list_in_session[self.selected_element_index] = element

        self.update_session_list()
        self.selected_element_index = -1
        self.reset_element()

    def save_on_closing(self):
        # TODO save
        self.window.destroy()

    def save_session(self):
        Config._CONFIG['session'] = []
        for element in self.list_in_session:
            Config._CONFIG['session'].append(element.__dict__)
        Config.save()
        self.app.set_custom(self.list_in_session)
        self.window.destroy()