you-can-build-macros 0.0.1

ಠ_ಠ
Documentation 
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

use quote::ToTokens;

#[track_caller]
pub fn print(tokens: &impl Print) {
    tokens.print();
}

pub trait Print {
    fn print(&self);
}

impl<T> Print for T
where
    T: ToTokens,
{
    fn print(&self) {
        let tokens = &proc_macro::TokenStream::from(self.to_token_stream());

        let printed: Result<(), Box<dyn std::error::Error>> = (|| {
            let pseudo_file: syn::File = syn::parse(tokens.clone())?;
            let pretty = prettyplease::unparse(&pseudo_file);
            let mut printer = bat::PrettyPrinter::new();
            printer.input_from_bytes(pretty.as_ref()).language("rust");
            printer.print()?;
            Ok(())
        })();

        if printed.is_err() {
            let pretty = tokens.to_string();
            let mut printer = bat::PrettyPrinter::new();
            printer.input_from_bytes(pretty.as_ref()).language("rust");
            printer.print().unwrap();
        }
    }
}