from typing import Tuple

from ..base import View
from ..button_interface import ButtonInterface

from .recipe import V60, ESPRESSO

from PIL import ImageDraw, Image

class RecipeSelection(View, ButtonInterface):

    recipes = [
        V60,
        ESPRESSO
    ]

    def __init__(self, parent, im_size, center, deactivate_command=None):
        self.selected_index = 0
        self.deactivate_command = deactivate_command
        super().__init__(parent, im_size, center)

    def _get_visual_recipes(self):
        recipes = self.recipes + ['+', 'BACK']
        if len(recipes) < 5:
            return recipes, 0
        
        start = max(0, self.selected_index - 2)
        end = min(len(recipes), start + 5)

        recipes = recipes[start:end]

        return recipes, start

    def update_weight(self, weight: float) -> Image.Image:
        im = self.bkg_im.copy()
        draw = ImageDraw.Draw(im)

        recipes, start = self._get_visual_recipes()
        for idx, recipe in enumerate(recipes):
            if idx + start == self.selected_index:
                r = 10
                offset = 15
                for i in range(0, 90, r // 2):
                    draw.circle((40 + i, offset + idx * 20), r, fill='black')
                if str(recipe) != 'BACK':
                    draw.text((40, 10 + idx * 20), str(recipe), fill='white')
                else:
                    draw.regular_polygon((40 + 5, 15 + idx * 20, 5), 3, fill='white', rotation=90)
            elif str(recipe) == 'BACK':
                    draw.regular_polygon((40 + 5, 15 + idx * 20, 5), 3, fill='black', rotation=90)
            else:
                draw.text((40, 10 + idx * 20), str(recipe), fill='black')

        
        return im

    def left_press(self):
        self.selected_index = (self.selected_index - 1) % (len(self.recipes) + 2)

    def right_press(self):
        self.selected_index = (self.selected_index + 1) % (len(self.recipes) + 2)

    def right_long_press(self):
        if self.selected_index < len(self.recipes):
            # activate selected recipe
            print(f"Activating recipe: {self.recipes[self.selected_index]}")
        elif self.selected_index == len(self.recipes):
            # add new recipe
            print("Adding new recipe")
        else:
            self.selected_index = 0
            self.deactivate_command()


    def has_button(self) -> Tuple[bool, bool, bool, bool]:
        return True, True, True, True

    def render_left_press(self, draw, x, y):
        draw.regular_polygon((x, y+2, 5), 3, fill='black')

    def render_left_long_press(self, draw, x, y):
        draw.text((x, y-5), 'Edit', fill='black')

    def render_right_press(self, draw, x, y):
        draw.regular_polygon((x, y+4, 5), 3, fill='black', rotation=180)

    def render_right_long_press(self, draw, x, y):
        draw.regular_polygon((x+2, y, 5), 3, fill='black', rotation=270)