use header::{Header, HeaderFormat};
use std::fmt::{mod};
use super::util::from_one_raw_str;
#[deriving(Clone, PartialEq, Show)]
pub struct Etag {
pub weak: bool,
pub tag: String
}
impl Header for Etag {
fn header_name(_: Option<Etag>) -> &'static str {
"Etag"
}
fn parse_header(raw: &[Vec<u8>]) -> Option<Etag> {
fn check_slice_validity(slice: &str) -> bool {
for c in slice.bytes() {
match c {
b'\x21' | b'\x23' ... b'\x7e' | b'\x80' ... b'\xff' => (),
_ => { return false; }
}
}
true
}
from_one_raw_str(raw).and_then(|s: String| {
let length: uint = s.len();
let slice = s[];
if slice.is_empty() || !slice.ends_with("\"") {
return None;
}
if slice.char_at(0) == '"' {
if check_slice_validity(slice.slice_chars(1, length-1)) {
return Some(Etag {
weak: false,
tag: slice.slice_chars(1, length-1).to_string()
});
} else {
return None;
}
}
if slice.slice_chars(0, 3) == "W/\"" {
if check_slice_validity(slice.slice_chars(3, length-1)) {
return Some(Etag {
weak: true,
tag: slice.slice_chars(3, length-1).to_string()
});
} else {
return None;
}
}
None
})
}
}
impl HeaderFormat for Etag {
fn fmt_header(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
if self.weak {
try!(fmt.write(b"W/"));
}
write!(fmt, "\"{}\"", self.tag)
}
}
#[cfg(test)]
mod tests {
use super::Etag;
use header::Header;
#[test]
fn test_etag_successes() {
let mut etag: Option<Etag>;
etag = Header::parse_header([b"\"foobar\"".to_vec()].as_slice());
assert_eq!(etag, Some(Etag {
weak: false,
tag: "foobar".to_string()
}));
etag = Header::parse_header([b"\"\"".to_vec()].as_slice());
assert_eq!(etag, Some(Etag {
weak: false,
tag: "".to_string()
}));
etag = Header::parse_header([b"W/\"weak-etag\"".to_vec()].as_slice());
assert_eq!(etag, Some(Etag {
weak: true,
tag: "weak-etag".to_string()
}));
etag = Header::parse_header([b"W/\"\x65\x62\"".to_vec()].as_slice());
assert_eq!(etag, Some(Etag {
weak: true,
tag: "\u{0065}\u{0062}".to_string()
}));
etag = Header::parse_header([b"W/\"\"".to_vec()].as_slice());
assert_eq!(etag, Some(Etag {
weak: true,
tag: "".to_string()
}));
}
#[test]
fn test_etag_failures() {
let mut etag: Option<Etag>;
etag = Header::parse_header([b"no-dquotes".to_vec()].as_slice());
assert_eq!(etag, None);
etag = Header::parse_header([b"w/\"the-first-w-is-case-sensitive\"".to_vec()].as_slice());
assert_eq!(etag, None);
etag = Header::parse_header([b"".to_vec()].as_slice());
assert_eq!(etag, None);
etag = Header::parse_header([b"\"unmatched-dquotes1".to_vec()].as_slice());
assert_eq!(etag, None);
etag = Header::parse_header([b"unmatched-dquotes2\"".to_vec()].as_slice());
assert_eq!(etag, None);
etag = Header::parse_header([b"matched-\"dquotes\"".to_vec()].as_slice());
assert_eq!(etag, None);
}
}
bench_header!(bench, Etag, { vec![b"W/\"nonemptytag\"".to_vec()] });