Load bare images within an HTML document (#3108)

servo · Aug 26, 2014 · 22ed3d6 · 22ed3d6
1 parent 2ce5b46
commit 22ed3d6
Showing 1 changed file with 17 additions and 12 deletions.
diff --git a/src/components/script/html/hubbub_html_parser.rs b/src/components/script/html/hubbub_html_parser.rs
@@ -526,19 +526,24 @@ pub fn parse_html(page: &Page,
     };
     parser.set_tree_handler(&mut tree_handler);
     debug!("set tree handler");
-
     debug!("loaded page");
-    loop {
-        match load_response.progress_port.recv() {
-            Payload(data) => {
-                debug!("received data");
-                parser.parse_chunk(data.as_slice());
-            }
-            Done(Err(err)) => {
-                fail!("Failed to load page URL {:s}, error: {:s}", url.serialize(), err);
-            }
-            Done(..) => {
-                break;
+    match load_response.metadata.content_type {
+        Some((ref t, _)) if t.as_slice().eq_ignore_ascii_case("image") => {
+            let page = format!("<html><body><img src='{:s}' /></body></html>", base_url.serialize());
+            parser.parse_chunk(page.into_bytes().as_slice());
+        },
+        _ => loop {
+            match load_response.progress_port.recv() {
+                Payload(data) => {
+                    debug!("received data");
+                    parser.parse_chunk(data.as_slice());
+                }
+                Done(Err(err)) => {
+                    fail!("Failed to load page URL {:s}, error: {:s}", url.serialize(), err);
+                }
+                Done(..) => {
+                    break;
+                }
             }
         }
     }