import argparse

from web3 import Web3

from mev_inspect import block
from mev_inspect.inspectors.uniswap import UniswapInspector
from mev_inspect.processor import Processor

parser = argparse.ArgumentParser(description="Inspect some blocks.")
parser.add_argument(
    "-block_number",
    metavar="b",
    type=int,
    nargs="+",
    help="the block number you are targetting, eventually this will need to be changed",
)
parser.add_argument(
    "-rpc", metavar="r", help="rpc endpoint, this needs to have parity style traces"
)
args = parser.parse_args()

## Set up the base provider, but don't wrap it in web3 so we can make requests to it with make_request()
base_provider = Web3.HTTPProvider(args.rpc)

## Get block data that we need
block_data = block.create_from_block_number(args.block_number[0], base_provider)
print(f"Total traces: {len(block_data.traces)}")

total_transactions = len(
    set(t.transaction_hash for t in block_data.traces if t.transaction_hash is not None)
)
print(f"Total transactions: {total_transactions}")

## Build a Uniswap inspector
uniswap_inspector = UniswapInspector(base_provider)

## Create a processor, pass in an ARRAY of inspects
processor = Processor([uniswap_inspector, uniswap_inspector])

classifications = processor.get_transaction_evaluations(block_data)
print(f"Returned {len(classifications)} classifications")