dojo-cairo-macros 1.7.0

use cairo_lang_macro::ProcMacroResult;

use crate::helpers::DiagnosticExt;

/// Print the output of a proc macro (output token stream and diagnostics),
/// if the global DOJO_DEBUG_MACRO or the specific `DOJO_DEBUG_<ELEMENT>_MACRO`
/// environment variable is set.
///
/// TODO: once fully integrated to the Dojo repo, this function should
/// be removed and `scarb expand` should be used instead.
pub fn debug_macro(element: &str, res: &ProcMacroResult) {
    let element = element.to_ascii_uppercase();

    if std::env::var("DOJO_DEBUG_MACRO").is_ok()
        || std::env::var(format!("DOJO_DEBUG_{element}_MACRO")).is_ok()
    {
        let content = format!("content:\n{}", res.token_stream);
        let diagnostics = if res.diagnostics.is_empty() {
            "".to_string()
        } else {
            format!(
                "diagnostics:\n{}",
                res.diagnostics.iter().map(|d| d.to_pretty_string()).collect::<Vec<_>>().join("\n")
            )
        };

        println!(
            "\n*> MACRO {element}\n>>>>>>>>>>>>>>>>>>>>>>>>>>>\n{content}\n{diagnostics}\\
             n<<<<<<<<<<<<<<<<<<<<<<<<<<<"
        );
    }
}

/// Prints the given string only if the 'DOJO_STORE_EXPAND' environment variable is set.
/// This is useful for debugging DojoStore implementation.
pub fn debug_store_expand(element_name: &str, code: &str) {
    if std::env::var("DOJO_STORE_EXPAND").is_ok() {
        println!(
            "\n*> EXPAND {} <*\n>>>>>>>>>>>>>>>>>>>>>>>>>>>\n{}\n<<<<<<<<<<<<<<<<<<<<<<<<<<<\n",
            element_name, code
        );
    }
}