update_manager: replace string choices with enums

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>

moonraker/components/update_manager/common.py

@@ -0,0 +1,102 @@
+# Moonraker/Klipper update configuration
+#
+# Copyright (C) 2022  Eric Callahan <arksine.code@gmail.com>
+#
+# This file may be distributed under the terms of the GNU GPLv3 license.
+
+from __future__ import annotations
+import os
+import sys
+import copy
+import pathlib
+from enum import Enum
+from ...utils import source_info
+from typing import (
+    TYPE_CHECKING,
+    Dict,
+    Union
+)
+
+if TYPE_CHECKING:
+    from ...confighelper import ConfigHelper
+    from ..database import MoonrakerDatabase
+
+KLIPPER_DEFAULT_PATH = os.path.expanduser("~/klipper")
+KLIPPER_DEFAULT_EXEC = os.path.expanduser("~/klippy-env/bin/python")
+
+BASE_CONFIG: Dict[str, Dict[str, str]] = {
+    "moonraker": {
+        "origin": "https://github.com/arksine/moonraker.git",
+        "requirements": "scripts/moonraker-requirements.txt",
+        "venv_args": "-p python3",
+        "system_dependencies": "scripts/system-dependencies.json",
+        "host_repo": "arksine/moonraker",
+        "virtualenv": sys.exec_prefix,
+        "path": str(source_info.source_path()),
+        "managed_services": "moonraker"
+    },
+    "klipper": {
+        "moved_origin": "https://github.com/kevinoconnor/klipper.git",
+        "origin": "https://github.com/Klipper3d/klipper.git",
+        "requirements": "scripts/klippy-requirements.txt",
+        "venv_args": "-p python2",
+        "install_script": "scripts/install-octopi.sh",
+        "host_repo": "arksine/moonraker",
+        "managed_services": "klipper"
+    }
+}
+
+class ExtEnum(Enum):
+    @classmethod
+    def from_string(cls, enum_name: str):
+        str_name = enum_name.upper()
+        for name, member in cls.__members__.items():
+            if name == str_name:
+                return cls(member.value)
+        raise ValueError(f"No enum member named {enum_name}")
+
+    def __str__(self) -> str:
+        return self._name_.lower()  # type: ignore
+
+class AppType(ExtEnum):
+    NONE = 1
+    WEB = 2
+    GIT_REPO = 3
+    ZIP = 4
+
+class Channel(ExtEnum):
+    STABLE = 1
+    BETA = 2
+    DEV = 3
+
+def get_app_type(app_path: Union[str, pathlib.Path]) -> AppType:
+    if isinstance(app_path, str):
+        app_path = pathlib.Path(app_path).expanduser()
+    # None type will perform checks on Moonraker
+    if source_info.is_git_repo(app_path):
+        return AppType.GIT_REPO
+    else:
+        return AppType.NONE
+
+def get_base_configuration(config: ConfigHelper) -> ConfigHelper:
+    server = config.get_server()
+    base_cfg = copy.deepcopy(BASE_CONFIG)
+    base_cfg["moonraker"]["type"] = str(get_app_type(source_info.source_path()))
+    db: MoonrakerDatabase = server.lookup_component('database')
+    base_cfg["klipper"]["path"] = db.get_item(
+        "moonraker", "update_manager.klipper_path", KLIPPER_DEFAULT_PATH
+    ).result()
+    base_cfg["klipper"]["env"] = db.get_item(
+        "moonraker", "update_manager.klipper_exec", KLIPPER_DEFAULT_EXEC
+    ).result()
+    base_cfg["klipper"]["type"] = str(get_app_type(base_cfg["klipper"]["path"]))
+    channel = config.get("channel", "dev")
+    base_cfg["moonraker"]["channel"] = channel
+    base_cfg["klipper"]["channel"] = channel
+    if config.has_section("update_manager moonraker"):
+        mcfg = config["update_manager moonraker"]
+        base_cfg["moonraker"]["channel"] = mcfg.get("channel", channel)
+    if config.has_section("update_manager klipper"):
+        kcfg = config["update_manager klipper"]
+        base_cfg["klipper"]["channel"] = kcfg.get("channel", channel)
+    return config.read_supplemental_dict(base_cfg)