consoleui.py

from argparse import ArgumentParser, Namespace

from src.config import SWAP
from src.interfaces.baseui import BaseUI
from src.modules.gameboard.resultlist import ResultList


class ConsoleUI(BaseUI):
    """Console UI"""

    def __init__(self) -> None:
        super().__init__()

        self.parser: ArgumentParser = ArgumentParser(
            description="Spellsolver", epilog="Word Finder for Discord Spellcast"
        )

        self.parser.add_argument(
            "game", type=str, default=None, nargs="?", help="gameboard string"
        )
        self.parser.add_argument(
            "--swap", type=int, required=False, help="enable swap mode", default=0
        )

        self.parser.add_argument(
            "--gems", type=str, required=False, help="gem coordinates", default=""
        )
        self.parser.add_argument(
            "--ices", type=str, required=False, help="ice coordinates", default=""
        )

        self.parser.add_argument(
            "--x2", type=str, required=False, help="x2 coordinates", default=""
        )
        self.parser.add_argument(
            "--dl", type=str, required=False, help="dl coordinates", default=""
        )
        self.parser.add_argument(
            "--tl", type=str, required=False, help="tl coordinates", default=""
        )

        self.opt = self.parser.parse_args()

        swap = self.opt.swap if self.opt.swap else SWAP
        self.init(swap=swap)

    def _mainargs(self, opt: Namespace) -> None:
        """Main loop of the Console UI using arguments"""
        gameboard_string = opt.game
        solver = self.load(gameboard_string)

        gems_string = opt.gems.replace(".", " ")
        ices_string = opt.ices.replace(".", " ")
        solver.set_modifiers(ices_string, gems_string)

        x2_string = opt.x2.replace(".", " ")
        dl_string = opt.dl.replace(".", " ")
        tl_string = opt.tl.replace(".", " ")
        solver.set_multipliers(x2_string, dl_string, tl_string)

        swap = opt.swap if opt.swap else SWAP
        results = solver.solve(swap=swap)
        self.print_results(results)

    def _maininput(self) -> None:
        """Main loop of the Console UI using inputs"""
        gameboard_string = input("Insert a gameboard: ")
        solver = self.load(gameboard_string)

        ices_string = input("Insert ice coordinates: ").replace(".", " ")
        gems_string = input("Insert gem coordinates: ").replace(".", " ")

        x2_string = input("Insert 2x coordinates: ").replace(".", " ")
        dl_string = input("Insert DL coordinates: ").replace(".", " ")
        tl_string = input("Insert TL coordinates: ").replace(".", " ")
        solver.set_multipliers(x2_string, dl_string, tl_string)

        swap = int(input("Use swap?: "))
        results = solver.solve(swap=swap)
        self.print_results(results)

    def mainloop(self) -> None:
        """Main loop of the Console UI"""
        if self.opt.game:
            self._mainargs(self.opt)
            return

        while True:
            self._maininput()

    @staticmethod
    def print_results(results: ResultList) -> None:
        """Print results to console"""
        sorted_words = results.sorted_words
        results.words_to_text(sorted_words[:10])


if __name__ == "__main__":
    app = ConsoleUI()
    app.mainloop()