RGBCube/uutils-coreutils
1
Fork 0
mirror of https://github.com/RGBCube/uutils-coreutils synced 2025-08-05 15:37:47 +00:00
Code Issues Activity

build/uucore_procs ~ add debugging statements

Browse source
This commit is contained in:
Roy Ivy III 2020-05-23 18:44:03 -05:00
parent a352657e1f
commit ad30781c0c
1 changed files with 24 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

46
src/uucore/src/uucore_procs/src/lib.rs
View file

@ -26,9 +26,9 @@ macro_rules! proc_dbg {
// main!( EXPR ) // main!( EXPR )
// generates a `main()` function for utilities within the uutils group // generates a `main()` function for utilities within the uutils group
// EXPR == syn::Expr::Lit::String | syn::Expr::Path::Ident ~ EXPR contains the location of the utility `uumain()` function // EXPR == syn::Expr::Lit::String | syn::Expr::Path::Ident ~ EXPR contains the lexical path to the utility `uumain()` function
//* for future use of "eager" macros and more generic use, EXPR may be in either STRING or IDENT form //* NOTE: EXPR is ultimately expected to be a multi-segment lexical path (eg, `crate::func`); so, if a single segment path is provided, a trailing "::uumain" is automatically added
//* for ease-of-use, the trailing "::uumain" is optional and will be added if needed //* for more generic use (and future use of "eager" macros), EXPR may be in either STRING or IDENT form
struct Tokens { struct Tokens {
expr: syn::Expr, expr: syn::Expr,
@ -45,30 +45,32 @@ impl syn::parse::Parse for Tokens {
#[proc_macro] #[proc_macro]
pub fn main(stream: proc_macro::TokenStream) -> proc_macro::TokenStream { pub fn main(stream: proc_macro::TokenStream) -> proc_macro::TokenStream {
let Tokens { expr } = syn::parse_macro_input!(stream as Tokens); let Tokens { expr } = syn::parse_macro_input!(stream as Tokens);
proc_dbg!(&expr);
const ARG_PANIC_TEXT: &str = "expected ident lexical path (or a literal string version) to 'uumain()' as argument";
// match EXPR as a string literal or an ident path, o/w panic!() // match EXPR as a string literal or an ident path, o/w panic!()
let expr = match expr { let mut expr = match expr {
syn::Expr::Lit(expr) => match expr.lit { syn::Expr::Lit(expr_lit) => {
syn::Lit::Str(ref lit) => { match expr_lit.lit {
let mut s = lit.value(); syn::Lit::Str(ref lit_str) => { lit_str.parse::<syn::ExprPath>().unwrap() },
if !s.ends_with("::uumain") { _ => panic!(ARG_PANIC_TEXT),
s += "::uumain";
} }
syn::LitStr::new(&s, proc_macro2::Span::call_site())
.parse()
.unwrap()
}
_ => panic!(),
}, },
syn::Expr::Path(expr) => { syn::Expr::Path(expr_path) => { expr_path },
if &expr.path.segments.last().unwrap().ident.to_string() != "uumain" { _ => panic!(ARG_PANIC_TEXT),
syn::parse_quote!( #expr::uumain )
} else {
expr
}
}
_ => panic!(),
}; };
proc_dbg!(&expr);
// for a single segment ExprPath argument, add trailing '::uumain' segment
if expr.path.segments.len() < 2 {
expr = syn::parse_quote!( #expr::uumain );
};
proc_dbg!(&expr);
let f = quote::quote! { #expr(uucore::args().collect()) }; let f = quote::quote! { #expr(uucore::args().collect()) };
proc_dbg!(&f);
// generate a uutils utility `main()` function, tailored for the calling utility // generate a uutils utility `main()` function, tailored for the calling utility
let result = quote::quote! { let result = quote::quote! {
fn main() { fn main() {