1234567891011121314151617#![allow(dead_code)]181920212223use quote::{quote, ToTokens};24use inflector::cases;25use std::fmt::Write;26use syn::{27 Expr, FnArg, GenericArgument, Generics, Ident, ImplItem, ImplItemMethod, ItemImpl, Lit, Meta,28 MetaNameValue, PatType, PathArguments, ReturnType, Type,29 spanned::Spanned,30 parse::{Parse, ParseStream},31 parenthesized, Token, LitInt, LitStr,32};3334use crate::{35 fn_selector_str, parse_ident_from_pat, parse_ident_from_path, parse_path, parse_path_segment,36 parse_result_ok, pascal_ident_to_call, pascal_ident_to_snake_call, snake_ident_to_pascal,37 snake_ident_to_screaming,38};3940struct Is {41 name: Ident,42 pascal_call_name: Ident,43 snake_call_name: Ident,44 via: Option<(Type, Ident)>,45 condition: Option<Expr>,46}47impl Is {48 fn expand_call_def(&self, gen_ref: &proc_macro2::TokenStream) -> proc_macro2::TokenStream {49 let name = &self.name;50 let pascal_call_name = &self.pascal_call_name;51 quote! {52 #name(#pascal_call_name #gen_ref)53 }54 }5556 fn expand_interface_id(&self) -> proc_macro2::TokenStream {57 let pascal_call_name = &self.pascal_call_name;58 quote! {59 interface_id ^= u32::from_be_bytes(#pascal_call_name::interface_id());60 }61 }6263 fn expand_supports_interface(64 &self,65 generics: &proc_macro2::TokenStream,66 ) -> proc_macro2::TokenStream {67 let pascal_call_name = &self.pascal_call_name;68 let condition = self.condition.as_ref().map(|condition| {69 quote! {70 (#condition) &&71 }72 });73 quote! {74 #condition <#pascal_call_name #generics>::supports_interface(this, interface_id)75 }76 }7778 fn expand_variant_weight(&self) -> proc_macro2::TokenStream {79 let name = &self.name;80 quote! {81 Self::#name(call) => call.weight()82 }83 }8485 fn expand_variant_call(86 &self,87 call_name: &proc_macro2::Ident,88 generics: &proc_macro2::TokenStream,89 ) -> proc_macro2::TokenStream {90 let name = &self.name;91 let pascal_call_name = &self.pascal_call_name;92 let via_typ = self93 .via94 .as_ref()95 .map(|(t, _)| quote! {#t})96 .unwrap_or_else(|| quote! {Self});97 let via_map = self98 .via99 .as_ref()100 .map(|(_, i)| quote! {.#i()})101 .unwrap_or_default();102 let condition = self.condition.as_ref().map(|condition| {103 quote! {104 if ({let this = &self; (#condition)})105 }106 });107 quote! {108 #call_name::#name(call) #condition => return <#via_typ as ::evm_coder::Callable<#pascal_call_name #generics>>::call(self #via_map, Msg {109 call,110 caller: c.caller,111 value: c.value,112 })113 }114 }115116 fn expand_parse(&self, generics: &proc_macro2::TokenStream) -> proc_macro2::TokenStream {117 let name = &self.name;118 let pascal_call_name = &self.pascal_call_name;119 quote! {120 if let Some(parsed_call) = <#pascal_call_name #generics>::parse(method_id, reader)? {121 return Ok(Some(Self::#name(parsed_call)))122 }123 }124 }125126 fn expand_generator(&self, generics: &proc_macro2::TokenStream) -> proc_macro2::TokenStream {127 let pascal_call_name = &self.pascal_call_name;128 quote! {129 <#pascal_call_name #generics>::generate_solidity_interface(tc, is_impl);130 }131 }132133 fn expand_event_generator(&self) -> proc_macro2::TokenStream {134 let name = &self.name;135 quote! {136 #name::generate_solidity_interface(tc, is_impl);137 }138 }139}140141#[derive(Default)]142struct IsList(Vec<Is>);143impl Parse for IsList {144 fn parse(input: ParseStream) -> syn::Result<Self> {145 let mut out = vec![];146 loop {147 if input.is_empty() {148 break;149 }150 let name = input.parse::<Ident>()?;151 let lookahead = input.lookahead1();152153 let mut condition: Option<Expr> = None;154 let mut via: Option<(Type, Ident)> = None;155156 if lookahead.peek(syn::token::Paren) {157 let contents;158 parenthesized!(contents in input);159 let input = contents;160161 while !input.is_empty() {162 let lookahead = input.lookahead1();163 if lookahead.peek(Token![if]) {164 input.parse::<Token![if]>()?;165 let contents;166 parenthesized!(contents in input);167 let contents = contents.parse::<Expr>()?;168169 if condition.replace(contents).is_some() {170 return Err(syn::Error::new(input.span(), "condition is already set"));171 }172 } else if lookahead.peek(kw::via) {173 input.parse::<kw::via>()?;174 let contents;175 parenthesized!(contents in input);176177 let method = contents.parse::<Ident>()?;178 contents.parse::<Token![,]>()?;179 let ty = contents.parse::<Type>()?;180181 if via.replace((ty, method)).is_some() {182 return Err(syn::Error::new(input.span(), "via is already set"));183 }184 } else if input.peek(Token![,]) {185 input.parse::<Token![,]>()?;186 } else {187 return Err(lookahead.error());188 }189 }190 } else if lookahead.peek(Token![,]) || input.is_empty() {191 192 } else {193 return Err(lookahead.error());194 };195 out.push(Is {196 pascal_call_name: pascal_ident_to_call(&name),197 snake_call_name: pascal_ident_to_snake_call(&name),198 name,199 via,200 condition,201 });202 if input.peek(Token![,]) {203 input.parse::<Token![,]>()?;204 continue;205 } else {206 break;207 }208 }209 Ok(Self(out))210 }211}212213pub struct InterfaceInfo {214 name: Ident,215 is: IsList,216 inline_is: IsList,217 events: IsList,218 expect_selector: Option<u32>,219}220impl Parse for InterfaceInfo {221 fn parse(input: ParseStream) -> syn::Result<Self> {222 let mut name = None;223 let mut is = None;224 let mut inline_is = None;225 let mut events = None;226 let mut expect_selector = None;227 228 loop {229 let lookahead = input.lookahead1();230 if lookahead.peek(kw::name) {231 let k = input.parse::<kw::name>()?;232 input.parse::<Token![=]>()?;233 if name.replace(input.parse::<Ident>()?).is_some() {234 return Err(syn::Error::new(k.span(), "name is already set"));235 }236 } else if lookahead.peek(kw::is) {237 let k = input.parse::<kw::is>()?;238 let contents;239 parenthesized!(contents in input);240 if is.replace(contents.parse::<IsList>()?).is_some() {241 return Err(syn::Error::new(k.span(), "is is already set"));242 }243 } else if lookahead.peek(kw::inline_is) {244 let k = input.parse::<kw::inline_is>()?;245 let contents;246 parenthesized!(contents in input);247 if inline_is.replace(contents.parse::<IsList>()?).is_some() {248 return Err(syn::Error::new(k.span(), "inline_is is already set"));249 }250 } else if lookahead.peek(kw::events) {251 let k = input.parse::<kw::events>()?;252 let contents;253 parenthesized!(contents in input);254 if events.replace(contents.parse::<IsList>()?).is_some() {255 return Err(syn::Error::new(k.span(), "events is already set"));256 }257 } else if lookahead.peek(kw::expect_selector) {258 let k = input.parse::<kw::expect_selector>()?;259 input.parse::<Token![=]>()?;260 let value = input.parse::<LitInt>()?;261 if expect_selector262 .replace(value.base10_parse::<u32>()?)263 .is_some()264 {265 return Err(syn::Error::new(k.span(), "expect_selector is already set"));266 }267 } else if input.is_empty() {268 break;269 } else {270 return Err(lookahead.error());271 }272 if input.peek(Token![,]) {273 input.parse::<Token![,]>()?;274 } else {275 break;276 }277 }278 Ok(Self {279 name: name.ok_or_else(|| syn::Error::new(input.span(), "missing name"))?,280 is: is.unwrap_or_default(),281 inline_is: inline_is.unwrap_or_default(),282 events: events.unwrap_or_default(),283 expect_selector,284 })285 }286}287288struct MethodInfo {289 rename_selector: Option<String>,290}291impl Parse for MethodInfo {292 fn parse(input: ParseStream) -> syn::Result<Self> {293 let mut rename_selector = None;294 let lookahead = input.lookahead1();295 if lookahead.peek(kw::rename_selector) {296 let k = input.parse::<kw::rename_selector>()?;297 input.parse::<Token![=]>()?;298 if rename_selector299 .replace(input.parse::<LitStr>()?.value())300 .is_some()301 {302 return Err(syn::Error::new(k.span(), "rename_selector is already set"));303 }304 }305 Ok(Self { rename_selector })306 }307}308309enum AbiType {310 311 Plain(Ident),312 313 Tuple(Vec<AbiType>),314 315 Vec(Box<AbiType>),316 317 Array(Box<AbiType>, usize),318}319impl AbiType {320 fn try_from(value: &Type) -> syn::Result<Self> {321 let value = Self::try_maybe_special_from(value)?;322 if value.is_special() {323 return Err(syn::Error::new(value.span(), "unexpected special type"));324 }325 Ok(value)326 }327 fn try_maybe_special_from(value: &Type) -> syn::Result<Self> {328 match value {329 Type::Array(arr) => {330 let wrapped = AbiType::try_from(&arr.elem)?;331 match &arr.len {332 Expr::Lit(l) => match &l.lit {333 Lit::Int(i) => {334 let num = i.base10_parse::<usize>()?;335 Ok(AbiType::Array(Box::new(wrapped), num as usize))336 }337 _ => Err(syn::Error::new(arr.len.span(), "should be int literal")),338 },339 _ => Err(syn::Error::new(arr.len.span(), "should be literal")),340 }341 }342 Type::Path(_) => {343 let path = parse_path(value)?;344 let segment = parse_path_segment(path)?;345 if segment.ident == "Vec" {346 let args = match &segment.arguments {347 PathArguments::AngleBracketed(e) => e,348 _ => {349 return Err(syn::Error::new(350 segment.arguments.span(),351 "missing Vec generic",352 ))353 }354 };355 let args = &args.args;356 if args.len() != 1 {357 return Err(syn::Error::new(358 args.span(),359 "expected only one generic for vec",360 ));361 }362 let arg = args.first().expect("first arg");363364 let ty = match arg {365 GenericArgument::Type(ty) => ty,366 _ => {367 return Err(syn::Error::new(368 arg.span(),369 "expected first generic to be type",370 ))371 }372 };373374 let wrapped = AbiType::try_from(ty)?;375 Ok(Self::Vec(Box::new(wrapped)))376 } else {377 if !segment.arguments.is_empty() {378 return Err(syn::Error::new(379 segment.arguments.span(),380 "unexpected generic arguments for non-vec type",381 ));382 }383 Ok(Self::Plain(segment.ident.clone()))384 }385 }386 Type::Tuple(t) => {387 let mut out = Vec::with_capacity(t.elems.len());388 for el in t.elems.iter() {389 out.push(AbiType::try_from(el)?)390 }391 Ok(Self::Tuple(out))392 }393 _ => Err(syn::Error::new(394 value.span(),395 "unexpected type, only arrays, plain types and tuples are supported",396 )),397 }398 }399 fn is_value(&self) -> bool {400 matches!(self, Self::Plain(v) if v == "value")401 }402 fn is_caller(&self) -> bool {403 matches!(self, Self::Plain(v) if v == "caller")404 }405 fn is_special(&self) -> bool {406 self.is_caller() || self.is_value()407 }408 fn selector_ty_buf(&self, buf: &mut String) -> std::fmt::Result {409 match self {410 AbiType::Plain(t) => {411 write!(buf, "{}", t)412 }413 AbiType::Tuple(t) => {414 write!(buf, "(")?;415 for (i, t) in t.iter().enumerate() {416 if i != 0 {417 write!(buf, ",")?;418 }419 t.selector_ty_buf(buf)?;420 }421 write!(buf, ")")422 }423 AbiType::Vec(v) => {424 v.selector_ty_buf(buf)?;425 write!(buf, "[]")426 }427 AbiType::Array(v, len) => {428 v.selector_ty_buf(buf)?;429 write!(buf, "[{}]", len)430 }431 }432 }433 fn selector_ty(&self) -> String {434 let mut out = String::new();435 self.selector_ty_buf(&mut out).expect("no fmt error");436 out437 }438}439impl ToTokens for AbiType {440 fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {441 match self {442 AbiType::Plain(t) => tokens.extend(quote! {#t}),443 AbiType::Tuple(t) => {444 tokens.extend(quote! {(445 #(#t),*446 )});447 }448 AbiType::Vec(v) => tokens.extend(quote! {Vec<#v>}),449 AbiType::Array(v, l) => tokens.extend(quote! {[#v; #l]}),450 }451 }452}453454struct MethodArg {455 name: Ident,456 camel_name: String,457 ty: AbiType,458}459impl MethodArg {460 fn try_from(value: &PatType) -> syn::Result<Self> {461 let name = parse_ident_from_pat(&value.pat)?.clone();462 Ok(Self {463 camel_name: cases::camelcase::to_camel_case(&name.to_string()),464 name,465 ty: AbiType::try_maybe_special_from(&value.ty)?,466 })467 }468 fn is_value(&self) -> bool {469 self.ty.is_value()470 }471 fn is_caller(&self) -> bool {472 self.ty.is_caller()473 }474 fn is_special(&self) -> bool {475 self.ty.is_special()476 }477 fn selector_ty(&self) -> String {478 assert!(!self.is_special());479 self.ty.selector_ty()480 }481482 fn expand_call_def(&self) -> proc_macro2::TokenStream {483 assert!(!self.is_special());484 let name = &self.name;485 let ty = &self.ty;486487 quote! {488 #name: #ty489 }490 }491492 fn expand_parse(&self) -> proc_macro2::TokenStream {493 assert!(!self.is_special());494 let name = &self.name;495 quote! {496 #name: reader.abi_read()?497 }498 }499500 fn expand_call_arg(&self) -> proc_macro2::TokenStream {501 if self.is_value() {502 quote! {503 c.value.clone()504 }505 } else if self.is_caller() {506 quote! {507 c.caller.clone()508 }509 } else {510 let name = &self.name;511 quote! {512 #name513 }514 }515 }516517 fn expand_solidity_argument(&self) -> proc_macro2::TokenStream {518 let camel_name = &self.camel_name.to_string();519 let ty = &self.ty;520 quote! {521 <NamedArgument<#ty>>::new(#camel_name)522 }523 }524}525526#[derive(PartialEq)]527enum Mutability {528 Mutable,529 View,530 Pure,531}532533534535mod kw {536 syn::custom_keyword!(weight);537538 syn::custom_keyword!(via);539 syn::custom_keyword!(name);540 syn::custom_keyword!(is);541 syn::custom_keyword!(inline_is);542 syn::custom_keyword!(events);543 syn::custom_keyword!(expect_selector);544545 syn::custom_keyword!(rename_selector);546}547548549struct Method {550 name: Ident,551 camel_name: String,552 pascal_name: Ident,553 screaming_name: Ident,554 selector_str: String,555 selector: u32,556 args: Vec<MethodArg>,557 has_normal_args: bool,558 mutability: Mutability,559 result: Type,560 weight: Option<Expr>,561 docs: Vec<String>,562}563impl Method {564 fn try_from(value: &ImplItemMethod) -> syn::Result<Self> {565 let mut info = MethodInfo {566 rename_selector: None,567 };568 let mut docs = Vec::new();569 let mut weight = None;570 for attr in &value.attrs {571 let ident = parse_ident_from_path(&attr.path, false)?;572 if ident == "solidity" {573 info = attr.parse_args::<MethodInfo>()?;574 } else if ident == "doc" {575 let args = attr.parse_meta().unwrap();576 let value = match args {577 Meta::NameValue(MetaNameValue {578 lit: Lit::Str(str), ..579 }) => str.value(),580 _ => unreachable!(),581 };582 docs.push(value);583 } else if ident == "weight" {584 weight = Some(attr.parse_args::<Expr>()?);585 }586 }587 let ident = &value.sig.ident;588 let ident_str = ident.to_string();589 if !cases::snakecase::is_snake_case(&ident_str) {590 return Err(syn::Error::new(ident.span(), "method name should be snake_cased\nif alternative solidity name needs to be set - use #[solidity] attribute"));591 }592593 let mut mutability = Mutability::Pure;594595 if let Some(FnArg::Receiver(receiver)) = value596 .sig597 .inputs598 .iter()599 .find(|arg| matches!(arg, FnArg::Receiver(_)))600 {601 if receiver.reference.is_none() {602 return Err(syn::Error::new(603 receiver.span(),604 "receiver should be by ref",605 ));606 }607 if receiver.mutability.is_some() {608 mutability = Mutability::Mutable;609 } else {610 mutability = Mutability::View;611 }612 }613 let mut args = Vec::new();614 for typ in value615 .sig616 .inputs617 .iter()618 .filter(|arg| matches!(arg, FnArg::Typed(_)))619 {620 let typ = match typ {621 FnArg::Typed(typ) => typ,622 _ => unreachable!(),623 };624 args.push(MethodArg::try_from(typ)?);625 }626627 if mutability != Mutability::Mutable && args.iter().any(|arg| arg.is_value()) {628 return Err(syn::Error::new(629 args.iter().find(|arg| arg.is_value()).unwrap().ty.span(),630 "payable function should be mutable",631 ));632 }633634 let result = match &value.sig.output {635 ReturnType::Type(_, ty) => ty,636 _ => 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)")),637 };638 let result = parse_result_ok(result)?;639640 let camel_name = info641 .rename_selector642 .unwrap_or_else(|| cases::camelcase::to_camel_case(&ident.to_string()));643 let mut selector_str = camel_name.clone();644 selector_str.push('(');645 let mut has_normal_args = false;646 for (i, arg) in args.iter().filter(|arg| !arg.is_special()).enumerate() {647 if i != 0 {648 selector_str.push(',');649 }650 write!(selector_str, "{}", arg.selector_ty()).unwrap();651 has_normal_args = true;652 }653 selector_str.push(')');654 let selector = fn_selector_str(&selector_str);655656 Ok(Self {657 name: ident.clone(),658 camel_name,659 pascal_name: snake_ident_to_pascal(ident),660 screaming_name: snake_ident_to_screaming(ident),661 selector_str,662 selector,663 args,664 has_normal_args,665 mutability,666 result: result.clone(),667 weight,668 docs,669 })670 }671 fn expand_call_def(&self) -> proc_macro2::TokenStream {672 let defs = self673 .args674 .iter()675 .filter(|a| !a.is_special())676 .map(|a| a.expand_call_def());677 let pascal_name = &self.pascal_name;678 let docs = &self.docs;679680 if self.has_normal_args {681 quote! {682 #(#[doc = #docs])*683 #[allow(missing_docs)]684 #pascal_name {685 #(686 #defs,687 )*688 }689 }690 } else {691 quote! {#pascal_name}692 }693 }694695 fn expand_const(&self) -> proc_macro2::TokenStream {696 let screaming_name = &self.screaming_name;697 let selector = u32::to_be_bytes(self.selector);698 let selector_str = &self.selector_str;699 quote! {700 #[doc = #selector_str]701 const #screaming_name: ::evm_coder::types::bytes4 = [#(#selector,)*];702 }703 }704705 fn expand_interface_id(&self) -> proc_macro2::TokenStream {706 let screaming_name = &self.screaming_name;707 quote! {708 interface_id ^= u32::from_be_bytes(Self::#screaming_name);709 }710 }711712 fn expand_parse(&self) -> proc_macro2::TokenStream {713 let pascal_name = &self.pascal_name;714 let screaming_name = &self.screaming_name;715 if self.has_normal_args {716 let parsers = self717 .args718 .iter()719 .filter(|a| !a.is_special())720 .map(|a| a.expand_parse());721 quote! {722 Self::#screaming_name => return Ok(Some(Self::#pascal_name {723 #(724 #parsers,725 )*726 }))727 }728 } else {729 quote! { Self::#screaming_name => return Ok(Some(Self::#pascal_name)) }730 }731 }732733 fn expand_variant_call(&self, call_name: &proc_macro2::Ident) -> proc_macro2::TokenStream {734 let pascal_name = &self.pascal_name;735 let name = &self.name;736737 let matcher = if self.has_normal_args {738 let names = self739 .args740 .iter()741 .filter(|a| !a.is_special())742 .map(|a| &a.name);743744 quote! {{745 #(746 #names,747 )*748 }}749 } else {750 quote! {}751 };752753 let receiver = match self.mutability {754 Mutability::Mutable | Mutability::View => quote! {self.},755 Mutability::Pure => quote! {Self::},756 };757 let args = self.args.iter().map(|a| a.expand_call_arg());758759 quote! {760 #call_name::#pascal_name #matcher => {761 let result = #receiver #name(762 #(763 #args,764 )*765 )?;766 (&result).to_result()767 }768 }769 }770771 fn expand_variant_weight(&self) -> proc_macro2::TokenStream {772 let pascal_name = &self.pascal_name;773 if let Some(weight) = &self.weight {774 let matcher = if self.has_normal_args {775 let names = self776 .args777 .iter()778 .filter(|a| !a.is_special())779 .map(|a| &a.name);780781 quote! {{782 #(783 #names,784 )*785 }}786 } else {787 quote! {}788 };789 quote! {790 Self::#pascal_name #matcher => (#weight).into()791 }792 } else {793 let matcher = if self.has_normal_args {794 quote! {{..}}795 } else {796 quote! {}797 };798 quote! {799 Self::#pascal_name #matcher => ().into()800 }801 }802 }803804 fn expand_solidity_function(&self) -> proc_macro2::TokenStream {805 let camel_name = &self.camel_name;806 let mutability = match self.mutability {807 Mutability::Mutable => quote! {SolidityMutability::Mutable},808 Mutability::View => quote! { SolidityMutability::View },809 Mutability::Pure => quote! {SolidityMutability::Pure},810 };811 let result = &self.result;812813 let args = self814 .args815 .iter()816 .filter(|a| !a.is_special())817 .map(MethodArg::expand_solidity_argument);818 let docs = &self.docs;819 let selector_str = &self.selector_str;820 let selector = self.selector;821822 quote! {823 SolidityFunction {824 docs: &[#(#docs),*],825 selector_str: #selector_str,826 selector: #selector,827 name: #camel_name,828 mutability: #mutability,829 args: (830 #(831 #args,832 )*833 ),834 result: <UnnamedArgument<#result>>::default(),835 }836 }837 }838}839840fn generics_list(gen: &Generics) -> proc_macro2::TokenStream {841 if gen.params.is_empty() {842 return quote! {};843 }844 let params = gen.params.iter().map(|p| match p {845 syn::GenericParam::Type(id) => {846 let v = &id.ident;847 quote! {#v}848 }849 syn::GenericParam::Lifetime(lt) => {850 let v = <.lifetime;851 quote! {#v}852 }853 syn::GenericParam::Const(c) => {854 let i = &c.ident;855 quote! {#i}856 }857 });858 quote! { #(#params),* }859}860fn generics_reference(gen: &Generics) -> proc_macro2::TokenStream {861 if gen.params.is_empty() {862 return quote! {};863 }864 let list = generics_list(gen);865 quote! { <#list> }866}867fn generics_data(gen: &Generics) -> proc_macro2::TokenStream {868 let list = generics_list(gen);869 if gen.params.len() == 1 {870 quote! {#list}871 } else {872 quote! { (#list) }873 }874}875876pub struct SolidityInterface {877 generics: Generics,878 name: Box<syn::Type>,879 info: InterfaceInfo,880 methods: Vec<Method>,881 docs: Vec<String>,882}883impl SolidityInterface {884 pub fn try_from(info: InterfaceInfo, value: &ItemImpl) -> syn::Result<Self> {885 let mut methods = Vec::new();886887 for item in &value.items {888 if let ImplItem::Method(method) = item {889 methods.push(Method::try_from(method)?)890 }891 }892 let mut docs = vec![];893 for attr in &value.attrs {894 let ident = parse_ident_from_path(&attr.path, false)?;895 if ident == "doc" {896 let args = attr.parse_meta().unwrap();897 let value = match args {898 Meta::NameValue(MetaNameValue {899 lit: Lit::Str(str), ..900 }) => str.value(),901 _ => unreachable!(),902 };903 docs.push(value);904 }905 }906 Ok(Self {907 generics: value.generics.clone(),908 name: value.self_ty.clone(),909 info,910 methods,911 docs,912 })913 }914 pub fn expand(self) -> proc_macro2::TokenStream {915 let name = self.name;916917 let solidity_name = self.info.name.to_string();918 let call_name = pascal_ident_to_call(&self.info.name);919 let generics = self.generics;920 let gen_ref = generics_reference(&generics);921 let gen_data = generics_data(&generics);922 let gen_where = &generics.where_clause;923924 let call_sub = self925 .info926 .inline_is927 .0928 .iter()929 .chain(self.info.is.0.iter())930 .map(|c| Is::expand_call_def(c, &gen_ref));931 let call_parse = self932 .info933 .inline_is934 .0935 .iter()936 .chain(self.info.is.0.iter())937 .map(|is| Is::expand_parse(is, &gen_ref));938 let call_variants = self939 .info940 .inline_is941 .0942 .iter()943 .chain(self.info.is.0.iter())944 .map(|c| Is::expand_variant_call(c, &call_name, &gen_ref));945 let weight_variants = self946 .info947 .inline_is948 .0949 .iter()950 .chain(self.info.is.0.iter())951 .map(Is::expand_variant_weight);952953 let inline_interface_id = self.info.inline_is.0.iter().map(Is::expand_interface_id);954 let supports_interface = self955 .info956 .is957 .0958 .iter()959 .map(|is| Is::expand_supports_interface(is, &gen_ref));960961 let calls = self.methods.iter().map(Method::expand_call_def);962 let consts = self.methods.iter().map(Method::expand_const);963 let interface_id = self.methods.iter().map(Method::expand_interface_id);964 let parsers = self.methods.iter().map(Method::expand_parse);965 let call_variants_this = self966 .methods967 .iter()968 .map(|m| Method::expand_variant_call(m, &call_name));969 let weight_variants_this = self.methods.iter().map(Method::expand_variant_weight);970 let solidity_functions = self.methods.iter().map(Method::expand_solidity_function);971972 973 let solidity_is = self974 .info975 .is976 .0977 .iter()978 .chain(self.info.inline_is.0.iter())979 .map(|is| is.name.to_string());980 let solidity_events_is = self.info.events.0.iter().map(|is| is.name.to_string());981 let solidity_generators = self982 .info983 .is984 .0985 .iter()986 .chain(self.info.inline_is.0.iter())987 .map(|is| Is::expand_generator(is, &gen_ref));988 let solidity_event_generators = self.info.events.0.iter().map(Is::expand_event_generator);989990 let docs = &self.docs;991992 if let Some(expect_selector) = &self.info.expect_selector {993 if !self.info.inline_is.0.is_empty() {994 return syn::Error::new(995 name.span(),996 "expect_selector is not compatible with inline_is",997 )998 .to_compile_error();999 }1000 let selector = self1001 .methods1002 .iter()1003 .map(|m| m.selector)1004 .fold(0, |a, b| a ^ b);10051006 if *expect_selector != selector {1007 let mut methods = String::new();1008 for meth in self.methods.iter() {1009 write!(methods, "\n- {}", meth.selector_str).expect("write to string");1010 }1011 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();1012 }1013 }1014 10151016 quote! {1017 #[derive(Debug)]1018 #(#[doc = #docs])*1019 pub enum #call_name #gen_ref {1020 1021 ERC165Call(::evm_coder::ERC165Call, ::core::marker::PhantomData<#gen_data>),1022 #(1023 #calls,1024 )*1025 #(1026 #call_sub,1027 )*1028 }1029 impl #gen_ref #call_name #gen_ref {1030 #(1031 #consts1032 )*1033 1034 pub fn interface_id() -> ::evm_coder::types::bytes4 {1035 let mut interface_id = 0;1036 #(#interface_id)*1037 #(#inline_interface_id)*1038 u32::to_be_bytes(interface_id)1039 }1040 1041 pub fn generate_solidity_interface(tc: &evm_coder::solidity::TypeCollector, is_impl: bool) {1042 use evm_coder::solidity::*;1043 use core::fmt::Write;1044 let interface = SolidityInterface {1045 docs: &[#(#docs),*],1046 name: #solidity_name,1047 selector: Self::interface_id(),1048 is: &["Dummy", "ERC165", #(1049 #solidity_is,1050 )* #(1051 #solidity_events_is,1052 )* ],1053 functions: (#(1054 #solidity_functions,1055 )*),1056 };10571058 let mut out = ::evm_coder::types::string::new();1059 if #solidity_name.starts_with("Inline") {1060 out.push_str("/// @dev inlined interface\n");1061 }1062 let _ = interface.format(is_impl, &mut out, tc);1063 tc.collect(out);1064 #(1065 #solidity_event_generators1066 )*1067 #(1068 #solidity_generators1069 )*1070 if is_impl {1071 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());1072 } else {1073 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());1074 }1075 }1076 }1077 impl #gen_ref ::evm_coder::Call for #call_name #gen_ref {1078 fn parse(method_id: ::evm_coder::types::bytes4, reader: &mut ::evm_coder::abi::AbiReader) -> ::evm_coder::execution::Result<Option<Self>> {1079 use ::evm_coder::abi::AbiRead;1080 match method_id {1081 ::evm_coder::ERC165Call::INTERFACE_ID => return Ok(1082 ::evm_coder::ERC165Call::parse(method_id, reader)?1083 .map(|c| Self::ERC165Call(c, ::core::marker::PhantomData))1084 ),1085 #(1086 #parsers,1087 )*1088 _ => {},1089 }1090 #(1091 #call_parse1092 )else*1093 return Ok(None);1094 }1095 }1096 impl #generics #call_name #gen_ref1097 #gen_where1098 {1099 1100 pub fn supports_interface(this: &#name, interface_id: ::evm_coder::types::bytes4) -> bool {1101 interface_id != u32::to_be_bytes(0xffffff) && (1102 interface_id == ::evm_coder::ERC165Call::INTERFACE_ID ||1103 interface_id == Self::interface_id()1104 #(1105 || #supports_interface1106 )*1107 )1108 }1109 }1110 impl #generics ::evm_coder::Weighted for #call_name #gen_ref1111 #gen_where1112 {1113 #[allow(unused_variables)]1114 fn weight(&self) -> ::evm_coder::execution::DispatchInfo {1115 match self {1116 #(1117 #weight_variants,1118 )*1119 1120 Self::ERC165Call(::evm_coder::ERC165Call::SupportsInterface {..}, _) => 100u64.into(),1121 #(1122 #weight_variants_this,1123 )*1124 }1125 }1126 }1127 impl #generics ::evm_coder::Callable<#call_name #gen_ref> for #name1128 #gen_where1129 {1130 #[allow(unreachable_code)] 1131 fn call(&mut self, c: Msg<#call_name #gen_ref>) -> ::evm_coder::execution::ResultWithPostInfo<::evm_coder::abi::AbiWriter> {1132 use ::evm_coder::abi::AbiWrite;1133 match c.call {1134 #(1135 #call_variants,1136 )*1137 #call_name::ERC165Call(::evm_coder::ERC165Call::SupportsInterface {interface_id}, _) => {1138 let mut writer = ::evm_coder::abi::AbiWriter::default();1139 writer.bool(&<#call_name #gen_ref>::supports_interface(self, interface_id));1140 return Ok(writer.into());1141 }1142 _ => {},1143 }1144 let mut writer = ::evm_coder::abi::AbiWriter::default();1145 match c.call {1146 #(1147 #call_variants_this,1148 )*1149 _ => Err(::evm_coder::execution::Error::from("method is not available").into()),1150 }1151 }1152 }1153 }1154 }1155}