cli-toolbox 0.8.0

Utility library for working with cli output ergonomically
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

use proc_macro2::TokenStream;
use quote::ToTokens;

use crate::common::Message;
use crate::common::tracing::trace_expansion;
use crate::debug_macro::{DebugLnMacro, DebugMacro};

impl ToTokens for DebugLnMacro {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        tokens.extend(trace_expansion(tokenize_debug_message_macro(&self.message)));
    }
}

impl ToTokens for DebugMacro {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        tokens.extend(trace_expansion(
            match self {
                Self::Message(message) =>
                    tokenize_debug_message_macro(message),
                Self::Expr(expr) =>
                    quote! {
                    #[cfg(debug_assertions)]
                    { #expr; }
                }
            }
        ));
    }
}

fn tokenize_debug_message_macro(message: &Message) -> TokenStream {
    let message = message.build_message(false);

    quote! {
        #[cfg(debug_assertions)]
        { #message; }
    }
}