1234567891011121314151617use inflector::cases;18use syn::{Data, DeriveInput, Field, Fields, Ident, Variant, spanned::Spanned};19use quote::quote;2021use crate::{parse_ident_from_path, parse_ident_from_type, snake_ident_to_screaming};2223struct EventField {24 name: Ident,25 camel_name: String,26 ty: Ident,27 indexed: bool,28}2930impl EventField {31 fn try_from(field: &Field) -> syn::Result<Self> {32 let name = field.ident.as_ref().unwrap();33 let ty = parse_ident_from_type(&field.ty, false)?;34 let mut indexed = false;35 for attr in &field.attrs {36 if let Ok(ident) = parse_ident_from_path(&attr.path, false) {37 if ident == "indexed" {38 indexed = true;39 }40 }41 }42 Ok(Self {43 name: name.to_owned(),44 camel_name: cases::camelcase::to_camel_case(&name.to_string()),45 ty: ty.to_owned(),46 indexed,47 })48 }49 fn expand_solidity_argument(&self) -> proc_macro2::TokenStream {50 let camel_name = &self.camel_name;51 let ty = &self.ty;52 let indexed = self.indexed;53 quote! {54 <SolidityEventArgument<#ty>>::new(#indexed, #camel_name)55 }56 }57}5859struct Event {60 name: Ident,61 name_screaming: Ident,62 fields: Vec<EventField>,63 selector: proc_macro2::TokenStream,64}6566impl Event {67 fn try_from(variant: &Variant) -> syn::Result<Self> {68 let name = &variant.ident;69 let name_lit = proc_macro2::Literal::string(name.to_string().as_str());70 let name_screaming = snake_ident_to_screaming(name);7172 let named = match &variant.fields {73 Fields::Named(named) => named,74 _ => {75 return Err(syn::Error::new(76 variant.fields.span(),77 "expected named fields",78 ))79 }80 };81 let mut fields = Vec::new();82 for field in &named.named {83 fields.push(EventField::try_from(field)?);84 }85 if fields.iter().filter(|f| f.indexed).count() > 3 {86 return Err(syn::Error::new(87 variant.fields.span(),88 "events can have at most 4 indexed fields (1 indexed field is reserved for event signature)"89 ));90 }9192 let args = fields.iter().map(|f| {93 let ty = &f.ty;94 quote! {nameof(<#ty as ::evm_coder::abi::AbiType>::SIGNATURE) fixed(",")}95 });96 97 let shift = (!fields.is_empty()).then(|| quote! {shift_left(1)});9899 let signature = quote! { ::evm_coder::make_signature!(new fixed(#name_lit) fixed("(") #(#args)* #shift fixed(")")) };100 let selector = quote! {101 {102 let signature = #signature;103 let mut sum = ::evm_coder::sha3_const::Keccak256::new();104 let mut pos = 0;105 while pos < signature.len {106 sum = sum.update(&[signature.data[pos]; 1]);107 pos += 1;108 }109 let a = sum.finalize();110 let mut selector_bytes = [0; 32];111 let mut i = 0;112 while i != 32 {113 selector_bytes[i] = a[i];114 i += 1;115 }116 selector_bytes117 }118 };119120 Ok(Self {121 name: name.to_owned(),122 name_screaming,123 fields,124 selector,125 })126 }127128 fn expand_serializers(&self) -> proc_macro2::TokenStream {129 let name = &self.name;130 let name_screaming = &self.name_screaming;131 let fields = self.fields.iter().map(|f| &f.name);132133 let indexed = self.fields.iter().filter(|f| f.indexed).map(|f| &f.name);134 let plain = self.fields.iter().filter(|f| !f.indexed).map(|f| &f.name);135136 quote! {137 Self::#name {#(138 #fields,139 )*} => {140 topics.push(topic::from(Self::#name_screaming));141 #(142 topics.push(#indexed.to_topic());143 )*144 #(145 #plain.abi_write(&mut writer);146 )*147 }148 }149 }150151 fn expand_consts(&self) -> proc_macro2::TokenStream {152 let name_screaming = &self.name_screaming;153 let selector = &self.selector;154155 quote! {156 const #name_screaming: [u8; 32] = #selector;157 }158 }159160 fn expand_solidity_function(&self) -> proc_macro2::TokenStream {161 let name = self.name.to_string();162 let args = self.fields.iter().map(EventField::expand_solidity_argument);163 quote! {164 SolidityEvent {165 name: #name,166 args: (167 #(168 #args,169 )*170 ),171 }172 }173 }174}175176pub struct Events {177 name: Ident,178 events: Vec<Event>,179}180181impl Events {182 pub fn try_from(data: &DeriveInput) -> syn::Result<Self> {183 let name = &data.ident;184 let en = match &data.data {185 Data::Enum(en) => en,186 _ => return Err(syn::Error::new(data.span(), "expected enum")),187 };188 let mut events = Vec::new();189 for variant in &en.variants {190 events.push(Event::try_from(variant)?);191 }192 Ok(Self {193 name: name.to_owned(),194 events,195 })196 }197 pub fn expand(&self) -> proc_macro2::TokenStream {198 let name = &self.name;199200 let consts = self.events.iter().map(Event::expand_consts);201 let serializers = self.events.iter().map(Event::expand_serializers);202 let solidity_name = self.name.to_string();203 let solidity_functions = self.events.iter().map(Event::expand_solidity_function);204205 quote! {206 impl #name {207 #(208 #consts209 )*210211 212 #[cfg(feature = "stubgen")]213 pub fn generate_solidity_interface(tc: &evm_coder::solidity::TypeCollector, is_impl: bool) {214 use evm_coder::solidity::*;215 use core::fmt::Write;216 let interface = SolidityInterface {217 docs: &[],218 selector: [0; 4],219 name: #solidity_name,220 is: &[],221 functions: (#(222 #solidity_functions,223 )*),224 };225 let mut out = string::new();226 out.push_str("/// @dev inlined interface\n");227 let _ = interface.format(is_impl, &mut out, tc);228 tc.collect(out);229 }230 }231232 #[automatically_derived]233 impl ::evm_coder::events::ToLog for #name {234 fn to_log(&self, contract: Address) -> ::ethereum::Log {235 use ::evm_coder::events::ToTopic;236 use ::evm_coder::abi::AbiWrite;237 let mut writer = ::evm_coder::abi::AbiWriter::new();238 let mut topics = Vec::new();239 match self {240 #(241 #serializers,242 )*243 }244 ::ethereum::Log {245 address: contract,246 topics,247 data: writer.finish(),248 }249 }250 }251 }252 }253}