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

kiauh/components/klipper/klipper_remove.py

@@ -0,0 +1,98 @@
+# ======================================================================= #
+#  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  #
+# ======================================================================= #
+from __future__ import annotations
+
+from typing import List
+
+from components.klipper import KLIPPER_DIR, KLIPPER_ENV_DIR
+from components.klipper.klipper import Klipper
+from components.klipper.klipper_dialogs import print_instance_overview
+from core.instance_manager.instance_manager import InstanceManager
+from core.logger import Logger
+from utils.fs_utils import run_remove_routines
+from utils.input_utils import get_selection_input
+from utils.instance_utils import get_instances
+from utils.sys_utils import unit_file_exists
+
+
+def run_klipper_removal(
+    remove_service: bool,
+    remove_dir: bool,
+    remove_env: bool,
+) -> None:
+    klipper_instances: List[Klipper] = get_instances(Klipper)
+
+    if remove_service:
+        Logger.print_status("Removing Klipper instances ...")
+        if klipper_instances:
+            instances_to_remove = select_instances_to_remove(klipper_instances)
+            remove_instances(instances_to_remove)
+        else:
+            Logger.print_info("No Klipper Services installed! Skipped ...")
+
+    if (remove_dir or remove_env) and unit_file_exists("klipper", suffix="service"):
+        Logger.print_info("There are still other Klipper services installed:")
+        Logger.print_info(f"● '{KLIPPER_DIR}' was not removed.", prefix=False)
+        Logger.print_info(f"● '{KLIPPER_ENV_DIR}' was not removed.", prefix=False)
+    else:
+        if remove_dir:
+            Logger.print_status("Removing Klipper local repository ...")
+            run_remove_routines(KLIPPER_DIR)
+        if remove_env:
+            Logger.print_status("Removing Klipper Python environment ...")
+            run_remove_routines(KLIPPER_ENV_DIR)
+
+
+def select_instances_to_remove(instances: List[Klipper]) -> List[Klipper] | None:
+    start_index = 1
+    options = [str(i + start_index) for i in range(len(instances))]
+    options.extend(["a", "b"])
+    instance_map = {options[i]: instances[i] for i in range(len(instances))}
+
+    print_instance_overview(
+        instances,
+        start_index=start_index,
+        show_index=True,
+        show_select_all=True,
+    )
+    selection = get_selection_input("Select Klipper instance to remove", options)
+
+    instances_to_remove = []
+    if selection == "b":
+        return None
+    elif selection == "a":
+        instances_to_remove.extend(instances)
+    else:
+        instances_to_remove.append(instance_map[selection])
+
+    return instances_to_remove
+
+
+def remove_instances(
+    instance_list: List[Klipper] | None,
+) -> None:
+    if not instance_list:
+        return
+
+    for instance in instance_list:
+        Logger.print_status(f"Removing instance {instance.service_file_path.stem} ...")
+        InstanceManager.remove(instance)
+
+
+def delete_klipper_logs(instances: List[Klipper]) -> None:
+    all_logfiles = []
+    for instance in instances:
+        all_logfiles = list(instance.base.log_dir.glob("klippy.log*"))
+    if not all_logfiles:
+        Logger.print_info("No Klipper logs found. Skipped ...")
+        return
+
+    for log in all_logfiles:
+        Logger.print_status(f"Remove '{log}'")
+        run_remove_routines(log)