lib.rs - OpenGrok cross reference for /aosp_15_r20/external/rust/android-crates-io/crates/debug_tree/src/lib.rs

use std::sync::{Arc, Mutex};

#[macro_use]
pub mod default;
mod internal;
pub mod scoped_branch;

pub mod defer;
mod test;
pub mod tree_config;

pub use default::default_tree;
use once_cell::sync::Lazy;
use scoped_branch::ScopedBranch;
use std::collections::BTreeMap;
use std::fs::File;
use std::io::Write;

pub use crate::tree_config::*;

/// Reference wrapper for `TreeBuilderBase`
#[derive(Debug, Clone)]
pub struct TreeBuilder(Arc<Mutex<internal::TreeBuilderBase>>);

impl TreeBuilder {
    /// Returns a new `TreeBuilder` with an empty `Tree`.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// ```
    pub fn new() -> TreeBuilder {
        TreeBuilder {
            0: Arc::new(Mutex::new(internal::TreeBuilderBase::new())),
        }
    }

    /// Set the configuration override for displaying trees
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::{TreeBuilder, add_branch_to, add_leaf_to, TreeSymbols, TreeConfig};
    /// let tree = TreeBuilder::new();
    /// {
    ///     add_branch_to!(tree, "1");
    ///     {
    ///         add_branch_to!(tree, "1.1");
    ///         add_leaf_to!(tree, "1.1.1");
    ///         add_leaf_to!(tree, "1.1.2");
    ///     }
    ///     add_leaf_to!(tree, "1.2");
    /// }
    /// add_leaf_to!(tree, "2");
    /// tree.set_config_override(TreeConfig::new()
    ///     .show_first_level()
    ///     .symbols(TreeSymbols::with_rounded()));
    /// tree.peek_print();
    /// assert_eq!("\
    /// ├╼ 1
    /// │ ├╼ 1.1
    /// │ │ ├╼ 1.1.1
    /// │ │ ╰╼ 1.1.2
    /// │ ╰╼ 1.2
    /// ╰╼ 2" , &tree.string());
    /// ```
    pub fn set_config_override(&self, config: TreeConfig) {
        let mut lock = self.0.lock().unwrap();
        lock.set_config_override(Some(config))
    }

    /// Remove the configuration override
    /// The default configuration will be used instead
    pub fn remove_config_override(&self) {
        self.0.lock().unwrap().set_config_override(None);
    }

    /// Update the configuration override for displaying trees
    /// If an override doesn't yet exist, it is created.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::{TreeBuilder, add_branch_to, add_leaf_to, TreeSymbols};
    /// let tree = TreeBuilder::new();
    /// {
    ///     add_branch_to!(tree, "1");
    ///     {
    ///         add_branch_to!(tree, "1.1");
    ///         add_leaf_to!(tree, "1.1.1");
    ///         add_leaf_to!(tree, "1.1.2");
    ///     }
    ///     add_leaf_to!(tree, "1.2");
    /// }
    /// add_leaf_to!(tree, "2");
    /// tree.update_config_override(|x|{
    ///     x.indent = 3;
    ///     x.symbols = TreeSymbols::with_rounded();
    ///     x.show_first_level = true;
    /// });
    /// tree.peek_print();
    /// assert_eq!("\
    /// ├─╼ 1
    /// │  ├─╼ 1.1
    /// │  │  ├─╼ 1.1.1
    /// │  │  ╰─╼ 1.1.2
    /// │  ╰─╼ 1.2
    /// ╰─╼ 2" , &tree.string());
    /// ```
    pub fn update_config_override<F: Fn(&mut TreeConfig)>(&self, update: F) {
        let mut lock = self.0.lock().unwrap();
        match lock.config_override_mut() {
            Some(x) => update(x),
            None => {
                let mut x = TreeConfig::default();
                update(&mut x);
                lock.set_config_override(Some(x));
            }
        }
    }

    /// Returns the optional configuration override.
    pub fn get_config_override(&self) -> Option<TreeConfig> {
        let lock = self.0.lock().unwrap();
        lock.config_override().clone()
    }

    /// Returns whether a configuration override is set.
    pub fn has_config_override(&self) -> bool {
        let lock = self.0.lock().unwrap();
        lock.config_override().is_some()
    }

    /// Adds a new branch with text, `text` and returns a `ScopedBranch`.
    /// When the returned `ScopedBranch` goes out of scope, (likely the end of the current block),
    /// or if its `release()` method is called, the tree will step back out of the added branch.
    ///
    /// # Arguments
    /// * `text` - A string slice to use as the newly added branch's text.
    ///
    /// # Examples
    ///
    /// Exiting branch when end of scope is reached.
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// {
    ///     let _branch = tree.add_branch("Branch"); // _branch enters scope
    ///     // tree is now pointed inside new branch.
    ///     tree.add_leaf("Child of Branch");
    ///     // _branch leaves scope, tree moves up to parent branch.
    /// }
    /// tree.add_leaf("Sibling of Branch");
    /// assert_eq!("\
    /// Branch
    /// └╼ Child of Branch
    /// Sibling of Branch" , &tree.string());
    /// ```
    ///
    /// Using `release()` before out of scope.
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// {
    ///     let mut branch = tree.add_branch("Branch"); // branch enters scope
    ///     // tree is now pointed inside new branch.
    ///     tree.add_leaf("Child of Branch");
    ///     branch.release();
    ///     tree.add_leaf("Sibling of Branch");
    ///     // branch leaves scope, but no effect because its `release()` method has already been called
    /// }
    /// assert_eq!("\
    /// Branch
    /// └╼ Child of Branch
    /// Sibling of Branch", &tree.string());
    /// ```
    pub fn add_branch(&self, text: &str) -> ScopedBranch {
        self.add_leaf(text);
        ScopedBranch::new(self.clone())
    }

    /// Adds a new branch with text, `text` and returns a `ScopedBranch`.
    /// When the returned `ScopedBranch` goes out of scope, (likely the end of the current block),
    /// or if its `release()` method is called, the tree tree will step back out of the added branch.
    ///
    /// # Arguments
    /// * `text` - A string slice to use as the newly added branch's text.
    ///
    /// # Examples
    ///
    /// Stepping out of branch when end of scope is reached.
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// {
    ///     tree.add_leaf("Branch");
    ///     let _branch = tree.enter_scoped(); // _branch enters scope
    ///     // tree is now pointed inside new branch.
    ///     tree.add_leaf("Child of Branch");
    ///     // _branch leaves scope, tree moves up to parent branch.
    /// }
    /// tree.add_leaf("Sibling of Branch");
    /// assert_eq!("\
    /// Branch
    /// └╼ Child of Branch
    /// Sibling of Branch", &tree.string());
    /// ```
    ///
    /// Using `release()` before out of scope.
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// {
    ///     tree.add_leaf("Branch");
    ///     let mut branch = tree.enter_scoped(); // branch enters scope
    ///     // tree is now pointed inside new branch.
    ///     tree.add_leaf("Child of Branch");
    ///     branch.release();
    ///     tree.add_leaf("Sibling of Branch");
    ///     // branch leaves scope, but no effect because its `release()` method has already been called
    /// }
    /// assert_eq!("\
    /// Branch
    /// └╼ Child of Branch
    /// Sibling of Branch", &tree.string());
    /// ```
    pub fn enter_scoped(&self) -> ScopedBranch {
        if self.is_enabled() {
            ScopedBranch::new(self.clone())
        } else {
            ScopedBranch::none()
        }
    }

    /// Adds a leaf to current branch with the given text, `text`.
    ///
    /// # Arguments
    /// * `text` - A string slice to use as the newly added leaf's text.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// tree.add_leaf("New leaf");
    /// ```
    pub fn add_leaf(&self, text: &str) {
        let mut x = self.0.lock().unwrap();
        if x.is_enabled() {
            x.add_leaf(&text);
        }
    }

    /// Steps into a new child branch.
    /// Stepping out of the branch requires calling `exit()`.
    ///
    /// # Example
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// tree.add_leaf("Branch");
    /// tree.enter();
    /// tree.add_leaf("Child of Branch");
    /// assert_eq!("\
    /// Branch
    /// └╼ Child of Branch", &tree.string());
    /// ```
    pub fn enter(&self) {
        let mut x = self.0.lock().unwrap();
        if x.is_enabled() {
            x.enter();
        }
    }

    /// Exits the current branch, to the parent branch.
    /// If no parent branch exists, no action is taken
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// tree.add_leaf("Branch");
    /// tree.enter();
    /// tree.add_leaf("Child of Branch");
    /// tree.exit();
    /// tree.add_leaf("Sibling of Branch");
    /// assert_eq!("\
    /// Branch
    /// └╼ Child of Branch
    /// Sibling of Branch", &tree.string());
    /// ```
    pub fn exit(&self) -> bool {
        let mut x = self.0.lock().unwrap();
        if x.is_enabled() {
            x.exit()
        } else {
            false
        }
    }

    /// Returns the depth of the current branch
    /// The initial depth when no branches have been adeed is 0.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// assert_eq!(0, tree.depth());
    /// let _b = tree.add_branch("Branch");
    /// assert_eq!(1, tree.depth());
    /// let _b = tree.add_branch("Child branch");
    /// assert_eq!(2, tree.depth());
    /// ```
    pub fn depth(&self) -> usize {
        self.0.lock().unwrap().depth()
    }

    /// Prints the tree without clearing.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// tree.add_leaf("Leaf");
    /// tree.peek_print();
    /// // Leaf
    /// tree.peek_print();
    /// // Leaf
    /// // Leaf 2
    /// ```
    pub fn peek_print(&self) {
        self.0.lock().unwrap().peek_print();
    }

    /// Prints the tree and then clears it.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// tree.add_leaf("Leaf");
    /// tree.print();
    /// // Leaf
    /// tree.add_leaf("Leaf 2");
    /// tree.print();
    /// // Leaf 2
    /// ```
    pub fn print(&self) {
        self.0.lock().unwrap().print();
    }

    /// Returns the tree as a string without clearing the tree.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// tree.add_leaf("Leaf");
    /// assert_eq!("Leaf", tree.peek_string());
    /// tree.add_leaf("Leaf 2");
    /// assert_eq!("Leaf\nLeaf 2", tree.peek_string());
    /// ```
    pub fn peek_string(&self) -> String {
        self.0.lock().unwrap().peek_string()
    }

    /// Returns the tree as a string and clears the tree.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// tree.add_leaf("Leaf");
    /// assert_eq!("Leaf", tree.string());
    /// tree.add_leaf("Leaf 2");
    /// assert_eq!("Leaf 2", tree.string());
    /// ```
    pub fn string(&self) -> String {
        self.0.lock().unwrap().string()
    }

    /// Writes the tree to file without clearing.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// use std::fs::{read_to_string, create_dir};
    /// use std::io::Read;
    /// let tree = TreeBuilder::new();
    /// create_dir("test_out").ok();
    /// tree.add_leaf("Leaf");
    /// assert_eq!(tree.peek_string(), "Leaf");
    /// tree.peek_write("test_out/peek_write.txt");
    /// assert_eq!(read_to_string("test_out/peek_write.txt").unwrap(), "Leaf");
    /// assert_eq!(tree.peek_string(), "Leaf");
    /// ```
    pub fn peek_write(&self, path: &str) -> std::io::Result<()> {
        let mut file = File::create(path)?;
        file.write_all(self.peek_string().as_bytes())
    }

    /// Writes the tree to file without clearing.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// use std::io::Read;
    /// use std::fs::{read_to_string, create_dir};
    /// let tree = TreeBuilder::new();
    /// create_dir("test_out").ok();
    /// tree.add_leaf("Leaf");
    /// assert_eq!(tree.peek_string(), "Leaf");
    /// tree.write("test_out/write.txt");
    /// assert_eq!(read_to_string("test_out/write.txt").unwrap(), "Leaf");
    /// assert_eq!(tree.peek_string(), "");
    /// ```
    pub fn write(&self, path: &str) -> std::io::Result<()> {
        let mut file = File::create(path)?;
        file.write_all(self.string().as_bytes())
    }

    /// Clears the tree.
    ///
    /// # Example
    ///
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let tree = TreeBuilder::new();
    /// tree.add_leaf("Leaf");
    /// assert_eq!("Leaf", tree.peek_string());
    /// tree.clear();
    /// assert_eq!("", tree.peek_string());
    /// ```
    pub fn clear(&self) {
        self.0.lock().unwrap().clear()
    }

    /// Sets the enabled state of the tree.
    ///
    /// If not enabled, the tree will not be modified by adding leaves or branches.
    /// Additionally, if called using the `add_`... macros, arguments will not be processed.
    /// This is particularly useful for suppressing output in production, with very little overhead.
    ///
    /// # Example
    /// ```
    /// #[macro_use]
    /// use debug_tree::{TreeBuilder, add_leaf_to};
    /// let mut tree = TreeBuilder::new();
    /// tree.add_leaf("Leaf 1");
    /// tree.set_enabled(false);
    /// add_leaf_to!(tree, "Leaf 2");
    /// tree.set_enabled(true);
    /// add_leaf_to!(tree, "Leaf 3");
    /// assert_eq!("Leaf 1\nLeaf 3", tree.peek_string());
    /// ```
    pub fn set_enabled(&self, enabled: bool) {
        self.0.lock().unwrap().set_enabled(enabled);
    }

    /// Returns the enabled state of the tree.
    ///
    /// # Example
    /// ```
    /// use debug_tree::TreeBuilder;
    /// let mut tree = TreeBuilder::new();
    /// assert_eq!(true, tree.is_enabled());
    /// tree.set_enabled(false);
    /// assert_eq!(false, tree.is_enabled());
    /// ```
    pub fn is_enabled(&self) -> bool {
        self.0.lock().unwrap().is_enabled()
    }
}

pub trait AsTree {
    fn as_tree(&self) -> TreeBuilder;
    fn is_tree_enabled(&self) -> bool {
        self.as_tree().is_enabled()
    }
}

impl AsTree for TreeBuilder {
    fn as_tree(&self) -> TreeBuilder {
        self.clone()
    }
}

pub(crate) fn get_or_add_tree<T: AsRef<str>>(name: T) -> TreeBuilder {
    let mut map = TREE_MAP.lock().unwrap();
    match map.get(name.as_ref()) {
        Some(x) => x.clone(),
        _ => {
            let val = TreeBuilder::new();
            map.insert(name.as_ref().to_string(), val.clone());
            val
        }
    }
}

pub(crate) fn get_tree<T: AsRef<str>>(name: T) -> Option<TreeBuilder> {
    TREE_MAP.lock().unwrap().get(name.as_ref()).cloned()
}

type TreeMap = BTreeMap<String, TreeBuilder>;

static TREE_MAP: Lazy<Arc<Mutex<TreeMap>>> =
    Lazy::new(|| -> Arc<Mutex<TreeMap>> { Arc::new(Mutex::new(TreeMap::new())) });

/// Sets the enabled state of the tree.
///
/// # Arguments
/// * `name` - The tree name
/// * `enabled` - The enabled state
///
pub fn set_enabled<T: AsRef<str>>(name: T, enabled: bool) {
    let mut map = TREE_MAP.lock().unwrap();
    match map.get_mut(name.as_ref()) {
        Some(x) => x.set_enabled(enabled),
        _ => {
            let tree = TreeBuilder::new();
            tree.set_enabled(enabled);
            map.insert(name.as_ref().to_string(), tree);
        }
    }
}

impl<T: AsRef<str>> AsTree for T {
    fn as_tree(&self) -> TreeBuilder {
        get_or_add_tree(self)
    }
    /// Check if the named tree is enabled and exists
    /// This does not create a new tree if non-existent
    ///
    /// # Arguments
    /// * `tree_name` - The tree name
    ///
    fn is_tree_enabled(&self) -> bool {
        get_tree(self).map(|x| x.is_enabled()).unwrap_or(false)
    }
}

/// Returns the tree
/// If there is no tree then one is created and then returned.
pub fn tree<T: AsTree>(tree: T) -> TreeBuilder {
    tree.as_tree()
}

/// Returns the tree named `name`
/// If there is no tree named `name` then one is created and then returned.
pub fn is_tree_enabled<T: AsTree>(tree: &T) -> bool {
    tree.is_tree_enabled()
}

/// Calls [clear](TreeBuilder::clear) for the tree named `name`
/// If there is no tree named `name` then one is created
pub fn clear<T: AsRef<str>>(name: T) {
    name.as_tree().clear();
}

/// Returns [string](TreeBuilder::string) for the tree named `name`
/// If there is no tree named `name` then one is created
pub fn string<T: AsRef<str>>(name: T) -> String {
    name.as_tree().string()
}

/// Returns [peek_string](TreeBuilder::peek_string) for the tree named `name`
/// If there is no tree named `name` then one is created
pub fn peek_string<T: AsRef<str>>(name: T) -> String {
    name.as_tree().peek_string()
}

/// Calls [print](TreeBuilder::print) for the tree named `name`
/// If there is no tree named `name` then one is created
pub fn print<T: AsRef<str>>(name: T) {
    name.as_tree().print();
}

/// Calls [peek_print](TreeBuilder::peek_print)  for the tree named `name`
/// If there is no tree named `name` then one is created
pub fn peek_print<T: AsRef<str>>(name: T) {
    name.as_tree().peek_print();
}

/// Calls [write](TreeBuilder::write) for the tree named `name`
/// If there is no tree named `name` then one is created
pub fn write<T: AsRef<str>, P: AsRef<str>>(name: T, path: P) -> std::io::Result<()> {
    name.as_tree().write(path.as_ref())
}

/// Calls [peek_print](TreeBuilder::peek_print)  for the tree named `name`
/// If there is no tree named `name` then one is created
pub fn peek_write<T: AsRef<str>, P: AsRef<str>>(name: T, path: P) -> std::io::Result<()> {
    name.as_tree().peek_write(path.as_ref())
}

/// Adds a leaf to given tree with the given text and formatting arguments
///
/// # Arguments
/// * `tree` - The tree that the leaf should be added to
/// * `text...` - Formatted text arguments, as per `format!(...)`.
///
/// # Example
///
/// ```
/// #[macro_use]
/// use debug_tree::{TreeBuilder, add_leaf_to};
/// fn main() {
///     let tree = TreeBuilder::new();
///     add_leaf_to!(tree, "A {} leaf", "new");
///     assert_eq!("A new leaf", &tree.peek_string());
/// }
/// ```
#[macro_export]
macro_rules! add_leaf_to {
    ($tree:expr, $($arg:tt)*) => (if $crate::is_tree_enabled(&$tree) {
        use $crate::AsTree;
        $tree.as_tree().add_leaf(&format!($($arg)*))
    });
}

/// Adds a leaf to given tree with the given `value` argument
///
/// # Arguments
/// * `tree` - The tree that the leaf should be added to
/// * `value` - An expression that implements the `Display` trait.
///
/// # Example
///
/// ```
/// #[macro_use]
/// use debug_tree::{TreeBuilder, add_leaf_value_to};
/// fn main() {
///     let tree = TreeBuilder::new();
///     let value = add_leaf_value_to!(tree, 5 * 4 * 3 * 2);
///     assert_eq!(120, value);
///     assert_eq!("120", &tree.peek_string());
/// }
/// ```
#[macro_export]
macro_rules! add_leaf_value_to {
    ($tree:expr, $value:expr) => {{
        let v = $value;
        if $crate::is_tree_enabled(&$tree) {
            use $crate::AsTree;
            $tree.as_tree().add_leaf(&format!("{}", &v));
        }
        v
    }};
}

/// Adds a scoped branch to given tree with the given text and formatting arguments
/// The branch will be exited at the end of the current block.
///
/// # Arguments
/// * `tree` - The tree that the leaf should be added to
/// * `text...` - Formatted text arguments, as per `format!(...)`.
///
/// # Example
///
/// ```
/// #[macro_use]
/// use debug_tree::{TreeBuilder, add_branch_to, add_leaf_to};
/// fn main() {
///     let tree = TreeBuilder::new();
///     {
///         add_branch_to!(tree, "New {}", "Branch"); // _branch enters scope
///         // tree is now pointed inside new branch.
///         add_leaf_to!(tree, "Child of {}", "Branch");
///         // Block ends, so tree exits the current branch.
///     }
///     add_leaf_to!(tree, "Sibling of {}", "Branch");
///     assert_eq!("\
/// New Branch
/// └╼ Child of Branch
/// Sibling of Branch" , &tree.string());
/// }
/// ```
#[macro_export]
macro_rules! add_branch_to {
    ($tree:expr) => {
        let _debug_tree_branch = if $crate::is_tree_enabled(&$tree) {
            use $crate::AsTree;
            $tree.as_tree().enter_scoped()
        } else {
            $crate::scoped_branch::ScopedBranch::none()
        };
    };
    ($tree:expr, $($arg:tt)*) => {
        let _debug_tree_branch = if $crate::is_tree_enabled(&$tree) {
            use $crate::AsTree;
            $tree.as_tree().add_branch(&format!($($arg)*))
        } else {
            $crate::scoped_branch::ScopedBranch::none()
        };
    };
}

/// Calls `function` with argument, `tree`, at the end of the current scope
/// The function will only be executed if the tree is enabled when this macro is called
#[macro_export]
macro_rules! defer {
    ($function:expr) => {
        let _debug_tree_defer = {
            use $crate::AsTree;
            if $crate::default::default_tree().is_enabled() {
                use $crate::AsTree;
                $crate::defer::DeferredFn::new($crate::default::default_tree(), $function)
            } else {
                $crate::defer::DeferredFn::none()
            }
        };
    };
    ($tree:expr, $function:expr) => {
        let _debug_tree_defer = {
            use $crate::AsTree;
            if $tree.as_tree().is_enabled() {
                $crate::defer::DeferredFn::new($tree.as_tree(), $function)
            } else {
                $crate::defer::DeferredFn::none()
            }
        };
    };
}

/// Calls [print](TreeBuilder::print) on `tree` at the end of the current scope.
/// The function will only be executed if the tree is enabled when this macro is called
#[macro_export]
macro_rules! defer_print {
    () => {
        $crate::defer!(|x| {
            x.print();
        })
    };
    ($tree:expr) => {
        $crate::defer!($tree, |x| {
            x.print();
        })
    };
}

/// Calls [peek_print](TreeBuilder::peek_print) on `tree` at the end of the current scope.
/// The function will only be executed if the tree is enabled when this macro is called
#[macro_export]
macro_rules! defer_peek_print {
    () => {
        $crate::defer!(|x| {
            x.peek_print();
        })
    };
    ($tree:expr) => {
        $crate::defer!($tree, |x| {
            x.peek_print();
        })
    };
}

/// Calls [write](TreeBuilder::write) on `tree` at the end of the current scope.
/// The function will only be executed if the tree is enabled when this macro is called
#[macro_export]
macro_rules! defer_write {
    ($tree:expr, $path:expr) => {
        $crate::defer!($tree, |x| {
            if let Err(err) = x.write($path) {
                eprintln!("error during `defer_write`: {}", err);
            }
        })
    };
    ($path:expr) => {
        $crate::defer!(|x| {
            if let Err(err) = x.write($path) {
                eprintln!("error during `defer_write`: {}", err);
            }
        })
    };
}

/// Calls [peek_write](TreeBuilder::peek_write) on `tree` at the end of the current scope.
/// The function will only be executed if the tree is enabled when this macro is called
#[macro_export]
macro_rules! defer_peek_write {
    ($tree:expr, $path:expr) => {
        $crate::defer!($tree, |x| {
            if let Err(err) = x.peek_write($path) {
                eprintln!("error during `defer_peek_write`: {}", err);
            }
        })
    };
    ($path:expr) => {
        $crate::defer!(|x| {
            if let Err(err) = x.peek_write($path) {
                eprintln!("error during `defer_peek_write`: {}", err);
            }
        })
    };
}