29 lines
848 B
Rust
29 lines
848 B
Rust
use proc_macro::TokenStream;
|
|
use quote::quote;
|
|
|
|
pub(super) fn flags_capture(args: TokenStream) -> TokenStream {
|
|
let cargo_crate_name = std::env::var("CARGO_CRATE_NAME");
|
|
let crate_name = match cargo_crate_name.as_ref() {
|
|
Err(_) => return args,
|
|
Ok(crate_name) => crate_name.trim_start_matches("conduwuit_"),
|
|
};
|
|
|
|
let flag = std::env::args().collect::<Vec<_>>();
|
|
let flag_len = flag.len();
|
|
let ret = quote! {
|
|
pub static RUSTC_FLAGS: [&str; #flag_len] = [#( #flag ),*];
|
|
|
|
#[conduwuit_core::ctor]
|
|
fn _set_rustc_flags() {
|
|
conduwuit_core::info::rustc::FLAGS.lock().expect("locked").insert(#crate_name, &RUSTC_FLAGS);
|
|
}
|
|
|
|
// static strings have to be yanked on module unload
|
|
#[conduwuit_core::dtor]
|
|
fn _unset_rustc_flags() {
|
|
conduwuit_core::info::rustc::FLAGS.lock().expect("locked").remove(#crate_name);
|
|
}
|
|
};
|
|
|
|
ret.into()
|
|
}
|