src/emit.rs


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

use proc_macro::TokenStream;
use proc_macro2::Span;
use quote::quote;
use syn::Error;

#[derive(Copy, Clone)]
pub enum Kind {
    Enum,
    Match,
    Let,
}

pub fn emit(err: Error, kind: Kind, original: TokenStream) -> TokenStream {
    let mut err = err;
    if !probably_has_spans(kind) {
        // Otherwise the error is printed without any line number.
        err = Error::new(Span::call_site(), &err.to_string());
    }

    let err = err.to_compile_error();
    let original = proc_macro2::TokenStream::from(original);

    let expanded = match kind {
        Kind::Enum | Kind::Let => quote!(#err #original),
        Kind::Match => quote!({ #err #original }),
    };

    TokenStream::from(expanded)
}

// Rustc is so bad at spans.
// https://github.com/rust-lang/rust/issues/43081
fn probably_has_spans(kind: Kind) -> bool {
    match kind {
        Kind::Enum => true,
        Kind::Match | Kind::Let => false,
    }
}