use std::cmp::Ordering;
use std::str::FromStr;
use http::header::{HeaderName, InvalidHeaderName};
#[derive(Debug, Copy, Clone, PartialOrd, Ord, PartialEq, Eq)]
#[non_exhaustive]
pub enum PseudoHeader {
RequestTarget,
Created,
Expires,
}
impl PseudoHeader {
pub fn as_str(&self) -> &str {
match self {
PseudoHeader::RequestTarget => "(request-target)",
PseudoHeader::Created => "(created)",
PseudoHeader::Expires => "(expires)",
}
}
}
impl FromStr for PseudoHeader {
type Err = ();
fn from_str(s: &str) -> Result<PseudoHeader, Self::Err> {
match s {
"(request-target)" => Ok(PseudoHeader::RequestTarget),
"(created)" => Ok(PseudoHeader::Created),
"(expires)" => Ok(PseudoHeader::Expires),
_ => Err(()),
}
}
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum Header {
Pseudo(PseudoHeader),
Normal(HeaderName),
}
impl Header {
pub fn as_str(&self) -> &str {
match self {
Header::Pseudo(h) => h.as_str(),
Header::Normal(h) => h.as_str(),
}
}
}
impl FromStr for Header {
type Err = InvalidHeaderName;
fn from_str(s: &str) -> Result<Header, Self::Err> {
PseudoHeader::from_str(s)
.map(Into::into)
.or_else(|_| HeaderName::from_str(s).map(Into::into))
}
}
impl Ord for Header {
fn cmp(&self, other: &Header) -> Ordering {
self.as_str().cmp(other.as_str())
}
}
impl PartialOrd for Header {
fn partial_cmp(&self, other: &Header) -> Option<Ordering> {
Some(self.cmp(other))
}
}
impl From<HeaderName> for Header {
fn from(other: HeaderName) -> Self {
Header::Normal(other)
}
}
impl From<PseudoHeader> for Header {
fn from(other: PseudoHeader) -> Self {
Header::Pseudo(other)
}
}