import uuid

import pytest

from conftest import option
from resources.constants import user_1, user_2
from test_cases import StatusBackendTestCase
from clients.signals import SignalType


@pytest.mark.rpc
@pytest.mark.transaction
@pytest.mark.wallet
class TestTransactionFromRoute(StatusBackendTestCase):
    await_signals = [
        SignalType.NODE_LOGIN.value,
        SignalType.WALLET_SUGGESTED_ROUTES.value,
        SignalType.WALLET_ROUTER_SIGN_TRANSACTIONS.value,
        SignalType.WALLET_ROUTER_SENDING_TRANSACTIONS_STARTED.value,
        SignalType.WALLET_TRANSACTION_STATUS_CHANGED.value,
        SignalType.WALLET_ROUTER_TRANSACTIONS_SENT.value,
    ]

    def test_tx_from_route(self):

        _uuid = str(uuid.uuid4())
        amount_in = "0xde0b6b3a7640000"

        method = "wallet_getSuggestedRoutesAsync"
        params = [
            {
                "uuid": _uuid,
                "sendType": 0,
                "addrFrom": user_1.address,
                "addrTo": user_2.address,
                "amountIn": amount_in,
                "amountOut": "0x0",
                "tokenID": "ETH",
                "tokenIDIsOwnerToken": False,
                "toTokenID": "",
                "disabledFromChainIDs": [10, 42161],
                "disabledToChainIDs": [10, 42161],
                "gasFeeMode": 1,
                "fromLockedAmount": {},
            }
        ]
        response = self.rpc_client.rpc_valid_request(method, params)

        routes = self.rpc_client.wait_for_signal(SignalType.WALLET_SUGGESTED_ROUTES.value)
        assert routes["event"]["Uuid"] == _uuid

        method = "wallet_buildTransactionsFromRoute"
        params = [{"uuid": _uuid, "slippagePercentage": 0}]
        response = self.rpc_client.rpc_valid_request(method, params)

        wallet_router_sign_transactions = self.rpc_client.wait_for_signal(SignalType.WALLET_ROUTER_SIGN_TRANSACTIONS.value)

        assert wallet_router_sign_transactions["event"]["signingDetails"]["signOnKeycard"] is False
        transaction_hashes = wallet_router_sign_transactions["event"]["signingDetails"]["hashes"]

        assert transaction_hashes, "Transaction hashes are empty!"

        tx_signatures = {}

        for hash in transaction_hashes:

            method = "wallet_signMessage"
            params = [hash, user_1.address, option.password]

            response = self.rpc_client.rpc_valid_request(method, params)

            result = response.json().get("result")
            assert result and result.startswith("0x"), f"Invalid transaction signature for hash {hash}: {result}"

            tx_signature = result[2:]

            signature = {
                "r": tx_signature[:64],
                "s": tx_signature[64:128],
                "v": tx_signature[128:],
            }

            tx_signatures[hash] = signature

        method = "wallet_sendRouterTransactionsWithSignatures"
        params = [{"uuid": _uuid, "Signatures": tx_signatures}]
        response = self.rpc_client.rpc_valid_request(method, params)

        tx_status = self.rpc_client.wait_for_signal(SignalType.WALLET_TRANSACTION_STATUS_CHANGED.value)

        assert tx_status["event"]["chainId"] == 31337
        assert tx_status["event"]["status"] == "Success"
        tx_hash = tx_status["event"]["hash"]

        method = "ethclient_transactionByHash"
        params = [self.network_id, tx_hash]

        response = self.rpc_client.rpc_valid_request(method, params)
        tx_details = response.json()["result"]["tx"]

        assert tx_details["value"] == amount_in
        assert tx_details["to"].upper() == user_2.address.upper()