82 lines
2.9 KiB
Rust
82 lines
2.9 KiB
Rust
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 <head> section of a
|
|
/// rendered Markdown file or generated documentation
|
|
pub in_header: String,
|
|
/// Content that will be included inline between <body> 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 </body> 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<Playground>)
|
|
-> Option<ExternalHtml> {
|
|
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<P: AsRef<Path>>(file_path: P, diag: &errors::Handler)
|
|
-> Result<String, LoadStringError>
|
|
{
|
|
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<String> {
|
|
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)
|
|
}
|