summary refs log tree commit diff
path: root/main.py
diff options
context:
space:
mode:
Diffstat (limited to 'main.py')
-rw-r--r--main.py18
1 files changed, 15 insertions, 3 deletions
diff --git a/main.py b/main.py
index 8aad57f..48cdb1c 100644
--- a/main.py
+++ b/main.py
@@ -9,11 +9,22 @@ from passlib.hash import scrypt
 import db

 import uuid

 import secrets

+import requests

 import time

 from urllib.parse import urlparse

 

+with open("config.json", "r") as f:

+    config_file = json.load(f)

+

 version = "Helium-1.0.0a"

-attachment_whitelist = ["u.cubeupload.com", "files.catbox.moe", "litter.catbox.moe", "i.ibb.co", "cubeupload.com", "media.tenor.com", "tenor.com", "c.tenor.com", "meower.fraudulent.loan", "fraudulent.loan", "deer.fraudulent.loan"]

+attachment_whitelist = config_file["whitelisted_urls"]

+

+contributors = []

+if config_file["gitlab_contributors_url"]:

+    r = requests.get(config_file["gitlab_contributors_url"])

+    if r.status_code == 200:

+        for i in r.json():

+            contributors.append(i["name"])

 

 addr = "localhost"

 port = 3636

@@ -85,7 +96,8 @@ class util:
             "version": version,

             "ulist": ulist,

             "messages": db.posts.get_recent(),

-            "locked": locked

+            "locked": locked,

+            "server_contributors": contributors

         })

     

     def ulist():

@@ -162,7 +174,7 @@ async def handler(websocket):
             if not ratelimits[str(websocket.id)] <= time.time():

                 await websocket.send(util.error("ratelimited", None))

                 continue

-            ratelimits[str(websocket.id)] = time.time() + 0.15

+            ratelimits[str(websocket.id)] = time.time() + 0.025

             try:

                 r = json.loads(message)

             except: