Support object element data.

1 files changed, 38 insertions, 2 deletions

diff --git a/src/components/script/dom/htmlobjectelement.rs b/src/components/script/dom/htmlobjectelement.rs
index aeb6ad754e9..74f40db40dd 100644
--- a/src/components/script/dom/htmlobjectelement.rs
+++ b/src/components/script/dom/htmlobjectelement.rs
@@ -12,14 +12,21 @@ use dom::validitystate::ValidityState;
 use dom::windowproxy::WindowProxy;
 use servo_util::str::DOMString;
 
+use extra::url::Url;
+use servo_net::image_cache_task;
+use servo_net::image_cache_task::ImageCacheTask;
+use servo_util::url::parse_url;
+use servo_util::namespace::Null;
+use servo_util::url::is_image_data;
+
 pub struct HTMLObjectElement {
-    htmlelement: HTMLElement
+    htmlelement: HTMLElement,
 }
 
 impl HTMLObjectElement {
     pub fn new_inherited(localName: DOMString, document: AbstractDocument) -> HTMLObjectElement {
         HTMLObjectElement {
-            htmlelement: HTMLElement::new_inherited(HTMLObjectElementTypeId, localName, document)
+            htmlelement: HTMLElement::new_inherited(HTMLObjectElementTypeId, localName, document),
         }
     }
 
@@ -30,6 +37,35 @@ impl HTMLObjectElement {
 }
 
 impl HTMLObjectElement {
+
+    // Makes the local `data` member match the status of the `data` attribute and starts
+    /// prefetching the image. This method must be called after `data` is changed.
+    pub fn process_data_url(&mut self, image_cache: ImageCacheTask, url: Option<Url>) {
+        let elem = &mut self.htmlelement.element;
+
+        // TODO: support other values
+        match (elem.get_attribute(Null, "type").map(|x| x.Value()),
+               elem.get_attribute(Null, "data").map(|x| x.Value())) {
+            (None, Some(uri)) => {
+                if is_image_data(uri) {
+                    let data_url = parse_url(uri, url);
+                    // Issue #84
+                    image_cache.send(image_cache_task::Prefetch(data_url));
+                }
+            }
+            _ => { }
+        }
+    }
+
+    pub fn AfterSetAttr(&mut self, name: DOMString, _value: DOMString) {
+        if "data" == name {
+            let document = self.htmlelement.element.node.owner_doc();
+            let window = document.document().window;
+            let url = window.page.url.as_ref().map(|&(ref url, _)| url.clone());
+            self.process_data_url(window.image_cache_task.clone(), url);
+        }
+    }
+
     pub fn Data(&self) -> DOMString {
         ~""
     }