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

kiauh/procedures/system.py

@@ -0,0 +1,103 @@
+# ======================================================================= #
+#  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 pathlib import Path
+from subprocess import PIPE, CalledProcessError, run
+
+from core.logger import DialogType, Logger
+from utils.common import check_install_dependencies, get_current_date
+from utils.fs_utils import check_file_exist
+from utils.input_utils import get_confirm, get_string_input
+
+
+def change_system_hostname() -> None:
+    """
+    Procedure to change the system hostname.
+    :return:
+    """
+
+    Logger.print_dialog(
+        DialogType.CUSTOM,
+        [
+            "Changing the hostname of this system allows you to access an installed "
+            "webinterface by simply typing the hostname like this in the browser:",
+            "\n\n",
+            "http://<hostname>.local",
+            "\n\n",
+            "Example: If you set your hostname to 'my-printer', you can access an "
+            "installed webinterface by tyoing 'http://my-printer.local' in the "
+            "browser.",
+        ],
+        custom_title="CHANGE SYSTEM HOSTNAME",
+    )
+    if not get_confirm("Do you want to change the hostname?", default_choice=False):
+        return
+
+    Logger.print_dialog(
+        DialogType.CUSTOM,
+        [
+            "Allowed characters: a-z, 0-9 and '-'",
+            "The name must not contain the following:",
+            "\n\n",
+            "● Any special characters",
+            "● No leading or trailing '-'",
+        ],
+    )
+    hostname = get_string_input(
+        "Enter the new hostname",
+        regex="^[a-z0-9]+([a-z0-9-]*[a-z0-9])?$",
+    )
+    if not get_confirm(f"Change the hostname to '{hostname}'?", default_choice=False):
+        Logger.print_info("Aborting hostname change ...")
+        return
+
+    try:
+        Logger.print_status("Changing hostname ...")
+
+        Logger.print_status("Checking for dependencies ...")
+        check_install_dependencies({"avahi-daemon"}, include_global=False)
+
+        # create or backup hosts file
+        Logger.print_status("Creating backup of hosts file ...")
+        hosts_file = Path("/etc/hosts")
+        if not check_file_exist(hosts_file, True):
+            cmd = ["sudo", "touch", hosts_file.as_posix()]
+            run(cmd, stderr=PIPE, check=True)
+        else:
+            date_time = get_current_date()
+            name = f"hosts.{date_time.get('date')}-{date_time.get('time')}.bak"
+            hosts_file_backup = Path(f"/etc/{name}")
+            cmd = [
+                "sudo",
+                "cp",
+                hosts_file.as_posix(),
+                hosts_file_backup.as_posix(),
+            ]
+            run(cmd, stderr=PIPE, check=True)
+        Logger.print_ok()
+
+        # call hostnamectl set-hostname <hostname>
+        Logger.print_status(f"Setting hostname to '{hostname}' ...")
+        cmd = ["sudo", "hostnamectl", "set-hostname", hostname]
+        run(cmd, stderr=PIPE, check=True)
+        Logger.print_ok()
+
+        # add hostname to hosts file at the end of the file
+        Logger.print_status("Writing new hostname to /etc/hosts ...")
+        stdin = f"127.0.0.1       {hostname}\n"
+        cmd = ["sudo", "tee", "-a", hosts_file.as_posix()]
+        run(cmd, input=stdin.encode(), stderr=PIPE, stdout=PIPE, check=True)
+        Logger.print_ok()
+
+        Logger.print_ok("New hostname successfully configured!")
+        Logger.print_ok("Remember to reboot for the changes to take effect!\n")
+
+    except CalledProcessError as e:
+        Logger.print_error(f"Error during change hostname procedure: {e}")
+        return