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
65
66
67
68
69
70
71
72
73

use std::{fmt::Display, process::Command};

use crate::{ADBCommand, ADBResult};

/// Wait for the device to be in the given state
pub struct ADBWait {
    shell: Command,
}

impl ADBWait {
    pub fn new(transport: Transport, state: State) -> Self {
        let mut cmd = Command::new("adb");
        cmd.arg(format!("wait-for-{}-{}", transport, state));

        ADBWait { shell: cmd }
    }
}

pub enum Transport {
    Usb,
    Local,
    Any,
}

impl Display for Transport {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "{}",
            match self {
                Transport::Usb => "usb",
                Transport::Local => "local",
                Transport::Any => "any",
            }
        )
    }
}

pub enum State {
    Bootloader,
    Device,
    Disconnect,
    Recovery,
    Rescue,
    Sideload,
}

impl Display for State {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "{}",
            match self {
                State::Bootloader => "bootloader",
                State::Device => "device",
                State::Disconnect => "disconnect",
                State::Recovery => "recovery",
                State::Rescue => "rescue",
                State::Sideload => "sideload",
            }
        )
    }
}

impl ADBCommand for ADBWait {
    fn build(&mut self) -> Result<&mut Command, String> {
        Ok(&mut self.shell)
    }

    fn process_output(&self, output: ADBResult) -> ADBResult {
        output
    }
}