1 use crate::{Applicability, Handler, Level, StashKey};
2 use crate::{Diagnostic, DiagnosticId, DiagnosticStyledString};
5 use rustc_span::{MultiSpan, Span};
6 use std::fmt::{self, Debug};
7 use std::ops::{Deref, DerefMut};
8 use std::thread::panicking;
10 /// Used for emitting structured error messages and other diagnostic information.
12 /// If there is some state in a downstream crate you would like to
13 /// access in the methods of `DiagnosticBuilder` here, consider
14 /// extending `HandlerFlags`, accessed via `self.handler.flags`.
17 pub struct DiagnosticBuilder<'a>(Box<DiagnosticBuilderInner<'a>>);
19 /// This is a large type, and often used as a return value, especially within
20 /// the frequently-used `PResult` type. In theory, return value optimization
21 /// (RVO) should avoid unnecessary copying. In practice, it does not (at the
22 /// time of writing). The split between `DiagnosticBuilder` and
23 /// `DiagnosticBuilderInner` exists to avoid many `memcpy` calls.
26 struct DiagnosticBuilderInner<'a> {
28 diagnostic: Diagnostic,
29 allow_suggestions: bool,
32 /// In general, the `DiagnosticBuilder` uses deref to allow access to
33 /// the fields and methods of the embedded `diagnostic` in a
34 /// transparent way. *However,* many of the methods are intended to
35 /// be used in a chained way, and hence ought to return `self`. In
36 /// that case, we can't just naively forward to the method on the
37 /// `diagnostic`, because the return type would be a `&Diagnostic`
38 /// instead of a `&DiagnosticBuilder<'a>`. This `forward!` macro makes
39 /// it easy to declare such methods on the builder.
40 macro_rules! forward {
41 // Forward pattern for &self -> &Self
44 pub fn $n:ident(&self, $($name:ident: $ty:ty),* $(,)?) -> &Self
47 pub fn $n(&self, $($name: $ty),*) -> &Self {
48 self.diagnostic.$n($($name),*);
53 // Forward pattern for &mut self -> &mut Self
56 pub fn $n:ident(&mut self, $($name:ident: $ty:ty),* $(,)?) -> &mut Self
59 pub fn $n(&mut self, $($name: $ty),*) -> &mut Self {
60 self.0.diagnostic.$n($($name),*);
65 // Forward pattern for &mut self -> &mut Self, with S: Into<MultiSpan>
66 // type parameter. No obvious way to make this more generic.
69 pub fn $n:ident<S: Into<MultiSpan>>(
71 $($name:ident: $ty:ty),*
76 pub fn $n<S: Into<MultiSpan>>(&mut self, $($name: $ty),*) -> &mut Self {
77 self.0.diagnostic.$n($($name),*);
83 impl<'a> Deref for DiagnosticBuilder<'a> {
84 type Target = Diagnostic;
86 fn deref(&self) -> &Diagnostic {
91 impl<'a> DerefMut for DiagnosticBuilder<'a> {
92 fn deref_mut(&mut self) -> &mut Diagnostic {
93 &mut self.0.diagnostic
97 impl<'a> DiagnosticBuilder<'a> {
98 /// Emit the diagnostic.
99 pub fn emit(&mut self) {
100 self.0.handler.emit_diagnostic(&self);
104 /// Emit the diagnostic unless `delay` is true,
105 /// in which case the emission will be delayed as a bug.
107 /// See `emit` and `delay_as_bug` for details.
108 pub fn emit_unless(&mut self, delay: bool) {
116 /// Stashes diagnostic for possible later improvement in a different,
117 /// later stage of the compiler. The diagnostic can be accessed with
118 /// the provided `span` and `key` through `.steal_diagnostic` on `Handler`.
120 /// As with `buffer`, this is unless the handler has disabled such buffering.
121 pub fn stash(self, span: Span, key: StashKey) {
122 if let Some((diag, handler)) = self.into_diagnostic() {
123 handler.stash_diagnostic(span, key, diag);
127 /// Converts the builder to a `Diagnostic` for later emission,
128 /// unless handler has disabled such buffering.
129 pub fn into_diagnostic(mut self) -> Option<(Diagnostic, &'a Handler)> {
130 if self.0.handler.flags.dont_buffer_diagnostics
131 || self.0.handler.flags.treat_err_as_bug.is_some()
137 let handler = self.0.handler;
139 // We must use `Level::Cancelled` for `dummy` to avoid an ICE about an
140 // unused diagnostic.
141 let dummy = Diagnostic::new(Level::Cancelled, "");
142 let diagnostic = std::mem::replace(&mut self.0.diagnostic, dummy);
144 // Logging here is useful to help track down where in logs an error was
146 debug!("buffer: diagnostic={:?}", diagnostic);
148 Some((diagnostic, handler))
151 /// Buffers the diagnostic for later emission,
152 /// unless handler has disabled such buffering.
153 pub fn buffer(self, buffered_diagnostics: &mut Vec<Diagnostic>) {
154 buffered_diagnostics.extend(self.into_diagnostic().map(|(diag, _)| diag));
157 /// Convenience function for internal use, clients should use one of the
158 /// span_* methods instead.
159 pub fn sub<S: Into<MultiSpan>>(
165 let span = span.map(|s| s.into()).unwrap_or_else(|| MultiSpan::new());
166 self.0.diagnostic.sub(level, message, span, None);
170 /// Delay emission of this diagnostic as a bug.
172 /// This can be useful in contexts where an error indicates a bug but
173 /// typically this only happens when other compilation errors have already
174 /// happened. In those cases this can be used to defer emission of this
175 /// diagnostic as a bug in the compiler only if no other errors have been
178 /// In the meantime, though, callsites are required to deal with the "bug"
179 /// locally in whichever way makes the most sense.
180 pub fn delay_as_bug(&mut self) {
181 self.level = Level::Bug;
182 self.0.handler.delay_as_bug(self.0.diagnostic.clone());
186 /// Adds a span/label to be included in the resulting snippet.
187 /// This is pushed onto the `MultiSpan` that was created when the
188 /// diagnostic was first built. If you don't call this function at
189 /// all, and you just supplied a `Span` to create the diagnostic,
190 /// then the snippet will just include that `Span`, which is
191 /// called the primary span.
192 pub fn span_label(&mut self, span: Span, label: impl Into<String>) -> &mut Self {
193 self.0.diagnostic.span_label(span, label);
197 /// Labels all the given spans with the provided label.
198 /// See `span_label` for more information.
201 spans: impl IntoIterator<Item = Span>,
202 label: impl AsRef<str>,
204 let label = label.as_ref();
206 self.0.diagnostic.span_label(span, label);
211 forward!(pub fn note_expected_found(
213 expected_label: &dyn fmt::Display,
214 expected: DiagnosticStyledString,
215 found_label: &dyn fmt::Display,
216 found: DiagnosticStyledString,
219 forward!(pub fn note_expected_found_extra(
221 expected_label: &dyn fmt::Display,
222 expected: DiagnosticStyledString,
223 found_label: &dyn fmt::Display,
224 found: DiagnosticStyledString,
225 expected_extra: &dyn fmt::Display,
226 found_extra: &dyn fmt::Display,
229 forward!(pub fn note_unsuccessfull_coercion(
231 expected: DiagnosticStyledString,
232 found: DiagnosticStyledString,
235 forward!(pub fn note(&mut self, msg: &str) -> &mut Self);
236 forward!(pub fn span_note<S: Into<MultiSpan>>(
241 forward!(pub fn warn(&mut self, msg: &str) -> &mut Self);
242 forward!(pub fn span_warn<S: Into<MultiSpan>>(&mut self, sp: S, msg: &str) -> &mut Self);
243 forward!(pub fn help(&mut self, msg: &str) -> &mut Self);
244 forward!(pub fn span_help<S: Into<MultiSpan>>(
250 pub fn multipart_suggestion(
253 suggestion: Vec<(Span, String)>,
254 applicability: Applicability,
256 if !self.0.allow_suggestions {
259 self.0.diagnostic.multipart_suggestion(msg, suggestion, applicability);
263 pub fn tool_only_multipart_suggestion(
266 suggestion: Vec<(Span, String)>,
267 applicability: Applicability,
269 if !self.0.allow_suggestions {
272 self.0.diagnostic.tool_only_multipart_suggestion(msg, suggestion, applicability);
276 pub fn span_suggestion(
281 applicability: Applicability,
283 if !self.0.allow_suggestions {
286 self.0.diagnostic.span_suggestion(sp, msg, suggestion, applicability);
290 pub fn span_suggestions(
294 suggestions: impl Iterator<Item = String>,
295 applicability: Applicability,
297 if !self.0.allow_suggestions {
300 self.0.diagnostic.span_suggestions(sp, msg, suggestions, applicability);
304 pub fn span_suggestion_short(
309 applicability: Applicability,
311 if !self.0.allow_suggestions {
314 self.0.diagnostic.span_suggestion_short(sp, msg, suggestion, applicability);
318 pub fn span_suggestion_hidden(
323 applicability: Applicability,
325 if !self.0.allow_suggestions {
328 self.0.diagnostic.span_suggestion_hidden(sp, msg, suggestion, applicability);
332 pub fn tool_only_span_suggestion(
337 applicability: Applicability,
339 if !self.0.allow_suggestions {
342 self.0.diagnostic.tool_only_span_suggestion(sp, msg, suggestion, applicability);
346 forward!(pub fn set_span<S: Into<MultiSpan>>(&mut self, sp: S) -> &mut Self);
347 forward!(pub fn code(&mut self, s: DiagnosticId) -> &mut Self);
349 pub fn allow_suggestions(&mut self, allow: bool) -> &mut Self {
350 self.0.allow_suggestions = allow;
354 /// Convenience function for internal use, clients should use one of the
355 /// struct_* methods on Handler.
356 crate fn new(handler: &'a Handler, level: Level, message: &str) -> DiagnosticBuilder<'a> {
357 DiagnosticBuilder::new_with_code(handler, level, None, message)
360 /// Convenience function for internal use, clients should use one of the
361 /// struct_* methods on Handler.
362 crate fn new_with_code(
363 handler: &'a Handler,
365 code: Option<DiagnosticId>,
367 ) -> DiagnosticBuilder<'a> {
368 let diagnostic = Diagnostic::new_with_code(level, code, message);
369 DiagnosticBuilder::new_diagnostic(handler, diagnostic)
372 /// Creates a new `DiagnosticBuilder` with an already constructed
374 crate fn new_diagnostic(handler: &'a Handler, diagnostic: Diagnostic) -> DiagnosticBuilder<'a> {
375 debug!("Created new diagnostic");
376 DiagnosticBuilder(Box::new(DiagnosticBuilderInner {
379 allow_suggestions: true,
384 impl<'a> Debug for DiagnosticBuilder<'a> {
385 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
386 self.0.diagnostic.fmt(f)
390 /// Destructor bomb - a `DiagnosticBuilder` must be either emitted or canceled
391 /// or we emit a bug.
392 impl<'a> Drop for DiagnosticBuilder<'a> {
394 if !panicking() && !self.cancelled() {
395 let mut db = DiagnosticBuilder::new(
398 "the following error was constructed but not emitted",
408 macro_rules! struct_span_err {
409 ($session:expr, $span:expr, $code:ident, $($message:tt)*) => ({
410 $session.struct_span_err_with_code(
412 &format!($($message)*),
413 $crate::error_code!($code),
419 macro_rules! error_code {
420 ($code:ident) => {{ $crate::DiagnosticId::Error(stringify!($code).to_owned()) }};