Skip to content

Commit

Permalink
feat(header): implement fmt::Display for several headers
Browse files Browse the repository at this point in the history
Specifically, `CacheControl`, `Expect`, `Origin`, `Pragma`, `Prefer`,
`PreferenceApplied`, `ReferrerPolicy`, `StrictTransportSecurity`.
  • Loading branch information
seanmonstar committed Feb 1, 2017
1 parent b4b2fb7 commit e9e7381
Show file tree
Hide file tree
Showing 11 changed files with 77 additions and 17 deletions.
33 changes: 19 additions & 14 deletions src/header/common/authorization.rs
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,12 @@ impl<S: Scheme + Any> Header for Authorization<S> where <S as FromStr>::Err: 'st
}

fn fmt_header(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}

impl<S: Scheme> fmt::Display for Authorization<S> {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
if let Some(scheme) = <S as Scheme>::scheme() {
try!(write!(f, "{} ", scheme))
};
Expand Down Expand Up @@ -152,7 +158,6 @@ impl Scheme for Basic {
if let Some(ref pass) = self.password {
text.push_str(&pass[..]);
}

f.write_str(&encode(text.as_ref()))
}
}
Expand Down Expand Up @@ -193,25 +198,25 @@ impl FromStr for Basic {
#[derive(Clone, PartialEq, Debug)]
///Token holder for Bearer Authentication, most often seen with oauth
pub struct Bearer {
///Actual bearer token as a string
pub token: String
///Actual bearer token as a string
pub token: String
}

impl Scheme for Bearer {
fn scheme() -> Option<&'static str> {
Some("Bearer")
}
fn scheme() -> Option<&'static str> {
Some("Bearer")
}

fn fmt_scheme(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{}", self.token)
}
fn fmt_scheme(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{}", self.token)
}
}

impl FromStr for Bearer {
type Err = ::Error;
fn from_str(s: &str) -> ::Result<Bearer> {
Ok(Bearer { token: s.to_owned()})
}
type Err = ::Error;
fn from_str(s: &str) -> ::Result<Bearer> {
Ok(Bearer { token: s.to_owned()})
}
}

#[cfg(test)]
Expand Down Expand Up @@ -265,7 +270,7 @@ mod tests {
assert_eq!(auth.0.password, Some("".to_owned()));
}

#[test]
#[test]
fn test_bearer_auth() {
let mut headers = Headers::new();
headers.set(Authorization(
Expand Down
6 changes: 6 additions & 0 deletions src/header/common/cache_control.rs
Original file line number Diff line number Diff line change
Expand Up @@ -65,6 +65,12 @@ impl Header for CacheControl {
}

fn fmt_header(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}

impl fmt::Display for CacheControl {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt_comma_delimited(f, &self[..])
}
}
Expand Down
7 changes: 7 additions & 0 deletions src/header/common/cookie.rs
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,12 @@ impl Header for Cookie {
}

fn fmt_header(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}

impl fmt::Display for Cookie {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
let cookies = &self.0;
for (i, cookie) in cookies.iter().enumerate() {
if i != 0 {
Expand All @@ -63,6 +69,7 @@ impl Header for Cookie {
try!(Display::fmt(&cookie, f));
}
Ok(())

}
}

Expand Down
6 changes: 6 additions & 0 deletions src/header/common/expect.rs
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,12 @@ impl Header for Expect {
}

fn fmt_header(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}

impl fmt::Display for Expect {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.write_str("100-continue")
}
}
3 changes: 1 addition & 2 deletions src/header/common/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -85,9 +85,8 @@ macro_rules! bench_header(
fn bench_format(b: &mut Bencher) {
let raw = $value.into();
let val: $ty = Header::parse_header(&raw).unwrap();
let fmt = ::header::HeaderFormatter(&val);
b.iter(|| {
format!("{}", fmt);
format!("{}", val);
});
}
}
Expand Down
8 changes: 7 additions & 1 deletion src/header/common/origin.rs
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ impl Header for Origin {
}

fn fmt_header(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{}://{}", self.scheme, self.host)
fmt::Display::fmt(self, f)
}
}

Expand All @@ -83,6 +83,12 @@ impl FromStr for Origin {
}
}

impl fmt::Display for Origin {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{}://{}", self.scheme, self.host)
}
}

impl PartialEq for Origin {
fn eq(&self, other: &Origin) -> bool {
self.scheme == other.scheme && self.host == other.host
Expand Down
6 changes: 6 additions & 0 deletions src/header/common/pragma.rs
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,12 @@ impl Header for Pragma {
}

fn fmt_header(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}

impl fmt::Display for Pragma {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.write_str(match *self {
Pragma::NoCache => "no-cache",
Pragma::Ext(ref string) => &string[..],
Expand Down
6 changes: 6 additions & 0 deletions src/header/common/prefer.rs
Original file line number Diff line number Diff line change
Expand Up @@ -66,6 +66,12 @@ impl Header for Prefer {
}

fn fmt_header(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}

impl fmt::Display for Prefer {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt_comma_delimited(f, &self[..])
}
}
Expand Down
7 changes: 7 additions & 0 deletions src/header/common/preference_applied.rs
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,13 @@ impl Header for PreferenceApplied {
}

fn fmt_header(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}

impl fmt::Display for PreferenceApplied {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
//TODO: format this without allocating a Vec and cloning contents
let preferences: Vec<_> = self.0.iter().map(|pref| match pref {
// The spec ignores parameters in `Preferences-Applied`
&Preference::Extension(ref name, ref value, _) => Preference::Extension(
Expand Down
6 changes: 6 additions & 0 deletions src/header/common/referrer_policy.rs
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,12 @@ impl Header for ReferrerPolicy {
}

fn fmt_header(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}

impl fmt::Display for ReferrerPolicy {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
use self::ReferrerPolicy::*;
f.write_str(match *self {
NoReferrer => "no-referrer",
Expand Down
6 changes: 6 additions & 0 deletions src/header/common/strict_transport_security.rs
Original file line number Diff line number Diff line change
Expand Up @@ -130,6 +130,12 @@ impl Header for StrictTransportSecurity {
}

fn fmt_header(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt::Display::fmt(self, f)
}
}

impl fmt::Display for StrictTransportSecurity {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
if self.include_subdomains {
write!(f, "max-age={}; includeSubdomains", self.max_age)
} else {
Expand Down

0 comments on commit e9e7381

Please sign in to comment.