From 18f4446344a595ede13261a254362ec6f0a1238c Mon Sep 17 00:00:00 2001
From: zizou <111426680+flopell@users.noreply.github.com>
Date: Thu, 17 Oct 2024 12:49:03 +0200
Subject: [PATCH] fix(balancer): index `PoolBalanceManaged` balance changes

---
 substreams/Cargo.lock                        |  2 +-
 substreams/ethereum-balancer/Cargo.toml      |  2 +-
 substreams/ethereum-balancer/src/modules.rs  | 11 +++++++++++
 substreams/ethereum-balancer/substreams.yaml |  2 +-
 4 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/substreams/Cargo.lock b/substreams/Cargo.lock
index 3132472..b65aee5 100644
--- a/substreams/Cargo.lock
+++ b/substreams/Cargo.lock
@@ -222,7 +222,7 @@ dependencies = [
 
 [[package]]
 name = "ethereum-balancer"
-version = "0.2.1"
+version = "0.2.2"
 dependencies = [
  "anyhow",
  "bytes",
diff --git a/substreams/ethereum-balancer/Cargo.toml b/substreams/ethereum-balancer/Cargo.toml
index 8b9ba80..0669c73 100644
--- a/substreams/ethereum-balancer/Cargo.toml
+++ b/substreams/ethereum-balancer/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "ethereum-balancer"
-version = "0.2.1"
+version = "0.2.2"
 edition = "2021"
 
 [lib]
diff --git a/substreams/ethereum-balancer/src/modules.rs b/substreams/ethereum-balancer/src/modules.rs
index d85102d..89ef7d7 100644
--- a/substreams/ethereum-balancer/src/modules.rs
+++ b/substreams/ethereum-balancer/src/modules.rs
@@ -113,6 +113,17 @@ pub fn map_relative_balances(
                         },
                     ]);
                 }
+            } else if let Some(ev) =
+                abi::vault::events::PoolBalanceManaged::match_and_decode(vault_log.log)
+            {
+                let component_id = format!("0x{}", hex::encode(ev.pool_id));
+                deltas.extend_from_slice(&[BalanceDelta {
+                    ord: vault_log.ordinal(),
+                    tx: Some(vault_log.receipt.transaction.into()),
+                    token: ev.token.to_vec(),
+                    delta: ev.cash_delta.to_signed_bytes_be(),
+                    component_id: component_id.as_bytes().to_vec(),
+                }]);
             }
 
             deltas
diff --git a/substreams/ethereum-balancer/substreams.yaml b/substreams/ethereum-balancer/substreams.yaml
index 0bbb66b..4e17b5e 100644
--- a/substreams/ethereum-balancer/substreams.yaml
+++ b/substreams/ethereum-balancer/substreams.yaml
@@ -1,7 +1,7 @@
 specVersion: v0.1.0
 package:
   name: "ethereum_balancer"
-  version: v0.2.1
+  version: v0.2.2
 
 protobuf:
   files: