From 939b9db16edece54d44ac953e1cbc656c2201c65 Mon Sep 17 00:00:00 2001
From: Luke Van Seters <lukevanseters@gmail.com>
Date: Thu, 15 Jul 2021 16:43:40 -0400
Subject: [PATCH 1/2] Add types for ABIs

---
 mev_inspect/abi.py              | 20 +++++++++++++
 mev_inspect/schemas/__init__.py |  1 +
 mev_inspect/schemas/abi.py      | 52 +++++++++++++++++++++++++++++++++
 3 files changed, 73 insertions(+)
 create mode 100644 mev_inspect/abi.py
 create mode 100644 mev_inspect/schemas/abi.py

diff --git a/mev_inspect/abi.py b/mev_inspect/abi.py
new file mode 100644
index 0000000..614704e
--- /dev/null
+++ b/mev_inspect/abi.py
@@ -0,0 +1,20 @@
+import json
+from typing import Optional
+
+from pydantic import parse_obj_as
+
+from mev_inspect.config import load_config
+from mev_inspect.schemas import ABI
+
+
+ABI_CONFIG_KEY = "ABI"
+
+config = load_config()
+
+
+def get_abi(abi_name: str) -> Optional[ABI]:
+    if abi_name in config[ABI_CONFIG_KEY]:
+        abi_json = json.loads(config[ABI_CONFIG_KEY][abi_name])
+        return parse_obj_as(ABI, abi_json)
+
+    return None
diff --git a/mev_inspect/schemas/__init__.py b/mev_inspect/schemas/__init__.py
index 38980cb..f56e7ec 100644
--- a/mev_inspect/schemas/__init__.py
+++ b/mev_inspect/schemas/__init__.py
@@ -1 +1,2 @@
+from .abi import ABI
 from .blocks import Block, BlockCall, BlockCallType
diff --git a/mev_inspect/schemas/abi.py b/mev_inspect/schemas/abi.py
new file mode 100644
index 0000000..d337480
--- /dev/null
+++ b/mev_inspect/schemas/abi.py
@@ -0,0 +1,52 @@
+from enum import Enum
+from typing import List, Union
+from typing_extensions import Literal
+
+from hexbytes import HexBytes
+from pydantic import BaseModel
+from web3 import Web3
+
+from .utils import CamelModel
+
+
+class ABIDescriptionType(str, Enum):
+    function = "function"
+    constructor = "constructor"
+    fallback = "fallback"
+    event = "event"
+    receive = "receive"
+
+
+NON_FUNCTION_DESCRIPTION_TYPES = Union[
+    Literal[ABIDescriptionType.constructor],
+    Literal[ABIDescriptionType.fallback],
+    Literal[ABIDescriptionType.event],
+    Literal[ABIDescriptionType.receive],
+]
+
+
+class ABIDescriptionInput(CamelModel):
+    internal_type: str
+
+
+class ABIGenericDescription(BaseModel):
+    type: NON_FUNCTION_DESCRIPTION_TYPES
+
+
+class ABIFunctionDescription(BaseModel):
+    type: Literal[ABIDescriptionType.function]
+    name: str
+    inputs: List[ABIDescriptionInput]
+
+    def get_selector(self) -> HexBytes:
+        signature = self.get_signature()
+        return Web3.sha3(text=signature)[0:4]
+
+    def get_signature(self) -> str:
+        joined_input_types = ",".join(input.internal_type for input in self.inputs)
+
+        return f"{self.name}({joined_input_types})"
+
+
+ABIDescription = Union[ABIFunctionDescription, ABIGenericDescription]
+ABI = List[ABIDescription]

From c93ae8193245d3cf2fb4c01babcb417524cb0564 Mon Sep 17 00:00:00 2001
From: Luke Van Seters <lukevanseters@gmail.com>
Date: Thu, 15 Jul 2021 16:54:43 -0400
Subject: [PATCH 2/2] Use type not internal type

---
 mev_inspect/schemas/abi.py | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/mev_inspect/schemas/abi.py b/mev_inspect/schemas/abi.py
index d337480..3c54837 100644
--- a/mev_inspect/schemas/abi.py
+++ b/mev_inspect/schemas/abi.py
@@ -6,8 +6,6 @@ from hexbytes import HexBytes
 from pydantic import BaseModel
 from web3 import Web3
 
-from .utils import CamelModel
-
 
 class ABIDescriptionType(str, Enum):
     function = "function"
@@ -25,8 +23,8 @@ NON_FUNCTION_DESCRIPTION_TYPES = Union[
 ]
 
 
-class ABIDescriptionInput(CamelModel):
-    internal_type: str
+class ABIDescriptionInput(BaseModel):
+    type: str
 
 
 class ABIGenericDescription(BaseModel):
@@ -43,8 +41,7 @@ class ABIFunctionDescription(BaseModel):
         return Web3.sha3(text=signature)[0:4]
 
     def get_signature(self) -> str:
-        joined_input_types = ",".join(input.internal_type for input in self.inputs)
-
+        joined_input_types = ",".join(input.type for input in self.inputs)
         return f"{self.name}({joined_input_types})"