mirror of
https://github.com/YGGverse/ggemini.git
synced 2026-04-01 01:25:32 +00:00
81 lines
2.7 KiB
Rust
81 lines
2.7 KiB
Rust
pub mod error;
|
|
pub use error::Error;
|
|
|
|
/// [Server Unavailable](https://geminiprotocol.net/docs/protocol-specification.gmi#status-41-server-unavailable)
|
|
/// temporary error status code
|
|
pub const CODE: &[u8] = b"41";
|
|
|
|
/// Default message if the optional value was not provided by the server
|
|
/// * useful to skip match cases in external applications,
|
|
/// by using `super::message_or_default` method.
|
|
pub const DEFAULT_MESSAGE: &str = "Server unavailable";
|
|
|
|
/// Hold header `String` for [Server Unavailable](https://geminiprotocol.net/docs/protocol-specification.gmi#status-41-server-unavailable)
|
|
/// temporary error status code
|
|
///
|
|
/// * this response type does not contain body data
|
|
/// * the header member is closed to require valid construction
|
|
pub struct ServerUnavailable(String);
|
|
|
|
impl ServerUnavailable {
|
|
// Constructors
|
|
|
|
/// Parse `Self` from buffer contains header bytes
|
|
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(),
|
|
))
|
|
}
|
|
|
|
// Getters
|
|
|
|
/// Get optional message for `Self`
|
|
/// * return `None` if the message is empty
|
|
pub fn message(&self) -> Option<&str> {
|
|
self.0.get(2..).map(|s| s.trim()).filter(|x| !x.is_empty())
|
|
}
|
|
|
|
/// Get optional message for `Self`
|
|
/// * if the optional message not provided by the server, return `DEFAULT_MESSAGE`
|
|
pub fn message_or_default(&self) -> &str {
|
|
self.message().unwrap_or(DEFAULT_MESSAGE)
|
|
}
|
|
|
|
/// Get header string of `Self`
|
|
pub fn as_str(&self) -> &str {
|
|
&self.0
|
|
}
|
|
|
|
/// Get header bytes of `Self`
|
|
pub fn as_bytes(&self) -> &[u8] {
|
|
self.0.as_bytes()
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test() {
|
|
// ok
|
|
let su = ServerUnavailable::from_utf8("41 Message\r\n".as_bytes()).unwrap();
|
|
assert_eq!(su.message(), Some("Message"));
|
|
assert_eq!(su.message_or_default(), "Message");
|
|
assert_eq!(su.as_str(), "41 Message\r\n");
|
|
assert_eq!(su.as_bytes(), "41 Message\r\n".as_bytes());
|
|
|
|
let su = ServerUnavailable::from_utf8("41\r\n".as_bytes()).unwrap();
|
|
assert_eq!(su.message(), None);
|
|
assert_eq!(su.message_or_default(), DEFAULT_MESSAGE);
|
|
assert_eq!(su.as_str(), "41\r\n");
|
|
assert_eq!(su.as_bytes(), "41\r\n".as_bytes());
|
|
|
|
// err
|
|
assert!(ServerUnavailable::from_utf8("13 Fail\r\n".as_bytes()).is_err());
|
|
assert!(ServerUnavailable::from_utf8("Fail\r\n".as_bytes()).is_err());
|
|
assert!(ServerUnavailable::from_utf8("Fail".as_bytes()).is_err());
|
|
}
|