DeathGarden_API_Rebirth/src/endpoints/web.py
2024-07-21 00:25:01 +02:00

679 lines
28 KiB
Python

from flask_definitions import *
from logic.queue_handler import matchmaking_queue
@app.route('/', methods=["GET"])
def index():
check_for_game_client("soft")
return render_template("index.html")
@app.route('/robots.txt', methods=["GET"])
def robots():
return send_from_directory(app.static_folder, request.path[1:])
@app.route('/.well-known/security.txt', methods=["GET"])
def security():
return send_from_directory(app.static_folder, request.path[1:])
@app.route('/.well-known/gpc.json', methods=["GET"])
def gpc():
return send_from_directory(app.static_folder, request.path[1:])
@app.route('/.well-known/dnt-policy.txt', methods=["GET"])
def dnt():
return send_from_directory(app.static_folder, request.path[1:])
@app.route('/favicon.ico')
def favicon():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'favicon.ico',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-favicon", message=e)
@app.route('/apple-touch-icon.png')
def apple_touch_icon():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'apple-touch-icon.png',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-apple_touch_icon", message=e)
@app.route('/android-chrome-192x192.png')
def android_chrome_192x192():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'android-chrome-192x192.png',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-android-chrome-192x192", message=e)
@app.route('/android-chrome-256x256.png')
def android_chrome_256x256():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'android-chrome-256x256.png',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-android-chrome-256x256", message=e)
@app.route('/browserconfig.xml')
def browserconfig():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'browserconfig.xml')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-browserconfig", message=e)
@app.route('/favicon-16x16.png')
def favicon_16x16():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'favicon-16x16.png',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-favicon-16x16", message=e)
@app.route('/favicon-32x32.png')
def favicon_32x32():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'favicon-32x32.png',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-favicon-32x32", message=e)
@app.route('/favicon.png')
def favicon_png():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'favicon.png',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-favicon", message=e)
@app.route('/mstile-150x150.png')
def mstile_150x150():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'mstile-150x150.png',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-mstile-150x150", message=e)
@app.route('/safari-pinned-tab.svg')
def safari_pinned_tab():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'safari-pinned-tab.svg',
mimetype='image/svg+xml')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-safari-pinned-tab", message=e)
@app.route('/site.webmanifest')
def site_webmanifest():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'site.webmanifest')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-site.webmanifest", message=e)
@app.route('/apple-touch-icon-precomposed.png')
def apple_touch_icon_precomposed():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'favicon.png',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-apple-touch-icon-precomposed", message=e)
@app.route('/apple-touch-icon-120x120.png')
def apple_touch_icon_120x120():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'favicon.png',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-apple-touch-icon-120x120", message=e)
@app.route('/images/icons/fugue/tick-circle.png')
def tick_circle():
try:
return send_from_directory(os.path.join(app.root_path, 'image'), 'tick-circle.png',
mimetype='image/vnd.microsoft.icon')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="general-tick-circle-png", message=e)
@app.route("/admin/ban", methods=["POST", "GET"])
def admin_ban():
if request.method == "POST":
login_token = sanitize_input(request.form.get("login_token"))
if login_token is None:
return jsonify({"status": "error", "message": "No api token found"}), 401
if login_token not in allowed_tokens:
ip = check_for_game_client("remote")
logger.graylog_logger(level="error", handler="web_admin_ban", message={"IP": ip, "message": "Invalid api token"})
error_message = "Invalid api token"
return render_template('ban.html', error=error_message)
chosen_date = sanitize_input(request.form.get("datetime"))
ban_reason = sanitize_input(request.form.get("ban_reason"))
steam_id = sanitize_input(request.form.get("steam_id"))
if not chosen_date or not ban_reason or not steam_id:
error_message = "Please fill out all fields."
return render_template('ban.html', error=error_message)
try:
create_epoch, expire_epoch = create_ban_time(chosen_date)
except Exception as e:
logger.graylog_logger(level="error", handler="web_admin_ban", message=e)
error_message = "An Error occurred. Are you sure the date is correct?"
return render_template('ban.html', error=error_message)
ret = mongo.write_data_with_list(login=steam_id,
login_steam=True,
items_dict={"is_banned": True,
"ban_reason": ban_reason,
"ban_start": create_epoch,
"ban_expire": expire_epoch})
if ret:
logger.graylog_logger(level="info", handler="web_admin_ban", message={"IP": check_for_game_client("remote"),
"steamid": steam_id,
"ban_reason": ban_reason,
"ban_start": create_epoch,
"ban_expire": expire_epoch,
"login_token": login_token})
success_message = "User banned successfully."
return render_template('ban.html', success=success_message)
else:
error_message = "An Error occurred. Are you sure the SteamID is correct?"
return render_template('ban.html', error=error_message)
elif request.method == "GET":
return render_template('ban.html')
else:
return abort(405)
@app.route("/debug", methods=["Get"])
def debug_root():
check_for_game_client("soft")
endpoint_descriptions = {
'/debug/user': 'Endpoint to debug user information'}
endpoint_list = []
for endpoint, description in endpoint_descriptions.items():
endpoint_list.append(f'<li>{endpoint}: {description}</li>')
endpoints_html = '<ul>' + ''.join(endpoint_list) + '</ul>'
return f'<h1>Debug Endpoints:</h1>{endpoints_html}'
@app.route('/debug/user/', methods=['GET'], defaults={'steamid_unsanitized': None})
@app.route('/debug/user/<steamid_unsanitized>', methods=['GET'])
def debug_user(steamid_unsanitized):
steamid = sanitize_input(steamid_unsanitized)
if steamid is None:
return render_template('debug/user.html', is_id_set=False, id_not_found=True)
user_data = mongo.get_debug(steamid=steamid)
if user_data is None:
return render_template('debug/user.html', is_id_set=True, id_not_found=False, steam_id=steamid)
return render_template(
'debug/user.html',
is_id_set=True,
is_not_found=False,
steam_id=user_data.get('steamid'),
eula=user_data.get('eula'),
currency_blood_cells=user_data.get('currency_blood_cells'),
currency_ink_cells=user_data.get('currency_ink_cells'),
currency_iron=user_data.get('currency_iron'),
userId=user_data.get('userId'),
account_xp=user_data.get('account_xp'),
prestige_xp=user_data.get('prestige_xp'),
is_banned=user_data.get('is_banned'),
ban_reason=user_data.get('ban_reason'),
ban_start=user_data.get('ban_start'),
ban_expire=user_data.get('ban_expire'),
special_unlocks=user_data.get('special_unlocks'),
finished_challanges=user_data.get('finished_challanges'),
open_challanges=user_data.get('open_challanges'),
unread_msg_ids=user_data.get('unread_msg_ids')
)
@app.route("/debug/mirrors/write", methods=["POST", "GET"])
def debug_mirrors_write():
check_for_game_client("soft")
try:
if request.method == "POST":
try:
api_token = sanitize_input(request.cookies.get("api_token"))
if api_token is None:
return jsonify({"status": "error", "message": "No api token found"}, 401)
if api_token not in allowed_tokens:
return jsonify({"status": "error", "message": "Invalid api token"}), 401
steam_user_id = sanitize_input(request.json.get("steamid"))
data_b = request.json.get("data")
if not data_b:
return jsonify({"status": "error", "message": "No data found."}), 400
if not steam_user_id:
return jsonify({"status": "error", "message": "No Steamid found."}), 400
logger.graylog_logger(level="info", handler="logging_debug_mirror_write",
message={"IP": check_for_game_client("remote"), "steamid": steam_user_id,
"data": data_b})
return_val = mongo.write_data_with_list(login=steam_user_id, login_steam=True, items_dict=data_b)
if return_val is None:
return jsonify({"status": "error", "message": "There was a error on our End. "
"Please try again later."}), 400
return jsonify({"status": "success", "message": "Data written."}), 200
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="logging_debug_mirror_write", message=e)
if request.method == "GET":
return jsonify({"message": "Endpoint not found"}), 404
except Exception as e:
logger.graylog_logger(level="error", handler="web_debug_mirrors", message=e)
@app.route("/debug/mirrors/get", methods=["POST"])
def debug_mirrors_get():
check_for_game_client("soft")
try:
if request.method == "POST":
try:
api_token = sanitize_input(request.cookies.get("api_token"))
if api_token is None:
return jsonify({"status": "error", "message": "No api token found"}, 401)
if api_token not in allowed_tokens:
return jsonify({"status": "error", "message": "Invalid api token"}), 401
steam_user_id = sanitize_input(request.json.get("steamid"))
if not steam_user_id:
return jsonify({"status": "error", "message": "No Steamid found."}), 400
# logger.graylog_logger(level="info", handler="logging_debug_mirror_read",
# message={"IP": check_for_game_client("remote"), "steamid": steam_user_id})
return_val = mongo.get_debug(steam_user_id)
if return_val is None:
return jsonify({"status": "error", "message": "There was a error on our End. "
"Please try again later."}), 400
return_val.pop("_id")
return jsonify({"Data": return_val}), 200
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="logging_debug_mirror_write", message=e)
if request.method == "GET":
return jsonify({"message": "Endpoint not found"}), 404
except Exception as e:
logger.graylog_logger(level="error", handler="web_debug_mirrors", message=e)
@app.route("/debug/mirrors/clear", methods=["POST"])
def debug_mirrors_clear():
check_for_game_client("soft")
try:
if request.method == "POST":
try:
api_token = sanitize_input(request.cookies.get("api_token"))
if api_token is None:
return jsonify({"status": "error", "message": "No api token found"}, 401)
if api_token not in allowed_tokens:
return jsonify({"status": "error", "message": "Invalid api token"}), 401
# logger.graylog_logger(level="info", handler="logging_debug_mirror_read",
# message={"IP": check_for_game_client("remote"), "steamid": steam_user_id})
return_val = matchmaking_queue.clear_queue()
if return_val is None:
return jsonify({"status": "error", "message": "There was a error on our End. "
"Please try again later."}), 400
return jsonify({"status": "success", "message": "Queues Cleared."}), 200
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="logging_debug_mirror_write", message=e)
if request.method == "GET":
return jsonify({"message": "Endpoint not found"}), 404
except Exception as e:
logger.graylog_logger(level="error", handler="web_debug_mirrors", message=e)
@app.route('/updater/', methods=["GET"])
def updater_root():
return render_template("updater.html")
@app.route("/updater/versions", methods=["GET"])
def updater_version():
return jsonify({
"script_version": 3,
"pak_version": 1,
"sig_version": 1
})
@app.route("/updater/files/", methods=["GET"])
def updater_files():
return jsonify({"Error": "Please read the Docs about this Endpoint."}), 400
@app.route("/updater/files/pak/", methods=["GET"])
def updater_pak():
try:
return send_from_directory(os.path.join(app.root_path, 'files'),
'TheExitRebirthBackendAPI-WindowsNoEditor_P.pak')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="web-updater-PAK", message=e)
@app.route("/updater/files/sig/", methods=["GET"])
def updater_sig():
try:
return send_from_directory(os.path.join(app.root_path, 'files'),
'TheExitRebirthBackendAPI-WindowsNoEditor_P.sig')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="web-updater-SIG", message=e)
@app.route("/updater/files/script/", methods=["GET"])
def updater_script():
try:
return send_from_directory(os.path.join(app.root_path, 'files'), 'TheExit-Rebirth-Updater.bat')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="web-updater-script", message=e)
@app.route("/debug/matchmaking", methods=["GET"])
@app.route("/debug/MatchMaking", methods=["GET"])
def debug_matchmaking():
matchmaking_queue.cleanup_queue_players()
len_queue = matchmaking_queue.get_len_of_queue()
len_killed_lobbies = matchmaking_queue.get_len_of_killed_lobbies()
len_queued_runners = matchmaking_queue.get_len_of_queued_runners()
len_queued_hunters = matchmaking_queue.get_len_of_queued_hunters()
len_open_lobbies = matchmaking_queue.get_len_of_open_lobbies()
len_rich_presence = rich_presence_handler.get_presence_length()
current_players_locker_room = rich_presence_handler.get_locker_room_presence()
current_players_arena = rich_presence_handler.get_arena_presence()
lobby_data = matchmaking_queue.get_lobby_data()
if dev_env == "true":
return render_template('debug/matchmaking.html',
len_queue=len_queue,
len_killed_lobbies=len_killed_lobbies,
len_queued_runners=len_queued_runners,
len_queued_hunters=len_queued_hunters,
len_open_lobbies=len_open_lobbies,
lobby_data=lobby_data,
len_rich_presence=len_rich_presence,
current_players_locker_room=current_players_locker_room,
current_players_arena=current_players_arena)
return render_template('debug/anon_matchmaking.html',
len_queue=len_queue,
len_killed_lobbies=len_killed_lobbies,
len_queued_runners=len_queued_runners,
len_queued_hunters=len_queued_hunters,
len_open_lobbies=len_open_lobbies,
lobby_data=lobby_data,
len_rich_presence=len_rich_presence,
current_players_locker_room=current_players_locker_room,
current_players_arena=current_players_arena)
@app.route("/api/debug/MatchMaking", methods=["GET"])
def api_debug_matchmaking():
matchmaking_queue.cleanup_queue_players()
len_queue = matchmaking_queue.get_len_of_queue()
len_killed_lobbies = matchmaking_queue.get_len_of_killed_lobbies()
len_queued_runners = matchmaking_queue.get_len_of_queued_runners()
len_queued_hunters = matchmaking_queue.get_len_of_queued_hunters()
len_open_lobbies = matchmaking_queue.get_len_of_open_lobbies()
len_rich_presence = rich_presence_handler.get_presence_length()
lobby_data = matchmaking_queue.get_lobby_data()
lobby_return_data = []
for item in lobby_data:
# Hunters needs to be changed from STRING to LIST some day
lobby_info = {
"id": item.id,
"Hunters": 1,
"Runners": len(item.nonHosts),
"status": item.status
}
lobby_return_data.append(lobby_info)
return jsonify({
"len_queue": len_queue,
"len_killed_lobbies": len_killed_lobbies,
"len_queued_runners": len_queued_runners,
"len_queued_hunters": len_queued_hunters,
"len_open_lobbies": len_open_lobbies,
"lobby_data": lobby_return_data,
"len_rich_presence": len_rich_presence
})
@app.route("/api/matchmaking/json", methods=["GET"])
def api_matchmaking_json():
try:
matchmaking_queue.cleanup_queue_players()
len_queue = matchmaking_queue.get_len_of_queue()
len_killed_lobbies = matchmaking_queue.get_len_of_killed_lobbies()
len_queued_runners = matchmaking_queue.get_len_of_queued_runners()
len_queued_hunters = matchmaking_queue.get_len_of_queued_hunters()
len_open_lobbies = matchmaking_queue.get_len_of_open_lobbies()
lobby_data = matchmaking_queue.get_lobby_data()
lobby_return_data = []
for item in lobby_data:
# Hunters needs to be changed from STRING to LIST some day
lobby_info = {
"id": item.id,
"Hunters": 1,
"Runners": item.nonHosts,
"status": item.status
}
lobby_return_data.append(lobby_info)
return jsonify({
"len_queue": len_queue,
"len_killed_lobbies": len_killed_lobbies,
"len_queued_runners": len_queued_runners,
"len_queued_hunters": len_queued_hunters,
"len_open_lobbies": len_open_lobbies,
"lobby_data": lobby_return_data
})
except Exception as e:
logger.graylog_logger(level="error", handler="web-api-matchmaking-json", message=e)
return jsonify({"status": "error"}), 500
@app.route("/en/eula", methods=["GET"])
def eula():
check_for_game_client("soft")
try:
return render_template("dg_eula.html")
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="web-eula", message=e)
@app.route("/js/manifest.js", methods=["GET"])
def manifest():
check_for_game_client("soft")
try:
return send_from_directory(os.path.join(app.root_path, 'static', 'dg'), 'manifest.js')
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="web-manifest", message=e)
@app.route("/js/vendor.js", methods=["GET"])
def vendor():
return jsonify({"message": "This file has sadly been lost to time..."}), 404
@app.route("/js/app.js", methods=["GET"])
def app_js():
return jsonify({"message": "This file has sadly been lost to time..."}), 404
@app.route("/css/app.css", methods=["GET"])
def app_css():
return jsonify({"message": "This file has sadly been lost to time..."}), 404
@app.route("/api/sha", methods=["GET"])
def api_sha():
try:
hashes = hash_handler.get_hash()
return jsonify({"pak": hashes[0], "sig": hashes[1]})
except Exception as e:
logger.graylog_logger(level="error", handler="web-api-sha", message=e)
return jsonify({"status": "error"}), 500
@app.route("/api/v1/upload", methods=["POST", "GET"])
def upload():
check_for_game_client("soft")
try:
return jsonify({"status": "error", "message": "ENDPOINT WIP"}), 404
if request.method == "POST":
try:
api_token = sanitize_input(request.headers.get("api_token"))
if api_token is None:
return jsonify({"status": "error", "message": "No api token found"}), 401
if api_token not in allowed_tokens:
return jsonify({"status": "error", "message": "Invalid api token"}), 401
file = request.files['file']
user_id = sanitize_input("test")
if not user_id:
return jsonify({"status": "error", "message": "No User ID found."}), 400
if not file:
return jsonify({"status": "error", "message": "No file found."}), 400
file_id = file_handler.create_file(file, user_id)
return jsonify({"status": "success", "file_id": file_id}), 200
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="web-upload", message=e)
if request.method == "GET":
return render_template("debug/upload.html")
except Exception as e:
logger.graylog_logger(level="error", handler="web-upload", message=e)
@app.route("/api/v1/download/<file_id>", methods=["GET"])
def download(file_id):
check_for_game_client("soft")
try:
#file = file_handler.get_file(file_id)
file = None
if file is None:
return jsonify({"status": "error", "message": "File not found."}), 404
return file
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="web-download", message=e)
@app.route("/api/v1/metadata/<file_id>", methods=["GET"])
def metadata(file_id):
check_for_game_client("soft")
try:
#metadata = file_handler.get_metadata(file_id)
metadata_data = None
if metadata_data is None:
return jsonify({"status": "error", "message": "File not found."}), 404
return jsonify(metadata_data)
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="web-metadata", message=e)
@app.route("/api/v1/sha256/<file_id>", methods=["GET"])
def sha256(file_id):
check_for_game_client("soft")
try:
#sha256 = file_handler.get_file_sha256(file_id)
sha256_data = None
if sha256_data is None:
return jsonify({"status": "error", "message": "File not found."}), 404
return jsonify({"sha256": sha256_data})
except TimeoutError:
return jsonify({"status": "error"})
except Exception as e:
logger.graylog_logger(level="error", handler="web-sha256", message=e)