use std::path::Path; use std::fs::File; use std::io::Read; pub static INDEX: &'static [u8] = include_bytes!("index.hbs"); pub static CSS: &'static [u8] = include_bytes!("book.css"); pub static FAVICON: &'static [u8] = include_bytes!("favicon.png"); pub static JS: &'static [u8] = include_bytes!("book.js"); pub static HIGHLIGHT_JS: &'static [u8] = include_bytes!("highlight.js"); pub static TOMORROW_NIGHT_CSS: &'static [u8] = include_bytes!("tomorrow-night.css"); pub static HIGHLIGHT_CSS: &'static [u8] = include_bytes!("highlight.css"); pub static JQUERY: &'static [u8] = include_bytes!("jquery-2.1.4.min.js"); pub static FONT_AWESOME: &'static [u8] = include_bytes!("_FontAwesome/css/font-awesome.min.css"); pub static FONT_AWESOME_EOT: &'static [u8] = include_bytes!("_FontAwesome/fonts/fontawesome-webfont.eot"); pub static FONT_AWESOME_SVG: &'static [u8] = include_bytes!("_FontAwesome/fonts/fontawesome-webfont.svg"); pub static FONT_AWESOME_TTF: &'static [u8] = include_bytes!("_FontAwesome/fonts/fontawesome-webfont.ttf"); pub static FONT_AWESOME_WOFF: &'static [u8] = include_bytes!("_FontAwesome/fonts/fontawesome-webfont.woff"); pub static FONT_AWESOME_WOFF2: &'static [u8] = include_bytes!("_FontAwesome/fonts/fontawesome-webfont.woff2"); pub static FONT_AWESOME_OTF: &'static [u8] = include_bytes!("_FontAwesome/fonts/FontAwesome.otf"); /// The `Theme` struct should be used instead of the static variables because the `new()` method /// will look if the user has a theme directory in his source folder and use the users theme instead /// of the default. /// /// You should exceptionnaly use the static variables only if you need the default theme even if the /// user has specified another theme. pub struct Theme { pub index: Vec, pub css: Vec, pub favicon: Vec, pub js: Vec, pub highlight_css: Vec, pub tomorrow_night_css: Vec, pub highlight_js: Vec, pub jquery: Vec, } impl Theme { pub fn new(src: &Path) -> Self { // Default theme let mut theme = Theme { index: INDEX.to_owned(), css: CSS.to_owned(), favicon: FAVICON.to_owned(), js: JS.to_owned(), highlight_css: HIGHLIGHT_CSS.to_owned(), tomorrow_night_css: TOMORROW_NIGHT_CSS.to_owned(), highlight_js: HIGHLIGHT_JS.to_owned(), jquery: JQUERY.to_owned(), }; // Check if the given path exists if !src.exists() || !src.is_dir() { return theme } let src = src.join("theme"); // If src does exist, check if there is a theme directory in it if !src.exists() || !src.is_dir() { return theme } // Check for individual files if they exist // index.hbs if let Ok(mut f) = File::open(&src.join("index.hbs")) { theme.index.clear(); // Reset the value, because read_to_string appends... let _ = f.read_to_end(&mut theme.index); } // book.js if let Ok(mut f) = File::open(&src.join("book.js")) { theme.js.clear(); let _ = f.read_to_end(&mut theme.js); } // book.css if let Ok(mut f) = File::open(&src.join("book.css")) { theme.css.clear(); let _ = f.read_to_end(&mut theme.css); } // favicon.png if let Ok(mut f) = File::open(&src.join("favicon.png")) { theme.favicon.clear(); let _ = f.read_to_end(&mut theme.favicon); } // highlight.js if let Ok(mut f) = File::open(&src.join("highlight.js")) { theme.highlight_js.clear(); let _ = f.read_to_end(&mut theme.highlight_js); } // highlight.css if let Ok(mut f) = File::open(&src.join("highlight.css")) { theme.highlight_css.clear(); let _ = f.read_to_end(&mut theme.highlight_css); } // tomorrow-night.css if let Ok(mut f) = File::open(&src.join("tomorrow-night.css")) { theme.tomorrow_night_css.clear(); let _ = f.read_to_end(&mut theme.tomorrow_night_css); } theme } }