import os
import re
from copy import copy

from vinetrimmer.services.BaseService import BaseService

SERVICE_MAP = {}

for service in os.listdir(os.path.dirname(__file__)):
    if service.startswith("_") or not service.endswith(".py"):
        continue

    service = os.path.splitext(service)[0]

    if service in ("__init__", "BaseService"):
        continue

    with open(os.path.join(os.path.dirname(__file__), f"{service}.py"), encoding="utf-8") as fd:
        code = ""
        for line in fd.readlines():
            if re.match(r"\s*(?:import(?! click)|from)\s", line):
                continue
            code += line
            if re.match(r"\s*super\(\)\.__init__\(", line):
                break
        exec(code)

for x in copy(globals()).values():
    if isinstance(x, type) and issubclass(x, BaseService) and x != BaseService:
        SERVICE_MAP[x.__name__] = x.ALIASES


def get_service_key(value):
    """
    Get the Service Key name (e.g. DisneyPlus, not dsnp, disney+, etc.) from the SERVICE_MAP.
    Input value can be of any case-sensitivity and can be either the key itself or an alias.
    """
    value = value.lower()
    for key, aliases in SERVICE_MAP.items():
        if value in map(str.lower, aliases) or value == key.lower():
            return key