diff options
author | Lars Wirzenius <liw@liw.fi> | 2022-09-03 09:25:26 +0300 |
---|---|---|
committer | Lars Wirzenius <liw@liw.fi> | 2022-09-03 09:30:16 +0300 |
commit | 2341869c78cea3647eadd46bbfc7f703541cee0b (patch) | |
tree | 2d870c6eec99a782bbc421bd678003b3ae323515 /src/visitor/embedded.rs | |
parent | ba31a2ad25fa0ae3f3fd49eaa5174f9520248b90 (diff) | |
download | subplot-2341869c78cea3647eadd46bbfc7f703541cee0b.tar.gz |
refactor: rename DataFile{,s} to EmbeddedFile{,s} to be systematic
Sponsored-by: author
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); + } + } + } + } +} |