use proc_macro2::TokenStream as TokenStream2;
use quote::quote;

use crate::{
    generate::lua_wrapper::{
        lua_functions_to_globals::generate_add_lua_functions_to_globals,
        rust_wrapper_functions::generate_rust_wrapper_functions,
    },
    DataCommandEnum,
};

mod lua_functions_to_globals;
mod rust_wrapper_functions;

pub fn lua_wrapper(input: &DataCommandEnum) -> TokenStream2 {
    let add_lua_functions_to_globals = generate_add_lua_functions_to_globals(input);
    let rust_wrapper_functions = generate_rust_wrapper_functions(None, input);
    quote! {
        #add_lua_functions_to_globals
        #rust_wrapper_functions
    }
}