import curses
from typing import Generic, TypeVar

from pqcli.ui.curses.event_handler import EventHandler
from pqcli.ui.curses.util import Choice
from pqcli.ui.curses.views.base_view import BaseView
from pqcli.ui.curses.widgets import Menu

ItemT = TypeVar("ItemT")


class MenuView(BaseView, Generic[ItemT]):
    def __init__(
        self, screen: curses.window, title: str, active_choice: int = 0
    ) -> None:
        super().__init__(screen)

        self.on_cancel = EventHandler[[]]()
        self.on_confirm = EventHandler[[ItemT]]()

        self._title = f"{title}:"
        self._choices: list[Choice] = []
        self._active_choice = active_choice

    def start(self) -> None:
        scr_height, scr_width = self.screen.getmaxyx()
        self.screen.erase()
        self.screen.noutrefresh()

        self.main_menu = Menu(
            header=self._title,
            choices=self._choices,
            active_choice=self._active_choice,
            scr_height=scr_height,
            scr_width=scr_width,
        )
        self.main_menu.render()

    def stop(self) -> None:
        self.main_menu.stop()

    def keypress(self, key: int) -> None:
        self.main_menu.keypress(key)
        self.main_menu.render()