1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
mod codegen;
mod parser;
use proc_macro::TokenStream;
use syn::Expr;
use syn::punctuated::Punctuated;
/// Format a string with Symfony Console-style tags.
///
/// Supported tags: `<info>`, `<comment>`, `<error>`, `<question>`, `<highlight>`, `<warning>`.
///
/// # Examples
///
/// ```ignore
/// // Single tagged segment
/// console_format!("<info>All packages are up to date.</info>")
///
/// // With format arguments
/// console_format!("<info>Removing {name} from require-dev</info>")
///
/// // Mixed tags
/// console_format!("<info>{}</info> : <comment>{}</comment>", label, value)
///
/// // Plain text (equivalent to format!)
/// console_format!("plain text {}", x)
/// ```
#[proc_macro]
pub fn console_format(input: TokenStream) -> TokenStream {
let input2: proc_macro2::TokenStream = input.into();
match console_format_impl(input2) {
Ok(tokens) => tokens.into(),
Err(err) => err.into_compile_error().into(),
}
}
fn console_format_impl(
input: proc_macro2::TokenStream,
) -> Result<proc_macro2::TokenStream, syn::Error> {
let args: ConsoleFormatArgs = syn::parse2(input)?;
let segments = parser::parse_format_string(&args.format_str)
.map_err(|msg| syn::Error::new(args.format_str_span, msg))?;
Ok(codegen::generate(
&segments,
&args.extra_args,
args.format_str_span,
))
}
struct ConsoleFormatArgs {
format_str: String,
format_str_span: proc_macro2::Span,
extra_args: Punctuated<Expr, syn::Token![,]>,
}
impl syn::parse::Parse for ConsoleFormatArgs {
fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
let lit: syn::LitStr = input.parse()?;
let format_str = lit.value();
let format_str_span = lit.span();
let extra_args = if input.peek(syn::Token![,]) {
input.parse::<syn::Token![,]>()?;
Punctuated::parse_terminated(input)?
} else {
Punctuated::new()
};
Ok(ConsoleFormatArgs {
format_str,
format_str_span,
extra_args,
})
}
}
|
