29 lines
657 B
Rust
29 lines
657 B
Rust
|
//@ force-host
|
||
|
|
//@ no-prefer-dynamic
|
||
|
|
|
||
|
|
#![crate_type = "proc-macro"]
|
||
|
|
#![feature(proc_macro_tracked_env)]
|
||
|
|
|
||
|
|
extern crate proc_macro;
|
||
|
|
|
||
|
|
use proc_macro::TokenStream;
|
||
|
|
use proc_macro::tracked_env::var;
|
||
|
|
|
||
|
|
#[proc_macro]
|
||
|
|
pub fn generate_const(input: TokenStream) -> TokenStream {
|
||
|
|
let the_const = match var("THE_CONST") {
|
||
|
|
Ok(x) if x == "12" => {
|
||
|
|
"const THE_CONST: u32 = 12;"
|
||
|
|
}
|
||
|
|
_ => {
|
||
|
|
"const THE_CONST: u32 = 0;"
|
||
|
|
}
|
||
|
|
};
|
||
|
|
let another = if var("ANOTHER").is_ok() {
|
||
|
|
"const ANOTHER: u32 = 1;"
|
||
|
|
} else {
|
||
|
|
"const ANOTHER: u32 = 2;"
|
||
|
|
};
|
||
|
|
format!("{the_const}{another}").parse().unwrap()
|
||
|
|
}
|