proto_cli 0.59.0

A multi-language version manager, a unified toolchain.
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

use proto_core::reporter::ProtoConsole;
use starbase_console::ConsoleError;
use starbase_console::ui::{ProgressReporter, ProgressState};
use std::ops::Deref;
use tokio::task::JoinHandle;

pub struct ProgressInstance {
    pub handle: JoinHandle<Result<(), ConsoleError>>,
    pub reporter: ProgressReporter,
}

impl ProgressInstance {
    pub async fn stop(self) -> Result<(), ConsoleError> {
        self.reporter.exit();

        if let Ok(result) = self.handle.await {
            result?;
        }

        Ok(())
    }
}

impl Deref for ProgressInstance {
    type Target = ProgressReporter;

    fn deref(&self) -> &Self::Target {
        &self.reporter
    }
}

pub fn monitor_non_tty_progress(
    console: ProtoConsole,
    reporter: ProgressReporter,
    id: Option<String>,
) -> JoinHandle<Result<(), ConsoleError>> {
    tokio::spawn(Box::pin(async move {
        let mut receiver = reporter.subscribe();

        while let Ok(state) = receiver.recv().await {
            match state {
                ProgressState::Exit => {
                    break;
                }
                ProgressState::Message(message) if !console.out.is_quiet() => {
                    let _ = console.progress(message, id.clone());
                }
                _ => {}
            }
        }

        Ok(())
    }))
}