use std::time::Duration;
use bytes::Bytes;
#[derive(Debug, Eq, PartialEq, Clone)]
pub enum CmdWaitFor {
Nothing,
StdOutMessage { message: Bytes },
StdErrMessage { message: Bytes },
Duration { length: Duration },
Exit { code: Option<i64> },
}
impl CmdWaitFor {
pub fn message_on_stdout(message: impl AsRef<[u8]>) -> Self {
Self::StdOutMessage {
message: Bytes::from(message.as_ref().to_vec()),
}
}
pub fn message_on_stderr(message: impl AsRef<[u8]>) -> Self {
Self::StdErrMessage {
message: Bytes::from(message.as_ref().to_vec()),
}
}
pub fn exit() -> Self {
Self::Exit { code: None }
}
pub fn exit_code(code: i64) -> Self {
Self::Exit { code: Some(code) }
}
pub fn duration(duration: Duration) -> Self {
Self::Duration { length: duration }
}
pub fn seconds(secs: u64) -> Self {
Self::duration(Duration::from_secs(secs))
}
pub fn millis(millis: u64) -> Self {
Self::duration(Duration::from_millis(millis))
}
}