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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

extern crate proc_macro;


mod util;
mod ast;
mod generator;

use proc_macro::TokenStream;

 

/// This turns regular rust function into N-API compatible native module
/// 
/// For example; given rust following here
/// 
///      fn sum(first: i32, second: i32) -> i32 {
///           return first+second
///      }
/// 
/// into N-API module
///     #[no_mangle]
///     pub extern "C" fn n_sum(env: napi_env, cb_info: napi_callback_info) -> napi_value {
///         fn sum(first: i32, second: i32) -> i32 {
///           return first+second
///         }
///         let js_env = JsEnv::new(env);
///         let js_cb = result_to_napi!(js_env.get_cb_info(cb_info, 2),&js_env);
///         let first = result_to_napi!(js_cb.get_value::<i32>(0),&js_env);
///         let second = result_to_napi!(js_cb.get_value::<i32>(0),&js_env);
///         sum(msg).to_js(&js_env)
///     }
#[proc_macro_attribute]
pub fn node_bindgen(args: TokenStream, item: TokenStream) -> TokenStream {

    use syn::AttributeArgs;

    use ast::FunctionAttributes;
    use ast::NodeItem;    
    use generator::generate_function;
    use generator::generate_class;

    let attribute_args = syn::parse_macro_input!(args as AttributeArgs);
    
    let attribute: FunctionAttributes = 
        match FunctionAttributes::from_ast(attribute_args) {
            Ok(attr) => attr,
            Err(err) => return err.to_compile_error().into()
        };

    let parsed_item = syn::parse_macro_input!(item as NodeItem);
    
    let out_express = match parsed_item {
        NodeItem::Function(fn_item) => {           
            generate_function(fn_item,attribute)
        }
        NodeItem::Impl(impl_item) => {
            generate_class(impl_item)
        }
    };
    
    // used for debugging, if error occurs println do not work so should uncomment express
    //println!("{}",out_express);
    //let out_express = quote::quote! {};
    
    out_express.into()
}