(feat) Implemented OFAC list address check

shibaeff · shibaeff · commit 8069fb6f07c6 · 2024-09-06T16:21:54.000+02:00
diff --git a/pyinjective/core/broadcaster.py b/pyinjective/core/broadcaster.py
@@ -3,6 +3,7 @@
 from decimal import Decimal
 from typing import List, Optional
 
+import requests
 from google.protobuf import any_pb2
 from grpc import RpcError
 
@@ -12,6 +13,9 @@
 from pyinjective.constant import GAS_PRICE
 from pyinjective.core.gas_limit_estimator import GasLimitEstimator
 from pyinjective.core.network import Network
+from pyinjective.exceptions import BannedAddressError, OfacListFetchError
+
+OFAC_LIST_URL = "https://raw.githubusercontent.com/InjectiveLabs/injective-lists/master/wallets/ofac.json"
 
 
 class BroadcasterAccountConfig(ABC):
@@ -62,6 +66,17 @@ def __init__(
         self._client = client
         self._composer = composer
         self._fee_calculator = fee_calculator
+        self._ofac_list = self.load_ofac_list()
+
+    @classmethod
+    def load_ofac_list(cls) -> List[str]:
+        try:
+            response = requests.get(OFAC_LIST_URL)
+            response.raise_for_status()
+            ofac_list = response.json()
+            return ofac_list
+        except requests.exceptions.RequestException as e:
+            raise OfacListFetchError(f"Error fetching OFAC list: {e}")
 
     @classmethod
     def new_using_simulation(
@@ -157,6 +172,10 @@ async def broadcast(self, messages: List[any_pb2.Any]):
 
         messages_for_transaction = self._account_config.messages_prepared_for_transaction(messages=messages)
 
+        # before contraucting the transaction, check if sender address is in the OFAC list
+        if self._account_config.trading_injective_address in self._ofac_list:
+            raise BannedAddressError(f"Address {self._account_config.trading_injective_address} is in the OFAC list")
+
         transaction = Transaction()
         transaction.with_messages(*messages_for_transaction)
         transaction.with_sequence(self._client.get_sequence())
diff --git a/pyinjective/exceptions.py b/pyinjective/exceptions.py
@@ -28,3 +28,11 @@ class ConvertError(PyInjectiveError):
 
 class SchemaError(PyInjectiveError):
     pass
+
+
+class BannedAddressError(PyInjectiveError):
+    pass
+
+
+class OfacListFetchError(PyInjectiveError):
+    pass
