import io
import tkinter as tk
from tkinter import Frame, Canvas, ttk, PhotoImage
from enum import Enum

from PIL import ImageChops

from ..config import DISPLAY_TYPES, DISPLAY_MODES
from . import NumberView, CircleView, TimerView
from .button_interface import ButtonInterface
from .buttons_manager import ButtonsManager
from .recipe_selection import RecipeSelection
from .draw_utils import draw_clock

class MainView(tk.Frame, ButtonInterface):
    def __init__(self, parent,
                 tare_command=None,
                 calibrate_command=None,
                 **kwargs):
        super().__init__(parent, **kwargs)
        self.views = []
        self.timer_view = None  # Timer view is always active
        self.tare_command = tare_command
        self.calibrate_command = calibrate_command
        
        self.actions = Frame(self)
        self.actions.pack()
        self.calibrate_button = ttk.Button(self.actions, text="Calibrate", command=calibrate_command)
        self.calibrate_button.pack()

        self.im_size = (168, 144)
        self.center = (168 // 2, 144 // 2)

        # Create timer view that's always active
        self.canvas = Canvas(self, width=168, height=144, background='white',
                             highlightthickness=1, highlightbackground="black")
        self.canvas.pack()


        self.timer_view = TimerView(self.actions, self.im_size, self.center)
        self.recipe_selection = RecipeSelection(self, self.im_size, self.center)
        self.buttons = ButtonsManager(self, self.im_size, self.center,
                                      default=self,
                                      select_recipe=self.recipe_selection)
        
        self.curr_mode = DISPLAY_MODES.MAIN
    
    def has_button(self):
        return True, True, True, True

    def left_press(self):
        self.timer_view.toggle_timer()
    
    def left_long_press(self):
        self.timer_view.reset_timer()
    
    def right_press(self):
        self.tare_command()
    
    def right_long_press(self):
        self.enter_recipe_selection()


    def render_left_press(self, draw, x, y):
        draw_clock(draw, (x, y), radius=3)

    def render_left_long_press(self, draw, x, y):
        draw_clock(draw, (x, y), radius=3)
        draw.text((x + 6, y - 5), "R", fill='black')

    def render_right_press(self, draw, x, y):
        draw.text((x, y), "T", fill='black')

    def render_right_long_press(self, draw, x, y):
        draw.text((x, y - 5), "R", fill='black')



    def enter_recipe_selection(self):
        self.curr_mode = DISPLAY_MODES.RECIPE_SELECTION
        self.buttons.current_view = self.recipe_selection
        self.refresh(0.0)


    ################ VIEW MANAGEMENT ################

    def update_views(self, selected_types: DISPLAY_TYPES):
        # Clear only the selectable views, not the timer
        for v in self.views:
            if v.ui is not None:
                v.ui.destroy()
        self.views.clear()

        if selected_types & DISPLAY_TYPES.NUMBER:
            number_view = NumberView(self.actions, self.im_size, self.center)
            self.views.append(number_view)

        if selected_types & DISPLAY_TYPES.CIRCLE:
            circle_view = CircleView(self.actions, self.im_size, self.center)
            self.views.append(circle_view)


    def refresh(self, weight: float):
        ims = []

        if self.curr_mode == DISPLAY_MODES.MAIN:
            # Always include timer and button view
            if self.timer_view:
                timer_im = self.timer_view.update_weight(weight)
                button_im = self.buttons.update_weight(weight)
                ims.append(timer_im)
                ims.append(button_im)
            
            # Add other selected views
            for view in self.views:
                im = view.update_weight(weight)
                ims.append(im)
        
        elif self.curr_mode == DISPLAY_MODES.RECIPE_SELECTION:
            button_im = self.buttons.update_weight(weight)
            ims.append(button_im)
            recipe_im = self.recipe_selection.update_weight(weight)
            ims.append(recipe_im)

            
        self.canvas.delete("all")
        # Combine images by logical_and
        if ims:
            combined_im = ims[0]
            for im in ims[1:]:
                combined_im = ImageChops.invert(ImageChops.logical_xor(combined_im, im))
            
            # Convert PIL image to bytes
            buffer = io.BytesIO()
            combined_im.save(buffer, format='PNG')
            buffer.seek(0)
            
            # Load into PhotoImage and display on canvas
            self.photo = PhotoImage(data=buffer.getvalue())
            self.canvas.create_image(0, 0, anchor="nw", image=self.photo)