rustfmt_config_proc_macro/
lib.rs1#![recursion_limit = "256"]
4
5mod attrs;
6mod config_type;
7mod item_enum;
8mod item_struct;
9mod utils;
10
11use std::str::FromStr;
12
13use proc_macro::TokenStream;
14use syn::parse_macro_input;
15
16#[proc_macro_attribute]
17pub fn config_type(_args: TokenStream, input: TokenStream) -> TokenStream {
18 let input = parse_macro_input!(input as syn::Item);
19 let output = config_type::define_config_type(&input);
20
21 #[cfg(feature = "debug-with-rustfmt")]
22 {
23 utils::debug_with_rustfmt(&output);
24 }
25
26 TokenStream::from(output)
27}
28
29#[proc_macro_attribute]
41pub fn nightly_only_test(_args: TokenStream, input: TokenStream) -> TokenStream {
42 if option_env!("CFG_RELEASE_CHANNEL").map_or(true, |c| c == "nightly" || c == "dev") {
44 input
45 } else {
46 TokenStream::from_str("").unwrap()
48 }
49}
50
51#[proc_macro_attribute]
63pub fn stable_only_test(_args: TokenStream, input: TokenStream) -> TokenStream {
64 if option_env!("CFG_RELEASE_CHANNEL").map_or(false, |c| c == "stable") {
66 input
67 } else {
68 TokenStream::from_str("").unwrap()
70 }
71}
72
73#[proc_macro_attribute]
76pub fn rustfmt_only_ci_test(_args: TokenStream, input: TokenStream) -> TokenStream {
77 if option_env!("RUSTFMT_CI").is_some() {
78 input
79 } else {
80 let mut token_stream = TokenStream::from_str("#[ignore]").unwrap();
81 token_stream.extend(input);
82 token_stream
83 }
84}