MonkeyBusiness/modules/__init__.py

from importlib import util
from os import path
from glob import glob

from fastapi import APIRouter, Request
from typing import Optional

routers = []
for module_path in [
    f
    for f in glob(path.join(path.dirname(__file__), "**/*.py"), recursive=True)
    if path.basename(f) != "__init__.py"
]:
    spec = util.spec_from_file_location("", module_path)
    module = util.module_from_spec(spec)
    spec.loader.exec_module(module)

    router = getattr(module, "router", None)
    if router is not None:
        routers.append(router)

    if path.basename(module_path) != "api.py":
        for obj in dir(module):
            globals()[obj] = module.__dict__[obj]

router = APIRouter(tags=["slashless_forwarder"])


@router.post("/fwdr")
async def forward_slashless(
    request: Request,
    model: Optional[str] = None,
    f: Optional[str] = None,
    module: Optional[str] = None,
    method: Optional[str] = None,
):
    if f != None:
        module, method = f.split(".")

    try:
        find_response = globals()[f"{module}_{method}".lower()]
        return await find_response(request)
    except KeyError:
        try:
            game_code = model.split(":")[0]
            # TODO: check for more edge cases
            if game_code == "MDX" and module == "eventlog" or module == "eventlog_2":
                find_response = globals()[f"ddr_{module}_{method}"]
            elif game_code == "REC":
                find_response = globals()[f"drs_{module}_{method}"]
            elif game_code == "KFC" and module == "eventlog":
                find_response = globals()[f"sdvx_{module}_{method}"]
            elif game_code == "M32":
                if module == "lobby":
                    find_response = globals()[f"gitadora_{module}_{method}"]
                else:
                    gd_module = module.split("_")
                    find_response = globals()[f"gitadora_{gd_module[-1]}_{method}"]
                    return await find_response(gd_module[0], request)
            return await find_response(request)
        except (KeyError, UnboundLocalError):
            print("Try URL Slash 1 (On) if this game is supported.")
            return Response(status_code=404)


routers.append(router)