diff options
author | Daniel Silverstone <dsilvers+gitlab@digital-scurf.org> | 2023-02-01 23:49:11 +0000 |
---|---|---|
committer | Daniel Silverstone <dsilvers+gitlab@digital-scurf.org> | 2023-02-01 23:49:11 +0000 |
commit | b6df0fa5b7046bad97f63dc928a9776cecaee623 (patch) | |
tree | 7f2fe57569bdf9d77d0bc30e3fac78ba2d32ff92 /src/md/visitor/block_class.rs | |
parent | e1d63019fdd6f0d24703197b6a6029f26e6f089f (diff) | |
parent | b160ad5b9f0e38859b0a6d3c3262afbb39067584 (diff) | |
download | subplot-b6df0fa5b7046bad97f63dc928a9776cecaee623.tar.gz |
Merge branch 'liw/refactor-md' into 'main'
improve Markdown API
See merge request subplot/subplot!307
Diffstat (limited to 'src/md/visitor/block_class.rs')
-rw-r--r-- | src/md/visitor/block_class.rs | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/src/md/visitor/block_class.rs b/src/md/visitor/block_class.rs new file mode 100644 index 0000000..303616b --- /dev/null +++ b/src/md/visitor/block_class.rs @@ -0,0 +1,25 @@ +use std::collections::HashSet; + +use pandoc_ast::{Block, MutVisitor}; + +#[derive(Default)] +pub struct BlockClassVisitor { + pub classes: HashSet<String>, +} + +impl MutVisitor for BlockClassVisitor { + fn visit_vec_block(&mut self, vec_block: &mut Vec<Block>) { + for block in vec_block { + match block { + Block::CodeBlock(attr, _) => { + for class in &attr.1 { + self.classes.insert(class.to_string()); + } + } + _ => { + self.visit_block(block); + } + } + } + } +} |