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

improve Markdown API

See merge request subplot/subplot!307

1 files changed, 35 insertions, 0 deletions

diff --git a/src/md/visitor/embedded.rs b/src/md/visitor/embedded.rs
new file mode 100644
index 0000000..840d9ed
--- /dev/null
+++ b/src/md/visitor/embedded.rs
@@ -0,0 +1,35 @@
+use crate::md::panhelper;
+use crate::EmbeddedFile;
+use crate::EmbeddedFiles;
+
+use pandoc_ast::{Block, MutVisitor};
+
+impl MutVisitor for EmbeddedFiles {
+    fn visit_vec_block(&mut self, vec_block: &mut Vec<Block>) {
+        use panhelper::is_class;
+        for block in vec_block {
+            match block {
+                Block::CodeBlock(attr, contents) => {
+                    if is_class(attr, "file") {
+                        let add_newline = match panhelper::find_attr_kv(attr, "add-newline").next()
+                        {
+                            None | Some("auto") => !contents.ends_with('\n'),
+                            Some("yes") => true,
+                            Some("no") => false,
+                            _ => unreachable!(),
+                        };
+                        let contents = if add_newline {
+                            format!("{}\n", contents)
+                        } else {
+                            contents.clone()
+                        };
+                        self.push(EmbeddedFile::new(panhelper::get_filename(attr), contents));
+                    }
+                }
+                _ => {
+                    self.visit_block(block);
+                }
+            }
+        }
+    }
+}