use std::collections::HashMap;
use std::future::Future;
use alloy_primitives::{Address, B256, Bytes, U256};
use alloy_provider::Provider;
use r402_core::cache::{Duplicate, SettlementCache};
use r402_core::chain::ChainProvider;
use r402_core::error::VerificationError;
use r402_core::facilitator::{DynFacilitator, Facilitator, FacilitatorError};
use r402_core::scheme::{SchemeBuilder, SchemeId};
use r402_core::wire;
use r402_core::wire::UnixTimestamp;
use crate::chain::Eip155MetaTransactionProvider;
use crate::exact::types;
use crate::exact::{Eip155Exact, ExactPayload, ExactScheme};
macro_rules! traced {
($fut:expr, $span:expr) => {{
#[cfg(feature = "telemetry")]
{
use tracing::Instrument;
$fut.instrument($span).await
}
#[cfg(not(feature = "telemetry"))]
{
$fut.await
}
}};
}
macro_rules! transfer_span {
($name:expr, $call:expr $(, $key:ident = $val:expr)*) => {
tracing::info_span!($name,
from = %$call.from,
to = %$call.to,
value = %$call.value,
valid_after = %$call.valid_after,
valid_before = %$call.valid_before,
nonce = %$call.nonce,
signature = %$call.signature,
token_contract = %$call.contract_address,
$($key = $val,)*
otel.kind = "client",
)
};
}
pub(crate) mod contract;
mod error;
mod settle;
pub(crate) mod signature;
mod verify;
pub use error::Eip155ExactError;
use settle::{settle_payment, settle_permit2_payment};
pub use signature::StructuredSignatureFormatError;
pub(crate) use verify::assert_time;
use verify::{verify_payment, verify_permit2_payment};
pub use crate::exact::types::VALIDATOR_ADDRESS;
#[derive(Debug)]
pub struct Eip3009Payment {
pub from: Address,
pub to: Address,
pub value: U256,
pub valid_after: UnixTimestamp,
pub valid_before: UnixTimestamp,
pub nonce: B256,
pub signature: Bytes,
}
#[derive(Debug)]
pub struct Permit2Payment {
pub from: Address,
pub to: Address,
pub token: Address,
pub amount: U256,
pub spender: Address,
pub nonce: U256,
pub deadline: U256,
pub valid_after: U256,
pub signature: Bytes,
}
const DEFAULT_CLOCK_SKEW_TOLERANCE: u64 = crate::EVM_DEFAULT_CLOCK_SKEW_TOLERANCE_SECS;
pub struct Eip155ExactFacilitator<P> {
provider: P,
clock_skew_tolerance: u64,
settlement_cache: SettlementCache,
}
impl<P> Eip155ExactFacilitator<P> {
pub fn new(provider: P) -> Self {
Self::with_settlement_cache(provider, SettlementCache::new())
}
pub const fn with_settlement_cache(provider: P, settlement_cache: SettlementCache) -> Self {
Self {
provider,
clock_skew_tolerance: DEFAULT_CLOCK_SKEW_TOLERANCE,
settlement_cache,
}
}
#[must_use]
pub const fn with_clock_skew_tolerance(mut self, seconds: u64) -> Self {
self.clock_skew_tolerance = seconds;
self
}
fn eip3009_cache_key(&self, nonce: B256) -> String
where
P: ChainProvider,
{
format!("{}:{nonce}", self.provider.chain_id())
}
fn permit2_cache_key(&self, nonce: U256) -> String
where
P: ChainProvider,
{
format!("{}:permit2:{nonce:#x}", self.provider.chain_id())
}
}
impl<P> std::fmt::Debug for Eip155ExactFacilitator<P> {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.debug_struct("Eip155ExactFacilitator")
.finish_non_exhaustive()
}
}
impl<P> SchemeBuilder<P> for Eip155Exact
where
P: Eip155MetaTransactionProvider + ChainProvider + Send + Sync + 'static,
Eip155ExactError: From<P::Error>,
{
fn build(
&self,
provider: P,
_config: Option<serde_json::Value>,
) -> Result<Box<dyn DynFacilitator>, Box<dyn std::error::Error + Send + Sync>> {
Ok(Box::new(Eip155ExactFacilitator::new(provider)))
}
}
impl<P> Facilitator for Eip155ExactFacilitator<P>
where
P: Eip155MetaTransactionProvider + ChainProvider + Send + Sync,
P::Inner: Provider,
Eip155ExactError: From<P::Error>,
{
async fn verify(
&self,
request: wire::VerifyRequest,
) -> Result<wire::VerifyResponse, FacilitatorError> {
let request = types::v2::VerifyRequest::from_verify(request)?;
let payload = &request.payment_payload;
let requirements = &request.payment_requirements;
match &payload.payload {
ExactPayload::Eip3009(eip3009) => {
let (contract, payment, eip712_domain) = verify::assert_valid_payment(
self.provider.inner(),
self.provider.chain(),
eip3009,
payload,
requirements,
self.clock_skew_tolerance,
)
.await?;
let payer =
verify_payment(self.provider.inner(), &contract, &payment, &eip712_domain)
.await?;
Ok(wire::VerifyResponse::valid(payer.to_string()))
}
ExactPayload::Permit2(permit2) => {
let (_erc20, payment, eip712_domain) = verify::assert_valid_permit2_payment(
self.provider.inner(),
self.provider.chain(),
permit2,
payload,
requirements,
self.clock_skew_tolerance,
)
.await?;
let payer =
verify_permit2_payment(self.provider.inner(), &payment, &eip712_domain).await?;
Ok(wire::VerifyResponse::valid(payer.to_string()))
}
}
}
async fn settle(
&self,
request: wire::SettleRequest,
) -> Result<wire::SettleResponse, FacilitatorError> {
let request = types::v2::SettleRequest::from_settle(request)?;
let payload = &request.payment_payload;
let requirements = &request.payment_requirements;
let cache_key = match &payload.payload {
ExactPayload::Eip3009(eip3009) => self.eip3009_cache_key(eip3009.authorization.nonce),
ExactPayload::Permit2(permit2) => {
self.permit2_cache_key(permit2.permit2_authorization.nonce.into())
}
};
if self.settlement_cache.reserve(cache_key) == Duplicate::Yes {
return Err(VerificationError::DuplicateSettlement.into());
}
match &payload.payload {
ExactPayload::Eip3009(eip3009) => {
let (contract, payment, eip712_domain) = verify::assert_valid_payment(
self.provider.inner(),
self.provider.chain(),
eip3009,
payload,
requirements,
self.clock_skew_tolerance,
)
.await?;
let tx_hash =
settle_payment(&self.provider, &contract, &payment, &eip712_domain).await?;
Ok(wire::SettleResponse::Success {
payer: payment.from.to_string().into(),
transaction: tx_hash.to_string().into(),
network: payload.accepted.network.to_string().into(),
amount: Some(requirements.amount.0.to_string().into()),
extensions: wire::Extensions::new(),
})
}
ExactPayload::Permit2(permit2) => {
let (_erc20, payment, _eip712_domain) = verify::assert_valid_permit2_payment(
self.provider.inner(),
self.provider.chain(),
permit2,
payload,
requirements,
self.clock_skew_tolerance,
)
.await?;
let tx_hash = settle_permit2_payment(&self.provider, &payment).await?;
Ok(wire::SettleResponse::Success {
payer: payment.from.to_string().into(),
transaction: tx_hash.to_string().into(),
network: payload.accepted.network.to_string().into(),
amount: Some(requirements.amount.0.to_string().into()),
extensions: wire::Extensions::new(),
})
}
}
}
fn supported(
&self,
) -> impl Future<Output = Result<wire::SupportedResponse, FacilitatorError>> + Send {
use compact_str::CompactString;
let chain_id = self.provider.chain_id();
let kinds = vec![wire::SupportedPaymentKind::new(
wire::V2.into(),
ExactScheme.to_string(),
chain_id.to_string(),
)];
let mut signers: HashMap<CompactString, Vec<CompactString>> = HashMap::with_capacity(1);
let _ = signers.insert(
Eip155Exact.caip_family().into(),
self.provider
.signer_addresses()
.into_iter()
.map(CompactString::from)
.collect(),
);
std::future::ready(Ok(wire::SupportedResponse::new()
.with_kinds(kinds)
.with_signers(signers)))
}
}