diff options
Diffstat (limited to 'src/md/visitor/image.rs')
-rw-r--r-- | src/md/visitor/image.rs | 25 |
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)); + } + } +} |