update/
monitor_state.rs

1// Copyright 2021 The Fuchsia Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5use fidl_fuchsia_update::{MonitorRequest, MonitorRequestStream};
6use fidl_fuchsia_update_ext::State;
7use futures::prelude::*;
8use std::io::Write;
9
10fn print_state(state: &State) {
11    if termion::is_tty(&std::io::stdout()) {
12        print!("\r{}\x1b[K", state);
13        if state.is_terminal() {
14            println!();
15        }
16    } else {
17        println!("State: {state:?}")
18    }
19    std::io::stdout().flush().unwrap();
20}
21
22pub async fn monitor_state(mut stream: MonitorRequestStream) -> Result<(), anyhow::Error> {
23    while let Some(event) = stream.try_next().await? {
24        match event {
25            MonitorRequest::OnState { state, responder } => {
26                responder.send()?;
27
28                let state = State::from(state);
29
30                // Exit if we encounter an error during an update.
31                if state.is_error() {
32                    anyhow::bail!("Update failed: {:?}", state)
33                } else {
34                    print_state(&state);
35                }
36            }
37        }
38    }
39    Ok(())
40}
update/monitor_state.rs

update/
monitor_state.rs
