1
2#![doc = include_str!("../README.md")]
3
4extern crate proc_macro;
7
8use proc_macro::{LexError, TokenStream};
9use std::process::Command;
10use proc_macro_error::*;
11
12fn run_node_macro(
13 macro_input: TokenStream,
14 macro_args: Option<TokenStream>,
15 node_script_path: &str,
16) -> Result<TokenStream, LexError> {
17 let output = Command::new("node")
18 .arg("-e")
19 .arg(format!(
20 r"
21 MACRO_INPUT = {:?};
22 MACRO_ARGUMENTS = {:?};
23 MACRO_OUTPUT = '';
24 require({:?});
25 console.log('MACRO_OUTPUT: ' + MACRO_OUTPUT);
26 ",
27 macro_input.to_string(),
28 macro_args.map(|a| a.to_string()).unwrap_or_default(),
29 node_script_path
30 ))
31 .output()
32 .unwrap_or_else(|e| {
33 if let std::io::ErrorKind::NotFound = e.kind() {
34 panic!("No `node` command found in PATH");
35 }
36
37 panic!("IO Error running JS macro: {}", e)
38 });
39
40 if !output.status.success() {
41 emit_call_site_error!(format!("Error occurred running JS macro:\n{}\n", String::from_utf8_lossy(&output.stderr)));
42 return Ok(Default::default());
43 }
44
45 let macro_output = std::str::from_utf8(&output.stdout)
46 .expect("JS Macro had unexpected non-utf8 output")
47 .split("MACRO_OUTPUT: ")
48 .nth(1)
49 .expect("JS Macro unexpectedly missing macro output");
50
51 macro_output.parse()
52}
53#[proc_macro_error]
54 #[proc_macro_derive(SayHello, attributes(hello_message))]
55 #[allow(non_snake_case)] #[doc(hidden)]
56 pub fn SayHello(input: TokenStream) -> TokenStream {
57 run_node_macro(input, None, "/Users/agrant/w/js-macros/target/debug/build/js-macros-6770ecadfd1cfef1/out/custom_derive.js").unwrap()
58 }#[proc_macro_error]
59 #[proc_macro]
60 #[allow(non_snake_case)] #[doc(hidden)]
61 pub fn hello_world(input: TokenStream) -> TokenStream {
62 run_node_macro(input, None, "/Users/agrant/w/js-macros/target/debug/build/js-macros-6770ecadfd1cfef1/out/custom_function.js").unwrap()
63 }#[proc_macro_error]
64 #[proc_macro_attribute]
65 #[allow(non_snake_case)] #[doc(hidden)]
66 pub fn say_hello(args: TokenStream, input: TokenStream) -> TokenStream {
67 run_node_macro(input, Some(args), "/Users/agrant/w/js-macros/target/debug/build/js-macros-6770ecadfd1cfef1/out/custom_attribute.js").unwrap()
68 }