diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/ast.rs | 70 | ||||
-rw-r--r-- | src/bin/sp-meta.rs | 11 | ||||
-rw-r--r-- | src/codegen.rs | 10 |
3 files changed, 47 insertions, 44 deletions
@@ -143,11 +143,11 @@ impl<'a> Document { pub fn sources(&mut self) -> Vec<PathBuf> { let mut names = vec![]; - if let Some(x) = self.meta().bindings_filename() { + for x in self.meta().bindings_filenames() { names.push(PathBuf::from(x)) } - if let Some(x) = self.meta().functions_filename() { + for x in self.meta().functions_filenames() { names.push(PathBuf::from(x)) } @@ -426,9 +426,9 @@ mod test_extract { pub struct Metadata { title: String, date: Option<String>, - bindings_filename: Option<PathBuf>, + bindings_filenames: Vec<PathBuf>, bindings: Bindings, - functions_filename: Option<PathBuf>, + functions_filenames: Vec<PathBuf>, template: Option<String>, bibliographies: Vec<PathBuf>, /// Extra class names which should be considered 'correct' for this document @@ -443,21 +443,21 @@ impl Metadata { { let title = get_title(&doc.meta)?; let date = get_date(&doc.meta); - let bindings_filename = get_bindings_filename(basedir.as_ref(), &doc.meta)?; - let functions_filename = get_functions_filename(basedir.as_ref(), &doc.meta)?; + let bindings_filenames = get_bindings_filenames(basedir.as_ref(), &doc.meta); + let functions_filenames = get_functions_filenames(basedir.as_ref(), &doc.meta); let template = get_template_name(&doc.meta)?; let mut bindings = Bindings::new(); - if let Some(ref filename) = bindings_filename { - get_bindings(filename, &mut bindings)?; - } + let bibliographies = get_bibliographies(basedir.as_ref(), &doc.meta); let classes = get_classes(&doc.meta); + + get_bindings(&bindings_filenames, &mut bindings)?; Ok(Metadata { title, date, - bindings_filename, + bindings_filenames, bindings, - functions_filename, + functions_filenames, template, bibliographies, classes, @@ -479,19 +479,16 @@ impl Metadata { } /// Return filename where bindings are specified. - pub fn bindings_filename(&self) -> Option<&Path> { - match &self.bindings_filename { - Some(filename) => Some(&filename), - None => None, - } + pub fn bindings_filenames(&self) -> Vec<&Path> { + self.bindings_filenames.iter().map(|f| f.as_ref()).collect() } /// Return filename where functions are specified. - pub fn functions_filename(&self) -> Option<&Path> { - match &self.functions_filename { - Some(filename) => Some(&filename), - None => None, - } + pub fn functions_filenames(&self) -> Vec<&Path> { + self.functions_filenames + .iter() + .map(|f| f.as_ref()) + .collect() } /// Return the name of the code template, if specified. @@ -536,24 +533,18 @@ fn get_date(map: &Mapp) -> Option<String> { } } -fn get_bindings_filename<P>(basedir: P, map: &Mapp) -> Result<Option<PathBuf>> +fn get_bindings_filenames<P>(basedir: P, map: &Mapp) -> Vec<PathBuf> where P: AsRef<Path>, { - match get_path(map, "bindings") { - None => Ok(None), - Some(path) => Ok(Some(basedir.as_ref().join(path))), - } + get_paths(basedir, map, "bindings") } -fn get_functions_filename<P>(basedir: P, map: &Mapp) -> Result<Option<PathBuf>> +fn get_functions_filenames<P>(basedir: P, map: &Mapp) -> Vec<PathBuf> where P: AsRef<Path>, { - match get_path(map, "functions") { - None => Ok(None), - Some(path) => Ok(Some(basedir.as_ref().join(path))), - } + get_paths(basedir, map, "functions") } fn get_template_name(map: &Mapp) -> Result<Option<String>> { @@ -563,10 +554,13 @@ fn get_template_name(map: &Mapp) -> Result<Option<String>> { } } -fn get_path(map: &Mapp, field: &str) -> Option<PathBuf> { - match get_string(map, field) { - None => None, - Some(s) => Some(Path::new(&s).to_path_buf()), +fn get_paths<P>(basedir: P, map: &Mapp, field: &str) -> Vec<PathBuf> +where + P: AsRef<Path>, +{ + match map.get(field) { + None => vec![], + Some(v) => pathbufs(basedir, v), } } @@ -687,11 +681,13 @@ mod test_join { } } -fn get_bindings<P>(filename: P, bindings: &mut Bindings) -> Result<()> +fn get_bindings<P>(filenames: &[P], bindings: &mut Bindings) -> Result<()> where P: AsRef<Path>, { - bindings.add_from_file(filename)?; + for filename in filenames { + bindings.add_from_file(filename)?; + } Ok(()) } diff --git a/src/bin/sp-meta.rs b/src/bin/sp-meta.rs index 979f18a..cfa3b3d 100644 --- a/src/bin/sp-meta.rs +++ b/src/bin/sp-meta.rs @@ -22,8 +22,15 @@ fn main() -> Result<()> { } println!("title: {}", doc.meta().title()); - println!("bindings: {}", filename(doc.meta().bindings_filename())); - println!("functions: {}", filename(doc.meta().functions_filename())); + + for filename in doc.meta().bindings_filenames() { + println!("bindings: {}", filename.display()); + } + + for filename in doc.meta().functions_filenames() { + println!("functions: {}", filename.display()); + } + for bib in doc.meta().bibliographies().iter() { println!("bibliography: {}", filename(Some(bib))); } diff --git a/src/codegen.rs b/src/codegen.rs index 70d1966..db31765 100644 --- a/src/codegen.rs +++ b/src/codegen.rs @@ -41,12 +41,12 @@ fn context(doc: &mut Document) -> Result<Context> { context.insert("scenarios", &doc.matched_scenarios()?); context.insert("files", doc.files()); - let (funcs_filename, funcs) = match doc.meta().functions_filename() { - Some(filename) => (filename, cat(filename)?), - None => (Path::new(""), "".to_string()), - }; + let funcs_filenames = doc.meta().functions_filenames(); + let mut funcs = String::new(); + for filename in funcs_filenames { + funcs.push_str(&cat(filename)?); + } context.insert("functions", &funcs); - context.insert("functions_filename", funcs_filename); Ok(context) } |