Restructure classifier
This commit is contained in:
parent
d75e9b76ab
commit
d7872db45c
@ -1,4 +1,4 @@
|
|||||||
from typing import Optional, List, Sequence
|
from typing import Optional, List, Sequence, Tuple
|
||||||
|
|
||||||
from mev_inspect.schemas.swaps import Swap
|
from mev_inspect.schemas.swaps import Swap
|
||||||
from mev_inspect.schemas.transfers import Transfer, ETH_TOKEN_ADDRESS
|
from mev_inspect.schemas.transfers import Transfer, ETH_TOKEN_ADDRESS
|
||||||
@ -6,6 +6,17 @@ from mev_inspect.schemas.transfers import Transfer, ETH_TOKEN_ADDRESS
|
|||||||
from mev_inspect.schemas.traces import DecodedCallTrace, ClassifiedTrace
|
from mev_inspect.schemas.traces import DecodedCallTrace, ClassifiedTrace
|
||||||
|
|
||||||
|
|
||||||
|
RFQ_SIGNATURES = [
|
||||||
|
"fillRfqOrder((address,address,uint128,uint128,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128)",
|
||||||
|
"_fillRfqOrder((address,address,uint128,uint128,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128,address,bool,address)",
|
||||||
|
]
|
||||||
|
LIMIT_SIGNATURES = [
|
||||||
|
"fillOrKillLimitOrder((address,address,uint128,uint128,uint128,address,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128)",
|
||||||
|
"fillLimitOrder((address,address,uint128,uint128,uint128,address,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128)",
|
||||||
|
"_fillLimitOrder((address,address,uint128,uint128,uint128,address,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128,address,address)",
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
def create_swap_from_transfers(
|
def create_swap_from_transfers(
|
||||||
trace: DecodedCallTrace,
|
trace: DecodedCallTrace,
|
||||||
recipient_address: str,
|
recipient_address: str,
|
||||||
@ -86,19 +97,67 @@ def _filter_transfers(
|
|||||||
return filtered_transfers
|
return filtered_transfers
|
||||||
|
|
||||||
|
|
||||||
def get_first_amount_transferred_to_address(
|
def is_valid_0x_swap(
|
||||||
address: str, transfers: List[Transfer]
|
trace: DecodedCallTrace,
|
||||||
|
child_transfers: List[Transfer],
|
||||||
|
) -> bool:
|
||||||
|
|
||||||
|
# 1. There should be 2 child transfers, one for each settled leg of the order
|
||||||
|
if len(child_transfers) != 2:
|
||||||
|
raise ValueError(
|
||||||
|
f"A settled order should consist of 2 child transfers, not {len(child_transfers)}."
|
||||||
|
)
|
||||||
|
|
||||||
|
# 2. The function signature must be in the lists of supported signatures
|
||||||
|
if trace.function_signature not in (LIMIT_SIGNATURES + RFQ_SIGNATURES):
|
||||||
|
raise RuntimeError(
|
||||||
|
f"0x orderbook function {trace.function_signature} is not supported"
|
||||||
|
)
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
def _get_taker_token_in_amount(
|
||||||
|
taker_address: str, token_in_address: str, child_transfers: List[Transfer]
|
||||||
) -> int:
|
) -> int:
|
||||||
for transfer in transfers:
|
|
||||||
if transfer.to_address == address:
|
ANY_TAKER = "0x0000000000000000000000000000000000000000"
|
||||||
return transfer.amount
|
|
||||||
|
if taker_address == ANY_TAKER:
|
||||||
|
for transfer in child_transfers:
|
||||||
|
if transfer.token_address == token_in_address:
|
||||||
|
return transfer.amount
|
||||||
|
else:
|
||||||
|
for transfer in child_transfers:
|
||||||
|
if transfer.to_address == taker_address:
|
||||||
|
return transfer.amount
|
||||||
return 0
|
return 0
|
||||||
|
|
||||||
|
|
||||||
def get_first_amount_transferred_by_token_address(
|
def get_0x_token_in_data(
|
||||||
token_address: str, transfers: List[Transfer]
|
trace: DecodedCallTrace, child_transfers: List[Transfer]
|
||||||
) -> int:
|
) -> Tuple[str, int]:
|
||||||
for transfer in transfers:
|
|
||||||
if transfer.token_address == token_address:
|
order: List = trace.inputs["order"]
|
||||||
return transfer.amount
|
token_in_address = order[0]
|
||||||
return 0
|
|
||||||
|
if trace.function_signature in RFQ_SIGNATURES:
|
||||||
|
taker_address = order[5]
|
||||||
|
|
||||||
|
elif trace.function_signature in LIMIT_SIGNATURES:
|
||||||
|
taker_address = order[6]
|
||||||
|
|
||||||
|
token_in_amount = _get_taker_token_in_amount(
|
||||||
|
taker_address, token_in_address, child_transfers
|
||||||
|
)
|
||||||
|
|
||||||
|
return token_in_address, token_in_amount
|
||||||
|
|
||||||
|
|
||||||
|
def get_0x_token_out_data(trace: DecodedCallTrace) -> Tuple[str, int]:
|
||||||
|
|
||||||
|
order: List = trace.inputs["order"]
|
||||||
|
token_out_address = order[1]
|
||||||
|
token_out_amount = trace.inputs["takerTokenFillAmount"]
|
||||||
|
|
||||||
|
return token_out_address, token_out_amount
|
||||||
|
@ -10,8 +10,9 @@ from mev_inspect.schemas.classifiers import (
|
|||||||
SwapClassifier,
|
SwapClassifier,
|
||||||
)
|
)
|
||||||
from mev_inspect.classifiers.helpers import (
|
from mev_inspect.classifiers.helpers import (
|
||||||
get_first_amount_transferred_to_address,
|
is_valid_0x_swap,
|
||||||
get_first_amount_transferred_by_token_address,
|
get_0x_token_in_data,
|
||||||
|
get_0x_token_out_data,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@ -23,58 +24,11 @@ class ZeroExSwapClassifier(SwapClassifier):
|
|||||||
child_transfers: List[Transfer],
|
child_transfers: List[Transfer],
|
||||||
) -> Optional[Swap]:
|
) -> Optional[Swap]:
|
||||||
|
|
||||||
ANY_TAKER = "0x0000000000000000000000000000000000000000"
|
assert is_valid_0x_swap(trace, child_transfers)
|
||||||
RFQ_SIGNATURES = [
|
|
||||||
"fillRfqOrder((address,address,uint128,uint128,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128)",
|
|
||||||
"_fillRfqOrder((address,address,uint128,uint128,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128,address,bool,address)",
|
|
||||||
]
|
|
||||||
LIMIT_SIGNATURES = [
|
|
||||||
"fillOrKillLimitOrder((address,address,uint128,uint128,uint128,address,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128)",
|
|
||||||
"fillLimitOrder((address,address,uint128,uint128,uint128,address,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128)",
|
|
||||||
"_fillLimitOrder((address,address,uint128,uint128,uint128,address,address,address,address,bytes32,uint64,uint256),(uint8,uint8,bytes32,bytes32),uint128,address,address)",
|
|
||||||
]
|
|
||||||
|
|
||||||
# Assumptions:
|
token_in_address, token_in_amount = get_0x_token_in_data(trace, child_transfers)
|
||||||
# 1. There should be 2 child transfers, one for each settled leg of the order
|
|
||||||
if len(child_transfers) != 2:
|
|
||||||
raise ValueError(
|
|
||||||
f"A settled order should consist of 2 child transfers, not {len(child_transfers)}."
|
|
||||||
)
|
|
||||||
|
|
||||||
# 2. The function signature must be in the lists of supported signatures
|
token_out_address, token_out_amount = get_0x_token_out_data(trace)
|
||||||
if trace.function_signature not in (LIMIT_SIGNATURES + RFQ_SIGNATURES):
|
|
||||||
raise RuntimeError(
|
|
||||||
f"0x orderbook function {trace.function_signature} is not supported"
|
|
||||||
)
|
|
||||||
|
|
||||||
# The position of the token addresses and presence of takerTokenFillAmount
|
|
||||||
# is always ensured across order formats
|
|
||||||
|
|
||||||
order: List = trace.inputs["order"]
|
|
||||||
|
|
||||||
token_in_address: str = order[0]
|
|
||||||
|
|
||||||
token_out_address: str = order[1]
|
|
||||||
token_out_amount = trace.inputs["takerTokenFillAmount"]
|
|
||||||
|
|
||||||
# Find token in amount
|
|
||||||
if trace.function_signature in RFQ_SIGNATURES:
|
|
||||||
taker_address = order[5]
|
|
||||||
|
|
||||||
elif trace.function_signature in LIMIT_SIGNATURES:
|
|
||||||
taker_address = order[6]
|
|
||||||
|
|
||||||
token_out_amount = trace.inputs["takerTokenFillAmount"]
|
|
||||||
|
|
||||||
if taker_address == ANY_TAKER:
|
|
||||||
token_in_amount = get_first_amount_transferred_by_token_address(
|
|
||||||
token_in_address, child_transfers
|
|
||||||
)
|
|
||||||
|
|
||||||
else:
|
|
||||||
token_in_amount = get_first_amount_transferred_to_address(
|
|
||||||
taker_address, child_transfers
|
|
||||||
)
|
|
||||||
|
|
||||||
return Swap(
|
return Swap(
|
||||||
abi_name=trace.abi_name,
|
abi_name=trace.abi_name,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user