use alloy_rlp::{RlpDecodableWrapper, RlpEncodableWrapper};
use derive_more::{AsRef, Deref, DerefMut, From, IntoIterator};
use reth_codecs::{reth_codec, Compact};
#[cfg(not(feature = "std"))]
use alloc::vec::Vec;
#[doc(inline)]
pub use alloy_eips::eip4895::Withdrawal;
use serde::{Deserialize, Serialize};
#[reth_codec]
#[derive(
Debug,
Clone,
PartialEq,
Eq,
Default,
Hash,
From,
AsRef,
Deref,
DerefMut,
IntoIterator,
RlpEncodableWrapper,
RlpDecodableWrapper,
Serialize,
Deserialize,
)]
#[as_ref(forward)]
pub struct Withdrawals(Vec<Withdrawal>);
impl Withdrawals {
pub const fn new(withdrawals: Vec<Withdrawal>) -> Self {
Self(withdrawals)
}
#[inline]
pub fn total_size(&self) -> usize {
self.capacity() * core::mem::size_of::<Withdrawal>()
}
#[inline]
pub fn size(&self) -> usize {
self.len() * core::mem::size_of::<Withdrawal>()
}
pub fn iter(&self) -> core::slice::Iter<'_, Withdrawal> {
self.0.iter()
}
pub fn iter_mut(&mut self) -> core::slice::IterMut<'_, Withdrawal> {
self.0.iter_mut()
}
pub fn into_inner(self) -> Vec<Withdrawal> {
self.0
}
}
impl<'a> IntoIterator for &'a Withdrawals {
type Item = &'a Withdrawal;
type IntoIter = core::slice::Iter<'a, Withdrawal>;
fn into_iter(self) -> Self::IntoIter {
self.iter()
}
}
impl<'a> IntoIterator for &'a mut Withdrawals {
type Item = &'a mut Withdrawal;
type IntoIter = core::slice::IterMut<'a, Withdrawal>;
fn into_iter(self) -> Self::IntoIter {
self.iter_mut()
}
}
#[cfg(test)]
mod tests {
use super::*;
use alloy_primitives::Address;
use alloy_rlp::{RlpDecodable, RlpEncodable};
use proptest::proptest;
use proptest_arbitrary_interop::arb;
#[reth_codec]
#[derive(
Debug,
Clone,
PartialEq,
Eq,
Default,
Hash,
RlpEncodable,
RlpDecodable,
Serialize,
Deserialize,
)]
struct RethWithdrawal {
index: u64,
validator_index: u64,
address: Address,
amount: u64,
}
impl PartialEq<Withdrawal> for RethWithdrawal {
fn eq(&self, other: &Withdrawal) -> bool {
self.index == other.index &&
self.validator_index == other.validator_index &&
self.address == other.address &&
self.amount == other.amount
}
}
#[test]
fn test_withdrawal_serde_roundtrip() {
let input = r#"[{"index":"0x0","validatorIndex":"0x0","address":"0x0000000000000000000000000000000000001000","amount":"0x1"},{"index":"0x1","validatorIndex":"0x1","address":"0x0000000000000000000000000000000000001001","amount":"0x1"},{"index":"0x2","validatorIndex":"0x2","address":"0x0000000000000000000000000000000000001002","amount":"0x1"},{"index":"0x3","validatorIndex":"0x3","address":"0x0000000000000000000000000000000000001003","amount":"0x1"},{"index":"0x4","validatorIndex":"0x4","address":"0x0000000000000000000000000000000000001004","amount":"0x1"},{"index":"0x5","validatorIndex":"0x5","address":"0x0000000000000000000000000000000000001005","amount":"0x1"},{"index":"0x6","validatorIndex":"0x6","address":"0x0000000000000000000000000000000000001006","amount":"0x1"},{"index":"0x7","validatorIndex":"0x7","address":"0x0000000000000000000000000000000000001007","amount":"0x1"},{"index":"0x8","validatorIndex":"0x8","address":"0x0000000000000000000000000000000000001008","amount":"0x1"},{"index":"0x9","validatorIndex":"0x9","address":"0x0000000000000000000000000000000000001009","amount":"0x1"},{"index":"0xa","validatorIndex":"0xa","address":"0x000000000000000000000000000000000000100a","amount":"0x1"},{"index":"0xb","validatorIndex":"0xb","address":"0x000000000000000000000000000000000000100b","amount":"0x1"},{"index":"0xc","validatorIndex":"0xc","address":"0x000000000000000000000000000000000000100c","amount":"0x1"},{"index":"0xd","validatorIndex":"0xd","address":"0x000000000000000000000000000000000000100d","amount":"0x1"},{"index":"0xe","validatorIndex":"0xe","address":"0x000000000000000000000000000000000000100e","amount":"0x1"},{"index":"0xf","validatorIndex":"0xf","address":"0x000000000000000000000000000000000000100f","amount":"0x1"}]"#;
let withdrawals: Vec<Withdrawal> = serde_json::from_str(input).unwrap();
let s = serde_json::to_string(&withdrawals).unwrap();
assert_eq!(input, s);
}
proptest!(
#[test]
fn test_roundtrip_withdrawal_compat(withdrawal in arb::<RethWithdrawal>()) {
let mut compacted_reth_withdrawal = Vec::<u8>::new();
let len = withdrawal.to_compact(&mut compacted_reth_withdrawal);
let alloy_withdrawal = Withdrawal::from_compact(&compacted_reth_withdrawal, len).0;
assert_eq!(withdrawal, alloy_withdrawal);
let mut compacted_alloy_withdrawal = Vec::<u8>::new();
let alloy_len = alloy_withdrawal.to_compact(&mut compacted_alloy_withdrawal);
assert_eq!(len, alloy_len);
assert_eq!(compacted_reth_withdrawal, compacted_alloy_withdrawal);
}
);
}