#!/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)