import json
from typing import Optional

from web3 import Web3

from typing import List, Optional

from mev_inspect import utils
from mev_inspect.config import load_config
from mev_inspect.schemas.blocks import NestedTrace, TraceType
from mev_inspect.schemas.classified_traces import Classification, ClassifiedTrace
from mev_inspect.schemas.strategy import StrategyType, Strategy, Liquidation
from mev_inspect.classifier_specs import CLASSIFIER_SPECS
from mev_inspect.trace_classifier import TraceClassifier
from mev_inspect import block

# poetry run inspect -b 12498502 -r 'http://162.55.96.141:8546/'
all_traces = []
result = []
addrs = ['0x3dfd23A6c5E8BbcFc9581d2E864a68feb6a076d3']
transfers = []
# Inspect list of classified traces and identify liquidation
def liquidations(traces: List[ClassifiedTrace]):
	event = []
	# For each trace
	for k in range(len(traces)):
		trace = traces[k]

		# Check for liquidation and register trace and unique liquidator
		if trace.classification == Classification.liquidate:

			all_traces.append(trace)
			liquidator = trace.from_address
			if liquidator not in addrs: addrs.append(liquidator)
			print(f"\nLiquidation found: \n\t{liquidator}")
			print(f"\nHash: \n\t{trace.transaction_hash}\n")

			# Found liquidation, now parse inputs for data
			for i in trace.inputs:

				if(i == '_purchaseAmount'):
					liquidation_amount = trace.inputs[i]
				elif (i == '_collateral'):
					collateral_type = trace.inputs[i]
				elif (i == '_reserve'):
					reserve = trace.inputs[i]
				elif(i == '_user'):
					liquidated_usr = trace.inputs[i]

			# Register liquidation
			result.append(
							Liquidation(collateral_type=collateral_type,
			    							collateral_amount=liquidation_amount,
			    							collateral_source="",
			    							reserve=reserve,
											strategy=StrategyType.liquidation,
											traces=all_traces,
											protocols=[trace.protocol],)
						  )

		# Check for transfer from a registered liquidator
		elif (
			  trace.classification  ==  Classification.transfer and
			  'sender' in trace.inputs and
			  trace.inputs['sender'] in addrs
			 ):

			liquidator = next(addrs[i] for i in range(len(addrs)) if trace.inputs['sender'] == addrs[i])
			transfers.append(trace)

			print(f"""
				  \nTransfer from liquidator {liquidator}:
				  \n\tAmount in received token: {trace.inputs['amount']} to
				  \n\t{trace.inputs['recipient']}
				  \n\tTransaction: {trace.transaction_hash}\n
				  """)

		# Check for transfer to a registered liquidator
		elif (
			  trace.classification == Classification.transfer and
			  trace.inputs['recipient'] in addrs
			 ):

			liquidator = next(addrs[i] for i in range(len(addrs)) if trace.inputs['recipient'] == addrs[i])
			transfers.append(trace)

			print(f"""
				  \nTransfer to liquidator {liquidator}:
				  \n\tAmount in received token: {trace.inputs['amount']} from
				  \n\t{trace.from_address}
				  \n\tTransaction: {trace.transaction_hash}\n
				  """)

	return result


rpc = 'http://162.55.96.141:8546/'
block_number = 12498502
base_provider = Web3.HTTPProvider(rpc)
block_data = block.create_from_block_number(block_number, base_provider)
trace_clasifier = TraceClassifier(CLASSIFIER_SPECS)
classified_traces = trace_clasifier.classify(block_data.traces)
fin = liquidations(classified_traces)