use hyper;
use hyper::header::{self, Header, Raw};
use std::fmt;
#[derive(Debug, Clone, Copy)]
pub enum Trailer {
StreamError,
}
impl Header for Trailer {
fn header_name() -> &'static str {
"Trailer"
}
fn parse_header(raw: &Raw) -> hyper::Result<Trailer> {
if let Some(bytes) = raw.one() {
let value = String::from_utf8_lossy(bytes);
match value.as_ref() {
"X-Stream-Error" => Ok(Trailer::StreamError),
_ => Err(hyper::Error::Header),
}
} else {
Err(hyper::Error::Header)
}
}
fn fmt_header(&self, f: &mut header::Formatter) -> fmt::Result {
let value = match *self {
Trailer::StreamError => "X-Stream-Error",
};
f.fmt_line(&value)
}
}
#[derive(Debug, Clone)]
pub struct XStreamError {
pub error: String,
}
impl Header for XStreamError {
fn header_name() -> &'static str {
"X-Stream-Error"
}
fn parse_header(raw: &Raw) -> hyper::Result<XStreamError> {
if let Some(bytes) = raw.one() {
let value = String::from_utf8_lossy(bytes);
Ok(XStreamError { error: value.into_owned() })
} else {
Err(hyper::Error::Header)
}
}
fn fmt_header(&self, f: &mut header::Formatter) -> fmt::Result {
f.fmt_line(&self.error)
}
}