diff options
Diffstat (limited to 'src/visitor/embedded.rs')
-rw-r--r-- | src/visitor/embedded.rs | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/src/visitor/embedded.rs b/src/visitor/embedded.rs new file mode 100644 index 0000000..891240b --- /dev/null +++ b/src/visitor/embedded.rs @@ -0,0 +1,35 @@ +use crate::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); + } + } + } + } +} |