feat: KIAUH v6 - full rewrite of KIAUH in Python (#428)

2025-12-27 01:33:36 +05:00 · 2024-08-31 19:16:52 +02:00
parent 8547942986
commit 0ee0fa3325
159 changed files with 13461 additions and 54 deletions
--- a/kiauh/extensions/pretty_gcode/pretty_gcode_extension.py
+++ b/kiauh/extensions/pretty_gcode/pretty_gcode_extension.py
@@ -0,0 +1,101 @@
+# ======================================================================= #
+#  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 shutil
+from pathlib import Path
+
+from components.webui_client.client_utils import create_nginx_cfg
+from core.constants import NGINX_SITES_AVAILABLE, NGINX_SITES_ENABLED
+from core.logger import DialogType, Logger
+from extensions.base_extension import BaseExtension
+from utils.common import check_install_dependencies
+from utils.fs_utils import (
+    remove_file,
+)
+from utils.git_utils import git_clone_wrapper, git_pull_wrapper
+from utils.input_utils import get_number_input
+from utils.sys_utils import cmd_sysctl_service, get_ipv4_addr
+
+MODULE_PATH = Path(__file__).resolve().parent
+PGC_DIR = Path.home().joinpath("pgcode")
+PGC_REPO = "https://github.com/Kragrathea/pgcode"
+PGC_CONF = "pgcode.local.conf"
+
+
+# noinspection PyMethodMayBeStatic
+class PrettyGcodeExtension(BaseExtension):
+    def install_extension(self, **kwargs) -> None:
+        Logger.print_status("Installing PrettyGCode for Klipper ...")
+        Logger.print_dialog(
+            DialogType.ATTENTION,
+            [
+                "Make sure you don't select a port which is already in use by "
+                "another application. Your input will not be validated! Choosing a port "
+                "which is already in use by another application may cause issues!",
+                "The default port is 7136.",
+            ],
+        )
+
+        port = get_number_input(
+            "On which port should PrettyGCode run",
+            min_count=0,
+            default=7136,
+            allow_go_back=True,
+        )
+
+        check_install_dependencies({"nginx"})
+
+        try:
+            if PGC_DIR.exists():
+                shutil.rmtree(PGC_DIR)
+
+            git_clone_wrapper(PGC_REPO, PGC_DIR)
+
+            create_nginx_cfg(
+                "PrettyGCode for Klipper",
+                cfg_name=PGC_CONF,
+                template_src=MODULE_PATH.joinpath(f"assets/{PGC_CONF}"),
+                ROOT_DIR=PGC_DIR,
+                PORT=port,
+            )
+
+            cmd_sysctl_service("nginx", "restart")
+
+            log = f"Open PrettyGCode now on: http://{get_ipv4_addr()}:{port}"
+            Logger.print_ok("PrettyGCode installation complete!", start="\n")
+            Logger.print_ok(log, prefix=False, end="\n\n")
+
+        except Exception as e:
+            Logger.print_error(
+                f"Error during PrettyGCode for Klipper installation: {e}"
+            )
+
+    def update_extension(self, **kwargs) -> None:
+        Logger.print_status("Updating PrettyGCode for Klipper ...")
+        try:
+            git_pull_wrapper(PGC_REPO, PGC_DIR)
+
+        except Exception as e:
+            Logger.print_error(f"Error during PrettyGCode for Klipper update: {e}")
+
+    def remove_extension(self, **kwargs) -> None:
+        try:
+            Logger.print_status("Removing PrettyGCode for Klipper ...")
+
+            # remove pgc dir
+            shutil.rmtree(PGC_DIR)
+            # remove nginx config
+            remove_file(NGINX_SITES_AVAILABLE.joinpath(PGC_CONF), True)
+            remove_file(NGINX_SITES_ENABLED.joinpath(PGC_CONF), True)
+            # restart nginx
+            cmd_sysctl_service("nginx", "restart")
+
+            Logger.print_ok("PrettyGCode for Klipper removed!")
+
+        except Exception as e:
+            Logger.print_error(f"Error during PrettyGCode for Klipper removal: {e}")