refactor: use params instead of hardcoded addresses in substreams

- Remove hardcoded MAINNET_PROXY_ADDRESS and NILE_PROXY_ADDRESS constants
- Add parse_proxy_addresses function to read addresses from params
- Update substreams.yaml to pass params to the map module

This allows the proxy addresses to be configured without recompiling.

Author: rodrigopavezi

packages/substreams-tron/src/lib.rs

@@ -10,19 +10,37 @@ use pb::request::tron::v1::{Payment, Payments};
 use substreams::log;
 use substreams_tron::pb::sf::tron::r#type::v1::Block;
 
-/// ERC20FeeProxy contract addresses
-const MAINNET_PROXY_ADDRESS: &str = "TCUDPYnS9dH3WvFEaE7wN7vnDa51J4R4fd";
-const NILE_PROXY_ADDRESS: &str = "THK5rNmrvCujhmrXa5DB1dASepwXTr9cJs";
-
 /// TransferWithReferenceAndFee event signature (keccak256 hash of event signature)
 /// Event: TransferWithReferenceAndFee(address,address,uint256,bytes indexed,uint256,address)
 /// keccak256("TransferWithReferenceAndFee(address,address,uint256,bytes,uint256,address)")
 const TRANSFER_WITH_REF_AND_FEE_TOPIC: &str =
     "9f16cbcc523c67a60c450e5ffe4f3b7b6dbe772e7abcadb2686ce029a9a0a2b6";
 
+/// Parses proxy addresses from the params string
+/// Expected format: "mainnet_proxy_address=ADDR1\nnile_proxy_address=ADDR2"
+fn parse_proxy_addresses(params: &str) -> (String, String) {
+    let mut mainnet = String::new();
+    let mut nile = String::new();
+    
+    for line in params.lines() {
+        let parts: Vec<&str> = line.splitn(2, '=').collect();
+        if parts.len() == 2 {
+            match parts[0].trim() {
+                "mainnet_proxy_address" => mainnet = parts[1].trim().to_string(),
+                "nile_proxy_address" => nile = parts[1].trim().to_string(),
+                _ => {}
+            }
+        }
+    }
+    
+    (mainnet, nile)
+}
+
 /// Maps TRON blocks to extract ERC20FeeProxy payment events
 #[substreams::handlers::map]
-fn map_erc20_fee_proxy_payments(block: Block) -> Result<Payments, substreams::errors::Error> {
+fn map_erc20_fee_proxy_payments(params: String, block: Block) -> Result<Payments, substreams::errors::Error> {
+    let (mainnet_proxy, nile_proxy) = parse_proxy_addresses(&params);
+    
     let mut payments = Vec::new();
     let block_number = block.number;
     let block_timestamp = block.header.as_ref().map(|h| h.timestamp).unwrap_or(0) / 1000; // Convert from ms to seconds
@@ -36,8 +54,7 @@ fn map_erc20_fee_proxy_payments(block: Block) -> Result<Payments, substreams::er
                 // Check if this log is from one of our proxy contracts
                 let contract_address = base58_encode(&log_entry.address);
 
-                if contract_address != MAINNET_PROXY_ADDRESS 
-                    && contract_address != NILE_PROXY_ADDRESS {
+                if contract_address != mainnet_proxy && contract_address != nile_proxy {
                     continue;
                 }
 

packages/substreams-tron/substreams.yaml

@@ -27,6 +27,7 @@ modules:
     kind: map
     initialBlock: 63208782 # Nile testnet deployment block (use lower for both networks)
     inputs:
+      - params: string
       - source: sf.tron.type.v1.Block
     output:
       type: proto:request.tron.v1.Payments
@@ -35,6 +36,10 @@ modules:
       Filters transactions to the known proxy contract addresses and parses
       the event logs to extract payment details.
 
+      Params format:
+        mainnet_proxy_address=<address>
+        nile_proxy_address=<address>
+
   - name: store_payments
     kind: store
     updatePolicy: set