omaha_client/state_machine/
observer.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
39
40
41
42
43
44
45
46
47
48
49
// Copyright 2019 The Fuchsia Authors
//
// Licensed under a BSD-style license <LICENSE-BSD>, Apache License, Version 2.0
// <LICENSE-APACHE or https://www.apache.org/licenses/LICENSE-2.0>, or the MIT
// license <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your option.
// This file may not be copied, modified, or distributed except according to
// those terms.

use crate::{
    common::{ProtocolState, UpdateCheckSchedule},
    installer::ProgressObserver,
    protocol::response::Response,
    state_machine::{update_check, State, UpdateCheckError},
};
use futures::{channel::mpsc, future::BoxFuture, prelude::*};

/// Events emitted by the state machine.
#[derive(Debug)]
pub enum StateMachineEvent {
    StateChange(State),
    ScheduleChange(UpdateCheckSchedule),
    ProtocolStateChange(ProtocolState),
    UpdateCheckResult(Result<update_check::Response, UpdateCheckError>),
    InstallProgressChange(InstallProgress),
    OmahaServerResponse(Response),
    InstallerError(Option<Box<dyn std::error::Error + Send + 'static>>),
}

#[derive(Debug)]
pub struct InstallProgress {
    pub progress: f32,
}

pub(super) struct StateMachineProgressObserver(pub(super) mpsc::Sender<InstallProgress>);

impl ProgressObserver for StateMachineProgressObserver {
    fn receive_progress(
        &self,
        _operation: Option<&str>,
        progress: f32,
        _total_size: Option<u64>,
        _size_so_far: Option<u64>,
    ) -> BoxFuture<'_, ()> {
        async move {
            let _ = self.0.clone().send(InstallProgress { progress }).await;
        }
        .boxed()
    }
}