initial commit

Dockerfile

@@ -0,0 +1,9 @@
+FROM python:alpine3.11
+
+COPY requirements.txt .
+
+RUN pip install -r requirements.txt
+
+COPY . .
+
+CMD ["python", "./main.py"]

main.py

@@ -0,0 +1,59 @@
+import datetime
+from json import dumps, loads
+from flask import Flask, request
+import redis
+import base64
+import re
+import requests
+
+app = Flask(__name__)
+DEV = False
+DB_KEY = "scores" if not DEV else "scores_dev"
+WEBHOOK_URL = "https://discord.com/api/webhooks/1233484325962518709/yUxyEYZzfxOvOhd3G892pSd4XHbwdj96UnLmL9oJXeimc7sqMfUPkGhs0VA-k34o-56J"
+
+db = redis.Redis(host='redis',
+                 port=50658,
+                 decode_responses=True,
+                 ssl=True,
+                 username="default", 
+                 password="default") #! probably fine?
+
+def seconds2mmss(total_seconds: float):
+	seconds: float = total_seconds % 60
+	minutes: int   =  int(total_seconds / 60.0) % 60
+	return "%02d:%05.2f" % (minutes, seconds)
+
+@app.route("/leaderboard", methods=["POST"])
+def submit_score():
+    data = request.args.get('score')
+    print(data)
+    if not data.startswith("EGH"):
+        return "Decryption Failure!", 403
+    
+    score = base64.urlsafe_b64decode(data[3:].split("eHa")[0]).decode()
+    uname = base64.urlsafe_b64decode(data.split("eHa")[1].split("lAx")[0]).decode()
+    wave = base64.urlsafe_b64decode(data.split("eHa")[1].split("lAx")[1]).decode()
+
+    if not re.match("^[A-Z]{3}$", uname):
+        return "Invalid username", 405
+    db.zadd(DB_KEY, {(uname + "|" + score + "|" + wave): int(wave) * 1e12 - float(score)})
+    
+    scores = [x.split("|") for x in db.zrevrange(DB_KEY, start=0, end=50)]
+
+    position = db.zrevrank(DB_KEY, uname + "|" + score + "|" + wave) + 1
+    print(position)
+    if position == 1:
+        print("WR ALERT!!!!")
+        if DEV:
+             return
+        requests.post(WEBHOOK_URL, json={"content": f":warning: <@&1234952370387943517> WR ALERT :warning:\n`{uname.upper()}` set a new WR by reaching wave {wave} in `{seconds2mmss(float(score))}`"})
+
+    if position > 50:
+        position /= db.zcount(DB_KEY, "-inf", "inf")
+
+    print(scores)
+
+    return dumps({
+        "scores": scores,
+        "position": position
+    })

requirements.txt

@@ -0,0 +1,4 @@
+flask
+dotenv
+redis
+requests