use alloy_primitives::{Address, Bytes};
use r402_core::wire::{ExtensionEntry, Extensions};
use serde::{Deserialize, Serialize};
use thiserror::Error;
use crate::chain::TokenAmount;
pub const ERC20_APPROVAL_GAS_SPONSORING_KEY: &str = "erc20ApprovalGasSponsoring";
pub const ERC20_APPROVAL_GAS_SPONSORING_VERSION: &str = "1";
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Erc20ApprovalGasSponsoringInfo {
pub from: Address,
pub asset: Address,
pub spender: Address,
pub amount: TokenAmount,
pub signed_transaction: Bytes,
pub version: String,
}
#[derive(Debug, Error)]
pub enum Erc20ApprovalParseError {
#[error("invalid erc20ApprovalGasSponsoring extension payload: {0}")]
Invalid(#[from] serde_json::Error),
#[error("erc20ApprovalGasSponsoring.signedTransaction is empty")]
EmptyTransaction,
}
impl Erc20ApprovalGasSponsoringInfo {
pub fn from_extensions(
extensions: &Extensions,
) -> Result<Option<Self>, Erc20ApprovalParseError> {
let Some(entry) = extensions.get(ERC20_APPROVAL_GAS_SPONSORING_KEY) else {
return Ok(None);
};
let parsed: Self = match entry {
ExtensionEntry::Structured { info, .. } => serde_json::from_value(info.clone())?,
ExtensionEntry::Raw(value) => serde_json::from_value(value.clone())?,
};
if parsed.signed_transaction.is_empty() {
return Err(Erc20ApprovalParseError::EmptyTransaction);
}
Ok(Some(parsed))
}
pub fn to_extension_entry(&self) -> Result<ExtensionEntry, serde_json::Error> {
Ok(ExtensionEntry::info(serde_json::to_value(self)?))
}
}
#[cfg(test)]
mod tests {
use alloy_primitives::{Address, Bytes, U256};
use r402_core::wire::Extensions;
use super::*;
fn sample() -> Erc20ApprovalGasSponsoringInfo {
Erc20ApprovalGasSponsoringInfo {
from: Address::repeat_byte(0x11),
asset: Address::repeat_byte(0x22),
spender: Address::repeat_byte(0x33),
amount: TokenAmount::from(U256::MAX),
signed_transaction: Bytes::from(vec![0x02, 0xf8, 0x01]),
version: ERC20_APPROVAL_GAS_SPONSORING_VERSION.into(),
}
}
#[test]
fn wire_field_names() {
let v = serde_json::to_value(sample()).unwrap();
assert!(v.get("signedTransaction").is_some());
assert_eq!(v.get("version"), Some(&serde_json::json!("1")));
}
#[test]
fn round_trip() {
let info = sample();
let mut ext = Extensions::new();
ext.insert(
ERC20_APPROVAL_GAS_SPONSORING_KEY,
info.to_extension_entry().unwrap(),
);
let out = Erc20ApprovalGasSponsoringInfo::from_extensions(&ext)
.unwrap()
.unwrap();
assert_eq!(out, info);
}
}