feat(headers): adds Accept
Moved utils to shared/. Added quality_value.
This commit is contained in:
		| @@ -1,7 +1,9 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use std::fmt::{mod, Show}; | ||||
| use std::str::from_utf8; | ||||
| use mime::Mime; | ||||
| use std::fmt; | ||||
|  | ||||
| use header; | ||||
| use header::shared; | ||||
|  | ||||
| use mime; | ||||
|  | ||||
| /// The `Accept` header. | ||||
| /// | ||||
| @@ -14,60 +16,53 @@ use mime::Mime; | ||||
| /// ``` | ||||
| /// # use hyper::header::Headers; | ||||
| /// # use hyper::header::common::Accept; | ||||
| /// # use hyper::header::shared::qitem; | ||||
| /// use hyper::mime::Mime; | ||||
| /// use hyper::mime::TopLevel::Text; | ||||
| /// use hyper::mime::SubLevel::{Html, Xml}; | ||||
| /// # let mut headers = Headers::new(); | ||||
| /// headers.set(Accept(vec![ Mime(Text, Html, vec![]), Mime(Text, Xml, vec![]) ])); | ||||
| /// headers.set(Accept(vec![ | ||||
| ///     qitem(Mime(Text, Html, vec![])), | ||||
| ///     qitem(Mime(Text, Xml, vec![])) ])); | ||||
| /// ``` | ||||
| #[deriving(Clone, PartialEq, Show)] | ||||
| pub struct Accept(pub Vec<Mime>); | ||||
| pub struct Accept(pub Vec<shared::QualityItem<mime::Mime>>); | ||||
|  | ||||
| deref!(Accept -> Vec<Mime>); | ||||
| deref!(Accept -> Vec<shared::QualityItem<mime::Mime>>); | ||||
|  | ||||
| impl Header for Accept { | ||||
| impl header::Header for Accept { | ||||
|     fn header_name(_: Option<Accept>) -> &'static str { | ||||
|         "Accept" | ||||
|     } | ||||
|  | ||||
|     fn parse_header(raw: &[Vec<u8>]) -> Option<Accept> { | ||||
|         let mut mimes: Vec<Mime> = vec![]; | ||||
|         for mimes_raw in raw.iter() { | ||||
|             match from_utf8(mimes_raw.as_slice()) { | ||||
|                 Ok(mimes_str) => { | ||||
|                     for mime_str in mimes_str.split(',') { | ||||
|                         match mime_str.trim().parse() { | ||||
|                             Some(mime) => mimes.push(mime), | ||||
|                             None => return None | ||||
|                         } | ||||
|                     } | ||||
|                 }, | ||||
|                 Err(_) => return None | ||||
|             }; | ||||
|         } | ||||
|  | ||||
|         if !mimes.is_empty() { | ||||
|             Some(Accept(mimes)) | ||||
|         } else { | ||||
|             // Currently is just a None, but later it can be Accept for */* | ||||
|             None | ||||
|         } | ||||
|         // TODO: Return */* if no value is given. | ||||
|         shared::from_comma_delimited(raw).map(Accept) | ||||
|     } | ||||
| } | ||||
|  | ||||
| impl HeaderFormat for Accept { | ||||
| impl header::HeaderFormat for Accept { | ||||
|     fn fmt_header(&self, fmt: &mut fmt::Formatter) -> fmt::Result { | ||||
|         let Accept(ref value) = *self; | ||||
|         let last = value.len() - 1; | ||||
|         for (i, mime) in value.iter().enumerate() { | ||||
|             try!(mime.fmt(fmt)); | ||||
|             if i < last { | ||||
|                 try!(", ".fmt(fmt)); | ||||
|             } | ||||
|         } | ||||
|         Ok(()) | ||||
|         shared::fmt_comma_delimited(fmt, self[]) | ||||
|     } | ||||
| } | ||||
|  | ||||
| bench_header!(bench, Accept, { vec![b"text/plain; q=0.5, text/html".to_vec()] }); | ||||
|  | ||||
| #[test] | ||||
| fn test_parse_header_no_quality() { | ||||
|     let a: Accept = header::Header::parse_header([b"text/plain; charset=utf-8".to_vec()].as_slice()).unwrap(); | ||||
|     let b = Accept(vec![ | ||||
|         shared::QualityItem{item: mime::Mime(mime::TopLevel::Text, mime::SubLevel::Plain, vec![(mime::Attr::Charset, mime::Value::Utf8)]), quality: 1f32}, | ||||
|     ]); | ||||
|     assert_eq!(a, b); | ||||
| } | ||||
|  | ||||
| #[test] | ||||
| fn test_parse_header_with_quality() { | ||||
|     let a: Accept = header::Header::parse_header([b"text/plain; charset=utf-8; q=0.5".to_vec()].as_slice()).unwrap(); | ||||
|     let b = Accept(vec![ | ||||
|         shared::QualityItem{item: mime::Mime(mime::TopLevel::Text, mime::SubLevel::Plain, vec![(mime::Attr::Charset, mime::Value::Utf8)]), quality: 0.5f32}, | ||||
|     ]); | ||||
|     assert_eq!(a, b); | ||||
| } | ||||
|   | ||||
							
								
								
									
										39
									
								
								src/header/common/accept_encoding.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										39
									
								
								src/header/common/accept_encoding.rs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,39 @@ | ||||
| use std::fmt; | ||||
|  | ||||
| use header; | ||||
| use header::shared; | ||||
|  | ||||
| /// The `Accept-Encoding` header | ||||
| /// | ||||
| /// The `Accept-Encoding` header can be used by clients to indicate what | ||||
| /// response encodings they accept. | ||||
| #[deriving(Clone, PartialEq, Show)] | ||||
| pub struct AcceptEncoding(pub Vec<shared::QualityItem<shared::Encoding>>); | ||||
|  | ||||
| deref!(AcceptEncoding -> Vec<shared::QualityItem<shared::Encoding>>); | ||||
|  | ||||
| impl header::Header for AcceptEncoding { | ||||
|     fn header_name(_: Option<AcceptEncoding>) -> &'static str { | ||||
|         "AcceptEncoding" | ||||
|     } | ||||
|  | ||||
|     fn parse_header(raw: &[Vec<u8>]) -> Option<AcceptEncoding> { | ||||
|         shared::from_comma_delimited(raw).map(AcceptEncoding) | ||||
|     } | ||||
| } | ||||
|  | ||||
| impl header::HeaderFormat for AcceptEncoding { | ||||
|     fn fmt_header(&self, fmt: &mut fmt::Formatter) -> fmt::Result { | ||||
|         shared::fmt_comma_delimited(fmt, self[]) | ||||
|     } | ||||
| } | ||||
|  | ||||
| #[test] | ||||
| fn test_parse_header() { | ||||
|     let a: AcceptEncoding = header::Header::parse_header([b"gzip;q=1.0, identity; q=0.5".to_vec()].as_slice()).unwrap(); | ||||
|     let b = AcceptEncoding(vec![ | ||||
|         shared::QualityItem{item: shared::Gzip, quality: 1f32}, | ||||
|         shared::QualityItem{item: shared::Identity, quality: 0.5f32}, | ||||
|     ]); | ||||
|     assert_eq!(a, b); | ||||
| } | ||||
| @@ -1,7 +1,7 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use method::Method; | ||||
| use std::fmt::{mod}; | ||||
| use super::util::{from_comma_delimited, fmt_comma_delimited}; | ||||
| use header::shared::util::{from_comma_delimited, fmt_comma_delimited}; | ||||
|  | ||||
| /// The `Allow` header. | ||||
| /// See also https://tools.ietf.org/html/rfc7231#section-7.4.1 | ||||
| @@ -46,4 +46,3 @@ mod tests { | ||||
| } | ||||
|  | ||||
| bench_header!(bench, Allow, { vec![b"OPTIONS,GET,PUT,POST,DELETE,HEAD,TRACE,CONNECT,PATCH,fOObAr".to_vec()] }); | ||||
|  | ||||
|   | ||||
| @@ -1,7 +1,7 @@ | ||||
| use std::fmt; | ||||
| use std::str::FromStr; | ||||
| use header::{Header, HeaderFormat}; | ||||
| use super::util::{from_one_comma_delimited, fmt_comma_delimited}; | ||||
| use header::shared::util::{from_one_comma_delimited, fmt_comma_delimited}; | ||||
|  | ||||
| /// The Cache-Control header. | ||||
| #[deriving(PartialEq, Clone, Show)] | ||||
|   | ||||
| @@ -1,7 +1,7 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use std::fmt::{mod, Show}; | ||||
| use std::str::FromStr; | ||||
| use super::util::{from_comma_delimited, fmt_comma_delimited}; | ||||
| use header::shared::util::{from_comma_delimited, fmt_comma_delimited}; | ||||
|  | ||||
| pub use self::ConnectionOption::{KeepAlive, Close, ConnectionHeader}; | ||||
|  | ||||
|   | ||||
| @@ -1,7 +1,7 @@ | ||||
| use std::fmt::{mod, Show}; | ||||
|  | ||||
| use header::{Header, HeaderFormat}; | ||||
| use super::util::from_one_raw_str; | ||||
| use header::shared::util::from_one_raw_str; | ||||
|  | ||||
| /// The `Content-Length` header. | ||||
| /// | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use std::fmt::{mod, Show}; | ||||
| use super::util::from_one_raw_str; | ||||
| use header::shared::util::from_one_raw_str; | ||||
| use mime::Mime; | ||||
|  | ||||
| /// The `Content-Type` header. | ||||
|   | ||||
| @@ -2,7 +2,8 @@ use std::fmt::{mod, Show}; | ||||
| use std::str::FromStr; | ||||
| use time::Tm; | ||||
| use header::{Header, HeaderFormat}; | ||||
| use super::util::{from_one_raw_str, tm_from_str}; | ||||
| use header::shared::util::from_one_raw_str; | ||||
| use header::shared::time::tm_from_str; | ||||
|  | ||||
| // Egh, replace as soon as something better than time::Tm exists. | ||||
| /// The `Date` header field. | ||||
| @@ -41,4 +42,3 @@ impl FromStr for Date { | ||||
| bench_header!(imf_fixdate, Date, { vec![b"Sun, 07 Nov 1994 08:48:37 GMT".to_vec()] }); | ||||
| bench_header!(rfc_850, Date, { vec![b"Sunday, 06-Nov-94 08:49:37 GMT".to_vec()] }); | ||||
| bench_header!(asctime, Date, { vec![b"Sun Nov  6 08:49:37 1994".to_vec()] }); | ||||
|  | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use std::fmt::{mod}; | ||||
| use super::util::from_one_raw_str; | ||||
| use header::shared::util::from_one_raw_str; | ||||
|  | ||||
| /// The `Etag` header. | ||||
| /// | ||||
|   | ||||
| @@ -2,7 +2,8 @@ use std::fmt::{mod, Show}; | ||||
| use std::str::FromStr; | ||||
| use time::Tm; | ||||
| use header::{Header, HeaderFormat}; | ||||
| use super::util::{from_one_raw_str, tm_from_str}; | ||||
| use header::shared::util::from_one_raw_str; | ||||
| use header::shared::time::tm_from_str; | ||||
|  | ||||
| /// The `Expires` header field. | ||||
| #[deriving(Copy, PartialEq, Clone)] | ||||
| @@ -40,4 +41,3 @@ impl FromStr for Expires { | ||||
| bench_header!(imf_fixdate, Expires, { vec![b"Sun, 07 Nov 1994 08:48:37 GMT".to_vec()] }); | ||||
| bench_header!(rfc_850, Expires, { vec![b"Sunday, 06-Nov-94 08:49:37 GMT".to_vec()] }); | ||||
| bench_header!(asctime, Expires, { vec![b"Sun Nov  6 08:49:37 1994".to_vec()] }); | ||||
|  | ||||
|   | ||||
| @@ -1,7 +1,7 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use Port; | ||||
| use std::fmt::{mod, Show}; | ||||
| use super::util::from_one_raw_str; | ||||
| use header::shared::util::from_one_raw_str; | ||||
|  | ||||
| /// The `Host` header. | ||||
| /// | ||||
|   | ||||
| @@ -2,7 +2,8 @@ use std::fmt::{mod, Show}; | ||||
| use std::str::FromStr; | ||||
| use time::Tm; | ||||
| use header::{Header, HeaderFormat}; | ||||
| use super::util::{from_one_raw_str, tm_from_str}; | ||||
| use header::shared::util::from_one_raw_str; | ||||
| use header::shared::time::tm_from_str; | ||||
|  | ||||
| /// The `If-Modified-Since` header field. | ||||
| #[deriving(Copy, PartialEq, Clone)] | ||||
| @@ -40,4 +41,3 @@ impl FromStr for IfModifiedSince { | ||||
| bench_header!(imf_fixdate, IfModifiedSince, { vec![b"Sun, 07 Nov 1994 08:48:37 GMT".to_vec()] }); | ||||
| bench_header!(rfc_850, IfModifiedSince, { vec![b"Sunday, 06-Nov-94 08:49:37 GMT".to_vec()] }); | ||||
| bench_header!(asctime, IfModifiedSince, { vec![b"Sun Nov  6 08:49:37 1994".to_vec()] }); | ||||
|  | ||||
|   | ||||
| @@ -2,7 +2,8 @@ use std::fmt::{mod, Show}; | ||||
| use std::str::FromStr; | ||||
| use time::Tm; | ||||
| use header::{Header, HeaderFormat}; | ||||
| use super::util::{from_one_raw_str, tm_from_str}; | ||||
| use header::shared::util::from_one_raw_str; | ||||
| use header::shared::time::tm_from_str; | ||||
|  | ||||
| /// The `LastModified` header field. | ||||
| #[deriving(Copy, PartialEq, Clone)] | ||||
| @@ -40,4 +41,3 @@ impl FromStr for LastModified { | ||||
| bench_header!(imf_fixdate, LastModified, { vec![b"Sun, 07 Nov 1994 08:48:37 GMT".to_vec()] }); | ||||
| bench_header!(rfc_850, LastModified, { vec![b"Sunday, 06-Nov-94 08:49:37 GMT".to_vec()] }); | ||||
| bench_header!(asctime, LastModified, { vec![b"Sun Nov  6 08:49:37 1994".to_vec()] }); | ||||
|  | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use std::fmt::{mod, Show}; | ||||
| use super::util::from_one_raw_str; | ||||
| use header::shared::util::from_one_raw_str; | ||||
|  | ||||
| /// The `Location` header. | ||||
| /// | ||||
|   | ||||
| @@ -7,6 +7,7 @@ | ||||
| //! is used, such as `ContentType(pub Mime)`. | ||||
|  | ||||
| pub use self::accept::Accept; | ||||
| pub use self::accept_encoding::AcceptEncoding; | ||||
| pub use self::allow::Allow; | ||||
| pub use self::authorization::Authorization; | ||||
| pub use self::cache_control::CacheControl; | ||||
| @@ -76,6 +77,9 @@ macro_rules! deref( | ||||
| /// Exposes the Accept header. | ||||
| pub mod accept; | ||||
|  | ||||
| /// Exposes the AcceptEncoding header. | ||||
| pub mod accept_encoding; | ||||
|  | ||||
| /// Exposes the Allow header. | ||||
| pub mod allow; | ||||
|  | ||||
| @@ -135,5 +139,3 @@ pub mod user_agent; | ||||
|  | ||||
| /// Exposes the Vary header. | ||||
| pub mod vary; | ||||
|  | ||||
| pub mod util; | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use std::fmt::{mod, Show}; | ||||
| use super::util::from_one_raw_str; | ||||
| use header::shared::util::from_one_raw_str; | ||||
|  | ||||
| /// The `Server` header field. | ||||
| /// | ||||
| @@ -28,4 +28,3 @@ impl HeaderFormat for Server { | ||||
| } | ||||
|  | ||||
| bench_header!(bench, Server, { vec![b"Some String".to_vec()] }); | ||||
|  | ||||
|   | ||||
| @@ -103,16 +103,15 @@ fn test_fmt() { | ||||
|  | ||||
| #[test] | ||||
| fn cookie_jar() { | ||||
|     let jar = CookieJar::new("secret".as_bytes()); | ||||
|     let jar = CookieJar::new(b"secret"); | ||||
|     let cookie = Cookie::new("foo".to_string(), "bar".to_string()); | ||||
|     jar.encrypted().add(cookie); | ||||
|  | ||||
|     let cookies = SetCookie::from_cookie_jar(&jar); | ||||
|  | ||||
|     let mut new_jar = CookieJar::new("secret".as_bytes()); | ||||
|     let mut new_jar = CookieJar::new(b"secret"); | ||||
|     cookies.apply_to_cookie_jar(&mut new_jar); | ||||
|  | ||||
|     assert_eq!(jar.encrypted().find("foo"), new_jar.encrypted().find("foo")); | ||||
|     assert_eq!(jar.iter().collect::<Vec<Cookie>>(), new_jar.iter().collect::<Vec<Cookie>>()); | ||||
| } | ||||
|  | ||||
|   | ||||
| @@ -1,7 +1,7 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use std::fmt; | ||||
| use std::str::FromStr; | ||||
| use super::util::{from_comma_delimited, fmt_comma_delimited}; | ||||
| use header::shared::util::{from_comma_delimited, fmt_comma_delimited}; | ||||
|  | ||||
| use self::Encoding::{Chunked, Gzip, Deflate, Compress, EncodingExt}; | ||||
|  | ||||
|   | ||||
| @@ -1,7 +1,7 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use std::fmt::{mod, Show}; | ||||
| use std::str::FromStr; | ||||
| use super::util::{from_comma_delimited, fmt_comma_delimited}; | ||||
| use header::shared::util::{from_comma_delimited, fmt_comma_delimited}; | ||||
|  | ||||
| use self::Protocol::{WebSocket, ProtocolExt}; | ||||
|  | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| use header::{Header, HeaderFormat}; | ||||
| use std::fmt::{mod, Show}; | ||||
| use super::util::from_one_raw_str; | ||||
| use header::shared::util::from_one_raw_str; | ||||
|  | ||||
| /// The `User-Agent` header field. | ||||
| /// | ||||
|   | ||||
| @@ -1,84 +0,0 @@ | ||||
| //! Utility functions for Header implementations. | ||||
|  | ||||
| use std::str::{FromStr, from_utf8}; | ||||
| use std::fmt::{mod, Show}; | ||||
| use time::{Tm, strptime}; | ||||
|  | ||||
| /// Reads a single raw string when parsing a header | ||||
| pub fn from_one_raw_str<T: FromStr>(raw: &[Vec<u8>]) -> Option<T> { | ||||
|     if raw.len() != 1 { | ||||
|         return None; | ||||
|     } | ||||
|     // we JUST checked that raw.len() == 1, so raw[0] WILL exist. | ||||
|     match from_utf8(unsafe { raw[].get_unchecked(0)[] }) { | ||||
|         Ok(s) => FromStr::from_str(s), | ||||
|         Err(_) => None | ||||
|     } | ||||
| } | ||||
|  | ||||
| /// Reads a comma-delimited raw header into a Vec. | ||||
| #[inline] | ||||
| pub fn from_comma_delimited<T: FromStr>(raw: &[Vec<u8>]) -> Option<Vec<T>> { | ||||
|     if raw.len() != 1 { | ||||
|         return None; | ||||
|     } | ||||
|     // we JUST checked that raw.len() == 1, so raw[0] WILL exist. | ||||
|     from_one_comma_delimited(unsafe { raw.as_slice().get_unchecked(0).as_slice() }) | ||||
| } | ||||
|  | ||||
| /// Reads a comma-delimited raw string into a Vec. | ||||
| pub fn from_one_comma_delimited<T: FromStr>(raw: &[u8]) -> Option<Vec<T>> { | ||||
|     match from_utf8(raw) { | ||||
|         Ok(s) => { | ||||
|             Some(s.as_slice() | ||||
|                  .split(',') | ||||
|                  .map(|x| x.trim()) | ||||
|                  .filter_map(FromStr::from_str) | ||||
|                  .collect()) | ||||
|         } | ||||
|         Err(_) => None | ||||
|     } | ||||
| } | ||||
|  | ||||
| /// Format an array into a comma-delimited string. | ||||
| pub fn fmt_comma_delimited<T: Show>(fmt: &mut fmt::Formatter, parts: &[T]) -> fmt::Result { | ||||
|     let last = parts.len() - 1; | ||||
|     for (i, part) in parts.iter().enumerate() { | ||||
|         try!(part.fmt(fmt)); | ||||
|         if i < last { | ||||
|             try!(", ".fmt(fmt)); | ||||
|         } | ||||
|     } | ||||
|     Ok(()) | ||||
| } | ||||
|  | ||||
| /// Get a Tm from HTTP date formats. | ||||
| //    Prior to 1995, there were three different formats commonly used by | ||||
| //   servers to communicate timestamps.  For compatibility with old | ||||
| //   implementations, all three are defined here.  The preferred format is | ||||
| //   a fixed-length and single-zone subset of the date and time | ||||
| //   specification used by the Internet Message Format [RFC5322]. | ||||
| // | ||||
| //     HTTP-date    = IMF-fixdate / obs-date | ||||
| // | ||||
| //   An example of the preferred format is | ||||
| // | ||||
| //     Sun, 06 Nov 1994 08:49:37 GMT    ; IMF-fixdate | ||||
| // | ||||
| //   Examples of the two obsolete formats are | ||||
| // | ||||
| //     Sunday, 06-Nov-94 08:49:37 GMT   ; obsolete RFC 850 format | ||||
| //     Sun Nov  6 08:49:37 1994         ; ANSI C's asctime() format | ||||
| // | ||||
| //   A recipient that parses a timestamp value in an HTTP header field | ||||
| //   MUST accept all three HTTP-date formats.  When a sender generates a | ||||
| //   header field that contains one or more timestamps defined as | ||||
| //   HTTP-date, the sender MUST generate those timestamps in the | ||||
| //   IMF-fixdate format. | ||||
| pub fn tm_from_str(s: &str) -> Option<Tm> { | ||||
|     strptime(s, "%a, %d %b %Y %T %Z").or_else(|_| { | ||||
|         strptime(s, "%A, %d-%b-%y %T %Z") | ||||
|     }).or_else(|_| { | ||||
|         strptime(s, "%c") | ||||
|     }).ok() | ||||
| } | ||||
| @@ -1,6 +1,6 @@ | ||||
| use header::{Header, HeaderFormat, CaseInsensitive}; | ||||
| use std::fmt::{mod}; | ||||
| use super::util::{from_comma_delimited, fmt_comma_delimited, from_one_raw_str}; | ||||
| use header::shared::util::{from_comma_delimited, fmt_comma_delimited, from_one_raw_str}; | ||||
|  | ||||
| /// The `Allow` header. | ||||
| /// See also https://tools.ietf.org/html/rfc7231#section-7.1.4 | ||||
|   | ||||
| @@ -22,10 +22,13 @@ use http::{mod, LineEnding}; | ||||
| use {HttpResult}; | ||||
|  | ||||
| pub use self::common::*; | ||||
| pub use self::shared::*; | ||||
|  | ||||
| /// Common Headers | ||||
| pub mod common; | ||||
|  | ||||
| pub mod shared; | ||||
|  | ||||
| /// A trait for any object that will represent a header field and value. | ||||
| /// | ||||
| /// This trait represents the construction and identification of headers, | ||||
| @@ -193,7 +196,7 @@ impl Headers { | ||||
|     /// ``` | ||||
|     /// # use hyper::header::Headers; | ||||
|     /// # let mut headers = Headers::new(); | ||||
|     /// headers.set_raw("content-length", vec!["5".as_bytes().to_vec()]); | ||||
|     /// headers.set_raw("content-length", vec![b"5".to_vec()]); | ||||
|     /// ``` | ||||
|     pub fn set_raw<K: IntoCow<'static, String, str>>(&mut self, name: K, value: Vec<Vec<u8>>) { | ||||
|         self.data.insert(CaseInsensitive(name.into_cow()), MuCell::new(Item::raw(value))); | ||||
| @@ -518,6 +521,7 @@ mod tests { | ||||
|     use super::CaseInsensitive; | ||||
|     use super::{Headers, Header, HeaderFormat}; | ||||
|     use super::common::{ContentLength, ContentType, Accept, Host}; | ||||
|     use super::shared::{QualityItem}; | ||||
|  | ||||
|     use test::Bencher; | ||||
|  | ||||
| @@ -542,13 +546,13 @@ mod tests { | ||||
|  | ||||
|     #[test] | ||||
|     fn test_content_type() { | ||||
|         let content_type = Header::parse_header(["text/plain".as_bytes().to_vec()].as_slice()); | ||||
|         let content_type = Header::parse_header([b"text/plain".to_vec()].as_slice()); | ||||
|         assert_eq!(content_type, Some(ContentType(Mime(Text, Plain, vec![])))); | ||||
|     } | ||||
|  | ||||
|     #[test] | ||||
|     fn test_accept() { | ||||
|         let text_plain = Mime(Text, Plain, vec![]); | ||||
|         let text_plain = QualityItem{item: Mime(Text, Plain, vec![]), quality: 1f32}; | ||||
|         let application_vendor = "application/vnd.github.v3.full+json; q=0.5".parse().unwrap(); | ||||
|  | ||||
|         let accept = Header::parse_header([b"text/plain".to_vec()].as_slice()); | ||||
|   | ||||
							
								
								
									
										50
									
								
								src/header/shared/encoding.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										50
									
								
								src/header/shared/encoding.rs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,50 @@ | ||||
| //! Provides an Encoding enum. | ||||
|  | ||||
| use std::fmt; | ||||
| use std::str; | ||||
|  | ||||
| pub use self::Encoding::{Chunked, Gzip, Deflate, Compress, Identity, EncodingExt}; | ||||
|  | ||||
| /// A value to represent an encoding used in `Transfer-Encoding` | ||||
| /// or `Accept-Encoding` header. | ||||
| #[deriving(Clone, PartialEq)] | ||||
| pub enum Encoding { | ||||
|     /// The `chunked` encoding. | ||||
|     Chunked, | ||||
|     /// The `gzip` encoding. | ||||
|     Gzip, | ||||
|     /// The `deflate` encoding. | ||||
|     Deflate, | ||||
|     /// The `compress` encoding. | ||||
|     Compress, | ||||
|     /// The `identity` encoding. | ||||
|     Identity, | ||||
|     /// Some other encoding that is less common, can be any String. | ||||
|     EncodingExt(String) | ||||
| } | ||||
|  | ||||
| impl fmt::Show for Encoding { | ||||
|     fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result { | ||||
|         match *self { | ||||
|             Chunked => "chunked", | ||||
|             Gzip => "gzip", | ||||
|             Deflate => "deflate", | ||||
|             Compress => "compress", | ||||
|             Identity => "identity", | ||||
|             EncodingExt(ref s) => s.as_slice() | ||||
|             }.fmt(fmt) | ||||
|         } | ||||
|     } | ||||
|  | ||||
| impl str::FromStr for Encoding { | ||||
|     fn from_str(s: &str) -> Option<Encoding> { | ||||
|         match s { | ||||
|             "chunked" => Some(Chunked), | ||||
|             "deflate" => Some(Deflate), | ||||
|             "gzip" => Some(Gzip), | ||||
|             "compress" => Some(Compress), | ||||
|             "identity" => Some(Identity), | ||||
|             _ => Some(EncodingExt(s.to_string())) | ||||
|         } | ||||
|     } | ||||
| } | ||||
							
								
								
									
										26
									
								
								src/header/shared/mod.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										26
									
								
								src/header/shared/mod.rs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,26 @@ | ||||
| //! Various functions, structs and enums useful for many headers. | ||||
|  | ||||
| pub use self::encoding::Encoding; | ||||
| pub use self::encoding::Encoding::{ | ||||
|     Chunked, | ||||
|     Gzip, | ||||
|     Deflate, | ||||
|     Compress, | ||||
|     Identity, | ||||
|     EncodingExt}; | ||||
|  | ||||
| pub use self::quality_item::QualityItem; | ||||
| pub use self::quality_item::qitem; | ||||
|  | ||||
| pub use self::time::tm_from_str; | ||||
|  | ||||
| pub use self::util::{ | ||||
|     from_one_raw_str, | ||||
|     from_comma_delimited, | ||||
|     from_one_comma_delimited, | ||||
|     fmt_comma_delimited}; | ||||
|  | ||||
| pub mod encoding; | ||||
| pub mod quality_item; | ||||
| pub mod time; | ||||
| pub mod util; | ||||
							
								
								
									
										123
									
								
								src/header/shared/quality_item.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										123
									
								
								src/header/shared/quality_item.rs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,123 @@ | ||||
| //! Provides a struct for quality values. | ||||
| //! | ||||
| //! [RFC7231 Section 5.3.1](https://tools.ietf.org/html/rfc7231#section-5.3.1) | ||||
| //! gives more information on quality values in HTTP header fields. | ||||
|  | ||||
| use std::fmt; | ||||
| use std::str; | ||||
| #[cfg(test)] use super::encoding::*; | ||||
|  | ||||
| /// Represents an item with a quality value as defined in | ||||
| /// [RFC7231](https://tools.ietf.org/html/rfc7231#section-5.3.1). | ||||
| #[deriving(Clone, PartialEq)] | ||||
| pub struct QualityItem<T> { | ||||
|     /// The actual contents of the field. | ||||
|     pub item: T, | ||||
|     /// The quality (client or server preference) for the value. | ||||
|     pub quality: f32, | ||||
| } | ||||
|  | ||||
| impl<T> QualityItem<T> { | ||||
|     /// Creates a new `QualityItem` from an item and a quality. | ||||
|     /// The item can be of any type. | ||||
|     /// The quality should be a value in the range [0, 1]. | ||||
|     pub fn new(item: T, quality: f32) -> QualityItem<T> { | ||||
|         QualityItem{item: item, quality: quality} | ||||
|     } | ||||
| } | ||||
|  | ||||
| impl<T: fmt::Show> fmt::Show for QualityItem<T> { | ||||
|     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { | ||||
|         write!(f, "{}; q={}", self.item, format!("{:.3}", self.quality).trim_right_matches(['0', '.'].as_slice())) | ||||
|     } | ||||
| } | ||||
|  | ||||
| impl<T: str::FromStr> str::FromStr for QualityItem<T> { | ||||
|     fn from_str(s: &str) -> Option<Self> { | ||||
|         // Set defaults used if parsing fails. | ||||
|         let mut raw_item = s; | ||||
|         let mut quality = 1f32; | ||||
|  | ||||
|         let parts: Vec<&str> = s.rsplitn(1, ';').map(|x| x.trim()).collect(); | ||||
|         if parts.len() == 2 { | ||||
|             let start = parts[0].slice(0, 2); | ||||
|             if start == "q=" || start == "Q=" { | ||||
|                 let q_part = parts[0].slice(2, parts[0].len()); | ||||
|                 if q_part.len() > 5 { | ||||
|                     return None; | ||||
|                 } | ||||
|                 let x: Option<f32> = q_part.parse(); | ||||
|                 match x { | ||||
|                     Some(q_value) => { | ||||
|                         if 0f32 <= q_value && q_value <= 1f32 { | ||||
|                             quality = q_value; | ||||
|                             raw_item = parts[1]; | ||||
|                             } else { | ||||
|                                 return None; | ||||
|                             } | ||||
|                         }, | ||||
|                     None => return None, | ||||
|                 } | ||||
|             } | ||||
|         } | ||||
|         let x: Option<T> = raw_item.parse(); | ||||
|         match x { | ||||
|             Some(item) => { | ||||
|                 Some(QualityItem{ item: item, quality: quality, }) | ||||
|             }, | ||||
|             None => return None, | ||||
|         } | ||||
|     } | ||||
| } | ||||
|  | ||||
| /// Convinience function to wrap a value in a `QualityItem` | ||||
| /// Sets `q` to the default 1.0 | ||||
| pub fn qitem<T>(item: T) -> QualityItem<T> { | ||||
|     QualityItem::new(item, 1.0) | ||||
| } | ||||
|  | ||||
| #[test] | ||||
| fn test_quality_item_show1() { | ||||
|     let x = qitem(Chunked); | ||||
|     assert_eq!(format!("{}", x), "chunked; q=1.000"); | ||||
| } | ||||
| #[test] | ||||
| fn test_quality_item_show2() { | ||||
|     let x = QualityItem::new(Chunked, 0.001); | ||||
|     assert_eq!(format!("{}", x), "chunked; q=0.001"); | ||||
| } | ||||
| #[test] | ||||
| fn test_quality_item_show3() { | ||||
|     // Custom value | ||||
|     let x = QualityItem{ | ||||
|         item: EncodingExt("identity".to_string()), | ||||
|         quality: 0.5f32, | ||||
|     }; | ||||
|     assert_eq!(format!("{}", x), "identity; q=0.500"); | ||||
| } | ||||
|  | ||||
| #[test] | ||||
| fn test_quality_item_from_str1() { | ||||
|     let x: Option<QualityItem<Encoding>> = "chunked".parse(); | ||||
|     assert_eq!(x.unwrap(), QualityItem{ item: Chunked, quality: 1f32, }); | ||||
| } | ||||
| #[test] | ||||
| fn test_quality_item_from_str2() { | ||||
|     let x: Option<QualityItem<Encoding>> = "chunked; q=1".parse(); | ||||
|     assert_eq!(x.unwrap(), QualityItem{ item: Chunked, quality: 1f32, }); | ||||
| } | ||||
| #[test] | ||||
| fn test_quality_item_from_str3() { | ||||
|     let x: Option<QualityItem<Encoding>> = "gzip; q=0.5".parse(); | ||||
|     assert_eq!(x.unwrap(), QualityItem{ item: Gzip, quality: 0.5f32, }); | ||||
| } | ||||
| #[test] | ||||
| fn test_quality_item_from_str4() { | ||||
|     let x: Option<QualityItem<Encoding>> = "gzip; q=0.273".parse(); | ||||
|     assert_eq!(x.unwrap(), QualityItem{ item: Gzip, quality: 0.273f32, }); | ||||
| } | ||||
| #[test] | ||||
| fn test_quality_item_from_str5() { | ||||
|     let x: Option<QualityItem<Encoding>> = "gzip; q=0.2739999".parse(); | ||||
|     assert_eq!(x, None); | ||||
| } | ||||
							
								
								
									
										34
									
								
								src/header/shared/time.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								src/header/shared/time.rs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,34 @@ | ||||
| //! Provides utility function to parse HTTP header value time. | ||||
|  | ||||
| extern crate time; | ||||
|  | ||||
| /// Get a Tm from HTTP date formats. | ||||
| //    Prior to 1995, there were three different formats commonly used by | ||||
| //   servers to communicate timestamps.  For compatibility with old | ||||
| //   implementations, all three are defined here.  The preferred format is | ||||
| //   a fixed-length and single-zone subset of the date and time | ||||
| //   specification used by the Internet Message Format [RFC5322]. | ||||
| // | ||||
| //     HTTP-date    = IMF-fixdate / obs-date | ||||
| // | ||||
| //   An example of the preferred format is | ||||
| // | ||||
| //     Sun, 06 Nov 1994 08:49:37 GMT    ; IMF-fixdate | ||||
| // | ||||
| //   Examples of the two obsolete formats are | ||||
| // | ||||
| //     Sunday, 06-Nov-94 08:49:37 GMT   ; obsolete RFC 850 format | ||||
| //     Sun Nov  6 08:49:37 1994         ; ANSI C's asctime() format | ||||
| // | ||||
| //   A recipient that parses a timestamp value in an HTTP header field | ||||
| //   MUST accept all three HTTP-date formats.  When a sender generates a | ||||
| //   header field that contains one or more timestamps defined as | ||||
| //   HTTP-date, the sender MUST generate those timestamps in the | ||||
| //   IMF-fixdate format. | ||||
| pub fn tm_from_str(s: &str) -> Option<time::Tm> { | ||||
|     time::strptime(s, "%a, %d %b %Y %T %Z").or_else(|_| { | ||||
|         time::strptime(s, "%A, %d-%b-%y %T %Z") | ||||
|     }).or_else(|_| { | ||||
|         time::strptime(s, "%c") | ||||
|     }).ok() | ||||
| } | ||||
							
								
								
									
										52
									
								
								src/header/shared/util.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										52
									
								
								src/header/shared/util.rs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,52 @@ | ||||
| //! Utility functions for Header implementations. | ||||
|  | ||||
| use std::str; | ||||
| use std::fmt; | ||||
|  | ||||
| /// Reads a single raw string when parsing a header | ||||
| pub fn from_one_raw_str<T: str::FromStr>(raw: &[Vec<u8>]) -> Option<T> { | ||||
|     if raw.len() != 1 { | ||||
|         return None; | ||||
|     } | ||||
|     // we JUST checked that raw.len() == 1, so raw[0] WILL exist. | ||||
|     match str::from_utf8(unsafe { raw[].unsafe_get(0)[] }) { | ||||
|         Ok(s) => str::FromStr::from_str(s), | ||||
|         Err(_) => None | ||||
|     } | ||||
| } | ||||
|  | ||||
| /// Reads a comma-delimited raw header into a Vec. | ||||
| #[inline] | ||||
| pub fn from_comma_delimited<T: str::FromStr>(raw: &[Vec<u8>]) -> Option<Vec<T>> { | ||||
|     if raw.len() != 1 { | ||||
|         return None; | ||||
|     } | ||||
|     // we JUST checked that raw.len() == 1, so raw[0] WILL exist. | ||||
|     from_one_comma_delimited(unsafe { raw.as_slice().unsafe_get(0).as_slice() }) | ||||
| } | ||||
|  | ||||
| /// Reads a comma-delimited raw string into a Vec. | ||||
| pub fn from_one_comma_delimited<T: str::FromStr>(raw: &[u8]) -> Option<Vec<T>> { | ||||
|     match str::from_utf8(raw) { | ||||
|         Ok(s) => { | ||||
|             Some(s.as_slice() | ||||
|                  .split(',') | ||||
|                  .map(|x| x.trim()) | ||||
|                  .filter_map(str::FromStr::from_str) | ||||
|                  .collect()) | ||||
|         } | ||||
|         Err(_) => None | ||||
|     } | ||||
| } | ||||
|  | ||||
| /// Format an array into a comma-delimited string. | ||||
| pub fn fmt_comma_delimited<T: fmt::Show>(fmt: &mut fmt::Formatter, parts: &[T]) -> fmt::Result { | ||||
|     let last = parts.len() - 1; | ||||
|     for (i, part) in parts.iter().enumerate() { | ||||
|         try!(write!(fmt, "{}", part)); | ||||
|         if i < last { | ||||
|             try!(write!(fmt, ", ")); | ||||
|         } | ||||
|     } | ||||
|     Ok(()) | ||||
| } | ||||
| @@ -789,7 +789,7 @@ mod tests { | ||||
|         } | ||||
|  | ||||
|         read("Host: rust-lang.org\r\n", Ok(Some(("Host".to_string(), | ||||
|                                                 "rust-lang.org".as_bytes().to_vec())))); | ||||
|                                                  b"rust-lang.org".to_vec())))); | ||||
|     } | ||||
|  | ||||
|     #[test] | ||||
|   | ||||
		Reference in New Issue
	
	Block a user