Move uniswap inspector. Implement the inspector interface

mev_inspect/inspector_uniswap.py

@@ -1,96 +0,0 @@
-import json
-
-from web3 import Web3
-
-from mev_inspect import utils
-from mev_inspect.config import load_config
-
-config = load_config()
-
-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.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
-
-    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):
-        for call in calls:
-            print("\n", call)
-            if (
-                call["type"] == "call"
-                and (
-                    call["action"]["to"] == uniswap_router_address.lower()
-                    or call["action"]["to"] == sushiswap_router_address.lower()
-                )
-                and utils.check_trace_for_signature(
-                    call, self.uniswap_router_trade_signatures
-                )
-            ):
-                # print("WIP, here is where there is a call that matches what we are looking for")
-                1 == 1

mev_inspect/inspectors/__init__.py

@@ -0,0 +1 @@
+from .base import Inspector