ProgressQuest-cli/pqcli/ui/new_game_view.py

import random
import typing as T

import urwid
import urwid_readline

from pqcli.config import CLASSES, PRIME_STATS, RACES
from pqcli.mechanic import Stats, StatsBuilder, generate_name
from pqcli.ui.button import MenuButton
from pqcli.ui.line_box import LineBox


class StatsBox(urwid.Filler):
    def __init__(self) -> None:
        self.stats_builder = StatsBuilder()
        self.stats = self.stats_builder.roll()
        self.stat_labels = {stat: urwid.Text("0") for stat in PRIME_STATS}
        self.total_label = urwid.Text("0")
        self.update_values()

        roll_button = MenuButton("Roll", on_press=self.on_roll_press)
        unroll_button = MenuButton("Unroll", on_press=self.on_unroll_press)

        value_texts = list(self.stat_labels.values())
        label_texts = [urwid.Text(f"{stat.value}: ") for stat in PRIME_STATS]

        super().__init__(
            urwid.Pile(
                [
                    (
                        13,
                        LineBox(
                            urwid.ListBox(
                                [
                                    urwid.Columns(
                                        [
                                            urwid.Pile(label_texts),
                                            urwid.Pile(value_texts),
                                        ]
                                    ),
                                    urwid.Divider(),
                                    urwid.Columns(
                                        [
                                            urwid.Text("Total: "),
                                            self.total_label,
                                        ]
                                    ),
                                    urwid.Divider(),
                                    roll_button,
                                    unroll_button,
                                ]
                            ),
                            title="Stats",
                        ),
                    )
                ]
            ),
            valign="top",
        )

    def on_roll_press(self, _user_data: T.Any) -> None:
        self.stats = self.stats_builder.roll()
        self.update_values()

    def on_unroll_press(self, _user_data: T.Any) -> None:
        try:
            self.stats = self.stats_builder.unroll()
        except IndexError:
            return
        self.update_values()

    def update_values(self) -> None:
        for stat, edit in self.stat_labels.items():
            edit.set_text(str(self.stats[stat]))
        self.total_label.set_text(
            str(sum(self.stats[stat] for stat in self.stat_labels.keys()))
        )


class RaceBox(LineBox):
    def __init__(self) -> None:
        race_checkboxes: T.List[urwid.RadioButton] = []
        self.race = random.choice(RACES)
        for race in RACES:
            urwid.RadioButton(
                group=race_checkboxes,
                label=race.name,
                state=race.name == self.race.name,
            )
        super().__init__(urwid.ListBox(race_checkboxes), title="Race")


class ClassBox(LineBox):
    def __init__(self) -> None:
        class_checkboxes: T.List[urwid.RadioButton] = []
        self.class_ = random.choice(CLASSES)
        for class_ in CLASSES:
            urwid.RadioButton(
                group=class_checkboxes,
                label=class_.name,
                state=class_.name == self.class_.name,
            )
        super().__init__(urwid.ListBox(class_checkboxes), title="Class")


class NewGameView(urwid.Pile):
    def __init__(self, on_confirm: T.Callable, on_cancel: T.Callable) -> None:
        self.on_confirm = on_confirm
        self.on_cancel = on_cancel

        race_box = RaceBox()
        class_box = ClassBox()
        stats_box = StatsBox()

        self.char_name_edit = urwid_readline.ReadlineEdit(
            "Name: ", generate_name()
        )
        generate_char_name_btn = MenuButton(
            "Generate random name", on_press=self.on_generate_char_name_press
        )

        super().__init__(
            [
                (1, urwid.Filler(self.char_name_edit)),
                (
                    1,
                    urwid.Filler(
                        urwid.Padding(
                            generate_char_name_btn,
                            left=len(self.char_name_edit.caption),
                        )
                    ),
                ),
                urwid.Columns(
                    [
                        ("weight", 2, race_box),
                        ("weight", 2, class_box),
                        stats_box,
                    ]
                ),
            ]
        )

    def generate_random_char_name(self) -> None:
        self.char_name_edit.edit_text = generate_name()
        self.char_name_edit.edit_pos = len(self.char_name_edit.edit_text)

    def on_generate_char_name_press(self, _user_data: T.Any) -> None:
        self.generate_random_char_name()

    def unhandled_input(self, key: str) -> bool:
        if key == "esc":
            self.on_cancel()
            return True

        return False