Merge remote-tracking branch 'origin/kiauh-v6-dev' into kiauh-v6-dev

* use service_instance_exists function

* change header in __init__.py

* remove is_service_installed function

* small fix

.editorconfig

@@ -6,10 +6,6 @@ indent_size = 4
 insert_final_newline = true
 trim_trailing_whitespace = true
 charset = utf-8
-end_of_line = lf
 
 [*.py]
 max_line_length = 88
-
-[*.{sh,yml,yaml}]
-indent_size = 2

.github/workflows/fast-forward.yml

@@ -1,28 +0,0 @@
-name: fast-forward
-on:
-  issue_comment:
-    types: [ created, edited ]
-jobs:
-  fast-forward:
-    # Only run if the comment contains the /fast-forward command.
-    if: |
-      contains(github.event.comment.body, '/fast-forward') &&
-      github.event.issue.pull_request &&
-      github.base_ref == 'develop'
-    runs-on: ubuntu-latest
-
-    permissions:
-      contents: write
-      pull-requests: write
-      issues: write
-
-    steps:
-      - name: Fast forwarding
-        uses: sequoia-pgp/fast-forward@v1
-        with:
-          merge: true
-          # To reduce the workflow's verbosity, use 'on-error'
-          # to only post a comment when an error occurs, or 'never' to
-          # never post a comment.  (In all cases the information is
-          # still available in the step's summary.)
-          comment: on-error

.github/workflows/pull-request.yml

@@ -1,27 +0,0 @@
-name: pull-request
-on:
-  pull_request:
-    branches:
-      - develop
-    types: [ opened, reopened, synchronize ]
-jobs:
-  check-fast-forward:
-    runs-on: ubuntu-latest
-
-    permissions:
-      contents: read
-      # We appear to need write permission for both pull-requests and
-      # issues in order to post a comment to a pull request.
-      pull-requests: write
-      issues: write
-
-    steps:
-      - name: Checking if fast forwarding is possible
-        uses: sequoia-pgp/fast-forward@v1
-        with:
-          merge: false
-          # To reduce the workflow's verbosity, use 'on-error'
-          # to only post a comment when an error occurs, or 'never' to
-          # never post a comment.  (In all cases the information is
-          # still available in the step's summary.)
-          comment: on-error

.gitignore

@@ -1,10 +1,6 @@
 .idea
 .vscode
 .pytest_cache
-.jupyter
-*.ipynb
-*.ipynb_checkpoints
-*.tmp
 __pycache__
 .kiauh-env
 *.code-workspace

README.md

@@ -28,12 +28,12 @@
 
 ### 📋 Prerequisites
 KIAUH is a script that assists you in installing Klipper on a Linux operating system that has
-already been flashed to your Raspberry Pi's (or other SBC's) SD card. As a result, you must ensure
+already been flashed to your Raspberry Pi's (or other SBC's) SD card. As a result, you must ensure 
-that you have a functional Linux system on hand. `Raspberry Pi OS Lite (either 32bit or 64bit)` is a recommended Linux image
+that you have a functional Linux system on hand. `Raspberry Pi OS Lite (either 32bit or 64bit)` is a recommended Linux image 
-if you are using a Raspberry Pi. The [official Raspberry Pi Imager](https://www.raspberrypi.com/software/)
+if you are using a Raspberry Pi. The [official Raspberry Pi Imager](https://www.raspberrypi.com/software/) 
 is the simplest way to flash an image like this to an SD card.
 
-* Once you have downloaded, installed and launched the Raspberry Pi Imager,
+* Once you have downloaded, installed and launched the Raspberry Pi Imager, 
 select `Choose OS -> Raspberry Pi OS (other)`: \
 <p align="center">
   <img src="https://raw.githubusercontent.com/dw-0/kiauh/master/resources/screenshots/rpi_imager1.png" alt="KIAUH logo" height="350">
@@ -44,7 +44,7 @@ select `Choose OS -> Raspberry Pi OS (other)`: \
   <img src="https://raw.githubusercontent.com/dw-0/kiauh/master/resources/screenshots/rpi_imager2.png" alt="KIAUH logo" height="350">
 </p>
 
-* Back in the Raspberry Pi Imager's main menu, select the corresponding SD card to which
+* Back in the Raspberry Pi Imager's main menu, select the corresponding SD card to which 
 you want to flash the image.
 
 * Make sure to go into the Advanced Option (the cog icon in the lower left corner of the main menu)
@@ -52,9 +52,9 @@ and enable SSH and configure Wi-Fi.
 
 * If you need more help for using the Raspberry Pi Imager, please visit the [official documentation](https://www.raspberrypi.com/documentation/computers/getting-started.html).
 
-These steps **only** apply if you are actually using a Raspberry Pi. In case you want
+These steps **only** apply if you are actually using a Raspberry Pi. In case you want 
-to use a different SBC (like an Orange Pi or any other Pi derivates), please look up on how to get an appropriate Linux image flashed
+to use a different SBC (like an Orange Pi or any other Pi derivates), please look up on how to get an appropriate Linux image flashed 
-to the SD card before proceeding further (usually done with Balena Etcher in those cases). Also make sure that KIAUH will be able to run
+to the SD card before proceeding further (usually done with Balena Etcher in those cases). Also make sure that KIAUH will be able to run 
 and operate on the Linux Distribution you are going to flash. You likely will have the most success with
 distributions based on Debian 11 Bullseye. Read the notes further down below in this document.
 
@@ -82,8 +82,8 @@ Finally, start KIAUH by running the next command:
 ```
 
 * **Step 4:** \
-You should now find yourself in the main menu of KIAUH. You will see several actions to choose from depending
+You should now find yourself in the main menu of KIAUH. You will see several actions to choose from depending 
-on what you want to do. To choose an action, simply type the corresponding number into the "Perform action"
+on what you want to do. To choose an action, simply type the corresponding number into the "Perform action" 
 prompt and confirm by hitting ENTER.
 
 <hr>
@@ -101,83 +101,77 @@ prompt and confirm by hitting ENTER.
 
 <h2 align="center">🌐 Sources & Further Information</h2>
 
-<table align="center">
+<table>
 <tr>
-    <th><h3><a href="https://github.com/Klipper3d/klipper">Klipper</a></h3></th>
+<th><h3><a href="https://github.com/Klipper3d/klipper">Klipper</a></h3></th>
-    <th><h3><a href="https://github.com/Arksine/moonraker">Moonraker</a></h3></th>
+<th><h3><a href="https://github.com/Arksine/moonraker">Moonraker</a></h3></th>
-    <th><h3><a href="https://github.com/mainsail-crew/mainsail">Mainsail</a></h3></th>
+<th><h3><a href="https://github.com/mainsail-crew/mainsail">Mainsail</a></h3></th>
 </tr>
 <tr>
-    <th><img src="https://raw.githubusercontent.com/Klipper3d/klipper/master/docs/img/klipper-logo.png" alt="Klipper Logo" height="64"></th>
+<th><img src="https://raw.githubusercontent.com/Klipper3d/klipper/master/docs/img/klipper-logo.png" alt="Klipper Logo" height="64"></th>
-    <th><img src="https://avatars.githubusercontent.com/u/9563098?v=4" alt="Arksine avatar" height="64"></th>
+<th><img src="https://avatars.githubusercontent.com/u/9563098?v=4" alt="Arksine avatar" height="64"></th>
-    <th><img src="https://raw.githubusercontent.com/mainsail-crew/docs/master/assets/img/logo.png" alt="Mainsail Logo" height="64"></th>
+<th><img src="https://raw.githubusercontent.com/mainsail-crew/docs/master/assets/img/logo.png" alt="Mainsail Logo" height="64"></th>
 </tr>
 <tr>
-    <th>by <a href="https://github.com/KevinOConnor">KevinOConnor</a></th>
+<th>by <a href="https://github.com/KevinOConnor">KevinOConnor</a></th>
-    <th>by <a href="https://github.com/Arksine">Arksine</a></th>
+<th>by <a href="https://github.com/Arksine">Arksine</a></th>
-    <th>by <a href="https://github.com/mainsail-crew">mainsail-crew</a></th>
+<th>by <a href="https://github.com/mainsail-crew">mainsail-crew</a></th>
+</tr>
+<tr>
+<th><h3><a href="https://github.com/fluidd-core/fluidd">Fluidd</a></h3></th>
+<th><h3><a href="https://github.com/jordanruthe/KlipperScreen">KlipperScreen</a></h3></th>
+<th><h3><a href="https://github.com/OctoPrint/OctoPrint">OctoPrint</a></h3></th>
+</tr>
+<tr>
+<th><img src="https://raw.githubusercontent.com/fluidd-core/fluidd/master/docs/assets/images/logo.svg" alt="Fluidd Logo" height="64"></th>
+<th><img src="https://avatars.githubusercontent.com/u/31575189?v=4" alt="jordanruthe avatar" height="64"></th>
+<th><img src="https://raw.githubusercontent.com/OctoPrint/OctoPrint/master/docs/images/octoprint-logo.png" alt="OctoPrint Logo" height="64"></th>
+</tr>
+<tr>
+<th>by <a href="https://github.com/fluidd-core">fluidd-core</a></th>
+<th>by <a href="https://github.com/jordanruthe">jordanruthe</a></th>
+<th>by <a href="https://github.com/OctoPrint">OctoPrint</a></th>
 </tr>
 
 <tr>
-    <th><h3><a href="https://github.com/fluidd-core/fluidd">Fluidd</a></h3></th>
+<th><h3><a href="https://github.com/nlef/moonraker-telegram-bot">Moonraker-Telegram-Bot</a></h3></th>
-    <th><h3><a href="https://github.com/jordanruthe/KlipperScreen">KlipperScreen</a></h3></th>
+<th><h3><a href="https://github.com/Kragrathea/pgcode">PrettyGCode for Klipper</a></h3></th>
-    <th><h3><a href="https://github.com/OctoPrint/OctoPrint">OctoPrint</a></h3></th>
+<th><h3><a href="https://github.com/TheSpaghettiDetective/moonraker-obico">Obico for Klipper</a></h3></th>
-</tr>
-<tr>
-    <th><img src="https://raw.githubusercontent.com/fluidd-core/fluidd/master/docs/assets/images/logo.svg" alt="Fluidd Logo" height="64"></th>
-    <th><img src="https://avatars.githubusercontent.com/u/31575189?v=4" alt="jordanruthe avatar" height="64"></th>
-    <th><img src="https://raw.githubusercontent.com/OctoPrint/OctoPrint/master/docs/images/octoprint-logo.png" alt="OctoPrint Logo" height="64"></th>
-</tr>
-<tr>
-    <th>by <a href="https://github.com/fluidd-core">fluidd-core</a></th>
-    <th>by <a href="https://github.com/jordanruthe">jordanruthe</a></th>
-    <th>by <a href="https://github.com/OctoPrint">OctoPrint</a></th>
 </tr>
 
 <tr>
-    <th><h3><a href="https://github.com/nlef/moonraker-telegram-bot">Moonraker-Telegram-Bot</a></h3></th>
+<th><img src="https://avatars.githubusercontent.com/u/52351624?v=4" alt="nlef avatar" height="64"></th>
-    <th><h3><a href="https://github.com/Kragrathea/pgcode">PrettyGCode for Klipper</a></h3></th>
+<th><img src="https://avatars.githubusercontent.com/u/5917231?v=4" alt="Kragrathea avatar" height="64"></th>
-    <th><h3><a href="https://github.com/TheSpaghettiDetective/moonraker-obico">Obico for Klipper</a></h3></th>
+<th><img src="https://avatars.githubusercontent.com/u/46323662?s=200&v=4" alt="Obico logo" height="64"></th>
-</tr>
-<tr>
-    <th><img src="https://avatars.githubusercontent.com/u/52351624?v=4" alt="nlef avatar" height="64"></th>
-    <th><img src="https://avatars.githubusercontent.com/u/5917231?v=4" alt="Kragrathea avatar" height="64"></th>
-    <th><img src="https://avatars.githubusercontent.com/u/46323662?s=200&v=4" alt="Obico logo" height="64"></th>
-</tr>
-<tr>
-    <th>by <a href="https://github.com/nlef">nlef</a></th>
-    <th>by <a href="https://github.com/Kragrathea">Kragrathea</a></th>
-    <th>by <a href="https://github.com/TheSpaghettiDetective">Obico</a></th>
 </tr>
 
 <tr>
-    <th><h3><a href="https://github.com/Clon1998/mobileraker_companion">Mobileraker's Companion</a></h3></th>
+<th>by <a href="https://github.com/nlef">nlef</a></th>
-    <th><h3><a href="https://octoeverywhere.com/?source=kiauh_readme">OctoEverywhere For Klipper</a></h3></th>
+<th>by <a href="https://github.com/Kragrathea">Kragrathea</a></th>
-    <th><h3><a href="https://github.com/crysxd/OctoApp-Plugin">OctoApp For Klipper</a></h3></th>
+<th>by <a href="https://github.com/TheSpaghettiDetective">Obico</a></th>
-</tr>
-<tr>
-    <th><a href="https://github.com/Clon1998/mobileraker_companion"><img src="https://raw.githubusercontent.com/Clon1998/mobileraker/master/assets/icon/mr_appicon.png" alt="Mobileraker Logo" height="64"></a></th>
-    <th><a href="https://octoeverywhere.com/?source=kiauh_readme"><img src="https://octoeverywhere.com/img/logo.svg" alt="OctoEverywhere Logo" height="64"></a></th>
-    <th><a href="https://octoapp.eu/?source=kiauh_readme"><img src="https://octoapp.eu/octoapp.webp" alt="OctoApp Logo" height="64"></a></th>
-</tr>
-<tr>
-    <th>by <a href="https://github.com/Clon1998">Patrick Schmidt</a></th>
-    <th>by <a href="https://github.com/QuinnDamerell">Quinn Damerell</a></th>
-    <th>by <a href="https://github.com/crysxd">Christian Würthner</a></th>
 </tr>
 
 <tr>
-    <th><h3><a href="https://github.com/staubgeborener/klipper-backup">Klipper-Backup</a></h3></th>
+<th><h3><a href="https://github.com/Clon1998/mobileraker_companion">Mobileraker's Companion</a></h3></th>
-    <th><h3><a href="https://simplyprint.io/">SimplyPrint for Klipper</a></h3></th>
+<th><h3><a href="https://octoeverywhere.com/?source=kiauh_readme">OctoEverywhere For Klipper</a></h3></th>
+<th><h3><a href="https://github.com/crysxd/OctoPrint-OctoApp">OctoApp For Klipper</a></h3></th>
+<th><h3></h3></th>
 </tr>
+
 <tr>
-    <th><a href="https://github.com/staubgeborener/klipper-backup"><img src="https://avatars.githubusercontent.com/u/28908603?v=4" alt="Staubgeroner Avatar" height="64"></a></th>
+<th><a href="https://github.com/Clon1998/mobileraker_companion"><img src="https://raw.githubusercontent.com/Clon1998/mobileraker/master/assets/icon/mr_appicon.png" alt="OctoEverywhere Logo" height="64"></a></th>
-    <th><a href="https://github.com/SimplyPrint"><img src="https://avatars.githubusercontent.com/u/64896552?s=200&v=4" alt="" height="64"></a></th>
+<th><a href="https://octoeverywhere.com/?source=kiauh_readme"><img src="https://octoeverywhere.com/img/logo.svg" alt="OctoEverywhere Logo" height="64"></a></th>
+<th><a href="https://octoapp.eu/?source=kiauh_readme"><img src="https://octoapp.eu/octoapp.webp" alt="OctoApp Logo" height="64"></a></th>
 </tr>
+
 <tr>
-    <th>by <a href="https://github.com/Staubgeborener">Staubgeborener</a></th>
+<th>by <a href="https://github.com/Clon1998">Patrick Schmidt</a></th>
-    <th>by <a href="https://github.com/SimplyPrint">SimplyPrint</a></th>
+<th>by <a href="https://github.com/QuinnDamerell">Quinn Damerell</a></th>
+<th>by <a href="https://github.com/crysxd">Christian Würthner</a></th>
+<th></th>
 </tr>
+
+
 </table>
 
 <hr>
@@ -192,12 +186,6 @@ prompt and confirm by hitting ENTER.
 
 <hr>
 
-<div align="center">
-    <img src="https://repobeats.axiom.co/api/embed/a1afbda9190c04a90cf4bd3061e5573bc836cb05.svg" alt="Repobeats analytics image"/>
-</div>
-
-<hr>
-
 <h2 align="center">✨ Credits ✨</h2>
 
 * A big thank you to [lixxbox](https://github.com/lixxbox) for that awesome KIAUH-Logo!

default_kiauh.cfg

@@ -14,5 +14,5 @@ port: 80
 unstable_releases: False
 
 [fluidd]
-port: 80
+port: 81
 unstable_releases: False

docs/changelog.md

@@ -2,54 +2,13 @@
 
 This document covers possible important changes to KIAUH.
 
-### 2024-08-31 (v6.0.0-alpha.1)
-Long time no see, but here we are again!
-A lot has happened in the background, but now it is time to take it out into the wild.
-
-#### KIAUH has now reached version 6! Well, at least in an alpha state...
-
-The project has seen a complete rewrite of the script from scratch in Python.
-It requires Python 3.8 or newer to run. Because this update is still in an alpha state, bugs may or will occur.
-During startup, you will be asked if you want to start the new version 6 or the old version 5.
-As long as version 6 is in a pre-release state, version 5 will still be available. If there are any critical issues
-with the new version that were overlooked, you can always switch back to the old version.
-
-In case you selected not to get asked about which version to start (option 3 or 4 in the startup dialog) and you want to
-revert that decision, you will find a line called `version_to_launch=` within the `.kiauh.ini` file in your home directory.
-Just delete that line, save the file and restart KIAUH. KIAUH will then ask you again which version you want to start.
-
-Here is a list of the most important changes to KIAUH in regard to version 6:
-- The majority of features available in KIAUH v5 are still available; they just got migrated from Bash to Python.
-- It is now possible to add new/remove instances to/from existing multi-instance installations of Klipper and Moonraker
-- KIAUH now has an Extension-System. This allows contributors to add new installers to KIAUH without having to modify the main script.
-    - You will now find some of the features that were previously available in the Installer-Menu in the Extensions-Menu.
-    - The current extensions are:
-        - G-Code Shell Command (previously found in the Advanced-Menu)
-        - Mainsail Theme Installer (previously found in the Advanced-Menu)
-        - Klipper-Backup (new in v6!)
-        - Moonraker Telegram Bot (previously found in the Installer-Menu)
-        - PrettyGCode for Klipper (previously found in the Installer-Menu)
-        - Obico for Klipper (previously found in the Installer-Menu)
-    - The following additional extensions are planned, but not yet available:
-        - Spoolman (available in v5 in the Installer-Menu)
-        - OctoApp (available in v5 in the Installer-Menu)
-- KIAUH has its own config file now
-    - The file has some default values for the currently supported options
-    - There might be more options in the future
-    - It is located in KIAUH's root directory and is called `default.kiauh.cfg`
-        - DO NOT EDIT the default file directly, instead make a copy of it and call it `kiauh.cfg`
-        - Settings changed via the Advanced-Menu will be written to the `kiauh.cfg`
-- Support for OctoPrint was removed
-
-Feel free to give version 6 a try and report any bugs or issues you encounter! Every feedback is appreciated.
-
 ### 2023-06-17
-KIAUH has now added support for installing Mobileraker's companion!
+KIAUH has now added support for installing Mobileraker's companion! 
 Mobileraker is a free and Open Source Android and iOS App for Klipper, utilizing the Moonraker API, allowing you
 to control your printer. Thank you to [Clon1998](https://github.com/Clon1998) for adding this feature!
 
 ### 2023-02-03
-The installer for MJPG-Streamer got replaced by crowsnest. It is an improved webcam service, utilizing ustreamer.
+The installer for MJPG-Streamer got replaced by crowsnest. It is an improved webcam service, utilizing ustreamer. 
 Please have a look here for additional info about crowsnest and how to configure it: https://github.com/mainsail-crew/crowsnest \
 It's unsure if the previous MJPG-Streamer installer will be updated and make its way back into KIAUH.
 A big thanks to [KwadFan](https://github.com/KwadFan) for writing the crowsnest implementation.
@@ -156,7 +115,7 @@ membership for example caused issues when installing mjpg-streamer while not usi
 Other issues could occur when trying to flash an MCU on Debian or Ubuntu distributions where a user might not be part
 of the dialout group by default. A check for the tty group is also done. The tty group is needed for setting
 up a linux MCU (currently not yet supported by KIAUH).
-* There is an issue when trying to install Mainsail or Fluidd on Ubuntu 21.10. Permissions on that distro seem to have seen a rework
+* There is an issue when trying to install Mainsail or Fluidd on Ubuntu 21.10. Permissions on that distro seem to have seen a rework 
  in comparison to 20.04 and users will be greeted with an "Error 403 - Permission denied" message after installing one of Klippers webinterfaces.
 I still have to figure out a viable solution for that.
 

kiauh.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #

kiauh.sh

@@ -10,168 +10,99 @@
 #=======================================================================#
 
 set -e
-clear -x
+clear
 
-# make sure we have the correct permissions while running the script
+function main() {
-umask 022
+  local python_command
-
-### sourcing all additional scripts
-KIAUH_SRCDIR="$(dirname -- "$(readlink -f "${BASH_SOURCE[0]}")")"
-for script in "${KIAUH_SRCDIR}/scripts/"*.sh; do . "${script}"; done
-for script in "${KIAUH_SRCDIR}/scripts/ui/"*.sh; do . "${script}"; done
-
-#===================================================#
-#=================== UPDATE KIAUH ==================#
-#===================================================#
-
-function update_kiauh() {
-  status_msg "Updating KIAUH ..."
-
-  cd "${KIAUH_SRCDIR}"
-  git reset --hard && git pull
-
-  ok_msg "Update complete! Please restart KIAUH."
-  exit 0
-}
-
-#===================================================#
-#=================== KIAUH STATUS ==================#
-#===================================================#
-
-function kiauh_update_avail() {
-  [[ ! -d "${KIAUH_SRCDIR}/.git" ]] && return
-  local origin head
-
-  cd "${KIAUH_SRCDIR}"
-
-  ### abort if not on master branch
-  ! git branch -a | grep -q "\* master" && return
-
-  ### compare commit hash
-  git fetch -q
-  origin=$(git rev-parse --short=8 origin/master)
-  head=$(git rev-parse --short=8 HEAD)
-
-  if [[ ${origin} != "${head}" ]]; then
-    echo "true"
-  fi
-}
-
-function save_startup_version() {
-  local launch_version
-
-  echo "${1}"
-
-  sed -i "/^version_to_launch=/d" "${INI_FILE}"
-  sed -i '$a'"version_to_launch=${1}" "${INI_FILE}"
-}
-
-function kiauh_update_dialog() {
-  [[ ! $(kiauh_update_avail) == "true" ]] && return
-  top_border
-  echo -e "|${green}              New KIAUH update available!              ${white}|"
-  hr
-  echo -e "|${green}  View Changelog: https://git.io/JnmlX                 ${white}|"
-  blank_line
-  echo -e "|${yellow}  It is recommended to keep KIAUH up to date. Updates  ${white}|"
-  echo -e "|${yellow}  usually contain bugfixes, important changes or new   ${white}|"
-  echo -e "|${yellow}  features. Please consider updating!                  ${white}|"
-  bottom_border
-
-  local yn
-  read -p "${cyan}###### Do you want to update now? (Y/n):${white} " yn
-  while true; do
-    case "${yn}" in
-     Y|y|Yes|yes|"")
-       do_action "update_kiauh"
-       break;;
-     N|n|No|no)
-       break;;
-     *)
-       deny_action "kiauh_update_dialog";;
-    esac
-  done
-}
-
-function launch_kiauh_v5() {
-    main_menu
-}
-
-function launch_kiauh_v6() {
   local entrypoint
 
-  if ! command -v python3 &>/dev/null || [[ $(python3 -V | cut -d " " -f2 | cut -d "." -f2) -lt 8 ]]; then
+  if command -v python3 &>/dev/null; then
-    echo "Python 3.8 or higher is not installed!"
+    python_command="python3"
-    echo "Please install Python 3.8 or higher and try again."
+  elif command -v python &>/dev/null; then
+    python_command="python"
+  else
+    echo "Python is not installed. Please install Python and try again."
     exit 1
   fi
 
   entrypoint=$(dirname "$(readlink -f "${BASH_SOURCE[0]}")")
 
-  export PYTHONPATH="${entrypoint}"
+  ${python_command} "${entrypoint}/kiauh.py"
-
-  clear -x
-  python3 "${entrypoint}/kiauh.py"
 }
 
-function main() {
-  read_kiauh_ini "${FUNCNAME[0]}"
-
-  if [[ ${version_to_launch} -eq 5 ]]; then
-    launch_kiauh_v5
-  elif [[ ${version_to_launch} -eq 6 ]]; then
-    launch_kiauh_v6
-  else
-    top_border
-    echo -e "|         ${green}KIAUH v6.0.0-alpha1 is available now!${white}         |"
-    hr
-    echo -e "|         View Changelog: ${magenta}https://git.io/JnmlX${white}          |"
-    blank_line
-    echo -e "| KIAUH v6 was completely rewritten from the ground up. |"
-    echo -e "| It's based on Python 3.8 and has many improvements.   |"
-    blank_line
-    echo -e "| ${yellow}NOTE: Version 6 is still in alpha, so bugs may occur!${white} |"
-    echo -e "| ${yellow}Yet, your feedback and bug reports are very much${white}      |"
-    echo -e "| ${yellow}appreciated and will help finalize the release.${white}       |"
-    hr
-    echo -e "| Would you like to try out KIAUH v6?                   |"
-    echo -e "| 1) Yes                                                |"
-    echo -e "| 2) No                                                 |"
-    echo -e "| 3) Yes, remember my choice for next time              |"
-    echo -e "| 4) No, remember my choice for next time               |"
-    quit_footer
-    while true; do
-      read -p "${cyan}###### Select action:${white} " -e input
-      case "${input}" in
-        1)
-          launch_kiauh_v6
-          break;;
-        2)
-          launch_kiauh_v5
-          break;;
-        3)
-          save_startup_version 6
-          launch_kiauh_v6
-          break;;
-        4)
-          save_startup_version 5
-          launch_kiauh_v5
-          break;;
-        Q|q)
-          echo -e "${green}###### Happy printing! ######${white}"; echo
-          exit 0;;
-        *)
-          error_msg "Invalid Input!\n";;
-      esac
-    done && input=""
-  fi
-}
-
-check_if_ratos
-check_euid
-init_logfile
-set_globals
-kiauh_update_dialog
-read_kiauh_ini
-init_ini
 main
+
+#### sourcing all additional scripts
+#KIAUH_SRCDIR="$(dirname -- "$(readlink -f "${BASH_SOURCE[0]}")")"
+#for script in "${KIAUH_SRCDIR}/scripts/"*.sh; do . "${script}"; done
+#for script in "${KIAUH_SRCDIR}/scripts/ui/"*.sh; do . "${script}"; done
+#
+##===================================================#
+##=================== UPDATE KIAUH ==================#
+##===================================================#
+#
+#function update_kiauh() {
+#  status_msg "Updating KIAUH ..."
+#
+#  cd "${KIAUH_SRCDIR}"
+#  git reset --hard && git pull
+#
+#  ok_msg "Update complete! Please restart KIAUH."
+#  exit 0
+#}
+#
+##===================================================#
+##=================== KIAUH STATUS ==================#
+##===================================================#
+#
+#function kiauh_update_avail() {
+#  [[ ! -d "${KIAUH_SRCDIR}/.git" ]] && return
+#  local origin head
+#
+#  cd "${KIAUH_SRCDIR}"
+#
+#  ### abort if not on master branch
+#  ! git branch -a | grep -q "\* master" && return
+#
+#  ### compare commit hash
+#  git fetch -q
+#  origin=$(git rev-parse --short=8 origin/master)
+#  head=$(git rev-parse --short=8 HEAD)
+#
+#  if [[ ${origin} != "${head}" ]]; then
+#    echo "true"
+#  fi
+#}
+#
+#function kiauh_update_dialog() {
+#  [[ ! $(kiauh_update_avail) == "true" ]] && return
+#  top_border
+#  echo -e "|${green}              New KIAUH update available!              ${white}|"
+#  hr
+#  echo -e "|${green}  View Changelog: https://git.io/JnmlX                 ${white}|"
+#  blank_line
+#  echo -e "|${yellow}  It is recommended to keep KIAUH up to date. Updates  ${white}|"
+#  echo -e "|${yellow}  usually contain bugfixes, important changes or new   ${white}|"
+#  echo -e "|${yellow}  features. Please consider updating!                  ${white}|"
+#  bottom_border
+#
+#  local yn
+#  read -p "${cyan}###### Do you want to update now? (Y/n):${white} " yn
+#  while true; do
+#    case "${yn}" in
+#      Y|y|Yes|yes|"")
+#        do_action "update_kiauh"
+#        break;;
+#      N|n|No|no)
+#        break;;
+#      *)
+#        deny_action "kiauh_update_dialog";;
+#    esac
+#  done
+#}
+#
+#check_euid
+#init_logfile
+#set_globals
+#kiauh_update_dialog
+#main_menu

kiauh/__init__.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #

kiauh/components/crowsnest/__init__.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -10,21 +10,7 @@
 from pathlib import Path
 
 from core.backup_manager import BACKUP_ROOT_DIR
-from core.constants import SYSTEMD
 
-# repo
-CROWSNEST_REPO = "https://github.com/mainsail-crew/crowsnest.git"
-
-# names
-CROWSNEST_SERVICE_NAME = "crowsnest.service"
-
-# directories
 CROWSNEST_DIR = Path.home().joinpath("crowsnest")
+CROWSNEST_REPO = "https://github.com/mainsail-crew/crowsnest.git"
 CROWSNEST_BACKUP_DIR = BACKUP_ROOT_DIR.joinpath("crowsnest-backups")
-
-# files
-CROWSNEST_MULTI_CONFIG = CROWSNEST_DIR.joinpath("tools/.config")
-CROWSNEST_INSTALL_SCRIPT = CROWSNEST_DIR.joinpath("tools/install.sh")
-CROWSNEST_BIN_FILE = Path("/usr/local/bin/crowsnest")
-CROWSNEST_LOGROTATE_FILE = Path("/etc/logrotate.d/crowsnest")
-CROWSNEST_SERVICE_FILE = SYSTEMD.joinpath(CROWSNEST_SERVICE_NAME)

kiauh/components/crowsnest/crowsnest.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -14,36 +14,27 @@ from pathlib import Path
 from subprocess import CalledProcessError, run
 from typing import List
 
-from components.crowsnest import (
+from components.crowsnest import CROWSNEST_BACKUP_DIR, CROWSNEST_DIR, CROWSNEST_REPO
-    CROWSNEST_BACKUP_DIR,
-    CROWSNEST_BIN_FILE,
-    CROWSNEST_DIR,
-    CROWSNEST_INSTALL_SCRIPT,
-    CROWSNEST_LOGROTATE_FILE,
-    CROWSNEST_MULTI_CONFIG,
-    CROWSNEST_REPO,
-    CROWSNEST_SERVICE_FILE,
-    CROWSNEST_SERVICE_NAME,
-)
 from components.klipper.klipper import Klipper
 from core.backup_manager.backup_manager import BackupManager
-from core.logger import DialogType, Logger
+from core.instance_manager.instance_manager import InstanceManager
 from core.settings.kiauh_settings import KiauhSettings
-from core.types.component_status import ComponentStatus
 from utils.common import (
     check_install_dependencies,
     get_install_status,
 )
+from utils.constants import CURRENT_USER
 from utils.git_utils import (
     git_clone_wrapper,
     git_pull_wrapper,
 )
 from utils.input_utils import get_confirm
-from utils.instance_utils import get_instances
+from utils.logger import DialogType, Logger
 from utils.sys_utils import (
     cmd_sysctl_service,
     parse_packages_from_file,
 )
+from utils.types import ComponentStatus
 
 
 def install_crowsnest() -> None:
@@ -51,10 +42,11 @@ def install_crowsnest() -> None:
     git_clone_wrapper(CROWSNEST_REPO, CROWSNEST_DIR, "master")
 
     # Step 2: Install dependencies
-    check_install_dependencies({"make"})
+    check_install_dependencies(["make"])
 
     # Step 3: Check for Multi Instance
-    instances: List[Klipper] = get_instances(Klipper)
+    im = InstanceManager(Klipper)
+    instances: List[Klipper] = im.instances
 
     if len(instances) > 1:
         print_multi_instance_warning(instances)
@@ -72,7 +64,7 @@ def install_crowsnest() -> None:
     Logger.print_info("Installer will prompt you for sudo password!")
     try:
         run(
-            f"sudo make install",
+            f"sudo make install BASE_USER={CURRENT_USER}",
             cwd=CROWSNEST_DIR,
             shell=True,
             check=True,
@@ -83,6 +75,7 @@ def install_crowsnest() -> None:
 
 
 def print_multi_instance_warning(instances: List[Klipper]) -> None:
+    _instances = [f"● {instance.data_dir_name}" for instance in instances]
     Logger.print_dialog(
         DialogType.WARNING,
         [
@@ -93,12 +86,14 @@ def print_multi_instance_warning(instances: List[Klipper]) -> None:
             "this instance to set up your 'crowsnest.conf' and steering it's service.",
             "\n\n",
             "The following instances were found:",
-            *[f"● {instance.data_dir.name}" for instance in instances],
+            *_instances,
         ],
+        end="",
     )
 
 
 def configure_multi_instance() -> None:
+    config = Path(CROWSNEST_DIR).joinpath("tools/.config")
     try:
         run(
             "make config",
@@ -108,17 +103,17 @@ def configure_multi_instance() -> None:
         )
     except CalledProcessError as e:
         Logger.print_error(f"Something went wrong! Please try again...\n{e}")
-        if CROWSNEST_MULTI_CONFIG.exists():
+        if config.exists():
-            Path.unlink(CROWSNEST_MULTI_CONFIG)
+            Path.unlink(config)
         return
 
-    if not CROWSNEST_MULTI_CONFIG.exists():
+    if not config.exists():
         Logger.print_error("Generating .config failed, installation aborted")
 
 
 def update_crowsnest() -> None:
     try:
-        cmd_sysctl_service(CROWSNEST_SERVICE_NAME, "stop")
+        cmd_sysctl_service("crowsnest", "stop")
 
         if not CROWSNEST_DIR.exists():
             git_clone_wrapper(CROWSNEST_REPO, CROWSNEST_DIR, "master")
@@ -129,17 +124,18 @@ def update_crowsnest() -> None:
             if settings.kiauh.backup_before_update:
                 bm = BackupManager()
                 bm.backup_directory(
-                    CROWSNEST_DIR.name,
+                    "crowsnest",
                     source=CROWSNEST_DIR,
                     target=CROWSNEST_BACKUP_DIR,
                 )
 
             git_pull_wrapper(CROWSNEST_REPO, CROWSNEST_DIR)
 
-            deps = parse_packages_from_file(CROWSNEST_INSTALL_SCRIPT)
+            script = CROWSNEST_DIR.joinpath("tools/install.sh")
-            check_install_dependencies({*deps})
+            deps = parse_packages_from_file(script)
+            check_install_dependencies(deps)
 
-        cmd_sysctl_service(CROWSNEST_SERVICE_NAME, "restart")
+        cmd_sysctl_service("crowsnest", "restart")
 
         Logger.print_ok("Crowsnest updated successfully.", end="\n\n")
     except CalledProcessError as e:
@@ -149,9 +145,9 @@ def update_crowsnest() -> None:
 
 def get_crowsnest_status() -> ComponentStatus:
     files = [
-        CROWSNEST_BIN_FILE,
+        Path("/usr/local/bin/crowsnest"),
-        CROWSNEST_LOGROTATE_FILE,
+        Path("/etc/logrotate.d/crowsnest"),
-        CROWSNEST_SERVICE_FILE,
+        Path("/etc/systemd/system/crowsnest.service"),
     ]
     return get_install_status(CROWSNEST_DIR, files=files)
 

kiauh/components/klipper/__init__.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -13,27 +13,9 @@ from core.backup_manager import BACKUP_ROOT_DIR
 
 MODULE_PATH = Path(__file__).resolve().parent
 
-KLIPPER_REPO_URL = "https://github.com/Klipper3d/klipper.git"
-
-# names
-KLIPPER_LOG_NAME = "klippy.log"
-KLIPPER_CFG_NAME = "printer.cfg"
-KLIPPER_SERIAL_NAME = "klippy.serial"
-KLIPPER_UDS_NAME = "klippy.sock"
-KLIPPER_ENV_FILE_NAME = "klipper.env"
-KLIPPER_SERVICE_NAME = "klipper.service"
-
-# directories
 KLIPPER_DIR = Path.home().joinpath("klipper")
-KLIPPER_KCONFIGS_DIR = Path.home().joinpath("klipper-kconfigs")
 KLIPPER_ENV_DIR = Path.home().joinpath("klippy-env")
 KLIPPER_BACKUP_DIR = BACKUP_ROOT_DIR.joinpath("klipper-backups")
-
+KLIPPER_REQUIREMENTS_TXT = KLIPPER_DIR.joinpath("scripts/klippy-requirements.txt")
-# files
-KLIPPER_REQ_FILE = KLIPPER_DIR.joinpath("scripts/klippy-requirements.txt")
-KLIPPER_INSTALL_SCRIPT = KLIPPER_DIR.joinpath("scripts/install-ubuntu-22.04.sh")
-KLIPPER_SERVICE_TEMPLATE = MODULE_PATH.joinpath(f"assets/{KLIPPER_SERVICE_NAME}")
-KLIPPER_ENV_FILE_TEMPLATE = MODULE_PATH.joinpath(f"assets/{KLIPPER_ENV_FILE_NAME}")
-
 
 EXIT_KLIPPER_SETUP = "Exiting Klipper setup ..."

kiauh/components/klipper/klipper.py

@@ -1,142 +1,152 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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 dataclasses import dataclass, field
+import subprocess
 from pathlib import Path
-from subprocess import CalledProcessError
+from typing import List
 
-from components.klipper import (
+from components.klipper import KLIPPER_DIR, KLIPPER_ENV_DIR, MODULE_PATH
-    KLIPPER_CFG_NAME,
-    KLIPPER_DIR,
-    KLIPPER_ENV_DIR,
-    KLIPPER_ENV_FILE_NAME,
-    KLIPPER_ENV_FILE_TEMPLATE,
-    KLIPPER_LOG_NAME,
-    KLIPPER_SERIAL_NAME,
-    KLIPPER_SERVICE_TEMPLATE,
-    KLIPPER_UDS_NAME,
-)
-from core.constants import CURRENT_USER
 from core.instance_manager.base_instance import BaseInstance
-from core.logger import Logger
+from utils.constants import SYSTEMD
-from utils.fs_utils import create_folders, get_data_dir
+from utils.logger import Logger
-from utils.sys_utils import get_service_file_path
 
 
 # noinspection PyMethodMayBeStatic
-@dataclass(repr=True)
+class Klipper(BaseInstance):
-class Klipper:
+    @classmethod
-    suffix: str
+    def blacklist(cls) -> List[str]:
-    base: BaseInstance = field(init=False, repr=False)
+        return ["None", "mcu"]
-    service_file_path: Path = field(init=False)
-    log_file_name: str = KLIPPER_LOG_NAME
-    klipper_dir: Path = KLIPPER_DIR
-    env_dir: Path = KLIPPER_ENV_DIR
-    data_dir: Path = field(init=False)
-    cfg_file: Path = field(init=False)
-    env_file: Path = field(init=False)
-    serial: Path = field(init=False)
-    uds: Path = field(init=False)
 
-    def __post_init__(self):
+    def __init__(self, suffix: str = ""):
-        self.base: BaseInstance = BaseInstance(Klipper, self.suffix)
+        super().__init__(instance_type=self, suffix=suffix)
-        self.base.log_file_name = self.log_file_name
+        self.klipper_dir: Path = KLIPPER_DIR
+        self.env_dir: Path = KLIPPER_ENV_DIR
+        self._cfg_file = self.cfg_dir.joinpath("printer.cfg")
+        self._log = self.log_dir.joinpath("klippy.log")
+        self._serial = self.comms_dir.joinpath("klippy.serial")
+        self._uds = self.comms_dir.joinpath("klippy.sock")
 
-        self.service_file_path: Path = get_service_file_path(Klipper, self.suffix)
+    @property
-        self.data_dir: Path = get_data_dir(Klipper, self.suffix)
+    def cfg_file(self) -> Path:
-        self.cfg_file: Path = self.base.cfg_dir.joinpath(KLIPPER_CFG_NAME)
+        return self._cfg_file
-        self.env_file: Path = self.base.sysd_dir.joinpath(KLIPPER_ENV_FILE_NAME)
+
-        self.serial: Path = self.base.comms_dir.joinpath(KLIPPER_SERIAL_NAME)
+    @property
-        self.uds: Path = self.base.comms_dir.joinpath(KLIPPER_UDS_NAME)
+    def log(self) -> Path:
+        return self._log
+
+    @property
+    def serial(self) -> Path:
+        return self._serial
+
+    @property
+    def uds(self) -> Path:
+        return self._uds
 
     def create(self) -> None:
-        from utils.sys_utils import create_env_file, create_service_file
-
         Logger.print_status("Creating new Klipper Instance ...")
+        service_template_path = MODULE_PATH.joinpath("assets/klipper.service")
+        service_file_name = self.get_service_file_name(extension=True)
+        service_file_target = SYSTEMD.joinpath(service_file_name)
+        env_template_file_path = MODULE_PATH.joinpath("assets/klipper.env")
+        env_file_target = self.sysd_dir.joinpath("klipper.env")
 
         try:
-            create_folders(self.base.base_folders)
+            self.create_folders()
-
+            self.write_service_file(
-            create_service_file(
+                service_template_path, service_file_target, env_file_target
-                name=self.service_file_path.name,
-                content=self._prep_service_file_content(),
             )
+            self.write_env_file(env_template_file_path, env_file_target)
 
-            create_env_file(
+        except subprocess.CalledProcessError as e:
-                path=self.base.sysd_dir.joinpath(KLIPPER_ENV_FILE_NAME),
+            Logger.print_error(
-                content=self._prep_env_file_content(),
+                f"Error creating service file {service_file_target}: {e}"
             )
-
-        except CalledProcessError as e:
-            Logger.print_error(f"Error creating instance: {e}")
             raise
         except OSError as e:
-            Logger.print_error(f"Error creating env file: {e}")
+            Logger.print_error(f"Error creating env file {env_file_target}: {e}")
             raise
 
-    def _prep_service_file_content(self) -> str:
+    def delete(self) -> None:
-        template = KLIPPER_SERVICE_TEMPLATE
+        service_file = self.get_service_file_name(extension=True)
+        service_file_path = self.get_service_file_path()
+
+        Logger.print_status(f"Deleting Klipper Instance: {service_file}")
 
         try:
-            with open(template, "r") as template_file:
+            command = ["sudo", "rm", "-f", service_file_path]
+            subprocess.run(command, check=True)
+            Logger.print_ok(f"Service file deleted: {service_file_path}")
+        except subprocess.CalledProcessError as e:
+            Logger.print_error(f"Error deleting service file: {e}")
+            raise
+
+    def write_service_file(
+        self,
+        service_template_path: Path,
+        service_file_target: Path,
+        env_file_target: Path,
+    ) -> None:
+        service_content = self._prep_service_file(
+            service_template_path, env_file_target
+        )
+        command = ["sudo", "tee", service_file_target]
+        subprocess.run(
+            command,
+            input=service_content.encode(),
+            stdout=subprocess.DEVNULL,
+            check=True,
+        )
+        Logger.print_ok(f"Service file created: {service_file_target}")
+
+    def write_env_file(
+        self, env_template_file_path: Path, env_file_target: Path
+    ) -> None:
+        env_file_content = self._prep_env_file(env_template_file_path)
+        with open(env_file_target, "w") as env_file:
+            env_file.write(env_file_content)
+        Logger.print_ok(f"Env file created: {env_file_target}")
+
+    def _prep_service_file(
+        self, service_template_path: Path, env_file_path: Path
+    ) -> str:
+        try:
+            with open(service_template_path, "r") as template_file:
                 template_content = template_file.read()
         except FileNotFoundError:
-            Logger.print_error(f"Unable to open {template} - File not found")
+            Logger.print_error(
+                f"Unable to open {service_template_path} - File not found"
+            )
             raise
-
+        service_content = template_content.replace("%USER%", self.user)
-        service_content = template_content.replace(
-            "%USER%",
-            CURRENT_USER,
-        )
         service_content = service_content.replace(
-            "%KLIPPER_DIR%",
+            "%KLIPPER_DIR%", str(self.klipper_dir)
-            self.klipper_dir.as_posix(),
-        )
-        service_content = service_content.replace(
-            "%ENV%",
-            self.env_dir.as_posix(),
-        )
-        service_content = service_content.replace(
-            "%ENV_FILE%",
-            self.base.sysd_dir.joinpath(KLIPPER_ENV_FILE_NAME).as_posix(),
         )
+        service_content = service_content.replace("%ENV%", str(self.env_dir))
+        service_content = service_content.replace("%ENV_FILE%", str(env_file_path))
         return service_content
 
-    def _prep_env_file_content(self) -> str:
+    def _prep_env_file(self, env_template_file_path: Path) -> str:
-        template = KLIPPER_ENV_FILE_TEMPLATE
-
         try:
-            with open(template, "r") as env_file:
+            with open(env_template_file_path, "r") as env_file:
                 env_template_file_content = env_file.read()
         except FileNotFoundError:
-            Logger.print_error(f"Unable to open {template} - File not found")
+            Logger.print_error(
+                f"Unable to open {env_template_file_path} - File not found"
+            )
             raise
-
         env_file_content = env_template_file_content.replace(
-            "%KLIPPER_DIR%", self.klipper_dir.as_posix()
+            "%KLIPPER_DIR%", str(self.klipper_dir)
         )
         env_file_content = env_file_content.replace(
-            "%CFG%",
+            "%CFG%", f"{self.cfg_dir}/printer.cfg"
-            f"{self.base.cfg_dir}/{KLIPPER_CFG_NAME}",
         )
-        env_file_content = env_file_content.replace(
+        env_file_content = env_file_content.replace("%SERIAL%", str(self.serial))
-            "%SERIAL%",
+        env_file_content = env_file_content.replace("%LOG%", str(self.log))
-            self.serial.as_posix() if self.serial else "",
+        env_file_content = env_file_content.replace("%UDS%", str(self.uds))
-        )
-        env_file_content = env_file_content.replace(
-            "%LOG%",
-            self.base.log_dir.joinpath(self.log_file_name).as_posix(),
-        )
-        env_file_content = env_file_content.replace(
-            "%UDS%",
-            self.uds.as_posix() if self.uds else "",
-        )
-
         return env_file_content

kiauh/components/klipper/klipper_dialogs.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -11,9 +11,9 @@ import textwrap
 from enum import Enum, unique
 from typing import List
 
+from core.instance_manager.base_instance import BaseInstance
 from core.menus.base_menu import print_back_footer
-from core.types.color import Color
+from utils.constants import COLOR_CYAN, COLOR_GREEN, COLOR_YELLOW, RESET_FORMAT
-from utils.instance_type import InstanceType
 
 
 @unique
@@ -23,59 +23,52 @@ class DisplayType(Enum):
 
 
 def print_instance_overview(
-    instances: List[InstanceType],
+    instances: List[BaseInstance],
     display_type: DisplayType = DisplayType.SERVICE_NAME,
     show_headline=True,
     show_index=False,
-    start_index=0,
     show_select_all=False,
-) -> None:
+):
-    dialog = "╔═══════════════════════════════════════════════════════╗\n"
+    dialog = "/=======================================================\\\n"
     if show_headline:
         d_type = (
             "Klipper instances"
             if display_type is DisplayType.SERVICE_NAME
             else "printer directories"
         )
-        headline = Color.apply(f"The following {d_type} were found:", Color.GREEN)
+        headline = f"{COLOR_GREEN}The following {d_type} were found:{RESET_FORMAT}"
-        dialog += f"║{headline:^64}║\n"
+        dialog += f"|{headline:^64}|\n"
-        dialog += "╟───────────────────────────────────────────────────────╢\n"
+        dialog += "|-------------------------------------------------------|\n"
 
     if show_select_all:
-        select_all = Color.apply("a) Select all", Color.YELLOW)
+        select_all = f"{COLOR_YELLOW}a) Select all{RESET_FORMAT}"
-        dialog += f"║ {select_all:<63}║\n"
+        dialog += f"| {select_all:<63}|\n"
-        dialog += "║                                                       ║\n"
+        dialog += "|                                                       |\n"
 
     for i, s in enumerate(instances):
         if display_type is DisplayType.SERVICE_NAME:
-            name = s.service_file_path.stem
+            name = s.get_service_file_name()
         else:
             name = s.data_dir
-        line = Color.apply(
+        line = f"{COLOR_CYAN}{f'{i})' if show_index else '●'} {name}{RESET_FORMAT}"
-            f"{f'{i + start_index})' if show_index else '●'} {name}", Color.CYAN
+        dialog += f"| {line:<63}|\n"
-        )
-        dialog += f"║ {line:<63}║\n"
-    dialog += "╟───────────────────────────────────────────────────────╢\n"
 
     print(dialog, end="")
     print_back_footer()
 
 
-def print_select_instance_count_dialog() -> None:
+def print_select_instance_count_dialog():
-    line1 = Color.apply("WARNING:", Color.YELLOW)
+    line1 = f"{COLOR_YELLOW}WARNING:{RESET_FORMAT}"
-    line2 = Color.apply(
+    line2 = f"{COLOR_YELLOW}Setting up too many instances may crash your system.{RESET_FORMAT}"
-        "Setting up too many instances may crash your system.", Color.YELLOW
-    )
     dialog = textwrap.dedent(
         f"""
-        ╔═══════════════════════════════════════════════════════╗
+        /=======================================================\\
-        ║ Please select the number of Klipper instances to set  ║
+        | Please select the number of Klipper instances to set  |
-        ║ up. The number of Klipper instances will determine    ║
+        | up. The number of Klipper instances will determine    |
-        ║ the amount of printers you can run from this host.    ║
+        | the amount of printers you can run from this host.    |
-        ║                                                       ║
+        |                                                       |
-        ║ {line1:<63}║
+        | {line1:<63}|
-        ║ {line2:<63}║
+        | {line2:<63}|
-        ╟───────────────────────────────────────────────────────╢
         """
     )[1:]
 
@@ -83,31 +76,76 @@ def print_select_instance_count_dialog() -> None:
     print_back_footer()
 
 
-def print_select_custom_name_dialog() -> None:
+def print_select_custom_name_dialog():
-    line1 = Color.apply("INFO:", Color.YELLOW)
+    line1 = f"{COLOR_YELLOW}INFO:{RESET_FORMAT}"
-    line2 = Color.apply("Only alphanumeric characters are allowed!", Color.YELLOW)
+    line2 = f"{COLOR_YELLOW}Only alphanumeric characters are allowed!{RESET_FORMAT}"
     dialog = textwrap.dedent(
         f"""
-        ╔═══════════════════════════════════════════════════════╗
+        /=======================================================\\
-        ║ Do you want to assign a custom name to each instance? ║
+        | You can now assign a custom name to each instance.    |
-        ║                                                       ║
+        | If skipped, each instance will get an index assigned  |
-        ║ Assigning a custom name will create a Klipper service ║
+        | in ascending order, starting at index '1'.            |
-        ║ and a printer directory with the chosen name.         ║
+        |                                                       |
-        ║                                                       ║
+        | {line1:<63}|
-        ║ Example for custom name 'kiauh':                      ║
+        | {line2:<63}|
-        ║  ● Klipper service:   klipper-kiauh.service           ║
-        ║  ● Printer directory: printer_kiauh_data              ║
-        ║                                                       ║
-        ║ If skipped, each instance will get an index assigned  ║
-        ║ in ascending order, starting at '1' in case of a new  ║
-        ║ installation. Otherwise, the index will be derived    ║
-        ║ from amount of already existing instances.            ║
-        ║                                                       ║
-        ║ {line1:<63}║
-        ║ {line2:<63}║
-        ╟───────────────────────────────────────────────────────╢
         """
     )[1:]
 
     print(dialog, end="")
     print_back_footer()
+
+
+def print_missing_usergroup_dialog(missing_groups) -> None:
+    line1 = f"{COLOR_YELLOW}WARNING: Your current user is not in group:{RESET_FORMAT}"
+    line2 = f"{COLOR_CYAN}● tty{RESET_FORMAT}"
+    line3 = f"{COLOR_CYAN}● dialout{RESET_FORMAT}"
+    line4 = f"{COLOR_YELLOW}INFO:{RESET_FORMAT}"
+    line5 = f"{COLOR_YELLOW}Relog required for group assignments to take effect!{RESET_FORMAT}"
+
+    dialog = textwrap.dedent(
+        f"""
+        /=======================================================\\
+        | {line1:<63}|
+        """
+    )[1:]
+
+    if "tty" in missing_groups:
+        dialog += f"| {line2:<63}|\n"
+    if "dialout" in missing_groups:
+        dialog += f"| {line3:<63}|\n"
+
+    dialog += textwrap.dedent(
+        f"""
+        |                                                       |
+        | It is possible that you won't be able to successfully |
+        | connect and/or flash the controller board without     |
+        | your user being a member of that group.               |
+        | If you want to add the current user to the group(s)   |
+        | listed above, answer with 'Y'. Else skip with 'n'.    |
+        |                                                       |
+        | {line4:<63}|
+        | {line5:<63}|
+        \\=======================================================/
+        """
+    )[1:]
+
+    print(dialog, end="")
+
+
+def print_update_warn_dialog() -> None:
+    line1 = f"{COLOR_YELLOW}WARNING:{RESET_FORMAT}"
+    line2 = f"{COLOR_YELLOW}Do NOT continue if there are ongoing prints running!{RESET_FORMAT}"
+    line3 = f"{COLOR_YELLOW}All Klipper instances will be restarted during the {RESET_FORMAT}"
+    line4 = f"{COLOR_YELLOW}update process and ongoing prints WILL FAIL.{RESET_FORMAT}"
+    dialog = textwrap.dedent(
+        f"""
+        /=======================================================\\
+        | {line1:<63}|
+        | {line2:<63}|
+        | {line3:<63}|
+        | {line4:<63}|
+        \\=======================================================/
+        """
+    )[1:]
+
+    print(dialog, end="")

kiauh/components/klipper/klipper_remove.py

@@ -0,0 +1,131 @@
+# ======================================================================= #
+#  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 typing import List, Union
+
+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 utils.fs_utils import remove_file
+from utils.input_utils import get_selection_input
+from utils.logger import Logger
+from utils.sys_utils import cmd_sysctl_manage
+
+
+def run_klipper_removal(
+    remove_service: bool,
+    remove_dir: bool,
+    remove_env: bool,
+    delete_logs: bool,
+) -> None:
+    im = InstanceManager(Klipper)
+
+    if remove_service:
+        Logger.print_status("Removing Klipper instances ...")
+        if im.instances:
+            instances_to_remove = select_instances_to_remove(im.instances)
+            remove_instances(im, instances_to_remove)
+        else:
+            Logger.print_info("No Klipper Services installed! Skipped ...")
+
+    if (remove_dir or remove_env) and im.instances:
+        Logger.print_warn("There are still other Klipper services installed!")
+        Logger.print_warn("Therefor the following parts cannot be removed:")
+        Logger.print_warn(
+            """
+            ● Klipper local repository
+            ● Klipper Python environment
+            """,
+            False,
+        )
+    else:
+        if remove_dir:
+            Logger.print_status("Removing Klipper local repository ...")
+            remove_klipper_dir()
+        if remove_env:
+            Logger.print_status("Removing Klipper Python environment ...")
+            remove_klipper_env()
+
+    # delete klipper logs of all instances
+    if delete_logs:
+        Logger.print_status("Removing all Klipper logs ...")
+        delete_klipper_logs(im.instances)
+
+
+def select_instances_to_remove(
+    instances: List[Klipper],
+) -> Union[List[Klipper], None]:
+    print_instance_overview(instances, show_index=True, show_select_all=True)
+
+    options = [str(i) for i in range(len(instances))]
+    options.extend(["a", "A", "b", "B"])
+
+    selection = get_selection_input("Select Klipper instance to remove", options)
+
+    instances_to_remove = []
+    if selection == "b".lower():
+        return None
+    elif selection == "a".lower():
+        instances_to_remove.extend(instances)
+    else:
+        instance = instances[int(selection)]
+        instances_to_remove.append(instance)
+
+    return instances_to_remove
+
+
+def remove_instances(
+    instance_manager: InstanceManager,
+    instance_list: List[Klipper],
+) -> None:
+    for instance in instance_list:
+        Logger.print_status(f"Removing instance {instance.get_service_file_name()} ...")
+        instance_manager.current_instance = instance
+        instance_manager.stop_instance()
+        instance_manager.disable_instance()
+        instance_manager.delete_instance()
+
+    cmd_sysctl_manage("daemon-reload")
+
+
+def remove_klipper_dir() -> None:
+    if not KLIPPER_DIR.exists():
+        Logger.print_info(f"'{KLIPPER_DIR}' does not exist. Skipped ...")
+        return
+
+    try:
+        shutil.rmtree(KLIPPER_DIR)
+    except OSError as e:
+        Logger.print_error(f"Unable to delete '{KLIPPER_DIR}':\n{e}")
+
+
+def remove_klipper_env() -> None:
+    if not KLIPPER_ENV_DIR.exists():
+        Logger.print_info(f"'{KLIPPER_ENV_DIR}' does not exist. Skipped ...")
+        return
+
+    try:
+        shutil.rmtree(KLIPPER_ENV_DIR)
+    except OSError as e:
+        Logger.print_error(f"Unable to delete '{KLIPPER_ENV_DIR}':\n{e}")
+
+
+def delete_klipper_logs(instances: List[Klipper]) -> None:
+    all_logfiles = []
+    for instance in instances:
+        all_logfiles = list(instance.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}'")
+        remove_file(log)

kiauh/components/klipper/klipper_setup.py

@@ -0,0 +1,173 @@
+# ======================================================================= #
+#  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 components.klipper import (
+    EXIT_KLIPPER_SETUP,
+    KLIPPER_DIR,
+    KLIPPER_ENV_DIR,
+    KLIPPER_REQUIREMENTS_TXT,
+)
+from components.klipper.klipper import Klipper
+from components.klipper.klipper_dialogs import print_update_warn_dialog
+from components.klipper.klipper_utils import (
+    add_to_existing,
+    backup_klipper_dir,
+    check_is_single_to_multi_conversion,
+    check_user_groups,
+    create_example_printer_cfg,
+    get_install_count,
+    handle_disruptive_system_packages,
+    handle_instance_naming,
+    handle_to_multi_instance_conversion,
+    init_name_scheme,
+    update_name_scheme,
+)
+from components.moonraker.moonraker import Moonraker
+from components.webui_client.client_utils import (
+    get_existing_clients,
+)
+from core.instance_manager.instance_manager import InstanceManager
+from core.settings.kiauh_settings import KiauhSettings
+from utils.common import check_install_dependencies
+from utils.git_utils import git_clone_wrapper, git_pull_wrapper
+from utils.input_utils import get_confirm
+from utils.logger import Logger
+from utils.sys_utils import (
+    cmd_sysctl_manage,
+    create_python_venv,
+    install_python_requirements,
+    parse_packages_from_file,
+)
+
+
+def install_klipper() -> None:
+    kl_im = InstanceManager(Klipper)
+
+    # ask to add new instances, if there are existing ones
+    if kl_im.instances and not add_to_existing():
+        Logger.print_status(EXIT_KLIPPER_SETUP)
+        return
+
+    install_count = get_install_count()
+    if install_count is None:
+        Logger.print_status(EXIT_KLIPPER_SETUP)
+        return
+
+    # create a dict of the size of the existing instances + install count
+    name_dict = {c: "" for c in range(len(kl_im.instances) + install_count)}
+    name_scheme = init_name_scheme(kl_im.instances, install_count)
+    mr_im = InstanceManager(Moonraker)
+    name_scheme = update_name_scheme(
+        name_scheme, name_dict, kl_im.instances, mr_im.instances
+    )
+
+    handle_instance_naming(name_dict, name_scheme)
+
+    create_example_cfg = get_confirm("Create example printer.cfg?")
+
+    try:
+        if not kl_im.instances:
+            check_install_dependencies(["git"])
+            setup_klipper_prerequesites()
+
+        count = 0
+        for name in name_dict:
+            if name_dict[name] in [n.suffix for n in kl_im.instances]:
+                continue
+
+            if check_is_single_to_multi_conversion(kl_im.instances):
+                handle_to_multi_instance_conversion(name_dict[name])
+                continue
+
+            count += 1
+            create_klipper_instance(name_dict[name], create_example_cfg)
+
+            if count == install_count:
+                break
+
+        cmd_sysctl_manage("daemon-reload")
+
+    except Exception as e:
+        Logger.print_error(e)
+        Logger.print_error("Klipper installation failed!")
+        return
+
+    # step 4: check/handle conflicting packages/services
+    handle_disruptive_system_packages()
+
+    # step 5: check for required group membership
+    check_user_groups()
+
+
+def setup_klipper_prerequesites() -> None:
+    settings = KiauhSettings()
+    repo = settings.klipper.repo_url
+    branch = settings.klipper.branch
+
+    git_clone_wrapper(repo, KLIPPER_DIR, branch)
+
+    # install klipper dependencies and create python virtualenv
+    try:
+        install_klipper_packages(KLIPPER_DIR)
+        create_python_venv(KLIPPER_ENV_DIR)
+        install_python_requirements(KLIPPER_ENV_DIR, KLIPPER_REQUIREMENTS_TXT)
+    except Exception:
+        Logger.print_error("Error during installation of Klipper requirements!")
+        raise
+
+
+def install_klipper_packages(klipper_dir: Path) -> None:
+    script = klipper_dir.joinpath("scripts/install-debian.sh")
+    packages = parse_packages_from_file(script)
+    packages = [pkg.replace("python-dev", "python3-dev") for pkg in packages]
+    packages.append("python3-venv")
+    # Add dfu-util for octopi-images
+    packages.append("dfu-util")
+    # Add dbus requirement for DietPi distro
+    if Path("/boot/dietpi/.version").exists():
+        packages.append("dbus")
+
+    check_install_dependencies(packages)
+
+
+def update_klipper() -> None:
+    print_update_warn_dialog()
+    if not get_confirm("Update Klipper now?"):
+        return
+
+    settings = KiauhSettings()
+    if settings.kiauh.backup_before_update:
+        backup_klipper_dir()
+
+    instance_manager = InstanceManager(Klipper)
+    instance_manager.stop_all_instance()
+
+    git_pull_wrapper(repo=settings.klipper.repo_url, target_dir=KLIPPER_DIR)
+
+    # install possible new system packages
+    install_klipper_packages(KLIPPER_DIR)
+    # install possible new python dependencies
+    install_python_requirements(KLIPPER_ENV_DIR, KLIPPER_REQUIREMENTS_TXT)
+
+    instance_manager.start_all_instance()
+
+
+def create_klipper_instance(name: str, create_example_cfg: bool) -> None:
+    kl_im = InstanceManager(Klipper)
+    new_instance = Klipper(suffix=name)
+    kl_im.current_instance = new_instance
+    kl_im.create_instance()
+    kl_im.enable_instance()
+    if create_example_cfg:
+        # if a client-config is installed, include it in the new example cfg
+        clients = get_existing_clients()
+        create_example_printer_cfg(new_instance, clients)
+    kl_im.start_instance()

kiauh/components/klipper/klipper_utils.py

@@ -1,117 +1,218 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import grp
 import os
+import re
 import shutil
-from pathlib import Path
 from subprocess import CalledProcessError, run
-from typing import Dict, List
+from typing import Dict, List, Optional, Union
 
 from components.klipper import (
     KLIPPER_BACKUP_DIR,
     KLIPPER_DIR,
     KLIPPER_ENV_DIR,
-    KLIPPER_INSTALL_SCRIPT,
     MODULE_PATH,
 )
 from components.klipper.klipper import Klipper
 from components.klipper.klipper_dialogs import (
     print_instance_overview,
+    print_missing_usergroup_dialog,
+    print_select_custom_name_dialog,
     print_select_instance_count_dialog,
 )
+from components.moonraker.moonraker import Moonraker
+from components.moonraker.moonraker_utils import moonraker_to_multi_conversion
 from components.webui_client.base_data import BaseWebClient
 from components.webui_client.client_config.client_config_setup import (
     create_client_config_symlink,
 )
 from core.backup_manager.backup_manager import BackupManager
-from core.constants import CURRENT_USER
+from core.instance_manager.base_instance import BaseInstance
-from core.instance_manager.base_instance import SUFFIX_BLACKLIST
+from core.instance_manager.instance_manager import InstanceManager
-from core.logger import DialogType, Logger
+from core.instance_manager.name_scheme import NameScheme
 from core.submodules.simple_config_parser.src.simple_config_parser.simple_config_parser import (
     SimpleConfigParser,
 )
-from core.types.component_status import ComponentStatus
+from utils import PRINTER_CFG_BACKUP_DIR
-from utils.common import check_install_dependencies, get_install_status
+from utils.common import get_install_status
-from utils.fs_utils import check_file_exist
+from utils.constants import CURRENT_USER
 from utils.input_utils import get_confirm, get_number_input, get_string_input
-from utils.instance_utils import get_instances
+from utils.logger import DialogType, Logger
-from utils.sys_utils import (
+from utils.sys_utils import cmd_sysctl_service
-    cmd_sysctl_service,
+from utils.types import ComponentStatus
-    install_python_packages,
-    parse_packages_from_file,
-)
 
 
 def get_klipper_status() -> ComponentStatus:
     return get_install_status(KLIPPER_DIR, KLIPPER_ENV_DIR, Klipper)
 
 
-def add_to_existing() -> bool | None:
+def check_is_multi_install(
-    kl_instances: List[Klipper] = get_instances(Klipper)
+    existing_instances: List[Klipper], install_count: int
+) -> bool:
+    return not existing_instances and install_count > 1
+
+
+def check_is_single_to_multi_conversion(
+    existing_instances: List[Klipper],
+) -> bool:
+    return len(existing_instances) == 1 and existing_instances[0].suffix == ""
+
+
+def init_name_scheme(
+    existing_instances: List[Klipper], install_count: int
+) -> NameScheme:
+    if check_is_multi_install(
+        existing_instances, install_count
+    ) or check_is_single_to_multi_conversion(existing_instances):
+        print_select_custom_name_dialog()
+        if get_confirm("Assign custom names?", False, allow_go_back=True):
+            return NameScheme.CUSTOM
+        else:
+            return NameScheme.INDEX
+    else:
+        return NameScheme.SINGLE
+
+
+def update_name_scheme(
+    name_scheme: NameScheme,
+    name_dict: Dict[int, str],
+    klipper_instances: List[Klipper],
+    moonraker_instances: List[Moonraker],
+) -> NameScheme:
+    # if there are more moonraker instances installed
+    # than klipper, we load their names into the name_dict,
+    # as we will detect and enforce that naming scheme
+    if len(moonraker_instances) > len(klipper_instances):
+        update_name_dict(name_dict, moonraker_instances)
+        return detect_name_scheme(moonraker_instances)
+    elif len(klipper_instances) > 1:
+        update_name_dict(name_dict, klipper_instances)
+        return detect_name_scheme(klipper_instances)
+    else:
+        return name_scheme
+
+
+def update_name_dict(name_dict: Dict[int, str], instances: List[BaseInstance]) -> None:
+    for k, v in enumerate(instances):
+        name_dict[k] = v.suffix
+
+
+def handle_instance_naming(name_dict: Dict[int, str], name_scheme: NameScheme) -> None:
+    if name_scheme == NameScheme.SINGLE:
+        return
+
+    for k in name_dict:
+        if name_dict[k] == "" and name_scheme == NameScheme.INDEX:
+            name_dict[k] = str(k + 1)
+        elif name_dict[k] == "" and name_scheme == NameScheme.CUSTOM:
+            assign_custom_name(k, name_dict)
+
+
+def add_to_existing() -> bool:
+    kl_instances = InstanceManager(Klipper).instances
     print_instance_overview(kl_instances)
-    _input: bool | None = get_confirm("Add new instances?", allow_go_back=True)
+    return get_confirm("Add new instances?", allow_go_back=True)
-    return _input
 
 
-def get_install_count() -> int | None:
+def get_install_count() -> Union[int, None]:
     """
     Print a dialog for selecting the amount of Klipper instances
     to set up with an option to navigate back. Returns None if the
     user selected to go back, otherwise an integer greater or equal than 1 |
     :return: Integer >= 1 or None
     """
-    kl_instances = get_instances(Klipper)
+    kl_instances = InstanceManager(Klipper).instances
     print_select_instance_count_dialog()
     question = (
         f"Number of"
         f"{' additional' if len(kl_instances) > 0 else ''} "
         f"Klipper instances to set up"
     )
-    _input: int | None = get_number_input(question, 1, default=1, allow_go_back=True)
+    return get_number_input(question, 1, default=1, allow_go_back=True)
-    return _input
 
 
 def assign_custom_name(key: int, name_dict: Dict[int, str]) -> None:
     existing_names = []
-    existing_names.extend(SUFFIX_BLACKLIST)
+    existing_names.extend(Klipper.blacklist())
     existing_names.extend(name_dict[n] for n in name_dict)
-    pattern = r"^[a-zA-Z0-9]+$"
+    question = f"Enter name for instance {key + 1}"
-
+    name_dict[key] = get_string_input(question, exclude=existing_names)
-    question = f"Enter name for instance {key}"
-    name_dict[key] = get_string_input(question, exclude=existing_names, regex=pattern)
 
 
-def check_user_groups() -> None:
+def handle_to_multi_instance_conversion(new_name: str) -> None:
-    user_groups = [grp.getgrgid(gid).gr_name for gid in os.getgroups()]
+    Logger.print_status("Converting single instance to multi instances ...")
-    missing_groups = [g for g in ["tty", "dialout"] if g not in user_groups]
+    klipper_to_multi_conversion(new_name)
+    moonraker_to_multi_conversion(new_name)
+
+
+def klipper_to_multi_conversion(new_name: str) -> None:
+    Logger.print_status("Convert Klipper single to multi instance ...")
+    im = InstanceManager(Klipper)
+    im.current_instance = im.instances[0]
+
+    # temporarily store the data dir path
+    old_data_dir = im.instances[0].data_dir
+    old_data_dir_name = im.instances[0].data_dir_name
+
+    # backup the old data_dir
+    bm = BackupManager()
+    name = f"config-{old_data_dir_name}"
+    bm.backup_directory(
+        name,
+        source=im.current_instance.cfg_dir,
+        target=PRINTER_CFG_BACKUP_DIR,
+    )
+
+    # remove the old single instance
+    im.stop_instance()
+    im.disable_instance()
+    im.delete_instance()
+
+    # create a new klipper instance with the new name
+    new_instance = Klipper(suffix=new_name)
+    im.current_instance = new_instance
+
+    if not new_instance.data_dir.is_dir():
+        # rename the old data dir and use it for the new instance
+        Logger.print_status(f"Rename '{old_data_dir}' to '{new_instance.data_dir}' ...")
+        old_data_dir.rename(new_instance.data_dir)
+    else:
+        Logger.print_info(f"Existing '{new_instance.data_dir}' found ...")
+
+    # patch the virtual_sdcard sections path
+    # value to match the new printer_data foldername
+    scp = SimpleConfigParser()
+    scp.read(new_instance.cfg_file)
+    if scp.has_section("virtual_sdcard"):
+        scp.set("virtual_sdcard", "path", str(new_instance.gcodes_dir))
+        scp.write(new_instance.cfg_file)
+
+    # finalize creating the new instance
+    im.create_instance()
+    im.enable_instance()
+    im.start_instance()
+
+
+def check_user_groups():
+    current_groups = [grp.getgrgid(gid).gr_name for gid in os.getgroups()]
+
+    missing_groups = []
+    if "tty" not in current_groups:
+        missing_groups.append("tty")
+    if "dialout" not in current_groups:
+        missing_groups.append("dialout")
 
     if not missing_groups:
         return
 
-    Logger.print_dialog(
+    print_missing_usergroup_dialog(missing_groups)
-        DialogType.ATTENTION,
-        [
-            "Your current user is not in group:",
-            *[f"● {g}" for g in missing_groups],
-            "\n\n",
-            "It is possible that you won't be able to successfully connect and/or "
-            "flash the controller board without your user being a member of that "
-            "group. If you want to add the current user to the group(s) listed above, "
-            "answer with 'Y'. Else skip with 'n'.",
-            "\n\n",
-            "INFO:",
-            "Relog required for group assignments to take effect!",
-        ],
-    )
-
     if not get_confirm(f"Add user '{CURRENT_USER}' to group(s) now?"):
         log = "Skipped adding user to required groups. You might encounter issues."
         Logger.warn(log)
@@ -161,13 +262,27 @@ def handle_disruptive_system_packages() -> None:
                     "Please fix the problem manually. Otherwise, this may have "
                     "undesirable effects on the operation of Klipper."
                 ],
+                end="",
             )
 
 
+def detect_name_scheme(instance_list: List[BaseInstance]) -> NameScheme:
+    pattern = re.compile("^\d+$")
+    for instance in instance_list:
+        if not pattern.match(instance.suffix):
+            return NameScheme.CUSTOM
+
+    return NameScheme.INDEX
+
+
+def get_highest_index(instance_list: List[Klipper]) -> int:
+    return max([int(instance.suffix.split("-")[-1]) for instance in instance_list])
+
+
 def create_example_printer_cfg(
-    instance: Klipper, clients: List[BaseWebClient] | None = None
+    instance: Klipper, clients: Optional[List[BaseWebClient]] = None
 ) -> None:
-    Logger.print_status(f"Creating example printer.cfg in '{instance.base.cfg_dir}'")
+    Logger.print_status(f"Creating example printer.cfg in '{instance.cfg_dir}'")
     if instance.cfg_file.is_file():
         Logger.print_info(f"'{instance.cfg_file}' already exists.")
         return
@@ -181,8 +296,8 @@ def create_example_printer_cfg(
         return
 
     scp = SimpleConfigParser()
-    scp.read_file(target)
+    scp.read(target)
-    scp.set_option("virtual_sdcard", "path", str(instance.base.gcodes_dir))
+    scp.set("virtual_sdcard", "path", str(instance.gcodes_dir))
 
     # include existing client configs in the example config
     if clients is not None and len(clients) > 0:
@@ -192,65 +307,12 @@ def create_example_printer_cfg(
             scp.add_section(section=section)
             create_client_config_symlink(client_config, [instance])
 
-    scp.write_file(target)
+    scp.write(target)
 
-    Logger.print_ok(f"Example printer.cfg created in '{instance.base.cfg_dir}'")
+    Logger.print_ok(f"Example printer.cfg created in '{instance.cfg_dir}'")
 
 
 def backup_klipper_dir() -> None:
     bm = BackupManager()
     bm.backup_directory("klipper", source=KLIPPER_DIR, target=KLIPPER_BACKUP_DIR)
     bm.backup_directory("klippy-env", source=KLIPPER_ENV_DIR, target=KLIPPER_BACKUP_DIR)
-
-
-def install_klipper_packages() -> None:
-    script = KLIPPER_INSTALL_SCRIPT
-    packages = parse_packages_from_file(script)
-
-    # Add pkg-config for rp2040 build
-    packages.append("pkg-config")
-
-    # Add dbus requirement for DietPi distro
-    if check_file_exist(Path("/boot/dietpi/.version")):
-        packages.append("dbus")
-
-    check_install_dependencies({*packages})
-
-
-def install_input_shaper_deps() -> None:
-    if not KLIPPER_ENV_DIR.exists():
-        Logger.print_warn("Required Klipper python environment not found!")
-        return
-
-    Logger.print_dialog(
-        DialogType.CUSTOM,
-        [
-            "Resonance measurements and shaper auto-calibration require additional "
-            "software dependencies which are not installed by default. "
-            "If you agree, the following additional system packages will be installed:",
-            "● python3-numpy",
-            "● python3-matplotlib",
-            "● libatlas-base-dev",
-            "● libopenblas-dev",
-            "\n\n",
-            "Also, the following Python package will be installed:",
-            "● numpy",
-        ],
-        custom_title="Install Input Shaper Dependencies",
-    )
-    if not get_confirm(
-        "Do you want to install the required packages?", default_choice=False
-    ):
-        return
-
-    apt_deps = (
-        "python3-numpy",
-        "python3-matplotlib",
-        "libatlas-base-dev",
-        "libopenblas-dev",
-    )
-    check_install_dependencies({*apt_deps})
-
-    py_deps = ("numpy",)
-
-    install_python_packages(KLIPPER_ENV_DIR, {*py_deps})

kiauh/components/klipper/menus/klipper_remove_menu.py

@@ -1,102 +1,116 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
-from typing import Type
+from typing import Optional, Type
 
-from components.klipper.services.klipper_setup_service import KlipperSetupService
+from components.klipper import klipper_remove
 from core.menus import FooterType, Option
 from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import COLOR_CYAN, COLOR_RED, RESET_FORMAT
 
 
 # noinspection PyUnusedLocal
 class KlipperRemoveMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
+        self.previous_menu = previous_menu
+        self.footer_type = FooterType.BACK_HELP
+        self.remove_klipper_service = False
+        self.remove_klipper_dir = False
+        self.remove_klipper_env = False
+        self.delete_klipper_logs = False
 
-        self.title = "Remove Klipper"
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
-        self.title_color = Color.RED
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
-        self.footer_type = FooterType.BACK
-
-        self.rm_svc = False
-        self.rm_dir = False
-        self.rm_env = False
-        self.select_state = False
-
-        self.klsvc = KlipperSetupService()
-
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
         from core.menus.remove_menu import RemoveMenu
 
-        self.previous_menu = previous_menu if previous_menu is not None else RemoveMenu
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else RemoveMenu
+        )
 
     def set_options(self) -> None:
         self.options = {
-            "a": Option(method=self.toggle_all),
+            "0": Option(method=self.toggle_all, menu=False),
-            "1": Option(method=self.toggle_remove_klipper_service),
+            "1": Option(method=self.toggle_remove_klipper_service, menu=False),
-            "2": Option(method=self.toggle_remove_klipper_dir),
+            "2": Option(method=self.toggle_remove_klipper_dir, menu=False),
-            "3": Option(method=self.toggle_remove_klipper_env),
+            "3": Option(method=self.toggle_remove_klipper_env, menu=False),
-            "c": Option(method=self.run_removal_process),
+            "4": Option(method=self.toggle_delete_klipper_logs, menu=False),
+            "c": Option(method=self.run_removal_process, menu=False),
         }
 
     def print_menu(self) -> None:
-        checked = f"[{Color.apply('x', Color.CYAN)}]"
+        header = " [ Remove Klipper ] "
+        color = COLOR_RED
+        count = 62 - len(color) - len(RESET_FORMAT)
+        checked = f"[{COLOR_CYAN}x{RESET_FORMAT}]"
         unchecked = "[ ]"
-        o1 = checked if self.rm_svc else unchecked
+        o1 = checked if self.remove_klipper_service else unchecked
-        o2 = checked if self.rm_dir else unchecked
+        o2 = checked if self.remove_klipper_dir else unchecked
-        o3 = checked if self.rm_env else unchecked
+        o3 = checked if self.remove_klipper_env else unchecked
-        sel_state = f"{'Select' if not self.select_state else 'Deselect'} everything"
+        o4 = checked if self.delete_klipper_logs else unchecked
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ Enter a number and hit enter to select / deselect     ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║ the specific option for removal.                      ║
+            |-------------------------------------------------------|
-            ╟───────────────────────────────────────────────────────╢
+            | Enter a number and hit enter to select / deselect     |
-            ║  a) {sel_state:49} ║
+            | the specific option for removal.                      |
-            ╟───────────────────────────────────────────────────────╢
+            |-------------------------------------------------------|
-            ║  1) {o1} Remove Service                                ║
+            |  0) Select everything                                 |
-            ║  2) {o2} Remove Local Repository                       ║
+            |-------------------------------------------------------|
-            ║  3) {o3} Remove Python Environment                     ║
+            |  1) {o1} Remove Service                                |
-            ╟───────────────────────────────────────────────────────╢
+            |  2) {o2} Remove Local Repository                       |
-            ║  C) Continue                                          ║
+            |  3) {o3} Remove Python Environment                     |
-            ╟───────────────────────────────────────────────────────╢
+            |  4) {o4} Delete all Log-Files                          |
+            |-------------------------------------------------------|
+            |  C) Continue                                          |
             """
         )[1:]
         print(menu, end="")
 
     def toggle_all(self, **kwargs) -> None:
-        self.select_state = not self.select_state
+        self.remove_klipper_service = True
-        self.rm_svc = self.select_state
+        self.remove_klipper_dir = True
-        self.rm_dir = self.select_state
+        self.remove_klipper_env = True
-        self.rm_env = self.select_state
+        self.delete_klipper_logs = True
 
     def toggle_remove_klipper_service(self, **kwargs) -> None:
-        self.rm_svc = not self.rm_svc
+        self.remove_klipper_service = not self.remove_klipper_service
 
     def toggle_remove_klipper_dir(self, **kwargs) -> None:
-        self.rm_dir = not self.rm_dir
+        self.remove_klipper_dir = not self.remove_klipper_dir
 
     def toggle_remove_klipper_env(self, **kwargs) -> None:
-        self.rm_env = not self.rm_env
+        self.remove_klipper_env = not self.remove_klipper_env
+
+    def toggle_delete_klipper_logs(self, **kwargs) -> None:
+        self.delete_klipper_logs = not self.delete_klipper_logs
 
     def run_removal_process(self, **kwargs) -> None:
-        if not self.rm_svc and not self.rm_dir and not self.rm_env:
+        if (
-            msg = "Nothing selected! Select options to remove first."
+            not self.remove_klipper_service
-            print(Color.apply(msg, Color.RED))
+            and not self.remove_klipper_dir
+            and not self.remove_klipper_env
+            and not self.delete_klipper_logs
+        ):
+            error = f"{COLOR_RED}Nothing selected! Select options to remove first.{RESET_FORMAT}"
+            print(error)
             return
 
-        self.klsvc.remove(self.rm_svc, self.rm_dir, self.rm_env)
+        klipper_remove.run_klipper_removal(
+            self.remove_klipper_service,
+            self.remove_klipper_dir,
+            self.remove_klipper_env,
+            self.delete_klipper_logs,
+        )
 
-        self.rm_svc = False
+        self.remove_klipper_service = False
-        self.rm_dir = False
+        self.remove_klipper_dir = False
-        self.rm_env = False
+        self.remove_klipper_env = False
-        self.select_state = False
+        self.delete_klipper_logs = False

kiauh/components/klipper/services/klipper_instance_service.py

@@ -1,46 +0,0 @@
-# ======================================================================= #
-#  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
-
-from typing import List
-
-from components.klipper.klipper import Klipper
-from utils.instance_utils import get_instances
-
-
-class KlipperInstanceService:
-    __cls_instance = None
-    __instances: List[Klipper] = []
-
-    def __new__(cls) -> "KlipperInstanceService":
-        if cls.__cls_instance is None:
-            cls.__cls_instance = super(KlipperInstanceService, cls).__new__(cls)
-        return cls.__cls_instance
-
-    def __init__(self) -> None:
-        if not hasattr(self, "__initialized"):
-            self.__initialized = False
-        if self.__initialized:
-            return
-        self.__initialized = True
-
-    def load_instances(self) -> None:
-        self.__instances = get_instances(Klipper)
-
-    def create_new_instance(self, suffix: str) -> Klipper:
-        instance = Klipper(suffix)
-        self.__instances.append(instance)
-        return instance
-
-    def get_all_instances(self) -> List[Klipper]:
-        return self.__instances
-
-    def get_instance_by_suffix(self, suffix: str) -> Klipper | None:
-        instances: List[Klipper] = [i for i in self.__instances if i.suffix == suffix]
-        return instances[0] if instances else None

kiauh/components/klipper/services/klipper_setup_service.py

@@ -1,362 +0,0 @@
-# ======================================================================= #
-#  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
-
-from copy import copy
-from typing import Dict, List, Tuple
-
-from components.klipper import (
-    EXIT_KLIPPER_SETUP,
-    KLIPPER_DIR,
-    KLIPPER_ENV_DIR,
-    KLIPPER_REQ_FILE,
-)
-from components.klipper.klipper import Klipper
-from components.klipper.klipper_dialogs import (
-    print_instance_overview,
-    print_select_custom_name_dialog,
-)
-from components.klipper.klipper_utils import (
-    assign_custom_name,
-    backup_klipper_dir,
-    check_user_groups,
-    create_example_printer_cfg,
-    get_install_count,
-    handle_disruptive_system_packages,
-    install_klipper_packages,
-)
-from components.klipper.services.klipper_instance_service import KlipperInstanceService
-from components.moonraker.moonraker import Moonraker
-from components.moonraker.services.moonraker_instance_service import (
-    MoonrakerInstanceService,
-)
-from components.webui_client.client_utils import (
-    get_existing_clients,
-)
-from core.instance_manager.instance_manager import InstanceManager
-from core.logger import DialogType, Logger
-from core.services.message_service import Message, MessageService
-from core.settings.kiauh_settings import KiauhSettings
-from core.types.color import Color
-from utils.fs_utils import run_remove_routines
-from utils.git_utils import git_clone_wrapper, git_pull_wrapper
-from utils.input_utils import get_confirm, get_selection_input
-from utils.sys_utils import (
-    cmd_sysctl_manage,
-    create_python_venv,
-    install_python_requirements,
-    unit_file_exists,
-)
-
-
-# noinspection PyMethodMayBeStatic
-class KlipperSetupService:
-    __cls_instance = None
-
-    kisvc: KlipperInstanceService
-    misvc: MoonrakerInstanceService
-    msgsvc = MessageService
-
-    settings: KiauhSettings
-    klipper_list: List[Klipper]
-    moonraker_list: List[Moonraker]
-
-    def __new__(cls) -> "KlipperSetupService":
-        if cls.__cls_instance is None:
-            cls.__cls_instance = super(KlipperSetupService, cls).__new__(cls)
-        return cls.__cls_instance
-
-    def __init__(self) -> None:
-        if not hasattr(self, "__initialized"):
-            self.__initialized = False
-        if self.__initialized:
-            return
-        self.__initialized = True
-        self.__init_state()
-
-    def __init_state(self) -> None:
-        self.settings = KiauhSettings()
-
-        self.kisvc = KlipperInstanceService()
-        self.kisvc.load_instances()
-        self.klipper_list = self.kisvc.get_all_instances()
-
-        self.misvc = MoonrakerInstanceService()
-        self.misvc.load_instances()
-        self.moonraker_list = self.misvc.get_all_instances()
-
-        self.msgsvc = MessageService()
-
-    def __refresh_state(self) -> None:
-        self.kisvc.load_instances()
-        self.klipper_list = self.kisvc.get_all_instances()
-
-        self.misvc.load_instances()
-        self.moonraker_list = self.misvc.get_all_instances()
-
-    def install(self) -> None:
-        Logger.print_status("Installing Klipper ...")
-
-        match_moonraker: bool = False
-
-        # if there are more moonraker instances than klipper instances, ask the user to
-        # match the klipper instance count to the count of moonraker instances with the same suffix
-        if len(self.moonraker_list) > len(self.klipper_list):
-            is_confirmed = self.__display_moonraker_info()
-            if not is_confirmed:
-                Logger.print_status(EXIT_KLIPPER_SETUP)
-                return
-            match_moonraker = True
-
-        install_count, name_dict = self.__get_install_count_and_name_dict()
-
-        if install_count == 0:
-            Logger.print_status(EXIT_KLIPPER_SETUP)
-            return
-
-        is_multi_install = install_count > 1 or (
-            len(name_dict) >= 1 and install_count >= 1
-        )
-        if not name_dict and install_count == 1:
-            name_dict = {0: ""}
-        elif is_multi_install and not match_moonraker:
-            custom_names = self.__use_custom_names_or_go_back()
-            if custom_names is None:
-                Logger.print_status(EXIT_KLIPPER_SETUP)
-                return
-
-            self.__handle_instance_names(install_count, name_dict, custom_names)
-
-        create_example_cfg = get_confirm("Create example printer.cfg?")
-        # run the actual installation
-        try:
-            self.__run_setup(name_dict, create_example_cfg)
-        except Exception as e:
-            Logger.print_error(e)
-            Logger.print_error("Klipper installation failed!")
-            return
-
-    def update(self) -> None:
-        Logger.print_dialog(
-            DialogType.WARNING,
-            [
-                "Do NOT continue if there are ongoing prints running!",
-                "All Klipper instances will be restarted during the update process and "
-                "ongoing prints WILL FAIL.",
-            ],
-        )
-
-        if not get_confirm("Update Klipper now?"):
-            return
-
-        self.__refresh_state()
-
-        if self.settings.kiauh.backup_before_update:
-            backup_klipper_dir()
-
-        InstanceManager.stop_all(self.klipper_list)
-        git_pull_wrapper(self.settings.klipper.repo_url, KLIPPER_DIR)
-        install_klipper_packages()
-        install_python_requirements(KLIPPER_ENV_DIR, KLIPPER_REQ_FILE)
-        InstanceManager.start_all(self.klipper_list)
-
-    def remove(
-        self,
-        remove_service: bool,
-        remove_dir: bool,
-        remove_env: bool,
-    ) -> None:
-        self.__refresh_state()
-
-        completion_msg = Message(
-            title="Klipper Removal Process completed",
-            color=Color.GREEN,
-        )
-
-        if remove_service:
-            Logger.print_status("Removing Klipper instances ...")
-            if self.klipper_list:
-                instances_to_remove = self.__get_instances_to_remove()
-                self.__remove_instances(instances_to_remove)
-                if instances_to_remove:
-                    instance_names = [
-                        i.service_file_path.stem for i in instances_to_remove
-                    ]
-                    txt = f"● Klipper instances removed: {', '.join(instance_names)}"
-                    completion_msg.text.append(txt)
-            else:
-                Logger.print_info("No Klipper Services installed! Skipped ...")
-
-        if (remove_dir or remove_env) and unit_file_exists("klipper", suffix="service"):
-            completion_msg.text = [
-                "Some Klipper services are still installed:",
-                f"● '{KLIPPER_DIR}' was not removed, even though selected for removal.",
-                f"● '{KLIPPER_ENV_DIR}' was not removed, even though selected for removal.",
-            ]
-        else:
-            if remove_dir:
-                Logger.print_status("Removing Klipper local repository ...")
-                if run_remove_routines(KLIPPER_DIR):
-                    completion_msg.text.append("● Klipper local repository removed")
-            if remove_env:
-                Logger.print_status("Removing Klipper Python environment ...")
-                if run_remove_routines(KLIPPER_ENV_DIR):
-                    completion_msg.text.append("● Klipper Python environment removed")
-
-        if completion_msg.text:
-            completion_msg.text.insert(0, "The following actions were performed:")
-        else:
-            completion_msg.color = Color.YELLOW
-            completion_msg.centered = True
-            completion_msg.text = ["Nothing to remove."]
-
-        self.msgsvc.set_message(completion_msg)
-
-    def __get_install_count_and_name_dict(self) -> Tuple[int, Dict[int, str]]:
-        install_count: int | None
-        if len(self.moonraker_list) > len(self.klipper_list):
-            install_count = len(self.moonraker_list)
-            name_dict = {
-                i: moonraker.suffix for i, moonraker in enumerate(self.moonraker_list)
-            }
-        else:
-            install_count = get_install_count()
-            name_dict = {
-                i: klipper.suffix for i, klipper in enumerate(self.klipper_list)
-            }
-
-            if install_count is None:
-                Logger.print_status(EXIT_KLIPPER_SETUP)
-                return 0, {}
-
-        return install_count, name_dict
-
-    def __run_setup(self, name_dict: Dict[int, str], create_example_cfg: bool) -> None:
-        if not self.klipper_list:
-            self.__install_deps()
-
-        for i in name_dict:
-            # skip this iteration if there is already an instance with the name
-            if name_dict[i] in [n.suffix for n in self.klipper_list]:
-                continue
-
-            instance = Klipper(suffix=name_dict[i])
-            instance.create()
-            InstanceManager.enable(instance)
-
-            if create_example_cfg:
-                # if a client-config is installed, include it in the new example cfg
-                clients = get_existing_clients()
-                create_example_printer_cfg(instance, clients)
-
-            InstanceManager.start(instance)
-
-        cmd_sysctl_manage("daemon-reload")
-
-        # step 4: check/handle conflicting packages/services
-        handle_disruptive_system_packages()
-
-        # step 5: check for required group membership
-        check_user_groups()
-
-    def __install_deps(self) -> None:
-        repo = self.settings.klipper.repo_url
-        branch = self.settings.klipper.branch
-
-        git_clone_wrapper(repo, KLIPPER_DIR, branch)
-
-        try:
-            install_klipper_packages()
-            if create_python_venv(KLIPPER_ENV_DIR):
-                install_python_requirements(KLIPPER_ENV_DIR, KLIPPER_REQ_FILE)
-        except Exception:
-            Logger.print_error("Error during installation of Klipper requirements!")
-            raise
-
-    def __display_moonraker_info(self) -> bool:
-        # todo: only show the klipper instances that are not already installed
-        Logger.print_dialog(
-            DialogType.INFO,
-            [
-                "Existing Moonraker instances detected:",
-                *[f"● {m.service_file_path.stem}" for m in self.moonraker_list],
-                "\n\n",
-                "The following Klipper instances will be installed:",
-                *[f"● klipper-{m.suffix}" for m in self.moonraker_list],
-            ],
-        )
-        _input: bool = get_confirm("Proceed with installation?")
-        return _input
-
-    def __handle_instance_names(
-        self, install_count: int, name_dict: Dict[int, str], custom_names: bool
-    ) -> None:
-        for i in range(install_count):  # 3
-            key: int = len(name_dict.keys()) + 1
-            if custom_names:
-                assign_custom_name(key, name_dict)
-            else:
-                name_dict[key] = str(len(name_dict) + 1)
-
-    def __use_custom_names_or_go_back(self) -> bool | None:
-        print_select_custom_name_dialog()
-        _input: bool | None = get_confirm(
-            "Assign custom names?",
-            False,
-            allow_go_back=True,
-        )
-        return _input
-
-    def __get_instances_to_remove(self) -> List[Klipper] | None:
-        start_index = 1
-        curr_instances: List[Klipper] = self.klipper_list
-        instance_count = len(curr_instances)
-
-        options = [str(i + start_index) for i in range(instance_count)]
-        options.extend(["a", "b"])
-        instance_map = {options[i]: self.klipper_list[i] for i in range(instance_count)}
-
-        print_instance_overview(
-            self.klipper_list,
-            start_index=start_index,
-            show_index=True,
-            show_select_all=True,
-        )
-        selection = get_selection_input("Select Klipper instance to remove", options)
-
-        if selection == "b":
-            return None
-        elif selection == "a":
-            return copy(self.klipper_list)
-
-        return [instance_map[selection]]
-
-    def __remove_instances(
-        self,
-        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)
-            self.__delete_klipper_env_file(instance)
-
-        self.__refresh_state()
-
-    def __delete_klipper_env_file(self, instance: Klipper):
-        Logger.print_status(f"Remove '{instance.env_file}'")
-        if not instance.env_file.exists():
-            msg = f"Env file in {instance.base.sysd_dir} not found. Skipped ..."
-            Logger.print_info(msg)
-            return
-        run_remove_routines(instance.env_file)

kiauh/components/klipper_firmware/__init__.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #

kiauh/components/klipper_firmware/firmware_utils.py

@@ -1,22 +1,13 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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 re
+
-from pathlib import Path
+from subprocess import PIPE, STDOUT, CalledProcessError, Popen, check_output, run
-from subprocess import (
-    DEVNULL,
-    PIPE,
-    STDOUT,
-    CalledProcessError,
-    Popen,
-    check_output,
-    run,
-)
 from typing import List
 
 from components.klipper import KLIPPER_DIR
@@ -27,32 +18,28 @@ from components.klipper_firmware.flash_options import (
     FlashOptions,
 )
 from core.instance_manager.instance_manager import InstanceManager
-from core.logger import Logger
+from utils.logger import Logger
-from utils.instance_utils import get_instances
 from utils.sys_utils import log_process
 
 
 def find_firmware_file() -> bool:
     target = KLIPPER_DIR.joinpath("out")
-    target_exists: bool = target.exists()
+    target_exists = target.exists()
 
     f1 = "klipper.elf.hex"
     f2 = "klipper.elf"
     f3 = "klipper.bin"
-    f4 = "klipper.uf2"
+    fw_file_exists = (
-    fw_file_exists: bool = (
+        target.joinpath(f1).exists() and target.joinpath(f2).exists()
-        (target.joinpath(f1).exists() and target.joinpath(f2).exists())
+    ) or target.joinpath(f3).exists()
-        or target.joinpath(f3).exists()
-        or target.joinpath(f4).exists()
-    )
 
     return target_exists and fw_file_exists
 
 
 def find_usb_device_by_id() -> List[str]:
     try:
-        command = "find /dev/serial/by-id/*"
+        command = "find /dev/serial/by-id/* 2>/dev/null"
-        output = check_output(command, shell=True, text=True, stderr=DEVNULL)
+        output = check_output(command, shell=True, text=True)
         return output.splitlines()
     except CalledProcessError as e:
         Logger.print_error("Unable to find a USB device!")
@@ -62,14 +49,9 @@ def find_usb_device_by_id() -> List[str]:
 
 def find_uart_device() -> List[str]:
     try:
-        cmd = "find /dev -maxdepth 1"
+        command = '"find /dev -maxdepth 1 -regextype posix-extended -regex "^\/dev\/tty(AMA0|S0)$" 2>/dev/null"'
-        output = check_output(cmd, shell=True, text=True, stderr=DEVNULL)
+        output = check_output(command, shell=True, text=True)
-        device_list = []
+        return output.splitlines()
-        if output:
-            pattern = r"^/dev/tty(AMA0|S0)$"
-            devices = output.splitlines()
-            device_list = [d for d in devices if re.search(pattern, d)]
-        return device_list
     except CalledProcessError as e:
         Logger.print_error("Unable to find a UART device!")
         Logger.print_error(e, prefix=False)
@@ -78,45 +60,25 @@ def find_uart_device() -> List[str]:
 
 def find_usb_dfu_device() -> List[str]:
     try:
-        output = check_output("lsusb", shell=True, text=True, stderr=DEVNULL)
+        command = '"lsusb | grep "DFU" | cut -d " " -f 6 2>/dev/null"'
-        device_list = []
+        output = check_output(command, shell=True, text=True)
-        if output:
+        return output.splitlines()
-            devices = output.splitlines()
-            device_list = [d.split(" ")[5] for d in devices if "DFU" in d]
-        return device_list
-
     except CalledProcessError as e:
         Logger.print_error("Unable to find a USB DFU device!")
         Logger.print_error(e, prefix=False)
         return []
 
 
-def find_usb_rp2_boot_device() -> List[str]:
-    try:
-        output = check_output("lsusb", shell=True, text=True, stderr=DEVNULL)
-        device_list = []
-        if output:
-            devices = output.splitlines()
-            device_list = [d.split(" ")[5] for d in devices if "RP2 Boot" in d]
-        return device_list
-
-    except CalledProcessError as e:
-        Logger.print_error("Unable to find a USB RP2 Boot device!")
-        Logger.print_error(e, prefix=False)
-        return []
-
-
 def get_sd_flash_board_list() -> List[str]:
     if not KLIPPER_DIR.exists() or not SD_FLASH_SCRIPT.exists():
         return []
 
     try:
         cmd = f"{SD_FLASH_SCRIPT} -l"
-        blist: List[str] = check_output(cmd, shell=True, text=True).splitlines()[1:]
+        blist = check_output(cmd, shell=True, text=True)
-        return blist
+        return blist.splitlines()[1:]
     except CalledProcessError as e:
         Logger.print_error(f"An unexpected error occured:\n{e}")
-        return []
 
 
 def start_flash_process(flash_options: FlashOptions) -> None:
@@ -139,7 +101,6 @@ def start_flash_process(flash_options: FlashOptions) -> None:
         if flash_options.flash_method is FlashMethod.REGULAR:
             cmd = [
                 "make",
-                f"KCONFIG_CONFIG={flash_options.selected_kconfig}",
                 flash_options.flash_command.value,
                 f"FLASH_DEVICE={flash_options.selected_mcu}",
             ]
@@ -155,29 +116,29 @@ def start_flash_process(flash_options: FlashOptions) -> None:
         else:
             raise Exception("Invalid value for flash_method!")
 
-        instances = get_instances(Klipper)
+        instance_manager = InstanceManager(Klipper)
-        InstanceManager.stop_all(instances)
+        instance_manager.stop_all_instance()
 
         process = Popen(cmd, cwd=KLIPPER_DIR, stdout=PIPE, stderr=STDOUT, text=True)
         log_process(process)
 
-        InstanceManager.start_all(instances)
+        instance_manager.start_all_instance()
 
         rc = process.returncode
         if rc != 0:
             raise Exception(f"Flashing failed with returncode: {rc}")
         else:
-            Logger.print_ok("Flashing successful!", start="\n", end="\n\n")
+            Logger.print_ok("Flashing successfull!", start="\n", end="\n\n")
 
     except (Exception, CalledProcessError):
         Logger.print_error("Flashing failed!", start="\n")
         Logger.print_error("See the console output above!", end="\n\n")
 
 
-def run_make_clean(kconfig=Path(KLIPPER_DIR.joinpath(".config"))) -> None:
+def run_make_clean() -> None:
     try:
         run(
-            f"make KCONFIG_CONFIG={kconfig} clean",
+            "make clean",
             cwd=KLIPPER_DIR,
             shell=True,
             check=True,
@@ -187,10 +148,10 @@ def run_make_clean(kconfig=Path(KLIPPER_DIR.joinpath(".config"))) -> None:
         raise
 
 
-def run_make_menuconfig(kconfig=Path(KLIPPER_DIR.joinpath(".config"))) -> None:
+def run_make_menuconfig() -> None:
     try:
         run(
-            f"make PYTHON=python3 KCONFIG_CONFIG={kconfig} menuconfig",
+            "make PYTHON=python3 menuconfig",
             cwd=KLIPPER_DIR,
             shell=True,
             check=True,
@@ -200,10 +161,10 @@ def run_make_menuconfig(kconfig=Path(KLIPPER_DIR.joinpath(".config"))) -> None:
         raise
 
 
-def run_make(kconfig=Path(KLIPPER_DIR.joinpath(".config"))) -> None:
+def run_make() -> None:
     try:
         run(
-            f"make PYTHON=python3 KCONFIG_CONFIG={kconfig}",
+            "make PYTHON=python3",
             cwd=KLIPPER_DIR,
             shell=True,
             check=True,

kiauh/components/klipper_firmware/flash_options.py

@@ -1,16 +1,15 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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 dataclasses import field
 from enum import Enum
-from typing import List
+from typing import List, Union
 
 
 class FlashMethod(Enum):
@@ -26,20 +25,18 @@ class FlashCommand(Enum):
 class ConnectionType(Enum):
     USB = "USB"
     USB_DFU = "USB (DFU)"
-    USB_RP2040 = "USB (RP2040)"
     UART = "UART"
 
 
 class FlashOptions:
     _instance = None
-    _flash_method: FlashMethod | None = None
+    _flash_method: Union[FlashMethod, None] = None
-    _flash_command: FlashCommand | None = None
+    _flash_command: Union[FlashCommand, None] = None
-    _connection_type: ConnectionType | None = None
+    _connection_type: Union[ConnectionType, None] = None
     _mcu_list: List[str] = field(default_factory=list)
     _selected_mcu: str = ""
     _selected_board: str = ""
     _selected_baudrate: int = 250000
-    _selected_kconfig: str = ".config"
 
     def __new__(cls, *args, **kwargs):
         if not cls._instance:
@@ -47,31 +44,31 @@ class FlashOptions:
         return cls._instance
 
     @classmethod
-    def destroy(cls) -> None:
+    def destroy(cls):
         cls._instance = None
 
     @property
-    def flash_method(self) -> FlashMethod | None:
+    def flash_method(self) -> Union[FlashMethod, None]:
         return self._flash_method
 
     @flash_method.setter
-    def flash_method(self, value: FlashMethod | None):
+    def flash_method(self, value: Union[FlashMethod, None]):
         self._flash_method = value
 
     @property
-    def flash_command(self) -> FlashCommand | None:
+    def flash_command(self) -> Union[FlashCommand, None]:
         return self._flash_command
 
     @flash_command.setter
-    def flash_command(self, value: FlashCommand | None):
+    def flash_command(self, value: Union[FlashCommand, None]):
         self._flash_command = value
 
     @property
-    def connection_type(self) -> ConnectionType | None:
+    def connection_type(self) -> Union[ConnectionType, None]:
         return self._connection_type
 
     @connection_type.setter
-    def connection_type(self, value: ConnectionType | None):
+    def connection_type(self, value: Union[ConnectionType, None]):
         self._connection_type = value
 
     @property
@@ -105,11 +102,3 @@ class FlashOptions:
     @selected_baudrate.setter
     def selected_baudrate(self, value: int) -> None:
         self._selected_baudrate = value
-
-    @property
-    def selected_kconfig(self) -> str:
-        return self._selected_kconfig
-
-    @selected_kconfig.setter
-    def selected_kconfig(self, value: str) -> None:
-        self._selected_kconfig = value

kiauh/components/klipper_firmware/menus/klipper_build_menu.py

@@ -1,30 +1,25 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
-from pathlib import Path
+from typing import Optional, Type
-from shutil import copyfile
-from typing import List, Set, Type
 
-from components.klipper import KLIPPER_DIR, KLIPPER_KCONFIGS_DIR
+from components.klipper import KLIPPER_DIR
 from components.klipper_firmware.firmware_utils import (
     run_make,
     run_make_clean,
     run_make_menuconfig,
 )
-from components.klipper_firmware.flash_options import FlashOptions
-from core.logger import DialogType, Logger
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import COLOR_CYAN, COLOR_GREEN, COLOR_RED, RESET_FORMAT
-from utils.input_utils import get_confirm, get_string_input
+from utils.logger import Logger
 from utils.sys_utils import (
     check_package_install,
     install_system_packages,
@@ -32,153 +27,59 @@ from utils.sys_utils import (
 )
 
 
-# noinspection PyUnusedLocal
-# noinspection PyMethodMayBeStatic
-class KlipperKConfigMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
-        super().__init__()
-        self.title = "Firmware Config Menu"
-        self.title_color = Color.CYAN
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
-        self.flash_options = FlashOptions()
-        self.kconfigs_dirname = KLIPPER_KCONFIGS_DIR
-        self.kconfig_default = KLIPPER_DIR.joinpath(".config")
-        self.configs: List[Path] = []
-        self.kconfig = (
-            self.kconfig_default if not Path(self.kconfigs_dirname).is_dir() else None
-        )
-
-    def run(self) -> None:
-        if not self.kconfig:
-            super().run()
-        else:
-            self.flash_options.selected_kconfig = self.kconfig
-
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
-        from core.menus.advanced_menu import AdvancedMenu
-
-        self.previous_menu = (
-            previous_menu if previous_menu is not None else AdvancedMenu
-        )
-
-    def set_options(self) -> None:
-        if not Path(self.kconfigs_dirname).is_dir():
-            return
-
-        self.input_label_txt = "Select config or action to continue (default=N)"
-        self.default_option = Option(
-            method=self.select_config, opt_data=self.kconfig_default
-        )
-
-        option_index = 1
-        for kconfig in Path(self.kconfigs_dirname).iterdir():
-            if not kconfig.name.endswith(".config"):
-                continue
-            kconfig_path = self.kconfigs_dirname.joinpath(kconfig)
-            if Path(kconfig_path).is_file():
-                self.configs += [kconfig]
-                self.options[str(option_index)] = Option(
-                    method=self.select_config, opt_data=kconfig_path
-                )
-                option_index += 1
-        self.options["n"] = Option(
-            method=self.select_config, opt_data=self.kconfig_default
-        )
-
-    def print_menu(self) -> None:
-        cfg_found_str = Color.apply(
-            "Previously saved firmware configs found!", Color.GREEN
-        )
-        menu = textwrap.dedent(
-            f"""
-            ╟───────────────────────────────────────────────────────╢
-            ║ {cfg_found_str:^62} ║
-            ║                                                       ║
-            ║    Select an existing config or create a new one.     ║
-            ╟───────────────────────────────────────────────────────╢
-            ║ Available firmware configs:                           ║
-            """
-        )[1:]
-
-        start_index = 1
-        for i, s in enumerate(self.configs):
-            line = f"{start_index + i}) {s.name}"
-            menu += f"║ {line:<54}║\n"
-
-        new_config = Color.apply("N) Create new firmware config", Color.GREEN)
-        menu += "║                                                       ║\n"
-        menu += f"║ {new_config:<62} ║\n"
-
-        menu += "╟───────────────────────────────────────────────────────╢\n"
-
-        print(menu, end="")
-
-    def select_config(self, **kwargs) -> None:
-        selection: str | None = kwargs.get("opt_data", None)
-        if selection is None:
-            raise Exception("opt_data is None")
-        if not Path(selection).is_file() and selection != self.kconfig_default:
-            raise Exception("opt_data does not exists")
-        self.kconfig = selection
-
-
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class KlipperBuildFirmwareMenu(BaseMenu):
-    def __init__(
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
-        self, kconfig: str | None = None, previous_menu: Type[BaseMenu] | None = None
-    ):
         super().__init__()
-        self.title = "Build Firmware Menu"
+        self.previous_menu = previous_menu
-        self.title_color = Color.CYAN
+        self.deps = ["build-essential", "dpkg-dev", "make"]
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
+        self.missing_deps = check_package_install(self.deps)
-        self.deps: Set[str] = {"build-essential", "dpkg-dev", "make"}
-        self.missing_deps: List[str] = check_package_install(self.deps)
-        self.flash_options = FlashOptions()
-        self.kconfigs_dirname = KLIPPER_KCONFIGS_DIR
-        self.kconfig_default = KLIPPER_DIR.joinpath(".config")
-        self.kconfig = self.flash_options.selected_kconfig
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.advanced_menu import AdvancedMenu
 
-        self.previous_menu = (
+        self.previous_menu: Type[BaseMenu] = (
             previous_menu if previous_menu is not None else AdvancedMenu
         )
 
     def set_options(self) -> None:
-        self.input_label_txt = "Press ENTER to install dependencies"
-        self.default_option = Option(method=self.install_missing_deps)
-
-    def run(self):
-        # immediately start the build process if all dependencies are met
         if len(self.missing_deps) == 0:
-            self.start_build_process()
+            self.input_label_txt = "Press ENTER to continue"
+            self.default_option = Option(method=self.start_build_process, menu=False)
         else:
-            super().run()
+            self.input_label_txt = "Press ENTER to install dependencies"
+            self.default_option = Option(method=self.install_missing_deps, menu=False)
 
     def print_menu(self) -> None:
-        txt = Color.apply("Dependencies are missing!", Color.RED)
+        header = " [ Build Firmware Menu ] "
+        color = COLOR_CYAN
+        count = 62 - len(color) - len(RESET_FORMAT)
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ {txt:^62} ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ╟───────────────────────────────────────────────────────╢
+            |-------------------------------------------------------|
-            ║ The following dependencies are required:              ║
+            | The following dependencies are required:              |
-            ║                                                       ║
+            |                                                       |
             """
         )[1:]
 
         for d in self.deps:
-            status_ok = Color.apply("*INSTALLED*", Color.GREEN)
+            status_ok = f"{COLOR_GREEN}*INSTALLED*{RESET_FORMAT}"
-            status_missing = Color.apply("*MISSING*", Color.RED)
+            status_missing = f"{COLOR_RED}*MISSING*{RESET_FORMAT}"
             status = status_missing if d in self.missing_deps else status_ok
-            padding = 40 - len(d) + len(status) + (len(status_ok) - len(status))
+            padding = 39 - len(d) + len(status) + (len(status_ok) - len(status))
-            d = Color.apply(f"● {d}", Color.CYAN)
+            d = f" {COLOR_CYAN}● {d}{RESET_FORMAT}"
-            menu += f"║ {d}{status:>{padding}} ║\n"
+            menu += f"| {d}{status:>{padding}} |\n"
 
-        menu += "║                                                       ║\n"
+        menu += "|                                                       |\n"
-        menu += "╟───────────────────────────────────────────────────────╢\n"
+        if len(self.missing_deps) == 0:
+            line = f"{COLOR_GREEN}All dependencies are met!{RESET_FORMAT}"
+        else:
+            line = f"{COLOR_RED}Dependencies are missing!{RESET_FORMAT}"
+
+        menu += f"| {line:<62} |\n"
 
         print(menu, end="")
 
@@ -196,79 +97,16 @@ class KlipperBuildFirmwareMenu(BaseMenu):
 
     def start_build_process(self, **kwargs) -> None:
         try:
-            run_make_clean(self.kconfig)
+            run_make_clean()
-            run_make_menuconfig(self.kconfig)
+            run_make_menuconfig()
-            run_make(self.kconfig)
+            run_make()
 
             Logger.print_ok("Firmware successfully built!")
             Logger.print_ok(f"Firmware file located in '{KLIPPER_DIR}/out'!")
 
-            if self.kconfig == self.kconfig_default:
-                self.save_firmware_config()
-
         except Exception as e:
             Logger.print_error(e)
             Logger.print_error("Building Klipper Firmware failed!")
 
         finally:
-            if self.previous_menu is not None:
+            self.previous_menu().run()
-                self.previous_menu().run()
-
-    def save_firmware_config(self) -> None:
-        Logger.print_dialog(
-            DialogType.CUSTOM,
-            [
-                "You can save the firmware build configs for multiple MCUs,"
-                " and use them to update the firmware after a Klipper version upgrade"
-            ],
-            custom_title="Save firmware config",
-        )
-        if not get_confirm(
-            "Do you want to save firmware config?", default_choice=False
-        ):
-            return
-
-        filename = self.kconfig_default
-        while True:
-            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 '-'",
-                ],
-            )
-            input_name = get_string_input(
-                "Enter the new firmware config name",
-                regex=r"^[a-z0-9]+([a-z0-9-]*[a-z0-9])?$",
-            )
-            filename = self.kconfigs_dirname.joinpath(f"{input_name}.config")
-
-            if Path(filename).is_file():
-                if get_confirm(
-                    f"Firmware config {input_name} already exists, overwrite?",
-                    default_choice=False,
-                ):
-                    break
-
-            if Path(filename).is_dir():
-                Logger.print_error(f"Path {filename} exists and it's a directory")
-
-            if not Path(filename).exists():
-                break
-
-        if not get_confirm(
-            f"Save firmware config to '{filename}'?", default_choice=True
-        ):
-            Logger.print_info("Aborted saving firmware config ...")
-            return
-
-        if not Path(self.kconfigs_dirname).exists():
-            Path(self.kconfigs_dirname).mkdir()
-
-        copyfile(self.kconfig_default, filename)
-
-        Logger.print_ok()
-        Logger.print_ok(f"Firmware config successfully saved to {filename}")

kiauh/components/klipper_firmware/menus/klipper_flash_error_menu.py

@@ -1,60 +1,60 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
-
 import textwrap
-from typing import Type
+from typing import Optional, Type
 
 from components.klipper_firmware.flash_options import FlashMethod, FlashOptions
 from core.menus import FooterType, Option
-from core.menus.base_menu import BaseMenu, MenuTitleStyle
+from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import COLOR_RED, RESET_FORMAT
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class KlipperNoFirmwareErrorMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "!!! NO FIRMWARE FILE FOUND !!!"
+        self.previous_menu = previous_menu
-        self.title_color = Color.RED
-        self.title_style = MenuTitleStyle.PLAIN
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
 
         self.flash_options = FlashOptions()
         self.footer_type = FooterType.BLANK
         self.input_label_txt = "Press ENTER to go back to [Advanced Menu]"
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         self.previous_menu = previous_menu
 
     def set_options(self) -> None:
-        self.default_option = Option(method=self.go_back)
+        self.default_option = Option(self.go_back, False)
 
     def print_menu(self) -> None:
-        line1 = "Unable to find a compiled firmware file!"
+        header = "!!! NO FIRMWARE FILE FOUND !!!"
+        color = COLOR_RED
+        count = 62 - len(color) - len(RESET_FORMAT)
+        line1 = f"{color}Unable to find a compiled firmware file!{RESET_FORMAT}"
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ {Color.apply(line1, Color.RED):<62} ║
+            | {color}{header:^{count}}{RESET_FORMAT} |
-            ║                                                       ║
+            |-------------------------------------------------------|
-            ║ Make sure, that:                                      ║
+            | {line1:<62} |
-            ║ ● the folder '~/klipper/out' and its content exist    ║
+            |                                                       |
-            ║ ● the folder contains the following file:             ║
+            | Make sure, that:                                      |
+            | ● the folder '~/klipper/out' and its content exist    |
+            | ● the folder contains the following file:             |
             """
         )[1:]
 
         if self.flash_options.flash_method is FlashMethod.REGULAR:
-            menu += "║   ● 'klipper.elf'                                     ║\n"
+            menu += "|   ● 'klipper.elf'                                     |\n"
-            menu += "║   ● 'klipper.elf.hex'                                 ║\n"
+            menu += "|   ● 'klipper.elf.hex'                                 |\n"
         else:
-            menu += "║   ● 'klipper.bin'                                     ║\n"
+            menu += "|   ● 'klipper.bin'                                     |\n"
 
         print(menu, end="")
 
@@ -67,36 +67,38 @@ class KlipperNoFirmwareErrorMenu(BaseMenu):
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class KlipperNoBoardTypesErrorMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "!!! ERROR GETTING BOARD LIST !!!"
+        self.previous_menu = previous_menu
-        self.title_color = Color.RED
-        self.title_style = MenuTitleStyle.PLAIN
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
         self.footer_type = FooterType.BLANK
         self.input_label_txt = "Press ENTER to go back to [Main Menu]"
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         self.previous_menu = previous_menu
 
     def set_options(self) -> None:
-        self.default_option = Option(method=self.go_back)
+        self.default_option = Option(self.go_back, False)
 
     def print_menu(self) -> None:
-        line1 = "Reading the list of supported boards failed!"
+        header = "!!! ERROR GETTING BOARD LIST !!!"
+        color = COLOR_RED
+        count = 62 - len(color) - len(RESET_FORMAT)
+        line1 = f"{color}Reading the list of supported boards failed!{RESET_FORMAT}"
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ {Color.apply(line1, Color.RED):<62} ║
+            | {color}{header:^{count}}{RESET_FORMAT} |
-            ║                                                       ║
+            |-------------------------------------------------------|
-            ║ Make sure, that:                                      ║
+            | {line1:<62} |
-            ║ ● the folder '~/klipper' and all its content exist    ║
+            |                                                       |
-            ║ ● the content of folder '~/klipper' is not currupted  ║
+            | Make sure, that:                                      |
-            ║ ● the file '~/klipper/scripts/flash-sd.py' exist      ║
+            | ● the folder '~/klipper' and all its content exist    |
-            ║ ● your current user has access to those files/folders ║
+            | ● the content of folder '~/klipper' is not currupted  |
-            ║                                                       ║
+            | ● the file '~/klipper/scripts/flash-sd.py' exist      |
-            ║ If in doubt or this process continues to fail, please ║
+            | ● your current user has access to those files/folders |
-            ║ consider to download Klipper again.                   ║
+            |                                                       |
+            | If in doubt or this process continues to fail, please |
+            | consider to download Klipper again.                   |
             """
         )[1:]
         print(menu, end="")

kiauh/components/klipper_firmware/menus/klipper_flash_help_menu.py

@@ -1,37 +1,30 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
-
 import textwrap
-from typing import Tuple, Type
+from typing import Optional, Type
 
-from core.menus.base_menu import BaseMenu, MenuTitleStyle
+from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import COLOR_CYAN, COLOR_YELLOW, RESET_FORMAT
-
-
-def __title_config__() -> Tuple[str, Color, MenuTitleStyle]:
-    return "< ? > Help: Flash MCU < ? >", Color.YELLOW, MenuTitleStyle.PLAIN
 
 
 # noinspection DuplicatedCode
 class KlipperFlashMethodHelpMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title, self.title_color, self.title_style = __title_config__()
+        self.previous_menu = previous_menu
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from components.klipper_firmware.menus.klipper_flash_menu import (
             KlipperFlashMethodMenu,
         )
 
-        self.previous_menu = (
+        self.previous_menu: Type[BaseMenu] = (
             previous_menu if previous_menu is not None else KlipperFlashMethodMenu
         )
 
@@ -39,35 +32,39 @@ class KlipperFlashMethodHelpMenu(BaseMenu):
         pass
 
     def print_menu(self) -> None:
-        subheader1 = Color.apply("Regular flashing method:", Color.CYAN)
+        header = " < ? > Help: Flash MCU < ? > "
-        subheader2 = Color.apply("Updating via SD-Card Update:", Color.CYAN)
+        color = COLOR_YELLOW
+        count = 62 - len(color) - len(RESET_FORMAT)
+        subheader1 = f"{COLOR_CYAN}Regular flashing method:{RESET_FORMAT}"
+        subheader2 = f"{COLOR_CYAN}Updating via SD-Card Update:{RESET_FORMAT}"
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ {subheader1:<62} ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║ The default method to flash controller boards which   ║
+            |-------------------------------------------------------|
-            ║ are connected and updated over USB and not by placing ║
+            | {subheader1:<62} |
-            ║ a compiled firmware file onto an internal SD-Card.    ║
+            | The default method to flash controller boards which   |
-            ║                                                       ║
+            | are connected and updated over USB and not by placing |
-            ║ Common controllers that get flashed that way are:     ║
+            | a compiled firmware file onto an internal SD-Card.    |
-            ║ - Arduino Mega 2560                                   ║
+            |                                                       |
-            ║ - Fysetc F6 / S6 (used without a Display + SD-Slot)   ║
+            | Common controllers that get flashed that way are:     |
-            ║                                                       ║
+            | - Arduino Mega 2560                                   |
-            ║ {subheader2:<62} ║
+            | - Fysetc F6 / S6 (used without a Display + SD-Slot)   |
-            ║ Many popular controller boards ship with a bootloader ║
+            |                                                       |
-            ║ capable of updating the firmware via SD-Card.         ║
+            | {subheader2:<62} |
-            ║ Choose this method if your controller board supports  ║
+            | Many popular controller boards ship with a bootloader |
-            ║ this way of updating. This method ONLY works for up-  ║
+            | capable of updating the firmware via SD-Card.         |
-            ║ grading firmware. The initial flashing procedure must ║
+            | Choose this method if your controller board supports  |
-            ║ be done manually per the instructions that apply to   ║
+            | this way of updating. This method ONLY works for up-  |
-            ║ your controller board.                                ║
+            | grading firmware. The initial flashing procedure must |
-            ║                                                       ║
+            | be done manually per the instructions that apply to   |
-            ║ Common controllers that can be flashed that way are:  ║
+            | your controller board.                                |
-            ║ - BigTreeTech SKR 1.3 / 1.4 (Turbo) / E3 / Mini E3    ║
+            |                                                       |
-            ║ - Fysetc F6 / S6 (used with a Display + SD-Slot)      ║
+            | Common controllers that can be flashed that way are:  |
-            ║ - Fysetc Spider                                       ║
+            | - BigTreeTech SKR 1.3 / 1.4 (Turbo) / E3 / Mini E3    |
-            ║                                                       ║
+            | - Fysetc F6 / S6 (used with a Display + SD-Slot)      |
-            ╟───────────────────────────────────────────────────────╢
+            | - Fysetc Spider                                       |
+            |                                                       |
             """
         )[1:]
         print(menu, end="")
@@ -75,17 +72,16 @@ class KlipperFlashMethodHelpMenu(BaseMenu):
 
 # noinspection DuplicatedCode
 class KlipperFlashCommandHelpMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title, self.title_color, self.title_style = __title_config__()
+        self.previous_menu = previous_menu
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from components.klipper_firmware.menus.klipper_flash_menu import (
             KlipperFlashCommandMenu,
         )
 
-        self.previous_menu = (
+        self.previous_menu: Type[BaseMenu] = (
             previous_menu if previous_menu is not None else KlipperFlashCommandMenu
         )
 
@@ -93,21 +89,26 @@ class KlipperFlashCommandHelpMenu(BaseMenu):
         pass
 
     def print_menu(self) -> None:
-        subheader1 = Color.apply("make flash:", Color.CYAN)
+        header = " < ? > Help: Flash MCU < ? > "
-        subheader2 = Color.apply("make serialflash:", Color.CYAN)
+        color = COLOR_YELLOW
+        count = 62 - len(color) - len(RESET_FORMAT)
+        subheader1 = f"{COLOR_CYAN}make flash:{RESET_FORMAT}"
+        subheader2 = f"{COLOR_CYAN}make serialflash:{RESET_FORMAT}"
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ {subheader1:<62} ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║ The default command to flash controller board, it     ║
+            |-------------------------------------------------------|
-            ║ will detect selected microcontroller and use suitable ║
+            | {subheader1:<62} |
-            ║ tool for flashing it.                                 ║
+            | The default command to flash controller board, it     |
-            ║                                                       ║
+            | will detect selected microcontroller and use suitable |
-            ║ {subheader2:<62} ║
+            | tool for flashing it.                                 |
-            ║ Special command to flash STM32 microcontrollers in    ║
+            |                                                       |
-            ║ DFU mode but connected via serial. stm32flash command ║
+            | {subheader2:<62} |
-            ║ will be used internally.                              ║
+            | Special command to flash STM32 microcontrollers in    |
-            ║                                                       ║
+            | DFU mode but connected via serial. stm32flash command |
+            | will be used internally.                              |
+            |                                                       |
             """
         )[1:]
         print(menu, end="")
@@ -115,17 +116,16 @@ class KlipperFlashCommandHelpMenu(BaseMenu):
 
 # noinspection DuplicatedCode
 class KlipperMcuConnectionHelpMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title, self.title_color, self.title_style = __title_config__()
+        self.previous_menu = previous_menu
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from components.klipper_firmware.menus.klipper_flash_menu import (
             KlipperSelectMcuConnectionMenu,
         )
 
-        self.previous_menu = (
+        self.previous_menu: Type[BaseMenu] = (
             previous_menu
             if previous_menu is not None
             else KlipperSelectMcuConnectionMenu
@@ -135,43 +135,32 @@ class KlipperMcuConnectionHelpMenu(BaseMenu):
         pass
 
     def print_menu(self) -> None:
-        subheader1 = Color.apply("USB:", Color.CYAN)
+        header = " < ? > Help: Flash MCU < ? > "
-        subheader2 = Color.apply("UART:", Color.CYAN)
+        color = COLOR_YELLOW
-        subheader3 = Color.apply("USB DFU:", Color.CYAN)
+        count = 62 - len(color) - len(RESET_FORMAT)
-        subheader4 = Color.apply("USB RP2040 Boot:", Color.CYAN)
+        subheader1 = f"{COLOR_CYAN}USB:{RESET_FORMAT}"
+        subheader2 = f"{COLOR_CYAN}UART:{RESET_FORMAT}"
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ {subheader1:<62} ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║ Selecting USB as the connection method will scan the  ║
+            |-------------------------------------------------------|
-            ║ USB ports for connected controller boards. This will  ║
+            | {subheader1:<62} |
-            ║ be similar to the 'ls /dev/serial/by-id/*' command    ║
+            | Selecting USB as the connection method will scan the  |
-            ║ suggested by the official Klipper documentation for   ║
+            | USB ports for connected controller boards. This will  |
-            ║ determining successfull USB connections!              ║
+            | be similar to the 'ls /dev/serial/by-id/*' command    |
-            ║                                                       ║
+            | suggested by the official Klipper documentation for   |
-            ║ {subheader2:<62} ║
+            | determining successfull USB connections!              |
-            ║ Selecting UART as the connection method will list all ║
+            |                                                       |
-            ║ possible UART serial ports. Note: This method ALWAYS  ║
+            | {subheader2:<62} |
-            ║ returns something as it seems impossible to determine ║
+            | Selecting UART as the connection method will list all |
-            ║ if a valid Klipper controller board is connected or   ║
+            | possible UART serial ports. Note: This method ALWAYS  |
-            ║ not. Because of that, you MUST know which UART serial ║
+            | returns something as it seems impossible to determine |
-            ║ port your controller board is connected to when using ║
+            | if a valid Klipper controller board is connected or   |
-            ║ this connection method.                               ║
+            | not. Because of that, you MUST know which UART serial |
-            ║                                                       ║
+            | port your controller board is connected to when using |
-            ║ {subheader3:<62} ║
+            | this connection method.                               |
-            ║ Selecting USB DFU as the connection method will scan  ║
+            |                                                       |
-            ║ the USB ports for connected controller boards in      ║
-            ║ STM32 DFU mode, which is usually done by holding down ║
-            ║ the BOOT button or setting a special jumper on the    ║
-            ║ board before powering up.                             ║
-            ║                                                       ║
-            ║ {subheader4:<62} ║
-            ║ Selecting USB RP2 Boot as the connection method will  ║
-            ║ scan the USB ports for connected RP2040 controller    ║
-            ║ boards in Boot mode, which is usually done by holding ║
-            ║ down the BOOT button before powering up.              ║
-            ║                                                       ║
-            ╟───────────────────────────────────────────────────────╢
             """
         )[1:]
         print(menu, end="")

kiauh/components/klipper_firmware/menus/klipper_flash_menu.py

@@ -1,24 +1,21 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
 import time
-from pathlib import Path
+from typing import Optional, Type
-from typing import Type
 
 from components.klipper_firmware.firmware_utils import (
     find_firmware_file,
     find_uart_device,
     find_usb_device_by_id,
     find_usb_dfu_device,
-    find_usb_rp2_boot_device,
     get_sd_flash_board_list,
     start_flash_process,
 )
@@ -37,56 +34,59 @@ from components.klipper_firmware.menus.klipper_flash_help_menu import (
     KlipperFlashMethodHelpMenu,
     KlipperMcuConnectionHelpMenu,
 )
-from core.logger import DialogType, Logger
 from core.menus import FooterType, Option
-from core.menus.base_menu import BaseMenu, MenuTitleStyle
+from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import COLOR_CYAN, COLOR_RED, COLOR_YELLOW, RESET_FORMAT
 from utils.input_utils import get_number_input
+from utils.logger import Logger
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class KlipperFlashMethodMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "MCU Flash Menu"
-        self.title_color = Color.CYAN
         self.help_menu = KlipperFlashMethodHelpMenu
         self.input_label_txt = "Select flash method"
         self.footer_type = FooterType.BACK_HELP
         self.flash_options = FlashOptions()
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.advanced_menu import AdvancedMenu
 
-        self.previous_menu = (
+        self.previous_menu: Type[BaseMenu] = (
             previous_menu if previous_menu is not None else AdvancedMenu
         )
 
     def set_options(self) -> None:
         self.options = {
-            "1": Option(self.select_regular),
+            "1": Option(self.select_regular, menu=False),
-            "2": Option(self.select_sdcard),
+            "2": Option(self.select_sdcard, menu=False),
         }
 
     def print_menu(self) -> None:
-        subheader = Color.apply("ATTENTION:", Color.YELLOW)
+        header = " [ MCU Flash Menu ] "
-        subline1 = Color.apply(
+        subheader = f"{COLOR_YELLOW}ATTENTION:{RESET_FORMAT}"
-            "Make sure to select the correct method for the MCU!", Color.YELLOW
+        subline1 = f"{COLOR_YELLOW}Make sure to select the correct method for the  MCU!{RESET_FORMAT}"
-        )
+        subline2 = f"{COLOR_YELLOW}Not all MCUs support both methods!{RESET_FORMAT}"
-        subline2 = Color.apply("Not all MCUs support both methods!", Color.YELLOW)
+
+        color = COLOR_CYAN
+        count = 62 - len(color) - len(RESET_FORMAT)
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ Select the flash method for flashing the MCU.         ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║                                                       ║
+            |-------------------------------------------------------|
-            ║ {subheader:<62} ║
+            | Select the flash method for flashing the MCU.         |
-            ║ {subline1:<62} ║
+            |                                                       |
-            ║ {subline2:<62} ║
+            | {subheader:<62} |
-            ╟───────────────────────────────────────────────────────╢
+            | {subline1:<62} |
-            ║ 1) Regular flashing method                            ║
+            | {subline2:<62} |
-            ║ 2) Updating via SD-Card Update                        ║
+            |-------------------------------------------------------|
-            ╟───────────────────────────┬───────────────────────────╢
+            |                                                       |
+            | 1) Regular flashing method                            |
+            | 2) Updating via SD-Card Update                        |
+            |                                                       |
             """
         )[1:]
         print(menu, end="")
@@ -109,35 +109,34 @@ class KlipperFlashMethodMenu(BaseMenu):
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class KlipperFlashCommandMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "Which flash command to use for flashing the MCU?"
-        self.title_style = MenuTitleStyle.PLAIN
-        self.title_color = Color.YELLOW
         self.help_menu = KlipperFlashCommandHelpMenu
         self.input_label_txt = "Select flash command"
         self.footer_type = FooterType.BACK_HELP
         self.flash_options = FlashOptions()
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
-        self.previous_menu = (
+        self.previous_menu: Type[BaseMenu] = (
             previous_menu if previous_menu is not None else KlipperFlashMethodMenu
         )
 
     def set_options(self) -> None:
         self.options = {
-            "1": Option(self.select_flash),
+            "1": Option(self.select_flash, menu=False),
-            "2": Option(self.select_serialflash),
+            "2": Option(self.select_serialflash, menu=False),
         }
-        self.default_option = Option(self.select_flash)
+        self.default_option = Option(self.select_flash, menu=False)
 
     def print_menu(self) -> None:
         menu = textwrap.dedent(
             """
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ 1) make flash (default)                               ║
+            |                                                       |
-            ║ 2) make serialflash (stm32flash)                      ║
+            | Which flash command to use for flashing the MCU?      |
-            ╟───────────────────────────┬───────────────────────────╢
+            | 1) make flash (default)                               |
+            | 2) make serialflash (stm32flash)                      |
+            |                                                       |
             """
         )[1:]
         print(menu, end="")
@@ -158,43 +157,43 @@ class KlipperFlashCommandMenu(BaseMenu):
 # noinspection PyMethodMayBeStatic
 class KlipperSelectMcuConnectionMenu(BaseMenu):
     def __init__(
-        self, previous_menu: Type[BaseMenu] | None = None, standalone: bool = False
+        self, previous_menu: Optional[Type[BaseMenu]] = None, standalone: bool = False
     ):
         super().__init__()
-        self.title = "Make sure that the controller board is connected now!"
+        self.previous_menu = previous_menu
-        self.title_style = MenuTitleStyle.PLAIN
-        self.title_color = Color.YELLOW
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
         self.__standalone = standalone
         self.help_menu = KlipperMcuConnectionHelpMenu
         self.input_label_txt = "Select connection type"
         self.footer_type = FooterType.BACK_HELP
         self.flash_options = FlashOptions()
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
-        self.previous_menu = (
+        self.previous_menu: Type[BaseMenu] = (
             previous_menu if previous_menu is not None else KlipperFlashCommandMenu
         )
 
     def set_options(self) -> None:
         self.options = {
-            "1": Option(method=self.select_usb),
+            "1": Option(method=self.select_usb, menu=False),
-            "2": Option(method=self.select_dfu),
+            "2": Option(method=self.select_dfu, menu=False),
-            "3": Option(method=self.select_usb_dfu),
+            "3": Option(method=self.select_usb_dfu, menu=False),
-            "4": Option(method=self.select_usb_rp2040),
         }
 
     def print_menu(self) -> None:
+        header = "Make sure that the controller board is connected now!"
+        color = COLOR_YELLOW
+        count = 62 - len(color) - len(RESET_FORMAT)
         menu = textwrap.dedent(
-            """
+            f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ How is the controller board connected to the host?    ║
+            | {color}{header:^{count}}{RESET_FORMAT} |
-            ╟───────────────────────────────────────────────────────╢
+            |-------------------------------------------------------|
-            ║ 1) USB                                                ║
+            |                                                       |
-            ║ 2) UART                                               ║
+            | How is the controller board connected to the host?    |
-            ║ 3) USB (DFU mode)                                     ║
+            | 1) USB                                                |
-            ║ 4) USB (RP2040 mode)                                  ║
+            | 2) UART                                               |
-            ╟───────────────────────────┬───────────────────────────╢
+            | 3) USB (DFU mode)                                     |
+            |                                                       |
             """
         )[1:]
         print(menu, end="")
@@ -211,10 +210,6 @@ class KlipperSelectMcuConnectionMenu(BaseMenu):
         self.flash_options.connection_type = ConnectionType.USB_DFU
         self.get_mcu_list()
 
-    def select_usb_rp2040(self, **kwargs):
-        self.flash_options.connection_type = ConnectionType.USB_RP2040
-        self.get_mcu_list()
-
     def get_mcu_list(self, **kwargs):
         conn_type = self.flash_options.connection_type
 
@@ -227,11 +222,6 @@ class KlipperSelectMcuConnectionMenu(BaseMenu):
         elif conn_type is ConnectionType.USB_DFU:
             Logger.print_status("Identifying MCU connected via USB in DFU mode ...")
             self.flash_options.mcu_list = find_usb_dfu_device()
-        elif conn_type is ConnectionType.USB_RP2040:
-            Logger.print_status(
-                "Identifying MCU connected via USB in RP2 Boot mode ..."
-            )
-            self.flash_options.mcu_list = find_usb_rp2_boot_device()
 
         if len(self.flash_options.mcu_list) < 1:
             Logger.print_warn("No MCUs found!")
@@ -241,7 +231,7 @@ class KlipperSelectMcuConnectionMenu(BaseMenu):
         if self.__standalone and len(self.flash_options.mcu_list) > 0:
             Logger.print_ok("The following MCUs were found:", prefix=False)
             for i, mcu in enumerate(self.flash_options.mcu_list):
-                print(f"   ● MCU #{i}: {Color.CYAN}{mcu}{Color.RST}")
+                print(f"   ● MCU #{i}: {COLOR_CYAN}{mcu}{RESET_FORMAT}")
             time.sleep(3)
             return
 
@@ -254,18 +244,15 @@ class KlipperSelectMcuConnectionMenu(BaseMenu):
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class KlipperSelectMcuIdMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "!!! ATTENTION !!!"
-        self.title_style = MenuTitleStyle.PLAIN
-        self.title_color = Color.RED
         self.flash_options = FlashOptions()
         self.mcu_list = self.flash_options.mcu_list
         self.input_label_txt = "Select MCU to flash"
-        self.footer_type = FooterType.BACK
+        self.footer_type = FooterType.BACK_HELP
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
-        self.previous_menu = (
+        self.previous_menu: Type[BaseMenu] = (
             previous_menu
             if previous_menu is not None
             else KlipperSelectMcuConnectionMenu
@@ -273,70 +260,62 @@ class KlipperSelectMcuIdMenu(BaseMenu):
 
     def set_options(self) -> None:
         self.options = {
-            f"{i}": Option(self.flash_mcu, f"{i}") for i in range(len(self.mcu_list))
+            f"{i}": Option(self.flash_mcu, False, f"{i}")
+            for i in range(len(self.mcu_list))
         }
 
     def print_menu(self) -> None:
-        header2 = f"[{Color.apply('List of detected MCUs', Color.CYAN)}]"
+        header = "!!! ATTENTION !!!"
+        header2 = f"[{COLOR_CYAN}List of available MCUs{RESET_FORMAT}]"
+        color = COLOR_RED
+        count = 62 - len(color) - len(RESET_FORMAT)
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ Make sure, to select the correct MCU!                 ║
+            | {color}{header:^{count}}{RESET_FORMAT} |
-            ║ ONLY flash a firmware created for the respective MCU! ║
+            |-------------------------------------------------------|
-            ║                                                       ║
+            | Make sure, to select the correct MCU!                 |
-            ╟{header2:─^64}╢
+            | ONLY flash a firmware created for the respective MCU! |
-            ║                                                       ║
+            |                                                       |
+            |{header2:-^64}|
+            
             """
         )[1:]
 
         for i, mcu in enumerate(self.mcu_list):
             mcu = mcu.split("/")[-1]
-            menu += f"║ {i}) {Color.apply(f'{mcu:<51}', Color.CYAN)}║\n"
+            menu += f"   ● MCU #{i}: {COLOR_CYAN}{mcu}{RESET_FORMAT}\n"
 
-        menu += textwrap.dedent(
+        print(menu, end="\n")
-            """
-            ║                                                       ║
-            ╟───────────────────────────────────────────────────────╢
-            """
-        )[1:]
-        print(menu, end="")
 
     def flash_mcu(self, **kwargs):
-        try:
+        index = int(kwargs.get("opt_index"))
-            index: int | None = kwargs.get("opt_index", None)
+        selected_mcu = self.mcu_list[index]
-            if index is None:
+        self.flash_options.selected_mcu = selected_mcu
-                raise Exception("opt_index is None")
 
-            index = int(index)
+        if self.flash_options.flash_method == FlashMethod.SD_CARD:
-            selected_mcu = self.mcu_list[index]
+            KlipperSelectSDFlashBoardMenu(previous_menu=self.__class__).run()
-            self.flash_options.selected_mcu = selected_mcu
+        elif self.flash_options.flash_method == FlashMethod.REGULAR:
-
+            KlipperFlashOverviewMenu(previous_menu=self.__class__).run()
-            if self.flash_options.flash_method == FlashMethod.SD_CARD:
-                KlipperSelectSDFlashBoardMenu(previous_menu=self.__class__).run()
-            elif self.flash_options.flash_method == FlashMethod.REGULAR:
-                KlipperFlashOverviewMenu(previous_menu=self.__class__).run()
-        except Exception as e:
-            Logger.print_error(e)
-            Logger.print_error("Flashing failed!")
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class KlipperSelectSDFlashBoardMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
         self.flash_options = FlashOptions()
         self.available_boards = get_sd_flash_board_list()
         self.input_label_txt = "Select board type"
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
-        self.previous_menu = (
+        self.previous_menu: Type[BaseMenu] = (
             previous_menu if previous_menu is not None else KlipperSelectMcuIdMenu
         )
 
     def set_options(self) -> None:
         self.options = {
-            f"{i}": Option(self.board_select, f"{i}")
+            f"{i}": Option(self.board_select, False, f"{i}")
             for i in range(len(self.available_boards))
         }
 
@@ -346,43 +325,38 @@ class KlipperSelectSDFlashBoardMenu(BaseMenu):
         else:
             menu = textwrap.dedent(
                 """
-                ║ Please select the type of board that corresponds to   ║
+                /=======================================================\\
-                ║ the currently selected MCU ID you chose before.       ║
+                | Please select the type of board that corresponds to   |
-                ║                                                       ║
+                | the currently selected MCU ID you chose before.       |
-                ║ The following boards are currently supported:         ║
+                |                                                       |
-                ╟───────────────────────────────────────────────────────╢
+                | The following boards are currently supported:         |
+                |-------------------------------------------------------|
                 """
             )[1:]
 
             for i, board in enumerate(self.available_boards):
                 line = f" {i}) {board}"
-                menu += f"║{line:<55}║\n"
+                menu += f"|{line:<55}|\n"
-            menu += "╟───────────────────────────────────────────────────────╢"
+
             print(menu, end="")
 
     def board_select(self, **kwargs):
-        try:
+        board = int(kwargs.get("opt_index"))
-            index: int | None = kwargs.get("opt_index", None)
+        self.flash_options.selected_board = self.available_boards[board]
-            if index is None:
+        self.baudrate_select()
-                raise Exception("opt_index is None")
-
-            index = int(index)
-            self.flash_options.selected_board = self.available_boards[index]
-            self.baudrate_select()
-        except Exception as e:
-            Logger.print_error(e)
-            Logger.print_error("Board selection failed!")
 
     def baudrate_select(self, **kwargs):
-        Logger.print_dialog(
+        menu = textwrap.dedent(
-            DialogType.CUSTOM,
+            """
-            [
+            /=======================================================\\
-                "If your board is flashed with firmware that connects "
+            | If your board is flashed with firmware that connects  |
-                "at a custom baud rate, please change it now.",
+            | at a custom baud rate, please change it now.          |
-                "\n\n",
+            |                                                       |
-                "If you are unsure, stick to the default 250000!",
+            | If you are unsure, stick to the default 250000!       |
-            ],
+            \\=======================================================/
-        )
+            """
+        )[1:]
+        print(menu, end="")
         self.flash_options.selected_baudrate = get_number_input(
             question="Please set the baud rate",
             default=250000,
@@ -395,81 +369,65 @@ class KlipperSelectSDFlashBoardMenu(BaseMenu):
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class KlipperFlashOverviewMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "!!! ATTENTION !!!"
-        self.title_style = MenuTitleStyle.PLAIN
-        self.title_color = Color.RED
         self.flash_options = FlashOptions()
         self.input_label_txt = "Perform action (default=Y)"
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
+        self.previous_menu: Type[BaseMenu] = previous_menu
 
     def set_options(self) -> None:
         self.options = {
-            "y": Option(self.execute_flash),
+            "Y": Option(self.execute_flash, menu=False),
-            "n": Option(self.abort_process),
+            "N": Option(self.abort_process, menu=False),
         }
 
-        self.default_option = Option(self.execute_flash)
+        self.default_option = Option(self.execute_flash, menu=False)
 
     def print_menu(self) -> None:
+        header = "!!! ATTENTION !!!"
+        color = COLOR_RED
+        count = 62 - len(color) - len(RESET_FORMAT)
+
         method = self.flash_options.flash_method.value
         command = self.flash_options.flash_command.value
         conn_type = self.flash_options.connection_type.value
-        mcu = self.flash_options.selected_mcu.split("/")[-1]
+        mcu = self.flash_options.selected_mcu
         board = self.flash_options.selected_board
         baudrate = self.flash_options.selected_baudrate
-        kconfig = Path(self.flash_options.selected_kconfig).name
+        subheader = f"[{COLOR_CYAN}Overview{RESET_FORMAT}]"
-        color = Color.CYAN
-        subheader = f"[{Color.apply('Overview', color)}]"
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ Before contuining the flashing process, please check  ║
+            | {color}{header:^{count}}{RESET_FORMAT} |
-            ║ if all parameters were set correctly! Once you made   ║
+            |-------------------------------------------------------|
-            ║ sure everything is correct, start the process. If any ║
+            | Before contuining the flashing process, please check  |
-            ║ parameter needs to be changed, you can go back (B)    ║
+            | if all parameters were set correctly! Once you made   |
-            ║ step by step or abort and start from the beginning.   ║
+            | sure everything is correct, start the process. If any |
-            ║{subheader:─^64}║
+            | parameter needs to be changed, you can go back (B)    |
-            ║                                                       ║
+            | step by step or abort and start from the beginning.   |
+            |{subheader:-^64}|
+            
             """
         )[1:]
 
-        menu += textwrap.dedent(
+        menu += f"   ● MCU: {COLOR_CYAN}{mcu}{RESET_FORMAT}\n"
-            f"""
+        menu += f"   ● Connection: {COLOR_CYAN}{conn_type}{RESET_FORMAT}\n"
-            ║ MCU: {Color.apply(f"{mcu:<48}", color)} ║
+        menu += f"   ● Flash method: {COLOR_CYAN}{method}{RESET_FORMAT}\n"
-            ║ Connection: {Color.apply(f"{conn_type:<41}", color)} ║
+        menu += f"   ● Flash command: {COLOR_CYAN}{command}{RESET_FORMAT}\n"
-            ║ Flash method: {Color.apply(f"{method:<39}", color)} ║
-            ║ Flash command: {Color.apply(f"{command:<38}", color)} ║
-            """
-        )[1:]
 
         if self.flash_options.flash_method is FlashMethod.SD_CARD:
-            menu += textwrap.dedent(
+            menu += f"   ● Board type: {COLOR_CYAN}{board}{RESET_FORMAT}\n"
-                f"""
+            menu += f"   ● Baudrate: {COLOR_CYAN}{baudrate}{RESET_FORMAT}\n"
-                ║ Board type: {Color.apply(f"{board:<41}", color)} ║
-                ║ Baudrate: {Color.apply(f"{baudrate:<43}", color)} ║
-                """
-            )[1:]
-
-        if self.flash_options.flash_method is FlashMethod.REGULAR:
-            menu += textwrap.dedent(
-                f"""
-                ║ Firmware config: {Color.apply(f"{kconfig:<36}", color)} ║
-                """
-            )[1:]
 
         menu += textwrap.dedent(
             """
-            ║                                                       ║
+            |-------------------------------------------------------|
-            ╟───────────────────────────────────────────────────────╢
+            |  Y) Start flash process                               |
-            ║  Y) Start flash process                               ║
+            |  N) Abort - Return to Advanced Menu                   |
-            ║  N) Abort - Return to Advanced Menu                   ║
-            ╟───────────────────────────────────────────────────────╢
             """
-        )[1:]
+        )
         print(menu, end="")
 
     def execute_flash(self, **kwargs):

kiauh/components/klipperscreen/__init__.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -9,26 +9,8 @@
 from pathlib import Path
 
 from core.backup_manager import BACKUP_ROOT_DIR
-from core.constants import SYSTEMD
 
-# repo
 KLIPPERSCREEN_REPO = "https://github.com/KlipperScreen/KlipperScreen.git"
-
-# names
-KLIPPERSCREEN_SERVICE_NAME = "KlipperScreen.service"
-KLIPPERSCREEN_UPDATER_SECTION_NAME = "update_manager KlipperScreen"
-KLIPPERSCREEN_LOG_NAME = "KlipperScreen.log"
-
-# directories
 KLIPPERSCREEN_DIR = Path.home().joinpath("KlipperScreen")
-KLIPPERSCREEN_ENV_DIR = Path.home().joinpath(".KlipperScreen-env")
+KLIPPERSCREEN_ENV = Path.home().joinpath(".KlipperScreen-env")
 KLIPPERSCREEN_BACKUP_DIR = BACKUP_ROOT_DIR.joinpath("klipperscreen-backups")
-
-# files
-KLIPPERSCREEN_REQ_FILE = KLIPPERSCREEN_DIR.joinpath(
-    "scripts/KlipperScreen-requirements.txt"
-)
-KLIPPERSCREEN_INSTALL_SCRIPT = KLIPPERSCREEN_DIR.joinpath(
-    "scripts/KlipperScreen-install.sh"
-)
-KLIPPERSCREEN_SERVICE_FILE = SYSTEMD.joinpath(KLIPPERSCREEN_SERVICE_NAME)

kiauh/components/klipperscreen/klipperscreen.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -15,40 +15,33 @@ from components.klipper.klipper import Klipper
 from components.klipperscreen import (
     KLIPPERSCREEN_BACKUP_DIR,
     KLIPPERSCREEN_DIR,
-    KLIPPERSCREEN_ENV_DIR,
+    KLIPPERSCREEN_ENV,
-    KLIPPERSCREEN_INSTALL_SCRIPT,
-    KLIPPERSCREEN_LOG_NAME,
     KLIPPERSCREEN_REPO,
-    KLIPPERSCREEN_REQ_FILE,
-    KLIPPERSCREEN_SERVICE_FILE,
-    KLIPPERSCREEN_SERVICE_NAME,
-    KLIPPERSCREEN_UPDATER_SECTION_NAME,
 )
 from components.moonraker.moonraker import Moonraker
 from core.backup_manager.backup_manager import BackupManager
-from core.constants import SYSTEMD
 from core.instance_manager.instance_manager import InstanceManager
-from core.logger import DialogType, Logger
 from core.settings.kiauh_settings import KiauhSettings
-from core.types.component_status import ComponentStatus
 from utils.common import (
     check_install_dependencies,
     get_install_status,
 )
 from utils.config_utils import add_config_section, remove_config_section
+from utils.constants import SYSTEMD
 from utils.fs_utils import remove_with_sudo
 from utils.git_utils import (
     git_clone_wrapper,
     git_pull_wrapper,
 )
 from utils.input_utils import get_confirm
-from utils.instance_utils import get_instances
+from utils.logger import DialogType, Logger
 from utils.sys_utils import (
     check_python_version,
+    cmd_sysctl_manage,
     cmd_sysctl_service,
     install_python_requirements,
-    remove_system_service,
 )
+from utils.types import ComponentStatus
 
 
 def install_klipperscreen() -> None:
@@ -57,7 +50,8 @@ def install_klipperscreen() -> None:
     if not check_python_version(3, 7):
         return
 
-    mr_instances = get_instances(Moonraker)
+    mr_im = InstanceManager(Moonraker)
+    mr_instances = mr_im.instances
     if not mr_instances:
         Logger.print_dialog(
             DialogType.WARNING,
@@ -68,6 +62,7 @@ def install_klipperscreen() -> None:
                 "KlipperScreens update manager configuration for Moonraker "
                 "will not be added to any moonraker.conf.",
             ],
+            end="",
         )
         if not get_confirm(
             "Continue KlipperScreen installation?",
@@ -76,15 +71,17 @@ def install_klipperscreen() -> None:
         ):
             return
 
-    check_install_dependencies()
+    package_list = ["git", "wget", "curl", "unzip", "dfu-util"]
+    check_install_dependencies(package_list)
 
     git_clone_wrapper(KLIPPERSCREEN_REPO, KLIPPERSCREEN_DIR)
 
     try:
-        run(KLIPPERSCREEN_INSTALL_SCRIPT.as_posix(), shell=True, check=True)
+        script = f"{KLIPPERSCREEN_DIR}/scripts/KlipperScreen-install.sh"
+        run(script, shell=True, check=True)
         if mr_instances:
             patch_klipperscreen_update_manager(mr_instances)
-            InstanceManager.restart_all(mr_instances)
+            mr_im.restart_all_instance()
         else:
             Logger.print_info(
                 "Moonraker is not installed! Cannot add "
@@ -97,30 +94,34 @@ def install_klipperscreen() -> None:
 
 
 def patch_klipperscreen_update_manager(instances: List[Moonraker]) -> None:
+    env_py = f"{KLIPPERSCREEN_ENV}/bin/python"
     add_config_section(
-        section=KLIPPERSCREEN_UPDATER_SECTION_NAME,
+        section="update_manager KlipperScreen",
         instances=instances,
         options=[
             ("type", "git_repo"),
-            ("path", KLIPPERSCREEN_DIR.as_posix()),
+            ("path", str(KLIPPERSCREEN_DIR)),
-            ("origin", KLIPPERSCREEN_REPO),
+            ("orgin", KLIPPERSCREEN_REPO),
-            ("managed_services", "KlipperScreen"),
+            ("env", env_py),
-            ("env", f"{KLIPPERSCREEN_ENV_DIR}/bin/python"),
+            ("requirements", "scripts/KlipperScreen-requirements.txt"),
-            ("requirements", KLIPPERSCREEN_REQ_FILE.as_posix()),
+            ("install_script", "scripts/KlipperScreen-install.sh"),
-            ("install_script", KLIPPERSCREEN_INSTALL_SCRIPT.as_posix()),
         ],
     )
 
 
 def update_klipperscreen() -> None:
-    if not KLIPPERSCREEN_DIR.exists():
-        Logger.print_info("KlipperScreen does not seem to be installed! Skipping ...")
-        return
-
     try:
+        cmd_sysctl_service("KlipperScreen", "stop")
+
+        if not KLIPPERSCREEN_DIR.exists():
+            Logger.print_info(
+                "KlipperScreen does not seem to be installed! Skipping ..."
+            )
+            return
+
         Logger.print_status("Updating KlipperScreen ...")
 
-        cmd_sysctl_service(KLIPPERSCREEN_SERVICE_NAME, "stop")
+        cmd_sysctl_service("KlipperScreen", "stop")
 
         settings = KiauhSettings()
         if settings.kiauh.backup_before_update:
@@ -128,9 +129,12 @@ def update_klipperscreen() -> None:
 
         git_pull_wrapper(KLIPPERSCREEN_REPO, KLIPPERSCREEN_DIR)
 
-        install_python_requirements(KLIPPERSCREEN_ENV_DIR, KLIPPERSCREEN_REQ_FILE)
+        requirements = KLIPPERSCREEN_DIR.joinpath(
+            "/scripts/KlipperScreen-requirements.txt"
+        )
+        install_python_requirements(KLIPPERSCREEN_ENV, requirements)
 
-        cmd_sysctl_service(KLIPPERSCREEN_SERVICE_NAME, "start")
+        cmd_sysctl_service("KlipperScreen", "start")
 
         Logger.print_ok("KlipperScreen updated successfully.", end="\n\n")
     except CalledProcessError as e:
@@ -141,8 +145,8 @@ def update_klipperscreen() -> None:
 def get_klipperscreen_status() -> ComponentStatus:
     return get_install_status(
         KLIPPERSCREEN_DIR,
-        KLIPPERSCREEN_ENV_DIR,
+        KLIPPERSCREEN_ENV,
-        files=[SYSTEMD.joinpath(KLIPPERSCREEN_SERVICE_NAME)],
+        files=[SYSTEMD.joinpath("KlipperScreen.service")],
     )
 
 
@@ -156,31 +160,40 @@ def remove_klipperscreen() -> None:
         else:
             Logger.print_warn("KlipperScreen directory not found!")
 
-        if KLIPPERSCREEN_ENV_DIR.exists():
+        if KLIPPERSCREEN_ENV.exists():
             Logger.print_status("Removing KlipperScreen environment ...")
-            shutil.rmtree(KLIPPERSCREEN_ENV_DIR)
+            shutil.rmtree(KLIPPERSCREEN_ENV)
             Logger.print_ok("KlipperScreen environment successfully removed!")
         else:
             Logger.print_warn("KlipperScreen environment not found!")
 
-        if KLIPPERSCREEN_SERVICE_FILE.exists():
+        service = SYSTEMD.joinpath("KlipperScreen.service")
-            remove_system_service(KLIPPERSCREEN_SERVICE_NAME)
+        if service.exists():
+            Logger.print_status("Removing KlipperScreen service ...")
+            cmd_sysctl_service(service, "stop")
+            cmd_sysctl_service(service, "disable")
+            remove_with_sudo(service)
+            cmd_sysctl_manage("daemon-reload")
+            cmd_sysctl_manage("reset-failed")
+            Logger.print_ok("KlipperScreen service successfully removed!")
 
-        logfile = Path(f"/tmp/{KLIPPERSCREEN_LOG_NAME}")
+        logfile = Path("/tmp/KlipperScreen.log")
         if logfile.exists():
             Logger.print_status("Removing KlipperScreen log file ...")
             remove_with_sudo(logfile)
             Logger.print_ok("KlipperScreen log file successfully removed!")
 
-        kl_instances: List[Klipper] = get_instances(Klipper)
+        kl_im = InstanceManager(Klipper)
+        kl_instances: List[Klipper] = kl_im.instances
         for instance in kl_instances:
-            logfile = instance.base.log_dir.joinpath(KLIPPERSCREEN_LOG_NAME)
+            logfile = instance.log_dir.joinpath("KlipperScreen.log")
             if logfile.exists():
                 Logger.print_status(f"Removing {logfile} ...")
                 Path(logfile).unlink()
                 Logger.print_ok(f"{logfile} successfully removed!")
 
-        mr_instances: List[Moonraker] = get_instances(Moonraker)
+        mr_im = InstanceManager(Moonraker)
+        mr_instances: List[Moonraker] = mr_im.instances
         if mr_instances:
             Logger.print_status("Removing KlipperScreen from update manager ...")
             remove_config_section("update_manager KlipperScreen", mr_instances)
@@ -195,12 +208,12 @@ def remove_klipperscreen() -> None:
 def backup_klipperscreen_dir() -> None:
     bm = BackupManager()
     bm.backup_directory(
-        KLIPPERSCREEN_DIR.name,
+        "KlipperScreen",
         source=KLIPPERSCREEN_DIR,
         target=KLIPPERSCREEN_BACKUP_DIR,
     )
     bm.backup_directory(
-        KLIPPERSCREEN_ENV_DIR.name,
+        "KlipperScreen-env",
-        source=KLIPPERSCREEN_ENV_DIR,
+        source=KLIPPERSCREEN_ENV,
         target=KLIPPERSCREEN_BACKUP_DIR,
     )

kiauh/components/log_uploads/__init__.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #

kiauh/components/log_uploads/log_upload_utils.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -13,14 +13,13 @@ from typing import List
 
 from components.klipper.klipper import Klipper
 from components.log_uploads import LogFile
-from core.logger import Logger
+from core.instance_manager.instance_manager import InstanceManager
-from utils.instance_utils import get_instances
+from utils.logger import Logger
 
 
 def get_logfile_list() -> List[LogFile]:
-    log_dirs: List[Path] = [
+    cm = InstanceManager(Klipper)
-        instance.base.log_dir for instance in get_instances(Klipper)
+    log_dirs: List[Path] = [instance.log_dir for instance in cm.instances]
-    ]
 
     logfiles: List[LogFile] = []
     for _dir in log_dirs:

kiauh/components/log_uploads/menus/log_upload_menu.py

@@ -1,67 +1,61 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
-from typing import Type
+from typing import Optional, Type
 
 from components.log_uploads.log_upload_utils import get_logfile_list, upload_logfile
-from core.logger import Logger
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import COLOR_YELLOW, RESET_FORMAT
 
 
 # noinspection PyMethodMayBeStatic
 class LogUploadMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "Log Upload"
+        self.previous_menu = previous_menu
-        self.title_color = Color.YELLOW
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
         self.logfile_list = get_logfile_list()
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.main_menu import MainMenu
 
-        self.previous_menu = previous_menu if previous_menu is not None else MainMenu
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else MainMenu
+        )
 
     def set_options(self) -> None:
         self.options = {
-            f"{index}": Option(self.upload, opt_index=f"{index}")
+            f"{index}": Option(self.upload, False, opt_index=f"{index}")
             for index in range(len(self.logfile_list))
         }
 
-    def print_menu(self) -> None:
+    def print_menu(self):
+        header = " [ Log Upload ] "
+        color = COLOR_YELLOW
+        count = 62 - len(color) - len(RESET_FORMAT)
         menu = textwrap.dedent(
-            """
+            f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ You can select the following logfiles for uploading:  ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║                                                       ║
+            |-------------------------------------------------------|
+            | You can select the following logfiles for uploading:  |
+            |                                                       |
             """
         )[1:]
 
         for logfile in enumerate(self.logfile_list):
             line = f"{logfile[0]}) {logfile[1].get('display_name')}"
-            menu += f"║ {line:<54}║\n"
+            menu += f"| {line:<54}|\n"
-        menu += "╟───────────────────────────────────────────────────────╢\n"
 
         print(menu, end="")
 
     def upload(self, **kwargs):
-        try:
+        index = int(kwargs.get("opt_index"))
-            index: int | None = kwargs.get("opt_index", None)
+        upload_logfile(self.logfile_list[index])
-            if index is None:
-                raise Exception("opt_index is None")
-
-            index = int(index)
-            upload_logfile(self.logfile_list[index])
-        except Exception as e:
-            Logger.print_error(e)
-            Logger.print_error("Log upload failed!")

kiauh/components/mobileraker/__init__.py

@@ -1,31 +1,16 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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 core.backup_manager import BACKUP_ROOT_DIR
-from core.constants import SYSTEMD
 
-# repo
 MOBILERAKER_REPO = "https://github.com/Clon1998/mobileraker_companion.git"
-
-# names
-MOBILERAKER_SERVICE_NAME = "mobileraker.service"
-MOBILERAKER_UPDATER_SECTION_NAME = "update_manager mobileraker"
-MOBILERAKER_LOG_NAME = "mobileraker.log"
-
-# directories
 MOBILERAKER_DIR = Path.home().joinpath("mobileraker_companion")
-MOBILERAKER_ENV_DIR = Path.home().joinpath("mobileraker-env")
+MOBILERAKER_ENV = Path.home().joinpath("mobileraker-env")
 MOBILERAKER_BACKUP_DIR = BACKUP_ROOT_DIR.joinpath("mobileraker-backups")
-
-# files
-MOBILERAKER_INSTALL_SCRIPT = MOBILERAKER_DIR.joinpath("scripts/install.sh")
-MOBILERAKER_REQ_FILE = MOBILERAKER_DIR.joinpath("scripts/mobileraker-requirements.txt")
-MOBILERAKER_SERVICE_FILE = SYSTEMD.joinpath(MOBILERAKER_SERVICE_NAME)

kiauh/components/mobileraker/mobileraker.py

@@ -0,0 +1,211 @@
+# ======================================================================= #
+#  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 subprocess import CalledProcessError, run
+from typing import List
+
+from components.klipper.klipper import Klipper
+from components.mobileraker import (
+    MOBILERAKER_BACKUP_DIR,
+    MOBILERAKER_DIR,
+    MOBILERAKER_ENV,
+    MOBILERAKER_REPO,
+)
+from components.moonraker.moonraker import Moonraker
+from core.backup_manager.backup_manager import BackupManager
+from core.instance_manager.instance_manager import InstanceManager
+from core.settings.kiauh_settings import KiauhSettings
+from utils.common import check_install_dependencies, get_install_status
+from utils.config_utils import add_config_section, remove_config_section
+from utils.constants import SYSTEMD
+from utils.fs_utils import remove_with_sudo
+from utils.git_utils import (
+    git_clone_wrapper,
+    git_pull_wrapper,
+)
+from utils.input_utils import get_confirm
+from utils.logger import DialogType, Logger
+from utils.sys_utils import (
+    check_python_version,
+    cmd_sysctl_manage,
+    cmd_sysctl_service,
+    install_python_requirements,
+)
+from utils.types import ComponentStatus
+
+
+def install_mobileraker() -> None:
+    Logger.print_status("Installing Mobileraker's companion ...")
+
+    if not check_python_version(3, 7):
+        return
+
+    mr_im = InstanceManager(Moonraker)
+    mr_instances = mr_im.instances
+    if not mr_instances:
+        Logger.print_dialog(
+            DialogType.WARNING,
+            [
+                "Moonraker not found! Mobileraker's companion will not properly work "
+                "without a working Moonraker installation.",
+                "Mobileraker's companion's update manager configuration for Moonraker "
+                "will not be added to any moonraker.conf.",
+            ],
+            end="",
+        )
+        if not get_confirm(
+            "Continue Mobileraker's companion installation?",
+            default_choice=False,
+            allow_go_back=True,
+        ):
+            return
+
+    package_list = ["git", "wget", "curl", "unzip", "dfu-util"]
+    check_install_dependencies(package_list)
+
+    git_clone_wrapper(MOBILERAKER_REPO, MOBILERAKER_DIR)
+
+    try:
+        script = f"{MOBILERAKER_DIR}/scripts/install.sh"
+        run(script, shell=True, check=True)
+        if mr_instances:
+            patch_mobileraker_update_manager(mr_instances)
+            mr_im.restart_all_instance()
+        else:
+            Logger.print_info(
+                "Moonraker is not installed! Cannot add Mobileraker's "
+                "companion to update manager!"
+            )
+        Logger.print_ok("Mobileraker's companion successfully installed!")
+    except CalledProcessError as e:
+        Logger.print_error(f"Error installing Mobileraker's companion:\n{e}")
+        return
+
+
+def patch_mobileraker_update_manager(instances: List[Moonraker]) -> None:
+    env_py = f"{MOBILERAKER_ENV}/bin/python"
+    add_config_section(
+        section="update_manager mobileraker",
+        instances=instances,
+        options=[
+            ("type", "git_repo"),
+            ("path", "mobileraker_companion"),
+            ("orgin", MOBILERAKER_REPO),
+            ("primary_branch", "main"),
+            ("managed_services", "mobileraker"),
+            ("env", env_py),
+            ("requirements", "scripts/mobileraker-requirements.txt"),
+            ("install_script", "scripts/install.sh"),
+        ],
+    )
+
+
+def update_mobileraker() -> None:
+    try:
+        if not MOBILERAKER_DIR.exists():
+            Logger.print_info(
+                "Mobileraker's companion does not seem to be installed! Skipping ..."
+            )
+            return
+
+        Logger.print_status("Updating Mobileraker's companion ...")
+
+        cmd_sysctl_service("mobileraker", "stop")
+
+        settings = KiauhSettings()
+        if settings.kiauh.backup_before_update:
+            backup_mobileraker_dir()
+
+        git_pull_wrapper(MOBILERAKER_REPO, MOBILERAKER_DIR)
+
+        requirements = MOBILERAKER_DIR.joinpath("/scripts/mobileraker-requirements.txt")
+        install_python_requirements(MOBILERAKER_ENV, requirements)
+
+        cmd_sysctl_service("mobileraker", "start")
+
+        Logger.print_ok("Mobileraker's companion updated successfully.", end="\n\n")
+    except CalledProcessError as e:
+        Logger.print_error(f"Error updating Mobileraker's companion:\n{e}")
+        return
+
+
+def get_mobileraker_status() -> ComponentStatus:
+    return get_install_status(
+        MOBILERAKER_DIR,
+        MOBILERAKER_ENV,
+        files=[SYSTEMD.joinpath("mobileraker.service")],
+    )
+
+
+def remove_mobileraker() -> None:
+    Logger.print_status("Removing Mobileraker's companion ...")
+    try:
+        if MOBILERAKER_DIR.exists():
+            Logger.print_status("Removing Mobileraker's companion directory ...")
+            shutil.rmtree(MOBILERAKER_DIR)
+            Logger.print_ok("Mobileraker's companion directory successfully removed!")
+        else:
+            Logger.print_warn("Mobileraker's companion directory not found!")
+
+        if MOBILERAKER_ENV.exists():
+            Logger.print_status("Removing Mobileraker's companion environment ...")
+            shutil.rmtree(MOBILERAKER_ENV)
+            Logger.print_ok("Mobileraker's companion environment successfully removed!")
+        else:
+            Logger.print_warn("Mobileraker's companion environment not found!")
+
+        service = SYSTEMD.joinpath("mobileraker.service")
+        if service.exists():
+            Logger.print_status("Removing mobileraker service ...")
+            cmd_sysctl_service(service, "stop")
+            cmd_sysctl_service(service, "disable")
+            remove_with_sudo(service)
+            cmd_sysctl_manage("daemon-reload")
+            cmd_sysctl_manage("reset-failed")
+            Logger.print_ok("Mobileraker's companion service successfully removed!")
+
+        kl_im = InstanceManager(Klipper)
+        kl_instances: List[Klipper] = kl_im.instances
+        for instance in kl_instances:
+            logfile = instance.log_dir.joinpath("mobileraker.log")
+            if logfile.exists():
+                Logger.print_status(f"Removing {logfile} ...")
+                Path(logfile).unlink()
+                Logger.print_ok(f"{logfile} successfully removed!")
+
+        mr_im = InstanceManager(Moonraker)
+        mr_instances: List[Moonraker] = mr_im.instances
+        if mr_instances:
+            Logger.print_status(
+                "Removing Mobileraker's companion from update manager ..."
+            )
+            remove_config_section("update_manager mobileraker", mr_instances)
+            Logger.print_ok(
+                "Mobileraker's companion successfully removed from update manager!"
+            )
+
+        Logger.print_ok("Mobileraker's companion successfully removed!")
+
+    except Exception as e:
+        Logger.print_error(f"Error removing Mobileraker's companion:\n{e}")
+
+
+def backup_mobileraker_dir() -> None:
+    bm = BackupManager()
+    bm.backup_directory(
+        "mobileraker_companion",
+        source=MOBILERAKER_DIR,
+        target=MOBILERAKER_BACKUP_DIR,
+    )
+    bm.backup_directory(
+        "mobileraker-env",
+        source=MOBILERAKER_ENV,
+        target=MOBILERAKER_BACKUP_DIR,
+    )

kiauh/components/moonraker/__init__.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -13,35 +13,21 @@ from core.backup_manager import BACKUP_ROOT_DIR
 
 MODULE_PATH = Path(__file__).resolve().parent
 
-MOONRAKER_REPO_URL = "https://github.com/Arksine/moonraker.git"
-
-# names
-MOONRAKER_CFG_NAME = "moonraker.conf"
-MOONRAKER_LOG_NAME = "moonraker.log"
-MOONRAKER_SERVICE_NAME = "moonraker.service"
-MOONRAKER_DEFAULT_PORT = 7125
-MOONRAKER_ENV_FILE_NAME = "moonraker.env"
-
-# directories
 MOONRAKER_DIR = Path.home().joinpath("moonraker")
 MOONRAKER_ENV_DIR = Path.home().joinpath("moonraker-env")
 MOONRAKER_BACKUP_DIR = BACKUP_ROOT_DIR.joinpath("moonraker-backups")
 MOONRAKER_DB_BACKUP_DIR = BACKUP_ROOT_DIR.joinpath("moonraker-db-backups")
+MOONRAKER_REQUIREMENTS_TXT = MOONRAKER_DIR.joinpath(
+    "scripts/moonraker-requirements.txt"
+)
+DEFAULT_MOONRAKER_PORT = 7125
 
-# files
-MOONRAKER_INSTALL_SCRIPT = MOONRAKER_DIR.joinpath("scripts/install-moonraker.sh")
-MOONRAKER_REQ_FILE = MOONRAKER_DIR.joinpath("scripts/moonraker-requirements.txt")
-MOONRAKER_SPEEDUPS_REQ_FILE = MOONRAKER_DIR.joinpath("scripts/moonraker-speedups.txt")
-MOONRAKER_DEPS_JSON_FILE = MOONRAKER_DIR.joinpath("scripts/system-dependencies.json")
 # introduced due to
 # https://github.com/Arksine/moonraker/issues/349
 # https://github.com/Arksine/moonraker/pull/346
 POLKIT_LEGACY_FILE = Path("/etc/polkit-1/localauthority/50-local.d/10-moonraker.pkla")
 POLKIT_FILE = Path("/etc/polkit-1/rules.d/moonraker.rules")
 POLKIT_USR_FILE = Path("/usr/share/polkit-1/rules.d/moonraker.rules")
-POLKIT_SCRIPT = MOONRAKER_DIR.joinpath("scripts/set-policykit-rules.sh")
+POLKIT_SCRIPT = Path.home().joinpath("moonraker/scripts/set-policykit-rules.sh")
-MOONRAKER_SERVICE_TEMPLATE = MODULE_PATH.joinpath(f"assets/{MOONRAKER_SERVICE_NAME}")
-MOONRAKER_ENV_FILE_TEMPLATE = MODULE_PATH.joinpath(f"assets/{MOONRAKER_ENV_FILE_NAME}")
-
 
 EXIT_MOONRAKER_SETUP = "Exiting Moonraker setup ..."

kiauh/components/moonraker/assets/moonraker.conf

@@ -10,7 +10,6 @@ trusted_clients:
     169.254.0.0/16
     172.16.0.0/12
     192.168.0.0/16
-    FC00::/7
     FE80::/10
     ::1/128
 cors_domains:

kiauh/components/moonraker/menus/moonraker_remove_menu.py

@@ -1,82 +1,88 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
-from typing import Type
+from typing import Optional, Type
 
 from components.moonraker import moonraker_remove
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import COLOR_CYAN, COLOR_RED, RESET_FORMAT
 
 
 # noinspection PyUnusedLocal
 class MoonrakerRemoveMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None):
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "Remove Moonraker"
+        self.previous_menu = previous_menu
-        self.title_color = Color.RED
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
         self.remove_moonraker_service = False
         self.remove_moonraker_dir = False
         self.remove_moonraker_env = False
         self.remove_moonraker_polkit = False
-        self.selection_state = False
+        self.delete_moonraker_logs = False
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.remove_menu import RemoveMenu
 
-        self.previous_menu = previous_menu if previous_menu is not None else RemoveMenu
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else RemoveMenu
+        )
 
     def set_options(self) -> None:
         self.options = {
-            "a": Option(method=self.toggle_all),
+            "0": Option(method=self.toggle_all, menu=False),
-            "1": Option(method=self.toggle_remove_moonraker_service),
+            "1": Option(method=self.toggle_remove_moonraker_service, menu=False),
-            "2": Option(method=self.toggle_remove_moonraker_dir),
+            "2": Option(method=self.toggle_remove_moonraker_dir, menu=False),
-            "3": Option(method=self.toggle_remove_moonraker_env),
+            "3": Option(method=self.toggle_remove_moonraker_env, menu=False),
-            "4": Option(method=self.toggle_remove_moonraker_polkit),
+            "4": Option(method=self.toggle_remove_moonraker_polkit, menu=False),
-            "c": Option(method=self.run_removal_process),
+            "5": Option(method=self.toggle_delete_moonraker_logs, menu=False),
+            "c": Option(method=self.run_removal_process, menu=False),
         }
 
     def print_menu(self) -> None:
-        checked = f"[{Color.apply('x', Color.CYAN)}]"
+        header = " [ Remove Moonraker ] "
+        color = COLOR_RED
+        count = 62 - len(color) - len(RESET_FORMAT)
+        checked = f"[{COLOR_CYAN}x{RESET_FORMAT}]"
         unchecked = "[ ]"
         o1 = checked if self.remove_moonraker_service else unchecked
         o2 = checked if self.remove_moonraker_dir else unchecked
         o3 = checked if self.remove_moonraker_env else unchecked
         o4 = checked if self.remove_moonraker_polkit else unchecked
+        o5 = checked if self.delete_moonraker_logs else unchecked
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ Enter a number and hit enter to select / deselect     ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║ the specific option for removal.                      ║
+            |-------------------------------------------------------|
-            ╟───────────────────────────────────────────────────────╢
+            | Enter a number and hit enter to select / deselect     |
-            ║  a) {self._get_selection_state_str():37}             ║
+            | the specific option for removal.                      |
-            ╟───────────────────────────────────────────────────────╢
+            |-------------------------------------------------------|
-            ║  1) {o1} Remove Service                                ║
+            |  0) Select everything                                 |
-            ║  2) {o2} Remove Local Repository                       ║
+            |-------------------------------------------------------|
-            ║  3) {o3} Remove Python Environment                     ║
+            |  1) {o1} Remove Service                                |
-            ║  4) {o4} Remove Policy Kit Rules                       ║
+            |  2) {o2} Remove Local Repository                       |
-            ╟───────────────────────────────────────────────────────╢
+            |  3) {o3} Remove Python Environment                     |
-            ║  C) Continue                                          ║
+            |  4) {o4} Remove Policy Kit Rules                       |
-            ╟───────────────────────────────────────────────────────╢
+            |  5) {o5} Delete all Log-Files                          |
+            |-------------------------------------------------------|
+            |  C) Continue                                          |
             """
         )[1:]
         print(menu, end="")
 
     def toggle_all(self, **kwargs) -> None:
-        self.selection_state = not self.selection_state
+        self.remove_moonraker_service = True
-        self.remove_moonraker_service = self.selection_state
+        self.remove_moonraker_dir = True
-        self.remove_moonraker_dir = self.selection_state
+        self.remove_moonraker_env = True
-        self.remove_moonraker_env = self.selection_state
+        self.remove_moonraker_polkit = True
-        self.remove_moonraker_polkit = self.selection_state
+        self.delete_moonraker_logs = True
 
     def toggle_remove_moonraker_service(self, **kwargs) -> None:
         self.remove_moonraker_service = not self.remove_moonraker_service
@@ -90,18 +96,19 @@ class MoonrakerRemoveMenu(BaseMenu):
     def toggle_remove_moonraker_polkit(self, **kwargs) -> None:
         self.remove_moonraker_polkit = not self.remove_moonraker_polkit
 
+    def toggle_delete_moonraker_logs(self, **kwargs) -> None:
+        self.delete_moonraker_logs = not self.delete_moonraker_logs
+
     def run_removal_process(self, **kwargs) -> None:
         if (
             not self.remove_moonraker_service
             and not self.remove_moonraker_dir
             and not self.remove_moonraker_env
             and not self.remove_moonraker_polkit
+            and not self.delete_moonraker_logs
         ):
-            print(
+            error = f"{COLOR_RED}Nothing selected! Select options to remove first.{RESET_FORMAT}"
-                Color.apply(
+            print(error)
-                    "Nothing selected! Select options to remove first.", Color.RED
-                )
-            )
             return
 
         moonraker_remove.run_moonraker_removal(
@@ -109,20 +116,11 @@ class MoonrakerRemoveMenu(BaseMenu):
             self.remove_moonraker_dir,
             self.remove_moonraker_env,
             self.remove_moonraker_polkit,
+            self.delete_moonraker_logs,
         )
 
         self.remove_moonraker_service = False
         self.remove_moonraker_dir = False
         self.remove_moonraker_env = False
         self.remove_moonraker_polkit = False
-
+        self.delete_moonraker_logs = False
-        self._go_back()
-
-    def _get_selection_state_str(self) -> str:
-        return (
-            "Select everything" if not self.selection_state else "Deselect everything"
-        )
-
-    def _go_back(self, **kwargs) -> None:
-        if self.previous_menu is not None:
-            self.previous_menu().run()

kiauh/components/moonraker/moonraker.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -8,139 +8,151 @@
 # ======================================================================= #
 from __future__ import annotations
 
-from dataclasses import dataclass, field
+import subprocess
 from pathlib import Path
-from subprocess import CalledProcessError
+from typing import List
 
-from components.klipper.klipper import Klipper
+from components.moonraker import MODULE_PATH, MOONRAKER_DIR, MOONRAKER_ENV_DIR
-from components.moonraker import (
-    MOONRAKER_CFG_NAME,
-    MOONRAKER_DIR,
-    MOONRAKER_ENV_DIR,
-    MOONRAKER_ENV_FILE_NAME,
-    MOONRAKER_ENV_FILE_TEMPLATE,
-    MOONRAKER_LOG_NAME,
-    MOONRAKER_SERVICE_TEMPLATE,
-)
-from core.constants import CURRENT_USER
 from core.instance_manager.base_instance import BaseInstance
-from core.logger import Logger
 from core.submodules.simple_config_parser.src.simple_config_parser.simple_config_parser import (
     SimpleConfigParser,
 )
-from utils.fs_utils import create_folders
+from utils.constants import SYSTEMD
-from utils.sys_utils import get_service_file_path
+from utils.logger import Logger
 
 
 # noinspection PyMethodMayBeStatic
-@dataclass
+class Moonraker(BaseInstance):
-class Moonraker:
+    @classmethod
-    suffix: str
+    def blacklist(cls) -> List[str]:
-    base: BaseInstance = field(init=False, repr=False)
+        return ["None", "mcu"]
-    service_file_path: Path = field(init=False)
-    log_file_name: str = MOONRAKER_LOG_NAME
-    moonraker_dir: Path = MOONRAKER_DIR
-    env_dir: Path = MOONRAKER_ENV_DIR
-    data_dir: Path = field(init=False)
-    cfg_file: Path = field(init=False)
-    env_file: Path = field(init=False)
-    backup_dir: Path = field(init=False)
-    certs_dir: Path = field(init=False)
-    db_dir: Path = field(init=False)
-    port: int | None = field(init=False)
 
-    def __post_init__(self):
+    def __init__(self, suffix: str = ""):
-        self.base: BaseInstance = BaseInstance(Klipper, self.suffix)
+        super().__init__(instance_type=self, suffix=suffix)
-        self.base.log_file_name = self.log_file_name
+        self.moonraker_dir: Path = MOONRAKER_DIR
+        self.env_dir: Path = MOONRAKER_ENV_DIR
+        self.cfg_file = self.cfg_dir.joinpath("moonraker.conf")
+        self.port = self._get_port()
+        self.backup_dir = self.data_dir.joinpath("backup")
+        self.certs_dir = self.data_dir.joinpath("certs")
+        self._db_dir = self.data_dir.joinpath("database")
+        self._comms_dir = self.data_dir.joinpath("comms")
+        self.log = self.log_dir.joinpath("moonraker.log")
 
-        self.service_file_path: Path = get_service_file_path(Moonraker, self.suffix)
+    @property
-        self.data_dir: Path = self.base.data_dir
+    def db_dir(self) -> Path:
-        self.cfg_file: Path = self.base.cfg_dir.joinpath(MOONRAKER_CFG_NAME)
+        return self._db_dir
-        self.env_file: Path = self.base.sysd_dir.joinpath(MOONRAKER_ENV_FILE_NAME)
-        self.backup_dir: Path = self.base.data_dir.joinpath("backup")
-        self.certs_dir: Path = self.base.data_dir.joinpath("certs")
-        self.db_dir: Path = self.base.data_dir.joinpath("database")
-        self.port: int | None = self._get_port()
 
-    def create(self) -> None:
+    @property
-        from utils.sys_utils import create_env_file, create_service_file
+    def comms_dir(self) -> Path:
+        return self._comms_dir
 
+    def create(self, create_example_cfg: bool = False) -> None:
         Logger.print_status("Creating new Moonraker Instance ...")
+        service_template_path = MODULE_PATH.joinpath("assets/moonraker.service")
+        env_template_file_path = MODULE_PATH.joinpath("assets/moonraker.env")
+        service_file_name = self.get_service_file_name(extension=True)
+        service_file_target = SYSTEMD.joinpath(service_file_name)
+        env_file_target = self.sysd_dir.joinpath("moonraker.env")
 
         try:
-            create_folders(self.base.base_folders)
+            self.create_folders([self.backup_dir, self.certs_dir, self._db_dir])
-
+            self.write_service_file(
-            create_service_file(
+                service_template_path, service_file_target, env_file_target
-                name=self.service_file_path.name,
-                content=self._prep_service_file_content(),
-            )
-            create_env_file(
-                path=self.base.sysd_dir.joinpath(MOONRAKER_ENV_FILE_NAME),
-                content=self._prep_env_file_content(),
             )
+            self.write_env_file(env_template_file_path, env_file_target)
 
-        except CalledProcessError as e:
+        except subprocess.CalledProcessError as e:
-            Logger.print_error(f"Error creating instance: {e}")
+            Logger.print_error(
+                f"Error creating service file {service_file_target}: {e}"
+            )
             raise
         except OSError as e:
-            Logger.print_error(f"Error creating env file: {e}")
+            Logger.print_error(f"Error writing file: {e}")
             raise
 
-    def _prep_service_file_content(self) -> str:
+    def delete(self) -> None:
-        template = MOONRAKER_SERVICE_TEMPLATE
+        service_file = self.get_service_file_name(extension=True)
+        service_file_path = self.get_service_file_path()
+
+        Logger.print_status(f"Deleting Moonraker Instance: {service_file}")
 
         try:
-            with open(template, "r") as template_file:
+            command = ["sudo", "rm", "-f", service_file_path]
+            subprocess.run(command, check=True)
+            Logger.print_ok(f"Service file deleted: {service_file_path}")
+        except subprocess.CalledProcessError as e:
+            Logger.print_error(f"Error deleting service file: {e}")
+            raise
+
+    def write_service_file(
+        self,
+        service_template_path: Path,
+        service_file_target: Path,
+        env_file_target: Path,
+    ) -> None:
+        service_content = self._prep_service_file(
+            service_template_path, env_file_target
+        )
+        command = ["sudo", "tee", service_file_target]
+        subprocess.run(
+            command,
+            input=service_content.encode(),
+            stdout=subprocess.DEVNULL,
+            check=True,
+        )
+        Logger.print_ok(f"Service file created: {service_file_target}")
+
+    def write_env_file(
+        self, env_template_file_path: Path, env_file_target: Path
+    ) -> None:
+        env_file_content = self._prep_env_file(env_template_file_path)
+        with open(env_file_target, "w") as env_file:
+            env_file.write(env_file_content)
+        Logger.print_ok(f"Env file created: {env_file_target}")
+
+    def _prep_service_file(
+        self, service_template_path: Path, env_file_path: Path
+    ) -> str:
+        try:
+            with open(service_template_path, "r") as template_file:
                 template_content = template_file.read()
         except FileNotFoundError:
-            Logger.print_error(f"Unable to open {template} - File not found")
+            Logger.print_error(
+                f"Unable to open {service_template_path} - File not found"
+            )
             raise
-
+        service_content = template_content.replace("%USER%", self.user)
-        service_content = template_content.replace(
-            "%USER%",
-            CURRENT_USER,
-        )
         service_content = service_content.replace(
-            "%MOONRAKER_DIR%",
+            "%MOONRAKER_DIR%", str(self.moonraker_dir)
-            self.moonraker_dir.as_posix(),
-        )
-        service_content = service_content.replace(
-            "%ENV%",
-            self.env_dir.as_posix(),
-        )
-        service_content = service_content.replace(
-            "%ENV_FILE%",
-            self.base.sysd_dir.joinpath(MOONRAKER_ENV_FILE_NAME).as_posix(),
         )
+        service_content = service_content.replace("%ENV%", str(self.env_dir))
+        service_content = service_content.replace("%ENV_FILE%", str(env_file_path))
         return service_content
 
-    def _prep_env_file_content(self) -> str:
+    def _prep_env_file(self, env_template_file_path: Path) -> str:
-        template = MOONRAKER_ENV_FILE_TEMPLATE
-
         try:
-            with open(template, "r") as env_file:
+            with open(env_template_file_path, "r") as env_file:
                 env_template_file_content = env_file.read()
         except FileNotFoundError:
-            Logger.print_error(f"Unable to open {template} - File not found")
+            Logger.print_error(
+                f"Unable to open {env_template_file_path} - File not found"
+            )
             raise
-
         env_file_content = env_template_file_content.replace(
-            "%MOONRAKER_DIR%",
+            "%MOONRAKER_DIR%", str(self.moonraker_dir)
-            self.moonraker_dir.as_posix(),
         )
         env_file_content = env_file_content.replace(
-            "%PRINTER_DATA%",
+            "%PRINTER_DATA%", str(self.data_dir)
-            self.base.data_dir.as_posix(),
         )
-
         return env_file_content
 
     def _get_port(self) -> int | None:
-        if not self.cfg_file or not self.cfg_file.is_file():
+        if not self.cfg_file.is_file():
             return None
 
         scp = SimpleConfigParser()
-        scp.read_file(self.cfg_file)
+        scp.read(self.cfg_file)
-        port: int | None = scp.getint("server", "port", fallback=None)
+        port = scp.getint("server", "port", fallback=None)
 
         return port

kiauh/components/moonraker/moonraker_dialogs.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -13,7 +13,7 @@ from typing import List
 from components.klipper.klipper import Klipper
 from components.moonraker.moonraker import Moonraker
 from core.menus.base_menu import print_back_footer
-from core.types.color import Color
+from utils.constants import COLOR_CYAN, COLOR_GREEN, COLOR_YELLOW, RESET_FORMAT
 
 
 def print_moonraker_overview(
@@ -22,23 +22,23 @@ def print_moonraker_overview(
     show_index=False,
     show_select_all=False,
 ):
-    headline = Color.apply("The following instances were found:", Color.GREEN)
+    headline = f"{COLOR_GREEN}The following instances were found:{RESET_FORMAT}"
     dialog = textwrap.dedent(
         f"""
-        ╔═══════════════════════════════════════════════════════╗
+        /=======================================================\\
-        ║{headline:^64}║
+        |{headline:^64}|
-        ╟───────────────────────────────────────────────────────╢
+        |-------------------------------------------------------|
         """
     )[1:]
 
     if show_select_all:
-        select_all = Color.apply("a) Select all", Color.YELLOW)
+        select_all = f"{COLOR_YELLOW}a) Select all{RESET_FORMAT}"
-        dialog += f"║ {select_all:<63}║\n"
+        dialog += f"| {select_all:<63}|\n"
-        dialog += "║                                                       ║\n"
+        dialog += "|                                                       |\n"
 
     instance_map = {
-        k.service_file_path.stem: (
+        k.get_service_file_name(): (
-            k.service_file_path.stem.replace("klipper", "moonraker")
+            k.get_service_file_name().replace("klipper", "moonraker")
             if k.suffix in [m.suffix for m in moonraker_instances]
             else ""
         )
@@ -48,24 +48,19 @@ def print_moonraker_overview(
     for i, k in enumerate(instance_map):
         mr_name = instance_map.get(k)
         m = f"<-> {mr_name}" if mr_name != "" else ""
-        line = Color.apply(f"{f'{i + 1})' if show_index else '●'} {k} {m}", Color.CYAN)
+        line = f"{COLOR_CYAN}{f'{i})' if show_index else '●'} {k} {m} {RESET_FORMAT}"
-        dialog += f"║ {line:<63}║\n"
+        dialog += f"| {line:<63}|\n"
 
-    warn_l1 = Color.apply("PLEASE NOTE:", Color.YELLOW)
+    warn_l1 = f"{COLOR_YELLOW}PLEASE NOTE: {RESET_FORMAT}"
-    warn_l2 = Color.apply(
+    warn_l2 = f"{COLOR_YELLOW}If you select an instance with an existing Moonraker{RESET_FORMAT}"
-        "If you select an instance with an existing Moonraker", Color.YELLOW
+    warn_l3 = f"{COLOR_YELLOW}instance, that Moonraker instance will be re-created!{RESET_FORMAT}"
-    )
-    warn_l3 = Color.apply(
-        "instance, that Moonraker instance will be re-created!", Color.YELLOW
-    )
     warning = textwrap.dedent(
         f"""
-        ║                                                       ║
+        |                                                       |
-        ╟───────────────────────────────────────────────────────╢
+        |-------------------------------------------------------|
-        ║ {warn_l1:<63}║
+        | {warn_l1:<63}|
-        ║ {warn_l2:<63}║
+        | {warn_l2:<63}|
-        ║ {warn_l3:<63}║
+        | {warn_l3:<63}|
-        ╟───────────────────────────────────────────────────────╢
         """
     )[1:]
 

kiauh/components/moonraker/moonraker_remove.py

@@ -1,25 +1,24 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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 subprocess import DEVNULL, PIPE, CalledProcessError, run
+import shutil
-from typing import List
+import subprocess
+from typing import List, Union
 
 from components.klipper.klipper_dialogs import print_instance_overview
 from components.moonraker import MOONRAKER_DIR, MOONRAKER_ENV_DIR
 from components.moonraker.moonraker import Moonraker
 from core.instance_manager.instance_manager import InstanceManager
-from core.logger import Logger
+from utils.fs_utils import remove_file
-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.logger import Logger
-from utils.sys_utils import unit_file_exists
+from utils.sys_utils import cmd_sysctl_manage
 
 
 def run_moonraker_removal(
@@ -27,74 +26,102 @@ def run_moonraker_removal(
     remove_dir: bool,
     remove_env: bool,
     remove_polkit: bool,
+    delete_logs: bool,
 ) -> None:
-    instances = get_instances(Moonraker)
+    im = InstanceManager(Moonraker)
 
     if remove_service:
         Logger.print_status("Removing Moonraker instances ...")
-        if instances:
+        if im.instances:
-            instances_to_remove = select_instances_to_remove(instances)
+            instances_to_remove = select_instances_to_remove(im.instances)
-            remove_instances(instances_to_remove)
+            remove_instances(im, instances_to_remove)
         else:
             Logger.print_info("No Moonraker Services installed! Skipped ...")
 
-    delete_remaining: bool = remove_polkit or remove_dir or remove_env
+    if (remove_polkit or remove_dir or remove_env) and im.instances:
-    if delete_remaining and unit_file_exists("moonraker", suffix="service"):
+        Logger.print_warn("There are still other Moonraker services installed!")
-        Logger.print_info("There are still other Moonraker services installed")
+        Logger.print_warn("Therefor the following parts cannot be removed:")
-        Logger.print_info(
+        Logger.print_warn(
-            "●  Moonraker PolicyKit rules were not removed.", prefix=False
+            """
+            ● Moonraker PolicyKit rules
+            ● Moonraker local repository
+            ● Moonraker Python environment
+            """,
+            False,
         )
-        Logger.print_info(f"● '{MOONRAKER_DIR}' was not removed.", prefix=False)
-        Logger.print_info(f"● '{MOONRAKER_ENV_DIR}' was not removed.", prefix=False)
     else:
         if remove_polkit:
             Logger.print_status("Removing all Moonraker policykit rules ...")
             remove_polkit_rules()
         if remove_dir:
             Logger.print_status("Removing Moonraker local repository ...")
-            run_remove_routines(MOONRAKER_DIR)
+            remove_moonraker_dir()
         if remove_env:
             Logger.print_status("Removing Moonraker Python environment ...")
-            run_remove_routines(MOONRAKER_ENV_DIR)
+            remove_moonraker_env()
+
+    # delete moonraker logs of all instances
+    if delete_logs:
+        Logger.print_status("Removing all Moonraker logs ...")
+        delete_moonraker_logs(im.instances)
 
 
 def select_instances_to_remove(
     instances: List[Moonraker],
-) -> List[Moonraker] | None:
+) -> Union[List[Moonraker], None]:
-    start_index = 1
+    print_instance_overview(instances, show_index=True, show_select_all=True)
-    options = [str(i + start_index) for i in range(len(instances))]
+
-    options.extend(["a", "b"])
+    options = [str(i) for i in range(len(instances))]
-    instance_map = {options[i]: instances[i] for i in range(len(instances))}
+    options.extend(["a", "A", "b", "B"])
 
-    print_instance_overview(
-        instances,
-        start_index=start_index,
-        show_index=True,
-        show_select_all=True,
-    )
     selection = get_selection_input("Select Moonraker instance to remove", options)
 
     instances_to_remove = []
-    if selection == "b":
+    if selection == "b".lower():
         return None
-    elif selection == "a":
+    elif selection == "a".lower():
         instances_to_remove.extend(instances)
     else:
-        instances_to_remove.append(instance_map[selection])
+        instance = instances[int(selection)]
+        instances_to_remove.append(instance)
 
     return instances_to_remove
 
 
 def remove_instances(
-    instance_list: List[Moonraker] | None,
+    instance_manager: InstanceManager,
+    instance_list: List[Moonraker],
 ) -> None:
-    if not instance_list:
-        Logger.print_info("No Moonraker instances found. Skipped ...")
-        return
     for instance in instance_list:
-        Logger.print_status(f"Removing instance {instance.service_file_path.stem} ...")
+        Logger.print_status(f"Removing instance {instance.get_service_file_name()} ...")
-        InstanceManager.remove(instance)
+        instance_manager.current_instance = instance
-        delete_moonraker_env_file(instance)
+        instance_manager.stop_instance()
+        instance_manager.disable_instance()
+        instance_manager.delete_instance()
+
+    cmd_sysctl_manage("daemon-reload")
+
+
+def remove_moonraker_dir() -> None:
+    if not MOONRAKER_DIR.exists():
+        Logger.print_info(f"'{MOONRAKER_DIR}' does not exist. Skipped ...")
+        return
+
+    try:
+        shutil.rmtree(MOONRAKER_DIR)
+    except OSError as e:
+        Logger.print_error(f"Unable to delete '{MOONRAKER_DIR}':\n{e}")
+
+
+def remove_moonraker_env() -> None:
+    if not MOONRAKER_ENV_DIR.exists():
+        Logger.print_info(f"'{MOONRAKER_ENV_DIR}' does not exist. Skipped ...")
+        return
+
+    try:
+        shutil.rmtree(MOONRAKER_ENV_DIR)
+    except OSError as e:
+        Logger.print_error(f"Unable to delete '{MOONRAKER_ENV_DIR}':\n{e}")
 
 
 def remove_polkit_rules() -> None:
@@ -104,18 +131,30 @@ def remove_polkit_rules() -> None:
         return
 
     try:
-        cmd = [f"{MOONRAKER_DIR}/scripts/set-policykit-rules.sh", "--clear"]
+        command = [
-        run(cmd, stderr=PIPE, stdout=DEVNULL, check=True)
+            f"{MOONRAKER_DIR}/scripts/set-policykit-rules.sh",
-    except CalledProcessError as e:
+            "--clear",
+        ]
+        subprocess.run(
+            command,
+            stderr=subprocess.PIPE,
+            stdout=subprocess.DEVNULL,
+            check=True,
+        )
+    except subprocess.CalledProcessError as e:
         Logger.print_error(f"Error while removing policykit rules: {e}")
 
     Logger.print_ok("Policykit rules successfully removed!")
 
 
-def delete_moonraker_env_file(instance: Moonraker):
+def delete_moonraker_logs(instances: List[Moonraker]) -> None:
-    Logger.print_status(f"Remove '{instance.env_file}'")
+    all_logfiles = []
-    if not instance.env_file.exists():
+    for instance in instances:
-        msg = f"Env file in {instance.base.sysd_dir} not found. Skipped ..."
+        all_logfiles = list(instance.log_dir.glob("moonraker.log*"))
-        Logger.print_info(msg)
+    if not all_logfiles:
+        Logger.print_info("No Moonraker logs found. Skipped ...")
         return
-    run_remove_routines(instance.env_file)
+
+    for log in all_logfiles:
+        Logger.print_status(f"Remove '{log}'")
+        remove_file(log)

kiauh/components/moonraker/moonraker_setup.py

@@ -1,25 +1,21 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
+import json
-
 import subprocess
-from typing import List
+from pathlib import Path
 
 from components.klipper.klipper import Klipper
 from components.moonraker import (
     EXIT_MOONRAKER_SETUP,
-    MOONRAKER_DEPS_JSON_FILE,
     MOONRAKER_DIR,
     MOONRAKER_ENV_DIR,
-    MOONRAKER_INSTALL_SCRIPT,
+    MOONRAKER_REQUIREMENTS_TXT,
-    MOONRAKER_REQ_FILE,
-    MOONRAKER_SPEEDUPS_REQ_FILE,
     POLKIT_FILE,
     POLKIT_LEGACY_FILE,
     POLKIT_SCRIPT,
@@ -27,14 +23,9 @@ from components.moonraker import (
 )
 from components.moonraker.moonraker import Moonraker
 from components.moonraker.moonraker_dialogs import print_moonraker_overview
-from components.moonraker.services.moonraker_instance_service import (
+from components.moonraker.moonraker_utils import (
-    MoonrakerInstanceService,
-)
-from components.moonraker.utils.sysdeps_parser import SysDepsParser
-from components.moonraker.utils.utils import (
     backup_moonraker_dir,
     create_example_moonraker_conf,
-    load_sysdeps_json,
 )
 from components.webui_client.client_utils import (
     enable_mainsail_remotemode,
@@ -42,9 +33,7 @@ from components.webui_client.client_utils import (
 )
 from components.webui_client.mainsail_data import MainsailData
 from core.instance_manager.instance_manager import InstanceManager
-from core.logger import DialogType, Logger
 from core.settings.kiauh_settings import KiauhSettings
-from core.types.color import Color
 from utils.common import check_install_dependencies
 from utils.fs_utils import check_file_exist
 from utils.git_utils import git_clone_wrapper, git_pull_wrapper
@@ -52,124 +41,91 @@ from utils.input_utils import (
     get_confirm,
     get_selection_input,
 )
-from utils.instance_utils import get_instances
+from utils.logger import Logger
 from utils.sys_utils import (
     check_python_version,
     cmd_sysctl_manage,
-    cmd_sysctl_service,
     create_python_venv,
-    get_ipv4_addr,
     install_python_requirements,
     parse_packages_from_file,
 )
 
 
 def install_moonraker() -> None:
-    klipper_list: List[Klipper] = get_instances(Klipper)
+    if not check_moonraker_install_requirements():
-
-    if not check_moonraker_install_requirements(klipper_list):
         return
 
-    instance_service = MoonrakerInstanceService()
+    kl_im = InstanceManager(Klipper)
-    instance_service.load_instances()
+    klipper_instances = kl_im.instances
+    mr_im = InstanceManager(Moonraker)
+    moonraker_instances = mr_im.instances
 
-    moonraker_list: List[Moonraker] = instance_service.get_all_instances()
+    selected_klipper_instance = 0
-    new_instances: List[Moonraker] = []
+    if len(klipper_instances) > 1:
-    selected_option: str | Klipper
-
-    if len(klipper_list) == 1:
-        suffix: str = klipper_list[0].suffix
-        new_inst = instance_service.create_new_instance(suffix)
-        new_instances.append(new_inst)
-
-    else:
         print_moonraker_overview(
-            klipper_list,
+            klipper_instances,
-            moonraker_list,
+            moonraker_instances,
             show_index=True,
             show_select_all=True,
         )
-        options = {str(i + 1): k for i, k in enumerate(klipper_list)}
+        options = [str(i) for i in range(len(klipper_instances))]
-        additional_options = {"a": None, "b": None}
+        options.extend(["a", "A", "b", "B"])
-        options = {**options, **additional_options}
         question = "Select Klipper instance to setup Moonraker for"
-        selected_option = get_selection_input(question, options)
+        selected_klipper_instance = get_selection_input(question, options).lower()
 
-        if selected_option == "b":
+    instance_names = []
-            Logger.print_status(EXIT_MOONRAKER_SETUP)
+    if selected_klipper_instance == "b":
-            return
+        Logger.print_status(EXIT_MOONRAKER_SETUP)
+        return
 
-        if selected_option == "a":
+    elif selected_klipper_instance == "a":
-            new_inst_list: List[Moonraker] = [
+        for instance in klipper_instances:
-                instance_service.create_new_instance(k.suffix) for k in klipper_list
+            instance_names.append(instance.suffix)
-            ]
+
-            new_instances.extend(new_inst_list)
+    else:
-        else:
+        index = int(selected_klipper_instance)
-            klipper_instance: Klipper | None = options.get(selected_option)
+        instance_names.append(klipper_instances[index].suffix)
-            if klipper_instance is None:
-                raise Exception("Error selecting instance!")
-            new_inst = instance_service.create_new_instance(klipper_instance.suffix)
-            new_instances.append(new_inst)
 
     create_example_cfg = get_confirm("Create example moonraker.conf?")
 
     try:
-        check_install_dependencies()
+        check_install_dependencies(["git"])
         setup_moonraker_prerequesites()
         install_moonraker_polkit()
 
-        ports_map = instance_service.get_instance_port_map()
+        used_ports_map = {
-        for instance in new_instances:
+            instance.suffix: instance.port for instance in moonraker_instances
-            instance.create()
+        }
-            cmd_sysctl_service(instance.service_file_path.name, "enable")
+        for name in instance_names:
+            current_instance = Moonraker(suffix=name)
+
+            mr_im.current_instance = current_instance
+            mr_im.create_instance()
+            mr_im.enable_instance()
 
             if create_example_cfg:
                 # if a webclient and/or it's config is installed, patch
                 # its update section to the config
                 clients = get_existing_clients()
-                create_example_moonraker_conf(instance, ports_map, clients)
+                create_example_moonraker_conf(current_instance, used_ports_map, clients)
 
-            cmd_sysctl_service(instance.service_file_path.name, "start")
+            mr_im.start_instance()
 
         cmd_sysctl_manage("daemon-reload")
 
         # if mainsail is installed, and we installed
         # multiple moonraker instances, we enable mainsails remote mode
-        if MainsailData().client_dir.exists() and len(moonraker_list) > 1:
+        if MainsailData().client_dir.exists() and len(mr_im.instances) > 1:
             enable_mainsail_remotemode()
 
-        instance_service.load_instances()
-        new_instances = [
-            instance_service.get_instance_by_suffix(i.suffix) for i in new_instances
-        ]
-
-        ip: str = get_ipv4_addr()
-        # noinspection HttpUrlsUsage
-        url_list = [
-            f"● {i.service_file_path.stem}: http://{ip}:{i.port}"
-            for i in new_instances
-            if i.port
-        ]
-        dialog_content = []
-        if url_list:
-            dialog_content.append("You can access Moonraker via the following URL:")
-            dialog_content.extend(url_list)
-
-        Logger.print_dialog(
-            DialogType.CUSTOM,
-            custom_title="Moonraker successfully installed!",
-            custom_color=Color.GREEN,
-            content=dialog_content,
-        )
-
     except Exception as e:
         Logger.print_error(f"Error while installing Moonraker: {e}")
         return
 
 
-def check_moonraker_install_requirements(klipper_list: List[Klipper]) -> bool:
+def check_moonraker_install_requirements() -> bool:
     def check_klipper_instances() -> bool:
-        if len(klipper_list) >= 1:
+        if len(InstanceManager(Klipper).instances) >= 1:
             return True
 
         Logger.print_warn("Klipper not installed!")
@@ -187,35 +143,26 @@ def setup_moonraker_prerequesites() -> None:
     git_clone_wrapper(repo, MOONRAKER_DIR, branch)
 
     # install moonraker dependencies and create python virtualenv
-    install_moonraker_packages()
+    install_moonraker_packages(MOONRAKER_DIR)
-    if create_python_venv(MOONRAKER_ENV_DIR):
+    create_python_venv(MOONRAKER_ENV_DIR)
-        install_python_requirements(MOONRAKER_ENV_DIR, MOONRAKER_REQ_FILE)
+    install_python_requirements(MOONRAKER_ENV_DIR, MOONRAKER_REQUIREMENTS_TXT)
-        install_python_requirements(MOONRAKER_ENV_DIR, MOONRAKER_SPEEDUPS_REQ_FILE)
 
 
-def install_moonraker_packages() -> None:
+def install_moonraker_packages(moonraker_dir: Path) -> None:
-    Logger.print_status("Parsing Moonraker system dependencies  ...")
+    install_script = moonraker_dir.joinpath("scripts/install-moonraker.sh")
-
+    deps_json = MOONRAKER_DIR.joinpath("scripts/system-dependencies.json")
     moonraker_deps = []
-    if MOONRAKER_DEPS_JSON_FILE.exists():
-        Logger.print_info(
-            f"Parsing system dependencies from {MOONRAKER_DEPS_JSON_FILE.name} ..."
-        )
-        parser = SysDepsParser()
-        sysdeps = load_sysdeps_json(MOONRAKER_DEPS_JSON_FILE)
-        moonraker_deps.extend(parser.parse_dependencies(sysdeps))
 
-    elif MOONRAKER_INSTALL_SCRIPT.exists():
+    if deps_json.exists():
-        Logger.print_warn(f"{MOONRAKER_DEPS_JSON_FILE.name} not found!")
+        with open(deps_json, "r") as deps:
-        Logger.print_info(
+            moonraker_deps = json.load(deps).get("debian", [])
-            f"Parsing system dependencies from {MOONRAKER_INSTALL_SCRIPT.name} ..."
+    elif install_script.exists():
-        )
+        moonraker_deps = parse_packages_from_file(install_script)
-        moonraker_deps = parse_packages_from_file(MOONRAKER_INSTALL_SCRIPT)
 
     if not moonraker_deps:
-        raise ValueError("Error parsing Moonraker dependencies!")
+        raise ValueError("Error reading Moonraker dependencies!")
 
-    check_install_dependencies({*moonraker_deps})
+    check_install_dependencies(moonraker_deps)
 
 
 def install_moonraker_polkit() -> None:
@@ -256,14 +203,16 @@ def update_moonraker() -> None:
     if settings.kiauh.backup_before_update:
         backup_moonraker_dir()
 
-    instances = get_instances(Moonraker)
+    instance_manager = InstanceManager(Moonraker)
-    InstanceManager.stop_all(instances)
+    instance_manager.stop_all_instance()
 
-    git_pull_wrapper(repo=settings.moonraker.repo_url, target_dir=MOONRAKER_DIR)
+    git_pull_wrapper(
+        repo=settings.moonraker.repo_url, target_dir=MOONRAKER_DIR
+    )
 
     # install possible new system packages
-    install_moonraker_packages()
+    install_moonraker_packages(MOONRAKER_DIR)
     # install possible new python dependencies
-    install_python_requirements(MOONRAKER_ENV_DIR, MOONRAKER_REQ_FILE)
+    install_python_requirements(MOONRAKER_ENV_DIR, MOONRAKER_REQUIREMENTS_TXT)
 
-    InstanceManager.start_all(instances)
+    instance_manager.start_all_instance()

kiauh/components/moonraker/moonraker_utils.py

@@ -1,37 +1,38 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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 json
+
 import shutil
-from pathlib import Path
 from typing import Dict, List, Optional
 
 from components.moonraker import (
+    DEFAULT_MOONRAKER_PORT,
     MODULE_PATH,
     MOONRAKER_BACKUP_DIR,
     MOONRAKER_DB_BACKUP_DIR,
-    MOONRAKER_DEFAULT_PORT,
     MOONRAKER_DIR,
     MOONRAKER_ENV_DIR,
 )
 from components.moonraker.moonraker import Moonraker
 from components.webui_client.base_data import BaseWebClient
+from components.webui_client.client_utils import enable_mainsail_remotemode
+from components.webui_client.mainsail_data import MainsailData
 from core.backup_manager.backup_manager import BackupManager
-from core.logger import Logger
+from core.instance_manager.instance_manager import InstanceManager
 from core.submodules.simple_config_parser.src.simple_config_parser.simple_config_parser import (
     SimpleConfigParser,
 )
-from core.types.component_status import ComponentStatus
 from utils.common import get_install_status
-from utils.instance_utils import get_instances
+from utils.logger import Logger
 from utils.sys_utils import (
     get_ipv4_addr,
 )
+from utils.types import ComponentStatus
 
 
 def get_moonraker_status() -> ComponentStatus:
@@ -43,7 +44,7 @@ def create_example_moonraker_conf(
     ports_map: Dict[str, int],
     clients: Optional[List[BaseWebClient]] = None,
 ) -> None:
-    Logger.print_status(f"Creating example moonraker.conf in '{instance.base.cfg_dir}'")
+    Logger.print_status(f"Creating example moonraker.conf in '{instance.cfg_dir}'")
     if instance.cfg_file.is_file():
         Logger.print_info(f"'{instance.cfg_file}' already exists.")
         return
@@ -67,7 +68,7 @@ def create_example_moonraker_conf(
         # of moonraker-1 is 7125 and moonraker-3 is 7127 and there are moonraker.conf files for moonraker-1
         # and moonraker-3 already. though, there does not seem to be a very reliable way of always assigning
         # the correct port to each instance and the user will likely be required to correct the value manually.
-        port = max(ports) + 1 if ports else MOONRAKER_DEFAULT_PORT
+        port = max(ports) + 1 if ports else DEFAULT_MOONRAKER_PORT
     else:
         port = ports_map.get(instance.suffix)
 
@@ -75,18 +76,23 @@ def create_example_moonraker_conf(
 
     ip = get_ipv4_addr().split(".")[:2]
     ip.extend(["0", "0/16"])
-    uds = instance.base.comms_dir.joinpath("klippy.sock")
+    uds = instance.comms_dir.joinpath("klippy.sock")
 
     scp = SimpleConfigParser()
-    scp.read_file(target)
+    scp.read(target)
     trusted_clients: List[str] = [
-        f"    {'.'.join(ip)}\n",
+        ".".join(ip),
-        *scp.getval("authorization", "trusted_clients"),
+        *scp.get("authorization", "trusted_clients"),
     ]
 
-    scp.set_option("server", "port", str(port))
+    scp.set("server", "port", str(port))
-    scp.set_option("server", "klippy_uds_address", str(uds))
+    scp.set("server", "klippy_uds_address", str(uds))
-    scp.set_option("authorization", "trusted_clients", trusted_clients)
+    scp.set(
+        "authorization",
+        "trusted_clients",
+        "\n".join(trusted_clients),
+        True,
+    )
 
     # add existing client and client configs in the update section
     if clients is not None and len(clients) > 0:
@@ -101,7 +107,7 @@ def create_example_moonraker_conf(
             ]
             scp.add_section(section=c_section)
             for option in c_options:
-                scp.set_option(c_section, option[0], option[1])
+                scp.set(c_section, option[0], option[1])
 
             # client config part
             c_config = c.client_config
@@ -116,13 +122,65 @@ def create_example_moonraker_conf(
                 ]
                 scp.add_section(section=c_config_section)
                 for option in c_config_options:
-                    scp.set_option(c_config_section, option[0], option[1])
+                    scp.set(c_config_section, option[0], option[1])
 
-    scp.write_file(target)
+    scp.write(target)
-    Logger.print_ok(f"Example moonraker.conf created in '{instance.base.cfg_dir}'")
+    Logger.print_ok(f"Example moonraker.conf created in '{instance.cfg_dir}'")
 
 
-def backup_moonraker_dir() -> None:
+def moonraker_to_multi_conversion(new_name: str) -> None:
+    """
+    Converts the first instance in the List of Moonraker instances to an instance
+    with a new name. This method will be called when converting from a single Klipper
+    instance install to a multi instance install when Moonraker is also already
+    installed with a single instance.
+    :param new_name: new name the previous single instance is renamed to
+    :return: None
+    """
+    im = InstanceManager(Moonraker)
+    instances: List[Moonraker] = im.instances
+    if not instances:
+        return
+
+    # in case there are multiple Moonraker instances, we don't want to do anything
+    if len(instances) > 1:
+        Logger.print_info("More than a single Moonraker instance found. Skipped ...")
+        return
+
+    Logger.print_status("Convert Moonraker single to multi instance ...")
+
+    # remove the old single instance
+    im.current_instance = im.instances[0]
+    im.stop_instance()
+    im.disable_instance()
+    im.delete_instance()
+
+    # create a new moonraker instance with the new name
+    new_instance = Moonraker(suffix=new_name)
+    im.current_instance = new_instance
+
+    # patch the server sections klippy_uds_address value to match the new printer_data foldername
+    scp = SimpleConfigParser()
+    scp.read(new_instance.cfg_file)
+    if scp.has_section("server"):
+        scp.set(
+            "server",
+            "klippy_uds_address",
+            str(new_instance.comms_dir.joinpath("klippy.sock")),
+        )
+        scp.write(new_instance.cfg_file)
+
+    # create, enable and start the new moonraker instance
+    im.create_instance()
+    im.enable_instance()
+    im.start_instance()
+
+    # if mainsail is installed, we enable mainsails remote mode
+    if MainsailData().client_dir.exists() and len(im.instances) > 1:
+        enable_mainsail_remotemode()
+
+
+def backup_moonraker_dir():
     bm = BackupManager()
     bm.backup_directory("moonraker", source=MOONRAKER_DIR, target=MOONRAKER_BACKUP_DIR)
     bm.backup_directory(
@@ -131,21 +189,12 @@ def backup_moonraker_dir() -> None:
 
 
 def backup_moonraker_db_dir() -> None:
-    instances: List[Moonraker] = get_instances(Moonraker)
+    im = InstanceManager(Moonraker)
+    instances: List[Moonraker] = im.instances
     bm = BackupManager()
 
     for instance in instances:
-        name = f"database-{instance.data_dir.name}"
+        name = f"database-{instance.data_dir_name}"
         bm.backup_directory(
             name, source=instance.db_dir, target=MOONRAKER_DB_BACKUP_DIR
         )
-
-
-def load_sysdeps_json(file: Path) -> Dict[str, List[str]]:
-    try:
-        sysdeps: Dict[str, List[str]] = json.loads(file.read_bytes())
-    except json.JSONDecodeError as e:
-        Logger.print_error(f"Unable to parse {file.name}:\n{e}")
-        return {}
-    else:
-        return sysdeps

kiauh/components/moonraker/services/moonraker_instance_service.py

@@ -1,49 +0,0 @@
-# ======================================================================= #
-#  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
-
-from typing import Dict, List
-
-from components.moonraker.moonraker import Moonraker
-from utils.instance_utils import get_instances
-
-
-class MoonrakerInstanceService:
-    __cls_instance = None
-    __instances: List[Moonraker] = []
-
-    def __new__(cls) -> "MoonrakerInstanceService":
-        if cls.__cls_instance is None:
-            cls.__cls_instance = super(MoonrakerInstanceService, cls).__new__(cls)
-        return cls.__cls_instance
-
-    def __init__(self) -> None:
-        if not hasattr(self, "__initialized"):
-            self.__initialized = False
-        if self.__initialized:
-            return
-        self.__initialized = True
-
-    def load_instances(self) -> None:
-        self.__instances = get_instances(Moonraker)
-
-    def create_new_instance(self, suffix: str) -> Moonraker:
-        instance = Moonraker(suffix)
-        self.__instances.append(instance)
-        return instance
-
-    def get_all_instances(self) -> List[Moonraker]:
-        return self.__instances
-
-    def get_instance_by_suffix(self, suffix: str) -> Moonraker | None:
-        instances: List[Moonraker] = [i for i in self.__instances if i.suffix == suffix]
-        return instances[0] if instances else None
-
-    def get_instance_port_map(self) -> Dict[str, int]:
-        return {i.suffix: i.port for i in self.__instances}

kiauh/components/moonraker/utils/sysdeps_parser.py

@@ -1,173 +0,0 @@
-# ======================================================================= #
-#  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                                          #
-#  It was modified by Dominik Willner <th33xitus@gmail.com>               #
-#                                                                         #
-#  The original file is part of Moonraker:                                #
-#  https://github.com/Arksine/moonraker                                   #
-#  Copyright (C) 2025 Eric Callahan <arksine.code@gmail.com>              #
-#                                                                         #
-#  This file may be distributed under the terms of the GNU GPLv3 license  #
-# ======================================================================= #
-
-from __future__ import annotations
-
-import logging
-import pathlib
-import re
-import shlex
-from typing import Any, Dict, List, Tuple
-
-
-def _get_distro_info() -> Dict[str, Any]:
-    release_file = pathlib.Path("/etc/os-release")
-    release_info: Dict[str, str] = {}
-    with release_file.open("r") as f:
-        lexer = shlex.shlex(f, posix=True)
-        lexer.whitespace_split = True
-        for item in list(lexer):
-            if "=" in item:
-                key, val = item.split("=", maxsplit=1)
-                release_info[key] = val
-    return dict(
-        distro_id=release_info.get("ID", ""),
-        distro_version=release_info.get("VERSION_ID", ""),
-        aliases=release_info.get("ID_LIKE", "").split(),
-    )
-
-
-def _convert_version(version: str) -> Tuple[str | int, ...]:
-    version = version.strip()
-    ver_match = re.match(r"\d+(\.\d+)*((?:-|\.).+)?", version)
-    if ver_match is not None:
-        return tuple(
-            [
-                int(part) if part.isdigit() else part
-                for part in re.split(r"\.|-", version)
-            ]
-        )
-    return (version,)
-
-
-class SysDepsParser:
-    def __init__(self, distro_info: Dict[str, Any] | None = None) -> None:
-        if distro_info is None:
-            distro_info = _get_distro_info()
-        self.distro_id: str = distro_info.get("distro_id", "")
-        self.aliases: List[str] = distro_info.get("aliases", [])
-        self.distro_version: Tuple[int | str, ...] = tuple()
-        version = distro_info.get("distro_version")
-        if version:
-            self.distro_version = _convert_version(version)
-
-    def _parse_spec(self, full_spec: str) -> str | None:
-        parts = full_spec.split(";", maxsplit=1)
-        if len(parts) == 1:
-            return full_spec
-        pkg_name = parts[0].strip()
-        expressions = re.split(r"( and | or )", parts[1].strip())
-        if not len(expressions) & 1:
-            # There should always be an odd number of expressions.  Each
-            # expression is separated by an "and" or "or" operator
-            logging.info(
-                f"Requirement specifier is missing an expression "
-                f"between logical operators : {full_spec}"
-            )
-            return None
-        last_result: bool = True
-        last_logical_op: str | None = "and"
-        for idx, exp in enumerate(expressions):
-            if idx & 1:
-                if last_logical_op is not None:
-                    logging.info(
-                        "Requirement specifier contains sequential logical "
-                        f"operators: {full_spec}"
-                    )
-                    return None
-                logical_op = exp.strip()
-                if logical_op not in ("and", "or"):
-                    logging.info(
-                        f"Invalid logical operator {logical_op} in requirement "
-                        f"specifier: {full_spec}"
-                    )
-                    return None
-                last_logical_op = logical_op
-                continue
-            elif last_logical_op is None:
-                logging.info(
-                    f"Requirement specifier contains two seqential expressions "
-                    f"without a logical operator: {full_spec}"
-                )
-                return None
-            dep_parts = re.split(r"(==|!=|<=|>=|<|>)", exp.strip())
-            req_var = dep_parts[0].strip().lower()
-            if len(dep_parts) != 3:
-                logging.info(f"Invalid comparison, must be 3 parts: {full_spec}")
-                return None
-            elif req_var == "distro_id":
-                left_op: str | Tuple[int | str, ...] = self.distro_id
-                right_op = dep_parts[2].strip().strip("\"'")
-            elif req_var == "distro_version":
-                if not self.distro_version:
-                    logging.info(
-                        "Distro Version not detected, cannot satisfy requirement: "
-                        f"{full_spec}"
-                    )
-                    return None
-                left_op = self.distro_version
-                right_op = _convert_version(dep_parts[2].strip().strip("\"'"))
-            else:
-                logging.info(f"Invalid requirement specifier: {full_spec}")
-                return None
-            operator = dep_parts[1].strip()
-            try:
-                compfunc = {
-                    "<": lambda x, y: x < y,
-                    ">": lambda x, y: x > y,
-                    "==": lambda x, y: x == y,
-                    "!=": lambda x, y: x != y,
-                    ">=": lambda x, y: x >= y,
-                    "<=": lambda x, y: x <= y,
-                }.get(operator, lambda x, y: False)
-                result = compfunc(left_op, right_op)
-                if last_logical_op == "and":
-                    last_result &= result
-                else:
-                    last_result |= result
-                last_logical_op = None
-            except Exception:
-                logging.exception(f"Error comparing requirements: {full_spec}")
-                return None
-        if last_result:
-            return pkg_name
-        return None
-
-    def parse_dependencies(self, sys_deps: Dict[str, List[str]]) -> List[str]:
-        if not self.distro_id:
-            logging.info(
-                "Failed to detect current distro ID, cannot parse dependencies"
-            )
-            return []
-        all_ids = [self.distro_id] + self.aliases
-        for distro_id in all_ids:
-            if distro_id in sys_deps:
-                if not sys_deps[distro_id]:
-                    logging.info(
-                        f"Dependency data contains an empty package definition "
-                        f"for linux distro '{distro_id}'"
-                    )
-                    continue
-                processed_deps: List[str] = []
-                for dep in sys_deps[distro_id]:
-                    parsed_dep = self._parse_spec(dep)
-                    if parsed_dep is not None:
-                        processed_deps.append(parsed_dep)
-                return processed_deps
-        else:
-            logging.info(
-                f"Dependency data has no package definition for linux "
-                f"distro '{self.distro_id}'"
-            )
-        return []

kiauh/components/webui_client/__init__.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #

kiauh/components/webui_client/base_data.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -9,8 +9,7 @@
 
 from __future__ import annotations
 
-from abc import ABC
+from abc import ABC, abstractmethod
-from dataclasses import dataclass
 from enum import Enum
 from pathlib import Path
 
@@ -25,33 +24,89 @@ class WebClientConfigType(Enum):
     FLUIDD: str = "fluidd-config"
 
 
-@dataclass()
 class BaseWebClient(ABC):
     """Base class for webclient data"""
 
-    client: WebClientType
+    @property
-    name: str
+    @abstractmethod
-    display_name: str
+    def client(self) -> WebClientType:
-    client_dir: Path
+        raise NotImplementedError
-    config_file: Path
+
-    backup_dir: Path
+    @property
-    repo_path: str
+    @abstractmethod
-    download_url: str
+    def name(self) -> str:
-    nginx_config: Path
+        raise NotImplementedError
-    nginx_access_log: Path
+
-    nginx_error_log: Path
+    @property
-    client_config: BaseWebClientConfig
+    @abstractmethod
+    def display_name(self) -> str:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def client_dir(self) -> Path:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def backup_dir(self) -> Path:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def repo_path(self) -> str:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def download_url(self) -> str:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def client_config(self) -> BaseWebClientConfig:
+        raise NotImplementedError
 
 
-@dataclass()
 class BaseWebClientConfig(ABC):
     """Base class for webclient config data"""
 
-    client_config: WebClientConfigType
+    @property
-    name: str
+    @abstractmethod
-    display_name: str
+    def client_config(self) -> WebClientConfigType:
-    config_filename: str
+        raise NotImplementedError
-    config_dir: Path
+
-    backup_dir: Path
+    @property
-    repo_url: str
+    @abstractmethod
-    config_section: str
+    def name(self) -> str:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def display_name(self) -> str:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def config_filename(self) -> str:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def config_dir(self) -> Path:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def backup_dir(self) -> Path:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def repo_url(self) -> str:
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def config_section(self) -> str:
+        raise NotImplementedError

kiauh/components/webui_client/client_config/client_config_remove.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -8,84 +8,54 @@
 # ======================================================================= #
 
 
+import shutil
+import subprocess
 from typing import List
 
 from components.klipper.klipper import Klipper
 from components.moonraker.moonraker import Moonraker
 from components.webui_client.base_data import BaseWebClientConfig
-from core.logger import Logger
+from core.instance_manager.instance_manager import InstanceManager
-from core.services.message_service import Message
-from core.types.color import Color
 from utils.config_utils import remove_config_section
-from utils.fs_utils import run_remove_routines
+from utils.fs_utils import remove_file
-from utils.instance_type import InstanceType
+from utils.logger import Logger
-from utils.instance_utils import get_instances
 
 
 def run_client_config_removal(
     client_config: BaseWebClientConfig,
     kl_instances: List[Klipper],
     mr_instances: List[Moonraker],
-) -> Message:
+) -> None:
-    completion_msg = Message(
+    remove_client_config_dir(client_config)
-        title=f"{client_config.display_name} Removal Process completed",
+    remove_client_config_symlink(client_config)
-        color=Color.GREEN,
+    remove_config_section(f"update_manager {client_config.name}", mr_instances)
-    )
+    remove_config_section(client_config.config_section, kl_instances)
-    Logger.print_status(f"Removing {client_config.display_name} ...")
-    if run_remove_routines(client_config.config_dir):
-        completion_msg.text.append(f"● {client_config.display_name} removed")
-
-    completion_msg = remove_moonraker_config_section(
-        completion_msg, client_config, mr_instances
-    )
-
-    completion_msg = remove_printer_config_section(
-        completion_msg, client_config, kl_instances
-    )
-
-    if completion_msg.text:
-        completion_msg.text.insert(0, "The following actions were performed:")
-    else:
-        completion_msg.color = Color.YELLOW
-        completion_msg.centered = True
-        completion_msg.text = ["Nothing to remove."]
-
-    return completion_msg
 
 
-def remove_cfg_symlink(client_config: BaseWebClientConfig, message: Message) -> Message:
+def remove_client_config_dir(client_config: BaseWebClientConfig) -> None:
-    instances: List[Klipper] = get_instances(Klipper)
+    Logger.print_status(f"Removing {client_config.name} ...")
-    kl_instances = []
+    client_config_dir = client_config.config_dir
+    if not client_config_dir.exists():
+        Logger.print_info(f"'{client_config_dir}' does not exist. Skipping ...")
+        return
+
+    try:
+        shutil.rmtree(client_config_dir)
+    except OSError as e:
+        Logger.print_error(f"Unable to delete '{client_config_dir}':\n{e}")
+
+
+def remove_client_config_symlink(client_config: BaseWebClientConfig) -> None:
+    im = InstanceManager(Klipper)
+    instances: List[Klipper] = im.instances
     for instance in instances:
-        cfg = instance.base.cfg_dir.joinpath(client_config.config_filename)
+        Logger.print_status(f"Removing symlink from '{instance.cfg_dir}' ...")
-        if run_remove_routines(cfg):
+        symlink = instance.cfg_dir.joinpath(client_config.config_filename)
-            kl_instances.append(instance)
+        if not symlink.is_symlink():
-    text = f"{client_config.display_name} removed from instance"
+            Logger.print_info(f"'{symlink}' does not exist. Skipping ...")
-    return update_msg(kl_instances, message, text)
+            continue
 
-
+        try:
-def remove_printer_config_section(
+            remove_file(symlink)
-    message: Message, client_config: BaseWebClientConfig, kl_instances: List[Klipper]
+        except subprocess.CalledProcessError:
-) -> Message:
+            Logger.print_error("Failed to remove symlink!")
-    kl_section = client_config.config_section
-    kl_instances = remove_config_section(kl_section, kl_instances)
-    text = f"Klipper config section '{kl_section}' removed for instance"
-    return update_msg(kl_instances, message, text)
-
-
-def remove_moonraker_config_section(
-    message: Message, client_config: BaseWebClientConfig, mr_instances: List[Moonraker]
-) -> Message:
-    mr_section = f"update_manager {client_config.name}"
-    mr_instances = remove_config_section(mr_section, mr_instances)
-    text = f"Moonraker config section '{mr_section}' removed for instance"
-    return update_msg(mr_instances, message, text)
-
-
-def update_msg(instances: List[InstanceType], message: Message, text: str) -> Message:
-    if not instances:
-        return message
-
-    instance_names = [i.service_file_path.stem for i in instances]
-    message.text.append(f"● {text}: {', '.join(instance_names)}")
-    return message

kiauh/components/webui_client/client_config/client_config_setup.py

@@ -1,12 +1,11 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import shutil
 import subprocess
@@ -21,24 +20,23 @@ from components.webui_client.client_dialogs import (
 )
 from components.webui_client.client_utils import (
     backup_client_config_data,
-    detect_client_cfg_conflict,
+    config_for_other_client_exist,
 )
 from core.instance_manager.instance_manager import InstanceManager
-from core.logger import Logger
 from core.settings.kiauh_settings import KiauhSettings
 from utils.common import backup_printer_config_dir
 from utils.config_utils import add_config_section, add_config_section_at_top
 from utils.fs_utils import create_symlink
 from utils.git_utils import git_clone_wrapper, git_pull_wrapper
 from utils.input_utils import get_confirm
-from utils.instance_utils import get_instances
+from utils.logger import Logger
 
 
-def install_client_config(client_data: BaseWebClient, cfg_backup=True) -> None:
+def install_client_config(client_data: BaseWebClient) -> None:
     client_config: BaseWebClientConfig = client_data.client_config
     display_name = client_config.display_name
 
-    if detect_client_cfg_conflict(client_data):
+    if config_for_other_client_exist(client_data.client):
         Logger.print_info("Another Client-Config is already installed! Skipped ...")
         return
 
@@ -49,15 +47,16 @@ def install_client_config(client_data: BaseWebClient, cfg_backup=True) -> None:
         else:
             return
 
-    mr_instances: List[Moonraker] = get_instances(Moonraker)
+    mr_im = InstanceManager(Moonraker)
-    kl_instances = get_instances(Klipper)
+    mr_instances: List[Moonraker] = mr_im.instances
+    kl_im = InstanceManager(Klipper)
+    kl_instances = kl_im.instances
 
     try:
         download_client_config(client_config)
         create_client_config_symlink(client_config, kl_instances)
 
-        if cfg_backup:
+        backup_printer_config_dir()
-            backup_printer_config_dir()
 
         add_config_section(
             section=f"update_manager {client_config.name}",
@@ -71,7 +70,7 @@ def install_client_config(client_data: BaseWebClient, cfg_backup=True) -> None:
             ],
         )
         add_config_section_at_top(client_config.config_section, kl_instances)
-        InstanceManager.restart_all(kl_instances)
+        kl_im.restart_all_instance()
 
     except Exception as e:
         Logger.print_error(f"{display_name} installation failed!\n{e}")
@@ -113,12 +112,16 @@ def update_client_config(client: BaseWebClient) -> None:
 
 
 def create_client_config_symlink(
-    client_config: BaseWebClientConfig, klipper_instances: List[Klipper]
+    client_config: BaseWebClientConfig, klipper_instances: List[Klipper] = None
 ) -> None:
+    if klipper_instances is None:
+        kl_im = InstanceManager(Klipper)
+        klipper_instances = kl_im.instances
+
+    Logger.print_status(f"Create symlink for {client_config.config_filename} ...")
+    source = Path(client_config.config_dir, client_config.config_filename)
     for instance in klipper_instances:
-        Logger.print_status(f"Create symlink for {client_config.config_filename} ...")
+        target = instance.cfg_dir
-        source = Path(client_config.config_dir, client_config.config_filename)
-        target = instance.base.cfg_dir
         Logger.print_status(f"Linking {source} to {target}")
         try:
             create_symlink(source, target)

kiauh/components/webui_client/client_dialogs.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -7,87 +7,102 @@
 #  This file may be distributed under the terms of the GNU GPLv3 license  #
 # ======================================================================= #
 
+import textwrap
 from typing import List
 
 from components.webui_client.base_data import BaseWebClient
-from core.logger import DialogType, Logger
+from core.menus.base_menu import print_back_footer
+from utils.constants import COLOR_CYAN, COLOR_YELLOW, RESET_FORMAT
 
 
-def print_moonraker_not_found_dialog(name: str) -> None:
+def print_moonraker_not_found_dialog():
-    Logger.print_dialog(
+    line1 = f"{COLOR_YELLOW}WARNING:{RESET_FORMAT}"
-        DialogType.WARNING,
+    line2 = f"{COLOR_YELLOW}No local Moonraker installation was found!{RESET_FORMAT}"
-        [
+    dialog = textwrap.dedent(
-            "No local Moonraker installation was found!",
+        f"""
-            "\n\n",
+        /=======================================================\\
-            f"It is possible to install {name} without a local Moonraker installation. "
+        | {line1:<63}|
-            "If you continue, you need to make sure, that Moonraker is installed on "
+        | {line2:<63}|
-            f"another machine in your network. Otherwise {name} will NOT work "
+        |-------------------------------------------------------|
-            "correctly.",
+        | It is possible to install Mainsail without a local    |
-        ],
+        | Moonraker installation. If you continue, you need to  |
-    )
+        | make sure, that Moonraker is installed on another     |
+        | machine in your network. Otherwise Mainsail will NOT  |
+        | work correctly.                                       |
+        """
+    )[1:]
+
+    print(dialog, end="")
+    print_back_footer()
 
 
-def print_client_already_installed_dialog(name: str) -> None:
+def print_client_already_installed_dialog(name: str):
-    Logger.print_dialog(
+    line1 = f"{COLOR_YELLOW}WARNING:{RESET_FORMAT}"
-        DialogType.WARNING,
+    line2 = f"{COLOR_YELLOW}{name} seems to be already installed!{RESET_FORMAT}"
-        [
+    line3 = f"If you continue, your current {name}"
-            f"{name} seems to be already installed!",
+    dialog = textwrap.dedent(
-            f"If you continue, your current {name} installation will be overwritten.",
+        f"""
-        ],
+        /=======================================================\\
-    )
+        | {line1:<63}|
+        | {line2:<63}|
+        |-------------------------------------------------------|
+        | {line3:<54}|
+        | installation will be overwritten.                     |
+        """
+    )[1:]
+
+    print(dialog, end="")
+    print_back_footer()
 
 
-def print_client_port_select_dialog(
+def print_client_port_select_dialog(name: str, port: int, ports_in_use: List[int]):
-    name: str, port: int, ports_in_use: List[int]
+    port = f"{COLOR_CYAN}{port}{RESET_FORMAT}"
-) -> None:
+    line1 = f"Please select the port, {name} should be served on."
-    dialog_content: List[str] = [
+    line2 = f"In case you need {name} to be served on a specific"
-        f"Please select the port, {name} should be served on. If your are unsure "
+    dialog = textwrap.dedent(
-        f"what to select, hit Enter to apply the suggested value of: {port}",
+        f"""
-        "\n\n",
+        /=======================================================\\
-        f"In case you need {name} to be served on a specific port, you can set it "
+        | {line1:<54}|
-        f"now. Make sure that the port is not already used by another application "
+        | If you are unsure what to select, hit Enter to apply  |
-        f"on your system!",
+        | the suggested value of: {port:38} |
-    ]
+        |                                                       |
+        | {line2:<54}|
+        | port, you can set it now. Make sure the port is not   |
+        | used by any other application on your system!         |
+        """
+    )[1:]
 
-    if ports_in_use:
+    if len(ports_in_use) > 0:
-        dialog_content.extend(
+        dialog += "|-------------------------------------------------------|\n"
-            [
+        dialog += "| The following ports were found to be in use already:  |\n"
-                "\n\n",
+        for port in ports_in_use:
-                "The following ports were found to be already in use:",
+            port = f"{COLOR_CYAN}● {port}{RESET_FORMAT}"
-                *[f"● {p}" for p in ports_in_use if p != port],
+            dialog += f"|  {port:60}  |\n"
-            ]
-        )
 
-    Logger.print_dialog(DialogType.CUSTOM, dialog_content)
+    dialog += "\\=======================================================/\n"
+
+    print(dialog, end="")
 
 
-def print_install_client_config_dialog(client: BaseWebClient) -> None:
+def print_install_client_config_dialog(client: BaseWebClient):
     name = client.display_name
     url = client.client_config.repo_url.replace(".git", "")
-    Logger.print_dialog(
+    line1 = f"have {name} fully functional and working."
-        DialogType.INFO,
+    line2 = f"The recommended macros for {name} can be seen here:"
-        [
+    dialog = textwrap.dedent(
-            f"It is recommended to use special macros in order to have {name} fully "
+        f"""
-            f"functional and working.",
+        /=======================================================\\
-            "\n\n",
+        | It is recommended to use special macros in order to   |
-            f"The recommended macros for {name} can be seen here:",
+        | {line1:<54}|
-            url,
+        |                                                       |
-            "\n\n",
+        | {line2:<54}|
-            "If you already use these macros skip this step. Otherwise you should "
+        | {url:<54}|
-            "consider to answer with 'Y' to download the recommended macros.",
+        |                                                       |
-        ],
+        | If you already use these macros skip this step.       |
-    )
+        | Otherwise you should consider to answer with 'Y' to   |
+        | download the recommended macros.                      |
+        \\=======================================================/
+        """
+    )[1:]
 
-
+    print(dialog, end="")
-def print_ipv6_warning_dialog() -> None:
-    Logger.print_dialog(
-        DialogType.WARNING,
-        [
-            "It looks like IPv6 is enabled on this system!",
-            "This may cause issues with the installation of NGINX in the following "
-            "steps! It is recommended to disable IPv6 on your system to avoid this issue.",
-            "\n\n",
-            "If you think this warning is a false alarm, and you are sure that "
-            "IPv6 is disabled, you can continue with the installation.",
-        ],
-    )

kiauh/components/webui_client/client_remove.py

@@ -1,112 +1,74 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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 typing import List
 
 from components.klipper.klipper import Klipper
 from components.moonraker.moonraker import Moonraker
 from components.webui_client.base_data import (
     BaseWebClient,
+    WebClientType,
 )
 from components.webui_client.client_config.client_config_remove import (
     run_client_config_removal,
 )
-from core.backup_manager.backup_manager import BackupManager
+from components.webui_client.client_utils import backup_mainsail_config_json
-from core.constants import NGINX_SITES_AVAILABLE, NGINX_SITES_ENABLED
+from core.instance_manager.instance_manager import InstanceManager
-from core.logger import Logger
-from core.services.message_service import Message
-from core.types.color import Color
 from utils.config_utils import remove_config_section
 from utils.fs_utils import (
-    remove_with_sudo,
+    remove_nginx_config,
-    run_remove_routines,
+    remove_nginx_logs,
 )
-from utils.instance_utils import get_instances
+from utils.logger import Logger
 
 
 def run_client_removal(
     client: BaseWebClient,
-    remove_client: bool,
+    rm_client: bool,
-    remove_client_cfg: bool,
+    rm_client_config: bool,
-    backup_config: bool,
+    backup_ms_config_json: bool,
-) -> Message:
+) -> None:
-    completion_msg = Message(
+    mr_im = InstanceManager(Moonraker)
-        title=f"{client.display_name} Removal Process completed",
+    mr_instances: List[Moonraker] = mr_im.instances
-        color=Color.GREEN,
+    kl_im = InstanceManager(Klipper)
-    )
+    kl_instances: List[Klipper] = kl_im.instances
-    mr_instances: List[Moonraker] = get_instances(Moonraker)
-    kl_instances: List[Klipper] = get_instances(Klipper)
 
-    if backup_config:
+    if backup_ms_config_json and client.client == WebClientType.MAINSAIL:
-        bm = BackupManager()
+        backup_mainsail_config_json()
-        if bm.backup_file(client.config_file):
-            completion_msg.text.append(f"● {client.config_file.name} backup created")
 
-    if remove_client:
+    if rm_client:
         client_name = client.name
-        if remove_client_dir(client):
+        remove_client_dir(client)
-            completion_msg.text.append(f"● {client.display_name} removed")
+        remove_nginx_config(client_name)
-        if remove_client_nginx_config(client_name):
+        remove_nginx_logs(client_name, kl_instances)
-            completion_msg.text.append("● NGINX config removed")
-        if remove_client_nginx_logs(client, kl_instances):
-            completion_msg.text.append("● NGINX logs removed")
 
         section = f"update_manager {client_name}"
-        handled_instances: List[Moonraker] = remove_config_section(
+        remove_config_section(section, mr_instances)
-            section, mr_instances
-        )
-        if handled_instances:
-            names = [i.service_file_path.stem for i in handled_instances]
-            completion_msg.text.append(
-                f"● Moonraker config section '{section}' removed for instance: {', '.join(names)}"
-            )
 
-    if remove_client_cfg:
+    if rm_client_config:
-        cfg_completion_msg = run_client_config_removal(
+        run_client_config_removal(
             client.client_config,
             kl_instances,
             mr_instances,
         )
-        if cfg_completion_msg.color == Color.GREEN:
-            completion_msg.text.extend(cfg_completion_msg.text[1:])
-
-    if not completion_msg.text:
-        completion_msg.color = Color.YELLOW
-        completion_msg.centered = True
-        completion_msg.text.append("Nothing to remove.")
-    else:
-        completion_msg.text.insert(0, "The following actions were performed:")
-
-    return completion_msg
 
 
-def remove_client_dir(client: BaseWebClient) -> bool:
+def remove_client_dir(client: BaseWebClient) -> None:
     Logger.print_status(f"Removing {client.display_name} ...")
-    return run_remove_routines(client.client_dir)
+    client_dir = client.client_dir
+    if not client.client_dir.exists():
+        Logger.print_info(f"'{client_dir}' does not exist. Skipping ...")
+        return
 
-
+    try:
-def remove_client_nginx_config(name: str) -> bool:
+        shutil.rmtree(client_dir)
-    Logger.print_status(f"Removing NGINX config for {name.capitalize()} ...")
+    except OSError as e:
-    return remove_with_sudo(
+        Logger.print_error(f"Unable to delete '{client_dir}':\n{e}")
-        [
-            NGINX_SITES_AVAILABLE.joinpath(name),
-            NGINX_SITES_ENABLED.joinpath(name),
-        ]
-    )
-
-
-def remove_client_nginx_logs(client: BaseWebClient, instances: List[Klipper]) -> bool:
-    Logger.print_status(f"Removing NGINX logs for {client.display_name} ...")
-
-    files = [client.nginx_access_log, client.nginx_error_log]
-    if instances:
-        for instance in instances:
-            files.append(instance.base.log_dir.joinpath(client.nginx_access_log.name))
-            files.append(instance.base.log_dir.joinpath(client.nginx_error_log.name))
-
-    return remove_with_sudo(files)

kiauh/components/webui_client/client_setup.py

@@ -1,13 +1,12 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
+
-import tempfile
 from pathlib import Path
 from typing import List
 
@@ -23,27 +22,32 @@ from components.webui_client.client_config.client_config_setup import (
     install_client_config,
 )
 from components.webui_client.client_dialogs import (
+    print_client_port_select_dialog,
     print_install_client_config_dialog,
     print_moonraker_not_found_dialog,
 )
 from components.webui_client.client_utils import (
-    copy_common_vars_nginx_cfg,
+    backup_mainsail_config_json,
-    copy_upstream_nginx_cfg,
+    config_for_other_client_exist,
-    create_nginx_cfg,
-    detect_client_cfg_conflict,
     enable_mainsail_remotemode,
-    get_client_port_selection,
+    restore_mainsail_config_json,
     symlink_webui_nginx_log,
 )
 from core.instance_manager.instance_manager import InstanceManager
-from core.logger import DialogType, Logger
 from core.settings.kiauh_settings import KiauhSettings
-from core.types.color import Color
+from utils.common import check_install_dependencies
-from utils.common import backup_printer_config_dir, check_install_dependencies
 from utils.config_utils import add_config_section
-from utils.fs_utils import unzip
+from utils.fs_utils import (
-from utils.input_utils import get_confirm
+    copy_common_vars_nginx_cfg,
-from utils.instance_utils import get_instances
+    copy_upstream_nginx_cfg,
+    create_nginx_cfg,
+    get_next_free_port,
+    is_valid_port,
+    read_ports_from_nginx_configs,
+    unzip,
+)
+from utils.input_utils import get_confirm, get_number_input
+from utils.logger import Logger
 from utils.sys_utils import (
     cmd_sysctl_service,
     download_file,
@@ -51,17 +55,26 @@ from utils.sys_utils import (
 )
 
 
-def install_client(
+def install_client(client: BaseWebClient) -> None:
-    client: BaseWebClient,
+    if client is None:
-    settings: KiauhSettings,
+        raise ValueError("Missing parameter client_data!")
-    reinstall: bool = False,
+
-) -> None:
+    if client.client_dir.exists():
-    mr_instances: List[Moonraker] = get_instances(Moonraker)
+        Logger.print_info(
+            f"{client.display_name} seems to be already installed! Skipped ..."
+        )
+        return
+
+    mr_im = InstanceManager(Moonraker)
+    mr_instances: List[Moonraker] = mr_im.instances
 
     enable_remotemode = False
     if not mr_instances:
-        print_moonraker_not_found_dialog(client.display_name)
+        print_moonraker_not_found_dialog()
-        if not get_confirm(f"Continue {client.display_name} installation?"):
+        if not get_confirm(
+            f"Continue {client.display_name} installation?",
+            allow_go_back=True,
+        ):
             return
 
     # if moonraker is not installed or multiple instances
@@ -73,45 +86,55 @@ def install_client(
     ):
         enable_remotemode = True
 
-    kl_instances = get_instances(Klipper)
+    kl_im = InstanceManager(Klipper)
+    kl_instances = kl_im.instances
     install_client_cfg = False
     client_config: BaseWebClientConfig = client.client_config
     if (
         kl_instances
         and not client_config.config_dir.exists()
-        and not detect_client_cfg_conflict(client)
+        and not config_for_other_client_exist(client_to_ignore=client.client)
     ):
         print_install_client_config_dialog(client)
         question = f"Download the recommended {client_config.display_name}?"
         install_client_cfg = get_confirm(question, allow_go_back=False)
 
-    default_port: int = int(settings.get(client.name, "port"))
+    settings = KiauhSettings()
-    port: int = (
+    port: int = settings.get(client.name, "port")
-        default_port if reinstall else get_client_port_selection(client, settings)
+    ports_in_use: List[int] = read_ports_from_nginx_configs()
-    )
 
-    check_install_dependencies({"nginx"})
+    # check if configured port is a valid number and not in use already
+    valid_port = is_valid_port(port, ports_in_use)
+    while not valid_port:
+        next_port = get_next_free_port(ports_in_use)
+        print_client_port_select_dialog(client.display_name, next_port, ports_in_use)
+        port = get_number_input(
+            f"Configure {client.display_name} for port",
+            min_count=int(next_port),
+            default=next_port,
+        )
+        valid_port = is_valid_port(port, ports_in_use)
+
+    check_install_dependencies(["nginx", "unzip"])
 
     try:
         download_client(client)
         if enable_remotemode and client.client == WebClientType.MAINSAIL:
             enable_mainsail_remotemode()
-
+        if mr_instances:
-        backup_printer_config_dir()
+            add_config_section(
-        add_config_section(
+                section=f"update_manager {client.name}",
-            section=f"update_manager {client.name}",
+                instances=mr_instances,
-            instances=mr_instances,
+                options=[
-            options=[
+                    ("type", "web"),
-                ("type", "web"),
+                    ("channel", "stable"),
-                ("channel", "stable"),
+                    ("repo", str(client.repo_path)),
-                ("repo", str(client.repo_path)),
+                    ("path", str(client.client_dir)),
-                ("path", str(client.client_dir)),
+                ],
-            ],
+            )
-        )
+            mr_im.restart_all_instance()
-        InstanceManager.restart_all(mr_instances)
-
         if install_client_cfg and kl_instances:
-            install_client_config(client, False)
+            install_client_config(client)
 
         copy_upstream_nginx_cfg()
         copy_common_vars_nginx_cfg()
@@ -125,28 +148,16 @@ def install_client(
         )
 
         if kl_instances:
-            symlink_webui_nginx_log(client, kl_instances)
+            symlink_webui_nginx_log(kl_instances)
         cmd_sysctl_service("nginx", "restart")
 
     except Exception as e:
-        Logger.print_error(e)
+        Logger.print_error(f"{client.display_name} installation failed!\n{e}")
-        Logger.print_dialog(
-            DialogType.ERROR,
-            center_content=True,
-            content=[f"{client.display_name} installation failed!"],
-        )
         return
 
-    # noinspection HttpUrlsUsage
+    log = f"Open {client.display_name} now on: http://{get_ipv4_addr()}:{port}"
-    Logger.print_dialog(
+    Logger.print_ok(f"{client.display_name} installation complete!", start="\n")
-        DialogType.CUSTOM,
+    Logger.print_ok(log, prefix=False, end="\n\n")
-        custom_title=f"{client.display_name} installation complete!",
-        custom_color=Color.GREEN,
-        center_content=True,
-        content=[
-            f"Open {client.display_name} now on: http://{get_ipv4_addr()}{'' if port == 80 else f':{port}'}",
-        ],
-    )
 
 
 def download_client(client: BaseWebClient) -> None:
@@ -177,10 +188,10 @@ def update_client(client: BaseWebClient) -> None:
         )
         return
 
-    with tempfile.NamedTemporaryFile(suffix=".json") as tmp_file:
+    if client.client == WebClientType.MAINSAIL:
-        Logger.print_status(
+        backup_mainsail_config_json(is_temp=True)
-            f"Creating temporary backup of {client.config_file} as {tmp_file.name} ..."
+
-        )
+    download_client(client)
-        shutil.copy(client.config_file, tmp_file.name)
+
-        download_client(client)
+    if client.client == WebClientType.MAINSAIL:
-        shutil.copy(tmp_file.name, client.config_file)
+        restore_mainsail_config_json()

kiauh/components/webui_client/client_utils.py

@@ -1,50 +1,35 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
-import json
+import json  # noqa: I001
-import re
 import shutil
 from pathlib import Path
-from subprocess import PIPE, CalledProcessError, run
 from typing import List, get_args
 
 from components.klipper.klipper import Klipper
-from components.webui_client import MODULE_PATH
 from components.webui_client.base_data import (
     BaseWebClient,
+    BaseWebClientConfig,
     WebClientType,
 )
-from components.webui_client.client_dialogs import print_client_port_select_dialog
-from components.webui_client.fluidd_data import FluiddData
 from components.webui_client.mainsail_data import MainsailData
 from core.backup_manager.backup_manager import BackupManager
-from core.constants import (
+from core.settings.kiauh_settings import KiauhSettings
-    NGINX_CONFD,
+from utils import NGINX_CONFD, NGINX_SITES_AVAILABLE
-    NGINX_SITES_AVAILABLE,
-    NGINX_SITES_ENABLED,
-)
-from core.logger import Logger
-from core.settings.kiauh_settings import KiauhSettings, WebUiSettings
-from core.submodules.simple_config_parser.src.simple_config_parser.simple_config_parser import (
-    SimpleConfigParser,
-)
-from core.types.color import Color
-from core.types.component_status import ComponentStatus
 from utils.common import get_install_status
-from utils.fs_utils import create_symlink, remove_file
+from utils.constants import COLOR_CYAN, COLOR_YELLOW, RESET_FORMAT
 from utils.git_utils import (
-    get_latest_remote_tag,
+    get_latest_tag,
     get_latest_unstable_tag,
 )
-from utils.input_utils import get_number_input
+from utils.logger import Logger
-from utils.instance_utils import get_instances
+from utils.types import ComponentStatus, InstallStatus
 
 
 def get_client_status(
@@ -55,62 +40,57 @@ def get_client_status(
         NGINX_CONFD.joinpath("upstreams.conf"),
         NGINX_CONFD.joinpath("common_vars.conf"),
     ]
-    comp_status: ComponentStatus = get_install_status(client.client_dir, files=files)
+    status = get_install_status(client.client_dir, files=files)
 
     # if the client dir does not exist, set the status to not
     # installed even if the other files are present
     if not client.client_dir.exists():
-        comp_status.status = 0
+        status["status"] = InstallStatus.NOT_INSTALLED
 
-    comp_status.local = get_local_client_version(client)
+    status["local"] = get_local_client_version(client)
-    comp_status.remote = get_remote_client_version(client) if fetch_remote else None
+    status["remote"] = get_remote_client_version(client) if fetch_remote else None
-    return comp_status
+    return status
 
 
 def get_client_config_status(client: BaseWebClient) -> ComponentStatus:
     return get_install_status(client.client_config.config_dir)
 
 
-def get_current_client_config() -> str:
+def get_current_client_config(clients: List[BaseWebClient]) -> str:
-    mainsail, fluidd = MainsailData(), FluiddData()
+    installed = []
-    clients: List[BaseWebClient] = [mainsail, fluidd]
+    for client in clients:
-    installed = [c for c in clients if c.client_config.config_dir.exists()]
+        client_config = client.client_config
+        if client_config.config_dir.exists():
+            installed.append(client)
 
-    if not installed:
+    if len(installed) > 1:
-        return Color.apply("-", Color.CYAN)
+        return f"{COLOR_YELLOW}Conflict!{RESET_FORMAT}"
     elif len(installed) == 1:
         cfg = installed[0].client_config
-        return Color.apply(cfg.display_name, Color.CYAN)
+        return f"{COLOR_CYAN}{cfg.display_name}{RESET_FORMAT}"
 
-    # at this point, both client config folders exists, so we need to check
+    return f"{COLOR_CYAN}-{RESET_FORMAT}"
-    # which are actually included in the printer.cfg of all klipper instances
-    mainsail_includes, fluidd_includes = [], []
-    klipper_instances: List[Klipper] = get_instances(Klipper)
-    for instance in klipper_instances:
-        scp = SimpleConfigParser()
-        scp.read_file(instance.cfg_file)
-        includes_mainsail = scp.has_section(mainsail.client_config.config_section)
-        includes_fluidd = scp.has_section(fluidd.client_config.config_section)
 
-        if includes_mainsail:
-            mainsail_includes.append(instance)
-        if includes_fluidd:
-            fluidd_includes.append(instance)
 
-        # if both are included in the same file, we have a potential conflict
+def backup_mainsail_config_json(is_temp=False) -> None:
-        if includes_mainsail and includes_fluidd:
+    c_json = MainsailData().client_dir.joinpath("config.json")
-            return Color.apply("Conflict", Color.YELLOW)
+    Logger.print_status(f"Backup '{c_json}' ...")
-
+    bm = BackupManager()
-    if not mainsail_includes and not fluidd_includes:
+    if is_temp:
-        # there are no includes at all, even though the client config folders exist
+        fn = Path.home().joinpath("config.json.kiauh.bak")
-        return Color.apply("-", Color.CYAN)
+        bm.backup_file(c_json, custom_filename=fn)
-    elif len(fluidd_includes) > len(mainsail_includes):
-        # there are more instances that include fluidd than mainsail
-        return Color.apply(fluidd.client_config.display_name, Color.CYAN)
     else:
-        # there are the same amount of non-conflicting includes for each config
+        bm.backup_file(c_json)
-        # or more instances include mainsail than fluidd
+
-        return Color.apply(mainsail.client_config.display_name, Color.CYAN)
+
+def restore_mainsail_config_json() -> None:
+    try:
+        c_json = MainsailData().client_dir.joinpath("config.json")
+        Logger.print_status(f"Restore '{c_json}' ...")
+        source = Path.home().joinpath("config.json.kiauh.bak")
+        shutil.copy(source, c_json)
+    except OSError:
+        Logger.print_info("Unable to restore config.json. Skipped ...")
 
 
 def enable_mainsail_remotemode() -> None:
@@ -131,47 +111,45 @@ def enable_mainsail_remotemode() -> None:
     Logger.print_ok("Mainsails remote mode enabled!")
 
 
-def symlink_webui_nginx_log(
+def symlink_webui_nginx_log(klipper_instances: List[Klipper]) -> None:
-    client: BaseWebClient, klipper_instances: List[Klipper]
-) -> None:
     Logger.print_status("Link NGINX logs into log directory ...")
-    access_log = client.nginx_access_log
+    access_log = Path("/var/log/nginx/mainsail-access.log")
-    error_log = client.nginx_error_log
+    error_log = Path("/var/log/nginx/mainsail-error.log")
 
     for instance in klipper_instances:
-        desti_access = instance.base.log_dir.joinpath(access_log.name)
+        desti_access = instance.log_dir.joinpath("mainsail-access.log")
         if not desti_access.exists():
             desti_access.symlink_to(access_log)
 
-        desti_error = instance.base.log_dir.joinpath(error_log.name)
+        desti_error = instance.log_dir.joinpath("mainsail-error.log")
         if not desti_error.exists():
             desti_error.symlink_to(error_log)
 
 
-def get_local_client_version(client: BaseWebClient) -> str | None:
+def get_local_client_version(client: BaseWebClient) -> str:
     relinfo_file = client.client_dir.joinpath("release_info.json")
     version_file = client.client_dir.joinpath(".version")
 
     if not client.client_dir.exists():
-        return None
+        return "-"
     if not relinfo_file.is_file() and not version_file.is_file():
         return "n/a"
 
     if relinfo_file.is_file():
         with open(relinfo_file, "r") as f:
-            return str(json.load(f)["version"])
+            return json.load(f)["version"]
     else:
         with open(version_file, "r") as f:
             return f.readlines()[0]
 
 
-def get_remote_client_version(client: BaseWebClient) -> str | None:
+def get_remote_client_version(client: BaseWebClient) -> str:
     try:
-        if (tag := get_latest_remote_tag(client.repo_path)) != "":
+        if (tag := get_latest_tag(client.repo_path)) != "":
-            return str(tag)
+            return tag
-        return None
+        return "ERROR"
     except Exception:
-        return None
+        return "ERROR"
 
 
 def backup_client_data(client: BaseWebClient) -> None:
@@ -184,7 +162,9 @@ def backup_client_data(client: BaseWebClient) -> None:
 
     bm = BackupManager()
     bm.backup_directory(f"{name}-{version}", src, dest)
-    bm.backup_file(client.config_file, dest)
+    if name == "mainsail":
+        c_json = MainsailData().client_dir.joinpath("config.json")
+        bm.backup_file(c_json, dest)
     bm.backup_file(NGINX_SITES_AVAILABLE.joinpath(name), dest)
 
 
@@ -207,24 +187,30 @@ def get_existing_clients() -> List[BaseWebClient]:
     return installed_clients
 
 
-def detect_client_cfg_conflict(curr_client: BaseWebClient) -> bool:
+def get_existing_client_config() -> List[BaseWebClient]:
+    clients = list(get_args(WebClientType))
+    installed_client_configs: List[BaseWebClient] = []
+    for client in clients:
+        c_config_data: BaseWebClientConfig = client.client_config
+        if c_config_data.config_dir.exists():
+            installed_client_configs.append(client)
+
+    return installed_client_configs
+
+
+def config_for_other_client_exist(client_to_ignore: WebClientType) -> bool:
     """
     Check if any other client configs are present on the system.
     It is usually not harmful, but chances are they can conflict each other.
     Multiple client configs are, at least, redundant to have them installed
-    :param curr_client: The client name to check for the conflict
+    :param client_to_ignore: The client name to ignore for the check
     :return: True, if other client configs were found, else False
     """
 
-    mainsail_cfg_status: ComponentStatus = get_client_config_status(MainsailData())
+    clients = set([c.name for c in get_existing_client_config()])
-    fluidd_cfg_status: ComponentStatus = get_client_config_status(FluiddData())
+    clients = clients - {client_to_ignore.value}
 
-    if curr_client.client == WebClientType.MAINSAIL and fluidd_cfg_status.status == 2:
+    return True if len(clients) > 0 else False
-        return True
-    if curr_client.client == WebClientType.FLUIDD and mainsail_cfg_status.status == 2:
-        return True
-
-    return False
 
 
 def get_download_url(base_url: str, client: BaseWebClient) -> str:
@@ -242,212 +228,3 @@ def get_download_url(base_url: str, client: BaseWebClient) -> str:
         return f"{base_url}/download/{unstable_tag}/{client.name}.zip"
     except Exception:
         return stable_url
-
-
-#################################################
-## NGINX RELATED FUNCTIONS
-#################################################
-
-
-def copy_upstream_nginx_cfg() -> None:
-    """
-    Creates an upstream.conf in /etc/nginx/conf.d
-    :return: None
-    """
-    source = MODULE_PATH.joinpath("assets/upstreams.conf")
-    target = NGINX_CONFD.joinpath("upstreams.conf")
-    try:
-        command = ["sudo", "cp", source, target]
-        run(command, stderr=PIPE, check=True)
-    except CalledProcessError as e:
-        log = f"Unable to create upstreams.conf: {e.stderr.decode()}"
-        Logger.print_error(log)
-        raise
-
-
-def copy_common_vars_nginx_cfg() -> None:
-    """
-    Creates a common_vars.conf in /etc/nginx/conf.d
-    :return: None
-    """
-    source = MODULE_PATH.joinpath("assets/common_vars.conf")
-    target = NGINX_CONFD.joinpath("common_vars.conf")
-    try:
-        command = ["sudo", "cp", source, target]
-        run(command, stderr=PIPE, check=True)
-    except CalledProcessError as e:
-        log = f"Unable to create upstreams.conf: {e.stderr.decode()}"
-        Logger.print_error(log)
-        raise
-
-
-def generate_nginx_cfg_from_template(name: str, template_src: Path, **kwargs) -> None:
-    """
-    Creates an NGINX config from a template file and
-    replaces all placeholders passed as kwargs. A placeholder must be defined
-    in the template file as %{placeholder}%.
-    :param name: name of the config to create
-    :param template_src: the path to the template file
-    :return: None
-    """
-    tmp = Path.home().joinpath(f"{name}.tmp")
-    shutil.copy(template_src, tmp)
-    with open(tmp, "r+") as f:
-        content = f.read()
-
-        for key, value in kwargs.items():
-            content = content.replace(f"%{key}%", str(value))
-
-        f.seek(0)
-        f.write(content)
-        f.truncate()
-
-    target = NGINX_SITES_AVAILABLE.joinpath(name)
-    try:
-        command = ["sudo", "mv", tmp, target]
-        run(command, stderr=PIPE, check=True)
-    except CalledProcessError as e:
-        log = f"Unable to create '{target}': {e.stderr.decode()}"
-        Logger.print_error(log)
-        raise
-
-
-def create_nginx_cfg(
-    display_name: str,
-    cfg_name: str,
-    template_src: Path,
-    **kwargs,
-) -> None:
-    from utils.sys_utils import set_nginx_permissions
-
-    try:
-        Logger.print_status(f"Creating NGINX config for {display_name} ...")
-
-        source = NGINX_SITES_AVAILABLE.joinpath(cfg_name)
-        target = NGINX_SITES_ENABLED.joinpath(cfg_name)
-        remove_file(Path("/etc/nginx/sites-enabled/default"), True)
-        generate_nginx_cfg_from_template(cfg_name, template_src=template_src, **kwargs)
-        create_symlink(source, target, True)
-        set_nginx_permissions()
-
-        Logger.print_ok(f"NGINX config for {display_name} successfully created.")
-    except Exception:
-        Logger.print_error(f"Creating NGINX config for {display_name} failed!")
-        raise
-
-
-def get_nginx_config_list() -> List[Path]:
-    """
-    Get a list of all NGINX config files in /etc/nginx/sites-enabled
-    :return: List of NGINX config files
-    """
-    configs: List[Path] = []
-    for config in NGINX_SITES_ENABLED.iterdir():
-        if not config.is_file():
-            continue
-        configs.append(config)
-    return configs
-
-
-def get_nginx_listen_port(config: Path) -> int | None:
-    """
-    Get the listen port from an NGINX config file
-    :param config: The NGINX config file to read the port from
-    :return: The listen port as int or None if not found/parsable
-    """
-
-    # noinspection HttpUrlsUsage
-    pattern = r"default_server|http://|https://|[;\[\]]"
-    port = ""
-    with open(config, "r") as cfg:
-        for line in cfg.readlines():
-            line = re.sub(pattern, "", line.strip())
-            if line.startswith("listen"):
-                if ":" not in line:
-                    port = line.split()[-1]
-                else:
-                    port = line.split(":")[-1]
-        try:
-            return int(port)
-        except ValueError:
-            Logger.print_error(
-                f"Unable to parse listen port {port} from {config.name}!"
-            )
-            return None
-
-
-def read_ports_from_nginx_configs() -> List[int]:
-    """
-    Helper function to iterate over all NGINX configs
-    and read all ports defined for listen
-    :return: A sorted list of listen ports
-    """
-    if not NGINX_SITES_ENABLED.exists():
-        return []
-
-    port_list: List[int] = []
-    for config in get_nginx_config_list():
-        port = get_nginx_listen_port(config)
-        if port is not None:
-            port_list.append(port)
-
-    return sorted(port_list, key=lambda x: int(x))
-
-
-def get_client_port_selection(
-    client: BaseWebClient,
-    settings: KiauhSettings,
-    reconfigure=False,
-) -> int:
-    default_port: int = int(settings.get(client.name, "port"))
-    ports_in_use: List[int] = read_ports_from_nginx_configs()
-    next_free_port: int = get_next_free_port(ports_in_use)
-
-    port: int = (
-        next_free_port
-        if not reconfigure and default_port in ports_in_use
-        else default_port
-    )
-
-    print_client_port_select_dialog(client.display_name, port, ports_in_use)
-
-    while True:
-        _type = "Reconfigure" if reconfigure else "Configure"
-        question = f"{_type} {client.display_name} for port"
-        port_input = get_number_input(question, min_count=80, default=port)
-
-        if port_input not in ports_in_use:
-            client_settings: WebUiSettings = settings[client.name]
-            client_settings.port = port_input
-            settings.save()
-
-            return port_input
-
-        Logger.print_error("This port is already in use. Please select another one.")
-
-
-def get_next_free_port(ports_in_use: List[int]) -> int:
-    valid_ports = set(range(80, 7125))
-    used_ports = set(map(int, ports_in_use))
-
-    return min(valid_ports - used_ports)
-
-
-def set_listen_port(client: BaseWebClient, curr_port: int, new_port: int) -> None:
-    """
-    Set the port the client should listen on in the NGINX config
-    :param curr_port: The current port the client listens on
-    :param new_port: The new port to set
-    :param client: The client to set the port for
-    :return: None
-    """
-    config = NGINX_SITES_AVAILABLE.joinpath(client.name)
-    with open(config, "r") as f:
-        lines = f.readlines()
-
-    for i, line in enumerate(lines):
-        if "listen" in line:
-            lines[i] = line.replace(str(curr_port), str(new_port))
-
-    with open(config, "w") as f:
-        f.writelines(lines)

kiauh/components/webui_client/fluidd_data.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -18,11 +18,11 @@ from components.webui_client.base_data import (
     WebClientConfigType,
     WebClientType,
 )
+from components.webui_client.client_utils import get_download_url
 from core.backup_manager import BACKUP_ROOT_DIR
-from core.constants import NGINX_SITES_AVAILABLE
 
 
-@dataclass()
+@dataclass(frozen=True)
 class FluiddConfigWeb(BaseWebClientConfig):
     client_config: WebClientConfigType = WebClientConfigType.FLUIDD
     name: str = client_config.value
@@ -34,7 +34,7 @@ class FluiddConfigWeb(BaseWebClientConfig):
     repo_url: str = "https://github.com/fluidd-core/fluidd-config.git"
 
 
-@dataclass()
+@dataclass(frozen=True)
 class FluiddData(BaseWebClient):
     BASE_DL_URL = "https://github.com/fluidd-core/fluidd/releases"
 
@@ -42,17 +42,13 @@ class FluiddData(BaseWebClient):
     name: str = client.value
     display_name: str = name.capitalize()
     client_dir: Path = Path.home().joinpath("fluidd")
-    config_file: Path = client_dir.joinpath("config.json")
     backup_dir: Path = BACKUP_ROOT_DIR.joinpath("fluidd-backups")
     repo_path: str = "fluidd-core/fluidd"
-    nginx_config: Path = NGINX_SITES_AVAILABLE.joinpath("fluidd")
-    nginx_access_log: Path = Path("/var/log/nginx/fluidd-access.log")
-    nginx_error_log: Path = Path("/var/log/nginx/fluidd-error.log")
-    client_config: BaseWebClientConfig = None
-    download_url: str | None = None
 
-    def __post_init__(self):
+    @property
-        from components.webui_client.client_utils import get_download_url
+    def download_url(self) -> str:
+        return get_download_url(self.BASE_DL_URL, self)
 
-        self.client_config = FluiddConfigWeb()
+    @property
-        self.download_url = get_download_url(self.BASE_DL_URL, self)
+    def client_config(self) -> BaseWebClientConfig:
+        return FluiddConfigWeb()

kiauh/components/webui_client/mainsail_data.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -19,10 +19,9 @@ from components.webui_client.base_data import (
     WebClientType,
 )
 from core.backup_manager import BACKUP_ROOT_DIR
-from core.constants import NGINX_SITES_AVAILABLE
 
 
-@dataclass()
+@dataclass(frozen=True)
 class MainsailConfigWeb(BaseWebClientConfig):
     client_config: WebClientConfigType = WebClientConfigType.MAINSAIL
     name: str = client_config.value
@@ -34,7 +33,7 @@ class MainsailConfigWeb(BaseWebClientConfig):
     repo_url: str = "https://github.com/mainsail-crew/mainsail-config.git"
 
 
-@dataclass()
+@dataclass(frozen=True)
 class MainsailData(BaseWebClient):
     BASE_DL_URL: str = "https://github.com/mainsail-crew/mainsail/releases"
 
@@ -42,17 +41,15 @@ class MainsailData(BaseWebClient):
     name: str = WebClientType.MAINSAIL.value
     display_name: str = name.capitalize()
     client_dir: Path = Path.home().joinpath("mainsail")
-    config_file: Path = client_dir.joinpath("config.json")
     backup_dir: Path = BACKUP_ROOT_DIR.joinpath("mainsail-backups")
     repo_path: str = "mainsail-crew/mainsail"
-    nginx_config: Path = NGINX_SITES_AVAILABLE.joinpath("mainsail")
-    nginx_access_log: Path = Path("/var/log/nginx/mainsail-access.log")
-    nginx_error_log: Path = Path("/var/log/nginx/mainsail-error.log")
-    client_config: BaseWebClientConfig = None
-    download_url: str | None = None
 
-    def __post_init__(self):
+    @property
+    def download_url(self) -> str:
         from components.webui_client.client_utils import get_download_url
 
-        self.client_config = MainsailConfigWeb()
+        return get_download_url(self.BASE_DL_URL, self)
-        self.download_url = get_download_url(self.BASE_DL_URL, self)
+
+    @property
+    def client_config(self) -> BaseWebClientConfig:
+        return MainsailConfigWeb()

kiauh/components/webui_client/menus/client_install_menu.py

@@ -1,105 +0,0 @@
-# ======================================================================= #
-#  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

kiauh/components/webui_client/menus/client_remove_menu.py

@@ -1,114 +1,125 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
-from typing import Type
+from typing import Optional, Type
 
 from components.webui_client import client_remove
-from components.webui_client.base_data import BaseWebClient
+from components.webui_client.base_data import BaseWebClient, WebClientType
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import COLOR_CYAN, COLOR_RED, RESET_FORMAT
 
 
 # noinspection PyUnusedLocal
 class ClientRemoveMenu(BaseMenu):
     def __init__(
-        self, client: BaseWebClient, previous_menu: Type[BaseMenu] | None = None
+        self, client: BaseWebClient, previous_menu: Optional[Type[BaseMenu]] = None
     ):
         super().__init__()
-        self.title = f"Remove {client.display_name}"
+        self.previous_menu = previous_menu
-        self.title_color = Color.RED
+        self.client = client
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
+        self.rm_client = False
-        self.client: BaseWebClient = client
+        self.rm_client_config = False
-        self.remove_client: bool = False
+        self.backup_mainsail_config_json = False
-        self.remove_client_cfg: bool = False
-        self.backup_config_json: bool = False
-        self.select_state: bool = False
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.remove_menu import RemoveMenu
 
-        self.previous_menu = previous_menu if previous_menu is not None else RemoveMenu
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else RemoveMenu
+        )
 
     def set_options(self) -> None:
         self.options = {
-            "a": Option(method=self.toggle_all),
+            "0": Option(method=self.toggle_all, menu=False),
-            "1": Option(method=self.toggle_rm_client),
+            "1": Option(method=self.toggle_rm_client, menu=False),
-            "2": Option(method=self.toggle_rm_client_config),
+            "2": Option(method=self.toggle_rm_client_config, menu=False),
-            "3": Option(method=self.toggle_backup_config_json),
+            "c": Option(method=self.run_removal_process, menu=False),
-            "c": Option(method=self.run_removal_process),
         }
+        if self.client.client == WebClientType.MAINSAIL:
+            self.options["3"] = Option(self.toggle_backup_mainsail_config_json, False)
 
     def print_menu(self) -> None:
         client_name = self.client.display_name
         client_config = self.client.client_config
         client_config_name = client_config.display_name
 
-        checked = f"[{Color.apply('x', Color.CYAN)}]"
+        header = f" [ Remove {client_name} ] "
+        color = COLOR_RED
+        count = 62 - len(color) - len(RESET_FORMAT)
+        checked = f"[{COLOR_CYAN}x{RESET_FORMAT}]"
         unchecked = "[ ]"
-        o1 = checked if self.remove_client else unchecked
+        o1 = checked if self.rm_client else unchecked
-        o2 = checked if self.remove_client_cfg else unchecked
+        o2 = checked if self.rm_client_config else unchecked
-        o3 = checked if self.backup_config_json else unchecked
-        sel_state = f"{'Select' if not self.select_state else 'Deselect'} everything"
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ Enter a number and hit enter to select / deselect     ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║ the specific option for removal.                      ║
+            |-------------------------------------------------------|
-            ╟───────────────────────────────────────────────────────╢
+            | Enter a number and hit enter to select / deselect     |
-            ║  a) {sel_state:49} ║
+            | the specific option for removal.                      |
-            ╟───────────────────────────────────────────────────────╢
+            |-------------------------------------------------------|
-            ║  1) {o1} Remove {client_name:16}                       ║
+            |  0) Select everything                                 |
-            ║  2) {o2} Remove {client_config_name:24}               ║
+            |-------------------------------------------------------|
-            ║  3) {o3} Backup config.json                            ║
+            |  1) {o1} Remove {client_name:16}                       |
-            ╟───────────────────────────────────────────────────────╢
+            |  2) {o2} Remove {client_config_name:24}               |
-            ║  C) Continue                                          ║
+            """
-            ╟───────────────────────────────────────────────────────╢
+        )[1:]
+
+        if self.client.client == WebClientType.MAINSAIL:
+            o3 = checked if self.backup_mainsail_config_json else unchecked
+            menu += textwrap.dedent(
+                f"""
+                |  3) {o3} Backup config.json                            |
+                """
+            )[1:]
+
+        menu += textwrap.dedent(
+            """
+            |-------------------------------------------------------|
+            |  C) Continue                                          |
             """
         )[1:]
         print(menu, end="")
 
     def toggle_all(self, **kwargs) -> None:
-        self.select_state = not self.select_state
+        self.rm_client = True
-        self.remove_client = self.select_state
+        self.rm_client_config = True
-        self.remove_client_cfg = self.select_state
+        self.backup_mainsail_config_json = True
-        self.backup_config_json = self.select_state
 
     def toggle_rm_client(self, **kwargs) -> None:
-        self.remove_client = not self.remove_client
+        self.rm_client = not self.rm_client
 
     def toggle_rm_client_config(self, **kwargs) -> None:
-        self.remove_client_cfg = not self.remove_client_cfg
+        self.rm_client_config = not self.rm_client_config
 
-    def toggle_backup_config_json(self, **kwargs) -> None:
+    def toggle_backup_mainsail_config_json(self, **kwargs) -> None:
-        self.backup_config_json = not self.backup_config_json
+        self.backup_mainsail_config_json = not self.backup_mainsail_config_json
 
     def run_removal_process(self, **kwargs) -> None:
         if (
-            not self.remove_client
+            not self.rm_client
-            and not self.remove_client_cfg
+            and not self.rm_client_config
-            and not self.backup_config_json
+            and not self.backup_mainsail_config_json
         ):
-            print(Color.apply("Nothing selected ...", Color.RED))
+            error = f"{COLOR_RED}Nothing selected ...{RESET_FORMAT}"
+            print(error)
             return
 
-        completion_msg = client_remove.run_client_removal(
+        client_remove.run_client_removal(
             client=self.client,
-            remove_client=self.remove_client,
+            rm_client=self.rm_client,
-            remove_client_cfg=self.remove_client_cfg,
+            rm_client_config=self.rm_client_config,
-            backup_config=self.backup_config_json,
+            backup_ms_config_json=self.backup_mainsail_config_json,
         )
-        self.message_service.set_message(completion_msg)
 
-        self.remove_client = False
+        self.rm_client = False
-        self.remove_client_cfg = False
+        self.rm_client_config = False
-        self.backup_config_json = False
+        self.backup_mainsail_config_json = False
-        self.select_state = False

kiauh/core/backup_manager/__init__.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #

kiauh/core/backup_manager/backup_manager.py

@@ -1,32 +1,27 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import shutil
 from pathlib import Path
 from typing import List
 
 from core.backup_manager import BACKUP_ROOT_DIR
-from core.logger import Logger
 from utils.common import get_current_date
-
+from utils.logger import Logger
-
-class BackupManagerException(Exception):
-    pass
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class BackupManager:
     def __init__(self, backup_root_dir: Path = BACKUP_ROOT_DIR):
-        self._backup_root_dir: Path = backup_root_dir
+        self._backup_root_dir = backup_root_dir
-        self._ignore_folders: List[str] = []
+        self._ignore_folders = None
 
     @property
     def backup_root_dir(self) -> Path:
@@ -44,14 +39,12 @@ class BackupManager:
     def ignore_folders(self, value: List[str]):
         self._ignore_folders = value
 
-    def backup_file(
+    def backup_file(self, file: Path, target: Path = None, custom_filename=None):
-        self, file: Path, target: Path | None = None, custom_filename=None
-    ) -> bool:
         Logger.print_status(f"Creating backup of {file} ...")
 
         if not file.exists():
             Logger.print_info("File does not exist! Skipping ...")
-            return False
+            return
 
         target = self.backup_root_dir if target is None else target
 
@@ -64,45 +57,35 @@ class BackupManager:
                 Path(target).mkdir(exist_ok=True)
                 shutil.copyfile(file, target.joinpath(filename))
                 Logger.print_ok("Backup successful!")
-                return True
             except OSError as e:
                 Logger.print_error(f"Unable to backup '{file}':\n{e}")
-                return False
         else:
             Logger.print_info(f"File '{file}' not found ...")
-            return False
 
-    def backup_directory(
+    def backup_directory(self, name: str, source: Path, target: Path = None) -> None:
-        self, name: str, source: Path, target: Path | None = None
-    ) -> Path | None:
         Logger.print_status(f"Creating backup of {name} in {target} ...")
 
         if source is None or not Path(source).exists():
             Logger.print_info("Source directory does not exist! Skipping ...")
-            return None
+            return
 
         target = self.backup_root_dir if target is None else target
         try:
             date = get_current_date().get("date")
             time = get_current_date().get("time")
-            backup_target = target.joinpath(f"{name.lower()}-{date}-{time}")
             shutil.copytree(
                 source,
-                backup_target,
+                target.joinpath(f"{name.lower()}-{date}-{time}"),
                 ignore=self.ignore_folders_func,
-                ignore_dangling_symlinks=True,
             )
             Logger.print_ok("Backup successful!")
-
-            return backup_target
-
         except OSError as e:
             Logger.print_error(f"Unable to backup directory '{source}':\n{e}")
-            raise BackupManagerException(f"Unable to backup directory '{source}':\n{e}")
+            return
 
-    def ignore_folders_func(self, dirpath, filenames) -> List[str]:
+    def ignore_folders_func(self, dirpath, filenames):
         return (
             [f for f in filenames if f in self._ignore_folders]
-            if self._ignore_folders
+            if self._ignore_folders is not None
             else []
         )

kiauh/core/instance_manager/base_instance.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -9,50 +9,153 @@
 
 from __future__ import annotations
 
-import re
+from abc import ABC, abstractmethod
-from dataclasses import dataclass, field
 from pathlib import Path
-from typing import List
+from typing import List, Optional
 
-from utils.fs_utils import get_data_dir
+from utils.constants import CURRENT_USER, SYSTEMD
-
-SUFFIX_BLACKLIST: List[str] = ["None", "mcu", "obico", "bambu", "companion"]
 
 
-@dataclass(repr=True)
+class BaseInstance(ABC):
-class BaseInstance:
+    @classmethod
-    instance_type: type
+    def blacklist(cls) -> List[str]:
-    suffix: str
+        return []
-    log_file_name: str | None = None
-    data_dir: Path = field(init=False)
-    base_folders: List[Path] = field(init=False)
-    cfg_dir: Path = field(init=False)
-    log_dir: Path = field(init=False)
-    gcodes_dir: Path = field(init=False)
-    comms_dir: Path = field(init=False)
-    sysd_dir: Path = field(init=False)
-    is_legacy_instance: bool = field(init=False)
 
-    def __post_init__(self):
+    def __init__(
-        self.data_dir = get_data_dir(self.instance_type, self.suffix)
+        self,
-        # the following attributes require the data_dir to be set
+        suffix: str,
-        self.cfg_dir = self.data_dir.joinpath("config")
+        instance_type: BaseInstance,
-        self.log_dir = self.data_dir.joinpath("logs")
+    ):
-        self.gcodes_dir = self.data_dir.joinpath("gcodes")
+        self._instance_type = instance_type
-        self.comms_dir = self.data_dir.joinpath("comms")
+        self._suffix = suffix
-        self.sysd_dir = self.data_dir.joinpath("systemd")
+        self._user = CURRENT_USER
-        self.is_legacy_instance = self._set_is_legacy_instance()
+        self._data_dir_name = self.get_data_dir_name_from_suffix()
-        self.base_folders = [
+        self._data_dir = Path.home().joinpath(f"{self._data_dir_name}_data")
+        self._cfg_dir = self.data_dir.joinpath("config")
+        self._log_dir = self.data_dir.joinpath("logs")
+        self._comms_dir = self.data_dir.joinpath("comms")
+        self._sysd_dir = self.data_dir.joinpath("systemd")
+        self._gcodes_dir = self.data_dir.joinpath("gcodes")
+
+    @property
+    def instance_type(self) -> BaseInstance:
+        return self._instance_type
+
+    @instance_type.setter
+    def instance_type(self, value: BaseInstance) -> None:
+        self._instance_type = value
+
+    @property
+    def suffix(self) -> str:
+        return self._suffix
+
+    @suffix.setter
+    def suffix(self, value: str) -> None:
+        self._suffix = value
+
+    @property
+    def user(self) -> str:
+        return self._user
+
+    @user.setter
+    def user(self, value: str) -> None:
+        self._user = value
+
+    @property
+    def data_dir_name(self) -> str:
+        return self._data_dir_name
+
+    @data_dir_name.setter
+    def data_dir_name(self, value: str) -> None:
+        self._data_dir_name = value
+
+    @property
+    def data_dir(self) -> Path:
+        return self._data_dir
+
+    @data_dir.setter
+    def data_dir(self, value: Path) -> None:
+        self._data_dir = value
+
+    @property
+    def cfg_dir(self) -> Path:
+        return self._cfg_dir
+
+    @cfg_dir.setter
+    def cfg_dir(self, value: Path) -> None:
+        self._cfg_dir = value
+
+    @property
+    def log_dir(self) -> Path:
+        return self._log_dir
+
+    @log_dir.setter
+    def log_dir(self, value: Path) -> None:
+        self._log_dir = value
+
+    @property
+    def comms_dir(self) -> Path:
+        return self._comms_dir
+
+    @comms_dir.setter
+    def comms_dir(self, value: Path) -> None:
+        self._comms_dir = value
+
+    @property
+    def sysd_dir(self) -> Path:
+        return self._sysd_dir
+
+    @sysd_dir.setter
+    def sysd_dir(self, value: Path) -> None:
+        self._sysd_dir = value
+
+    @property
+    def gcodes_dir(self) -> Path:
+        return self._gcodes_dir
+
+    @gcodes_dir.setter
+    def gcodes_dir(self, value: Path) -> None:
+        self._gcodes_dir = value
+
+    @abstractmethod
+    def create(self) -> None:
+        raise NotImplementedError("Subclasses must implement the create method")
+
+    @abstractmethod
+    def delete(self) -> None:
+        raise NotImplementedError("Subclasses must implement the delete method")
+
+    def create_folders(self, add_dirs: Optional[List[Path]] = None) -> None:
+        dirs = [
             self.data_dir,
             self.cfg_dir,
             self.log_dir,
-            self.gcodes_dir,
             self.comms_dir,
             self.sysd_dir,
         ]
 
-    def _set_is_legacy_instance(self) -> bool:
+        if add_dirs:
-        legacy_pattern = r"^(?!printer)(.+)_data"
+            dirs.extend(add_dirs)
-        match = re.search(legacy_pattern, self.data_dir.name)
 
-        return True if (match and self.suffix != "") else False
+        for _dir in dirs:
+            _dir.mkdir(exist_ok=True)
+
+    def get_service_file_name(self, extension: bool = False) -> str:
+        from utils.common import convert_camelcase_to_kebabcase
+
+        name = convert_camelcase_to_kebabcase(self.__class__.__name__)
+        if self.suffix != "":
+            name += f"-{self.suffix}"
+
+        return name if not extension else f"{name}.service"
+
+    def get_service_file_path(self) -> Path:
+        return SYSTEMD.joinpath(self.get_service_file_name(extension=True))
+
+    def get_data_dir_name_from_suffix(self) -> str:
+        if self._suffix == "":
+            return "printer"
+        elif self._suffix.isdigit():
+            return f"printer_{self._suffix}"
+        else:
+            return self._suffix

kiauh/core/instance_manager/instance_manager.py

@@ -1,108 +1,188 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
+import re
+import subprocess
 from pathlib import Path
-from subprocess import CalledProcessError
+from typing import List, Optional, TypeVar, Union
-from typing import List
 
-from core.logger import Logger
+from core.instance_manager.base_instance import BaseInstance
-from utils.instance_type import InstanceType
+from utils.constants import SYSTEMD
+from utils.logger import Logger
 from utils.sys_utils import cmd_sysctl_service
 
+T = TypeVar(name="T", bound=BaseInstance, covariant=True)
 
+
+# noinspection PyMethodMayBeStatic
 class InstanceManager:
-    @staticmethod
+    def __init__(self, instance_type: T) -> None:
-    def enable(instance: InstanceType) -> None:
+        self._instance_type = instance_type
-        service_name: str = instance.service_file_path.name
+        self._current_instance: Optional[T] = None
+        self._instance_suffix: Optional[str] = None
+        self._instance_service: Optional[str] = None
+        self._instance_service_full: Optional[str] = None
+        self._instance_service_path: Optional[str] = None
+        self._instances: List[T] = []
+
+    @property
+    def instance_type(self) -> T:
+        return self._instance_type
+
+    @instance_type.setter
+    def instance_type(self, value: T):
+        self._instance_type = value
+
+    @property
+    def current_instance(self) -> T:
+        return self._current_instance
+
+    @current_instance.setter
+    def current_instance(self, value: T) -> None:
+        self._current_instance = value
+        self.instance_suffix = value.suffix
+        self.instance_service = value.get_service_file_name()
+        self.instance_service_path = value.get_service_file_path()
+
+    @property
+    def instance_suffix(self) -> str:
+        return self._instance_suffix
+
+    @instance_suffix.setter
+    def instance_suffix(self, value: str):
+        self._instance_suffix = value
+
+    @property
+    def instance_service(self) -> str:
+        return self._instance_service
+
+    @instance_service.setter
+    def instance_service(self, value: str):
+        self._instance_service = value
+
+    @property
+    def instance_service_full(self) -> str:
+        return f"{self._instance_service}.service"
+
+    @property
+    def instance_service_path(self) -> str:
+        return self._instance_service_path
+
+    @instance_service_path.setter
+    def instance_service_path(self, value: str):
+        self._instance_service_path = value
+
+    @property
+    def instances(self) -> List[T]:
+        return self.find_instances()
+
+    @instances.setter
+    def instances(self, value: List[T]):
+        self._instances = value
+
+    def create_instance(self) -> None:
+        if self.current_instance is not None:
+            try:
+                self.current_instance.create()
+            except (OSError, subprocess.CalledProcessError) as e:
+                Logger.print_error(f"Creating instance failed: {e}")
+                raise
+        else:
+            raise ValueError("current_instance cannot be None")
+
+    def delete_instance(self) -> None:
+        if self.current_instance is not None:
+            try:
+                self.current_instance.delete()
+            except (OSError, subprocess.CalledProcessError) as e:
+                Logger.print_error(f"Removing instance failed: {e}")
+                raise
+        else:
+            raise ValueError("current_instance cannot be None")
+
+    def enable_instance(self) -> None:
         try:
-            cmd_sysctl_service(service_name, "enable")
+            cmd_sysctl_service(self.instance_service_full, "enable")
-        except CalledProcessError as e:
+        except subprocess.CalledProcessError as e:
-            Logger.print_error(f"Error enabling service {service_name}:")
+            Logger.print_error(f"Error enabling service {self.instance_service_full}:")
             Logger.print_error(f"{e}")
 
-    @staticmethod
+    def disable_instance(self) -> None:
-    def disable(instance: InstanceType) -> None:
-        service_name: str = instance.service_file_path.name
         try:
-            cmd_sysctl_service(service_name, "disable")
+            cmd_sysctl_service(self.instance_service_full, "disable")
-        except CalledProcessError as e:
+        except subprocess.CalledProcessError as e:
-            Logger.print_error(f"Error disabling {service_name}: {e}")
+            Logger.print_error(f"Error disabling {self.instance_service_full}:")
+            Logger.print_error(f"{e}")
+
+    def start_instance(self) -> None:
+        try:
+            cmd_sysctl_service(self.instance_service_full, "start")
+        except subprocess.CalledProcessError as e:
+            Logger.print_error(f"Error starting {self.instance_service_full}:")
+            Logger.print_error(f"{e}")
+
+    def restart_instance(self) -> None:
+        try:
+            cmd_sysctl_service(self.instance_service_full, "restart")
+        except subprocess.CalledProcessError as e:
+            Logger.print_error(f"Error restarting {self.instance_service_full}:")
+            Logger.print_error(f"{e}")
+
+    def start_all_instance(self) -> None:
+        for instance in self.instances:
+            self.current_instance = instance
+            self.start_instance()
+
+    def restart_all_instance(self) -> None:
+        for instance in self.instances:
+            self.current_instance = instance
+            self.restart_instance()
+
+    def stop_instance(self) -> None:
+        try:
+            cmd_sysctl_service(self.instance_service_full, "stop")
+        except subprocess.CalledProcessError as e:
+            Logger.print_error(f"Error stopping {self.instance_service_full}:")
+            Logger.print_error(f"{e}")
             raise
 
-    @staticmethod
+    def stop_all_instance(self) -> None:
-    def start(instance: InstanceType) -> None:
+        for instance in self.instances:
-        service_name: str = instance.service_file_path.name
+            self.current_instance = instance
-        try:
+            self.stop_instance()
-            cmd_sysctl_service(service_name, "start")
-        except CalledProcessError as e:
-            Logger.print_error(f"Error starting {service_name}: {e}")
-            raise
 
-    @staticmethod
+    def find_instances(self) -> List[T]:
-    def stop(instance: InstanceType) -> None:
+        from utils.common import convert_camelcase_to_kebabcase
-        name: str = instance.service_file_path.name
-        try:
-            cmd_sysctl_service(name, "stop")
-        except CalledProcessError as e:
-            Logger.print_error(f"Error stopping {name}: {e}")
-            raise
 
-    @staticmethod
+        name = convert_camelcase_to_kebabcase(self.instance_type.__name__)
-    def restart(instance: InstanceType) -> None:
+        pattern = re.compile(f"^{name}(-[0-9a-zA-Z]+)?.service$")
-        name: str = instance.service_file_path.name
+        excluded = self.instance_type.blacklist()
-        try:
-            cmd_sysctl_service(name, "restart")
-        except CalledProcessError as e:
-            Logger.print_error(f"Error restarting {name}: {e}")
-            raise
 
-    @staticmethod
+        service_list = [
-    def start_all(instances: List[InstanceType]) -> None:
+            Path(SYSTEMD, service)
-        for instance in instances:
+            for service in SYSTEMD.iterdir()
-            InstanceManager.start(instance)
+            if pattern.search(service.name)
+            and not any(s in service.name for s in excluded)
+        ]
 
-    @staticmethod
+        instance_list = [
-    def stop_all(instances: List[InstanceType]) -> None:
+            self.instance_type(suffix=self._get_instance_suffix(service))
-        for instance in instances:
+            for service in service_list
-            InstanceManager.stop(instance)
+        ]
 
-    @staticmethod
+        return sorted(instance_list, key=lambda x: self._sort_instance_list(x.suffix))
-    def restart_all(instances: List[InstanceType]) -> None:
-        for instance in instances:
-            InstanceManager.restart(instance)
 
-    @staticmethod
+    def _get_instance_suffix(self, file_path: Path) -> str:
-    def remove(instance: InstanceType) -> None:
+        return file_path.stem.split("-")[-1] if "-" in file_path.stem else ""
-        from utils.fs_utils import run_remove_routines
-        from utils.sys_utils import remove_system_service
 
-        try:
+    def _sort_instance_list(self, s: Union[int, str, None]):
-            # remove the service file
+        if s is None:
-            service_file_path: Path = instance.service_file_path
+            return
-            if service_file_path is not None:
-                remove_system_service(service_file_path.name)
 
-            # then remove all the log files
+        return int(s) if s.isdigit() else s
-            if (
-                not instance.log_file_name
-                or not instance.base.log_dir
-                or not instance.base.log_dir.exists()
-            ):
-                return
-
-            files = instance.base.log_dir.iterdir()
-            logs = [f for f in files if f.name.startswith(instance.log_file_name)]
-            for log in logs:
-                Logger.print_status(f"Remove '{log}'")
-                run_remove_routines(log)
-
-        except Exception as e:
-            Logger.print_error(f"Error removing service: {e}")
-            raise

kiauh/core/instance_manager/name_scheme.py

@@ -0,0 +1,8 @@
+from enum import Enum, unique
+
+
+@unique
+class NameScheme(Enum):
+    SINGLE = "SINGLE"
+    INDEX = "INDEX"
+    CUSTOM = "CUSTOM"

kiauh/core/logger.py

@@ -1,167 +0,0 @@
-# ======================================================================= #
-#  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 enum import Enum
-from typing import List
-
-from core.types.color import Color
-
-
-class DialogType(Enum):
-    INFO = ("INFO", Color.WHITE)
-    SUCCESS = ("SUCCESS", Color.GREEN)
-    ATTENTION = ("ATTENTION", Color.YELLOW)
-    WARNING = ("WARNING", Color.YELLOW)
-    ERROR = ("ERROR", Color.RED)
-    CUSTOM = (None, None)
-
-
-LINE_WIDTH = 53
-
-
-BORDER_TOP: str = "┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓"
-BORDER_BOTTOM: str = "┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛"
-BORDER_TITLE: str = "┠───────────────────────────────────────────────────────┨"
-BORDER_LEFT: str = "┃"
-BORDER_RIGHT: str = "┃"
-
-
-class Logger:
-    @staticmethod
-    def print_info(msg, prefix=True, start="", end="\n") -> None:
-        message = f"[INFO] {msg}" if prefix else msg
-        Logger.__print(Color.WHITE, start, message, end)
-
-    @staticmethod
-    def print_ok(msg: str = "Success!", prefix=True, start="", end="\n") -> None:
-        message = f"[OK] {msg}" if prefix else msg
-        Logger.__print(Color.GREEN, start, message, end)
-
-    @staticmethod
-    def print_warn(msg, prefix=True, start="", end="\n") -> None:
-        message = f"[WARN] {msg}" if prefix else msg
-        Logger.__print(Color.YELLOW, start, message, end)
-
-    @staticmethod
-    def print_error(msg, prefix=True, start="", end="\n") -> None:
-        message = f"[ERROR] {msg}" if prefix else msg
-        Logger.__print(Color.RED, start, message, end)
-
-    @staticmethod
-    def print_status(msg, prefix=True, start="", end="\n") -> None:
-        message = f"\n###### {msg}" if prefix else msg
-        Logger.__print(Color.MAGENTA, start, message, end)
-
-    @staticmethod
-    def __print(color: Color, start: str, message: str, end: str) -> None:
-        print(Color.apply(f"{start}{message}", color), end=end)
-
-    @staticmethod
-    def print_dialog(
-        title: DialogType,
-        content: List[str],
-        center_content: bool = False,
-        custom_title: str | None = None,
-        custom_color: Color | None = None,
-        margin_top: int = 0,
-        margin_bottom: int = 0,
-    ) -> None:
-        """
-        Prints a dialog with the given title and content.
-        Those dialogs should be used to display verbose messages to the user which
-        require simple interaction like confirmation or input. Do not use this for
-        navigating through the application.
-
-        :param title: The type of the dialog.
-        :param content: The content of the dialog.
-        :param center_content: Whether to center the content or not.
-        :param custom_title: A custom title for the dialog.
-        :param custom_color: A custom color for the dialog.
-        :param margin_top: The number of empty lines to print before the dialog.
-        :param margin_bottom: The number of empty lines to print after the dialog.
-        """
-        color = Logger._get_dialog_color(title, custom_color)
-        dialog_title = Logger._get_dialog_title(title, custom_title)
-
-        print("\n" * margin_top)
-
-        print(Color.apply(BORDER_TOP, color))
-
-        if dialog_title:
-            print(Color.apply(f"┃ {dialog_title:^{LINE_WIDTH}} ┃", color))
-            print(Color.apply(BORDER_TITLE, color))
-
-        if content:
-            print(
-                Logger.format_content(
-                    content,
-                    LINE_WIDTH,
-                    color,
-                    center_content,
-                )
-            )
-
-        print(Color.apply(BORDER_BOTTOM, color))
-
-        print("\n" * margin_bottom)
-
-    @staticmethod
-    def _get_dialog_title(
-        title: DialogType, custom_title: str | None = None
-    ) -> str | None:
-        if title == DialogType.CUSTOM and custom_title:
-            return f"[ {custom_title} ]"
-        return f"[ {title.value[0]} ]" if title.value[0] else None
-
-    @staticmethod
-    def _get_dialog_color(
-        title: DialogType, custom_color: Color | None = None
-    ) -> Color:
-        if title == DialogType.CUSTOM and custom_color:
-            return custom_color
-
-        color: Color = title.value[1] if title.value[1] else Color.WHITE
-
-        return color
-
-    @staticmethod
-    def format_content(
-        content: List[str],
-        line_width: int,
-        color: Color = Color.WHITE,
-        center_content: bool = False,
-        border_left: str = "┃",
-        border_right: str = "┃",
-    ) -> str:
-        wrapper = textwrap.TextWrapper(line_width)
-
-        lines = []
-        for i, c in enumerate(content):
-            paragraph = wrapper.wrap(c)
-            lines.extend(paragraph)
-
-            # add a full blank line if we have a double newline
-            # character unless we are at the end of the list
-            if c == "\n\n" and i < len(content) - 1:
-                lines.append(" " * line_width)
-
-        if not center_content:
-            formatted_lines = [
-                Color.apply(f"{border_left} {line:<{line_width}} {border_right}", color)
-                for line in lines
-            ]
-        else:
-            formatted_lines = [
-                Color.apply(f"{border_left} {line:^{line_width}} {border_right}", color)
-                for line in lines
-            ]
-
-        return "\n".join(formatted_lines)

kiauh/core/menus/__init__.py

@@ -1,16 +1,15 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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 dataclasses import dataclass
 from enum import Enum
-from typing import Any, Callable, Type
+from typing import Any, Callable, Union
 
 
 @dataclass
@@ -18,14 +17,13 @@ class Option:
     """
     Represents a menu option.
     :param method: Method that will be used to call the menu option
+    :param menu: Flag for singaling that another menu will be opened
     :param opt_index: Can be used to pass the user input to the menu option
     :param opt_data: Can be used to pass any additional data to the menu option
     """
 
-    def __repr__(self):
+    method: Union[Callable, None] = None
-        return f"Option(method={self.method.__name__}, opt_index={self.opt_index}, opt_data={self.opt_data})"
+    menu: bool = False
-
-    method: Type[Callable]
     opt_index: str = ""
     opt_data: Any = None
 

kiauh/core/menus/advanced_menu.py

@@ -1,22 +1,19 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
-from typing import Type
+from typing import Optional, Type
 
 from components.klipper import KLIPPER_DIR
 from components.klipper.klipper import Klipper
-from components.klipper.klipper_utils import install_input_shaper_deps
 from components.klipper_firmware.menus.klipper_build_menu import (
     KlipperBuildFirmwareMenu,
-    KlipperKConfigMenu,
 )
 from components.klipper_firmware.menus.klipper_flash_menu import (
     KlipperFlashMethodMenu,
@@ -26,81 +23,74 @@ from components.moonraker import MOONRAKER_DIR
 from components.moonraker.moonraker import Moonraker
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import COLOR_YELLOW, RESET_FORMAT
-from procedures.system import change_system_hostname
 from utils.git_utils import rollback_repository
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class AdvancedMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None) -> None:
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "Advanced Menu"
+        self.previous_menu = previous_menu
-        self.title_color = Color.YELLOW
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.main_menu import MainMenu
 
-        self.previous_menu = previous_menu if previous_menu is not None else MainMenu
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else MainMenu
+        )
 
-    def set_options(self) -> None:
+    def set_options(self):
         self.options = {
-            "1": Option(method=self.build),
+            "1": Option(method=self.klipper_rollback, menu=True),
-            "2": Option(method=self.flash),
+            "2": Option(method=self.moonraker_rollback, menu=True),
-            "3": Option(method=self.build_flash),
+            "3": Option(method=self.build, menu=True),
-            "4": Option(method=self.get_id),
+            "4": Option(method=self.flash, menu=False),
-            "5": Option(method=self.input_shaper),
+            "5": Option(method=self.build_flash, menu=False),
-            "6": Option(method=self.klipper_rollback),
+            "6": Option(method=self.get_id, menu=False),
-            "7": Option(method=self.moonraker_rollback),
-            "8": Option(method=self.change_hostname),
         }
 
-    def print_menu(self) -> None:
+    def print_menu(self):
+        header = " [ Advanced Menu ] "
+        color = COLOR_YELLOW
+        count = 62 - len(color) - len(RESET_FORMAT)
         menu = textwrap.dedent(
-            """
+            f"""
-            ╟───────────────────────────┬───────────────────────────╢
+            /=======================================================\\
-            ║ Klipper Firmware:         │ Repository Rollback:      ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║  1) [Build]               │  6) [Klipper]             ║
+            |-------------------------------------------------------|
-            ║  2) [Flash]               │  7) [Moonraker]           ║
+            | Repo Rollback:                                        |
-            ║  3) [Build + Flash]       │                           ║
+            |  1) [Klipper]                                         |
-            ║  4) [Get MCU ID]          │ System:                   ║
+            |  2) [Moonraker]                                       |
-            ║                           │  8) [Change hostname]     ║
+            |                                                       |
-            ║ Extra Dependencies:       │                           ║
+            | Klipper Firmware:                                     |
-            ║  5) [Input Shaper]        │                           ║
+            |  3) [Build]                                           |
-            ╟───────────────────────────┴───────────────────────────╢
+            |  4) [Flash]                                           |
+            |  5) [Build + Flash]                                   |
+            |  6) [Get MCU ID]                                      |   
             """
         )[1:]
         print(menu, end="")
 
-    def klipper_rollback(self, **kwargs) -> None:
+    def klipper_rollback(self, **kwargs):
         rollback_repository(KLIPPER_DIR, Klipper)
 
-    def moonraker_rollback(self, **kwargs) -> None:
+    def moonraker_rollback(self, **kwargs):
         rollback_repository(MOONRAKER_DIR, Moonraker)
 
-    def build(self, **kwargs) -> None:
+    def build(self, **kwargs):
-        KlipperKConfigMenu().run()
         KlipperBuildFirmwareMenu(previous_menu=self.__class__).run()
 
-    def flash(self, **kwargs) -> None:
+    def flash(self, **kwargs):
-        KlipperKConfigMenu().run()
         KlipperFlashMethodMenu(previous_menu=self.__class__).run()
 
-    def build_flash(self, **kwargs) -> None:
+    def build_flash(self, **kwargs):
-        KlipperKConfigMenu().run()
         KlipperBuildFirmwareMenu(previous_menu=KlipperFlashMethodMenu).run()
         KlipperFlashMethodMenu(previous_menu=self.__class__).run()
 
-    def get_id(self, **kwargs) -> None:
+    def get_id(self, **kwargs):
         KlipperSelectMcuConnectionMenu(
             previous_menu=self.__class__,
             standalone=True,
         ).run()
-
-    def change_hostname(self, **kwargs) -> None:
-        change_system_hostname()
-
-    def input_shaper(self, **kwargs) -> None:
-        install_input_shaper_deps()

kiauh/core/menus/backup_menu.py

@@ -1,19 +1,18 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
-from typing import Type
+from typing import Optional, Type
 
 from components.klipper.klipper_utils import backup_klipper_dir
 from components.klipperscreen.klipperscreen import backup_klipperscreen_dir
-from components.moonraker.utils.utils import (
+from components.moonraker.moonraker_utils import (
     backup_moonraker_db_dir,
     backup_moonraker_dir,
 )
@@ -25,83 +24,85 @@ from components.webui_client.fluidd_data import FluiddData
 from components.webui_client.mainsail_data import MainsailData
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
-from core.types.color import Color
 from utils.common import backup_printer_config_dir
+from utils.constants import COLOR_CYAN, COLOR_YELLOW, RESET_FORMAT
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class BackupMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None) -> None:
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "Backup Menu"
+        self.previous_menu = previous_menu
-        self.title_color = Color.GREEN
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.main_menu import MainMenu
 
-        self.previous_menu = previous_menu if previous_menu is not None else MainMenu
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else MainMenu
+        )
 
     def set_options(self) -> None:
         self.options = {
-            "1": Option(method=self.backup_klipper),
+            "1": Option(method=self.backup_klipper, menu=False),
-            "2": Option(method=self.backup_moonraker),
+            "2": Option(method=self.backup_moonraker, menu=False),
-            "3": Option(method=self.backup_printer_config),
+            "3": Option(method=self.backup_printer_config, menu=False),
-            "4": Option(method=self.backup_moonraker_db),
+            "4": Option(method=self.backup_moonraker_db, menu=False),
-            "5": Option(method=self.backup_mainsail),
+            "5": Option(method=self.backup_mainsail, menu=False),
-            "6": Option(method=self.backup_fluidd),
+            "6": Option(method=self.backup_fluidd, menu=False),
-            "7": Option(method=self.backup_mainsail_config),
+            "7": Option(method=self.backup_mainsail_config, menu=False),
-            "8": Option(method=self.backup_fluidd_config),
+            "8": Option(method=self.backup_fluidd_config, menu=False),
-            "9": Option(method=self.backup_klipperscreen),
+            "9": Option(method=self.backup_klipperscreen, menu=False),
         }
 
-    def print_menu(self) -> None:
+    def print_menu(self):
-        line1 = Color.apply(
+        header = " [ Backup Menu ] "
-            "INFO: Backups are located in '~/kiauh-backups'", Color.YELLOW
+        line1 = f"{COLOR_YELLOW}INFO: Backups are located in '~/kiauh-backups'{RESET_FORMAT}"
-        )
+        color = COLOR_CYAN
+        count = 62 - len(color) - len(RESET_FORMAT)
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ {line1:^62} ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ╟───────────────────────────┬───────────────────────────╢
+            |-------------------------------------------------------|
-            ║ Klipper & Moonraker API:  │ Client-Config:            ║
+            | {line1:^62} |
-            ║  1) [Klipper]             │  7) [Mainsail-Config]     ║
+            |-------------------------------------------------------|
-            ║  2) [Moonraker]           │  8) [Fluidd-Config]       ║
+            | Klipper & Moonraker API:  | Client-Config:            |
-            ║  3) [Config Folder]       │                           ║
+            |  1) [Klipper]             |  7) [Mainsail-Config]     |
-            ║  4) [Moonraker Database]  │ Touchscreen GUI:          ║
+            |  2) [Moonraker]           |  8) [Fluidd-Config]       |
-            ║                           │  9) [KlipperScreen]       ║
+            |  3) [Config Folder]       |                           |
-            ║ Webinterface:             │                           ║
+            |  4) [Moonraker Database]  | Touchscreen GUI:          |
-            ║  5) [Mainsail]            │                           ║
+            |                           |  9) [KlipperScreen]       |
-            ║  6) [Fluidd]              │                           ║
+            | Webinterface:             |                           |
-            ╟───────────────────────────┴───────────────────────────╢
+            |  5) [Mainsail]            |                           |
+            |  6) [Fluidd]              |                           |
             """
         )[1:]
         print(menu, end="")
 
-    def backup_klipper(self, **kwargs) -> None:
+    def backup_klipper(self, **kwargs):
         backup_klipper_dir()
 
-    def backup_moonraker(self, **kwargs) -> None:
+    def backup_moonraker(self, **kwargs):
         backup_moonraker_dir()
 
-    def backup_printer_config(self, **kwargs) -> None:
+    def backup_printer_config(self, **kwargs):
         backup_printer_config_dir()
 
-    def backup_moonraker_db(self, **kwargs) -> None:
+    def backup_moonraker_db(self, **kwargs):
         backup_moonraker_db_dir()
 
-    def backup_mainsail(self, **kwargs) -> None:
+    def backup_mainsail(self, **kwargs):
         backup_client_data(MainsailData())
 
-    def backup_fluidd(self, **kwargs) -> None:
+    def backup_fluidd(self, **kwargs):
         backup_client_data(FluiddData())
 
-    def backup_mainsail_config(self, **kwargs) -> None:
+    def backup_mainsail_config(self, **kwargs):
         backup_client_config_data(MainsailData())
 
-    def backup_fluidd_config(self, **kwargs) -> None:
+    def backup_fluidd_config(self, **kwargs):
         backup_client_config_data(FluiddData())
 
-    def backup_klipperscreen(self, **kwargs) -> None:
+    def backup_klipperscreen(self, **kwargs):
         backup_klipperscreen_dir()

kiauh/core/menus/base_menu.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -14,87 +14,87 @@ import sys
 import textwrap
 import traceback
 from abc import abstractmethod
-from enum import Enum
+from typing import Dict, Optional, Type
-from typing import Dict, Type
 
-from core.logger import Logger
 from core.menus import FooterType, Option
-from core.services.message_service import MessageService
+from utils.constants import (
-from core.spinner import Spinner
+    COLOR_CYAN,
-from core.types.color import Color
+    COLOR_GREEN,
-from utils.input_utils import get_selection_input
+    COLOR_RED,
+    COLOR_YELLOW,
+    RESET_FORMAT,
+)
+from utils.logger import Logger
 
 
-def clear() -> None:
+def clear():
-    subprocess.call("clear -x", shell=True)
+    subprocess.call("clear", shell=True)
 
 
-def print_header() -> None:
+def print_header():
     line1 = " [ KIAUH ] "
     line2 = "Klipper Installation And Update Helper"
     line3 = ""
-    color = Color.CYAN
+    color = COLOR_CYAN
-    count = 62 - len(str(color)) - len(str(Color.RST))
+    count = 62 - len(color) - len(RESET_FORMAT)
     header = textwrap.dedent(
         f"""
-        ╔═══════════════════════════════════════════════════════╗
+        /=======================================================\\
-        ║ {Color.apply(f"{line1:~^{count}}", color)} ║
+        | {color}{line1:~^{count}}{RESET_FORMAT} |
-        ║ {Color.apply(f"{line2:^{count}}", color)} ║
+        | {color}{line2:^{count}}{RESET_FORMAT} |
-        ║ {Color.apply(f"{line3:~^{count}}", color)} ║
+        | {color}{line3:~^{count}}{RESET_FORMAT} |
-        ╚═══════════════════════════════════════════════════════╝
+        \=======================================================/
         """
     )[1:]
     print(header, end="")
 
 
-def print_quit_footer() -> None:
+def print_quit_footer():
     text = "Q) Quit"
-    color = Color.RED
+    color = COLOR_RED
-    count = 62 - len(str(color)) - len(str(Color.RST))
+    count = 62 - len(color) - len(RESET_FORMAT)
     footer = textwrap.dedent(
         f"""
-        ║ {color}{text:^{count}}{Color.RST} ║
+        |-------------------------------------------------------|
-        ╚═══════════════════════════════════════════════════════╝
+        | {color}{text:^{count}}{RESET_FORMAT} |
+        \=======================================================/
         """
     )[1:]
     print(footer, end="")
 
 
-def print_back_footer() -> None:
+def print_back_footer():
     text = "B) « Back"
-    color = Color.GREEN
+    color = COLOR_GREEN
-    count = 62 - len(str(color)) - len(str(Color.RST))
+    count = 62 - len(color) - len(RESET_FORMAT)
     footer = textwrap.dedent(
         f"""
-        ║ {color}{text:^{count}}{Color.RST} ║
+        |-------------------------------------------------------|
-        ╚═══════════════════════════════════════════════════════╝
+        | {color}{text:^{count}}{RESET_FORMAT} |
+        \=======================================================/
         """
     )[1:]
     print(footer, end="")
 
 
-def print_back_help_footer() -> None:
+def print_back_help_footer():
     text1 = "B) « Back"
     text2 = "H) Help [?]"
-    color1 = Color.GREEN
+    color1 = COLOR_GREEN
-    color2 = Color.YELLOW
+    color2 = COLOR_YELLOW
-    count = 34 - len(str(color1)) - len(str(Color.RST))
+    count = 34 - len(color1) - len(RESET_FORMAT)
     footer = textwrap.dedent(
         f"""
-        ║ {color1}{text1:^{count}}{Color.RST} │ {color2}{text2:^{count}}{Color.RST} ║
+        |-------------------------------------------------------|
-        ╚═══════════════════════════╧═══════════════════════════╝
+        | {color1}{text1:^{count}}{RESET_FORMAT} | {color2}{text2:^{count}}{RESET_FORMAT} |
+        \=======================================================/
         """
     )[1:]
     print(footer, end="")
 
 
-def print_blank_footer() -> None:
+def print_blank_footer():
-    print("╚═══════════════════════════════════════════════════════╝")
+    print("\=======================================================/")
-
-
-class MenuTitleStyle(Enum):
-    PLAIN = "plain"
-    STYLED = "styled"
 
 
 class PostInitCaller(type):
@@ -112,56 +112,42 @@ class BaseMenu(metaclass=PostInitCaller):
     default_option: Option = None
     input_label_txt: str = "Perform action"
     header: bool = False
-
+    previous_menu: Type[BaseMenu] = None
-    loading_msg: str = ""
+    help_menu: Type[BaseMenu] = None
-    spinner: Spinner | None = None
-
-    title: str = ""
-    title_style: MenuTitleStyle = MenuTitleStyle.STYLED
-    title_color: Color = Color.WHITE
-
-    previous_menu: Type[BaseMenu] | None = None
-    help_menu: Type[BaseMenu] | None = None
     footer_type: FooterType = FooterType.BACK
 
-    message_service = MessageService()
+    def __init__(self, **kwargs):
-
-    def __init__(self, **kwargs) -> None:
         if type(self) is BaseMenu:
             raise NotImplementedError("BaseMenu cannot be instantiated directly.")
 
-    def __post_init__(self) -> None:
+    def __post_init__(self):
         self.set_previous_menu(self.previous_menu)
         self.set_options()
 
         # conditionally add options based on footer type
         if self.footer_type is FooterType.QUIT:
-            self.options["q"] = Option(method=self.__exit)
+            self.options["q"] = Option(method=self.__exit, menu=False)
         if self.footer_type is FooterType.BACK:
-            self.options["b"] = Option(method=self.__go_back)
+            self.options["b"] = Option(method=self.__go_back, menu=False)
         if self.footer_type is FooterType.BACK_HELP:
-            self.options["b"] = Option(method=self.__go_back)
+            self.options["b"] = Option(method=self.__go_back, menu=False)
-            self.options["h"] = Option(method=self.__go_to_help)
+            self.options["h"] = Option(method=self.__go_to_help, menu=False)
         # if defined, add the default option to the options dict
         if self.default_option is not None:
             self.options[""] = self.default_option
 
-    def __go_back(self, **kwargs) -> None:
+    def __go_back(self, **kwargs):
-        if self.previous_menu is None:
-            return
         self.previous_menu().run()
 
-    def __go_to_help(self, **kwargs) -> None:
+    def __go_to_help(self, **kwargs):
-        if self.help_menu is None:
+        self.help_menu(previous_menu=self).run()
-            return
-        self.help_menu(previous_menu=self.__class__).run()
 
-    def __exit(self, **kwargs) -> None:
+    def __exit(self, **kwargs):
         Logger.print_ok("###### Happy printing!", False)
         sys.exit(0)
 
     @abstractmethod
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         raise NotImplementedError
 
     @abstractmethod
@@ -172,32 +158,7 @@ class BaseMenu(metaclass=PostInitCaller):
     def print_menu(self) -> None:
         raise NotImplementedError
 
-    def is_loading(self, state: bool) -> None:
+    def print_footer(self) -> None:
-        if not self.spinner and state:
-            self.spinner = Spinner(self.loading_msg)
-            self.spinner.start()
-        else:
-            self.spinner.stop()
-            self.spinner = None
-
-    def __print_menu_title(self) -> None:
-        count = 62 - len(str(self.title_color)) - len(str(Color.RST))
-        menu_title = "╔═══════════════════════════════════════════════════════╗\n"
-        if self.title:
-            title = (
-                f" [ {self.title} ] "
-                if self.title_style == MenuTitleStyle.STYLED
-                else self.title
-            )
-            line = (
-                f"{title:~^{count}}"
-                if self.title_style == MenuTitleStyle.STYLED
-                else f"{title:^{count}}"
-            )
-            menu_title += f"║ {Color.apply(line, self.title_color)} ║\n"
-        print(menu_title, end="")
-
-    def __print_footer(self) -> None:
         if self.footer_type is FooterType.QUIT:
             print_quit_footer()
         elif self.footer_type is FooterType.BACK:
@@ -207,32 +168,51 @@ class BaseMenu(metaclass=PostInitCaller):
         elif self.footer_type is FooterType.BLANK:
             print_blank_footer()
         else:
-            raise NotImplementedError("FooterType not correctly implemented!")
+            raise NotImplementedError
-
-    def __display_menu(self) -> None:
-        self.message_service.display_message()
 
+    def display_menu(self) -> None:
         if self.header:
             print_header()
-
-        self.__print_menu_title()
         self.print_menu()
-        self.__print_footer()
+        self.print_footer()
+
+    def validate_user_input(self, usr_input: str) -> Option:
+        """
+        Validate the user input and either return an Option, a string or None
+        :param usr_input: The user input in form of a string
+        :return: Option, str or None
+        """
+        usr_input = usr_input.lower()
+        option = self.options.get(usr_input, Option(None, False, "", None))
+
+        # if option/usr_input is None/empty string, we execute the menus default option if specified
+        if (option is None or usr_input == "") and self.default_option is not None:
+            self.default_option.opt_index = usr_input
+            return self.default_option
+
+        # user selected a regular option
+        option.opt_index = usr_input
+        return option
+
+    def handle_user_input(self) -> Option:
+        """Handle the user input, return the validated input or print an error."""
+        while True:
+            print(f"{COLOR_CYAN}###### {self.input_label_txt}: {RESET_FORMAT}", end="")
+            usr_input = input().lower()
+            validated_input = self.validate_user_input(usr_input)
+
+            if validated_input.method is not None:
+                return validated_input
+            else:
+                Logger.print_error("Invalid input!", False)
 
     def run(self) -> None:
         """Start the menu lifecycle. When this function returns, the lifecycle of the menu ends."""
         try:
-            self.__display_menu()
+            self.display_menu()
-            option = get_selection_input(self.input_label_txt, self.options)
+            option = self.handle_user_input()
-            selected_option: Option = self.options.get(option)
+            option.method(opt_index=option.opt_index, opt_data=option.opt_data)
-
-            selected_option.method(
-                opt_index=selected_option.opt_index,
-                opt_data=selected_option.opt_data,
-            )
-
             self.run()
-
         except Exception as e:
             Logger.print_error(
                 f"An unexpected error occured:\n{e}\n{traceback.format_exc()}"

kiauh/core/menus/install_menu.py

@@ -1,108 +1,104 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
-from typing import Type
+from typing import Optional, Type
 
 from components.crowsnest.crowsnest import install_crowsnest
-from components.klipper.services.klipper_setup_service import KlipperSetupService
+from components.klipper import klipper_setup
 from components.klipperscreen.klipperscreen import install_klipperscreen
+from components.mobileraker.mobileraker import install_mobileraker
 from components.moonraker import moonraker_setup
-from components.webui_client.client_config.client_config_setup import (
+from components.webui_client import client_setup
-    install_client_config,
+from components.webui_client.client_config import client_config_setup
-)
-from components.webui_client.client_setup import install_client
 from components.webui_client.fluidd_data import FluiddData
 from components.webui_client.mainsail_data import MainsailData
-from components.webui_client.menus.client_install_menu import ClientInstallMenu
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
-from core.settings.kiauh_settings import KiauhSettings
+from utils.constants import COLOR_GREEN, RESET_FORMAT
-from core.types.color import Color
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class InstallMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None) -> None:
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "Installation Menu"
+        self.previous_menu = previous_menu
-        self.title_color = Color.GREEN
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
-        self.klsvc = KlipperSetupService()
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.main_menu import MainMenu
 
-        self.previous_menu = previous_menu if previous_menu is not None else MainMenu
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else MainMenu
+        )
 
     def set_options(self) -> None:
         self.options = {
-            "1": Option(method=self.install_klipper),
+            "1": Option(method=self.install_klipper, menu=False),
-            "2": Option(method=self.install_moonraker),
+            "2": Option(method=self.install_moonraker, menu=False),
-            "3": Option(method=self.install_mainsail),
+            "3": Option(method=self.install_mainsail, menu=False),
-            "4": Option(method=self.install_fluidd),
+            "4": Option(method=self.install_fluidd, menu=False),
-            "5": Option(method=self.install_mainsail_config),
+            "5": Option(method=self.install_mainsail_config, menu=False),
-            "6": Option(method=self.install_fluidd_config),
+            "6": Option(method=self.install_fluidd_config, menu=False),
-            "7": Option(method=self.install_klipperscreen),
+            "7": Option(method=self.install_klipperscreen, menu=False),
-            "8": Option(method=self.install_crowsnest),
+            "8": Option(method=self.install_mobileraker, menu=False),
+            "9": Option(method=self.install_crowsnest, menu=False),
         }
 
-    def print_menu(self) -> None:
+    def print_menu(self):
+        header = " [ Installation Menu ] "
+        color = COLOR_GREEN
+        count = 62 - len(color) - len(RESET_FORMAT)
         menu = textwrap.dedent(
-            """
+            f"""
-            ╟───────────────────────────┬───────────────────────────╢
+            /=======================================================\\
-            ║ Firmware & API:           │ Touchscreen GUI:          ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║  1) [Klipper]             │  7) [KlipperScreen]       ║
+            |-------------------------------------------------------|
-            ║  2) [Moonraker]           │                           ║
+            | Firmware & API:           | Touchscreen GUI:          |
-            ║                           │ Webcam Streamer:          ║
+            |  1) [Klipper]             |  7) [KlipperScreen]       |
-            ║ Webinterface:             │  8) [Crowsnest]           ║
+            |  2) [Moonraker]           |                           |
-            ║  3) [Mainsail]            │                           ║
+            |                           | Android / iOS:            |
-            ║  4) [Fluidd]              │                           ║
+            | Webinterface:             |  8) [Mobileraker]         |
-            ║                           │                           ║
+            |  3) [Mainsail]            |                           |
-            ║ Client-Config:            │                           ║
+            |  4) [Fluidd]              | Webcam Streamer:          |
-            ║  5) [Mainsail-Config]     │                           ║
+            |                           |  9) [Crowsnest]           |
-            ║  6) [Fluidd-Config]       │                           ║
+            | Client-Config:            |                           |
-            ╟───────────────────────────┴───────────────────────────╢
+            |  5) [Mainsail-Config]     |                           |
+            |  6) [Fluidd-Config]       |                           |
+            |                           |                           |
             """
         )[1:]
         print(menu, end="")
 
-    def install_klipper(self, **kwargs) -> None:
+    def install_klipper(self, **kwargs):
-        self.klsvc.install()
+        klipper_setup.install_klipper()
 
-    def install_moonraker(self, **kwargs) -> None:
+    def install_moonraker(self, **kwargs):
         moonraker_setup.install_moonraker()
 
-    def install_mainsail(self, **kwargs) -> None:
+    def install_mainsail(self, **kwargs):
-        client: MainsailData = MainsailData()
+        client_setup.install_client(MainsailData())
-        if client.client_dir.exists():
-            ClientInstallMenu(client, self.__class__).run()
-        else:
-            install_client(client, settings=KiauhSettings())
 
-    def install_mainsail_config(self, **kwargs) -> None:
+    def install_mainsail_config(self, **kwargs):
-        install_client_config(MainsailData())
+        client_config_setup.install_client_config(MainsailData())
 
-    def install_fluidd(self, **kwargs) -> None:
+    def install_fluidd(self, **kwargs):
-        client: FluiddData = FluiddData()
+        client_setup.install_client(FluiddData())
-        if client.client_dir.exists():
-            ClientInstallMenu(client, self.__class__).run()
-        else:
-            install_client(client, settings=KiauhSettings())
 
-    def install_fluidd_config(self, **kwargs) -> None:
+    def install_fluidd_config(self, **kwargs):
-        install_client_config(FluiddData())
+        client_config_setup.install_client_config(FluiddData())
 
-    def install_klipperscreen(self, **kwargs) -> None:
+    def install_klipperscreen(self, **kwargs):
         install_klipperscreen()
 
-    def install_crowsnest(self, **kwargs) -> None:
+    def install_mobileraker(self, **kwargs):
+        install_mobileraker()
+
+    def install_crowsnest(self, **kwargs):
         install_crowsnest()

kiauh/core/menus/main_menu.py

@@ -1,29 +1,27 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
-import sys
 import textwrap
-from typing import Callable, Type
+from typing import Optional, Type
 
 from components.crowsnest.crowsnest import get_crowsnest_status
 from components.klipper.klipper_utils import get_klipper_status
 from components.klipperscreen.klipperscreen import get_klipperscreen_status
 from components.log_uploads.menus.log_upload_menu import LogUploadMenu
-from components.moonraker.utils.utils import get_moonraker_status
+from components.mobileraker.mobileraker import get_mobileraker_status
+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.logger import Logger
 from core.menus import FooterType
 from core.menus.advanced_menu import AdvancedMenu
 from core.menus.backup_menu import BackupMenu
@@ -32,148 +30,143 @@ 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 core.types.color import Color
-from core.types.component_status import ComponentStatus, StatusMap, StatusText
 from extensions.extensions_menu import ExtensionsMenu
-from utils.common import get_kiauh_version, trunc_string
+from utils.constants import (
+    COLOR_CYAN,
+    COLOR_GREEN,
+    COLOR_MAGENTA,
+    COLOR_RED,
+    COLOR_YELLOW,
+    RESET_FORMAT,
+)
+from utils.types import ComponentStatus
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class MainMenu(BaseMenu):
-    def __init__(self) -> None:
+    def __init__(self):
         super().__init__()
 
-        self.header: bool = True
+        self.header = True
-        self.title = "Main Menu"
+        self.footer_type = FooterType.QUIT
-        self.title_color = Color.CYAN
-        self.footer_type: FooterType = FooterType.QUIT
 
-        self.version = ""
+        self.kl_status = self.kl_repo = self.mr_status = self.mr_repo = ""
-        self.kl_status, self.kl_owner, self.kl_repo = "", "", ""
+        self.ms_status = self.fl_status = self.ks_status = self.mb_status = ""
-        self.mr_status, self.mr_owner, self.mr_repo = "", "", ""
+        self.cn_status = self.cc_status = ""
-        self.ms_status, self.fl_status, self.ks_status = "", "", ""
+        self.init_status()
-        self.cn_status, self.cc_status = "", ""
-        self._init_status()
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    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),
+            "0": Option(method=self.log_upload_menu, menu=True),
-            "1": Option(method=self.install_menu),
+            "1": Option(method=self.install_menu, menu=True),
-            "2": Option(method=self.update_menu),
+            "2": Option(method=self.update_menu, menu=True),
-            "3": Option(method=self.remove_menu),
+            "3": Option(method=self.remove_menu, menu=True),
-            "4": Option(method=self.advanced_menu),
+            "4": Option(method=self.advanced_menu, menu=True),
-            "5": Option(method=self.backup_menu),
+            "5": Option(method=self.backup_menu, menu=True),
-            "e": Option(method=self.extension_menu),
+            "e": Option(method=self.extension_menu, menu=True),
-            "s": Option(method=self.settings_menu),
+            "s": Option(method=self.settings_menu, menu=True),
         }
 
-    def _init_status(self) -> None:
+    def init_status(self) -> None:
-        status_vars = ["kl", "mr", "ms", "fl", "ks", "cn"]
+        status_vars = ["kl", "mr", "ms", "fl", "ks", "mb", "cn"]
         for var in status_vars:
             setattr(
                 self,
                 f"{var}_status",
-                Color.apply("Not installed", Color.RED),
+                f"{COLOR_RED}Not installed!{RESET_FORMAT}",
             )
 
-    def _fetch_status(self) -> None:
+    def fetch_status(self) -> None:
-        self.version = get_kiauh_version()
         self._get_component_status("kl", get_klipper_status)
         self._get_component_status("mr", get_moonraker_status)
         self._get_component_status("ms", get_client_status, MainsailData())
         self._get_component_status("fl", get_client_status, FluiddData())
+        self.cc_status = get_current_client_config([MainsailData(), FluiddData()])
         self._get_component_status("ks", get_klipperscreen_status)
+        self._get_component_status("mb", get_mobileraker_status)
         self._get_component_status("cn", get_crowsnest_status)
-        self.cc_status = get_current_client_config()
 
-    def _get_component_status(self, name: str, status_fn: Callable, *args) -> None:
+    def _get_component_status(self, name: str, status_fn: callable, *args) -> None:
         status_data: ComponentStatus = status_fn(*args)
-        code: int = status_data.status
+        code: int = status_data.get("status").value.code
-        status: StatusText = StatusMap[code]
+        status: str = status_data.get("status").value.txt
-        owner: str = trunc_string(status_data.owner, 23)
+        repo: str = status_data.get("repo")
-        repo: str = trunc_string(status_data.repo, 23)
+        instance_count: int = status_data.get("instances")
-        instance_count: int = status_data.instances
 
         count_txt: str = ""
-        if instance_count > 0 and code == 2:
+        if instance_count > 0 and code == 1:
             count_txt = f": {instance_count}"
 
         setattr(self, f"{name}_status", self._format_by_code(code, status, count_txt))
-        setattr(self, f"{name}_owner", Color.apply(owner, Color.CYAN))
+        setattr(self, f"{name}_repo", f"{COLOR_CYAN}{repo}{RESET_FORMAT}")
-        setattr(self, f"{name}_repo", Color.apply(repo, Color.CYAN))
 
     def _format_by_code(self, code: int, status: str, count: str) -> str:
-        color = Color.RED
+        if code == 1:
-        if code == 0:
+            return f"{COLOR_GREEN}{status}{count}{RESET_FORMAT}"
-            color = Color.RED
-        elif code == 1:
-            color = Color.YELLOW
         elif code == 2:
-            color = Color.GREEN
+            return f"{COLOR_RED}{status}{count}{RESET_FORMAT}"
 
-        return Color.apply(f"{status}{count}", color)
+        return f"{COLOR_YELLOW}{status}{count}{RESET_FORMAT}"
 
-    def print_menu(self) -> None:
+    def print_menu(self):
-        self._fetch_status()
+        self.fetch_status()
 
-        footer1 = Color.apply(self.version, Color.CYAN)
+        header = " [ Main Menu ] "
-        link = Color.apply("https://git.io/JnmlX", Color.MAGENTA)
+        footer1 = "KIAUH v6.0.0"
-        footer2 = f"Changelog: {link}"
+        footer2 = f"Changelog: {COLOR_MAGENTA}https://git.io/JnmlX{RESET_FORMAT}"
+        color = COLOR_CYAN
+        count = 62 - len(color) - len(RESET_FORMAT)
         pad1 = 32
         pad2 = 26
         menu = textwrap.dedent(
             f"""
-            ╟──────────────────┬────────────────────────────────────╢
+            /=======================================================\\
-            ║  0) [Log-Upload] │   Klipper: {self.kl_status:<{pad1}} ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║                  │     Owner: {self.kl_owner:<{pad1}} ║
+            |-------------------------------------------------------|
-            ║  1) [Install]    │      Repo: {self.kl_repo:<{pad1}} ║
+            |  0) [Log-Upload] |   Klipper: {self.kl_status:<{pad1}} |
-            ║  2) [Update]     ├────────────────────────────────────╢
+            |                  |      Repo: {self.kl_repo:<{pad1}} |
-            ║  3) [Remove]     │ Moonraker: {self.mr_status:<{pad1}} ║
+            |  1) [Install]    |------------------------------------|
-            ║  4) [Advanced]   │     Owner: {self.mr_owner:<{pad1}} ║
+            |  2) [Update]     | Moonraker: {self.mr_status:<{pad1}} |
-            ║  5) [Backup]     │      Repo: {self.mr_repo:<{pad1}} ║
+            |  3) [Remove]     |      Repo: {self.mr_repo:<{pad1}} |
-            ║                  ├────────────────────────────────────╢
+            |  4) [Advanced]   |------------------------------------|
-            ║  S) [Settings]   │        Mainsail: {self.ms_status:<{pad2}} ║
+            |  5) [Backup]     |        Mainsail: {self.ms_status:<{pad2}} |
-            ║                  │          Fluidd: {self.fl_status:<{pad2}} ║
+            |                  |          Fluidd: {self.fl_status:<{pad2}} |
-            ║ Community:       │   Client-Config: {self.cc_status:<{pad2}} ║
+            |  S) [Settings]   |   Client-Config: {self.cc_status:<{pad2}} |
-            ║  E) [Extensions] │                                    ║
+            |                  |                                    |
-            ║                  │   KlipperScreen: {self.ks_status:<{pad2}} ║
+            | Community:       |   KlipperScreen: {self.ks_status:<{pad2}} |
-            ║                  │       Crowsnest: {self.cn_status:<{pad2}} ║
+            |  E) [Extensions] |     Mobileraker: {self.mb_status:<{pad2}} |
-            ╟──────────────────┼────────────────────────────────────╢
+            |                  |       Crowsnest: {self.cn_status:<{pad2}} |
-            ║ {footer1:^25} │ {footer2:^43} ║
+            |-------------------------------------------------------|
-            ╟──────────────────┴────────────────────────────────────╢
+            | {COLOR_CYAN}{footer1:^16}{RESET_FORMAT} | {footer2:^43} |
             """
         )[1:]
         print(menu, end="")
 
-    def exit(self, **kwargs) -> None:
+    def log_upload_menu(self, **kwargs):
-        Logger.print_ok("###### Happy printing!", False)
-        sys.exit(0)
-
-    def log_upload_menu(self, **kwargs) -> None:
         LogUploadMenu().run()
 
-    def install_menu(self, **kwargs) -> None:
+    def install_menu(self, **kwargs):
         InstallMenu(previous_menu=self.__class__).run()
 
-    def update_menu(self, **kwargs) -> None:
+    def update_menu(self, **kwargs):
         UpdateMenu(previous_menu=self.__class__).run()
 
-    def remove_menu(self, **kwargs) -> None:
+    def remove_menu(self, **kwargs):
         RemoveMenu(previous_menu=self.__class__).run()
 
-    def advanced_menu(self, **kwargs) -> None:
+    def advanced_menu(self, **kwargs):
         AdvancedMenu(previous_menu=self.__class__).run()
 
-    def backup_menu(self, **kwargs) -> None:
+    def backup_menu(self, **kwargs):
         BackupMenu(previous_menu=self.__class__).run()
 
-    def settings_menu(self, **kwargs) -> None:
+    def settings_menu(self, **kwargs):
         SettingsMenu(previous_menu=self.__class__).run()
 
-    def extension_menu(self, **kwargs) -> None:
+    def extension_menu(self, **kwargs):
         ExtensionsMenu(previous_menu=self.__class__).run()

kiauh/core/menus/remove_menu.py

@@ -1,19 +1,19 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
-from typing import Type
+from typing import Optional, Type
 
 from components.crowsnest.crowsnest import remove_crowsnest
 from components.klipper.menus.klipper_remove_menu import KlipperRemoveMenu
 from components.klipperscreen.klipperscreen import remove_klipperscreen
+from components.mobileraker.mobileraker import remove_mobileraker
 from components.moonraker.menus.moonraker_remove_menu import (
     MoonrakerRemoveMenu,
 )
@@ -22,65 +22,74 @@ from components.webui_client.mainsail_data import MainsailData
 from components.webui_client.menus.client_remove_menu import ClientRemoveMenu
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import COLOR_RED, RESET_FORMAT
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class RemoveMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None) -> None:
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "Remove Menu"
+        self.previous_menu = previous_menu
-        self.title_color = Color.RED
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.main_menu import MainMenu
 
-        self.previous_menu = previous_menu if previous_menu is not None else MainMenu
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else MainMenu
+        )
 
-    def set_options(self) -> None:
+    def set_options(self):
         self.options = {
-            "1": Option(method=self.remove_klipper),
+            "1": Option(method=self.remove_klipper, menu=True),
-            "2": Option(method=self.remove_moonraker),
+            "2": Option(method=self.remove_moonraker, menu=True),
-            "3": Option(method=self.remove_mainsail),
+            "3": Option(method=self.remove_mainsail, menu=True),
-            "4": Option(method=self.remove_fluidd),
+            "4": Option(method=self.remove_fluidd, menu=True),
-            "5": Option(method=self.remove_klipperscreen),
+            "5": Option(method=self.remove_klipperscreen, menu=True),
-            "6": Option(method=self.remove_crowsnest),
+            "6": Option(method=self.remove_mobileraker, menu=True),
+            "7": Option(method=self.remove_crowsnest, menu=True),
         }
 
-    def print_menu(self) -> None:
+    def print_menu(self):
+        header = " [ Remove Menu ] "
+        color = COLOR_RED
+        count = 62 - len(color) - len(RESET_FORMAT)
         menu = textwrap.dedent(
-            """
+            f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ INFO: Configurations and/or any backups will be kept! ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ╟───────────────────────────┬───────────────────────────╢
+            |-------------------------------------------------------|
-            ║ Firmware & API:           │ Touchscreen GUI:          ║
+            | INFO: Configurations and/or any backups will be kept! |
-            ║  1) [Klipper]             │  5) [KlipperScreen]       ║
+            |-------------------------------------------------------|
-            ║  2) [Moonraker]           │                           ║
+            | Firmware & API:           | Touchscreen GUI:          |
-            ║                           │ Webcam Streamer:          ║
+            |  1) [Klipper]             |  5) [KlipperScreen]       |
-            ║ Klipper Webinterface:     │  6) [Crowsnest]           ║
+            |  2) [Moonraker]           |                           |
-            ║  3) [Mainsail]            │                           ║
+            |                           | Android / iOS:            |
-            ║  4) [Fluidd]              │                           ║
+            | Klipper Webinterface:     |  6) [Mobileraker]         |
-            ╟───────────────────────────┴───────────────────────────╢
+            |  3) [Mainsail]            |                           |
+            |  4) [Fluidd]              | Webcam Streamer:          |
+            |                           |  7) [Crowsnest]           |
             """
         )[1:]
         print(menu, end="")
 
-    def remove_klipper(self, **kwargs) -> None:
+    def remove_klipper(self, **kwargs):
         KlipperRemoveMenu(previous_menu=self.__class__).run()
 
-    def remove_moonraker(self, **kwargs) -> None:
+    def remove_moonraker(self, **kwargs):
         MoonrakerRemoveMenu(previous_menu=self.__class__).run()
 
-    def remove_mainsail(self, **kwargs) -> None:
+    def remove_mainsail(self, **kwargs):
         ClientRemoveMenu(previous_menu=self.__class__, client=MainsailData()).run()
 
-    def remove_fluidd(self, **kwargs) -> None:
+    def remove_fluidd(self, **kwargs):
         ClientRemoveMenu(previous_menu=self.__class__, client=FluiddData()).run()
 
-    def remove_klipperscreen(self, **kwargs) -> None:
+    def remove_klipperscreen(self, **kwargs):
         remove_klipperscreen()
 
-    def remove_crowsnest(self, **kwargs) -> None:
+    def remove_mobileraker(self, **kwargs):
+        remove_mobileraker()
+
+    def remove_crowsnest(self, **kwargs):
         remove_crowsnest()

kiauh/core/menus/settings_menu.py

@@ -1,162 +1,139 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
+import shutil
-
 import textwrap
 from pathlib import Path
-from typing import Literal, Tuple, Type
+from typing import Optional, Tuple, Type
 
-from components.klipper import KLIPPER_DIR, KLIPPER_REPO_URL
+from components.klipper import KLIPPER_DIR
-from components.klipper.klipper_utils import get_klipper_status
+from components.klipper.klipper import Klipper
-from components.moonraker import MOONRAKER_DIR, MOONRAKER_REPO_URL
+from components.moonraker import MOONRAKER_DIR
-from components.moonraker.utils.utils import get_moonraker_status
+from components.moonraker.moonraker import Moonraker
-from core.logger import DialogType, Logger
+from core.instance_manager.instance_manager import InstanceManager
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
-from core.settings.kiauh_settings import KiauhSettings, RepoSettings
+from core.settings.kiauh_settings import KiauhSettings
-from core.types.color import Color
+from utils.constants import COLOR_CYAN, COLOR_GREEN, RESET_FORMAT
-from procedures.switch_repo import run_switch_repo_routine
+from utils.git_utils import git_clone_wrapper
 from utils.input_utils import get_confirm, get_string_input
+from utils.logger import DialogType, Logger
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class SettingsMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None) -> None:
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.title = "Settings Menu"
+        self.previous_menu = previous_menu
-        self.title_color = Color.CYAN
+        self.klipper_repo = None
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
+        self.moonraker_repo = None
-
+        self.mainsail_unstable = None
-        self.mainsail_unstable: bool | None = None
+        self.fluidd_unstable = None
-        self.fluidd_unstable: bool | None = None
+        self.auto_backups_enabled = None
-        self.auto_backups_enabled: bool | None = None
         self._load_settings()
-        print(self.klipper_status)
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.main_menu import MainMenu
 
-        self.previous_menu = previous_menu if previous_menu is not None else MainMenu
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else MainMenu
+        )
 
     def set_options(self) -> None:
         self.options = {
-            "1": Option(method=self.set_klipper_repo),
+            "1": Option(method=self.set_klipper_repo, menu=True),
-            "2": Option(method=self.set_moonraker_repo),
+            "2": Option(method=self.set_moonraker_repo, menu=True),
-            "3": Option(method=self.toggle_mainsail_release),
+            "3": Option(method=self.toggle_mainsail_release, menu=True),
-            "4": Option(method=self.toggle_fluidd_release),
+            "4": Option(method=self.toggle_fluidd_release, menu=False),
-            "5": Option(method=self.toggle_backup_before_update),
+            "5": Option(method=self.toggle_backup_before_update, menu=False),
         }
 
-    def print_menu(self) -> None:
+    def print_menu(self):
-        color = Color.CYAN
+        header = " [ KIAUH Settings ] "
-        checked = f"[{Color.apply('x', Color.GREEN)}]"
+        color = COLOR_CYAN
+        count = 62 - len(color) - len(RESET_FORMAT)
+        checked = f"[{COLOR_GREEN}x{RESET_FORMAT}]"
         unchecked = "[ ]"
-
-        kl_repo: str = Color.apply(self.klipper_status.repo, color)
-        kl_branch: str = Color.apply(self.klipper_status.branch, color)
-        kl_owner: str = Color.apply(self.klipper_status.owner, color)
-        mr_repo: str = Color.apply(self.moonraker_status.repo, color)
-        mr_branch: str = Color.apply(self.moonraker_status.branch, color)
-        mr_owner: str = Color.apply(self.moonraker_status.owner, color)
         o1 = checked if self.mainsail_unstable else unchecked
         o2 = checked if self.fluidd_unstable else unchecked
         o3 = checked if self.auto_backups_enabled else unchecked
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────────────────────────────────────╢
+            /=======================================================\\
-            ║ Klipper:                                              ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║  ● Repo:   {kl_repo:51} ║
+            |-------------------------------------------------------|
-            ║  ● Owner:  {kl_owner:51} ║
+            | Klipper source repository:                            |
-            ║  ● Branch: {kl_branch:51} ║
+            |   ● {self.klipper_repo:<67} |
-            ╟───────────────────────────────────────────────────────╢
+            |                                                       |
-            ║ Moonraker:                                            ║
+            | Moonraker source repository:                          |
-            ║  ● Repo:   {mr_repo:51} ║
+            |   ● {self.moonraker_repo:<67} |
-            ║  ● Owner:  {mr_owner:51} ║
+            |                                                       |
-            ║  ● Branch: {mr_branch:51} ║
+            | Install unstable Webinterface releases:               |
-            ╟───────────────────────────────────────────────────────╢
+            |  {o1} Mainsail                                         |
-            ║ Install unstable releases:                            ║
+            |  {o2} Fluidd                                           |
-            ║  {o1} Mainsail                                         ║
+            |                                                       |
-            ║  {o2} Fluidd                                           ║
+            | Auto-Backup:                                          |
-            ╟───────────────────────────────────────────────────────╢
+            |  {o3} Automatic backup before update                   |
-            ║ Auto-Backup:                                          ║
+            |                                                       |
-            ║  {o3} Automatic backup before update                   ║
+            |-------------------------------------------------------|
-            ╟───────────────────────────────────────────────────────╢
+            | 1) Set Klipper source repository                      |
-            ║ 1) Set Klipper source repository                      ║
+            | 2) Set Moonraker source repository                    |
-            ║ 2) Set Moonraker source repository                    ║
+            |                                                       |
-            ║                                                       ║
+            | 3) Toggle unstable Mainsail releases                  |
-            ║ 3) Toggle unstable Mainsail releases                  ║
+            | 4) Toggle unstable Fluidd releases                    |
-            ║ 4) Toggle unstable Fluidd releases                    ║
+            |                                                       |
-            ║                                                       ║
+            | 5) Toggle automatic backups before updates            |
-            ║ 5) Toggle automatic backups before updates            ║
-            ╟───────────────────────────────────────────────────────╢
             """
         )[1:]
         print(menu, end="")
 
-    def _load_settings(self) -> None:
+    def _load_settings(self):
         self.settings = KiauhSettings()
+
+        self._format_repo_str("klipper")
+        self._format_repo_str("moonraker")
+
         self.auto_backups_enabled = self.settings.kiauh.backup_before_update
         self.mainsail_unstable = self.settings.mainsail.unstable_releases
         self.fluidd_unstable = self.settings.fluidd.unstable_releases
 
-        # by default, we show the status of the installed repositories
+    def _format_repo_str(self, repo_name: str) -> None:
-        self.klipper_status = get_klipper_status()
+        repo = self.settings.get(repo_name, "repo_url")
-        self.moonraker_status = get_moonraker_status()
+        repo = f"{'/'.join(repo.rsplit('/', 2)[-2:])}"
-        # if the repository is not installed, we show the status of the settings from the config file
+        branch = self.settings.get(repo_name, "branch")
-        if self.klipper_status.repo == "-":
+        branch = f"({COLOR_CYAN}@ {branch}{RESET_FORMAT})"
-            url_parts = self.settings.klipper.repo_url.split("/")
+        setattr(self, f"{repo_name}_repo", f"{COLOR_CYAN}{repo}{RESET_FORMAT} {branch}")
-            self.klipper_status.repo = url_parts[-1]
-            self.klipper_status.owner = url_parts[-2]
-            self.klipper_status.branch = self.settings.klipper.branch
-        if self.moonraker_status.repo == "-":
-            url_parts = self.settings.moonraker.repo_url.split("/")
-            self.moonraker_status.repo = url_parts[-1]
-            self.moonraker_status.owner = url_parts[-2]
-            self.moonraker_status.branch = self.settings.moonraker.branch
-
-    def _gather_input(
-        self, repo_name: Literal["klipper", "moonraker"], repo_dir: Path
-    ) -> Tuple[str, str]:
-        warn_msg = [
-            "There is only basic input validation in place! "
-            "Make sure your the input is valid and has no typos or invalid characters!"
-        ]
-
-        if repo_dir.exists():
-            warn_msg.extend(
-                [
-                    "For the change to take effect, the new repository will be cloned. "
-                    "A backup of the old repository will be created.",
-                    "\n\n",
-                    "Make sure you don't have any ongoing prints running, as the services "
-                    "will be restarted during this process! You will loose any ongoing print!",
-                ]
-            )
-
-        Logger.print_dialog(DialogType.ATTENTION, warn_msg)
 
+    def _gather_input(self) -> Tuple[str, str]:
+        Logger.print_dialog(
+            DialogType.ATTENTION,
+            [
+                "There is no input validation in place! Make sure your"
+                " input is valid and has no typos! For any change to"
+                " take effect, the repository must be cloned again. "
+                "Make sure you don't have any ongoing prints running, "
+                "as the services will be restarted!"
+            ],
+        )
         repo = get_string_input(
             "Enter new repository URL",
-            regex=r"^[\w/.:-]+$",
+            allow_special_chars=True,
-            default=KLIPPER_REPO_URL if repo_name == "klipper" else MOONRAKER_REPO_URL,
         )
         branch = get_string_input(
-            "Enter new branch name", regex=r"^.+$", default="master"
+            "Enter new branch name",
+            allow_special_chars=True,
         )
 
         return repo, branch
 
-    def _set_repo(
+    def _set_repo(self, repo_name: str):
-        self, repo_name: Literal["klipper", "moonraker"], repo_dir: Path
+        repo_url, branch = self._gather_input()
-    ) -> None:
-        repo_url, branch = self._gather_input(repo_name, repo_dir)
         display_name = repo_name.capitalize()
         Logger.print_dialog(
             DialogType.CUSTOM,
@@ -166,55 +143,66 @@ class SettingsMenu(BaseMenu):
                 f"New {display_name} repository branch:",
                 f"● {branch}",
             ],
+            end="",
         )
 
         if get_confirm("Apply changes?", allow_go_back=True):
-            repo: RepoSettings = self.settings[repo_name]
+            self.settings.set(repo_name, "repo_url", repo_url)
-            repo.repo_url = repo_url
+            self.settings.set(repo_name, "branch", branch)
-            repo.branch = branch
-
             self.settings.save()
             self._load_settings()
-
             Logger.print_ok("Changes saved!")
         else:
             Logger.print_info(
-                f"Changing of {display_name} source repository canceled ..."
+                f"Skipping change of {display_name} source repository  ..."
             )
             return
 
-        self._switch_repo(repo_name, repo_dir)
+        Logger.print_status(f"Switching to {display_name}'s new source repository ...")
+        self._switch_repo(repo_name)
+        Logger.print_ok(f"Switched to {repo_url} at branch {branch}!")
 
-    def _switch_repo(
+    def _switch_repo(self, name: str) -> None:
-        self, name: Literal["klipper", "moonraker"], repo_dir: Path
+        target_dir: Path
-    ) -> None:
+        if name == "klipper":
-        if not repo_dir.exists():
+            target_dir = KLIPPER_DIR
+            _type = Klipper
+        elif name == "moonraker":
+            target_dir = MOONRAKER_DIR
+            _type = Moonraker
+        else:
+            Logger.print_error("Invalid repository name!")
             return
 
-        Logger.print_status(
+        if target_dir.exists():
-            f"Switching to {name.capitalize()}'s new source repository ..."
+            shutil.rmtree(target_dir)
-        )
 
-        repo: RepoSettings = self.settings[name]
+        im = InstanceManager(_type)
-        run_switch_repo_routine(name, repo)
+        im.stop_all_instance()
 
-    def set_klipper_repo(self, **kwargs) -> None:
+        repo = self.settings.get(name, "repo_url")
-        self._set_repo("klipper", KLIPPER_DIR)
+        branch = self.settings.get(name, "branch")
+        git_clone_wrapper(repo, target_dir, branch)
 
-    def set_moonraker_repo(self, **kwargs) -> None:
+        im.start_all_instance()
-        self._set_repo("moonraker", MOONRAKER_DIR)
 
-    def toggle_mainsail_release(self, **kwargs) -> None:
+    def set_klipper_repo(self, **kwargs):
+        self._set_repo("klipper")
+
+    def set_moonraker_repo(self, **kwargs):
+        self._set_repo("moonraker")
+
+    def toggle_mainsail_release(self, **kwargs):
         self.mainsail_unstable = not self.mainsail_unstable
         self.settings.mainsail.unstable_releases = self.mainsail_unstable
         self.settings.save()
 
-    def toggle_fluidd_release(self, **kwargs) -> None:
+    def toggle_fluidd_release(self, **kwargs):
         self.fluidd_unstable = not self.fluidd_unstable
         self.settings.fluidd.unstable_releases = self.fluidd_unstable
         self.settings.save()
 
-    def toggle_backup_before_update(self, **kwargs) -> None:
+    def toggle_backup_before_update(self, **kwargs):
         self.auto_backups_enabled = not self.auto_backups_enabled
         self.settings.kiauh.backup_before_update = self.auto_backups_enabled
         self.settings.save()

kiauh/core/menus/update_menu.py

@@ -1,27 +1,30 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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
 
 import textwrap
-from typing import Callable, List, Type
+from typing import Optional, Type
 
 from components.crowsnest.crowsnest import get_crowsnest_status, update_crowsnest
+from components.klipper.klipper_setup import update_klipper
 from components.klipper.klipper_utils import (
     get_klipper_status,
 )
-from components.klipper.services.klipper_setup_service import KlipperSetupService
 from components.klipperscreen.klipperscreen import (
     get_klipperscreen_status,
     update_klipperscreen,
 )
+from components.mobileraker.mobileraker import (
+    get_mobileraker_status,
+    update_mobileraker,
+)
 from components.moonraker.moonraker_setup import update_moonraker
-from components.moonraker.utils.utils import get_moonraker_status
+from components.moonraker.moonraker_utils import get_moonraker_status
 from components.webui_client.client_config.client_config_setup import (
     update_client_config,
 )
@@ -32,295 +35,152 @@ from components.webui_client.client_utils import (
 )
 from components.webui_client.fluidd_data import FluiddData
 from components.webui_client.mainsail_data import MainsailData
-from core.logger import DialogType, Logger
 from core.menus import Option
 from core.menus.base_menu import BaseMenu
-from core.types.color import Color
+from utils.constants import (
-from core.types.component_status import ComponentStatus
+    COLOR_GREEN,
-from utils.input_utils import get_confirm
+    COLOR_RED,
-from utils.sys_utils import (
+    COLOR_YELLOW,
-    get_upgradable_packages,
+    RESET_FORMAT,
-    update_system_package_lists,
-    upgrade_system_packages,
 )
+from utils.types import ComponentStatus
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class UpdateMenu(BaseMenu):
-    def __init__(self, previous_menu: Type[BaseMenu] | None = None) -> None:
+    def __init__(self, previous_menu: Optional[Type[BaseMenu]] = None):
         super().__init__()
-        self.loading_msg = "Loading update menu, please wait"
+        self.previous_menu = previous_menu
-        self.is_loading(True)
 
-        self.title = "Update Menu"
+        self.kl_local = self.kl_remote = self.mr_local = self.mr_remote = ""
-        self.title_color = Color.GREEN
+        self.ms_local = self.ms_remote = self.fl_local = self.fl_remote = ""
-        self.previous_menu: Type[BaseMenu] | None = previous_menu
+        self.mc_local = self.mc_remote = self.fc_local = self.fc_remote = ""
-
+        self.ks_local = self.ks_remote = self.mb_local = self.mb_remote = ""
-        self.packages: List[str] = []
+        self.cn_local = self.cn_remote = ""
-        self.package_count: int = 0
-
-        self.klipper_local = self.klipper_remote = ""
-        self.moonraker_local = self.moonraker_remote = ""
-        self.mainsail_local = self.mainsail_remote = ""
-        self.mainsail_config_local = self.mainsail_config_remote = ""
-        self.fluidd_local = self.fluidd_remote = ""
-        self.fluidd_config_local = self.fluidd_config_remote = ""
-        self.klipperscreen_local = self.klipperscreen_remote = ""
-        self.crowsnest_local = self.crowsnest_remote = ""
 
         self.mainsail_data = MainsailData()
         self.fluidd_data = FluiddData()
-        self.status_data = {
-            "klipper": {
-                "display_name": "Klipper",
-                "installed": False,
-                "local": None,
-                "remote": None,
-            },
-            "moonraker": {
-                "display_name": "Moonraker",
-                "installed": False,
-                "local": None,
-                "remote": None,
-            },
-            "mainsail": {
-                "display_name": "Mainsail",
-                "installed": False,
-                "local": None,
-                "remote": None,
-            },
-            "mainsail_config": {
-                "display_name": "Mainsail-Config",
-                "installed": False,
-                "local": None,
-                "remote": None,
-            },
-            "fluidd": {
-                "display_name": "Fluidd",
-                "installed": False,
-                "local": None,
-                "remote": None,
-            },
-            "fluidd_config": {
-                "display_name": "Fluidd-Config",
-                "installed": False,
-                "local": None,
-                "remote": None,
-            },
-            "klipperscreen": {
-                "display_name": "KlipperScreen",
-                "installed": False,
-                "local": None,
-                "remote": None,
-            },
-            "crowsnest": {
-                "display_name": "Crowsnest",
-                "installed": False,
-                "local": None,
-                "remote": None,
-            },
-        }
-
         self._fetch_update_status()
-        self.is_loading(False)
 
-    def set_previous_menu(self, previous_menu: Type[BaseMenu] | None) -> None:
+    def set_previous_menu(self, previous_menu: Optional[Type[BaseMenu]]) -> None:
         from core.menus.main_menu import MainMenu
 
-        self.previous_menu = previous_menu if previous_menu is not None else MainMenu
+        self.previous_menu: Type[BaseMenu] = (
+            previous_menu if previous_menu is not None else MainMenu
+        )
 
     def set_options(self) -> None:
         self.options = {
-            "a": Option(self.update_all),
+            "0": Option(self.update_all, menu=False),
-            "1": Option(self.update_klipper),
+            "1": Option(self.update_klipper, menu=False),
-            "2": Option(self.update_moonraker),
+            "2": Option(self.update_moonraker, menu=False),
-            "3": Option(self.update_mainsail),
+            "3": Option(self.update_mainsail, menu=False),
-            "4": Option(self.update_fluidd),
+            "4": Option(self.update_fluidd, menu=False),
-            "5": Option(self.update_mainsail_config),
+            "5": Option(self.update_mainsail_config, menu=False),
-            "6": Option(self.update_fluidd_config),
+            "6": Option(self.update_fluidd_config, menu=False),
-            "7": Option(self.update_klipperscreen),
+            "7": Option(self.update_klipperscreen, menu=False),
-            "8": Option(self.update_crowsnest),
+            "8": Option(self.update_mobileraker, menu=False),
-            "9": Option(self.upgrade_system_packages),
+            "9": Option(self.update_crowsnest, menu=False),
+            "10": Option(self.upgrade_system_packages, menu=False),
         }
 
-    def print_menu(self) -> None:
+    def print_menu(self):
-        sysupgrades: str = "No upgrades available."
+        self._fetch_update_status()
-        padding = 29
-        if self.package_count > 0:
-            sysupgrades = Color.apply(
-                f"{self.package_count} upgrades available!", Color.GREEN
-            )
-            padding = 38
 
+        header = " [ Update Menu ] "
+        color = COLOR_GREEN
+        count = 62 - len(color) - len(RESET_FORMAT)
         menu = textwrap.dedent(
             f"""
-            ╟───────────────────────┬───────────────┬───────────────╢
+            /=======================================================\\
-            ║  a) Update all        │               │               ║
+            | {color}{header:~^{count}}{RESET_FORMAT} |
-            ║                       │ Current:      │ Latest:       ║
+            |-------------------------------------------------------|
-            ║ Klipper & API:        ├───────────────┼───────────────╢
+            |  0) Update all        |               |               |
-            ║  1) Klipper           │ {self.klipper_local:<22} │ {self.klipper_remote:<22} ║
+            |                       | Current:      | Latest:       |
-            ║  2) Moonraker         │ {self.moonraker_local:<22} │ {self.moonraker_remote:<22} ║
+            | Klipper & API:        |---------------|---------------|
-            ║                       │               │               ║
+            |  1) Klipper           | {self.kl_local:<22} | {self.kl_remote:<22} |
-            ║ Webinterface:         ├───────────────┼───────────────╢
+            |  2) Moonraker         | {self.mr_local:<22} | {self.mr_remote:<22} |
-            ║  3) Mainsail          │ {self.mainsail_local:<22} │ {self.mainsail_remote:<22} ║
+            |                       |               |               |
-            ║  4) Fluidd            │ {self.fluidd_local:<22} │ {self.fluidd_remote:<22} ║
+            | Webinterface:         |---------------|---------------|
-            ║                       │               │               ║
+            |  3) Mainsail          | {self.ms_local:<22} | {self.ms_remote:<22} |
-            ║ Client-Config:        ├───────────────┼───────────────╢
+            |  4) Fluidd            | {self.fl_local:<22} | {self.fl_remote:<22} |
-            ║  5) Mainsail-Config   │ {self.mainsail_config_local:<22} │ {self.mainsail_config_remote:<22} ║
+            |                       |               |               |
-            ║  6) Fluidd-Config     │ {self.fluidd_config_local:<22} │ {self.fluidd_config_remote:<22} ║
+            | Client-Config:        |---------------|---------------|
-            ║                       │               │               ║
+            |  5) Mainsail-Config   | {self.mc_local:<22} | {self.mc_remote:<22} |
-            ║ Other:                ├───────────────┼───────────────╢
+            |  6) Fluidd-Config     | {self.fc_local:<22} | {self.fc_remote:<22} |
-            ║  7) KlipperScreen     │ {self.klipperscreen_local:<22} │ {self.klipperscreen_remote:<22} ║
+            |                       |               |               |
-            ║  8) Crowsnest         │ {self.crowsnest_local:<22} │ {self.crowsnest_remote:<22} ║
+            | Other:                |---------------|---------------|
-            ║                       ├───────────────┴───────────────╢
+            |  7) KlipperScreen     | {self.ks_local:<22} | {self.ks_remote:<22} |
-            ║  9) System            │ {sysupgrades:^{padding}} ║
+            |  8) Mobileraker       | {self.mb_local:<22} | {self.mb_remote:<22} |
-            ╟───────────────────────┴───────────────────────────────╢
+            |  9) Crowsnest         | {self.cn_local:<22} | {self.cn_remote:<22} |
+            |                       |-------------------------------|
+            | 10) System            |                               |
             """
         )[1:]
         print(menu, end="")
 
-    def update_all(self, **kwargs) -> None:
+    def update_all(self, **kwargs):
-        Logger.print_status("Updating all components ...")
+        print("update_all")
-        self.update_klipper()
-        self.update_moonraker()
-        self.update_mainsail()
-        self.update_mainsail_config()
-        self.update_fluidd()
-        self.update_fluidd_config()
-        self.update_klipperscreen()
-        self.update_crowsnest()
-        self.upgrade_system_packages()
 
-    def update_klipper(self, **kwargs) -> None:
+    def update_klipper(self, **kwargs):
-        klsvc = KlipperSetupService()
+        update_klipper()
-        self._run_update_routine("klipper", klsvc.update)
 
-    def update_moonraker(self, **kwargs) -> None:
+    def update_moonraker(self, **kwargs):
-        self._run_update_routine("moonraker", update_moonraker)
+        update_moonraker()
 
-    def update_mainsail(self, **kwargs) -> None:
+    def update_mainsail(self, **kwargs):
-        self._run_update_routine(
+        update_client(self.mainsail_data)
-            "mainsail",
-            update_client,
-            self.mainsail_data,
-        )
 
-    def update_mainsail_config(self, **kwargs) -> None:
+    def update_mainsail_config(self, **kwargs):
-        self._run_update_routine(
+        update_client_config(self.mainsail_data)
-            "mainsail_config",
-            update_client_config,
-            self.mainsail_data,
-        )
 
-    def update_fluidd(self, **kwargs) -> None:
+    def update_fluidd(self, **kwargs):
-        self._run_update_routine(
+        update_client(self.fluidd_data)
-            "fluidd",
-            update_client,
-            self.fluidd_data,
-        )
 
-    def update_fluidd_config(self, **kwargs) -> None:
+    def update_fluidd_config(self, **kwargs):
-        self._run_update_routine(
+        update_client_config(self.fluidd_data)
-            "fluidd_config",
-            update_client_config,
-            self.fluidd_data,
-        )
 
-    def update_klipperscreen(self, **kwargs) -> None:
+    def update_klipperscreen(self, **kwargs):
-        self._run_update_routine("klipperscreen", update_klipperscreen)
+        update_klipperscreen()
 
-    def update_crowsnest(self, **kwargs) -> None:
+    def update_mobileraker(self, **kwargs):
-        self._run_update_routine("crowsnest", update_crowsnest)
+        update_mobileraker()
 
-    def upgrade_system_packages(self, **kwargs) -> None:
+    def update_crowsnest(self, **kwargs):
-        self._run_system_updates()
+        update_crowsnest()
 
-    def _fetch_update_status(self) -> None:
+    def upgrade_system_packages(self, **kwargs): ...
-        self._set_status_data("klipper", get_klipper_status)
-        self._set_status_data("moonraker", get_moonraker_status)
-        self._set_status_data("mainsail", get_client_status, self.mainsail_data, True)
-        self._set_status_data(
-            "mainsail_config", get_client_config_status, self.mainsail_data
-        )
-        self._set_status_data("fluidd", get_client_status, self.fluidd_data, True)
-        self._set_status_data(
-            "fluidd_config", get_client_config_status, self.fluidd_data
-        )
-        self._set_status_data("klipperscreen", get_klipperscreen_status)
-        self._set_status_data("crowsnest", get_crowsnest_status)
 
-        update_system_package_lists(silent=True)
+    def _fetch_update_status(self):
-        self.packages = get_upgradable_packages()
+        # klipper
-        self.package_count = len(self.packages)
+        self._get_update_status("kl", get_klipper_status)
+        # moonraker
+        self._get_update_status("mr", get_moonraker_status)
+        # mainsail
+        self._get_update_status("ms", get_client_status, self.mainsail_data, True)
+        # mainsail-config
+        self._get_update_status("mc", get_client_config_status, self.mainsail_data)
+        # fluidd
+        self._get_update_status("fl", get_client_status, self.fluidd_data, True)
+        # fluidd-config
+        self._get_update_status("fc", get_client_config_status, self.fluidd_data)
+        # klipperscreen
+        self._get_update_status("ks", get_klipperscreen_status)
+        # mobileraker
+        self._get_update_status("mb", get_mobileraker_status)
+        # crowsnest
+        self._get_update_status("cn", get_crowsnest_status)
 
     def _format_local_status(self, local_version, remote_version) -> str:
-        color = Color.RED
+        if local_version == remote_version:
-        if not local_version:
+            return f"{COLOR_GREEN}{local_version}{RESET_FORMAT}"
-            color = Color.RED
+        return f"{COLOR_YELLOW}{local_version}{RESET_FORMAT}"
-        elif local_version == remote_version:
-            color = Color.GREEN
-        elif local_version != remote_version:
-            color = Color.YELLOW
 
-        return Color.apply(local_version or "-", color)
+    def _get_update_status(self, name: str, status_fn: callable, *args) -> None:
-
+        status_data: ComponentStatus = status_fn(*args)
-    def _set_status_data(self, name: str, status_fn: Callable, *args) -> None:
+        local_ver = status_data.get("local")
-        comp_status: ComponentStatus = status_fn(*args)
+        remote_ver = status_data.get("remote")
-
+        color = COLOR_GREEN if remote_ver != "ERROR" else COLOR_RED
-        self.status_data[name]["installed"] = True if comp_status.status == 2 else False
+        setattr(self, f"{name}_local", self._format_local_status(local_ver, remote_ver))
-        self.status_data[name]["local"] = comp_status.local
+        setattr(self, f"{name}_remote", f"{color}{remote_ver}{RESET_FORMAT}")
-        self.status_data[name]["remote"] = comp_status.remote
-
-        self._set_status_string(name)
-
-    def _set_status_string(self, name: str) -> None:
-        local_status = self.status_data[name].get("local", None)
-        remote_status = self.status_data[name].get("remote", None)
-
-        color = Color.GREEN if remote_status else Color.RED
-        local_txt = self._format_local_status(local_status, remote_status)
-        remote_txt = Color.apply(remote_status or "-", color)
-
-        setattr(self, f"{name}_local", local_txt)
-        setattr(self, f"{name}_remote", remote_txt)
-
-    def _check_is_installed(self, name: str) -> bool:
-        return self.status_data[name]["installed"]
-
-    def _is_update_available(self, name: str) -> bool:
-        return self.status_data[name]["local"] != self.status_data[name]["remote"]
-
-    def _run_update_routine(self, name: str, update_fn: Callable, *args) -> None:
-        display_name = self.status_data[name]["display_name"]
-        is_installed = self._check_is_installed(name)
-        is_update_available = self._is_update_available(name)
-
-        if not is_installed:
-            Logger.print_info(f"{display_name} is not installed! Skipped ...")
-            return
-        elif not is_update_available:
-            Logger.print_info(f"{display_name} is already up to date! Skipped ...")
-            return
-
-        update_fn(*args)
-
-    def _run_system_updates(self) -> None:
-        if not self.packages:
-            Logger.print_info("No system upgrades available!")
-            return
-
-        try:
-            pkgs: str = ", ".join(self.packages)
-            Logger.print_dialog(
-                DialogType.CUSTOM,
-                ["The following packages will be upgraded:", "\n\n", pkgs],
-                custom_title="UPGRADABLE SYSTEM UPDATES",
-            )
-            if not get_confirm("Continue?"):
-                return
-            Logger.print_status("Upgrading system packages ...")
-            upgrade_system_packages(self.packages)
-        except Exception as e:
-            Logger.print_error(f"Error upgrading system packages:\n{e}")
-            raise

kiauh/core/services/message_service.py

@@ -1,61 +0,0 @@
-# ======================================================================= #
-#  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
-
-from dataclasses import dataclass, field
-from typing import List
-
-from core.logger import DialogType, Logger
-from core.types.color import Color
-
-
-@dataclass()
-class Message:
-    title: str = field(default="")
-    text: List[str] = field(default_factory=list)
-    color: Color = field(default=Color.WHITE)
-    centered: bool = field(default=False)
-
-
-class MessageService:
-    __cls_instance = None
-    __message: Message | None
-
-    def __new__(cls) -> "MessageService":
-        if cls.__cls_instance is None:
-            cls.__cls_instance = super(MessageService, cls).__new__(cls)
-        return cls.__cls_instance
-
-    def __init__(self) -> None:
-        if not hasattr(self, "__initialized"):
-            self.__initialized = False
-        if self.__initialized:
-            return
-        self.__initialized = True
-        self.__message = None
-
-    def set_message(self, message: Message) -> None:
-        self.__message = message
-
-    def display_message(self) -> None:
-        if self.__message is None:
-            return
-
-        Logger.print_dialog(
-            title=DialogType.CUSTOM,
-            content=self.__message.text,
-            custom_title=self.__message.title,
-            custom_color=self.__message.color,
-            center_content=self.__message.centered,
-        )
-
-        self.__clear_message()
-
-    def __clear_message(self) -> None:
-        self.__message = None

kiauh/core/settings/kiauh_settings.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2020 - 2025 Dominik Willner <th33xitus@gmail.com>        #
+#  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                                          #
@@ -8,79 +8,90 @@
 # ======================================================================= #
 from __future__ import annotations
 
-from dataclasses import dataclass, field
+import textwrap
-from typing import Any
+from typing import Union
 
-from core.logger import DialogType, Logger
 from core.submodules.simple_config_parser.src.simple_config_parser.simple_config_parser import (
     NoOptionError,
     NoSectionError,
     SimpleConfigParser,
 )
+from utils.constants import COLOR_RED, RESET_FORMAT
+from utils.logger import Logger
 from utils.sys_utils import kill
 
 from kiauh import PROJECT_ROOT
 
-DEFAULT_CFG = PROJECT_ROOT.joinpath("default.kiauh.cfg")
-CUSTOM_CFG = PROJECT_ROOT.joinpath("kiauh.cfg")
 
-
-@dataclass
 class AppSettings:
-    backup_before_update: bool | None = field(default=None)
+    def __init__(self) -> None:
+        self.backup_before_update = None
 
 
-@dataclass
+class KlipperSettings:
-class RepoSettings:
+    def __init__(self) -> None:
-    repo_url: str | None = field(default=None)
+        self.repo_url = None
-    branch: str | None = field(default=None)
+        self.branch = None
 
 
-@dataclass
+class MoonrakerSettings:
-class WebUiSettings:
+    def __init__(self) -> None:
-    port: str | None = field(default=None)
+        self.repo_url = None
-    unstable_releases: bool | None = field(default=None)
+        self.branch = None
+
+
+class MainsailSettings:
+    def __init__(self) -> None:
+        self.port = None
+        self.unstable_releases = None
+
+
+class FluiddSettings:
+    def __init__(self) -> None:
+        self.port = None
+        self.unstable_releases = None
 
 
 # noinspection PyUnusedLocal
 # noinspection PyMethodMayBeStatic
 class KiauhSettings:
     _instance = None
+    _default_cfg = PROJECT_ROOT.joinpath("default_kiauh.cfg")
+    _custom_cfg = PROJECT_ROOT.joinpath("kiauh.cfg")
 
     def __new__(cls, *args, **kwargs) -> "KiauhSettings":
         if cls._instance is None:
             cls._instance = super(KiauhSettings, cls).__new__(cls, *args, **kwargs)
+            cls._instance.__initialized = False
         return cls._instance
 
-    def __repr__(self) -> str:
-        return (
-            f"KiauhSettings(kiauh={self.kiauh}, klipper={self.klipper},"
-            f" moonraker={self.moonraker}, mainsail={self.mainsail},"
-            f" fluidd={self.fluidd})"
-        )
-
-    def __getitem__(self, item: str) -> Any:
-        return getattr(self, item)
-
     def __init__(self) -> None:
-        if not hasattr(self, "__initialized"):
-            self.__initialized = False
         if self.__initialized:
             return
         self.__initialized = True
         self.config = SimpleConfigParser()
         self.kiauh = AppSettings()
-        self.klipper = RepoSettings()
+        self.klipper = KlipperSettings()
-        self.moonraker = RepoSettings()
+        self.moonraker = MoonrakerSettings()
-        self.mainsail = WebUiSettings()
+        self.mainsail = MainsailSettings()
-        self.fluidd = WebUiSettings()
+        self.fluidd = FluiddSettings()
+
+        self.kiauh.backup_before_update = None
+        self.klipper.repo_url = None
+        self.klipper.branch = None
+        self.moonraker.repo_url = None
+        self.moonraker.branch = None
+        self.mainsail.port = None
+        self.mainsail.unstable_releases = None
+        self.fluidd.port = None
+        self.fluidd.unstable_releases = None
 
         self._load_config()
 
-    def get(self, section: str, option: str) -> str | int | bool:
+    def get(self, section: str, option: str) -> Union[str, int, bool]:
         """
-        Get a value from the settings state by providing the section and option name as
+        Get a value from the settings state by providing the section and option name as strings.
-        strings. Prefer direct access to the properties, as it is usually safer!
+        Prefer direct access to the properties, as it is usually safer!
         :param section: The section name as string.
         :param option: The option name as string.
         :return: The value of the option as string, int or bool.
@@ -89,24 +100,38 @@ class KiauhSettings:
         try:
             section = getattr(self, section)
             value = getattr(section, option)
-            return value  # type: ignore
+            return value
+        except AttributeError:
+            raise
+
+    def set(self, section: str, option: str, value: Union[str, int, bool]) -> None:
+        """
+        Set a value in the settings state by providing the section and option name as strings.
+        Prefer direct access to the properties, as it is usually safer!
+        :param section: The section name as string.
+        :param option: The option name as string.
+        :param value: The value to set as string, int or bool.
+        """
+        try:
+            section = getattr(self, section)
+            section.option = value
         except AttributeError:
             raise
 
     def save(self) -> None:
-        self._set_config_options_state()
+        self._set_config_options()
-        self.config.write_file(CUSTOM_CFG)
+        self.config.write(self._custom_cfg)
         self._load_config()
 
     def _load_config(self) -> None:
-        if not CUSTOM_CFG.exists() and not DEFAULT_CFG.exists():
+        if not self._custom_cfg.exists() or not self._default_cfg.exists():
             self._kill()
 
-        cfg = CUSTOM_CFG if CUSTOM_CFG.exists() else DEFAULT_CFG
+        cfg = self._custom_cfg if self._custom_cfg.exists() else self._default_cfg
-        self.config.read_file(cfg)
+        self.config.read(cfg)
 
         self._validate_cfg()
-        self._apply_settings_from_file()
+        self._read_settings()
 
     def _validate_cfg(self) -> None:
         try:
@@ -136,7 +161,7 @@ class KiauhSettings:
 
     def _validate_bool(self, section: str, option: str) -> None:
         self._v_section, self._v_option = (section, option)
-        (bool(self.config.getboolean(section, option)))
+        bool(self.config.getboolean(section, option))
 
     def _validate_int(self, section: str, option: str) -> None:
         self._v_section, self._v_option = (section, option)
@@ -144,19 +169,18 @@ class KiauhSettings:
 
     def _validate_str(self, section: str, option: str) -> None:
         self._v_section, self._v_option = (section, option)
-        v = self.config.getval(section, option)
+        v = self.config.get(section, option)
-
+        if v.isdigit() or v.lower() == "true" or v.lower() == "false":
-        if not v:
             raise ValueError
 
-    def _apply_settings_from_file(self) -> None:
+    def _read_settings(self):
         self.kiauh.backup_before_update = self.config.getboolean(
             "kiauh", "backup_before_update"
         )
-        self.klipper.repo_url = self.config.getval("klipper", "repo_url")
+        self.klipper.repo_url = self.config.get("klipper", "repo_url")
-        self.klipper.branch = self.config.getval("klipper", "branch")
+        self.klipper.branch = self.config.get("klipper", "branch")
-        self.moonraker.repo_url = self.config.getval("moonraker", "repo_url")
+        self.moonraker.repo_url = self.config.get("moonraker", "repo_url")
-        self.moonraker.branch = self.config.getval("moonraker", "branch")
+        self.moonraker.branch = self.config.get("moonraker", "branch")
         self.mainsail.port = self.config.getint("mainsail", "port")
         self.mainsail.unstable_releases = self.config.getboolean(
             "mainsail", "unstable_releases"
@@ -166,35 +190,37 @@ class KiauhSettings:
             "fluidd", "unstable_releases"
         )
 
-    def _set_config_options_state(self) -> None:
+    def _set_config_options(self):
-        self.config.set_option(
+        self.config.set(
             "kiauh",
             "backup_before_update",
             str(self.kiauh.backup_before_update),
         )
-        self.config.set_option("klipper", "repo_url", self.klipper.repo_url)
+        self.config.set("klipper", "repo_url", self.klipper.repo_url)
-        self.config.set_option("klipper", "branch", self.klipper.branch)
+        self.config.set("klipper", "branch", self.klipper.branch)
-        self.config.set_option("moonraker", "repo_url", self.moonraker.repo_url)
+        self.config.set("moonraker", "repo_url", self.moonraker.repo_url)
-        self.config.set_option("moonraker", "branch", self.moonraker.branch)
+        self.config.set("moonraker", "branch", self.moonraker.branch)
-        self.config.set_option("mainsail", "port", str(self.mainsail.port))
+        self.config.set("mainsail", "port", str(self.mainsail.port))
-        self.config.set_option(
+        self.config.set(
             "mainsail",
             "unstable_releases",
             str(self.mainsail.unstable_releases),
         )
-        self.config.set_option("fluidd", "port", str(self.fluidd.port))
+        self.config.set("fluidd", "port", str(self.fluidd.port))
-        self.config.set_option(
+        self.config.set(
             "fluidd", "unstable_releases", str(self.fluidd.unstable_releases)
         )
 
     def _kill(self) -> None:
-        Logger.print_dialog(
+        l1 = "!!! ERROR !!!"
-            DialogType.ERROR,
+        l2 = "No KIAUH configuration file found!"
-            [
+        error = textwrap.dedent(
-                "No KIAUH configuration file found! Please make sure you have at least "
+            f"""
-                "one of the following configuration files in KIAUH's root directory:",
+            {COLOR_RED}/=======================================================\\
-                "● default.kiauh.cfg",
+            | {l1:^53} |
-                "● kiauh.cfg",
+            | {l2:^53} |
-            ],
+            \=======================================================/{RESET_FORMAT}
-        )
+            """
+        )[1:]
+        print(error, end="")
         kill()

kiauh/core/spinner.py

@@ -1,42 +0,0 @@
-import sys
-import threading
-import time
-from typing import List, Literal
-
-from core.types.color import Color
-
-SpinnerColor = Literal["white", "red", "green", "yellow"]
-
-
-class Spinner:
-    def __init__(
-        self,
-        message: str = "Loading",
-        interval: float = 0.2,
-    ) -> None:
-        self.message = f"{message} ..."
-        self.interval = interval
-        self._stop_event = threading.Event()
-        self._thread = threading.Thread(target=self._animate)
-
-    def _animate(self) -> None:
-        animation: List[str] = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
-        while not self._stop_event.is_set():
-            for char in animation:
-                sys.stdout.write(f"\r{Color.GREEN}{char}{Color.RST} {self.message}")
-                sys.stdout.flush()
-                time.sleep(self.interval)
-                if self._stop_event.is_set():
-                    break
-        sys.stdout.write("\r" + " " * (len(self.message) + 1) + "\r")
-        sys.stdout.flush()
-
-    def start(self) -> None:
-        self._stop_event.clear()
-        if not self._thread.is_alive():
-            self._thread = threading.Thread(target=self._animate)
-            self._thread.start()
-
-    def stop(self) -> None:
-        self._stop_event.set()
-        self._thread.join()

kiauh/core/submodules/simple_config_parser/src/simple_config_parser/constants.py

@@ -1,62 +0,0 @@
-# ======================================================================= #
-#  Copyright (C) 2024 Dominik Willner <th33xitus@gmail.com>               #
-#                                                                         #
-#  https://github.com/dw-0/simple-config-parser                           #
-#                                                                         #
-#  This file may be distributed under the terms of the GNU GPLv3 license  #
-# ======================================================================= #
-import re
-
-# definition of section line:
-#  - then line MUST start with an opening square bracket - it is the first section marker
-#  - the section marker MUST be followed by at least one character - it is the section name
-#  - the section name MUST be followed by a closing square bracket - it is the second section marker
-#  - the second section marker MAY be followed by any amount of whitespace characters
-#  - the second section marker MAY be followed by a # or ; - it is the comment marker
-#  - the inline comment MAY be of any length and character
-SECTION_RE = re.compile(r"^\[(\S.*\S|\S)]\s*([#;].*)?$")
-
-# definition of option line:
-#  - the line MUST start with a word - it is the option name
-#  - the option name MUST be followed by a colon or an equal sign - it is the separator
-#  - the separator MUST be followed by a value
-#    - the separator MAY have any amount of leading or trailing whitespaces
-#    - the separator MUST NOT be directly followed by a colon or equal sign
-#  - the value MAY be of any length and character
-#    - the value MAY contain any amount of trailing whitespaces
-#    - the value MAY be followed by a # or ; - it is the comment marker
-#  - the inline comment MAY be of any length and character
-OPTION_RE = re.compile(r"^([^;#:=\s]+)\s?[:=]\s*([^;#:=\s][^;#]*?)\s*([#;].*)?$")
-# definition of options block start line:
-#  - the line MUST start with a word - it is the option name
-#  - the option name MUST be followed by a colon or an equal sign - it is the separator
-#  - the separator MUST NOT be followed by a value
-#    - the separator MAY have any amount of leading or trailing whitespaces
-#    - the separator MUST NOT be directly followed by a colon or equal sign
-#    - the separator MAY be followed by a # or ; - it is the comment marker
-#  - the inline comment MAY be of any length and character
-OPTIONS_BLOCK_START_RE = re.compile(r"^([^;#:=\s]+)\s*[:=]\s*([#;].*)?$")
-
-# definition of comment line:
-#  - the line MAY start with any amount of whitespace characters
-#  - the line MUST contain a # or ; - it is the comment marker
-#  - the comment marker MAY be followed by any amount of whitespace characters
-#  - the comment MAY be of any length and character
-LINE_COMMENT_RE = re.compile(r"^\s*[#;].*")
-
-# definition of empty line:
-#  - the line MUST contain only whitespace characters
-EMPTY_LINE_RE = re.compile(r"^\s*$")
-
-BOOLEAN_STATES = {
-    "1": True,
-    "yes": True,
-    "true": True,
-    "on": True,
-    "0": False,
-    "no": False,
-    "false": False,
-    "off": False,
-}
-
-HEADER_IDENT = "#_header"

kiauh/core/submodules/simple_config_parser/src/simple_config_parser/simple_config_parser.py

@@ -1,5 +1,5 @@
 # ======================================================================= #
-#  Copyright (C) 2024 Dominik Willner <th33xitus@gmail.com>               #
+#  Copyright (C) 2020 - 2024 Dominik Willner <th33xitus@gmail.com>        #
 #                                                                         #
 #  https://github.com/dw-0/simple-config-parser                           #
 #                                                                         #
@@ -8,24 +8,47 @@
 
 from __future__ import annotations
 
-import secrets
+import re
-import string
 from pathlib import Path
-from typing import Callable, Dict, List
+from typing import Callable, Dict, List, Match, Tuple, TypedDict
-
-from ..simple_config_parser.constants import (
-    BOOLEAN_STATES,
-    EMPTY_LINE_RE,
-    HEADER_IDENT,
-    LINE_COMMENT_RE,
-    OPTION_RE,
-    OPTIONS_BLOCK_START_RE,
-    SECTION_RE,
-)
 
 _UNSET = object()
 
 
+class Section(TypedDict):
+    """
+    A single section in the config file
+
+    - _raw: The raw representation of the section name
+    - options: A list of options in the section
+    """
+
+    _raw: str
+    options: List[Option]
+
+
+class Option(TypedDict, total=False):
+    """
+    A single option in a section in the config file
+
+    - is_multiline: Whether the option is a multiline option
+    - option: The name of the option
+    - value: The value of the option
+    - _raw: The raw representation of the option
+    - _raw_value: The raw value of the option
+
+    A multinline option is an option that contains multiple lines of text following
+    the option name in the next line. The value of a multiline option is a list of
+    strings, where each string represents a single line of text.
+    """
+
+    is_multiline: bool
+    option: str
+    value: str | List[str]
+    _raw: str
+    _raw_value: str | List[str]
+
+
 class NoSectionError(Exception):
     """Raised when a section is not defined"""
 
@@ -34,14 +57,6 @@ class NoSectionError(Exception):
         super().__init__(msg)
 
 
-class DuplicateSectionError(Exception):
-    """Raised when a section is defined more than once"""
-
-    def __init__(self, section: str):
-        msg = f"Section '{section}' is defined more than once"
-        super().__init__(msg)
-
-
 class NoOptionError(Exception):
     """Raised when an option is not defined in a section"""
 
@@ -50,212 +65,143 @@ class NoOptionError(Exception):
         super().__init__(msg)
 
 
+class DuplicateSectionError(Exception):
+    """Raised when a section is defined more than once"""
+
+    def __init__(self, section: str):
+        msg = f"Section '{section}' is defined more than once"
+        super().__init__(msg)
+
+
+class DuplicateOptionError(Exception):
+    """Raised when an option is defined more than once"""
+
+    def __init__(self, option: str, section: str):
+        msg = f"Option '{option}' in section '{section}' is defined more than once"
+        super().__init__(msg)
+
+
 # noinspection PyMethodMayBeStatic
 class SimpleConfigParser:
     """A customized config parser targeted at handling Klipper style config files"""
 
-    def __init__(self) -> None:
+    _SECTION_RE = re.compile(r"\s*\[(\w+ ?\w+)]\s*([#;].*)?$")
-        self.header: List[str] = []
+    _OPTION_RE = re.compile(r"^\s*(\w+)\s*[:=]\s*([^=:].*)\s*([#;].*)?$")
-        self.config: Dict = {}
+    _MLOPTION_RE = re.compile(r"^\s*(\w+)\s*[:=]\s*([#;].*)?$")
-        self.current_section: str | None = None
+    _COMMENT_RE = re.compile(r"^\s*([#;].*)?$")
-        self.current_opt_block: str | None = None
+    _EMPTY_LINE_RE = re.compile(r"^\s*$")
-        self.current_collector: str | None = None
-        self.in_option_block: bool = False
 
-    def _match_section(self, line: str) -> bool:
+    BOOLEAN_STATES = {
-        """Wheter or not the given line matches the definition of a section"""
+        "1": True,
-        return SECTION_RE.match(line) is not None
+        "yes": True,
+        "true": True,
+        "on": True,
+        "0": False,
+        "no": False,
+        "false": False,
+        "off": False,
+    }
 
-    def _match_option(self, line: str) -> bool:
+    def __init__(self):
-        """Wheter or not the given line matches the definition of an option"""
+        self._config: Dict = {}
-        return OPTION_RE.match(line) is not None
+        self._header: List[str] = []
+        self._all_sections: List[str] = []
+        self._all_options: Dict = {}
+        self.section_name: str = ""
+        self.in_option_block: bool = False  # whether we are in a multiline option block
 
-    def _match_options_block_start(self, line: str) -> bool:
+    def read(self, file: Path) -> None:
-        """Wheter or not the given line matches the definition of a multiline option"""
+        """
-        return OPTIONS_BLOCK_START_RE.match(line) is not None
+        Read the given file and store the result in the internal state.
+        Call this method before using any other methods. Calling this method
+        multiple times will reset the internal state on each call.
+        """
 
-    def _match_line_comment(self, line: str) -> bool:
+        self._reset_state()
-        """Wheter or not the given line matches the definition of a comment"""
-        return LINE_COMMENT_RE.match(line) is not None
 
-    def _match_empty_line(self, line: str) -> bool:
+        try:
-        """Wheter or not the given line matches the definition of an empty line"""
+            with open(file, "r") as f:
-        return EMPTY_LINE_RE.match(line) is not None
+                self._parse_config(f.readlines())
 
-    def _parse_line(self, line: str) -> None:
+        except OSError:
-        """Parses a line and determines its type"""
+            raise
-        if self._match_section(line):
-            self.current_collector = None
-            self.current_opt_block = None
-            self.current_section = SECTION_RE.match(line).group(1)
-            self.config[self.current_section] = {"_raw": line}
 
-        elif self._match_option(line):
+    def _reset_state(self):
-            self.current_collector = None
+        """Reset the internal state."""
-            self.current_opt_block = None
-            option = OPTION_RE.match(line).group(1)
-            value = OPTION_RE.match(line).group(2)
-            self.config[self.current_section][option] = {"_raw": line, "value": value}
 
-        elif self._match_options_block_start(line):
+        self._config.clear()
-            self.current_collector = None
+        self._header.clear()
-            option = OPTIONS_BLOCK_START_RE.match(line).group(1)
+        self._all_sections.clear()
-            self.current_opt_block = option
+        self._all_options.clear()
-            self.config[self.current_section][option] = {"_raw": line, "value": []}
+        self.section_name = ""
+        self.in_option_block = False
 
-        elif self.current_opt_block is not None:
+    def write(self, filename):
-            self.config[self.current_section][self.current_opt_block]["value"].append(
+        """Write the internal state to the given file"""
-                line
-            )
 
-        elif self._match_empty_line(line) or self._match_line_comment(line):
+        content = self._construct_content()
-            self.current_opt_block = None
 
-            # if current_section is None, we are at the beginning of the file,
+        with open(filename, "w") as f:
-            # so we consider the part up to the first section as the file header
+            f.write(content)
-            if not self.current_section:
-                self.config.setdefault(HEADER_IDENT, []).append(line)
-            else:
-                section = self.config[self.current_section]
 
-                # set the current collector to a new value, so that continuous
+    def _construct_content(self) -> str:
-                # empty lines or comments are collected into the same collector
+        """
-                if not self.current_collector:
+        Constructs the content of the configuration file based on the internal state of
-                    self.current_collector = self._generate_rand_id()
+        the _config object by iterating over the sections and their options. It starts
-                    section[self.current_collector] = []
+        by checking if a header is present and extends the content list with its elements.
+        Then, for each section, it appends the raw representation of the section to the
+        content list. If the section has a body, it iterates over its options and extends
+        the content list with their raw representations. If an option is multiline, it
+        also extends the content list with its raw value. Finally, the content list is
+        joined into a single string and returned.
 
-                section[self.current_collector].append(line)
+        :return: The content of the configuration file as a string
+        """
+        content: List[str] = []
+        if self._header is not None:
+            content.extend(self._header)
+        for section in self._config:
+            content.append(self._config[section]["_raw"])
 
-    def read_file(self, file: Path) -> None:
+            if (sec_body := self._config[section].get("body")) is not None:
-        """Read and parse a config file"""
+                for option in sec_body:
-        with open(file, "r") as file:
+                    content.extend(option["_raw"])
-            for line in file:
+                    if option["is_multiline"]:
-                self._parse_line(line)
+                        content.extend(option["_raw_value"])
+        content: str = "".join(content)
 
-        # print(json.dumps(self.config, indent=4))
+        return content
 
-    def write_file(self, file: Path) -> None:
+    def sections(self) -> List[str]:
-        """Write the current config to the config file"""
+        """Return a list of section names"""
-        if not file:
-            raise ValueError("No config file specified")
 
-        with open(file, "w") as file:
+        return self._all_sections
-            self._write_header(file)
-            self._write_sections(file)
-
-    def _write_header(self, file) -> None:
-        """Write the header to the config file"""
-        for line in self.config.get(HEADER_IDENT, []):
-            file.write(line)
-
-    def _write_sections(self, file) -> None:
-        """Write the sections to the config file"""
-        for section in self.get_sections():
-            for key, value in self.config[section].items():
-                self._write_section_content(file, key, value)
-
-    def _write_section_content(self, file, key, value) -> None:
-        """Write the content of a section to the config file"""
-        if key == "_raw":
-            file.write(value)
-        elif key.startswith("#_"):
-            for line in value:
-                file.write(line)
-        elif isinstance(value["value"], list):
-            file.write(value["_raw"])
-            for line in value["value"]:
-                file.write(line)
-        else:
-            file.write(value["_raw"])
-
-    def get_sections(self) -> List[str]:
-        """Return a list of all section names, but exclude any section starting with '#_'"""
-        return list(
-            filter(
-                lambda section: not section.startswith("#_"),
-                self.config.keys(),
-            )
-        )
-
-    def has_section(self, section: str) -> bool:
-        """Check if a section exists"""
-        return section in self.get_sections()
 
     def add_section(self, section: str) -> None:
-        """Add a new section to the config"""
+        """Add a new section to the internal state"""
-        if section in self.get_sections():
+
+        if section in self._all_sections:
             raise DuplicateSectionError(section)
-
+        self._all_sections.append(section)
-        if len(self.get_sections()) >= 1:
+        self._all_options[section] = {}
-            self._check_set_section_spacing()
+        self._config[section] = {"_raw": f"\n[{section}]\n", "body": []}
-
-        self.config[section] = {"_raw": f"[{section}]\n"}
-
-    def _check_set_section_spacing(self):
-        prev_section_name: str = self.get_sections()[-1]
-        prev_section_content: Dict = self.config[prev_section_name]
-        last_option_name: str = list(prev_section_content.keys())[-1]
-
-        if last_option_name.startswith("#_"):
-            last_elem_value: str = prev_section_content[last_option_name][-1]
-
-            # if the last section is a collector, we first check if the last element
-            # in the collector ends with a newline. if it does not, we append a newline.
-            # this can happen if the config file does not end with a newline.
-            if not last_elem_value.endswith("\n"):
-                prev_section_content[last_option_name][-1] = f"{last_elem_value}\n"
-
-            # if the last item in a collector is not a newline, we append a newline, so
-            # that the new section is seperated from the options of the previous section
-            # by a newline
-            if last_elem_value != "\n":
-                prev_section_content[last_option_name].append("\n")
-        else:
-            prev_section_content[self._generate_rand_id()] = ["\n"]
 
     def remove_section(self, section: str) -> None:
-        """Remove a section from the config"""
+        """Remove the given section"""
-        self.config.pop(section, None)
 
-    def get_options(self, section: str) -> List[str]:
+        if section not in self._all_sections:
-        """Return a list of all option names for a given section"""
+            raise NoSectionError(section)
-        return list(
-            filter(
-                lambda option: option != "_raw" and not option.startswith("#_"),
-                self.config[section].keys(),
-            )
-        )
 
-    def has_option(self, section: str, option: str) -> bool:
+        del self._all_sections[self._all_sections.index(section)]
-        """Check if an option exists in a section"""
+        del self._all_options[section]
-        return self.has_section(section) and option in self.get_options(section)
+        del self._config[section]
 
-    def set_option(self, section: str, option: str, value: str | List[str]) -> None:
+    def options(self, section) -> List[str]:
-        """
+        """Return a list of option names for the given section name"""
-        Set the value of an option in a section. If the section does not exist,
-        it is created. If the option does not exist, it is created.
-        """
-        if not self.has_section(section):
-            self.add_section(section)
 
-        if not self.has_option(section, option):
+        return self._all_options.get(section)
-            self.config[section][option] = {
-                "_raw": f"{option}:\n"
-                if isinstance(value, list)
-                else f"{option}: {value}\n",
-                "value": value,
-            }
-        else:
-            opt = self.config[section][option]
-            if not isinstance(value, list):
-                opt["_raw"] = opt["_raw"].replace(opt["value"], value)
-            opt["value"] = value
 
-    def remove_option(self, section: str, option: str) -> None:
+    def get(
-        """Remove an option from a section"""
-        self.config[section].pop(option, None)
-
-    def getval(
         self, section: str, option: str, fallback: str | _UNSET = _UNSET
     ) -> str | List[str]:
         """
@@ -264,12 +210,15 @@ class SimpleConfigParser:
         If the key is not found and 'fallback' is provided, it is used as
         a fallback value.
         """
+
         try:
-            if section not in self.get_sections():
+            if section not in self._all_sections:
                 raise NoSectionError(section)
-            if option not in self.get_options(section):
+
+            if option not in self._all_options.get(section):
                 raise NoOptionError(option, section)
-            return self.config[section][option]["value"]
+
+            return self._all_options[section][option]
         except (NoSectionError, NoOptionError):
             if fallback is _UNSET:
                 raise
@@ -277,29 +226,25 @@ class SimpleConfigParser:
 
     def getint(self, section: str, option: str, fallback: int | _UNSET = _UNSET) -> int:
         """Return the value of the given option in the given section as an int"""
+
         return self._get_conv(section, option, int, fallback=fallback)
 
     def getfloat(
         self, section: str, option: str, fallback: float | _UNSET = _UNSET
     ) -> float:
-        """Return the value of the given option in the given section as a float"""
         return self._get_conv(section, option, float, fallback=fallback)
 
     def getboolean(
         self, section: str, option: str, fallback: bool | _UNSET = _UNSET
     ) -> bool:
-        """Return the value of the given option in the given section as a boolean"""
         return self._get_conv(
             section, option, self._convert_to_boolean, fallback=fallback
         )
 
-    def _convert_to_boolean(self, value: str) -> bool:
+    def _convert_to_boolean(self, value) -> bool:
-        """Convert a string to a boolean"""
+        if value.lower() not in self.BOOLEAN_STATES:
-        if isinstance(value, bool):
-            return value
-        if value.lower() not in BOOLEAN_STATES:
             raise ValueError("Not a boolean: %s" % value)
-        return BOOLEAN_STATES[value.lower()]
+        return self.BOOLEAN_STATES[value.lower()]
 
     def _get_conv(
         self,
@@ -308,18 +253,299 @@ class SimpleConfigParser:
         conv: Callable[[str], int | float | bool],
         fallback: _UNSET = _UNSET,
     ) -> int | float | bool:
-        """Return the value of the given option in the given section as a converted value"""
         try:
-            return conv(self.getval(section, option, fallback))
+            return conv(self.get(section, option, fallback))
-        except (ValueError, TypeError, AttributeError) as e:
+        except:
             if fallback is not _UNSET:
                 return fallback
-            raise ValueError(
+            raise
-                f"Cannot convert {self.getval(section, option)} to {conv.__name__}"
-            ) from e
 
-    def _generate_rand_id(self) -> str:
+    def items(self, section: str) -> List[Tuple[str, str]]:
-        """Generate a random id with 6 characters"""
+        """Return a list of (option, value) tuples for a specific section"""
-        chars = string.ascii_letters + string.digits
+
-        rand_string = "".join(secrets.choice(chars) for _ in range(12))
+        if section not in self._all_sections:
-        return f"#_{rand_string}"
+            raise NoSectionError(section)
+
+        result = []
+        for _option in self._all_options[section]:
+            result.append((_option, self._all_options[section][_option]))
+
+        return result
+
+    def set(
+        self,
+        section: str,
+        option: str,
+        value: str,
+        multiline: bool = False,
+        indent: int = 4,
+    ) -> None:
+        """Set the given option to the given value in the given section
+
+        If the option is already defined, it will be overwritten. If the option
+        is not defined yet, it will be added to the section body.
+
+        The multiline parameter can be used to specify whether the value is
+        multiline or not. If it is not specified, the value will be considered
+        as multiline if it contains a newline character. The value will then be split
+        into multiple lines. If the value does not contain a newline character, it
+        will be considered as a single line value. The indent parameter can be used
+        to specify the indentation of the multiline value. Indentations are with spaces.
+
+        :param section: The section to set the option in
+        :param option: The option to set
+        :param value: The value to set
+        :param multiline: Whether the value is multiline or not
+        :param indent: The indentation for multiline values
+        """
+
+        if section not in self._all_sections:
+            raise NoSectionError(section)
+
+        # prepare the options value and raw value depending on the multiline flag
+        _raw_value: List[str] | None = None
+        if multiline or "\n" in value:
+            _multiline = True
+            _raw: str = f"{option}:\n"
+            _value: List[str] = value.split("\n")
+            _raw_value: List[str] = [f"{' ' * indent}{v}\n" for v in _value]
+        else:
+            _multiline = False
+            _raw: str = f"{option}: {value}\n"
+            _value: str = value
+
+        # the option does not exist yet
+        if option not in self._all_options.get(section):
+            _option: Option = {
+                "is_multiline": _multiline,
+                "option": option,
+                "value": _value,
+                "_raw": _raw,
+            }
+            if _raw_value is not None:
+                _option["_raw_value"] = _raw_value
+            self._config[section]["body"].insert(0, _option)
+
+        # the option exists and we need to update it
+        else:
+            for _option in self._config[section]["body"]:
+                if _option["option"] == option:
+                    if multiline:
+                        _option["_raw"] = _raw
+                    else:
+                        # we preserve inline comments by replacing the old value with the new one
+                        _option["_raw"] = _option["_raw"].replace(
+                            _option["value"], _value
+                        )
+                    _option["value"] = _value
+                    if _raw_value is not None:
+                        _option["_raw_value"] = _raw_value
+                    break
+
+        self._all_options[section][option] = _value
+
+    def remove_option(self, section: str, option: str) -> None:
+        """Remove the given option from the given section"""
+
+        if section not in self._all_sections:
+            raise NoSectionError(section)
+
+        if option not in self._all_options.get(section):
+            raise NoOptionError(option, section)
+
+        for _option in self._config[section]["body"]:
+            if _option["option"] == option:
+                del self._all_options[section][option]
+                self._config[section]["body"].remove(_option)
+                break
+
+    def has_section(self, section: str) -> bool:
+        """Return True if the given section exists, False otherwise"""
+        return section in self._all_sections
+
+    def has_option(self, section: str, option: str) -> bool:
+        """Return True if the given option exists in the given section, False otherwise"""
+        return option in self._all_options.get(section)
+
+    def _is_section(self, line: str) -> bool:
+        """Check if the given line contains a section definition"""
+        return self._SECTION_RE.match(line) is not None
+
+    def _is_option(self, line: str) -> bool:
+        """Check if the given line contains an option definition"""
+
+        match: Match[str] | None = self._OPTION_RE.match(line)
+
+        if not match:
+            return False
+
+        # if there is no value, it's not a regular option but a multiline option
+        if match.group(2).strip() == "":
+            return False
+
+        if not match.group(1).strip() == "":
+            return True
+
+        return False
+
+    def _is_comment(self, line: str) -> bool:
+        """Check if the given line is a comment"""
+        return self._COMMENT_RE.match(line) is not None
+
+    def _is_empty_line(self, line: str) -> bool:
+        """Check if the given line is an empty line"""
+        return self._EMPTY_LINE_RE.match(line) is not None
+
+    def _is_multiline_option(self, line: str) -> bool:
+        """Check if the given line starts a multiline option block"""
+
+        match: Match[str] | None = self._MLOPTION_RE.match(line)
+
+        if not match:
+            return False
+
+        return True
+
+    def _parse_config(self, content: List[str]) -> None:
+        """Parse the given content and store the result in the internal state"""
+
+        _curr_multi_opt = ""
+
+        # THE ORDER MATTERS, DO NOT REORDER THE CONDITIONS!
+        for line in content:
+            if self._is_section(line):
+                self._parse_section(line)
+
+            elif self._is_option(line):
+                self._parse_option(line)
+
+            # if it's not a regular option with the value inline,
+            # it might be a might be a multiline option block
+            elif self._is_multiline_option(line):
+                self.in_option_block = True
+                _curr_multi_opt = self._OPTION_RE.match(line).group(1).strip()
+                self._add_option_to_section_body(_curr_multi_opt, "", line)
+
+            elif self.in_option_block:
+                self._parse_multiline_option(_curr_multi_opt, line)
+
+            # if it's nothing from above, it's probably a comment or an empty line
+            elif self._is_comment(line) or self._is_empty_line(line):
+                self._parse_comment(line)
+
+    def _parse_section(self, line: str) -> None:
+        """Parse a section line and store the result in the internal state"""
+
+        match: Match[str] | None = self._SECTION_RE.match(line)
+        if not match:
+            return
+
+        self.in_option_block = False
+
+        section_name: str = match.group(1).strip()
+        self._store_internal_state_section(section_name, line)
+
+    def _store_internal_state_section(self, section: str, raw_value: str) -> None:
+        """Store the given section and its raw value in the internal state"""
+
+        if section in self._all_sections:
+            raise DuplicateSectionError(section)
+
+        self.section_name = section
+        self._all_sections.append(section)
+        self._config[section]: Section = {"_raw": raw_value, "body": []}
+
+    def _parse_option(self, line: str) -> None:
+        """Parse an option line and store the result in the internal state"""
+
+        self.in_option_block = False
+
+        match: Match[str] | None = self._OPTION_RE.match(line)
+        if not match:
+            return
+
+        option: str = match.group(1).strip()
+        value: str = match.group(2).strip()
+
+        if ";" in value:
+            i = value.index(";")
+            value = value[:i].strip()
+        elif "#" in value:
+            i = value.index("#")
+            value = value[:i].strip()
+
+        self._store_internal_state_option(option, value, line)
+
+    def _store_internal_state_option(
+        self, option: str, value: str, raw_value: str
+    ) -> None:
+        """Store the given option and its raw value in the internal state"""
+
+        section_options = self._all_options.setdefault(self.section_name, {})
+
+        if option in section_options:
+            raise DuplicateOptionError(option, self.section_name)
+
+        section_options[option] = value
+        self._add_option_to_section_body(option, value, raw_value)
+
+    def _parse_multiline_option(self, curr_ml_opt: str, line: str) -> None:
+        """Parse a multiline option line and store the result in the internal state"""
+
+        section_options = self._all_options.setdefault(self.section_name, {})
+        multiline_options = section_options.setdefault(curr_ml_opt, [])
+
+        _cleaned_line = line.strip().strip("\n")
+        if _cleaned_line and not self._is_comment(line):
+            multiline_options.append(_cleaned_line)
+
+        # add the option to the internal multiline option value state
+        self._ensure_section_body_exists()
+        for _option in self._config[self.section_name]["body"]:
+            if _option.get("option") == curr_ml_opt:
+                _option.update(
+                    is_multiline=True,
+                    _raw_value=_option.get("_raw_value", []) + [line],
+                    value=multiline_options,
+                )
+
+    def _parse_comment(self, line: str) -> None:
+        """
+        Parse a comment line and store the result in the internal state
+
+        If the there was no previous section parsed, the lines are handled as
+        the file header and added to the internal header list as it means, that
+        we are at the very top of the file.
+        """
+
+        self.in_option_block = False
+
+        if not self.section_name:
+            self._header.append(line)
+        else:
+            self._add_option_to_section_body("", "", line)
+
+    def _ensure_section_body_exists(self) -> None:
+        """
+        Ensure that the section body exists in the internal state.
+        If the section body does not exist, it is created as an empty list
+        """
+        if self.section_name not in self._config:
+            self._config.setdefault(self.section_name, {}).setdefault("body", [])
+
+    def _add_option_to_section_body(
+        self, option: str, value: str, line: str, is_multiline: bool = False
+    ) -> None:
+        """Add a raw option line to the internal state"""
+
+        self._ensure_section_body_exists()
+
+        new_option: Option = {
+            "is_multiline": is_multiline,
+            "option": option,
+            "value": value,
+            "_raw": line,
+        }
+
+        option_body = self._config[self.section_name]["body"]
+        option_body.append(new_option)

kiauh/core/submodules/simple_config_parser/tests/Test SimpleConfigParser.run.xml

@@ -0,0 +1,21 @@
+<component name="ProjectRunConfigurationManager">
+  <configuration default="false" name="Test SimpleConfigParser" type="tests" factoryName="py.test">
+    <module name="simple-config-parser" />
+    <option name="ENV_FILES" value="" />
+    <option name="INTERPRETER_OPTIONS" value="" />
+    <option name="PARENT_ENVS" value="true" />
+    <option name="SDK_HOME" value="" />
+    <option name="SDK_NAME" value="Python 3.8 (simple-config-parser)" />
+    <option name="WORKING_DIRECTORY" value="" />
+    <option name="IS_MODULE_SDK" value="false" />
+    <option name="ADD_CONTENT_ROOTS" value="true" />
+    <option name="ADD_SOURCE_ROOTS" value="true" />
+    <EXTENSION ID="PythonCoverageRunConfigurationExtension" runner="coverage.py" />
+    <option name="_new_keywords" value="&quot;&quot;" />
+    <option name="_new_parameters" value="&quot;&quot;" />
+    <option name="_new_additionalArguments" value="&quot;-s -vv&quot;" />
+    <option name="_new_target" value="&quot;&quot;" />
+    <option name="_new_targetType" value="&quot;PATH&quot;" />
+    <method v="2" />
+  </configuration>
+</component>

kiauh/core/submodules/simple_config_parser/tests/assets/test_config_1.cfg

@@ -1,32 +0,0 @@
-# a comment at the very top
-# should be treated as the file header
-
-# up to the first section, including all blank lines
-
-[section_1]
-option_1: value_1
-option_1_1: True # this is a boolean
-option_1_2: 5 ; this is an integer
-option_1_3: 1.123 #;this is a float
-
-[section_2] ; comment
-option_2: value_2
-
-; comment
-
-[section_3]
-option_3: value_3 # comment
-
-[section_4]
-# comment
-option_4: value_4
-
-[section number 5]
-#option_5: value_5
-option_5 = this.is.value-5
-multi_option:
-  # these are multi-line values
-  value_5_1
-  value_5_2 ; here is a comment
-  value_5_3
-option_5_1: value_5_1

kiauh/core/submodules/simple_config_parser/tests/assets/test_config_2.cfg

@@ -1,33 +0,0 @@
-# a comment at the very top
-# should be treated as the file header
-
-# up to the first section, including all blank lines
-
-[section_1]
-option_1: value_1
-option_1_1: True # this is a boolean
-option_1_2: 5 ; this is an integer
-option_1_3: 1.123 #;this is a float
-
-[section_2] ; comment
-option_2: value_2
-
-; comment
-
-[section_3]
-option_3: value_3 # comment
-
-[section_4]
-# comment
-option_4: value_4
-
-[section number 5]
-#option_5: value_5
-option_5 = this.is.value-5
-multi_option:
-  # these are multi-line values
-  value_5_1
-  value_5_2 ; here is a comment
-  value_5_3
-option_5_1: value_5_1
-# config ending with a comment

kiauh/core/submodules/simple_config_parser/tests/assets/test_config_3.cfg

@@ -1,94 +0,0 @@
-# a comment at the very top
-# should be treated as the file header
-
-# up to the first section, including all blank lines
-
-[section_1]
-option_1: value_1
-option_1_1: True # this is a boolean
-option_1_2: 5 ; this is an integer
-option_1_3: 1.123 #;this is a float
-
-[section_2] ; comment
-option_2: value_2
-
-; comment
-
-[section_3]
-option_3: value_3 # comment
-
-[section_4]
-# comment
-option_4: value_4
-
-[section number 5]
-#option_5: value_5
-option_5 = this.is.value-5
-multi_option:
-  # these are multi-line values
-  value_5_1
-  value_5_2 ; here is a comment
-  value_5_3
-option_5_1: value_5_1
-
-[gcode_macro M117]
-rename_existing: M117.1
-gcode:
-  {% if rawparams %}
-    {% set escaped_msg = rawparams.split(';', 1)[0].split('\x23', 1)[0]|replace('"', '\\"') %}
-    SET_DISPLAY_TEXT MSG="{escaped_msg}"
-    RESPOND TYPE=command MSG="{escaped_msg}"
-  {% else %}
-    SET_DISPLAY_TEXT
-  {% endif %}
-
-# SDCard 'looping' (aka Marlin M808 commands) support
-#
-# Support SDCard looping
-[sdcard_loop]
-[gcode_macro M486]
-gcode:
-  # Parameters known to M486 are as follows:
-  #   [C<flag>] Cancel the current object
-  #   [P<index>] Cancel the object with the given index
-  #   [S<index>] Set the index of the current object.
-  #       If the object with the given index has been canceled, this will cause
-  #       the firmware to skip to the next object. The value -1 is used to
-  #       indicate something that isn’t an object and shouldn’t be skipped.
-  #   [T<count>] Reset the state and set the number of objects
-  #   [U<index>] Un-cancel the object with the given index. This command will be
-  #       ignored if the object has already been skipped
-
-  {% if 'exclude_object' not in printer %}
-    {action_raise_error("[exclude_object] is not enabled")}
-  {% endif %}
-
-  {% if 'T' in params %}
-    EXCLUDE_OBJECT RESET=1
-
-    {% for i in range(params.T | int) %}
-      EXCLUDE_OBJECT_DEFINE NAME={i}
-    {% endfor %}
-  {% endif %}
-
-  {% if 'C' in params %}
-    EXCLUDE_OBJECT CURRENT=1
-  {% endif %}
-
-  {% if 'P' in params %}
-    EXCLUDE_OBJECT NAME={params.P}
-  {% endif %}
-
-  {% if 'S' in params %}
-    {% if params.S == '-1' %}
-      {% if printer.exclude_object.current_object %}
-        EXCLUDE_OBJECT_END NAME={printer.exclude_object.current_object}
-      {% endif %}
-    {% else %}
-      EXCLUDE_OBJECT_START NAME={params.S}
-    {% endif %}
-  {% endif %}
-
-  {% if 'U' in params %}
-    EXCLUDE_OBJECT RESET=1 NAME={params.U}
-  {% endif %}

kiauh/core/submodules/simple_config_parser/tests/features/internal_state/test_content_handling.py

@@ -0,0 +1,95 @@
+import pytest
+
+from src.simple_config_parser.simple_config_parser import SimpleConfigParser
+
+
+@pytest.fixture
+def parser():
+    parser = SimpleConfigParser()
+    parser._header = ["header1\n", "header2\n"]
+    parser._config = {
+        "section1": {
+            "_raw": "[section1]\n",
+            "body": [
+                {
+                    "_raw": "option1: value1\n",
+                    "_raw_value": "value1\n",
+                    "is_multiline": False,
+                    "option": "option1",
+                    "value": "value1",
+                },
+                {
+                    "_raw": "option2: value2\n",
+                    "_raw_value": "value2\n",
+                    "is_multiline": False,
+                    "option": "option2",
+                    "value": "value2",
+                },
+            ],
+        },
+        "section2": {
+            "_raw": "[section2]\n",
+            "body": [
+                {
+                    "_raw": "option3: value3\n",
+                    "_raw_value": "value3\n",
+                    "is_multiline": False,
+                    "option": "option3",
+                    "value": "value3",
+                },
+            ],
+        },
+        "section3": {
+            "_raw": "[section3]\n",
+            "body": [
+                {
+                    "_raw": "option4:\n",
+                    "_raw_value": ["    value4\n", "    value5\n", "    value6\n"],
+                    "is_multiline": True,
+                    "option": "option4",
+                    "value": ["value4", "value5", "value6"],
+                },
+            ],
+        },
+    }
+    return parser
+
+
+def test_construct_content(parser):
+    content = parser._construct_content()
+    assert (
+        content == "header1\nheader2\n"
+        "[section1]\n"
+        "option1: value1\n"
+        "option2: value2\n"
+        "[section2]\n"
+        "option3: value3\n"
+        "[section3]\n"
+        "option4:\n"
+        "    value4\n"
+        "    value5\n"
+        "    value6\n"
+    )
+
+
+def test_construct_content_no_header(parser):
+    parser._header = None
+    content = parser._construct_content()
+    assert (
+        content == "[section1]\n"
+        "option1: value1\n"
+        "option2: value2\n"
+        "[section2]\n"
+        "option3: value3\n"
+        "[section3]\n"
+        "option4:\n"
+        "    value4\n"
+        "    value5\n"
+        "    value6\n"
+    )
+
+
+def test_construct_content_no_sections(parser):
+    parser._config = {}
+    content = parser._construct_content()
+    assert content == "".join(parser._header)

kiauh/core/submodules/simple_config_parser/tests/features/internal_state/test_internal_state_changes.py

@@ -0,0 +1,83 @@
+import pytest
+
+from src.simple_config_parser.simple_config_parser import (
+    DuplicateOptionError,
+    DuplicateSectionError,
+    SimpleConfigParser,
+)
+
+
+@pytest.fixture
+def parser():
+    return SimpleConfigParser()
+
+
+class TestInternalStateChanges:
+    @pytest.mark.parametrize(
+        "given", ["dummy_section", "dummy_section 2", "another_section"]
+    )
+    def test_ensure_section_body_exists(self, parser, given):
+        parser._config = {}
+        parser.section_name = given
+        parser._ensure_section_body_exists()
+
+        assert parser._config[given] is not None
+        assert parser._config[given]["body"] == []
+
+    def test_add_option_to_section_body(self):
+        pass
+
+    @pytest.mark.parametrize(
+        "given", ["dummy_section", "dummy_section 2", "another_section\n"]
+    )
+    def test_store_internal_state_section(self, parser, given):
+        parser._store_internal_state_section(given, given)
+
+        assert parser._all_sections == [given]
+        assert parser._config[given]["body"] == []
+        assert parser._config[given]["_raw"] == given
+
+    def test_duplicate_section_error(self, parser):
+        section_name = "dummy_section"
+        parser._all_sections = [section_name]
+
+        with pytest.raises(DuplicateSectionError) as excinfo:
+            parser._store_internal_state_section(section_name, section_name)
+            message = f"Section '{section_name}' is defined more than once"
+            assert message in str(excinfo.value)
+
+        # Check that the internal state of the parser is correct
+        assert parser.in_option_block is False
+        assert parser.section_name == ""
+        assert parser._all_sections == [section_name]
+
+    @pytest.mark.parametrize(
+        "given_name, given_value, given_raw_value",
+        [("dummyoption", "dummyvalue", "dummyvalue\n")],
+    )
+    def test_store_internal_state_option(
+        self, parser, given_name, given_value, given_raw_value
+    ):
+        parser.section_name = "dummy_section"
+        parser._store_internal_state_option(given_name, given_value, given_raw_value)
+
+        assert parser._all_options[parser.section_name] == {given_name: given_value}
+
+        new_option = {
+            "is_multiline": False,
+            "option": given_name,
+            "value": given_value,
+            "_raw": given_raw_value,
+        }
+        assert parser._config[parser.section_name]["body"] == [new_option]
+
+    def test_duplicate_option_error(self, parser):
+        option_name = "dummyoption"
+        value = "dummyvalue"
+        parser.section_name = "dummy_section"
+        parser._all_options = {parser.section_name: {option_name: value}}
+
+        with pytest.raises(DuplicateOptionError) as excinfo:
+            parser._store_internal_state_option(option_name, value, value)
+            message = f"Option '{option_name}' in section '{parser.section_name}' is defined more than once"
+            assert message in str(excinfo.value)

kiauh/core/submodules/simple_config_parser/tests/features/line_parsing/data/case_parse_comment.py

@@ -0,0 +1,6 @@
+testcases = [
+    "# comment # 1",
+    "; comment # 2",
+    "    ; indented comment",
+    "        # another indented comment",
+]

kiauh/core/submodules/simple_config_parser/tests/features/line_parsing/data/case_parse_option.py

@@ -0,0 +1,19 @@
+testcases = [
+    ("option: value", "option", "value"),
+    ("option : value", "option", "value"),
+    ("option :value", "option", "value"),
+    ("option= value", "option", "value"),
+    ("option = value", "option", "value"),
+    ("option =value", "option", "value"),
+    ("option: value\n", "option", "value"),
+    ("option: value # inline comment", "option", "value"),
+    ("option: value # inline comment\n", "option", "value"),
+    ("description: homing!", "description", "homing!"),
+    ("description: inline macro :-)", "description", "inline macro :-)"),
+    ("path: %GCODES_DIR%", "path", "%GCODES_DIR%"),
+    (
+        "serial = /dev/serial/by-id/<your-mcu-id>",
+        "serial",
+        "/dev/serial/by-id/<your-mcu-id>",
+    ),
+]

kiauh/core/submodules/simple_config_parser/tests/features/line_parsing/data/case_parse_section.py

@@ -0,0 +1,6 @@
+testcases = [
+    ("[test_section]", "test_section"),
+    ("[test_section two]", "test_section two"),
+    ("[section1] # inline comment", "section1"),
+    ("[section2] ; second comment", "section2"),
+]

kiauh/core/submodules/simple_config_parser/tests/features/line_parsing/test_line_parsing.py

@@ -0,0 +1,92 @@
+import pytest
+from data.case_parse_comment import testcases as case_parse_comment
+from data.case_parse_option import testcases as case_parse_option
+from data.case_parse_section import testcases as case_parse_section
+
+from src.simple_config_parser.simple_config_parser import (
+    Option,
+    SimpleConfigParser,
+)
+
+
+@pytest.fixture
+def parser():
+    return SimpleConfigParser()
+
+
+class TestLineParsing:
+    @pytest.mark.parametrize("given, expected", [*case_parse_section])
+    def test_parse_section(self, parser, given, expected):
+        parser._parse_section(given)
+
+        # Check that the internal state of the parser is correct
+        assert parser.section_name == expected
+        assert parser.in_option_block is False
+        assert parser._all_sections == [expected]
+        assert parser._config[expected]["_raw"] == given
+        assert parser._config[expected]["body"] == []
+
+    @pytest.mark.parametrize(
+        "given, expected_option, expected_value", [*case_parse_option]
+    )
+    def test_parse_option(self, parser, given, expected_option, expected_value):
+        section_name = "test_section"
+        parser.section_name = section_name
+        parser._parse_option(given)
+
+        # Check that the internal state of the parser is correct
+        assert parser.section_name == section_name
+        assert parser.in_option_block is False
+        assert parser._all_options[section_name][expected_option] == expected_value
+
+        section_option = parser._config[section_name]["body"][0]
+        assert section_option["option"] == expected_option
+        assert section_option["value"] == expected_value
+        assert section_option["_raw"] == given
+
+    @pytest.mark.parametrize(
+        "option, next_line",
+        [("gcode", "next line"), ("gcode", "    {{% some jinja template %}}")],
+    )
+    def test_parse_multiline_option(self, parser, option, next_line):
+        parser.section_name = "dummy_section"
+        parser.in_option_block = True
+        parser._add_option_to_section_body(option, "", option)
+        parser._parse_multiline_option(option, next_line)
+        cleaned_next_line = next_line.strip().strip("\n")
+
+        assert parser._all_options[parser.section_name] is not None
+        assert parser._all_options[parser.section_name][option] == [cleaned_next_line]
+
+        expected_option: Option = {
+            "is_multiline": True,
+            "option": option,
+            "value": [cleaned_next_line],
+            "_raw": option,
+            "_raw_value": [next_line],
+        }
+        assert parser._config[parser.section_name]["body"] == [expected_option]
+
+    @pytest.mark.parametrize("given", [*case_parse_comment])
+    def test_parse_comment(self, parser, given):
+        parser.section_name = "dummy_section"
+        parser._parse_comment(given)
+
+        # internal state checks after parsing
+        assert parser.in_option_block is False
+
+        expected_option = {
+            "is_multiline": False,
+            "_raw": given,
+            "option": "",
+            "value": "",
+        }
+        assert parser._config[parser.section_name]["body"] == [expected_option]
+
+    @pytest.mark.parametrize("given", ["# header line", "; another header line"])
+    def test_parse_header_comment(self, parser, given):
+        parser.section_name = ""
+        parser._parse_comment(given)
+
+        assert parser.in_option_block is False
+        assert parser._header == [given]

kiauh/core/submodules/simple_config_parser/tests/features/line_type_detection/data/case_line_is_comment.py

@@ -0,0 +1,9 @@
+testcases = [
+    ("# an arbitrary comment", True),
+    ("; another arbitrary comment", True),
+    ("  ; indented comment", True),
+    ("  # indented comment", True),
+    ("not_a: comment", False),
+    ("also_not_a= comment", False),
+    ("[definitely_not_a_comment]", False),
+]

kiauh/core/submodules/simple_config_parser/tests/features/line_type_detection/data/case_line_is_empty.py

@@ -0,0 +1,9 @@
+testcases = [
+    ("", True),
+    (" ", True),
+    ("not empty", False),
+    ("  # indented comment", False),
+    ("not: empty", False),
+    ("also_not= empty", False),
+    ("[definitely_not_empty]", False),
+]

kiauh/core/submodules/simple_config_parser/tests/features/line_type_detection/data/case_line_is_multiline_option.py

@@ -0,0 +1,17 @@
+testcases = [
+    ("valid_option:", True),
+    ("valid_option:\n", True),
+    ("valid_option: ; inline comment", True),
+    ("valid_option: # inline comment", True),
+    ("valid_option :", True),
+    ("valid_option=", True),
+    ("valid_option= ", True),
+    ("valid_option =", True),
+    ("valid_option = ", True),
+    ("invalid_option ==", False),
+    ("invalid_option :=", False),
+    ("not_a_valid_option", False),
+    ("", False),
+    ("# that's a comment", False),
+    ("; that's a comment", False),
+]

kiauh/core/submodules/simple_config_parser/tests/features/line_type_detection/data/case_line_is_option.py

@@ -0,0 +1,30 @@
+testcases = [
+    ("valid_option: value", True),
+    ("valid_option: value\n", True),
+    ("valid_option: value ; inline comment", True),
+    ("valid_option: value # inline comment", True),
+    ("valid_option: value # inline comment\n", True),
+    ("valid_option : value", True),
+    ("valid_option :value", True),
+    ("valid_option= value", True),
+    ("valid_option = value", True),
+    ("valid_option =value", True),
+    ("invalid_option:", False),
+    ("invalid_option=", False),
+    ("invalid_option:: value", False),
+    ("invalid_option :: value", False),
+    ("invalid_option ::value", False),
+    ("invalid_option== value", False),
+    ("invalid_option == value", False),
+    ("invalid_option ==value", False),
+    ("invalid_option:= value", False),
+    ("invalid_option := value", False),
+    ("invalid_option :=value", False),
+    ("[that_is_a_section]", False),
+    ("[that_is_section two]", False),
+    ("not_a_valid_option", False),
+    ("description: homing!", True),
+    ("description: inline macro :-)", True),
+    ("path: %GCODES_DIR%", True),
+    ("serial = /dev/serial/by-id/<your-mcu-id>", True),
+]

kiauh/core/submodules/simple_config_parser/tests/features/line_type_detection/data/case_line_is_section.py

@@ -0,0 +1,6 @@
+testcases = [
+    ("[example_section]", True),
+    ("[example_section two]", True),
+    ("not_a_valid_section", False),
+    ("section: invalid", False),
+]

kiauh/core/submodules/simple_config_parser/tests/features/line_type_detection/test_line_type_detection.py

@@ -0,0 +1,37 @@
+import pytest
+from data.case_line_is_comment import testcases as case_line_is_comment
+from data.case_line_is_empty import testcases as case_line_is_empty
+from data.case_line_is_multiline_option import (
+    testcases as case_line_is_multiline_option,
+)
+from data.case_line_is_option import testcases as case_line_is_option
+from data.case_line_is_section import testcases as case_line_is_section
+
+from src.simple_config_parser.simple_config_parser import SimpleConfigParser
+
+
+@pytest.fixture
+def parser():
+    return SimpleConfigParser()
+
+
+class TestLineTypeDetection:
+    @pytest.mark.parametrize("given, expected", [*case_line_is_section])
+    def test_line_is_section(self, parser, given, expected):
+        assert parser._is_section(given) is expected
+
+    @pytest.mark.parametrize("given, expected", [*case_line_is_option])
+    def test_line_is_option(self, parser, given, expected):
+        assert parser._is_option(given) is expected
+
+    @pytest.mark.parametrize("given, expected", [*case_line_is_multiline_option])
+    def test_line_is_multiline_option(self, parser, given, expected):
+        assert parser._is_multiline_option(given) is expected
+
+    @pytest.mark.parametrize("given, expected", [*case_line_is_comment])
+    def test_line_is_comment(self, parser, given, expected):
+        assert parser._is_comment(given) is expected
+
+    @pytest.mark.parametrize("given, expected", [*case_line_is_empty])
+    def test_line_is_empty(self, parser, given, expected):
+        assert parser._is_empty_line(given) is expected

kiauh/core/submodules/simple_config_parser/tests/features/public_api/test_public_api.py

@@ -0,0 +1,196 @@
+import pytest
+
+from src.simple_config_parser.simple_config_parser import (
+    DuplicateSectionError,
+    NoOptionError,
+    NoSectionError,
+    SimpleConfigParser,
+)
+
+
+@pytest.fixture
+def parser():
+    return SimpleConfigParser()
+
+
+class TestPublicAPI:
+    def test_has_section(self, parser):
+        parser._all_sections = ["section1"]
+        assert parser.has_section("section1") is True
+
+    @pytest.mark.parametrize("section", ["section1", "section2", "section three"])
+    def test_add_section(self, parser, section):
+        parser.add_section(section)
+
+        assert section in parser._all_sections
+        assert parser._all_options[section] == {}
+
+        cfg_section = {"_raw": f"\n[{section}]\n", "body": []}
+        assert parser._config[section] == cfg_section
+
+    @pytest.mark.parametrize("section", ["section1", "section2", "section three"])
+    def test_add_existing_section(self, parser, section):
+        parser._all_sections = [section]
+
+        with pytest.raises(DuplicateSectionError):
+            parser.add_section(section)
+
+        assert parser._all_sections == [section]
+
+    @pytest.mark.parametrize("section", ["section1", "section2", "section three"])
+    def test_remove_section(self, parser, section):
+        parser.add_section(section)
+        parser.remove_section(section)
+
+        assert section not in parser._all_sections
+        assert section not in parser._all_options
+        assert section not in parser._config
+
+    @pytest.mark.parametrize("section", ["section1", "section2", "section three"])
+    def test_remove_non_existing_section(self, parser, section):
+        with pytest.raises(NoSectionError):
+            parser.remove_section(section)
+
+    def test_get_all_sections(self, parser):
+        parser.add_section("section1")
+        parser.add_section("section2")
+        parser.add_section("section three")
+
+        assert parser.sections() == ["section1", "section2", "section three"]
+
+    def test_has_option(self, parser):
+        parser.add_section("section1")
+        parser.set("section1", "option1", "value1")
+
+        assert parser.has_option("section1", "option1") is True
+
+    @pytest.mark.parametrize(
+        "section, option, value",
+        [
+            ("section1", "option1", "value1"),
+            ("section2", "option2", "value2"),
+            ("section three", "option3", "value three"),
+        ],
+    )
+    def test_set_new_option(self, parser, section, option, value):
+        parser.add_section(section)
+        parser.set(section, option, value)
+
+        assert section in parser._all_sections
+        assert option in parser._all_options[section]
+        assert parser._all_options[section][option] == value
+
+        assert parser._config[section]["body"][0]["is_multiline"] is False
+        assert parser._config[section]["body"][0]["option"] == option
+        assert parser._config[section]["body"][0]["value"] == value
+        assert parser._config[section]["body"][0]["_raw"] == f"{option}: {value}\n"
+
+    def test_set_existing_option(self, parser):
+        section, option, value1, value2 = "section1", "option1", "value1", "value2"
+
+        parser.add_section(section)
+        parser.set(section, option, value1)
+        parser.set(section, option, value2)
+
+        assert parser._all_options[section][option] == value2
+        assert parser._config[section]["body"][0]["is_multiline"] is False
+        assert parser._config[section]["body"][0]["option"] == option
+        assert parser._config[section]["body"][0]["value"] == value2
+        assert parser._config[section]["body"][0]["_raw"] == f"{option}: {value2}\n"
+
+    def test_set_new_multiline_option(self, parser):
+        section, option, value = "section1", "option1", "value1\nvalue2\nvalue3"
+
+        parser.add_section(section)
+        parser.set(section, option, value)
+
+        assert parser._config[section]["body"][0]["is_multiline"] is True
+        assert parser._config[section]["body"][0]["option"] == option
+
+        values = ["value1", "value2", "value3"]
+        raw_values = ["  value1\n", "  value2\n", "  value3\n"]
+        assert parser._config[section]["body"][0]["value"] == values
+        assert parser._config[section]["body"][0]["_raw"] == f"{option}:\n"
+        assert parser._config[section]["body"][0]["_raw_value"] == raw_values
+        assert parser._all_options[section][option] == values
+
+    def test_set_option_of_non_existing_section(self, parser):
+        with pytest.raises(NoSectionError):
+            parser.set("section1", "option1", "value1")
+
+    def test_remove_option(self, parser):
+        section, option, value = "section1", "option1", "value1"
+
+        parser.add_section(section)
+        parser.set(section, option, value)
+        parser.remove_option(section, option)
+
+        assert option not in parser._all_options[section]
+        assert option not in parser._config[section]["body"]
+
+    def test_remove_non_existing_option(self, parser):
+        parser.add_section("section1")
+        with pytest.raises(NoOptionError):
+            parser.remove_option("section1", "option1")
+
+    def test_remove_option_of_non_existing_section(self, parser):
+        with pytest.raises(NoSectionError):
+            parser.remove_option("section1", "option1")
+
+    def test_get_option(self, parser):
+        parser.add_section("section1")
+        parser.add_section("section2")
+        parser.set("section1", "option1", "value1")
+        parser.set("section2", "option2", "value2")
+        parser.set("section2", "option3", "value two")
+
+        assert parser.get("section1", "option1") == "value1"
+        assert parser.get("section2", "option2") == "value2"
+        assert parser.get("section2", "option3") == "value two"
+
+    def test_get_option_of_non_existing_section(self, parser):
+        with pytest.raises(NoSectionError):
+            parser.get("section1", "option1")
+
+    def test_get_option_of_non_existing_option(self, parser):
+        parser.add_section("section1")
+        with pytest.raises(NoOptionError):
+            parser.get("section1", "option1")
+
+    def test_get_option_fallback(self, parser):
+        parser.add_section("section1")
+        assert parser.get("section1", "option1", "fallback_value") == "fallback_value"
+
+    def test_get_options(self, parser):
+        parser.add_section("section1")
+        parser.set("section1", "option1", "value1")
+        parser.set("section1", "option2", "value2")
+        parser.set("section1", "option3", "value3")
+
+        options = {"option1": "value1", "option2": "value2", "option3": "value3"}
+        assert parser.options("section1") == options
+
+    def test_get_option_as_int(self, parser):
+        parser.add_section("section1")
+        parser.set("section1", "option1", "1")
+
+        option = parser.getint("section1", "option1")
+        assert isinstance(option, int) is True
+
+    def test_get_option_as_float(self, parser):
+        parser.add_section("section1")
+        parser.set("section1", "option1", "1.234")
+
+        option = parser.getfloat("section1", "option1")
+        assert isinstance(option, float) is True
+
+    @pytest.mark.parametrize(
+        "value",
+        ["True", "true", "on", "1", "yes", "False", "false", "off", "0", "no"],
+    )
+    def test_get_option_as_boolean(self, parser, value):
+        parser.add_section("section1")
+        parser.set("section1", "option1", value)
+
+        option = parser.getboolean("section1", "option1")
+        assert isinstance(option, bool) is True