stackwise 0.2.0

Drop-in Rust stack usage analysis with JSON reports and an interactive local UI
Documentation 
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

use std::borrow::Cow;
use std::io::{self, IsTerminal};
use std::time::Duration;

use indicatif::{ProgressBar, ProgressDrawTarget, ProgressStyle};

const TICK_INTERVAL: Duration = Duration::from_millis(120);

/// Terminal-aware progress indicator for the analysis pipeline.
///
/// Renders a stderr spinner only when stderr is attached to a terminal so
/// that piped output, CI runs, and tests stay quiet.
pub(crate) struct AnalysisProgress {
    bar: ProgressBar,
}

impl AnalysisProgress {
    pub(crate) fn new() -> Self {
        let bar = ProgressBar::with_draw_target(None, draw_target());
        bar.set_style(spinner_style());
        bar.enable_steady_tick(TICK_INTERVAL);
        Self { bar }
    }

    pub(crate) fn set_stage(&self, message: impl Into<Cow<'static, str>>) {
        self.bar.set_message(message);
    }

    pub(crate) fn finish(&self) {
        self.bar.finish_and_clear();
    }
}

impl Drop for AnalysisProgress {
    fn drop(&mut self) {
        // Safety net for early returns so the spinner never lingers above
        // an error message or a follow-up println.
        self.bar.finish_and_clear();
    }
}

fn draw_target() -> ProgressDrawTarget {
    if io::stderr().is_terminal() {
        ProgressDrawTarget::stderr()
    } else {
        ProgressDrawTarget::hidden()
    }
}

fn spinner_style() -> ProgressStyle {
    ProgressStyle::with_template("{spinner:.cyan} {msg}")
        .unwrap_or_else(|_| ProgressStyle::default_spinner())
}