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
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102

#![recursion_limit="128"]

extern crate proc_macro;
extern crate proc_macro2;
#[macro_use]
extern crate syn;
#[macro_use]
extern crate quote;
extern crate base_x;
extern crate serde;
#[macro_use]
extern crate serde_derive;
extern crate serde_json;
extern crate sha1;

#[cfg(test)]
mod testutils;

mod utils;

mod macro_js_export;
mod macro_async_test;
mod macro_js_raw;
mod macro_js;

mod attr_hack;
mod js_stringify;
mod js_shim;

use utils::Target;

fn emit( result: syn::parse::Result< proc_macro2::TokenStream > ) -> proc_macro::TokenStream {
    match result {
        Ok( stream ) => stream.into(),
        Err( error ) => proc_macro::TokenStream::from( error.to_compile_error() )
    }
}

#[proc_macro_attribute]
pub fn js_export( attrs: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    macro_js_export::js_export( attrs, input )
}

#[proc_macro_attribute]
pub fn async_test( attrs: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    macro_async_test::async_test( attrs, input )
}

#[proc_macro]
pub fn wasm32_unknown_unknown_js_raw( input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js_raw::js_raw( Target::NativeWebAssembly, input.into() ) )
}

#[proc_macro_attribute]
pub fn wasm32_unknown_unknown_js_raw_attr( _: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js_raw::js_raw_attr( Target::NativeWebAssembly, input.into() ) )
}

#[proc_macro_attribute]
pub fn wasm32_unknown_unknown_js_attr( _: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js::js_attr( Target::NativeWebAssembly, input.into(), false ) )
}

#[proc_macro_attribute]
pub fn wasm32_unknown_unknown_js_no_return_attr( _: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js::js_attr( Target::NativeWebAssembly, input.into(), true ) )
}

#[proc_macro]
pub fn emscripten_js_raw( input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js_raw::js_raw( Target::Emscripten, input.into() ) )
}

#[proc_macro_attribute]
pub fn emscripten_js_raw_attr( _: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js_raw::js_raw_attr( Target::Emscripten, input.into() ) )
}

#[proc_macro_attribute]
pub fn emscripten_js_attr( _: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js::js_attr( Target::Emscripten, input.into(), false ) )
}

#[proc_macro_attribute]
pub fn emscripten_js_no_return_attr( _: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js::js_attr( Target::Emscripten, input.into(), true ) )
}

#[proc_macro_attribute]
pub fn wasm_bindgen_js_raw_attr( _: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js_raw::js_raw_attr( Target::WasmBindgen, input.into() ) )
}

#[proc_macro_attribute]
pub fn wasm_bindgen_js_attr( _: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js::js_attr( Target::WasmBindgen, input.into(), false ) )
}

#[proc_macro_attribute]
pub fn wasm_bindgen_js_no_return_attr( _: proc_macro::TokenStream, input: proc_macro::TokenStream ) -> proc_macro::TokenStream {
    emit( macro_js::js_attr( Target::WasmBindgen, input.into(), true ) )
}