58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
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()
|