mirror of
https://github.com/dw-0/kiauh.git
synced 2025-12-23 15:53:36 +05:00
135 lines
5.2 KiB
Python
135 lines
5.2 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# ======================================================================= #
|
|
# 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 Dict, Literal, List, Union
|
|
|
|
from kiauh.core.config_manager.config_manager import ConfigManager
|
|
from kiauh.core.instance_manager.instance_manager import InstanceManager
|
|
from kiauh.core.repo_manager.repo_manager import RepoManager
|
|
from kiauh.modules.mainsail import MAINSAIL_DIR
|
|
from kiauh.modules.mainsail.mainsail_utils import enable_mainsail_remotemode
|
|
from kiauh.modules.moonraker import (
|
|
DEFAULT_MOONRAKER_PORT,
|
|
MODULE_PATH,
|
|
MOONRAKER_DIR,
|
|
MOONRAKER_ENV_DIR,
|
|
)
|
|
from kiauh.modules.moonraker.moonraker import Moonraker
|
|
from kiauh.utils.common import get_install_status_common
|
|
from kiauh.utils.logger import Logger
|
|
from kiauh.utils.system_utils import (
|
|
get_ipv4_addr,
|
|
)
|
|
|
|
|
|
def get_moonraker_status() -> (
|
|
Dict[
|
|
Literal["status", "status_code", "instances", "repo", "local", "remote"],
|
|
Union[str, int],
|
|
]
|
|
):
|
|
status = get_install_status_common(Moonraker, MOONRAKER_DIR, MOONRAKER_ENV_DIR)
|
|
return {
|
|
"status": status.get("status"),
|
|
"status_code": status.get("status_code"),
|
|
"instances": status.get("instances"),
|
|
"repo": RepoManager.get_repo_name(MOONRAKER_DIR),
|
|
"local": RepoManager.get_local_commit(MOONRAKER_DIR),
|
|
"remote": RepoManager.get_remote_commit(MOONRAKER_DIR),
|
|
}
|
|
|
|
|
|
def create_example_moonraker_conf(
|
|
instance: Moonraker, ports_map: Dict[str, int]
|
|
) -> None:
|
|
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
|
|
|
|
source = MODULE_PATH.joinpath("res/moonraker.conf")
|
|
target = instance.cfg_file
|
|
try:
|
|
shutil.copy(source, target)
|
|
except OSError as e:
|
|
Logger.print_error(f"Unable to create example moonraker.conf:\n{e}")
|
|
return
|
|
|
|
ports = [
|
|
ports_map.get(instance)
|
|
for instance in ports_map
|
|
if ports_map.get(instance) is not None
|
|
]
|
|
if ports_map.get(instance.suffix) is None:
|
|
# this could be improved to not increment the max value of the ports list and assign it as the port
|
|
# as it can lead to situation where the port for e.g. instance moonraker-2 becomes 7128 if the port
|
|
# 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 DEFAULT_MOONRAKER_PORT
|
|
else:
|
|
port = ports_map.get(instance.suffix)
|
|
|
|
ports_map[instance.suffix] = port
|
|
|
|
ip = get_ipv4_addr().split(".")[:2]
|
|
ip.extend(["0", "0/16"])
|
|
uds = instance.comms_dir.joinpath("klippy.sock")
|
|
|
|
cm = ConfigManager(target)
|
|
trusted_clients = f"\n{'.'.join(ip)}"
|
|
trusted_clients += cm.get_value("authorization", "trusted_clients")
|
|
|
|
cm.set_value("server", "port", str(port))
|
|
cm.set_value("server", "klippy_uds_address", str(uds))
|
|
cm.set_value("authorization", "trusted_clients", trusted_clients)
|
|
|
|
cm.write_config()
|
|
Logger.print_ok(f"Example moonraker.conf created in '{instance.cfg_dir}'")
|
|
|
|
|
|
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 klipper instance with the new name
|
|
im.current_instance = Moonraker(suffix=new_name)
|
|
# 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 MAINSAIL_DIR.exists() and len(im.instances) > 1:
|
|
enable_mainsail_remotemode()
|