CustomIWMServer/customiwmserver/database.py

from datetime import datetime
from typing import Literal
from pymongo import MongoClient


client = MongoClient("mongodb://root:catboys@mongo:27017")


db = client["IWM_CustomServer_DB"]

user_collection = db.users
maps_collection = db.maps
reports_collection = db.reports

general_collection = db.general
admin_log_collection = db.admin_log


def LogAdminAction(
    action_type: str, action_data: dict, UserID: int = None, success: bool = True
):
    """Log administrator action."""
    admin_log_collection.insert_one(
        {
            "date": datetime.utcnow(),
            "action_type": action_type,
            "action_data": action_data,
        }
    )


def auth_check(Authorization) -> (tuple[Literal[False], Literal["noauth"]] | tuple[Literal[True], dict]):
    """Checks credentials.
    Returns a tuple with result (for example False, "nouser").

    Results:
        - False if wrong username or password 
        - True, [dict] if correct
    """
    # FIXME: This function currently DOES NOT perform any authentication.
    # This means that ANYONE knowing the username could perform actions as the user.
    if Authorization is None:
        return False, "noauth"
    username, password = Authorization.split(":")

    query = user_collection.find_one({"Username": username})
    if not query:
        return False, "noauth"

    # if query["Password"] != password:
    #     return False, "wrongpass"

    return True, query



def id_to_mapcode(id_):
    return hex(id_).replace("0x", "").rjust(8, "0")[0:8].upper()