From 194cfe52966afb75d3a5e07a63a457428fb05c14 Mon Sep 17 00:00:00 2001
From: 0xMochan <mochanbot@gmail.com>
Date: Fri, 24 May 2024 17:51:49 -0400
Subject: [PATCH] feat: add WETH to ETH address swap

---
 substreams/ethereum-curve/.DS_Store           | Bin 0 -> 6148 bytes
 .../ethereum-curve/src/pool_factories.rs      |  25 ++++++++++++++++--
 2 files changed, 23 insertions(+), 2 deletions(-)
 create mode 100644 substreams/ethereum-curve/.DS_Store

diff --git a/substreams/ethereum-curve/.DS_Store b/substreams/ethereum-curve/.DS_Store
new file mode 100644
index 0000000000000000000000000000000000000000..2a7ee52672d117f3b56dbae20d3361206119b013
GIT binary patch
literal 6148
zcmeHKJ5Iwu5S<|@EYYN(pxi5@MHI#|C&&d*5($b#j)>?@#UVHk=YVri@#X_KGEh>$
z8)^1=XXfq7ukd(8M4P+yTx2F972HrBY#E#7%|~{WkttB^jQL{KZFe4RM%7;jj60FN
z+{$WmJ>;LAe`dB_+b%cV5|OHVzJEM<eYzNC*FR);H^Wn3K%-FsDnJFO02TPx3ZQ43
zRmXu`sQ?wA0v`(4_o2WI>%>0LKOI<n1OVD^yc?c<mH-wD0PDm)5D}OL6&O^_5krHH
zc*(jtu@4NosNalp@@CBmMg4Zfi<gVmfn2Eo6*yF265E;e|4aBc^Zy}<J1Rg0{*(eb
zZ5M5WSIXWxdO7R01%3s8H`IDL7H`EsZ^hVHD?T066?;a$PV57nj=0l-{1Gr+XjI@g
G6nFz&PbrfC

literal 0
HcmV?d00001

diff --git a/substreams/ethereum-curve/src/pool_factories.rs b/substreams/ethereum-curve/src/pool_factories.rs
index a87f40a..ab56ca0 100644
--- a/substreams/ethereum-curve/src/pool_factories.rs
+++ b/substreams/ethereum-curve/src/pool_factories.rs
@@ -18,6 +18,9 @@ const CRYPTO_SWAP_NG_FACTORY: [u8; 20] = hex!("6A8cbed756804B16E05E741eDaBd5cB54
 const TRICRYPTO_FACTORY: [u8; 20] = hex!("0c0e5f2fF0ff18a3be9b835635039256dC4B4963");
 const STABLESWAP_FACTORY: [u8; 20] = hex!("4F8846Ae9380B90d2E71D5e3D042dff3E7ebb40d");
 
+const WETH_ADDRESS: [u8; 20] = hex!("C02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2");
+const ETH_ADDRESS: [u8; 20] = hex!("EeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE");
+
 /// This trait defines some helpers for serializing and deserializing `Vec<BigInt>` which is needed
 ///  to be able to encode some of the `Attribute`s. This should also be handled by any downstream
 ///  application.
@@ -46,6 +49,20 @@ fn address_to_bytes_with_0x(address: &[u8; 20]) -> Vec<u8> {
     format!("0x{}", hex::encode(address)).into_bytes()
 }
 
+/// Function that swaps `WETH` addresses for `ETH` address for specific factory types that decide
+///  to use `WETH` address even though native `ETH` is stored. This is also extra weird bc ETH
+///  doesn't even have a real address, so we use the standard `0xEEEee...`.
+fn swap_weth_for_eth(tokens: Vec<Vec<u8>>) -> Vec<Vec<u8>> {
+    if tokens.contains(&WETH_ADDRESS.into()) {
+        tokens
+            .into_iter()
+            .map(|token| if token == WETH_ADDRESS { ETH_ADDRESS.into() } else { token })
+            .collect::<Vec<_>>()
+    } else {
+        tokens
+    }
+}
+
 /// This massive function matches factory address to specific logic to construct
 ///  `ProtocolComponent`s. While, most of the logic is readily replicable, several factories differ
 ///  in information density resulting in needing other information sources such as decoding calls
@@ -74,6 +91,8 @@ pub fn address_map(
             let pool_added =
                 abi::crypto_pool_factory::events::CryptoPoolDeployed::match_and_decode(log)?;
 
+            let tokens = swap_weth_for_eth(pool_added.coins.into());
+
             let component_id = &call.return_data[12..];
 
             Some(ProtocolComponent {
@@ -84,7 +103,7 @@ pub fn address_map(
                     hash: tx.hash.clone(),
                     index: tx.index.into(),
                 }),
-                tokens: pool_added.coins.into(),
+                tokens,
                 contracts: vec![component_id.into()],
                 static_att: vec![
                     Attribute {
@@ -449,6 +468,8 @@ pub fn address_map(
             if let Some(pool_added) =
                 abi::tricrypto_factory::events::TricryptoPoolDeployed::match_and_decode(log)
             {
+                let tokens = swap_weth_for_eth(pool_added.coins.into());
+
                 Some(ProtocolComponent {
                     id: hex::encode(&pool_added.pool),
                     tx: Some(Transaction {
@@ -457,7 +478,7 @@ pub fn address_map(
                         hash: tx.hash.clone(),
                         index: tx.index.into(),
                     }),
-                    tokens: pool_added.coins.into(),
+                    tokens,
                     contracts: vec![pool_added.pool.into()],
                     static_att: vec![
                         Attribute {