1 files changed, 84 insertions, 0 deletions
diff --git a/south_browser/rpc.py b/south_browser/rpc.py
new file mode 100644
index 0000000..fe76829
--- /dev/null
+++ b/south_browser/rpc.py
@@ -0,0 +1,84 @@
+import json
+
+from requests import post
+from time import time, sleep
+
+_PAYLOAD_ID = 0
+LOCAL_RPC = "http://localhost:26503/rpc"
+
+def cmd(method: str, params = None):
+    global _PAYLOAD_ID
+    _PAYLOAD_ID += 1
+
+    payload = {
+        "jsonrpc": "2.0",
+        "id": int(time()),
+        "method": method
+    }
+
+    if params is not None:
+        payload["params"] = params
+
+    r = post(LOCAL_RPC, data=json.dumps(payload))
+
+    if r.status_code != 200:
+        raise Exception(r.text)
+
+    data = r.json()
+    return data
+
+def NSRequestServerList():
+    return cmd("execute_squirrel", {
+        "code": "return NSRequestServerList()"
+    })["result"]
+
+def NSGetServerCount(repeat: bool = True):
+    data = cmd("execute_squirrel", {
+        "code": "return NSGetServerCount()"
+    })
+
+    c = data["result"]
+
+    if not c and repeat:
+        NSRequestServerList()
+        sleep(1) # give it a second to popualte
+        return NSGetServerCount(False)
+
+    return c
+
+def NSGetGameServers():
+    # work around Squirrel closure bullshit
+    server_count = NSGetServerCount()
+
+    for i in range(server_count):
+
+        primret = []
+
+        for c in range(server_count):
+            primret.append(f"[s[{c}].index, s[{c}].id, s[{c}].name, s[{c}].description, s[{c}].map, s[{c}].playlist, s[{c}].playerCount, s[{c}].maxPlayerCount, s[{c}].requiresPassword, s[{c}].region]")
+
+        ret = "[" + ",".join(primret) + "]"
+
+        return cmd("execute_squirrel", {
+            "code": f"array<ServerInfo> s = NSGetGameServers()\nreturn {ret}"
+        })["result"]
+
+    return []
+
+def NSTryAuthWithServer(index: int, password: str = None):
+    if password is None:
+        password = ""
+
+    return cmd("execute_squirrel", {
+        "code": f"return NSTryAuthWithServer({index}, {json.dumps(password)})"
+    })["result"]
+
+def NSWasAuthSuccessful():
+    return cmd("execute_squirrel", {
+        "code": "return NSWasAuthSuccessful()"
+    })["result"]
+
+def NSConnectToAuthedServer():
+    return cmd("execute_squirrel", {
+        "code": "return NSConnectToAuthedServer()"
+    })["result"]