ast: Build images into the partial AST

Signed-off-by: Daniel Silverstone <dsilvers@digital-scurf.org>

1 files changed, 10 insertions, 1 deletions

diff --git a/src/ast.rs b/src/ast.rs
index 7584acc..3840e0e 100644
--- a/src/ast.rs
+++ b/src/ast.rs
@@ -148,7 +148,7 @@ fn parse_blocks(markdown: &str) -> Vec<Block> {
                 Tag::Paragraph => blocks.push(paragraph(&mut inlines)),
                 Tag::Heading(level) => blocks.push(heading(level as i64, &mut inlines)),
                 Tag::CodeBlock(kind) => blocks.push(code_block(&kind, &mut inlines)),
-
+                Tag::Image(_link, dest, title) => blocks.push(image_block(&dest, &title)),
                 // We don't handle anything else yet.
                 _ => (),
             },
@@ -180,6 +180,15 @@ fn heading(level: i64, inlines: &mut Vec<Inline>) -> Block {
     Block::Header(level, attr, std::mem::take(inlines))
 }
 
+fn image_block(dest: &str, title: &str) -> Block {
+    let attr = ("".to_string(), vec![], vec![]);
+    Block::Para(vec![Inline::Image(
+        attr,
+        vec![],
+        (dest.to_string(), title.to_string()),
+    )])
+}
+
 fn code_block(kind: &CodeBlockKind, inlines: &mut Vec<Inline>) -> Block {
     event!(Level::TRACE, ?kind, "code block");
     let attr = if let CodeBlockKind::Fenced(lang) = kind {