import logging
from typing import Any, Dict

from sqlalchemy import orm
from web3 import Web3

from mev_inspect.arbitrages import get_arbitrages
from mev_inspect.block import get_classified_traces_from_events

logger = logging.getLogger(__name__)

TRAILING_ZEROS = "000000000000000000000000"


async def inspect_block(
    inspect_db_session: orm.Session,
    w3: Web3,
    block_number: int,
):
    await inspect_many_blocks(
        inspect_db_session,
        w3,
        block_number,
        block_number + 1,
    )


async def inspect_many_blocks(
    inspect_db_session: orm.Session,
    w3: Web3,
    after_block_number: int,
    before_block_number: int,
):
    count = 0
    arbitrages_payload = []
    liquidations_payload = []

    profits = []
    async for swaps, liquidations in get_classified_traces_from_events(
        w3, after_block_number, before_block_number
    ):
        arbitrages = get_arbitrages(swaps)

        if len(arbitrages) > 0:
            for arb in arbitrages:
                arb_payload: Dict[str, Any] = dict()
                arb_payload["block_number"] = arb.block_number
                arb_payload["transaction"] = arb.transaction_hash
                arb_payload["account"] = arb.account_address
                arb_payload["profit_amt"] = arb.profit_amount
                arb_payload["token"] = arb.profit_token_address
                arbitrages_payload.append(arb_payload)
                count += 1
                profits.append(
                    [
                        arb.block_number,
                        arb.transaction_hash,
                        "",
                        0,
                        str(arb.profit_token_address).replace(TRAILING_ZEROS, ""),
                        arb.profit_amount,
                    ]
                )

        if len(liquidations) > 0:
            for liq in liquidations:
                liq_payload: Dict[str, Any] = dict()
                liq_payload["block_number"] = liq.block_number
                liq_payload["transaction"] = liq.transaction_hash
                liq_payload["liquidator"] = liq.liquidator_user
                liq_payload["purchase_addr"] = liq.debt_token_address
                liq_payload["receive_addr"] = liq.received_token_address
                liq_payload["purchase_amount"] = liq.debt_purchase_amount
                liq_payload["receive_amount"] = liq.received_amount
                liquidations_payload.append(liq_payload)
                count += 1
                profits.append(
                    [
                        liq.block_number,
                        liq.transaction_hash,
                        str(liq.debt_token_address).replace(TRAILING_ZEROS, ""),
                        liq.debt_purchase_amount,
                        str(liq.received_amount).replace(TRAILING_ZEROS, ""),
                        liq.received_amount,
                    ]
                )

    if count > 0:
        print("writing profits of {0} mev transactions".format(count))
        # @TODO: Write profits to DB
        print(inspect_db_session.info)
        arbitrages_payload = []
        liquidations_payload = []
        count = 0