Merge pull request #115 from mendesfabio/balancer-v3-plasma

add balancer v3 plasma

Author: FJ-Riveros

.env.example

@@ -10,5 +10,5 @@ SWELL_NODE_URL='https://rpc.ankr.com/swell'
 SOLANA_NODE_URL='https://api.mainnet-beta.solana.com'
 BASE_NODE_URL="https://mainnet.base.org"
 HYPEREVM_NODE_URL="https://rpc.hyperlend.finance/archive"
-
+PLASMA_NODE_URL="https://rpc.plasma.to"
 DERIVE_SUBGRAPH_API_KEY=''

constants/balancer.py

@@ -13,6 +13,7 @@ class Token(Enum):
 AURA_VOTER_PROXY = {
     Chain.ETHEREUM: "0xaF52695E1bB01A16D33D7194C28C42b10e0Dbec2",
     Chain.FRAXTAL: "0xC181Edc719480bd089b94647c2Dc504e2700a2B0",
+    Chain.PLASMA: "0x0000000000000000000000000000000000000000",
 }
 
 BALANCER_V2_VAULT = "0xBA12222222228d8Ba445958a75a0704d566BF2C8"

constants/balancer_v3.py

@@ -3,17 +3,21 @@
 from typing import Dict, Optional
 
 from constants.chains import Chain
+from constants.balancer import Token
 from integrations.integration_ids import IntegrationID
 
 ## If you want to integrate another Balancer Pool, first add it to the IntegrationID enum in integration_ids.py
 ## Then, add a new entry to the INTEGRATION_CONFIGS dictionary below. Aura integration is optional.
 ## If the chain is not yet supported, add it to the Chain enum in chains.py and add RPCs to web3_utils.py.
 
 
-class Token(Enum):  # Ensure address is checksummed
+class WrappedToken(Enum):  # Ensure address is checksummed
     WA_ETH_USDE = (
         "0x5F9D59db355b4A60501544637b00e94082cA575b"  # Wrapped Aave Ethereum USDe
     )
+    WA_PLA_USDE = (
+        "0xC63F1a8c0cD4493E18f6f3371182BE01Ce0BeF02"  # Wrapped Aave Plasma USDe
+    )
 
 
 @dataclass
@@ -31,11 +35,27 @@ class IntegrationConfig:
     IntegrationID.BALANCER_V3_ETHEREUM_USDE_USDT: IntegrationConfig(
         chain=Chain.ETHEREUM,
         start_block=21467086,
-        incentivized_token=Token.WA_ETH_USDE.value,
+        incentivized_token=WrappedToken.WA_ETH_USDE.value,
         incentivized_token_decimals=18,
         pool_address="0xc1D48bB722a22Cc6Abf19faCbE27470F08B3dB8c",
         gauge_address="0x95d260ac86B58D458187C819f87aAd2c7c4203eF",
     ),
+    IntegrationID.BALANCER_V3_PLASMA_USDE_USDT: IntegrationConfig(
+        chain=Chain.PLASMA,
+        start_block=1727318,
+        incentivized_token=WrappedToken.WA_PLA_USDE.value,
+        incentivized_token_decimals=18,
+        pool_address="0x6a74BE33B5393D8A3EbA4D69B78f9D9da947C48c",
+        gauge_address=None,
+    ),
+    IntegrationID.BALANCER_V3_PLASMA_SUSDE_USDT: IntegrationConfig(
+        chain=Chain.PLASMA,
+        start_block=1726616,
+        incentivized_token=Token.SUSDE.value,
+        incentivized_token_decimals=18,
+        pool_address="0xd9c4e277c93374a9f8C877a9D06707a88092E8F0",
+        gauge_address=None,
+    ),
 }
 
 

constants/chains.py

@@ -18,3 +18,4 @@ class Chain(Enum):
     SEPOLIA = "Sepolia"
     TON = "Ton"
     HYPEREVM = "HyperEVM"
+    PLASMA = "Plasma"

integrations/balancer_v3_integration.py

@@ -40,8 +40,8 @@ def get_balance(self, user: str, block: int | str = "latest") -> float:
         Retrieve the balance of the user in the incentivized Ethena token.
 
         This method calculates the user's token balance based on the share of Balancer Pool Tokens (BPTs)
-        staked either directly in Balancer gauges or via Aura Finance.
         """
+        bpt_balance = get_user_balance(self.chain, user, self.pool_address, block)
         gauge_balance = (
             0
             if self.gauge_address is None
@@ -55,7 +55,7 @@ def get_balance(self, user: str, block: int | str = "latest") -> float:
 
         bpt_supply = get_token_supply(self.chain, self.pool_address, block)
 
-        user_balance = gauge_balance + aura_balance
+        user_balance = bpt_balance + gauge_balance + aura_balance
 
         incentivized_token_balance = get_vault_v3_pool_token_balance(
             self.chain, self.pool_address, self.incentivized_token, block
@@ -74,11 +74,12 @@ def get_participants(
         blocks: Optional[List[int]],
     ) -> Set[str]:
         """
-        Retrieve the set of all unique participants who might have staked Balancer Pool Tokens (BPTs).
-
-        This method identifies all addresses that have staked their BPT either directly
-        in Balancer gauges or via Aura Finance. Non-staked BPT holders are not included.
+        Retrieve the set of all unique participants who might have Balancer Pool Tokens (BPTs).
         """
+        bpt_holders = get_potential_token_holders(
+            self.chain, self.pool_address, self.start_block
+        )
+
         gauge_holders = (
             []
             if self.gauge_address is None
@@ -95,10 +96,10 @@ def get_participants(
             )
         )
 
-        return set(aura_holders + gauge_holders)
+        return set(bpt_holders + aura_holders + gauge_holders)
 
 
 if __name__ == "__main__":
-    balancer = BalancerV3Integration(IntegrationID.BALANCER_V3_ETHEREUM_USDE_USDT)
+    balancer = BalancerV3Integration(IntegrationID.BALANCER_V3_PLASMA_USDE_USDT)
     participants = balancer.get_participants(None)
     balances = balancer.get_balance(list(participants)[0])

integrations/integration_ids.py

@@ -343,6 +343,18 @@ class IntegrationID(Enum):
         Token.USDE,
     )
 
+    BALANCER_V3_PLASMA_USDE_USDT = (
+        "balancer_v3_plasma_usde_usdt",
+        "Balancer V3 Plasma USDe/USDT",
+        Token.USDE,
+    )
+
+    BALANCER_V3_PLASMA_SUSDE_USDT = (
+        "balancer_v3_plasma_susde_usdt",
+        "Balancer V3 Plasma sUSDe/USDT",
+        Token.SUSDE,
+    )
+
     # Nuri
     NURI_USDE_LP = ("nuri_usde_lp_bal", "Nuri USDe LP", Token.USDE)
 

utils/web3_utils.py

@@ -40,6 +40,8 @@
 w3_sepolia = Web3(Web3.HTTPProvider(SEPOLIA_NODE_URL))
 HYPEREVM_NODE_URL = os.getenv("HYPEREVM_NODE_URL")
 w3_hyperevm = Web3(Web3.HTTPProvider(HYPEREVM_NODE_URL))
+PLASMA_NODE_URL = os.getenv("PLASMA_NODE_URL")
+w3_plasma = Web3(Web3.HTTPProvider(PLASMA_NODE_URL))
 
 W3_BY_CHAIN = {
     Chain.ETHEREUM: {
@@ -84,6 +86,9 @@
     Chain.HYPEREVM: {
         "w3": w3_hyperevm,
     },
+    Chain.PLASMA: {
+        "w3": w3_plasma,
+    },
 }
 
 MULTICALL_ABI = [