feat: cluster client

test.py

@@ -0,0 +1,90 @@
+#!/bin/env /bin/python
+#  mCaptcha - A proof of work based DoS protection system
+#  Copyright © 2023 Aravinth Manivannan <realravinth@batsense.net>
+# 
+#  This program is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU Affero General Public License as
+#  published by the Free Software Foundation, either version 3 of the
+#  License, or (at your option) any later version.
+# 
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU Affero General Public License for more details.
+# 
+#  You should have received a copy of the GNU Affero General Public License
+#  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from pprint import pprint
+import requests
+
+
+def init(host: str):
+    resp = requests.post(f"http://{host}/init")
+    print(f"Initialization status: {resp.status_code}")
+
+
+def add_host(host: str, id: int, peer: str):
+    params = [id, peer]
+    resp = requests.post(f"http://{host}/add-learner", json=params)
+    print(f"Adding host {peer}. Status: {resp.status_code}")
+
+
+def switch_to_cluster(host: str, nodes: [int]):
+    resp = requests.post(f"http://{host}/change-membership", json=nodes)
+    print(f"Switching to cluster. Status: {resp.status_code}")
+
+
+def metrics(host: str):
+    resp = requests.get(f"http://{host}/metrics")
+    data = resp.json()
+    pprint(data)
+
+
+def write(host, data):
+    resp = requests.post(f"http://{host}/write", json=data)
+    print(f"RPC Status: {resp.status_code}")
+    data = resp.json()
+    if "Err" in data:
+        leader = data["Err"]["APIError"]["ForwardToLeader"]["leader_node"]["addr"]
+        write(leader, data)
+    return data['Ok']['data']
+
+
+def add_vote(host: str, captcha_id: str):
+    resp = write(host, data={"AddVisitor": captcha_id})
+    pprint(resp)
+
+
+def add_captcha(host: str, captcha_id: str):
+    params = {
+        "AddCaptcha": {
+            "id": captcha_id,
+            "mcaptcha": {
+                "visitor_threshold": 0,
+                "defense": {
+                    "levels": [
+                        {"visitor_threshold": 50, "difficulty_factor": 500},
+                        {"visitor_threshold": 5000, "difficulty_factor": 50000},
+                    ],
+                    "current_visitor_threshold": 0,
+                },
+                "duration": 30,
+            },
+        }
+    }
+    resp = write(host, data=params)
+    pprint(f"Captcha added {captcha_id}: {resp}")
+
+
+if __name__ == "__main__":
+    host = "localhost:9001"
+    #    init(host)
+    #    add_host(host, id=2, peer="localhost:9002")
+    #    switch_to_cluster(host, nodes=[1, 2])
+    #    metrics(host)
+    captcha_id = "test_1"
+    add_captcha(host, captcha_id)
+    add_vote(host, captcha_id)
+    for _ in range(0,600):
+        add_vote(host, captcha_id)