mirror of
https://github.com/dw-0/kiauh.git
synced 2025-12-14 11:04:29 +05:00
113 lines
4.2 KiB
Python
113 lines
4.2 KiB
Python
# ======================================================================= #
|
|
# 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 typing import List
|
|
|
|
from components.klipper.klipper import Klipper
|
|
from components.moonraker.moonraker import Moonraker
|
|
from components.webui_client.base_data import (
|
|
BaseWebClient,
|
|
)
|
|
from components.webui_client.client_config.client_config_remove import (
|
|
run_client_config_removal,
|
|
)
|
|
from core.backup_manager.backup_manager import BackupManager
|
|
from core.constants import NGINX_SITES_AVAILABLE, NGINX_SITES_ENABLED
|
|
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,
|
|
run_remove_routines,
|
|
)
|
|
from utils.instance_utils import get_instances
|
|
|
|
|
|
def run_client_removal(
|
|
client: BaseWebClient,
|
|
remove_client: bool,
|
|
remove_client_cfg: bool,
|
|
backup_config: bool,
|
|
) -> Message:
|
|
completion_msg = Message(
|
|
title=f"{client.display_name} Removal Process completed",
|
|
color=Color.GREEN,
|
|
)
|
|
mr_instances: List[Moonraker] = get_instances(Moonraker)
|
|
kl_instances: List[Klipper] = get_instances(Klipper)
|
|
|
|
if backup_config:
|
|
bm = BackupManager()
|
|
if bm.backup_file(client.config_file):
|
|
completion_msg.text.append(f"● {client.config_file.name} backup created")
|
|
|
|
if remove_client:
|
|
client_name = client.name
|
|
if remove_client_dir(client):
|
|
completion_msg.text.append(f"● {client.display_name} removed")
|
|
if remove_client_nginx_config(client_name):
|
|
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(
|
|
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:
|
|
cfg_completion_msg = 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:
|
|
Logger.print_status(f"Removing {client.display_name} ...")
|
|
return run_remove_routines(client.client_dir)
|
|
|
|
|
|
def remove_client_nginx_config(name: str) -> bool:
|
|
Logger.print_status(f"Removing NGINX config for {name.capitalize()} ...")
|
|
return remove_with_sudo(
|
|
[
|
|
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)
|