use std::convert::TryFrom;
use thiserror::Error;
use crate::body::Body;
use crate::Fragment;
use std::hash::{Hash, Hasher};
#[derive(Debug, Clone, Eq, Ord, PartialOrd)]
pub struct Trailer {
key: String,
value: String,
}
impl Trailer {
#[must_use]
pub fn new(key: &str, value: &str) -> Trailer {
Trailer {
key: String::from(key),
value: String::from(value),
}
}
#[must_use]
pub fn get_key(&self) -> String {
self.key.clone()
}
#[must_use]
pub fn get_value(&self) -> String {
self.value.clone()
}
}
impl PartialEq for Trailer {
fn eq(&self, other: &Self) -> bool {
self.key == other.key && self.value.trim_end() == other.value.trim_end()
}
}
impl Hash for Trailer {
fn hash<H: Hasher>(&self, state: &mut H) {
self.key.hash(state);
self.value.trim_end().hash(state);
}
}
impl From<Trailer> for String {
fn from(trailer: Trailer) -> Self {
format!("{}: {}", trailer.key, trailer.value)
}
}
impl From<Trailer> for Fragment {
fn from(trailer: Trailer) -> Self {
let trailer: String = trailer.into();
Fragment::Body(Body::from(trailer))
}
}
impl TryFrom<Body> for Trailer {
type Error = Error;
fn try_from(body: Body) -> Result<Self, Self::Error> {
let body_value = String::from(body.clone());
let value_and_key: Vec<&str> = body_value.split(": ").collect();
let key = value_and_key
.get(0)
.ok_or_else(|| Error::NotATrailer(body.clone()))?;
let value = value_and_key.get(1).ok_or(Error::NotATrailer(body))?;
Ok(Trailer::new(key, value))
}
}
#[derive(Error, Debug)]
pub enum Error {
#[error("no colon in body line, {0} is not a trailer")]
NotATrailer(Body),
}
#[cfg(test)]
mod tests {
use std::convert::TryFrom;
use pretty_assertions::assert_eq;
use crate::body::Body;
use crate::Fragment;
use super::Trailer;
use std::collections::hash_map::DefaultHasher;
use std::hash::{Hash, Hasher};
#[test]
fn it_can_tell_me_its_key() {
let trailer = Trailer::new("Relates-to", "#128");
assert_eq!(trailer.get_key(), String::from("Relates-to"))
}
#[test]
fn it_can_tell_me_its_value() {
let trailer = Trailer::new("Relates-to", "#128");
assert_eq!(trailer.get_value(), String::from("#128"))
}
#[test]
fn it_does_not_take_trailing_whitespace_into_account_in_equality_checks() {
let a = Trailer::new("Relates-to", "#128\n");
let b = Trailer::new("Relates-to", "#128");
assert_eq!(a, b)
}
#[test]
fn it_does_not_take_trailing_whitespace_into_account_in_hashing() {
let mut hasher_a = DefaultHasher::new();
Trailer::new("Relates-to", "#128\n").hash(&mut hasher_a);
let mut hasher_b = DefaultHasher::new();
Trailer::new("Relates-to", "#128").hash(&mut hasher_b);
assert_eq!(hasher_a.finish(), hasher_b.finish())
}
#[test]
fn it_can_give_me_itself_as_a_string() {
let trailer = Trailer::new("Relates-to", "#128");
assert_eq!(String::from(trailer), String::from("Relates-to: #128"))
}
#[test]
fn can_generate_itself_from_body() {
let trailer = Trailer::try_from(Body::from("Relates-to: #128"));
assert_eq!(
String::from(trailer.expect("Could not parse from string")),
String::from("Relates-to: #128")
)
}
#[test]
fn it_preserves_preceding_whitespace() {
let trailer = Trailer::try_from(Body::from("Relates-to: #128\n"));
assert_eq!(
String::from(trailer.expect("Could not parse from string")),
String::from("Relates-to: #128\n")
)
}
#[test]
fn can_generate_from_body() {
let trailer = Trailer::new("Relates-to", "#128");
let body: Fragment = Fragment::from(trailer);
assert_eq!(body, Fragment::Body(Body::from("Relates-to: #128")))
}
}