odra-macros 2.6.0

use quote::ToTokens;

pub mod mock {
    use crate::ir::{ModuleImplIR, ModuleStructIR, TypeIR};
    use quote::quote;

    pub fn module_impl() -> ModuleImplIR {
        let module = quote! {
            impl Erc20 {
                /// Initializes the contract with the given parameters.
                pub fn init(&mut self, total_supply: Option<U256>) {
                    if let Some(total_supply) = total_supply {
                        self.total_supply.set(total_supply);
                        self.balances.set(self.env().caller(), total_supply);
                    }
                }

                /// Upgrades the contract with the given parameters.
                pub fn upgrade(&mut self, total_supply: Option<U256>) {
                    if let Some(total_supply) = total_supply {
                        self.total_supply.set(total_supply);
                        self.balances.set(self.env().caller(), total_supply);
                    }
                }

                /// Returns the total supply of the token.
                pub fn total_supply(&self) -> U256 {
                    self.total_supply.get_or_default()
                }

                /// Pay to mint.
                #[odra(payable)]
                pub fn pay_to_mint(&mut self) {
                    let attached_value = self.env().attached_value();
                    self.total_supply
                        .set(self.total_supply() + U256::from(attached_value.as_u64()));
                }

                /// Approve.
                #[odra(non_reentrant)]
                pub fn approve(&mut self, to: &Address, amount: &U256, msg: Maybe<String>) {
                    self.env.emit_event(Approval {
                        owner: self.env.caller(),
                        spender: to,
                        value: amount
                    });
                }

                /// Airdrops the given amount to the given addresses.
                pub fn airdrop(to: &[Address], amount: &U256) {
                }

                fn private_function() {

                }

                /// Swaps the given amount to the given addresses.
                pub fn swap(&mut self, to: Address, amount: U256) -> U256 {
                    amount
                }
            }
        };

        let attr = quote!();
        ModuleImplIR::try_from((&attr, &module)).unwrap()
    }

    pub fn module_factory_on() -> ModuleImplIR {
        let module = quote! {
            impl Erc20 {
                /// Returns the total supply of the token.
                pub fn total_supply(&self) -> U256 {
                    self.total_supply.get_or_default()
                }

                /// Pay to mint.
                #[odra(payable)]
                pub fn pay_to_mint(&mut self) {
                    let attached_value = self.env().attached_value();
                    self.total_supply
                        .set(self.total_supply() + U256::from(attached_value.as_u64()));
                }
            }
        };

        let attr = quote!(factory = on);
        ModuleImplIR::try_from((&attr, &module)).unwrap()
    }

    pub fn module_factory_impl() -> ModuleImplIR {
        let module = quote! {
            impl Erc20Factory {
                pub fn init(&mut self, value: u32) {
                    self.value.set(value);
                }

                /// Returns the total supply of the token.
                pub fn total_supply(&self) -> U256 {
                    self.total_supply.get_or_default()
                }

                /// Pay to mint.
                #[odra(payable)]
                pub fn pay_to_mint(&mut self) {
                    let attached_value = self.env().attached_value();
                    self.total_supply
                        .set(self.total_supply() + U256::from(attached_value.as_u64()));
                }

                /// Approve.
                #[odra(non_reentrant)]
                pub fn approve(&mut self, to: &Address, amount: &U256, msg: Maybe<String>) {
                    self.env.emit_event(Approval {
                        owner: self.env.caller(),
                        spender: to,
                        value: amount
                    });
                }
            }
        };

        let attr = quote!();
        ModuleImplIR::try_from((&attr, &module)).unwrap()
    }

    pub fn module_trait_impl() -> ModuleImplIR {
        let module = quote! {
            impl IErc20 for Erc20 {
                fn total_supply(&self) -> U256 {
                    self.total_supply.get_or_default()
                }

                fn set_total_supply(&mut self) -> U256 {
                    self.total_supply.set(1000.into());
                    1000.into()
                }

                #[odra(payable)]
                fn pay_to_mint(&mut self) {
                    let attached_value = self.env().attached_value();
                    self.total_supply
                        .set(self.total_supply() + U256::from(attached_value.as_u64()));
                }
            }
        };

        let attr = quote!();
        ModuleImplIR::try_from((&attr, &module)).unwrap()
    }

    pub fn module_delegation() -> ModuleImplIR {
        let module = quote! {
            impl Erc20 {
                /// Returns the total supply of the token.
                pub fn total_supply(&self) -> U256 {
                    self.total_supply.get_or_default()
                }

                delegate! {
                    to self.ownable {
                        /// Returns the owner of the contract.
                        fn get_owner(&self) -> Address;
                        /// Sets the owner of the contract.
                        fn set_owner(&mut self, new_owner: Address);
                    }

                    to self.metadata {
                        /// Returns the name of the token.
                        fn name(&self) -> String;
                        fn symbol(&self) -> String;
                    }
                }
            }
        };

        let attr = quote!();
        ModuleImplIR::try_from((&attr, &module)).unwrap()
    }

    pub fn module_invalid_delegation() -> ModuleImplIR {
        let module = quote! {
            impl Erc20 {
                /// Returns the total supply of the token.
                pub fn total_supply(&self) -> U256 {
                    self.total_supply.get_or_default()
                }

                delegate! {
                    to self.ownable {
                        /// Returns the owner of the contract.
                        pub fn get_owner(&self) -> Address;
                        /// Sets the owner of the contract.
                        fn set_owner(&mut self, new_owner: Address);
                    }
                }
            }
        };

        let attr = quote!();
        ModuleImplIR::try_from((&attr, &module)).unwrap()
    }

    pub fn module_definition() -> ModuleStructIR {
        let module = quote!(
            pub struct CounterPack {
                counter0: SubModule<Counter>,
                counter1: SubModule<Counter>,
                counter2: SubModule<Counter>,
                counters: Var<u32>,
                counters_map: Mapping<u8, Counter>
            }
        );
        let attr = quote!(
            name = "MyCounterPack",
            version = "0.1.0",
            events = [OnTransfer, OnApprove],
            errors = Erc20Errors
        );
        ModuleStructIR::try_from((&attr, &module)).unwrap()
    }

    pub fn factory_module_definition() -> ModuleStructIR {
        let module = quote!(
            pub struct CounterPack {
                counter0: SubModule<Counter>,
                counters: Var<u32>,
                counters_map: Mapping<u8, Counter>
            }
        );
        let attr = quote!(factory = on);
        ModuleStructIR::try_from((&attr, &module)).unwrap()
    }

    pub fn invalid_module_definition() -> ModuleStructIR {
        let fields = (0..256).map(|i| {
            let field_name =
                syn::Ident::new(&format!("field{}", i), proc_macro2::Span::call_site());
            quote!(#field_name: Var<u32>,)
        });
        let module = quote!(
            pub struct Module {
                #(#fields)*
            }
        );
        let attr = quote!();
        ModuleStructIR::try_from((&attr, &module)).unwrap()
    }

    pub fn empty_module_definition() -> ModuleStructIR {
        let module = quote!(
            pub struct CounterPack;
        );
        let attr = quote!();
        ModuleStructIR::try_from((&attr, &module)).unwrap()
    }

    pub fn custom_struct() -> TypeIR {
        let ty = quote!(
            struct MyType {
                a: String,
                b: u32
            }
        );
        TypeIR::try_from(&ty).unwrap()
    }

    pub fn custom_enum() -> TypeIR {
        let ty = quote!(
            enum MyType {
                /// Description of A
                A = 10,
                /// Description of B
                B
            }
        );
        TypeIR::try_from(&ty).unwrap()
    }

    pub fn custom_complex_enum() -> TypeIR {
        let ty = quote!(
            enum MyType {
                /// Description of A
                A { a: String, b: u32 },
                /// Description of B
                B(u32, String),
                /// Description of C
                C(),
                /// Description of D
                D {}
            }
        );
        TypeIR::try_from(&ty).unwrap()
    }

    pub fn custom_union() -> TypeIR {
        let ty = quote!(
            union MyUnion {
                f1: u32,
                f2: f32,
            }
        );
        TypeIR::try_from(&ty).unwrap()
    }

    pub fn ext_contract() -> ModuleImplIR {
        let ext = quote!(
            pub trait Token {
                fn balance_of(&self, owner: Address) -> U256;
            }
        );
        let attr = quote!();
        ModuleImplIR::try_from((&attr, &ext)).unwrap()
    }
}

#[track_caller]
pub fn assert_eq<A: ToTokens, B: ToTokens>(a: A, b: B) {
    fn parse<T: ToTokens>(e: T) -> String {
        let e = e.to_token_stream().to_string();
        let e = syn::parse_file(&e).unwrap();
        prettyplease::unparse(&e)
    }
    pretty_assertions::assert_eq!(parse(a), parse(b));
}