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);
+                }
+            }
+        }
+    }
+}