# ======================================================================= #
#  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
from pathlib import Path
from typing import List

from core.constants import SYSTEMD
from core.instance_manager.base_instance import SUFFIX_BLACKLIST
from core.instance_type import InstanceType


def get_instances(instance_type: type) -> List[InstanceType]:
    from utils.common import convert_camelcase_to_kebabcase

    if not isinstance(instance_type, type):
        raise ValueError("instance_type must be a class")

    name = convert_camelcase_to_kebabcase(instance_type.__name__)
    pattern = re.compile(f"^{name}(-[0-9a-zA-Z]+)?.service$")

    service_list = [
        Path(SYSTEMD, service)
        for service in SYSTEMD.iterdir()
        if pattern.search(service.name)
        and not any(s in service.name for s in SUFFIX_BLACKLIST)
    ]

    instance_list = [
        instance_type(get_instance_suffix(name, service)) for service in service_list
    ]

    def _sort_instance_list(suffix: int | str | None):
        if suffix is None:
            return
        elif isinstance(suffix, str) and suffix.isdigit():
            return f"{int(suffix):04}"
        else:
            return suffix

    return sorted(instance_list, key=lambda x: _sort_instance_list(x.suffix))


def get_instance_suffix(name: str, file_path: Path) -> str:
    # to get the suffix of the instance, we remove the name of the instance from
    # the file name, if the remaining part an empty string we return it
    # otherwise there is and hyphen left, and we return the part after the hyphen
    suffix = file_path.stem[len(name) :]
    return suffix[1:] if suffix else ""