odra-codegen 0.7.1

Code generators for Odra IR.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

use derive_more::From;
use odra_ir::module::ImplItem;
use quote::ToTokens;

use crate::{poet::OdraPoetUsingImpl, GenerateCode};

use self::{def::ContractDef, deploy::Deploy, reference::ContractReference};

mod def;
mod deploy;
mod reference;

#[derive(From)]
pub struct ModuleImpl<'a> {
    contract: &'a odra_ir::module::ModuleImpl
}

as_ref_for_contract_impl_generator!(ModuleImpl);

impl GenerateCode for ModuleImpl<'_> {
    fn generate_code(&self) -> proc_macro2::TokenStream {
        let ident = self.contract.ident();
        let original_item_impls = self.contract.impl_items().iter().map(|item| match item {
            ImplItem::Constructor(item) => item.impl_item.to_token_stream(),
            ImplItem::Method(item) => item.impl_item.to_token_stream(),
            ImplItem::Other(item) => item.to_token_stream()
        });

        let contract_def = self.generate_code_using::<ContractDef>();
        let deploy = self.generate_code_using::<Deploy>();
        let contract_ref = self.generate_code_using::<ContractReference>();

        quote::quote! {
            impl #ident {
                # (#original_item_impls)*
            }

            #contract_def

            #deploy

            #contract_ref
        }
    }
}