use std::fs; use std::path::Path; use std::str; use errors; use rustc_feature::UnstableFeatures; use crate::syntax::edition::Edition; use crate::html::markdown::{IdMap, ErrorCodes, Markdown, Playground}; #[derive(Clone, Debug)] pub struct ExternalHtml { /// Content that will be included inline in the section of a /// rendered Markdown file or generated documentation pub in_header: String, /// Content that will be included inline between and the content of /// a rendered Markdown file or generated documentation pub before_content: String, /// Content that will be included inline between the content and of /// a rendered Markdown file or generated documentation pub after_content: String } impl ExternalHtml { pub fn load(in_header: &[String], before_content: &[String], after_content: &[String], md_before_content: &[String], md_after_content: &[String], diag: &errors::Handler, id_map: &mut IdMap, edition: Edition, playground: &Option) -> Option { let codes = ErrorCodes::from(UnstableFeatures::from_environment().is_nightly_build()); let ih = load_external_files(in_header, diag)?; let bc = load_external_files(before_content, diag)?; let m_bc = load_external_files(md_before_content, diag)?; let bc = format!("{}{}", bc, Markdown(&m_bc, &[], id_map, codes, edition, playground).to_string()); let ac = load_external_files(after_content, diag)?; let m_ac = load_external_files(md_after_content, diag)?; let ac = format!("{}{}", ac, Markdown(&m_ac, &[], id_map, codes, edition, playground).to_string()); Some(ExternalHtml { in_header: ih, before_content: bc, after_content: ac, }) } } pub enum LoadStringError { ReadFail, BadUtf8, } pub fn load_string>(file_path: P, diag: &errors::Handler) -> Result { let file_path = file_path.as_ref(); let contents = match fs::read(file_path) { Ok(bytes) => bytes, Err(e) => { diag.struct_err(&format!("error reading `{}`: {}", file_path.display(), e)).emit(); return Err(LoadStringError::ReadFail); } }; match str::from_utf8(&contents) { Ok(s) => Ok(s.to_string()), Err(_) => { diag.struct_err(&format!("error reading `{}`: not UTF-8", file_path.display())).emit(); Err(LoadStringError::BadUtf8) } } } fn load_external_files(names: &[String], diag: &errors::Handler) -> Option { let mut out = String::new(); for name in names { let s = match load_string(name, diag) { Ok(s) => s, Err(_) => return None, }; out.push_str(&s); out.push('\n'); } Some(out) }