Merge branch 'liw/refactor-md' into 'main'

improve Markdown API

See merge request subplot/subplot!307

1 files changed, 25 insertions, 0 deletions

diff --git a/src/md/visitor/image.rs b/src/md/visitor/image.rs
new file mode 100644
index 0000000..be49d66
--- /dev/null
+++ b/src/md/visitor/image.rs
@@ -0,0 +1,25 @@
+use std::path::PathBuf;
+
+use pandoc_ast::{Inline, MutVisitor};
+
+pub struct ImageVisitor {
+    images: Vec<PathBuf>,
+}
+
+impl ImageVisitor {
+    pub fn new() -> Self {
+        ImageVisitor { images: vec![] }
+    }
+
+    pub fn images(&self) -> Vec<PathBuf> {
+        self.images.clone()
+    }
+}
+
+impl MutVisitor for ImageVisitor {
+    fn visit_inline(&mut self, inline: &mut Inline) {
+        if let Inline::Image(_attr, _inlines, target) = inline {
+            self.images.push(PathBuf::from(&target.0));
+        }
+    }
+}