difftreelog
fix remove warning
in: master
2 files changed
crates/evm-coder/procedural/src/solidity_interface.rsdiffbeforeafterboth1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617#![allow(dead_code)]1819// NOTE: In order to understand this Rust macro better, first read this chapter20// about Procedural Macros in Rust book:21// https://doc.rust-lang.org/reference/procedural-macros.html2223use proc_macro2::TokenStream;24use quote::{quote, ToTokens, format_ident};25use inflector::cases;26use std::fmt::Write;27use syn::{28 Expr, FnArg, GenericArgument, Generics, Ident, ImplItem, ImplItemMethod, ItemImpl, Lit, Meta,29 MetaNameValue, PatType, PathArguments, ReturnType, Type,30 spanned::Spanned,31 parse::{Parse, ParseStream},32 parenthesized, Token, LitInt, LitStr,33};3435use crate::{36 fn_selector_str, parse_ident_from_pat, parse_ident_from_path, parse_path, parse_path_segment,37 parse_result_ok, pascal_ident_to_call, pascal_ident_to_snake_call, snake_ident_to_pascal,38 snake_ident_to_screaming,39};4041struct Is {42 name: Ident,43 pascal_call_name: Ident,44 snake_call_name: Ident,45 via: Option<(Type, Ident)>,46 condition: Option<Expr>,47}48impl Is {49 fn expand_call_def(&self, gen_ref: &proc_macro2::TokenStream) -> proc_macro2::TokenStream {50 let name = &self.name;51 let pascal_call_name = &self.pascal_call_name;52 quote! {53 #name(#pascal_call_name #gen_ref)54 }55 }5657 fn expand_interface_id(&self) -> proc_macro2::TokenStream {58 let pascal_call_name = &self.pascal_call_name;59 quote! {60 interface_id ^= u32::from_be_bytes(#pascal_call_name::interface_id());61 }62 }6364 fn expand_supports_interface(65 &self,66 generics: &proc_macro2::TokenStream,67 ) -> proc_macro2::TokenStream {68 let pascal_call_name = &self.pascal_call_name;69 let condition = self.condition.as_ref().map(|condition| {70 quote! {71 (#condition) &&72 }73 });74 quote! {75 #condition <#pascal_call_name #generics>::supports_interface(this, interface_id)76 }77 }7879 fn expand_variant_weight(&self) -> proc_macro2::TokenStream {80 let name = &self.name;81 quote! {82 Self::#name(call) => call.weight()83 }84 }8586 fn expand_variant_call(87 &self,88 call_name: &proc_macro2::Ident,89 generics: &proc_macro2::TokenStream,90 ) -> proc_macro2::TokenStream {91 let name = &self.name;92 let pascal_call_name = &self.pascal_call_name;93 let via_typ = self94 .via95 .as_ref()96 .map(|(t, _)| quote! {#t})97 .unwrap_or_else(|| quote! {Self});98 let via_map = self99 .via100 .as_ref()101 .map(|(_, i)| quote! {.#i()})102 .unwrap_or_default();103 let condition = self.condition.as_ref().map(|condition| {104 quote! {105 if ({let this = &self; (#condition)})106 }107 });108 quote! {109 #call_name::#name(call) #condition => return <#via_typ as ::evm_coder::Callable<#pascal_call_name #generics>>::call(self #via_map, Msg {110 call,111 caller: c.caller,112 value: c.value,113 })114 }115 }116117 fn expand_parse(&self, generics: &proc_macro2::TokenStream) -> proc_macro2::TokenStream {118 let name = &self.name;119 let pascal_call_name = &self.pascal_call_name;120 quote! {121 if let Some(parsed_call) = <#pascal_call_name #generics>::parse(method_id, reader)? {122 return Ok(Some(Self::#name(parsed_call)))123 }124 }125 }126127 fn expand_generator(&self, generics: &proc_macro2::TokenStream) -> proc_macro2::TokenStream {128 let pascal_call_name = &self.pascal_call_name;129 quote! {130 <#pascal_call_name #generics>::generate_solidity_interface(tc, is_impl);131 }132 }133134 fn expand_event_generator(&self) -> proc_macro2::TokenStream {135 let name = &self.name;136 quote! {137 #name::generate_solidity_interface(tc, is_impl);138 }139 }140}141142#[derive(Default)]143struct IsList(Vec<Is>);144impl Parse for IsList {145 fn parse(input: ParseStream) -> syn::Result<Self> {146 let mut out = vec![];147 loop {148 if input.is_empty() {149 break;150 }151 let name = input.parse::<Ident>()?;152 let lookahead = input.lookahead1();153154 let mut condition: Option<Expr> = None;155 let mut via: Option<(Type, Ident)> = None;156157 if lookahead.peek(syn::token::Paren) {158 let contents;159 parenthesized!(contents in input);160 let input = contents;161162 while !input.is_empty() {163 let lookahead = input.lookahead1();164 if lookahead.peek(Token![if]) {165 input.parse::<Token![if]>()?;166 let contents;167 parenthesized!(contents in input);168 let contents = contents.parse::<Expr>()?;169170 if condition.replace(contents).is_some() {171 return Err(syn::Error::new(input.span(), "condition is already set"));172 }173 } else if lookahead.peek(kw::via) {174 input.parse::<kw::via>()?;175 let contents;176 parenthesized!(contents in input);177178 let method = contents.parse::<Ident>()?;179 contents.parse::<kw::returns>()?;180 let ty = contents.parse::<Type>()?;181182 if via.replace((ty, method)).is_some() {183 return Err(syn::Error::new(input.span(), "via is already set"));184 }185 } else {186 return Err(lookahead.error());187 }188189 if input.peek(Token![,]) {190 input.parse::<Token![,]>()?;191 } else if !input.is_empty() {192 return Err(syn::Error::new(input.span(), "expected end"));193 }194 }195 } else if lookahead.peek(Token![,]) || input.is_empty() {196 // Pass197 } else {198 return Err(lookahead.error());199 };200 out.push(Is {201 pascal_call_name: pascal_ident_to_call(&name),202 snake_call_name: pascal_ident_to_snake_call(&name),203 name,204 via,205 condition,206 });207 if input.peek(Token![,]) {208 input.parse::<Token![,]>()?;209 continue;210 } else {211 break;212 }213 }214 Ok(Self(out))215 }216}217218pub struct InterfaceInfo {219 name: Ident,220 is: IsList,221 inline_is: IsList,222 events: IsList,223 expect_selector: Option<u32>,224}225impl Parse for InterfaceInfo {226 fn parse(input: ParseStream) -> syn::Result<Self> {227 let mut name = None;228 let mut is = None;229 let mut inline_is = None;230 let mut events = None;231 let mut expect_selector = None;232 // TODO: create proc-macro to optimize proc-macro boilerplate? :D233 loop {234 let lookahead = input.lookahead1();235 if lookahead.peek(kw::name) {236 let k = input.parse::<kw::name>()?;237 input.parse::<Token![=]>()?;238 if name.replace(input.parse::<Ident>()?).is_some() {239 return Err(syn::Error::new(k.span(), "name is already set"));240 }241 } else if lookahead.peek(kw::is) {242 let k = input.parse::<kw::is>()?;243 let contents;244 parenthesized!(contents in input);245 if is.replace(contents.parse::<IsList>()?).is_some() {246 return Err(syn::Error::new(k.span(), "is is already set"));247 }248 } else if lookahead.peek(kw::inline_is) {249 let k = input.parse::<kw::inline_is>()?;250 let contents;251 parenthesized!(contents in input);252 if inline_is.replace(contents.parse::<IsList>()?).is_some() {253 return Err(syn::Error::new(k.span(), "inline_is is already set"));254 }255 } else if lookahead.peek(kw::events) {256 let k = input.parse::<kw::events>()?;257 let contents;258 parenthesized!(contents in input);259 if events.replace(contents.parse::<IsList>()?).is_some() {260 return Err(syn::Error::new(k.span(), "events is already set"));261 }262 } else if lookahead.peek(kw::expect_selector) {263 let k = input.parse::<kw::expect_selector>()?;264 input.parse::<Token![=]>()?;265 let value = input.parse::<LitInt>()?;266 if expect_selector267 .replace(value.base10_parse::<u32>()?)268 .is_some()269 {270 return Err(syn::Error::new(k.span(), "expect_selector is already set"));271 }272 } else if input.is_empty() {273 break;274 } else {275 return Err(lookahead.error());276 }277 if input.peek(Token![,]) {278 input.parse::<Token![,]>()?;279 } else {280 break;281 }282 }283 Ok(Self {284 name: name.ok_or_else(|| syn::Error::new(input.span(), "missing name"))?,285 is: is.unwrap_or_default(),286 inline_is: inline_is.unwrap_or_default(),287 events: events.unwrap_or_default(),288 expect_selector,289 })290 }291}292293struct MethodInfo {294 rename_selector: Option<String>,295 hide: bool,296}297impl Parse for MethodInfo {298 fn parse(input: ParseStream) -> syn::Result<Self> {299 let mut rename_selector = None;300 let mut hide = false;301 while !input.is_empty() {302 let lookahead = input.lookahead1();303 if lookahead.peek(kw::rename_selector) {304 let k = input.parse::<kw::rename_selector>()?;305 input.parse::<Token![=]>()?;306 if rename_selector307 .replace(input.parse::<LitStr>()?.value())308 .is_some()309 {310 return Err(syn::Error::new(k.span(), "rename_selector is already set"));311 }312 } else if lookahead.peek(kw::hide) {313 input.parse::<kw::hide>()?;314 hide = true;315 } else {316 return Err(lookahead.error());317 }318319 if input.peek(Token![,]) {320 input.parse::<Token![,]>()?;321 } else if !input.is_empty() {322 return Err(syn::Error::new(input.span(), "expected end"));323 }324 }325 Ok(Self {326 rename_selector,327 hide,328 })329 }330}331332#[derive(Debug)]333enum AbiType {334 // type335 Plain(Ident),336 // (type1,type2)337 Tuple(Vec<AbiType>),338 // type[]339 Vec(Box<AbiType>),340 // type[20]341 Array(Box<AbiType>, usize),342}343impl AbiType {344 fn try_from(value: &Type) -> syn::Result<Self> {345 let value = Self::try_maybe_special_from(value)?;346 if value.is_special() {347 return Err(syn::Error::new(value.span(), "unexpected special type"));348 }349 Ok(value)350 }351 fn try_maybe_special_from(value: &Type) -> syn::Result<Self> {352 match value {353 Type::Array(arr) => {354 let wrapped = AbiType::try_from(&arr.elem)?;355 match &arr.len {356 Expr::Lit(l) => match &l.lit {357 Lit::Int(i) => {358 let num = i.base10_parse::<usize>()?;359 Ok(AbiType::Array(Box::new(wrapped), num as usize))360 }361 _ => Err(syn::Error::new(arr.len.span(), "should be int literal")),362 },363 _ => Err(syn::Error::new(arr.len.span(), "should be literal")),364 }365 }366 Type::Path(_) => {367 let path = parse_path(value)?;368 let segment = parse_path_segment(path)?;369 if segment.ident == "Vec" {370 let args = match &segment.arguments {371 PathArguments::AngleBracketed(e) => e,372 _ => {373 return Err(syn::Error::new(374 segment.arguments.span(),375 "missing Vec generic",376 ))377 }378 };379 let args = &args.args;380 if args.len() != 1 {381 return Err(syn::Error::new(382 args.span(),383 "expected only one generic for vec",384 ));385 }386 let arg = args.first().expect("first arg");387388 let ty = match arg {389 GenericArgument::Type(ty) => ty,390 _ => {391 return Err(syn::Error::new(392 arg.span(),393 "expected first generic to be type",394 ))395 }396 };397398 let wrapped = AbiType::try_from(ty)?;399 Ok(Self::Vec(Box::new(wrapped)))400 } else {401 if !segment.arguments.is_empty() {402 return Err(syn::Error::new(403 segment.arguments.span(),404 "unexpected generic arguments for non-vec type",405 ));406 }407 Ok(Self::Plain(segment.ident.clone()))408 }409 }410 Type::Tuple(t) => {411 let mut out = Vec::with_capacity(t.elems.len());412 for el in t.elems.iter() {413 out.push(AbiType::try_from(el)?)414 }415 Ok(Self::Tuple(out))416 }417 _ => Err(syn::Error::new(418 value.span(),419 "unexpected type, only arrays, plain types and tuples are supported",420 )),421 }422 }423 fn is_value(&self) -> bool {424 matches!(self, Self::Plain(v) if v == "value")425 }426 fn is_caller(&self) -> bool {427 matches!(self, Self::Plain(v) if v == "caller")428 }429 fn is_special(&self) -> bool {430 self.is_caller() || self.is_value()431 }432 fn selector_ty_buf(&self, buf: &mut String) -> std::fmt::Result {433 match self {434 AbiType::Plain(t) => {435 write!(buf, "{}", t)436 }437 AbiType::Tuple(t) => {438 write!(buf, "(")?;439 for (i, t) in t.iter().enumerate() {440 if i != 0 {441 write!(buf, ",")?;442 }443 t.selector_ty_buf(buf)?;444 }445 write!(buf, ")")446 }447 AbiType::Vec(v) => {448 v.selector_ty_buf(buf)?;449 write!(buf, "[]")450 }451 AbiType::Array(v, len) => {452 v.selector_ty_buf(buf)?;453 write!(buf, "[{}]", len)454 }455 }456 }457 fn selector_ty(&self) -> String {458 let mut out = String::new();459 self.selector_ty_buf(&mut out).expect("no fmt error");460 out461 }462}463impl ToTokens for AbiType {464 fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {465 match self {466 AbiType::Plain(t) => tokens.extend(quote! {#t}),467 AbiType::Tuple(t) => {468 tokens.extend(quote! {(469 #(#t),*470 )});471 }472 AbiType::Vec(v) => tokens.extend(quote! {Vec<#v>}),473 AbiType::Array(v, l) => tokens.extend(quote! {[#v; #l]}),474 }475 }476}477478#[derive(Debug)]479struct MethodArg {480 name: Ident,481 camel_name: String,482 ty: AbiType,483}484impl MethodArg {485 fn try_from(value: &PatType) -> syn::Result<Self> {486 let name = parse_ident_from_pat(&value.pat)?.clone();487 Ok(Self {488 camel_name: cases::camelcase::to_camel_case(&name.to_string()),489 name,490 ty: AbiType::try_maybe_special_from(&value.ty)?,491 })492 }493 fn is_value(&self) -> bool {494 self.ty.is_value()495 }496 fn is_caller(&self) -> bool {497 self.ty.is_caller()498 }499 fn is_special(&self) -> bool {500 self.ty.is_special()501 }502 fn selector_ty(&self) -> String {503 assert!(!self.is_special());504 self.ty.selector_ty()505 }506507 fn expand_call_def(&self) -> proc_macro2::TokenStream {508 assert!(!self.is_special());509 let name = &self.name;510 let ty = &self.ty;511512 quote! {513 #name: #ty514 }515 }516517 fn expand_parse(&self) -> proc_macro2::TokenStream {518 assert!(!self.is_special());519 let name = &self.name;520 quote! {521 #name: reader.abi_read()?522 }523 }524525 fn expand_call_arg(&self) -> proc_macro2::TokenStream {526 if self.is_value() {527 quote! {528 c.value.clone()529 }530 } else if self.is_caller() {531 quote! {532 c.caller.clone()533 }534 } else {535 let name = &self.name;536 quote! {537 #name538 }539 }540 }541542 fn expand_solidity_argument(&self) -> proc_macro2::TokenStream {543 let camel_name = &self.camel_name.to_string();544 let ty = &self.ty;545 quote! {546 <NamedArgument<#ty>>::new(#camel_name)547 }548 }549}550551#[derive(PartialEq)]552enum Mutability {553 Mutable,554 View,555 Pure,556}557558/// Group all keywords for this macro. Usage example:559/// #[solidity_interface(name = "B", inline_is(A))]560mod kw {561 syn::custom_keyword!(weight);562563 syn::custom_keyword!(via);564 syn::custom_keyword!(returns);565 syn::custom_keyword!(name);566 syn::custom_keyword!(is);567 syn::custom_keyword!(inline_is);568 syn::custom_keyword!(events);569 syn::custom_keyword!(expect_selector);570571 syn::custom_keyword!(rename_selector);572 syn::custom_keyword!(hide);573}574575/// Rust methods are parsed into this structure when Solidity code is generated576struct Method {577 name: Ident,578 camel_name: String,579 pascal_name: Ident,580 screaming_name: Ident,581 selector_str: String,582 selector: u32,583 hide: bool,584 args: Vec<MethodArg>,585 has_normal_args: bool,586 has_value_args: bool,587 mutability: Mutability,588 result: Type,589 weight: Option<Expr>,590 docs: Vec<String>,591}592impl Method {593 fn try_from(value: &ImplItemMethod) -> syn::Result<Self> {594 let mut info = MethodInfo {595 rename_selector: None,596 hide: false,597 };598 let mut docs = Vec::new();599 let mut weight = None;600 for attr in &value.attrs {601 let ident = parse_ident_from_path(&attr.path, false)?;602 if ident == "solidity" {603 info = attr.parse_args::<MethodInfo>()?;604 } else if ident == "doc" {605 let args = attr.parse_meta().unwrap();606 let value = match args {607 Meta::NameValue(MetaNameValue {608 lit: Lit::Str(str), ..609 }) => str.value(),610 _ => unreachable!(),611 };612 docs.push(value);613 } else if ident == "weight" {614 weight = Some(attr.parse_args::<Expr>()?);615 }616 }617 let ident = &value.sig.ident;618 let ident_str = ident.to_string();619 if !cases::snakecase::is_snake_case(&ident_str) {620 return Err(syn::Error::new(ident.span(), "method name should be snake_cased\nif alternative solidity name needs to be set - use #[solidity] attribute"));621 }622623 let mut mutability = Mutability::Pure;624625 if let Some(FnArg::Receiver(receiver)) = value626 .sig627 .inputs628 .iter()629 .find(|arg| matches!(arg, FnArg::Receiver(_)))630 {631 if receiver.reference.is_none() {632 return Err(syn::Error::new(633 receiver.span(),634 "receiver should be by ref",635 ));636 }637 if receiver.mutability.is_some() {638 mutability = Mutability::Mutable;639 } else {640 mutability = Mutability::View;641 }642 }643 let mut args = Vec::new();644 for typ in value645 .sig646 .inputs647 .iter()648 .filter(|arg| matches!(arg, FnArg::Typed(_)))649 {650 let typ = match typ {651 FnArg::Typed(typ) => typ,652 _ => unreachable!(),653 };654 args.push(MethodArg::try_from(typ)?);655 }656657 if mutability != Mutability::Mutable && args.iter().any(|arg| arg.is_value()) {658 return Err(syn::Error::new(659 args.iter().find(|arg| arg.is_value()).unwrap().ty.span(),660 "payable function should be mutable",661 ));662 }663664 let result = match &value.sig.output {665 ReturnType::Type(_, ty) => ty,666 _ => return Err(syn::Error::new(value.sig.output.span(), "interface method should return Result<value>\nif there is no value to return - specify void (which is alias to unit)")),667 };668 let result = parse_result_ok(result)?;669670 let camel_name = info671 .rename_selector672 .unwrap_or_else(|| cases::camelcase::to_camel_case(&ident.to_string()));673 let mut selector_str = camel_name.clone();674 selector_str.push('(');675 let mut has_normal_args = false;676 for (i, arg) in args.iter().filter(|arg| !arg.is_special()).enumerate() {677 if i != 0 {678 selector_str.push(',');679 }680 write!(selector_str, "{}", arg.selector_ty()).unwrap();681 has_normal_args = true;682 }683 let has_value_args = args.iter().any(|a| a.is_value());684 selector_str.push(')');685 let selector = fn_selector_str(&selector_str);686687 Ok(Self {688 name: ident.clone(),689 camel_name,690 pascal_name: snake_ident_to_pascal(ident),691 screaming_name: snake_ident_to_screaming(ident),692 selector_str,693 selector,694 hide: info.hide,695 args,696 has_normal_args,697 has_value_args,698 mutability,699 result: result.clone(),700 weight,701 docs,702 })703 }704 fn expand_call_def(&self) -> proc_macro2::TokenStream {705 let defs = self706 .args707 .iter()708 .filter(|a| !a.is_special())709 .map(|a| a.expand_call_def());710 let pascal_name = &self.pascal_name;711 let docs = &self.docs;712713 if self.has_normal_args {714 quote! {715 #(#[doc = #docs])*716 #[allow(missing_docs)]717 #pascal_name {718 #(719 #defs,720 )*721 }722 }723 } else {724 quote! {#pascal_name}725 }726 }727728 fn expand_const(&self) -> proc_macro2::TokenStream {729 let screaming_name = &self.screaming_name;730 let screaming_name_signature = format_ident!("{}_SIGNATURE", &self.screaming_name);731 let selector_str = &self.selector_str;732 let custom_signature = self.expand_custom_signature();733 quote! {734 const #screaming_name_signature: ::evm_coder::custom_signature::FunctionSignature = #custom_signature;735 #[doc = #selector_str]736 const #screaming_name: ::evm_coder::types::bytes4 = {737 let mut data = [0_u8; Self::#screaming_name_signature.unit.len];738 let mut pos = 0;739 while pos < Self::#screaming_name_signature.unit.len {740 data[pos] = Self::#screaming_name_signature.unit.data[pos];741 pos += 1;742 }743 let a = ::evm_coder::sha3_const::Keccak256::new()744 .update(&data)745 .finalize();746 [a[0], a[1], a[2], a[3]]747 };748 }749 }750751 fn expand_interface_id(&self) -> proc_macro2::TokenStream {752 let screaming_name = &self.screaming_name;753 quote! {754 interface_id ^= u32::from_be_bytes(Self::#screaming_name);755 }756 }757758 fn expand_parse(&self) -> proc_macro2::TokenStream {759 let pascal_name = &self.pascal_name;760 let screaming_name = &self.screaming_name;761 if self.has_normal_args {762 let parsers = self763 .args764 .iter()765 .filter(|a| !a.is_special())766 .map(|a| a.expand_parse());767 quote! {768 Self::#screaming_name => return Ok(Some(Self::#pascal_name {769 #(770 #parsers,771 )*772 }))773 }774 } else {775 quote! { Self::#screaming_name => return Ok(Some(Self::#pascal_name)) }776 }777 }778779 fn expand_variant_call(&self, call_name: &proc_macro2::Ident) -> proc_macro2::TokenStream {780 let pascal_name = &self.pascal_name;781 let name = &self.name;782783 let matcher = if self.has_normal_args {784 let names = self785 .args786 .iter()787 .filter(|a| !a.is_special())788 .map(|a| &a.name);789790 quote! {{791 #(792 #names,793 )*794 }}795 } else {796 quote! {}797 };798799 let receiver = match self.mutability {800 Mutability::Mutable | Mutability::View => quote! {self.},801 Mutability::Pure => quote! {Self::},802 };803 let args = self.args.iter().map(|a| a.expand_call_arg());804805 quote! {806 #call_name::#pascal_name #matcher => {807 let result = #receiver #name(808 #(809 #args,810 )*811 )?;812 (&result).to_result()813 }814 }815 }816817 fn expand_variant_weight(&self) -> proc_macro2::TokenStream {818 let pascal_name = &self.pascal_name;819 if let Some(weight) = &self.weight {820 let matcher = if self.has_normal_args {821 let names = self822 .args823 .iter()824 .filter(|a| !a.is_special())825 .map(|a| &a.name);826827 quote! {{828 #(829 #names,830 )*831 }}832 } else {833 quote! {}834 };835 quote! {836 Self::#pascal_name #matcher => (#weight).into()837 }838 } else {839 let matcher = if self.has_normal_args {840 quote! {{..}}841 } else {842 quote! {}843 };844 quote! {845 Self::#pascal_name #matcher => ().into()846 }847 }848 }849850 fn expand_type(851 ty: &AbiType,852 token_stream: &mut proc_macro2::TokenStream,853 read_signature: bool,854 ) {855 match ty {856 AbiType::Plain(ref ident) => {857 let plain_token = if read_signature {858 quote! {859 (<#ident>::SIGNATURE)860 }861 } else {862 quote! {863 #ident864 }865 };866867 token_stream.extend(plain_token);868 }869870 AbiType::Tuple(ref tuple_type) => {871 let mut tuple_types = proc_macro2::TokenStream::new();872 let mut is_first = true;873874 for ty in tuple_type {875 if is_first {876 is_first = false877 } else {878 tuple_types.extend(quote!(,));879 }880 Self::expand_type(ty, &mut tuple_types, false);881 }882 tuple_types = if read_signature {883 quote! { (<(#tuple_types)>::SIGNATURE) }884 } else {885 quote! { (#tuple_types) }886 };887 token_stream.extend(tuple_types);888 }889890 AbiType::Vec(ref vec_type) => {891 let mut vec_token = proc_macro2::TokenStream::new();892 Self::expand_type(vec_type.as_ref(), &mut vec_token, false);893 vec_token = if read_signature {894 quote! { (<Vec<#vec_token>>::SIGNATURE) }895 } else {896 quote! { <Vec<#vec_token>> }897 };898 token_stream.extend(vec_token);899 }900901 AbiType::Array(_, _) => todo!("Array eth signature"),902 };903 }904905 fn expand_custom_signature(&self) -> proc_macro2::TokenStream {906 let mut token_stream = TokenStream::new();907908 let mut is_first = true;909 for arg in &self.args {910 if arg.is_special() {911 continue;912 }913914 if is_first {915 is_first = false;916 } else {917 token_stream.extend(quote!(,));918 }919 Self::expand_type(&arg.ty, &mut token_stream, true);920 }921922 if !is_first {923 token_stream.extend(quote!(,));924 }925926 let func_name = self.camel_name.clone();927 let func_name = quote!(SignaturePreferences {928 open_name: Some(SignatureUnit::new(#func_name)),929 open_delimiter: Some(SignatureUnit::new("(")),930 param_delimiter: Some(SignatureUnit::new(",")),931 close_delimiter: Some(SignatureUnit::new(")")),932 close_name: None,933 });934 quote!({ ::evm_coder::make_signature!(new fn(#func_name), #token_stream) })935 }936937 fn expand_solidity_function(&self) -> proc_macro2::TokenStream {938 let camel_name = &self.camel_name;939 let mutability = match self.mutability {940 Mutability::Mutable => quote! {SolidityMutability::Mutable},941 Mutability::View => quote! { SolidityMutability::View },942 Mutability::Pure => quote! {SolidityMutability::Pure},943 };944 let result = &self.result;945946 let args = self947 .args948 .iter()949 .filter(|a| !a.is_special())950 .map(MethodArg::expand_solidity_argument);951 let docs = &self.docs;952 let selector_str = &self.selector_str;953 let screaming_name = &self.screaming_name;954 let hide = self.hide;955 let custom_signature = self.expand_custom_signature();956 let custom_signature = quote!(957 {958 const cs: FunctionSignature = #custom_signature;959 cs960 }961 );962 let is_payable = self.has_value_args;963964 quote! {965 SolidityFunction {966 docs: &[#(#docs),*],967 selector_str: #selector_str,968 hide: #hide,969 selector: u32::from_be_bytes(Self::#screaming_name),970 custom_signature: #custom_signature,971 name: #camel_name,972 mutability: #mutability,973 is_payable: #is_payable,974 args: (975 #(976 #args,977 )*978 ),979 result: <UnnamedArgument<#result>>::default(),980 }981 }982 }983}984985fn generics_list(gen: &Generics) -> proc_macro2::TokenStream {986 if gen.params.is_empty() {987 return quote! {};988 }989 let params = gen.params.iter().map(|p| match p {990 syn::GenericParam::Type(id) => {991 let v = &id.ident;992 quote! {#v}993 }994 syn::GenericParam::Lifetime(lt) => {995 let v = <.lifetime;996 quote! {#v}997 }998 syn::GenericParam::Const(c) => {999 let i = &c.ident;1000 quote! {#i}1001 }1002 });1003 quote! { #(#params),* }1004}1005fn generics_reference(gen: &Generics) -> proc_macro2::TokenStream {1006 if gen.params.is_empty() {1007 return quote! {};1008 }1009 let list = generics_list(gen);1010 quote! { <#list> }1011}1012fn generics_data(gen: &Generics) -> proc_macro2::TokenStream {1013 let list = generics_list(gen);1014 if gen.params.len() == 1 {1015 quote! {#list}1016 } else {1017 quote! { (#list) }1018 }1019}10201021pub struct SolidityInterface {1022 generics: Generics,1023 name: Box<syn::Type>,1024 info: InterfaceInfo,1025 methods: Vec<Method>,1026 docs: Vec<String>,1027}1028impl SolidityInterface {1029 pub fn try_from(info: InterfaceInfo, value: &ItemImpl) -> syn::Result<Self> {1030 let mut methods = Vec::new();10311032 for item in &value.items {1033 if let ImplItem::Method(method) = item {1034 methods.push(Method::try_from(method)?)1035 }1036 }1037 let mut docs = vec![];1038 for attr in &value.attrs {1039 let ident = parse_ident_from_path(&attr.path, false)?;1040 if ident == "doc" {1041 let args = attr.parse_meta().unwrap();1042 let value = match args {1043 Meta::NameValue(MetaNameValue {1044 lit: Lit::Str(str), ..1045 }) => str.value(),1046 _ => unreachable!(),1047 };1048 docs.push(value);1049 }1050 }1051 Ok(Self {1052 generics: value.generics.clone(),1053 name: value.self_ty.clone(),1054 info,1055 methods,1056 docs,1057 })1058 }1059 pub fn expand(self) -> proc_macro2::TokenStream {1060 let name = self.name;10611062 let solidity_name = self.info.name.to_string();1063 let call_name = pascal_ident_to_call(&self.info.name);1064 let generics = self.generics;1065 let gen_ref = generics_reference(&generics);1066 let gen_data = generics_data(&generics);1067 let gen_where = &generics.where_clause;10681069 let call_sub = self1070 .info1071 .inline_is1072 .01073 .iter()1074 .chain(self.info.is.0.iter())1075 .map(|c| Is::expand_call_def(c, &gen_ref));1076 let call_parse = self1077 .info1078 .inline_is1079 .01080 .iter()1081 .chain(self.info.is.0.iter())1082 .map(|is| Is::expand_parse(is, &gen_ref));1083 let call_variants = self1084 .info1085 .inline_is1086 .01087 .iter()1088 .chain(self.info.is.0.iter())1089 .map(|c| Is::expand_variant_call(c, &call_name, &gen_ref));1090 let weight_variants = self1091 .info1092 .inline_is1093 .01094 .iter()1095 .chain(self.info.is.0.iter())1096 .map(Is::expand_variant_weight);10971098 let inline_interface_id = self.info.inline_is.0.iter().map(Is::expand_interface_id);1099 let supports_interface = self1100 .info1101 .is1102 .01103 .iter()1104 .map(|is| Is::expand_supports_interface(is, &gen_ref));11051106 let calls = self.methods.iter().map(Method::expand_call_def);1107 let consts = self.methods.iter().map(Method::expand_const);1108 let interface_id = self.methods.iter().map(Method::expand_interface_id);1109 let parsers = self.methods.iter().map(Method::expand_parse);1110 let call_variants_this = self1111 .methods1112 .iter()1113 .map(|m| Method::expand_variant_call(m, &call_name));1114 let weight_variants_this = self.methods.iter().map(Method::expand_variant_weight);1115 let solidity_functions = self.methods.iter().map(Method::expand_solidity_function);11161117 // TODO: Inline inline_is1118 let solidity_is = self1119 .info1120 .is1121 .01122 .iter()1123 .chain(self.info.inline_is.0.iter())1124 .map(|is| is.name.to_string());1125 let solidity_events_is = self.info.events.0.iter().map(|is| is.name.to_string());1126 let solidity_generators = self1127 .info1128 .is1129 .01130 .iter()1131 .chain(self.info.inline_is.0.iter())1132 .map(|is| Is::expand_generator(is, &gen_ref));1133 let solidity_event_generators = self.info.events.0.iter().map(Is::expand_event_generator);11341135 let docs = &self.docs;11361137 if let Some(expect_selector) = &self.info.expect_selector {1138 if !self.info.inline_is.0.is_empty() {1139 return syn::Error::new(1140 name.span(),1141 "expect_selector is not compatible with inline_is",1142 )1143 .to_compile_error();1144 }1145 let selector = self1146 .methods1147 .iter()1148 .map(|m| m.selector)1149 .fold(0, |a, b| a ^ b);11501151 if *expect_selector != selector {1152 let mut methods = String::new();1153 for meth in self.methods.iter() {1154 write!(methods, "\n- {}", meth.selector_str).expect("write to string");1155 }1156 return syn::Error::new(name.span(), format!("expected selector mismatch, expected {expect_selector:0>8x}, but implementation has {selector:0>8x}{methods}")).to_compile_error();1157 }1158 }1159 // let methods = self.methods.iter().map(Method::solidity_def);11601161 quote! {1162 #[derive(Debug)]1163 #(#[doc = #docs])*1164 pub enum #call_name #gen_ref {1165 /// Inherited method1166 ERC165Call(::evm_coder::ERC165Call, ::core::marker::PhantomData<#gen_data>),1167 #(1168 #calls,1169 )*1170 #(1171 #call_sub,1172 )*1173 }1174 impl #gen_ref #call_name #gen_ref {1175 #(1176 #consts1177 )*1178 /// Return this call ERC165 selector1179 pub fn interface_id() -> ::evm_coder::types::bytes4 {1180 let mut interface_id = 0;1181 #(#interface_id)*1182 #(#inline_interface_id)*1183 u32::to_be_bytes(interface_id)1184 }1185 /// Generate solidity definitions for methods described in this interface1186 pub fn generate_solidity_interface(tc: &evm_coder::solidity::TypeCollector, is_impl: bool) {1187 use evm_coder::solidity::*;1188 use core::fmt::Write;1189 let interface = SolidityInterface {1190 docs: &[#(#docs),*],1191 name: #solidity_name,1192 selector: Self::interface_id(),1193 is: &["Dummy", "ERC165", #(1194 #solidity_is,1195 )* #(1196 #solidity_events_is,1197 )* ],1198 functions: (#(1199 #solidity_functions,1200 )*),1201 };12021203 let mut out = ::evm_coder::types::string::new();1204 if #solidity_name.starts_with("Inline") {1205 out.push_str("/// @dev inlined interface\n");1206 }1207 let _ = interface.format(is_impl, &mut out, tc);1208 tc.collect(out);1209 #(1210 #solidity_event_generators1211 )*1212 #(1213 #solidity_generators1214 )*1215 if is_impl {1216 tc.collect("/// @dev common stubs holder\ncontract Dummy {\n\tuint8 dummy;\n\tstring stub_error = \"this contract is implemented in native\";\n}\ncontract ERC165 is Dummy {\n\tfunction supportsInterface(bytes4 interfaceID) external view returns (bool) {\n\t\trequire(false, stub_error);\n\t\tinterfaceID;\n\t\treturn true;\n\t}\n}\n".into());1217 } else {1218 tc.collect("/// @dev common stubs holder\ninterface Dummy {\n}\ninterface ERC165 is Dummy {\n\tfunction supportsInterface(bytes4 interfaceID) external view returns (bool);\n}\n".into());1219 }1220 }1221 }1222 impl #gen_ref ::evm_coder::Call for #call_name #gen_ref {1223 fn parse(method_id: ::evm_coder::types::bytes4, reader: &mut ::evm_coder::abi::AbiReader) -> ::evm_coder::execution::Result<Option<Self>> {1224 use ::evm_coder::abi::AbiRead;1225 match method_id {1226 ::evm_coder::ERC165Call::INTERFACE_ID => return Ok(1227 ::evm_coder::ERC165Call::parse(method_id, reader)?1228 .map(|c| Self::ERC165Call(c, ::core::marker::PhantomData))1229 ),1230 #(1231 #parsers,1232 )*1233 _ => {},1234 }1235 #(1236 #call_parse1237 )else*1238 return Ok(None);1239 }1240 }1241 impl #generics #call_name #gen_ref1242 #gen_where1243 {1244 /// Is this contract implements specified ERC165 selector1245 pub fn supports_interface(this: &#name, interface_id: ::evm_coder::types::bytes4) -> bool {1246 interface_id != u32::to_be_bytes(0xffffff) && (1247 interface_id == ::evm_coder::ERC165Call::INTERFACE_ID ||1248 interface_id == Self::interface_id()1249 #(1250 || #supports_interface1251 )*1252 )1253 }1254 }1255 impl #generics ::evm_coder::Weighted for #call_name #gen_ref1256 #gen_where1257 {1258 #[allow(unused_variables)]1259 fn weight(&self) -> ::evm_coder::execution::DispatchInfo {1260 match self {1261 #(1262 #weight_variants,1263 )*1264 // TODO: It should be very cheap, but not free1265 Self::ERC165Call(::evm_coder::ERC165Call::SupportsInterface {..}, _) => ::frame_support::weights::Weight::from_ref_time(100).into(),1266 #(1267 #weight_variants_this,1268 )*1269 }1270 }1271 }1272 impl #generics ::evm_coder::Callable<#call_name #gen_ref> for #name1273 #gen_where1274 {1275 #[allow(unreachable_code)] // In case of no inner calls1276 fn call(&mut self, c: Msg<#call_name #gen_ref>) -> ::evm_coder::execution::ResultWithPostInfo<::evm_coder::abi::AbiWriter> {1277 use ::evm_coder::abi::AbiWrite;1278 match c.call {1279 #(1280 #call_variants,1281 )*1282 #call_name::ERC165Call(::evm_coder::ERC165Call::SupportsInterface {interface_id}, _) => {1283 let mut writer = ::evm_coder::abi::AbiWriter::default();1284 writer.bool(&<#call_name #gen_ref>::supports_interface(self, interface_id));1285 return Ok(writer.into());1286 }1287 _ => {},1288 }1289 let mut writer = ::evm_coder::abi::AbiWriter::default();1290 match c.call {1291 #(1292 #call_variants_this,1293 )*1294 _ => Err(::evm_coder::execution::Error::from("method is not available").into()),1295 }1296 }1297 }1298 }1299 }1300}1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617#![allow(dead_code)]1819// NOTE: In order to understand this Rust macro better, first read this chapter20// about Procedural Macros in Rust book:21// https://doc.rust-lang.org/reference/procedural-macros.html2223use proc_macro2::TokenStream;24use quote::{quote, ToTokens, format_ident};25use inflector::cases;26use std::fmt::Write;27use syn::{28 Expr, FnArg, GenericArgument, Generics, Ident, ImplItem, ImplItemMethod, ItemImpl, Lit, Meta,29 MetaNameValue, PatType, PathArguments, ReturnType, Type,30 spanned::Spanned,31 parse::{Parse, ParseStream},32 parenthesized, Token, LitInt, LitStr,33};3435use crate::{36 fn_selector_str, parse_ident_from_pat, parse_ident_from_path, parse_path, parse_path_segment,37 parse_result_ok, pascal_ident_to_call, pascal_ident_to_snake_call, snake_ident_to_pascal,38 snake_ident_to_screaming,39};4041struct Is {42 name: Ident,43 pascal_call_name: Ident,44 snake_call_name: Ident,45 via: Option<(Type, Ident)>,46 condition: Option<Expr>,47}48impl Is {49 fn expand_call_def(&self, gen_ref: &proc_macro2::TokenStream) -> proc_macro2::TokenStream {50 let name = &self.name;51 let pascal_call_name = &self.pascal_call_name;52 quote! {53 #name(#pascal_call_name #gen_ref)54 }55 }5657 fn expand_interface_id(&self) -> proc_macro2::TokenStream {58 let pascal_call_name = &self.pascal_call_name;59 quote! {60 interface_id ^= u32::from_be_bytes(#pascal_call_name::interface_id());61 }62 }6364 fn expand_supports_interface(65 &self,66 generics: &proc_macro2::TokenStream,67 ) -> proc_macro2::TokenStream {68 let pascal_call_name = &self.pascal_call_name;69 let condition = self.condition.as_ref().map(|condition| {70 quote! {71 (#condition) &&72 }73 });74 quote! {75 #condition <#pascal_call_name #generics>::supports_interface(this, interface_id)76 }77 }7879 fn expand_variant_weight(&self) -> proc_macro2::TokenStream {80 let name = &self.name;81 quote! {82 Self::#name(call) => call.weight()83 }84 }8586 fn expand_variant_call(87 &self,88 call_name: &proc_macro2::Ident,89 generics: &proc_macro2::TokenStream,90 ) -> proc_macro2::TokenStream {91 let name = &self.name;92 let pascal_call_name = &self.pascal_call_name;93 let via_typ = self94 .via95 .as_ref()96 .map(|(t, _)| quote! {#t})97 .unwrap_or_else(|| quote! {Self});98 let via_map = self99 .via100 .as_ref()101 .map(|(_, i)| quote! {.#i()})102 .unwrap_or_default();103 let condition = self.condition.as_ref().map(|condition| {104 quote! {105 if ({let this = &self; (#condition)})106 }107 });108 quote! {109 #call_name::#name(call) #condition => return <#via_typ as ::evm_coder::Callable<#pascal_call_name #generics>>::call(self #via_map, Msg {110 call,111 caller: c.caller,112 value: c.value,113 })114 }115 }116117 fn expand_parse(&self, generics: &proc_macro2::TokenStream) -> proc_macro2::TokenStream {118 let name = &self.name;119 let pascal_call_name = &self.pascal_call_name;120 quote! {121 if let Some(parsed_call) = <#pascal_call_name #generics>::parse(method_id, reader)? {122 return Ok(Some(Self::#name(parsed_call)))123 }124 }125 }126127 fn expand_generator(&self, generics: &proc_macro2::TokenStream) -> proc_macro2::TokenStream {128 let pascal_call_name = &self.pascal_call_name;129 quote! {130 <#pascal_call_name #generics>::generate_solidity_interface(tc, is_impl);131 }132 }133134 fn expand_event_generator(&self) -> proc_macro2::TokenStream {135 let name = &self.name;136 quote! {137 #name::generate_solidity_interface(tc, is_impl);138 }139 }140}141142#[derive(Default)]143struct IsList(Vec<Is>);144impl Parse for IsList {145 fn parse(input: ParseStream) -> syn::Result<Self> {146 let mut out = vec![];147 loop {148 if input.is_empty() {149 break;150 }151 let name = input.parse::<Ident>()?;152 let lookahead = input.lookahead1();153154 let mut condition: Option<Expr> = None;155 let mut via: Option<(Type, Ident)> = None;156157 if lookahead.peek(syn::token::Paren) {158 let contents;159 parenthesized!(contents in input);160 let input = contents;161162 while !input.is_empty() {163 let lookahead = input.lookahead1();164 if lookahead.peek(Token![if]) {165 input.parse::<Token![if]>()?;166 let contents;167 parenthesized!(contents in input);168 let contents = contents.parse::<Expr>()?;169170 if condition.replace(contents).is_some() {171 return Err(syn::Error::new(input.span(), "condition is already set"));172 }173 } else if lookahead.peek(kw::via) {174 input.parse::<kw::via>()?;175 let contents;176 parenthesized!(contents in input);177178 let method = contents.parse::<Ident>()?;179 contents.parse::<kw::returns>()?;180 let ty = contents.parse::<Type>()?;181182 if via.replace((ty, method)).is_some() {183 return Err(syn::Error::new(input.span(), "via is already set"));184 }185 } else {186 return Err(lookahead.error());187 }188189 if input.peek(Token![,]) {190 input.parse::<Token![,]>()?;191 } else if !input.is_empty() {192 return Err(syn::Error::new(input.span(), "expected end"));193 }194 }195 } else if lookahead.peek(Token![,]) || input.is_empty() {196 // Pass197 } else {198 return Err(lookahead.error());199 };200 out.push(Is {201 pascal_call_name: pascal_ident_to_call(&name),202 snake_call_name: pascal_ident_to_snake_call(&name),203 name,204 via,205 condition,206 });207 if input.peek(Token![,]) {208 input.parse::<Token![,]>()?;209 continue;210 } else {211 break;212 }213 }214 Ok(Self(out))215 }216}217218pub struct InterfaceInfo {219 name: Ident,220 is: IsList,221 inline_is: IsList,222 events: IsList,223 expect_selector: Option<u32>,224}225impl Parse for InterfaceInfo {226 fn parse(input: ParseStream) -> syn::Result<Self> {227 let mut name = None;228 let mut is = None;229 let mut inline_is = None;230 let mut events = None;231 let mut expect_selector = None;232 // TODO: create proc-macro to optimize proc-macro boilerplate? :D233 loop {234 let lookahead = input.lookahead1();235 if lookahead.peek(kw::name) {236 let k = input.parse::<kw::name>()?;237 input.parse::<Token![=]>()?;238 if name.replace(input.parse::<Ident>()?).is_some() {239 return Err(syn::Error::new(k.span(), "name is already set"));240 }241 } else if lookahead.peek(kw::is) {242 let k = input.parse::<kw::is>()?;243 let contents;244 parenthesized!(contents in input);245 if is.replace(contents.parse::<IsList>()?).is_some() {246 return Err(syn::Error::new(k.span(), "is is already set"));247 }248 } else if lookahead.peek(kw::inline_is) {249 let k = input.parse::<kw::inline_is>()?;250 let contents;251 parenthesized!(contents in input);252 if inline_is.replace(contents.parse::<IsList>()?).is_some() {253 return Err(syn::Error::new(k.span(), "inline_is is already set"));254 }255 } else if lookahead.peek(kw::events) {256 let k = input.parse::<kw::events>()?;257 let contents;258 parenthesized!(contents in input);259 if events.replace(contents.parse::<IsList>()?).is_some() {260 return Err(syn::Error::new(k.span(), "events is already set"));261 }262 } else if lookahead.peek(kw::expect_selector) {263 let k = input.parse::<kw::expect_selector>()?;264 input.parse::<Token![=]>()?;265 let value = input.parse::<LitInt>()?;266 if expect_selector267 .replace(value.base10_parse::<u32>()?)268 .is_some()269 {270 return Err(syn::Error::new(k.span(), "expect_selector is already set"));271 }272 } else if input.is_empty() {273 break;274 } else {275 return Err(lookahead.error());276 }277 if input.peek(Token![,]) {278 input.parse::<Token![,]>()?;279 } else {280 break;281 }282 }283 Ok(Self {284 name: name.ok_or_else(|| syn::Error::new(input.span(), "missing name"))?,285 is: is.unwrap_or_default(),286 inline_is: inline_is.unwrap_or_default(),287 events: events.unwrap_or_default(),288 expect_selector,289 })290 }291}292293struct MethodInfo {294 rename_selector: Option<String>,295 hide: bool,296}297impl Parse for MethodInfo {298 fn parse(input: ParseStream) -> syn::Result<Self> {299 let mut rename_selector = None;300 let mut hide = false;301 while !input.is_empty() {302 let lookahead = input.lookahead1();303 if lookahead.peek(kw::rename_selector) {304 let k = input.parse::<kw::rename_selector>()?;305 input.parse::<Token![=]>()?;306 if rename_selector307 .replace(input.parse::<LitStr>()?.value())308 .is_some()309 {310 return Err(syn::Error::new(k.span(), "rename_selector is already set"));311 }312 } else if lookahead.peek(kw::hide) {313 input.parse::<kw::hide>()?;314 hide = true;315 } else {316 return Err(lookahead.error());317 }318319 if input.peek(Token![,]) {320 input.parse::<Token![,]>()?;321 } else if !input.is_empty() {322 return Err(syn::Error::new(input.span(), "expected end"));323 }324 }325 Ok(Self {326 rename_selector,327 hide,328 })329 }330}331332#[derive(Debug)]333enum AbiType {334 // type335 Plain(Ident),336 // (type1,type2)337 Tuple(Vec<AbiType>),338 // type[]339 Vec(Box<AbiType>),340 // type[20]341 Array(Box<AbiType>, usize),342}343impl AbiType {344 fn try_from(value: &Type) -> syn::Result<Self> {345 let value = Self::try_maybe_special_from(value)?;346 if value.is_special() {347 return Err(syn::Error::new(value.span(), "unexpected special type"));348 }349 Ok(value)350 }351 fn try_maybe_special_from(value: &Type) -> syn::Result<Self> {352 match value {353 Type::Array(arr) => {354 let wrapped = AbiType::try_from(&arr.elem)?;355 match &arr.len {356 Expr::Lit(l) => match &l.lit {357 Lit::Int(i) => {358 let num = i.base10_parse::<usize>()?;359 Ok(AbiType::Array(Box::new(wrapped), num as usize))360 }361 _ => Err(syn::Error::new(arr.len.span(), "should be int literal")),362 },363 _ => Err(syn::Error::new(arr.len.span(), "should be literal")),364 }365 }366 Type::Path(_) => {367 let path = parse_path(value)?;368 let segment = parse_path_segment(path)?;369 if segment.ident == "Vec" {370 let args = match &segment.arguments {371 PathArguments::AngleBracketed(e) => e,372 _ => {373 return Err(syn::Error::new(374 segment.arguments.span(),375 "missing Vec generic",376 ))377 }378 };379 let args = &args.args;380 if args.len() != 1 {381 return Err(syn::Error::new(382 args.span(),383 "expected only one generic for vec",384 ));385 }386 let arg = args.first().expect("first arg");387388 let ty = match arg {389 GenericArgument::Type(ty) => ty,390 _ => {391 return Err(syn::Error::new(392 arg.span(),393 "expected first generic to be type",394 ))395 }396 };397398 let wrapped = AbiType::try_from(ty)?;399 Ok(Self::Vec(Box::new(wrapped)))400 } else {401 if !segment.arguments.is_empty() {402 return Err(syn::Error::new(403 segment.arguments.span(),404 "unexpected generic arguments for non-vec type",405 ));406 }407 Ok(Self::Plain(segment.ident.clone()))408 }409 }410 Type::Tuple(t) => {411 let mut out = Vec::with_capacity(t.elems.len());412 for el in t.elems.iter() {413 out.push(AbiType::try_from(el)?)414 }415 Ok(Self::Tuple(out))416 }417 _ => Err(syn::Error::new(418 value.span(),419 "unexpected type, only arrays, plain types and tuples are supported",420 )),421 }422 }423 fn is_value(&self) -> bool {424 matches!(self, Self::Plain(v) if v == "value")425 }426 fn is_caller(&self) -> bool {427 matches!(self, Self::Plain(v) if v == "caller")428 }429 fn is_special(&self) -> bool {430 self.is_caller() || self.is_value()431 }432 fn selector_ty_buf(&self, buf: &mut String) -> std::fmt::Result {433 match self {434 AbiType::Plain(t) => {435 write!(buf, "{}", t)436 }437 AbiType::Tuple(t) => {438 write!(buf, "(")?;439 for (i, t) in t.iter().enumerate() {440 if i != 0 {441 write!(buf, ",")?;442 }443 t.selector_ty_buf(buf)?;444 }445 write!(buf, ")")446 }447 AbiType::Vec(v) => {448 v.selector_ty_buf(buf)?;449 write!(buf, "[]")450 }451 AbiType::Array(v, len) => {452 v.selector_ty_buf(buf)?;453 write!(buf, "[{}]", len)454 }455 }456 }457 fn selector_ty(&self) -> String {458 let mut out = String::new();459 self.selector_ty_buf(&mut out).expect("no fmt error");460 out461 }462}463impl ToTokens for AbiType {464 fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {465 match self {466 AbiType::Plain(t) => tokens.extend(quote! {#t}),467 AbiType::Tuple(t) => {468 tokens.extend(quote! {(469 #(#t),*470 )});471 }472 AbiType::Vec(v) => tokens.extend(quote! {Vec<#v>}),473 AbiType::Array(v, l) => tokens.extend(quote! {[#v; #l]}),474 }475 }476}477478#[derive(Debug)]479struct MethodArg {480 name: Ident,481 camel_name: String,482 ty: AbiType,483}484impl MethodArg {485 fn try_from(value: &PatType) -> syn::Result<Self> {486 let name = parse_ident_from_pat(&value.pat)?.clone();487 Ok(Self {488 camel_name: cases::camelcase::to_camel_case(&name.to_string()),489 name,490 ty: AbiType::try_maybe_special_from(&value.ty)?,491 })492 }493 fn is_value(&self) -> bool {494 self.ty.is_value()495 }496 fn is_caller(&self) -> bool {497 self.ty.is_caller()498 }499 fn is_special(&self) -> bool {500 self.ty.is_special()501 }502 fn selector_ty(&self) -> String {503 assert!(!self.is_special());504 self.ty.selector_ty()505 }506507 fn expand_call_def(&self) -> proc_macro2::TokenStream {508 assert!(!self.is_special());509 let name = &self.name;510 let ty = &self.ty;511512 quote! {513 #name: #ty514 }515 }516517 fn expand_parse(&self) -> proc_macro2::TokenStream {518 assert!(!self.is_special());519 let name = &self.name;520 quote! {521 #name: reader.abi_read()?522 }523 }524525 fn expand_call_arg(&self) -> proc_macro2::TokenStream {526 if self.is_value() {527 quote! {528 c.value.clone()529 }530 } else if self.is_caller() {531 quote! {532 c.caller.clone()533 }534 } else {535 let name = &self.name;536 quote! {537 #name538 }539 }540 }541542 fn expand_solidity_argument(&self) -> proc_macro2::TokenStream {543 let camel_name = &self.camel_name.to_string();544 let ty = &self.ty;545 quote! {546 <NamedArgument<#ty>>::new(#camel_name)547 }548 }549}550551#[derive(PartialEq)]552enum Mutability {553 Mutable,554 View,555 Pure,556}557558/// Group all keywords for this macro. Usage example:559/// #[solidity_interface(name = "B", inline_is(A))]560mod kw {561 syn::custom_keyword!(weight);562563 syn::custom_keyword!(via);564 syn::custom_keyword!(returns);565 syn::custom_keyword!(name);566 syn::custom_keyword!(is);567 syn::custom_keyword!(inline_is);568 syn::custom_keyword!(events);569 syn::custom_keyword!(expect_selector);570571 syn::custom_keyword!(rename_selector);572 syn::custom_keyword!(hide);573}574575/// Rust methods are parsed into this structure when Solidity code is generated576struct Method {577 name: Ident,578 camel_name: String,579 pascal_name: Ident,580 screaming_name: Ident,581 selector_str: String,582 selector: u32,583 hide: bool,584 args: Vec<MethodArg>,585 has_normal_args: bool,586 has_value_args: bool,587 mutability: Mutability,588 result: Type,589 weight: Option<Expr>,590 docs: Vec<String>,591}592impl Method {593 fn try_from(value: &ImplItemMethod) -> syn::Result<Self> {594 let mut info = MethodInfo {595 rename_selector: None,596 hide: false,597 };598 let mut docs = Vec::new();599 let mut weight = None;600 for attr in &value.attrs {601 let ident = parse_ident_from_path(&attr.path, false)?;602 if ident == "solidity" {603 info = attr.parse_args::<MethodInfo>()?;604 } else if ident == "doc" {605 let args = attr.parse_meta().unwrap();606 let value = match args {607 Meta::NameValue(MetaNameValue {608 lit: Lit::Str(str), ..609 }) => str.value(),610 _ => unreachable!(),611 };612 docs.push(value);613 } else if ident == "weight" {614 weight = Some(attr.parse_args::<Expr>()?);615 }616 }617 let ident = &value.sig.ident;618 let ident_str = ident.to_string();619 if !cases::snakecase::is_snake_case(&ident_str) {620 return Err(syn::Error::new(ident.span(), "method name should be snake_cased\nif alternative solidity name needs to be set - use #[solidity] attribute"));621 }622623 let mut mutability = Mutability::Pure;624625 if let Some(FnArg::Receiver(receiver)) = value626 .sig627 .inputs628 .iter()629 .find(|arg| matches!(arg, FnArg::Receiver(_)))630 {631 if receiver.reference.is_none() {632 return Err(syn::Error::new(633 receiver.span(),634 "receiver should be by ref",635 ));636 }637 if receiver.mutability.is_some() {638 mutability = Mutability::Mutable;639 } else {640 mutability = Mutability::View;641 }642 }643 let mut args = Vec::new();644 for typ in value645 .sig646 .inputs647 .iter()648 .filter(|arg| matches!(arg, FnArg::Typed(_)))649 {650 let typ = match typ {651 FnArg::Typed(typ) => typ,652 _ => unreachable!(),653 };654 args.push(MethodArg::try_from(typ)?);655 }656657 if mutability != Mutability::Mutable && args.iter().any(|arg| arg.is_value()) {658 return Err(syn::Error::new(659 args.iter().find(|arg| arg.is_value()).unwrap().ty.span(),660 "payable function should be mutable",661 ));662 }663664 let result = match &value.sig.output {665 ReturnType::Type(_, ty) => ty,666 _ => return Err(syn::Error::new(value.sig.output.span(), "interface method should return Result<value>\nif there is no value to return - specify void (which is alias to unit)")),667 };668 let result = parse_result_ok(result)?;669670 let camel_name = info671 .rename_selector672 .unwrap_or_else(|| cases::camelcase::to_camel_case(&ident.to_string()));673 let mut selector_str = camel_name.clone();674 selector_str.push('(');675 let mut has_normal_args = false;676 for (i, arg) in args.iter().filter(|arg| !arg.is_special()).enumerate() {677 if i != 0 {678 selector_str.push(',');679 }680 write!(selector_str, "{}", arg.selector_ty()).unwrap();681 has_normal_args = true;682 }683 let has_value_args = args.iter().any(|a| a.is_value());684 selector_str.push(')');685 let selector = fn_selector_str(&selector_str);686687 Ok(Self {688 name: ident.clone(),689 camel_name,690 pascal_name: snake_ident_to_pascal(ident),691 screaming_name: snake_ident_to_screaming(ident),692 selector_str,693 selector,694 hide: info.hide,695 args,696 has_normal_args,697 has_value_args,698 mutability,699 result: result.clone(),700 weight,701 docs,702 })703 }704 fn expand_call_def(&self) -> proc_macro2::TokenStream {705 let defs = self706 .args707 .iter()708 .filter(|a| !a.is_special())709 .map(|a| a.expand_call_def());710 let pascal_name = &self.pascal_name;711 let docs = &self.docs;712713 if self.has_normal_args {714 quote! {715 #(#[doc = #docs])*716 #[allow(missing_docs)]717 #pascal_name {718 #(719 #defs,720 )*721 }722 }723 } else {724 quote! {#pascal_name}725 }726 }727728 fn expand_const(&self) -> proc_macro2::TokenStream {729 let screaming_name = &self.screaming_name;730 let screaming_name_signature = format_ident!("{}_SIGNATURE", &self.screaming_name);731 let selector_str = &self.selector_str;732 let custom_signature = self.expand_custom_signature();733 quote! {734 #[doc = #selector_str]735 const #screaming_name_signature: ::evm_coder::custom_signature::FunctionSignature = #custom_signature;736 const #screaming_name: ::evm_coder::types::bytes4 = {737 let mut sum = ::evm_coder::sha3_const::Keccak256::new();738 let mut pos = 0;739 while pos < Self::#screaming_name_signature.unit.len {740 sum = sum.update(&[Self::#screaming_name_signature.unit.data[pos]; 1]);741 pos += 1;742 }743 let a = sum.finalize();744 [a[0], a[1], a[2], a[3]]745 };746 }747 }748749 fn expand_interface_id(&self) -> proc_macro2::TokenStream {750 let screaming_name = &self.screaming_name;751 quote! {752 interface_id ^= u32::from_be_bytes(Self::#screaming_name);753 }754 }755756 fn expand_parse(&self) -> proc_macro2::TokenStream {757 let pascal_name = &self.pascal_name;758 let screaming_name = &self.screaming_name;759 if self.has_normal_args {760 let parsers = self761 .args762 .iter()763 .filter(|a| !a.is_special())764 .map(|a| a.expand_parse());765 quote! {766 Self::#screaming_name => return Ok(Some(Self::#pascal_name {767 #(768 #parsers,769 )*770 }))771 }772 } else {773 quote! { Self::#screaming_name => return Ok(Some(Self::#pascal_name)) }774 }775 }776777 fn expand_variant_call(&self, call_name: &proc_macro2::Ident) -> proc_macro2::TokenStream {778 let pascal_name = &self.pascal_name;779 let name = &self.name;780781 let matcher = if self.has_normal_args {782 let names = self783 .args784 .iter()785 .filter(|a| !a.is_special())786 .map(|a| &a.name);787788 quote! {{789 #(790 #names,791 )*792 }}793 } else {794 quote! {}795 };796797 let receiver = match self.mutability {798 Mutability::Mutable | Mutability::View => quote! {self.},799 Mutability::Pure => quote! {Self::},800 };801 let args = self.args.iter().map(|a| a.expand_call_arg());802803 quote! {804 #call_name::#pascal_name #matcher => {805 let result = #receiver #name(806 #(807 #args,808 )*809 )?;810 (&result).to_result()811 }812 }813 }814815 fn expand_variant_weight(&self) -> proc_macro2::TokenStream {816 let pascal_name = &self.pascal_name;817 if let Some(weight) = &self.weight {818 let matcher = if self.has_normal_args {819 let names = self820 .args821 .iter()822 .filter(|a| !a.is_special())823 .map(|a| &a.name);824825 quote! {{826 #(827 #names,828 )*829 }}830 } else {831 quote! {}832 };833 quote! {834 Self::#pascal_name #matcher => (#weight).into()835 }836 } else {837 let matcher = if self.has_normal_args {838 quote! {{..}}839 } else {840 quote! {}841 };842 quote! {843 Self::#pascal_name #matcher => ().into()844 }845 }846 }847848 fn expand_type(849 ty: &AbiType,850 token_stream: &mut proc_macro2::TokenStream,851 read_signature: bool,852 ) {853 match ty {854 AbiType::Plain(ref ident) => {855 let plain_token = if read_signature {856 quote! {857 (<#ident>::SIGNATURE)858 }859 } else {860 quote! {861 #ident862 }863 };864865 token_stream.extend(plain_token);866 }867868 AbiType::Tuple(ref tuple_type) => {869 let mut tuple_types = proc_macro2::TokenStream::new();870 let mut is_first = true;871872 for ty in tuple_type {873 if is_first {874 is_first = false875 } else {876 tuple_types.extend(quote!(,));877 }878 Self::expand_type(ty, &mut tuple_types, false);879 }880 tuple_types = if read_signature {881 quote! { (<(#tuple_types)>::SIGNATURE) }882 } else {883 quote! { (#tuple_types) }884 };885 token_stream.extend(tuple_types);886 }887888 AbiType::Vec(ref vec_type) => {889 let mut vec_token = proc_macro2::TokenStream::new();890 Self::expand_type(vec_type.as_ref(), &mut vec_token, false);891 vec_token = if read_signature {892 quote! { (<Vec<#vec_token>>::SIGNATURE) }893 } else {894 quote! { <Vec<#vec_token>> }895 };896 token_stream.extend(vec_token);897 }898899 AbiType::Array(_, _) => todo!("Array eth signature"),900 };901 }902903 fn expand_custom_signature(&self) -> proc_macro2::TokenStream {904 let mut token_stream = TokenStream::new();905906 let mut is_first = true;907 for arg in &self.args {908 if arg.is_special() {909 continue;910 }911912 if is_first {913 is_first = false;914 } else {915 token_stream.extend(quote!(,));916 }917 Self::expand_type(&arg.ty, &mut token_stream, true);918 }919920 if !is_first {921 token_stream.extend(quote!(,));922 }923924 let func_name = self.camel_name.clone();925 let func_name = quote!(SignaturePreferences {926 open_name: Some(SignatureUnit::new(#func_name)),927 open_delimiter: Some(SignatureUnit::new("(")),928 param_delimiter: Some(SignatureUnit::new(",")),929 close_delimiter: Some(SignatureUnit::new(")")),930 close_name: None,931 });932 quote!({ ::evm_coder::make_signature!(new fn(#func_name), #token_stream) })933 }934935 fn expand_solidity_function(&self) -> proc_macro2::TokenStream {936 let camel_name = &self.camel_name;937 let mutability = match self.mutability {938 Mutability::Mutable => quote! {SolidityMutability::Mutable},939 Mutability::View => quote! { SolidityMutability::View },940 Mutability::Pure => quote! {SolidityMutability::Pure},941 };942 let result = &self.result;943944 let args = self945 .args946 .iter()947 .filter(|a| !a.is_special())948 .map(MethodArg::expand_solidity_argument);949 let docs = &self.docs;950 let selector_str = &self.selector_str;951 let screaming_name = &self.screaming_name;952 let hide = self.hide;953 let custom_signature = self.expand_custom_signature();954 let custom_signature = quote!(955 {956 const cs: FunctionSignature = #custom_signature;957 cs958 }959 );960 let is_payable = self.has_value_args;961962 quote! {963 SolidityFunction {964 docs: &[#(#docs),*],965 selector_str: #selector_str,966 hide: #hide,967 selector: u32::from_be_bytes(Self::#screaming_name),968 custom_signature: #custom_signature,969 name: #camel_name,970 mutability: #mutability,971 is_payable: #is_payable,972 args: (973 #(974 #args,975 )*976 ),977 result: <UnnamedArgument<#result>>::default(),978 }979 }980 }981}982983fn generics_list(gen: &Generics) -> proc_macro2::TokenStream {984 if gen.params.is_empty() {985 return quote! {};986 }987 let params = gen.params.iter().map(|p| match p {988 syn::GenericParam::Type(id) => {989 let v = &id.ident;990 quote! {#v}991 }992 syn::GenericParam::Lifetime(lt) => {993 let v = <.lifetime;994 quote! {#v}995 }996 syn::GenericParam::Const(c) => {997 let i = &c.ident;998 quote! {#i}999 }1000 });1001 quote! { #(#params),* }1002}1003fn generics_reference(gen: &Generics) -> proc_macro2::TokenStream {1004 if gen.params.is_empty() {1005 return quote! {};1006 }1007 let list = generics_list(gen);1008 quote! { <#list> }1009}1010fn generics_data(gen: &Generics) -> proc_macro2::TokenStream {1011 let list = generics_list(gen);1012 if gen.params.len() == 1 {1013 quote! {#list}1014 } else {1015 quote! { (#list) }1016 }1017}10181019pub struct SolidityInterface {1020 generics: Generics,1021 name: Box<syn::Type>,1022 info: InterfaceInfo,1023 methods: Vec<Method>,1024 docs: Vec<String>,1025}1026impl SolidityInterface {1027 pub fn try_from(info: InterfaceInfo, value: &ItemImpl) -> syn::Result<Self> {1028 let mut methods = Vec::new();10291030 for item in &value.items {1031 if let ImplItem::Method(method) = item {1032 methods.push(Method::try_from(method)?)1033 }1034 }1035 let mut docs = vec![];1036 for attr in &value.attrs {1037 let ident = parse_ident_from_path(&attr.path, false)?;1038 if ident == "doc" {1039 let args = attr.parse_meta().unwrap();1040 let value = match args {1041 Meta::NameValue(MetaNameValue {1042 lit: Lit::Str(str), ..1043 }) => str.value(),1044 _ => unreachable!(),1045 };1046 docs.push(value);1047 }1048 }1049 Ok(Self {1050 generics: value.generics.clone(),1051 name: value.self_ty.clone(),1052 info,1053 methods,1054 docs,1055 })1056 }1057 pub fn expand(self) -> proc_macro2::TokenStream {1058 let name = self.name;10591060 let solidity_name = self.info.name.to_string();1061 let call_name = pascal_ident_to_call(&self.info.name);1062 let generics = self.generics;1063 let gen_ref = generics_reference(&generics);1064 let gen_data = generics_data(&generics);1065 let gen_where = &generics.where_clause;10661067 let call_sub = self1068 .info1069 .inline_is1070 .01071 .iter()1072 .chain(self.info.is.0.iter())1073 .map(|c| Is::expand_call_def(c, &gen_ref));1074 let call_parse = self1075 .info1076 .inline_is1077 .01078 .iter()1079 .chain(self.info.is.0.iter())1080 .map(|is| Is::expand_parse(is, &gen_ref));1081 let call_variants = self1082 .info1083 .inline_is1084 .01085 .iter()1086 .chain(self.info.is.0.iter())1087 .map(|c| Is::expand_variant_call(c, &call_name, &gen_ref));1088 let weight_variants = self1089 .info1090 .inline_is1091 .01092 .iter()1093 .chain(self.info.is.0.iter())1094 .map(Is::expand_variant_weight);10951096 let inline_interface_id = self.info.inline_is.0.iter().map(Is::expand_interface_id);1097 let supports_interface = self1098 .info1099 .is1100 .01101 .iter()1102 .map(|is| Is::expand_supports_interface(is, &gen_ref));11031104 let calls = self.methods.iter().map(Method::expand_call_def);1105 let consts = self.methods.iter().map(Method::expand_const);1106 let interface_id = self.methods.iter().map(Method::expand_interface_id);1107 let parsers = self.methods.iter().map(Method::expand_parse);1108 let call_variants_this = self1109 .methods1110 .iter()1111 .map(|m| Method::expand_variant_call(m, &call_name));1112 let weight_variants_this = self.methods.iter().map(Method::expand_variant_weight);1113 let solidity_functions = self.methods.iter().map(Method::expand_solidity_function);11141115 // TODO: Inline inline_is1116 let solidity_is = self1117 .info1118 .is1119 .01120 .iter()1121 .chain(self.info.inline_is.0.iter())1122 .map(|is| is.name.to_string());1123 let solidity_events_is = self.info.events.0.iter().map(|is| is.name.to_string());1124 let solidity_generators = self1125 .info1126 .is1127 .01128 .iter()1129 .chain(self.info.inline_is.0.iter())1130 .map(|is| Is::expand_generator(is, &gen_ref));1131 let solidity_event_generators = self.info.events.0.iter().map(Is::expand_event_generator);11321133 let docs = &self.docs;11341135 quote! {1136 #[derive(Debug)]1137 #(#[doc = #docs])*1138 pub enum #call_name #gen_ref {1139 /// Inherited method1140 ERC165Call(::evm_coder::ERC165Call, ::core::marker::PhantomData<#gen_data>),1141 #(1142 #calls,1143 )*1144 #(1145 #call_sub,1146 )*1147 }1148 impl #gen_ref #call_name #gen_ref {1149 #(1150 #consts1151 )*1152 /// Return this call ERC165 selector1153 pub fn interface_id() -> ::evm_coder::types::bytes4 {1154 let mut interface_id = 0;1155 #(#interface_id)*1156 #(#inline_interface_id)*1157 u32::to_be_bytes(interface_id)1158 }1159 /// Generate solidity definitions for methods described in this interface1160 pub fn generate_solidity_interface(tc: &evm_coder::solidity::TypeCollector, is_impl: bool) {1161 use evm_coder::solidity::*;1162 use core::fmt::Write;1163 let interface = SolidityInterface {1164 docs: &[#(#docs),*],1165 name: #solidity_name,1166 selector: Self::interface_id(),1167 is: &["Dummy", "ERC165", #(1168 #solidity_is,1169 )* #(1170 #solidity_events_is,1171 )* ],1172 functions: (#(1173 #solidity_functions,1174 )*),1175 };11761177 let mut out = ::evm_coder::types::string::new();1178 if #solidity_name.starts_with("Inline") {1179 out.push_str("/// @dev inlined interface\n");1180 }1181 let _ = interface.format(is_impl, &mut out, tc);1182 tc.collect(out);1183 #(1184 #solidity_event_generators1185 )*1186 #(1187 #solidity_generators1188 )*1189 if is_impl {1190 tc.collect("/// @dev common stubs holder\ncontract Dummy {\n\tuint8 dummy;\n\tstring stub_error = \"this contract is implemented in native\";\n}\ncontract ERC165 is Dummy {\n\tfunction supportsInterface(bytes4 interfaceID) external view returns (bool) {\n\t\trequire(false, stub_error);\n\t\tinterfaceID;\n\t\treturn true;\n\t}\n}\n".into());1191 } else {1192 tc.collect("/// @dev common stubs holder\ninterface Dummy {\n}\ninterface ERC165 is Dummy {\n\tfunction supportsInterface(bytes4 interfaceID) external view returns (bool);\n}\n".into());1193 }1194 }1195 }1196 impl #gen_ref ::evm_coder::Call for #call_name #gen_ref {1197 fn parse(method_id: ::evm_coder::types::bytes4, reader: &mut ::evm_coder::abi::AbiReader) -> ::evm_coder::execution::Result<Option<Self>> {1198 use ::evm_coder::abi::AbiRead;1199 match method_id {1200 ::evm_coder::ERC165Call::INTERFACE_ID => return Ok(1201 ::evm_coder::ERC165Call::parse(method_id, reader)?1202 .map(|c| Self::ERC165Call(c, ::core::marker::PhantomData))1203 ),1204 #(1205 #parsers,1206 )*1207 _ => {},1208 }1209 #(1210 #call_parse1211 )else*1212 return Ok(None);1213 }1214 }1215 impl #generics #call_name #gen_ref1216 #gen_where1217 {1218 /// Is this contract implements specified ERC165 selector1219 pub fn supports_interface(this: &#name, interface_id: ::evm_coder::types::bytes4) -> bool {1220 interface_id != u32::to_be_bytes(0xffffff) && (1221 interface_id == ::evm_coder::ERC165Call::INTERFACE_ID ||1222 interface_id == Self::interface_id()1223 #(1224 || #supports_interface1225 )*1226 )1227 }1228 }1229 impl #generics ::evm_coder::Weighted for #call_name #gen_ref1230 #gen_where1231 {1232 #[allow(unused_variables)]1233 fn weight(&self) -> ::evm_coder::execution::DispatchInfo {1234 match self {1235 #(1236 #weight_variants,1237 )*1238 // TODO: It should be very cheap, but not free1239 Self::ERC165Call(::evm_coder::ERC165Call::SupportsInterface {..}, _) => ::frame_support::weights::Weight::from_ref_time(100).into(),1240 #(1241 #weight_variants_this,1242 )*1243 }1244 }1245 }1246 impl #generics ::evm_coder::Callable<#call_name #gen_ref> for #name1247 #gen_where1248 {1249 #[allow(unreachable_code)] // In case of no inner calls1250 fn call(&mut self, c: Msg<#call_name #gen_ref>) -> ::evm_coder::execution::ResultWithPostInfo<::evm_coder::abi::AbiWriter> {1251 use ::evm_coder::abi::AbiWrite;1252 match c.call {1253 #(1254 #call_variants,1255 )*1256 #call_name::ERC165Call(::evm_coder::ERC165Call::SupportsInterface {interface_id}, _) => {1257 let mut writer = ::evm_coder::abi::AbiWriter::default();1258 writer.bool(&<#call_name #gen_ref>::supports_interface(self, interface_id));1259 return Ok(writer.into());1260 }1261 _ => {},1262 }1263 let mut writer = ::evm_coder::abi::AbiWriter::default();1264 match c.call {1265 #(1266 #call_variants_this,1267 )*1268 _ => Err(::evm_coder::execution::Error::from("method is not available").into()),1269 }1270 }1271 }1272 }1273 }1274}crates/evm-coder/src/custom_signature.rsdiffbeforeafterboth--- a/crates/evm-coder/src/custom_signature.rs
+++ b/crates/evm-coder/src/custom_signature.rs
@@ -241,7 +241,7 @@
/// ```
#[macro_export]
macro_rules! make_signature {
- (new fn($func:expr)$(,)+) => {
+ (new fn($func:expr)$(,)*) => {
{
let fs = FunctionSignature::new($func);
let fs = FunctionSignature::done(fs, false);
@@ -426,7 +426,7 @@
#[test]
fn make_func_without_args() {
const SIG: FunctionSignature = make_signature!(
- new fn(SIGNATURE_PREFERENCES),
+ new fn(SIGNATURE_PREFERENCES)
);
let name = SIG.as_str();
similar_asserts::assert_eq!(name, "some_funk()");