web3-wizzard-lib 1.6.14__py3-none-any.whl → 1.7.0__py3-none-any.whl

web3_wizzard_lib/core/contract/compound_v3.py

@@ -0,0 +1,15 @@
+from sybil_engine.contract.contract import Contract
+from sybil_engine.utils.file_loader import load_abi
+
+abi = load_abi("resources/abi/compound_v3.json")
+
+
+class CompoundV3Contract(Contract):
+    def __init__(self, contract_address, web3):
+        super().__init__(contract_address, web3, abi)
+
+    def user_collateral(self, account, token_address):
+        return self.contract.functions.userCollateral(
+            account.address,
+            token_address
+        ).call()[0]

web3_wizzard_lib/core/contract/compund_v3_bulker.py

@@ -0,0 +1,39 @@
+from sybil_engine.contract.contract import Contract
+from sybil_engine.contract.transaction_executor import evm_transaction
+from sybil_engine.utils.file_loader import load_abi
+from web3_wizzard_lib.core.contract.horizondex_router import decimal_to_padded_hexadecimal
+
+abi = load_abi("resources/abi/compound_v3_bulker.json")
+
+
+class CompoundV3BulkerContract(Contract):
+    def __init__(self, contract_address, web3):
+        super().__init__(contract_address, web3, abi)
+
+    @evm_transaction
+    def invoke(self, account, amount, actions):
+        hex_amount = decimal_to_padded_hexadecimal(amount, 32)
+        data = f'0x000000000000000000000000b2f97c1bd3bf02f5e74d13f02e3e26f93d77ce44000000000000000000000000ef0f48dadd0abe4f99b4c14862df303ba956bd1300000000000000000000000000000000{hex_amount}'
+
+        return self.contract.functions.invoke(
+            [hex_to_bytes(actions)],
+            [hex_to_bytes(data)]
+        ).build_transaction(self.build_generic_data(account.address, False))
+
+
+def hex_to_bytes(hex_string):
+    """
+    Convert a hexadecimal string to bytes.
+
+    Args:
+        hex_string (str): The hexadecimal string, with or without '0x' prefix
+
+    Returns:
+        bytes: The converted bytes object
+    """
+    # Remove '0x' prefix if it exists
+    if hex_string.startswith('0x'):
+        hex_string = hex_string[2:]
+
+    # Convert hex string to bytes
+    return bytes.fromhex(hex_string)

web3_wizzard_lib/core/modules/bank/compound_v3.py

@@ -0,0 +1,35 @@
+from sybil_engine.data.contracts import get_contracts_for_chain
+from sybil_engine.data.networks import get_ids_chain
+from sybil_engine.data.tokens import get_tokens_for_chain
+from sybil_engine.utils.utils import ConfigurationException
+
+from web3_wizzard_lib.core.contract.compound_v3 import CompoundV3Contract
+from web3_wizzard_lib.core.contract.compund_v3_bulker import CompoundV3BulkerContract
+from web3_wizzard_lib.core.modules.bank.bank import Bank
+
+
+class CompoundV3(Bank):
+    app_name = 'COMPOUND_V3'
+    supported_chains = ['SCROLL']
+
+    def __init__(self, chain, web3):
+        compound_v3_address = get_contracts_for_chain(chain)[self.app_name]
+        self.contract = CompoundV3Contract(compound_v3_address, web3)
+        compound_v3_bulker_address = get_contracts_for_chain(chain)['COMPOUND_V3_BULKER']
+        self.contract_bulker = CompoundV3BulkerContract(compound_v3_bulker_address, web3)
+
+    def supply(self, account, amount):
+        raise ConfigurationException("Only redeem supported for Compound V3")
+
+    def redeem(self, account, amount, token):
+        redeem_action = '0x414354494f4e5f57495448445241575f4e41544956455f544f4b454e00000000'
+        self.contract_bulker.invoke(account, amount, redeem_action)
+
+    def get_deposit_amount(self, account, token):
+        token_address = get_tokens_for_chain(get_ids_chain()[self.contract.web3.eth.chain_id])[token]
+        compound_v3 = get_contracts_for_chain(get_ids_chain()[self.contract.web3.eth.chain_id])[self.app_name]
+
+        return CompoundV3Contract(compound_v3, self.contract.web3).user_collateral(
+            account,
+            token_address
+        )

web3_wizzard_lib/core/modules/bank_module.py

@@ -9,6 +9,7 @@ from sybil_engine.utils.utils import ConfigurationException
 from sybil_engine.utils.web3_utils import init_web3
 
 from web3_wizzard_lib.core.modules.bank.basilisk import Basilisk
+from web3_wizzard_lib.core.modules.bank.compound_v3 import CompoundV3
 from web3_wizzard_lib.core.modules.bank.eralend import Eralend
 from web3_wizzard_lib.core.modules.bank.layerbank import LayerBank
 from web3_wizzard_lib.core.modules.bank.mendi_finance import MendiFinance
@@ -80,7 +81,8 @@ class Banking(Module):
             MendiFinance,
             LayerBank,
             Aave,
-            ZeroLend
+            ZeroLend,
+            CompoundV3
         }
 
     def log(self):