get signatures for uniswap router trading, add new abi, annotate

inspector_uniswap.py

@@ -1,25 +1,79 @@
 from web3 import Web3
 import configparser
 import json
+import utils
 
 ## Config file is used for addresses/ABIs
 config = configparser.ConfigParser()
 config.read('./utils/config.ini')
 
 uniswap_router_abi = json.loads(config['ABI']['UniswapV2Router'])
+uniswap_router_address = (config['ADDRESSES']['UniswapV2Router'])
+sushiswap_router_address = (config['ADDRESSES']['SushiswapV2Router'])
+
+uniswap_pair_abi = json.loads(config['ABI']['UniswapV2Pair'])
 
 class UniswapInspector:
     def __init__(self, base_provider) -> None:
         self.w3 = Web3(base_provider)
+        
         self.trading_functions = self.get_trading_functions()
-        self.uniswapV2RouterContract = self.w3.eth.contract(abi=uniswap_router_abi, address=config['ADDRESSES']['UniswapV2Router'])
+        self.uniswap_v2_router_contract = self.w3.eth.contract(abi=uniswap_router_abi, address=uniswap_router_address)
+        self.uniswap_router_trade_signatures = self.get_router_signatures()
 
+        self.uniswap_v2_pair_contract = self.w3.eth.contract(abi=uniswap_pair_abi)
+        self.uniswap_v2_pair_swap_signatures = self.uniswap_v2_pair_contract.functions.swap(0, 0, uniswap_router_address, "").selector ## Note the address here doesn't matter, but it must be filled out
+        self.uniswap_v2_pair_reserves_signatures = self.uniswap_v2_pair_contract.functions.getReserves().selector ## Called "checksigs" in mev-inpsect.ts
+
+        print("Built Uniswap inspector")
     def get_trading_functions(self):
         ## Gets all functions used for swapping
         result = []
        
+        ## For each entry in the ABI
         for abi in uniswap_router_abi:
+            ## Check to see if the entry is a function and if it is if the function's name starts with swap
             if abi['type'] == 'function' and abi['name'].startswith('swap'):
+                ## If so add it to our array
                 result.append(abi['name'])
         
-        return result
+        return result
+
+    def get_router_signatures(self):
+        ## Gets the selector / function signatures of all the router swap functions
+        result = []
+       
+        ## For each entry in the ABI
+        for abi in uniswap_router_abi:
+            ## Check to see if the entry is a function and if it is if the function's name starts with swap
+            if abi['type'] == 'function' and abi['name'].startswith('swap'):
+                ## Add a parantheses
+                function = abi['name'] + '('
+                
+                ## For each input in the function's input
+                for input in abi['inputs']:
+                    
+                    ## Concat them into a string with commas
+                    function = function + input['internalType'] + ','
+
+                ## Take off the last comma, add a ')' to close the parentheses
+                function = function[:-1] + ')'  
+
+                ## The result looks like this: 'swapETHForExactTokens(uint256,address[],address,uint256)'
+
+                ## Take the first 4 bytes of the sha3 hash of the above string.
+                selector = Web3.sha3(text=function)[0:4]
+
+                ## Add that to an array
+                result.append(selector)
+
+        return result
+
+    def inspect(self, calls):
+        result = []
+
+        trade_calls = []
+
+        for call in calls:
+            if (call.action.to == uniswap_router_address.lower() or call.action.to == sushiswap_router_address.lower()) and utils.check_call_for_signature(call, self.uniswap_router_trade_signatures):
+                print("WIP, here is where there is a call that matches what we are looking for")

utils.py

@@ -0,0 +1,14 @@
+
+def check_call_for_signature(call, signatures):
+    if (call.action.input == None):
+        return False
+    
+    ## By default set this to False
+    signature_present_boolean = False
+
+    ## Iterate over all signatures, and if our call matches any of them set it to True
+    for signature in signatures:
+        if call.action.input.startsWith(signature):
+            signature_present_boolean = True
+
+    return signature_present_boolean