use core::ops::{Deref, DerefMut};
use alloc::vec::Vec;
use alloy_consensus::{
BlobTransactionSidecar, BlobTransactionSidecarEip7594, BlobTransactionSidecarVariant,
};
use alloy_eips::eip7702::SignedAuthorization;
use alloy_serde::WithOtherFields;
pub trait TransactionBuilder4844: Default + Sized + Send + Sync + 'static {
fn max_fee_per_blob_gas(&self) -> Option<u128>;
fn set_max_fee_per_blob_gas(&mut self, max_fee_per_blob_gas: u128);
fn with_max_fee_per_blob_gas(mut self, max_fee_per_blob_gas: u128) -> Self {
self.set_max_fee_per_blob_gas(max_fee_per_blob_gas);
self
}
fn blob_sidecar(&self) -> Option<&BlobTransactionSidecarVariant>;
fn set_blob_sidecar(&mut self, sidecar: BlobTransactionSidecarVariant);
fn with_blob_sidecar(mut self, sidecar: BlobTransactionSidecarVariant) -> Self {
self.set_blob_sidecar(sidecar);
self
}
fn blob_sidecar_4844(&self) -> Option<&BlobTransactionSidecar> {
self.blob_sidecar().and_then(|s| s.as_eip4844())
}
fn has_blob_sidecar(&self) -> bool {
self.blob_sidecar().is_some()
}
fn set_blob_sidecar_4844(&mut self, sidecar: BlobTransactionSidecar) {
self.set_blob_sidecar(BlobTransactionSidecarVariant::Eip4844(sidecar));
}
fn with_blob_sidecar_4844(mut self, sidecar: BlobTransactionSidecar) -> Self {
self.set_blob_sidecar_4844(sidecar);
self
}
fn blob_sidecar_7594(&self) -> Option<&BlobTransactionSidecarEip7594> {
self.blob_sidecar().and_then(|s| s.as_eip7594())
}
fn set_blob_sidecar_7594(&mut self, sidecar: BlobTransactionSidecarEip7594) {
self.set_blob_sidecar(BlobTransactionSidecarVariant::Eip7594(sidecar));
}
fn with_blob_sidecar_7594(mut self, sidecar: BlobTransactionSidecarEip7594) -> Self {
self.set_blob_sidecar_7594(sidecar);
self
}
}
pub trait TransactionBuilder7702: Default + Sized + Send + Sync + 'static {
fn authorization_list(&self) -> Option<&Vec<SignedAuthorization>>;
fn set_authorization_list(&mut self, authorization_list: Vec<SignedAuthorization>);
fn with_authorization_list(mut self, authorization_list: Vec<SignedAuthorization>) -> Self {
self.set_authorization_list(authorization_list);
self
}
}
impl<T> TransactionBuilder7702 for WithOtherFields<T>
where
T: TransactionBuilder7702,
{
fn authorization_list(&self) -> Option<&Vec<SignedAuthorization>> {
self.deref().authorization_list()
}
fn set_authorization_list(&mut self, authorization_list: Vec<SignedAuthorization>) {
self.deref_mut().set_authorization_list(authorization_list)
}
}
impl<T> TransactionBuilder4844 for WithOtherFields<T>
where
T: TransactionBuilder4844,
{
fn max_fee_per_blob_gas(&self) -> Option<u128> {
self.deref().max_fee_per_blob_gas()
}
fn set_max_fee_per_blob_gas(&mut self, max_fee_per_blob_gas: u128) {
self.deref_mut().set_max_fee_per_blob_gas(max_fee_per_blob_gas)
}
fn blob_sidecar(&self) -> Option<&BlobTransactionSidecarVariant> {
self.deref().blob_sidecar()
}
fn has_blob_sidecar(&self) -> bool {
self.deref().has_blob_sidecar()
}
fn set_blob_sidecar(&mut self, sidecar: BlobTransactionSidecarVariant) {
self.deref_mut().set_blob_sidecar(sidecar)
}
}