summaryrefslogtreecommitdiff
path: root/src/md/visitor/image.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/md/visitor/image.rs')
-rw-r--r--src/md/visitor/image.rs25
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));
+ }
+ }
+}