Liquidity.Party/tycho-protocol-sdk
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(curve-substreams): miscellaneous fixes for balances extractions in Curve (#118)

Browse Source 
* fix(curve-substreams): filter out reverted calls in `emit_eth_deltas`

* feat(substreams-sdk): extract balances from `Deposit` events in `extract_balance_deltas_from_tx`

* ci: ignore fmt for abi files

* feat(sdk): also account for `Withdrawal` event from WETH

* chore: reorder mod.rs, remove unused import

* chore: cargo fmt

---------

Co-authored-by: zizou <111426680+flopell@users.noreply.github.com>
Co-authored-by: Thales <thales@datarevenue.com>
This commit is contained in:
Zizou
2024-12-06 22:21:07 +01:00
committed by GitHub
parent d766116421
commit 4d4d05203a
7 changed files with 1609 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
substreams/crates/tycho-substreams/src/abi/mod.rs
View File

@@ -1,2 +1,3 @@
#![allow(clippy::all)]
pub mod erc20;
pub mod weth;

1567
substreams/crates/tycho-substreams/src/abi/weth.rs Normal file
View File

File diff suppressed because it is too large Load Diff

35
substreams/crates/tycho-substreams/src/balances.rs
View File

@@ -172,10 +172,10 @@ pub fn aggregate_balances_changes(
/// Extracts balance deltas from a transaction trace based on a given address predicate.
///
/// This function processes the logs within a transaction trace to identify ERC-20 token transfer
/// events. It applies the given predicate to determine which addresses are of interest and extracts
/// the balance changes (deltas) for those addresses. The balance deltas are then returned as a
/// vector.
/// This function processes the logs within a transaction trace to identify ERC-20 token
/// transfer and weth specific events (Deposit and Withdrawal). It applies the given predicate to
/// determine which addresses are of interest and extracts the balance changes (deltas) for those
/// addresses. The balance deltas are then returned as a vector.
///
/// # Arguments
///
@@ -215,23 +215,30 @@ pub fn extract_balance_deltas_from_tx<F: Fn(&[u8], &[u8]) -> bool>(
tx.logs_with_calls()
.for_each(|(log, _)| {
let mut create_balance_delta = |transactor: &[u8], delta: BigInt| {
balance_deltas.push(BalanceDelta {
ord: log.ordinal,
tx: Some(tx.into()),
token: log.address.clone(),
delta: delta.to_signed_bytes_be(),
component_id: hex::encode(transactor).into(),
});
};
if let Some(transfer) = abi::erc20::events::Transfer::match_and_decode(log) {
let mut create_balance_delta = |transactor: &[u8], delta: BigInt| {
balance_deltas.push(BalanceDelta {
ord: log.ordinal,
tx: Some(tx.into()),
token: log.address.clone(),
delta: delta.to_signed_bytes_be(),
component_id: hex::encode(transactor).into(),
});
};
if address_predicate(&log.address, &transfer.from) {
create_balance_delta(&transfer.from, transfer.value.neg());
}
if address_predicate(&log.address, &transfer.to) {
create_balance_delta(&transfer.to, transfer.value);
}
} else if let Some(deposit) = abi::weth::events::Deposit::match_and_decode(log) {
if address_predicate(&log.address, &deposit.dst) {
create_balance_delta(&deposit.dst, deposit.wad);
}
} else if let Some(withdrawal) = abi::weth::events::Withdrawal::match_and_decode(log) {
if address_predicate(&log.address, &withdrawal.src) {
create_balance_delta(&withdrawal.src, withdrawal.wad.neg());
}
}
});