# ======================================================================= #
#  Copyright (C) 2020 - 2025 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  #
# ======================================================================= #
from __future__ import annotations

import textwrap
from typing import Type

from components.webui_client.base_data import BaseWebClient
from components.webui_client.client_setup import install_client
from components.webui_client.client_utils import (
    get_client_port_selection,
    get_nginx_listen_port,
    set_listen_port,
)
from core.logger import Logger
from core.menus import Option
from core.menus.base_menu import BaseMenu
from core.services.message_service import Message
from core.settings.kiauh_settings import KiauhSettings, WebUiSettings
from core.types.color import Color
from utils.sys_utils import cmd_sysctl_service, get_ipv4_addr


# noinspection PyUnusedLocal
class ClientInstallMenu(BaseMenu):
    def __init__(
        self, client: BaseWebClient, previous_menu: Type[BaseMenu] | None = None
    ):
        super().__init__()
        self.title = f"Installation Menu > {client.display_name}"
        self.title_color = Color.GREEN
        self.previous_menu: Type[BaseMenu] | None = previous_menu
        self.client: BaseWebClient = client
        self.settings = KiauhSettings()
        self.client_settings: WebUiSettings = self.settings[client.name]

    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
        from core.menus.install_menu import InstallMenu

        self.previous_menu = previous_menu if previous_menu is not None else InstallMenu

    def set_options(self) -> None:
        self.options = {
            "1": Option(method=self.reinstall_client),
            "2": Option(method=self.change_listen_port),
        }

    def print_menu(self) -> None:
        client_name = self.client.display_name
        port = f"(Current: {Color.apply(self._get_current_port(), Color.GREEN)})"
        menu = textwrap.dedent(
            f"""
            ╟───────────────────────────────────────────────────────╢
            ║  1) Reinstall {client_name:16}                        ║
            ║  2) Reconfigure Listen Port {port:<34} ║
            ╟───────────────────────────────────────────────────────╢
            """
        )[1:]
        print(menu, end="")

    def reinstall_client(self, **kwargs) -> None:
        install_client(self.client, settings=self.settings, reinstall=True)

    def change_listen_port(self, **kwargs) -> None:
        curr_port = self._get_current_port()
        new_port = get_client_port_selection(
            self.client,
            self.settings,
            reconfigure=True,
        )

        cmd_sysctl_service("nginx", "stop")
        set_listen_port(self.client, curr_port, new_port)

        Logger.print_status("Saving new port configuration ...")
        self.client_settings.port = new_port
        self.settings.save()
        Logger.print_ok("Port configuration saved!")

        cmd_sysctl_service("nginx", "start")

        # noinspection HttpUrlsUsage
        message = Message(
            title="Port reconfiguration complete!",
            text=[
                f"Open {self.client.display_name} now on: "
                f"http://{get_ipv4_addr()}:{new_port}",
            ],
            color=Color.GREEN,
        )
        self.message_service.set_message(message)

    def _get_current_port(self) -> int:
        curr_port = get_nginx_listen_port(self.client.nginx_config)
        if curr_port is None:
            # if the port is not found in the config file we use
            # the default port from the kiauh settings as fallback
            return int(self.client_settings.port)
        return curr_port