Consider title tags with attributes

1 files changed, 14 insertions, 4 deletions

diff --git a/src/plugins/url.rs b/src/plugins/url.rs
index e75d893..bff840f 100644
--- a/src/plugins/url.rs
+++ b/src/plugins/url.rs
@@ -37,10 +37,20 @@ impl Url {
     }
 
     fn get_title<'a>(&self, body: &str) -> Result<String, UrlError> {
-        let title = body.find("<title>")
-            .map(|start| body.find("</title>").map(|end| &body[start + 7..end]))
-            .and_then(|s| s).ok_or(ErrorKind::MissingTitle)?;
-
+        let title = body.find("<title")
+            .map(|tag| {
+                body[tag..]
+                    .find('>')
+                    .map(|offset| tag + offset + 1)
+                    .map(|start| {
+                        body[start..]
+                            .find("</title>")
+                            .map(|offset| start + offset)
+                            .map(|end| &body[start..end])
+                    })
+            })
+            .and_then(|s| s.and_then(|s| s))
+            .ok_or(ErrorKind::MissingTitle)?;
 
         debug!("Title: {:?}", title);