pub mod error;
pub use error::Error;
pub const CODE: &[u8] = b"10";
pub const DEFAULT_MESSAGE: &str = "Input expected";
pub struct Default(String);
impl Default {
pub fn from_utf8(buffer: &[u8]) -> Result<Self, Error> {
if !buffer.starts_with(CODE) {
return Err(Error::Code);
}
Ok(Self(
std::str::from_utf8(
crate::client::connection::response::header_bytes(buffer).map_err(Error::Header)?,
)
.map_err(Error::Utf8Error)?
.to_string(),
))
}
pub fn message(&self) -> Option<&str> {
self.0.get(2..).map(|s| s.trim()).filter(|x| !x.is_empty())
}
pub fn message_or_default(&self) -> &str {
self.message().unwrap_or(DEFAULT_MESSAGE)
}
pub fn as_str(&self) -> &str {
&self.0
}
pub fn as_bytes(&self) -> &[u8] {
self.0.as_bytes()
}
}
#[test]
fn test() {
let d = Default::from_utf8("10 Default\r\n".as_bytes()).unwrap();
assert_eq!(d.message(), Some("Default"));
assert_eq!(d.message_or_default(), "Default");
assert_eq!(d.as_str(), "10 Default\r\n");
assert_eq!(d.as_bytes(), "10 Default\r\n".as_bytes());
let d = Default::from_utf8("10\r\n".as_bytes()).unwrap();
assert_eq!(d.message(), None);
assert_eq!(d.message_or_default(), DEFAULT_MESSAGE);
assert_eq!(d.as_str(), "10\r\n");
assert_eq!(d.as_bytes(), "10\r\n".as_bytes());
assert!(Default::from_utf8("13 Fail\r\n".as_bytes()).is_err());
assert!(Default::from_utf8("Fail\r\n".as_bytes()).is_err());
assert!(Default::from_utf8("Fail".as_bytes()).is_err());
}