use crate::headers::{HeaderName, HeaderValue, Headers};
pub trait Header {
fn header_name(&self) -> HeaderName;
fn header_value(&self) -> HeaderValue;
fn apply_header<H: AsMut<Headers>>(&self, mut headers: H) {
let name = self.header_name();
let value = self.header_value();
headers.as_mut().insert(name, value).unwrap();
}
}
impl Header for (&str, &str) {
fn header_name(&self) -> HeaderName {
HeaderName::from(self.0)
}
fn header_value(&self) -> HeaderValue {
HeaderValue::from_bytes(self.1.to_owned().into_bytes())
.expect("String slice should be valid ASCII")
}
}
impl<T: Header> Header for &T {
fn header_name(&self) -> HeaderName {
(**self).header_name()
}
fn header_value(&self) -> HeaderValue {
(**self).header_value()
}
}
#[cfg(test)]
mod test {
use super::*;
#[test]
fn header_from_strings() {
let strings = ("Content-Length", "12");
assert_eq!(strings.header_name(), "Content-Length");
assert_eq!(strings.header_value(), "12");
}
}