fncmd_impl/
lib.rs

1#![feature(proc_macro_span)]
2
3use darling::FromMeta;
4use proc_macro::{Span, TokenStream};
5use proc_macro_error::proc_macro_error;
6use syn::{parse_macro_input, AttributeArgs, ItemFn};
7
8mod models;
9use models::*;
10mod context;
11use context::*;
12
13/// The `fncmd` macro. This can only be applied to the `main` function.
14///
15/// ```rust
16/// /// Description of the command line tool
17/// #[fncmd::fncmd]
18/// pub fn main(
19///   /// Argument foo
20///   #[opt(short, long)]
21///   foo: String,
22///   /// Argument bar
23///   #[opt(short, long)]
24///   bar: Option<String>,
25/// ) {
26///   println!("{:?} {:?}", foo, bar);
27/// }
28/// ```
29#[proc_macro_error]
30#[proc_macro_attribute]
31pub fn fncmd(attr: TokenStream, item: TokenStream) -> TokenStream {
32	// Get information about the target and package.
33	let call_site = Span::call_site();
34	let (target, package) = CONTEXT.get_target_and_package_of(&call_site);
35	let name = target.name.clone();
36	let version = package.version.to_string();
37	let subcmds = FncmdSubcmds::from((target, package));
38
39	// Parse the input tokens.
40	let attr = parse_macro_input!(attr as AttributeArgs);
41	let attr = FncmdAttr::from_list(&attr).unwrap();
42	let item = parse_macro_input!(item as ItemFn);
43	Fncmd::parse(name, version, attr, item, subcmds).into()
44}
fncmd_impl/lib.rs

fncmd_impl/
lib.rs
