Implement OFAC list address check

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,27 +13,26 @@
 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):
     @property
     @abstractmethod
-    def trading_injective_address(self) -> str:
-        ...
+    def trading_injective_address(self) -> str: ...
 
     @property
     @abstractmethod
-    def trading_private_key(self) -> PrivateKey:
-        ...
+    def trading_private_key(self) -> PrivateKey: ...
 
     @property
     @abstractmethod
-    def trading_public_key(self) -> PublicKey:
-        ...
+    def trading_public_key(self) -> PublicKey: ...
 
     @abstractmethod
-    def messages_prepared_for_transaction(self, messages: List[any_pb2.Any]) -> List[any_pb2.Any]:
-        ...
+    def messages_prepared_for_transaction(self, messages: List[any_pb2.Any]) -> List[any_pb2.Any]: ...
 
 
 class TransactionFeeCalculator(ABC):
@@ -44,8 +44,7 @@ async def configure_gas_fee_for_transaction(
         transaction: Transaction,
         private_key: PrivateKey,
         public_key: PublicKey,
-    ):
-        ...
+    ): ...
 
 
 class MsgBroadcasterWithPk:
@@ -62,6 +61,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 +167,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())

pyinjective/exceptions.py

@@ -28,3 +28,9 @@ class ConvertError(PyInjectiveError):
 
 class SchemaError(PyInjectiveError):
     pass
+
+class BannedAddressError(PyInjectiveError):
+    pass
+
+class OfacListFetchError(PyInjectiveError):
+    pass