Update protocols

DiscordGSM · Mar 15, 2024 · f1dd56f · f1dd56f
1 parent 72df2f2
commit f1dd56f
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 4 deletions.
diff --git a/discordgsm/protocols/beammp.py b/discordgsm/protocols/beammp.py
@@ -1,3 +1,4 @@
+import os
 import re
 import time
 from typing import TYPE_CHECKING
@@ -17,7 +18,9 @@ class BeamMP(Protocol):
     async def query(self):
         host, port = str(self.kv["host"]), int(str(self.kv["port"]))
         ip = await Socket.gethostbyname(host)
-        url = f"https://master-server.opengsq.com/beammp/search?host={ip}&port={port}"
+
+        base_url = os.getenv('OPENGSQ_MASTER_SERVER_URL', 'https://master-server.opengsq.com/').rstrip('/')
+        url = f"{base_url}/scum/search?host={ip}&port={port}"
         start = time.time()
 
         async with aiohttp.ClientSession() as session:

diff --git a/discordgsm/protocols/factorio.py b/discordgsm/protocols/factorio.py
@@ -1,3 +1,4 @@
+import os
 import re
 import time
 from typing import TYPE_CHECKING
@@ -65,7 +66,8 @@ async def query(self):
             data = dict(Factorio.master_servers[host_address])
             ping = 0
         else:
-            url = f"https://master-server.opengsq.com/factorio/search?host={ip}&port={port}"
+            base_url = os.getenv('OPENGSQ_MASTER_SERVER_URL', 'https://master-server.opengsq.com/').rstrip('/')
+            url = f"{base_url}/scum/search?host={ip}&port={port}"
             start = time.time()
 
             async with aiohttp.ClientSession() as session:

diff --git a/discordgsm/protocols/palworld.py b/discordgsm/protocols/palworld.py
@@ -1,3 +1,4 @@
+import os
 import time
 from typing import TYPE_CHECKING
 
@@ -16,7 +17,9 @@ class Palworld(Protocol):
     async def query(self):
         host, port = str(self.kv["host"]), int(str(self.kv["port"]))
         ip = await Socket.gethostbyname(host)
-        url = f"https://master-server.opengsq.com/palworld/search?host={ip}&port={port}"
+
+        base_url = os.getenv('OPENGSQ_MASTER_SERVER_URL', 'https://master-server.opengsq.com/').rstrip('/')
+        url = f"{base_url}/scum/search?host={ip}&port={port}"
         start = time.time()
 
         async with aiohttp.ClientSession() as session:

diff --git a/discordgsm/protocols/scum.py b/discordgsm/protocols/scum.py
@@ -1,3 +1,4 @@
+import os
 import time
 from typing import TYPE_CHECKING
 
@@ -17,7 +18,9 @@ class Scum(Protocol):
     async def query(self):
         host, port = str(self.kv["host"]), int(str(self.kv["port"]))
         ip = await Socket.gethostbyname(host)
-        url = f"https://master-server.opengsq.com/scum/search?host={ip}&port={port}"
+
+        base_url = os.getenv('OPENGSQ_MASTER_SERVER_URL', 'https://master-server.opengsq.com/').rstrip('/')
+        url = f"{base_url}/scum/search?host={ip}&port={port}"
         start = time.time()
 
         async with aiohttp.ClientSession() as session: