marine-rs-sdk-test/crates/fce-test-macro-impl/src/fce_test/module_generator.rs

/*
 * Copyright 2021 Fluence Labs Limited
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

mod methods_generator;
mod record_type_generator;

use crate::fce_test::utils;
use crate::fce_test::config_worker::Module;
use crate::TResult;

use fce_wit_parser::interface::FCEModuleInterface;

use proc_macro2::TokenStream;
use quote::quote;

pub(super) fn generate_module_definitions<'i>(
    modules: impl ExactSizeIterator<Item = &'i Module<'i>>,
) -> TResult<TokenStream> {
    let mut module_definitions = Vec::with_capacity(modules.len());

    for module in modules {
        let module_definition = generate_module_definition(&module.name, &module.interface)?;
        module_definitions.push(module_definition);
    }

    let module_definitions = quote! { #(#module_definitions),*};

    Ok(module_definitions)
}

fn generate_module_definition(
    module_name: &str,
    module_interface: &FCEModuleInterface,
) -> TResult<TokenStream> {
    let module_name_ident = utils::generate_module_name(module_name)?;
    let struct_name_ident = utils::generate_struct_name(module_name)?;
    let module_records = record_type_generator::generate_records(&module_interface.record_types)?;
    let module_functions = methods_generator::generate_module_methods(
        module_name,
        module_interface.function_signatures.iter(),
        &module_interface.record_types,
    )?;

    let module_definition = quote! {
        pub mod #module_name_ident {
            #module_records

            pub struct #struct_name_ident {
                pub fce: std::rc::Rc<std::cell::RefCell<fluence_test::internal::AppService>>,
            }

            impl #struct_name_ident {
                #(#module_functions)*
            }
        }
    };

    Ok(module_definition)
}