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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
pub(crate) mod context;
use alloc::vec::Vec;
use crate::{
fmt::{install_builtin_hooks, Hooks},
Report,
};
#[cfg(feature = "std")]
type RwLock<T> = std::sync::RwLock<T>;
// Generally the std mutex is faster than spin, so if both `std` and `hooks` is enabled we use the
// std variant.
#[cfg(all(not(feature = "std"), feature = "hooks"))]
type RwLock<T> = spin::rwlock::RwLock<T>;
static FMT_HOOK: RwLock<Hooks> = RwLock::new(Hooks { inner: Vec::new() });
impl Report<()> {
/// Can be used to globally set a [`Debug`] format hook, for a specific type `T`.
///
/// This hook will be called on every [`Debug`] call, if an attachment with the same type has
/// been found.
///
/// [`Debug`]: core::fmt::Debug
///
/// # Examples
///
/// ```rust
/// use std::io::{Error, ErrorKind};
///
/// use anystack::{
/// Report, IntoReport,
/// };
///
/// struct Suggestion(&'static str);
///
/// Report::install_debug_hook::<Suggestion>(|value, context| {
/// context.push_body(format!("suggestion: {}", value.0));
/// });
///
/// let report =
/// Error::from(ErrorKind::InvalidInput).into_report().attach(Suggestion("oh no, try again"));
///
/// # Report::set_color_mode(anystack::fmt::ColorMode::Emphasis);
/// # fn render(value: String) -> String {
/// # let backtrace = regex::Regex::new(r"backtrace no\. (\d+)\n(?: .*\n)* .*").unwrap();
/// # let backtrace_info = regex::Regex::new(r"backtrace( with (\d+) frames)? \((\d+)\)").unwrap();
/// #
/// # let value = backtrace.replace_all(&value, "backtrace no. $1\n [redacted]");
/// # let value = backtrace_info.replace_all(value.as_ref(), "backtrace ($3)");
/// #
/// # ansi_to_html::convert(value.as_ref()).unwrap()
/// # }
/// #
/// # expect_test::expect_file![concat!(env!("CARGO_MANIFEST_DIR"), "/tests/snapshots/doc/hook__debug_hook.snap")].assert_eq(&render(format!("{report:?}")));
/// #
/// println!("{report:?}");
/// ```
///
/// Which will result in something like:
///
/// <pre>
#[cfg_attr(doc, doc = include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/tests/snapshots/doc/hook__debug_hook.snap")))]
/// </pre>
///
/// This example showcases the ability of hooks to be invoked for attached values.
///
/// ```rust
/// # // this is a lot of boilerplate, if you find a better way, please change this!
///
/// use core::fmt;
/// use anystack::{Report, IntoReport};
///
/// struct Suggestion(&'static str);
///
/// #[derive(Debug)]
/// struct ErrorCode(u64);
///
///
/// #[derive(Debug)]
/// struct UserError;
///
/// impl fmt::Display for UserError {
/// fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
/// f.write_str("invalid user input")
/// }
/// }
///
/// impl core::error::Error for UserError {
/// }
///
/// # pub fn main() {
/// Report::install_debug_hook::<Suggestion>(|Suggestion(value), context| {
/// context.push_body(format!("suggestion: {value}"));
/// });
/// Report::install_debug_hook::<ErrorCode>(|ErrorCode(value), context| {
/// context.push_body(format!("error code: {value}"));
/// });
///
/// let report = UserError.into_report().attach(ErrorCode(420));
///
/// # Report::set_color_mode(anystack::fmt::ColorMode::Emphasis);
/// # fn render(value: String) -> String {
/// # let backtrace = regex::Regex::new(r"backtrace no\. (\d+)\n(?: .*\n)* .*").unwrap();
/// # let backtrace_info = regex::Regex::new(r"backtrace( with (\d+) frames)? \((\d+)\)").unwrap();
/// #
/// # let value = backtrace.replace_all(&value, "backtrace no. $1\n [redacted]");
/// # let value = backtrace_info.replace_all(value.as_ref(), "backtrace ($3)");
/// #
/// # ansi_to_html::convert(value.as_ref()).unwrap()
/// # }
/// #
/// # expect_test::expect_file![concat!(env!("CARGO_MANIFEST_DIR"), "/tests/snapshots/doc/hook__debug_hook_provide.snap")].assert_eq(&render(format!("{report:?}")));
/// #
/// println!("{report:?}");
/// # }
/// ```
///
/// Which will result in something like:
///
/// <pre>
#[cfg_attr(doc, doc = include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/tests/snapshots/doc/hook__debug_hook_provide.snap")))]
/// </pre>
///
/// `anystack` comes with some built-in hooks which can be overwritten. This is useful if you
/// want to change the output of the built-in hooks, or if you want to add additional
/// information to the output. For example, you can override the built-in hook for [`Location`]
/// to hide the file path:
///
/// ```rust
/// use std::{
/// io::{Error, ErrorKind},
/// panic::Location,
/// };
///
/// use anystack::IntoReport;
///
/// anystack::Report::install_debug_hook::<Location>(|_location, _context| {
/// // Intentionally left empty so nothing will be printed
/// });
///
/// let report = Error::from(ErrorKind::InvalidInput).into_report();
///
/// # anystack::Report::set_color_mode(anystack::fmt::ColorMode::Emphasis);
/// # fn render(value: String) -> String {
/// # let backtrace = regex::Regex::new(r"backtrace no\. (\d+)\n(?: .*\n)* .*").unwrap();
/// # let backtrace_info = regex::Regex::new(r"backtrace( with (\d+) frames)? \((\d+)\)").unwrap();
/// #
/// # let value = backtrace.replace_all(&value, "backtrace no. $1\n [redacted]");
/// # let value = backtrace_info.replace_all(value.as_ref(), "backtrace ($3)");
/// #
/// # ansi_to_html::convert(value.as_ref()).unwrap()
/// # }
/// #
/// # expect_test::expect_file![concat!(env!("CARGO_MANIFEST_DIR"), "/tests/snapshots/doc/hook__location_hook.snap")].assert_eq(&render(format!("{report:?}")));
/// #
/// println!("{report:?}");
/// ```
///
/// Which will result in something like:
///
/// <pre>
#[cfg_attr(doc, doc = include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/tests/snapshots/doc/hook__location_hook.snap")))]
/// </pre>
///
/// [`Location`]: std::panic::Location
#[cfg(any(feature = "std", feature = "hooks"))]
pub fn install_debug_hook<T: Send + Sync + 'static>(
hook: impl Fn(&T, &mut crate::fmt::HookContext<T>) + Send + Sync + 'static,
) {
install_builtin_hooks();
#[cfg(feature = "std")]
let Ok(mut lock) = FMT_HOOK.write() else {
unreachable!(
"Hook is poisoned. This is considered a bug and should be reported to \
https://github.com/zakstucke/anystack/issues/new/choose"
);
};
// The spin RwLock cannot panic
#[cfg(all(not(feature = "std"), feature = "hooks"))]
let mut lock = FMT_HOOK.write();
lock.insert(hook);
}
/// Returns the hook that was previously set by [`install_debug_hook`]
///
/// [`install_debug_hook`]: Self::install_debug_hook
#[cfg(any(feature = "std", feature = "hooks"))]
pub(crate) fn invoke_debug_format_hook<T>(closure: impl FnOnce(&Hooks) -> T) -> T {
install_builtin_hooks();
#[cfg(feature = "std")]
let Ok(hook) = FMT_HOOK.read() else {
unreachable!(
"Hook is poisoned. This is considered a bug and should be reported to \
https://github.com/zakstucke/anystack/issues/new/choose"
);
};
// The spin RwLock cannot panic
#[cfg(all(not(feature = "std"), feature = "hooks"))]
let hook = FMT_HOOK.read();
closure(&hook)
}
}