summaryrefslogtreecommitdiff
path: root/src/visitor/embedded.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/visitor/embedded.rs')
-rw-r--r--src/visitor/embedded.rs35
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);
+ }
+ }
+ }
+ }
+}