terminal.py

import curses

from gameplay_utils import Cell


cell_type_to_terminal_char = {
    Cell.EMPTY: ' ',
    Cell.TRACK: 'X',
    Cell.CONSIDER: ' ',
    Cell.MARKED: 'X',
    Cell.ENEMY: 'O',
}


class Terminal:
    def __init__(self):
        self._screen_obj = curses.initscr()
        self._screen_obj.nodelay(True)
        curses.noecho()
        curses.cbreak()
        self._screen_obj.keypad(True)

        self.max_y, self.max_x = self.get_max_y_and_x()

    def get_max_y_and_x(self):
        return self._screen_obj.getmaxyx()

    def get_pressed_key(self):
        return self._screen_obj.getch()

    def destroy(self, print_text_after_destroy: str = ''):
        curses.nocbreak()
        self._screen_obj.keypad(False)
        curses.echo()
        curses.endwin()

        if print_text_after_destroy:
            print(print_text_after_destroy)

    def print_changes(self, changes: list):
        for y, x, cell_type in changes:
            if cell_type == Cell.BORDER:
                self._print_border(y, x)
            else:
                char = cell_type_to_terminal_char[cell_type]
                self._print(y, x, char)

        self._screen_obj.refresh()

    def _print(self, y, x, char):
        try:
            self._screen_obj.addch(y, x, char)
        except curses.error:
            pass

    def _print_border(self, y: int, x: int):
        if y == self.max_y - 1 and x == self.max_x - 1:
            self._print(y, x, '┘')

        elif y == 0 and x == 0:
            self._print(y, x, '┌')

        elif y == self.max_y - 1 and x == 0:
            self._print(y, x, '└')

        elif y == 0 and x == self.max_x - 1:
            self._print(y, x, '┐')

        elif y == 0 or y == self.max_y - 1:
            self._print(y, x, '─')

        elif x == 0 or x == self.max_x - 1:
            self._print(y, x, '│')