support regex attrs

Cargo.toml

@@ -15,6 +15,7 @@ speculate = "0.1.2"
 html5ever = "0.25"
 bit-set = "0.5"
 markup5ever_rcdom = "0.1"
+regex = "1.3.9"
 
 [badges]
 travis-ci = { repository = "utkarshkukreti/select.rs" }

src/lib.rs

@@ -1,6 +1,7 @@
 extern crate bit_set;
 extern crate html5ever;
 extern crate markup5ever_rcdom;
+extern crate regex;
 
 pub mod document;
 pub mod node;

src/predicate.rs

@@ -1,4 +1,5 @@
 use node::{self, Node};
+use regex::Regex;
 
 /// A trait implemented by all `Node` matchers.
 pub trait Predicate {
@@ -94,6 +95,16 @@ impl<'a> Predicate for Attr<&'a str, ()> {
     }
 }
 
+impl<'a> Predicate for Attr<&'a str, Regex> {
+    fn matches(&self, node: &Node) -> bool {
+        if let Some(attr) = node.attr(self.0) {
+            return self.1.is_match(attr)
+        }
+        false
+    }
+}
+
+
 /// Matches if the function returns true.
 impl<F: Fn(&Node) -> bool> Predicate for F {
     fn matches(&self, node: &Node) -> bool {

tests/predicate_tests.rs

@@ -68,6 +68,9 @@ speculate! {
             assert_eq!(Attr("id", "post-0").matches(&article), true);
             assert_eq!(Attr("id", ()).matches(&html), false);
             assert_eq!(Attr("id", ()).matches(&article), true);
+            assert_eq!(Attr("id", regex::Regex::new("post").unwrap()).matches(&article), true);
+            assert_eq!(Attr("class", regex::Regex::new("^category$").unwrap()).matches(&article), false);
+            assert_eq!(Attr("class", regex::Regex::new("category").unwrap()).matches(&article), true);
         }
 
         test "Fn(&Node) -> bool" {