kiauh/kiauh/core/menus/main_menu.py

# ======================================================================= #
#  Copyright (C) 2020 - 2024 Dominik Willner <th33xitus@gmail.com>        #
#                                                                         #
#  This file is part of KIAUH - Klipper Installation And Update Helper    #
#  https://github.com/dw-0/kiauh                                          #
#                                                                         #
#  This file may be distributed under the terms of the GNU GPLv3 license  #
# ======================================================================= #

import textwrap
from typing import Type, Optional

from components.klipper.klipper_utils import get_klipper_status
from components.log_uploads.menus.log_upload_menu import LogUploadMenu
from components.moonraker.moonraker_utils import get_moonraker_status
from components.webui_client.client_utils import (
    get_client_status,
    get_current_client_config,
)
from components.webui_client.fluidd_data import FluiddData
from components.webui_client.mainsail_data import MainsailData
from core.menus import FooterType
from core.menus.advanced_menu import AdvancedMenu
from core.menus.backup_menu import BackupMenu
from core.menus.base_menu import BaseMenu, Option
from extensions.extensions_menu import ExtensionsMenu
from core.menus.install_menu import InstallMenu
from core.menus.remove_menu import RemoveMenu
from core.menus.settings_menu import SettingsMenu
from core.menus.update_menu import UpdateMenu
from utils.constants import (
    COLOR_MAGENTA,
    COLOR_CYAN,
    RESET_FORMAT,
    COLOR_RED,
    COLOR_GREEN,
    COLOR_YELLOW,
)


# noinspection PyUnusedLocal
# noinspection PyMethodMayBeStatic
class MainMenu(BaseMenu):
    def __init__(self):
        super().__init__()

        self.header = True
        self.footer_type = FooterType.QUIT

        self.kl_status = ""
        self.kl_repo = ""
        self.mr_status = ""
        self.mr_repo = ""
        self.ms_status = ""
        self.fl_status = ""
        self.ks_status = ""
        self.mb_status = ""
        self.cn_status = ""
        self.cc_status = ""
        self.init_status()

    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
        """MainMenu does not have a previous menu"""
        pass

    def set_options(self) -> None:
        self.options = {
            "0": Option(method=self.log_upload_menu, menu=True),
            "1": Option(method=self.install_menu, menu=True),
            "2": Option(method=self.update_menu, menu=True),
            "3": Option(method=self.remove_menu, menu=True),
            "4": Option(method=self.advanced_menu, menu=True),
            "5": Option(method=self.backup_menu, menu=True),
            "e": Option(method=self.extension_menu, menu=True),
            "s": Option(method=self.settings_menu, menu=True),
        }

    def init_status(self) -> None:
        status_vars = ["kl", "mr", "ms", "fl", "ks", "mb", "cn"]
        for var in status_vars:
            setattr(
                self,
                f"{var}_status",
                f"{COLOR_RED}Not installed!{RESET_FORMAT}",
            )

    def fetch_status(self) -> None:
        self._update_status("kl", get_klipper_status)
        self._update_status("mr", get_moonraker_status)
        self.ms_status = get_client_status(MainsailData())
        self.fl_status = get_client_status(FluiddData())
        self.cc_status = get_current_client_config([MainsailData(), FluiddData()])

    def _update_status(self, status_name: str, status_fn: callable) -> None:
        status_data = status_fn()
        status = status_data.get("status")
        code = status_data.get("status_code")
        instances = f" {status_data.get('instances')}" if code == 1 else ""
        setattr(
            self,
            f"{status_name}_status",
            self._format_status_by_code(code, status, instances),
        )
        setattr(
            self,
            f"{status_name}_repo",
            f"{COLOR_CYAN}{status_data.get('repo')}{RESET_FORMAT}",
        )

    def _format_status_by_code(self, code: int, status: str, count: str) -> str:
        if code == 1:
            return f"{COLOR_GREEN}{status}{count}{RESET_FORMAT}"
        elif code == 2:
            return f"{COLOR_RED}{status}{count}{RESET_FORMAT}"

        return f"{COLOR_YELLOW}{status}{count}{RESET_FORMAT}"

    def print_menu(self):
        self.fetch_status()

        header = " [ Main Menu ] "
        footer1 = "KIAUH v6.0.0"
        footer2 = f"Changelog: {COLOR_MAGENTA}https://git.io/JnmlX{RESET_FORMAT}"
        color = COLOR_CYAN
        count = 62 - len(color) - len(RESET_FORMAT)
        menu = textwrap.dedent(
            f"""
            /=======================================================\\
            | {color}{header:~^{count}}{RESET_FORMAT} |
            |-------------------------------------------------------|
            |  0) [Log-Upload] |   Klipper: {self.kl_status:<32} |
            |                  |      Repo: {self.kl_repo:<32} |
            |  1) [Install]    |------------------------------------|
            |  2) [Update]     | Moonraker: {self.mr_status:<32} |
            |  3) [Remove]     |      Repo: {self.mr_repo:<32} |
            |  4) [Advanced]   |------------------------------------|
            |  5) [Backup]     |        Mainsail: {self.ms_status:<26} |
            |                  |          Fluidd: {self.fl_status:<26} |
            |  S) [Settings]   |   Client-Config: {self.cc_status:<26} |
            |                  |                                    |
            | Community:       |   KlipperScreen: {self.ks_status:<26} |
            |  E) [Extensions] |     Mobileraker: {self.mb_status:<26} |
            |                  |       Crowsnest: {self.cn_status:<26} |
            |-------------------------------------------------------|
            | {COLOR_CYAN}{footer1:^16}{RESET_FORMAT} | {footer2:^43} |
            """
        )[1:]
        print(menu, end="")

    def log_upload_menu(self, **kwargs):
        LogUploadMenu().run()

    def install_menu(self, **kwargs):
        InstallMenu(previous_menu=self.__class__).run()

    def update_menu(self, **kwargs):
        UpdateMenu(previous_menu=self.__class__).run()

    def remove_menu(self, **kwargs):
        RemoveMenu(previous_menu=self.__class__).run()

    def advanced_menu(self, **kwargs):
        AdvancedMenu(previous_menu=self.__class__).run()

    def backup_menu(self, **kwargs):
        BackupMenu(previous_menu=self.__class__).run()

    def settings_menu(self, **kwargs):
        SettingsMenu(previous_menu=self.__class__).run()

    def extension_menu(self, **kwargs):
        ExtensionsMenu(previous_menu=self.__class__).run()