cargo-insert-docs 1.6.0

Inserts feature docs into crate docs, and crate docs into README.
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

//! We use our own span trace because our span formatting does not play well
//! with `Display` impls or `color-spantrace`.
//!
//! There can only be one formatter for `SpanTrace` and we can't get it's internal
//! span so we have to do this ourselves.

use core::fmt;
use std::{error::Error, panic};

use color_eyre::eyre::{EyreHandler, Report};
use tracing::{Level, Span};

pub fn wrap_hook(base_hook: HookFunc) -> HookFunc {
    Box::new(move |e| {
        Box::new(PrettyHandler {
            base: base_hook(e),
            level: Level::ERROR,
            span: Span::current(),
            location: None,
        })
    })
}

pub fn extract_span(report: &Report) -> Option<&Span> {
    report.handler().downcast_ref::<PrettyHandler>().map(|h| &h.span)
}

pub fn extract_location(report: &Report) -> Option<&'static panic::Location<'static>> {
    report.handler().downcast_ref::<PrettyHandler>().and_then(|h| h.location)
}

pub fn extract_severity(report: &Report) -> Level {
    report.handler().downcast_ref::<PrettyHandler>().map(|h| h.level).unwrap_or(Level::ERROR)
}

pub fn set_severity(report: &mut Report, level: Level) {
    if let Some(handler) = report.handler_mut().downcast_mut::<PrettyHandler>() {
        handler.level = level;
    }
}

struct PrettyHandler {
    base: Box<dyn EyreHandler>,
    span: Span,
    level: Level,
    location: Option<&'static panic::Location<'static>>,
}

impl EyreHandler for PrettyHandler {
    fn debug(&self, error: &(dyn Error + 'static), f: &mut fmt::Formatter<'_>) -> fmt::Result {
        self.base.debug(error, f)
    }

    fn display(&self, error: &(dyn Error + 'static), f: &mut fmt::Formatter<'_>) -> fmt::Result {
        self.base.display(error, f)
    }

    fn track_caller(&mut self, location: &'static panic::Location<'static>) {
        self.base.track_caller(location);
        self.location = Some(location);
    }
}

type HookFunc = Box<dyn Fn(&(dyn Error + 'static)) -> Box<dyn EyreHandler> + Send + Sync + 'static>;