squirrel-battle/dungeonbattle/display/menudisplay.py

from dungeonbattle.menus import Menu
from typing import Any
import curses


class MenuDisplay:
    def __init__(self, menu: Menu, screen: Any, height: int, width: int,
                 topleftx: int, toplefty: int):
        self.screen = screen
        self.position = menu.position
        self.values = menu.values
        self.width = width
        self.height = height
        self.trueheight = len(menu.values)
        self.truewidth = max(map(len, self.values))
        self.topleftx = topleftx
        self.toplefty = toplefty

        # Menu values are printed in pad
        self.pad = curses.newpad(self.trueheight, self.truewidth + 1)
        for i in range(self.trueheight - 1):
            self.pad.addstr(i, 0, " " + self.values[i])

        # Menu box
        self.menubox = curses.newpad(self.height, self.width)
        self.menubox.addstr(0, 0, "┏" + "━" * (self.width - 2) + "┓")
        for i in range(1, self.height - 2):
            self.menubox.addstr(i, 0, "┃" + " " * (self.width - 2) + "┃")
        self.menubox.addstr(self.height - 2, 0,
                            "┗" + "━" * (self.width - 2) + "┛")

    def update_pad(self, position: int) -> None:
        for i in range(self.trueheight):
            self.pad.addstr(i, 0, " ")
        # set a marker in front of the selected line
        self.pad.addstr(position, 0, ">")

    def refresh(self, position: int) -> None:
        if self.height - 2 >= position - 1:
            cornery = 0
        elif self.height - 2 >= self.trueheight - position:
            cornery = self.trueheight - self.height + 2

        self.menubox.refresh(0, 0, self.toplefty, self.topleftx,
                             self.height + self.toplefty,
                             self.width + self.topleftx)
        self.update_pad(position)
        self.pad.refresh(cornery, 0, self.toplefty + 1, self.topleftx + 1,
                         self.height - 2 + self.toplefty,
                         self.width - 2 + self.topleftx)