import random
import pytest
import jsonschema
import json
import time
from conftest import option, user_1, user_2
from test_cases import RpcTestCase, TransactionTestCase
from dataclasses import dataclass, field
from typing import List, Optional


@pytest.mark.rpc
@pytest.mark.wakuext
class TestRpc(RpcTestCase):

    @pytest.mark.parametrize(
        "method, params",
        [
            ("wakuext_peers", []),
        ],
    )
    def test_(self, method, params):
        _id = str(random.randint(1, 8888))

        response = self.rpc_request(method, params, _id, url=option.rpc_url_2)
        self.verify_is_valid_json_rpc_response(response)
        self.verify_json_schema(response, method)


@pytest.mark.rpc
@pytest.mark.accounts
@pytest.mark.wakuext
class TestRpcMessaging(RpcTestCase):

    @dataclass
    class User:
        rpc_url: str
        chat_public_key: Optional[str] = None
        chat_id: Optional[str] = None

    def test_add_contact(self):
        _id = str(random.randint(1, 8888))

        self.user_1 = self.User(rpc_url=option.rpc_url)
        self.user_2 = self.User(rpc_url=option.rpc_url_2)

        # get chat public key
        for user in self.user_1, self.user_2:
            response = self.rpc_request(
                "accounts_getAccounts", [], _id, url=user.rpc_url
            )
            self.verify_is_valid_json_rpc_response(response)

            user.chat_public_key = next(
                (
                    item["public-key"]
                    for item in response.json()["result"]
                    if item["chat"]
                ),
                None,
            )

        # send contact requests
        for sender in self.user_1, self.user_2:
            for receiver in self.user_1, self.user_2:
                if sender != receiver:
                    response = self.rpc_request(
                        method="wakuext_sendContactRequest",
                        params=[
                            {
                                "id": receiver.chat_public_key,
                                "message": f"contact request from {sender.chat_public_key}: sent at {time.time()}",
                            }
                        ],
                        _id=99,
                        url=sender.rpc_url,
                    )

                    self.verify_is_valid_json_rpc_response(response)
                    sender.chat_id = response.json()["result"]["chats"][0]["lastMessage"]["id"]

        # accept contact requests
        for user in self.user_1, self.user_2:
            response = self.rpc_request(
                method="wakuext_acceptContactRequest",
                params=[
                    {
                        "id": user.chat_id,
                    }
                ],
                _id=99,
                url=user.rpc_url,
            )
            self.verify_is_valid_json_rpc_response(response)

        # verify contacts
        for user in (self.user_1, self.user_2), (self.user_2, self.user_1):
            response = self.rpc_request(
                method="wakuext_contacts",
                params=[],
                _id=99,
                url=user[0].rpc_url,
            )
            self.verify_is_valid_json_rpc_response(response)

            response = response.json()
            assert response["result"][0]["added"] == True
            assert response["result"][0]["id"] == user[1].chat_public_key