|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933 |
- extern crate glob;
- extern crate rayon;
- extern crate serde;
- extern crate tera;
- #[macro_use]
- extern crate serde_derive;
- extern crate sass_rs;
-
- #[macro_use]
- extern crate errors;
- extern crate config;
- extern crate front_matter;
- extern crate imageproc;
- extern crate library;
- extern crate search;
- extern crate templates;
- extern crate utils;
-
- #[cfg(test)]
- extern crate tempfile;
-
- use std::collections::HashMap;
- use std::fs::{copy, create_dir_all, remove_dir_all};
- use std::path::{Path, PathBuf};
- use std::sync::{Arc, Mutex};
-
- use glob::glob;
- use rayon::prelude::*;
- use sass_rs::{compile_file, Options as SassOptions, OutputStyle};
- use tera::{Context, Tera};
-
- use config::{get_config, Config};
- use errors::{Result, ResultExt};
- use front_matter::InsertAnchor;
- use library::{
- find_taxonomies, sort_actual_pages_by_date, Library, Page, Paginator, Section, Taxonomy,
- };
- use templates::{global_fns, render_redirect_template, ZOLA_TERA};
- use utils::fs::{copy_directory, create_directory, create_file, ensure_directory_exists};
- use utils::net::get_available_port;
- use utils::templates::{render_template, rewrite_theme_paths};
-
- /// The sitemap only needs links and potentially date so we trim down
- /// all pages to only that
- #[derive(Debug, Serialize)]
- struct SitemapEntry {
- permalink: String,
- date: Option<String>,
- }
-
- impl SitemapEntry {
- pub fn new(permalink: String, date: Option<String>) -> SitemapEntry {
- SitemapEntry { permalink, date }
- }
- }
-
- #[derive(Debug)]
- pub struct Site {
- /// The base path of the zola site
- pub base_path: PathBuf,
- /// The parsed config for the site
- pub config: Config,
- pub tera: Tera,
- imageproc: Arc<Mutex<imageproc::Processor>>,
- // the live reload port to be used if there is one
- pub live_reload: Option<u16>,
- pub output_path: PathBuf,
- content_path: PathBuf,
- pub static_path: PathBuf,
- pub taxonomies: Vec<Taxonomy>,
- /// A map of all .md files (section and pages) and their permalink
- /// We need that if there are relative links in the content that need to be resolved
- pub permalinks: HashMap<String, String>,
- /// Contains all pages and sections of the site
- pub library: Library,
- }
-
- impl Site {
- /// Parse a site at the given path. Defaults to the current dir
- /// Passing in a path is only used in tests
- pub fn new<P: AsRef<Path>>(path: P, config_file: &str) -> Result<Site> {
- let path = path.as_ref();
- let mut config = get_config(path, config_file);
- config.load_extra_syntaxes(path)?;
-
- let tpl_glob =
- format!("{}/{}", path.to_string_lossy().replace("\\", "/"), "templates/**/*.*ml");
- // Only parsing as we might be extending templates from themes and that would error
- // as we haven't loaded them yet
- let mut tera = Tera::parse(&tpl_glob).chain_err(|| "Error parsing templates")?;
- if let Some(theme) = config.theme.clone() {
- // Grab data from the extra section of the theme
- config.merge_with_theme(&path.join("themes").join(&theme).join("theme.toml"))?;
-
- // Test that the templates folder exist for that theme
- let theme_path = path.join("themes").join(&theme);
- if !theme_path.join("templates").exists() {
- bail!("Theme `{}` is missing a templates folder", theme);
- }
-
- let theme_tpl_glob = format!(
- "{}/{}",
- path.to_string_lossy().replace("\\", "/"),
- format!("themes/{}/templates/**/*.*ml", theme)
- );
- let mut tera_theme =
- Tera::parse(&theme_tpl_glob).chain_err(|| "Error parsing templates from themes")?;
- rewrite_theme_paths(&mut tera_theme, &theme);
- // TODO: same as below
- if theme_path.join("templates").join("robots.txt").exists() {
- tera_theme
- .add_template_file(theme_path.join("templates").join("robots.txt"), None)?;
- }
- tera_theme.build_inheritance_chains()?;
- tera.extend(&tera_theme)?;
- }
- tera.extend(&ZOLA_TERA)?;
- // the `extend` above already does it but hey
- tera.build_inheritance_chains()?;
-
- // TODO: Tera doesn't use globset right now so we can load the robots.txt as part
- // of the glob above, therefore we load it manually if it exists.
- if path.join("templates").join("robots.txt").exists() {
- tera.add_template_file(path.join("templates").join("robots.txt"), Some("robots.txt"))?;
- }
-
- let content_path = path.join("content");
- let static_path = path.join("static");
- let imageproc =
- imageproc::Processor::new(content_path.clone(), &static_path, &config.base_url);
-
- let site = Site {
- base_path: path.to_path_buf(),
- config,
- tera,
- imageproc: Arc::new(Mutex::new(imageproc)),
- live_reload: None,
- output_path: path.join("public"),
- content_path,
- static_path,
- taxonomies: Vec::new(),
- permalinks: HashMap::new(),
- // We will allocate it properly later on
- library: Library::new(0, 0),
- };
-
- Ok(site)
- }
-
- /// The index section is ALWAYS at that path
- pub fn index_section_path(&self) -> PathBuf {
- self.content_path.join("_index.md")
- }
-
- /// We avoid the port the server is going to use as it's not bound yet
- /// when calling this function and we could end up having tried to bind
- /// both http and websocket server to the same port
- pub fn enable_live_reload(&mut self, port_to_avoid: u16) {
- self.live_reload = get_available_port(port_to_avoid);
- }
-
- /// Get all the orphan (== without section) pages in the site
- pub fn get_all_orphan_pages(&self) -> Vec<&Page> {
- self.library.get_all_orphan_pages()
- }
-
- pub fn set_base_url(&mut self, base_url: String) {
- let mut imageproc = self.imageproc.lock().unwrap();
- imageproc.set_base_url(&base_url);
- self.config.base_url = base_url;
- }
-
- pub fn set_output_path<P: AsRef<Path>>(&mut self, path: P) {
- self.output_path = path.as_ref().to_path_buf();
- }
-
- /// Reads all .md files in the `content` directory and create pages/sections
- /// out of them
- pub fn load(&mut self) -> Result<()> {
- let base_path = self.base_path.to_string_lossy().replace("\\", "/");
- let content_glob = format!("{}/{}", base_path, "content/**/*.md");
-
- let (section_entries, page_entries): (Vec<_>, Vec<_>) = glob(&content_glob)
- .unwrap()
- .filter_map(|e| e.ok())
- .filter(|e| !e.as_path().file_name().unwrap().to_str().unwrap().starts_with('.'))
- .partition(|entry| entry.as_path().file_name().unwrap() == "_index.md");
-
- self.library = Library::new(page_entries.len(), section_entries.len());
-
- let sections = {
- let config = &self.config;
-
- section_entries
- .into_par_iter()
- .map(|entry| {
- let path = entry.as_path();
- Section::from_file(path, config)
- })
- .collect::<Vec<_>>()
- };
-
- let pages = {
- let config = &self.config;
-
- page_entries
- .into_par_iter()
- .map(|entry| {
- let path = entry.as_path();
- Page::from_file(path, config)
- })
- .collect::<Vec<_>>()
- };
-
- // Kinda duplicated code for add_section/add_page but necessary to do it that
- // way because of the borrow checker
- for section in sections {
- let s = section?;
- self.add_section(s, false)?;
- }
-
- // Insert a default index section if necessary so we don't need to create
- // a _index.md to render the index page at the root of the site
- let index_path = self.index_section_path();
- if let Some(ref index_section) = self.library.get_section(&index_path) {
- if self.config.build_search_index && !index_section.meta.in_search_index {
- bail!(
- "You have enabled search in the config but disabled it in the index section: \
- either turn off the search in the config or remote `in_search_index = true` from the \
- section front-matter."
- )
- }
- }
- // Not in else because of borrow checker
- if !self.library.contains_section(&index_path) {
- let mut index_section = Section::default();
- index_section.permalink = self.config.make_permalink("");
- index_section.file.path = self.content_path.join("_index.md");
- index_section.file.parent = self.content_path.clone();
- index_section.file.relative = "_index.md".to_string();
- self.library.insert_section(index_section);
- }
-
- let mut pages_insert_anchors = HashMap::new();
- for page in pages {
- let p = page?;
- pages_insert_anchors.insert(
- p.file.path.clone(),
- self.find_parent_section_insert_anchor(&p.file.parent.clone()),
- );
- self.add_page(p, false)?;
- }
-
- self.register_early_global_fns();
- self.populate_sections();
- self.render_markdown()?;
- self.populate_taxonomies()?;
- self.register_tera_global_fns();
-
- Ok(())
- }
-
- /// Render the markdown of all pages/sections
- /// Used in a build and in `serve` if a shortcode has changed
- pub fn render_markdown(&mut self) -> Result<()> {
- // Another silly thing needed to not borrow &self in parallel and
- // make the borrow checker happy
- let permalinks = &self.permalinks;
- let tera = &self.tera;
- let config = &self.config;
-
- // This is needed in the first place because of silly borrow checker
- let mut pages_insert_anchors = HashMap::new();
- for (_, p) in self.library.pages() {
- pages_insert_anchors.insert(
- p.file.path.clone(),
- self.find_parent_section_insert_anchor(&p.file.parent.clone()),
- );
- }
-
- self.library
- .pages_mut()
- .values_mut()
- .collect::<Vec<_>>()
- .par_iter_mut()
- .map(|page| {
- let insert_anchor = pages_insert_anchors[&page.file.path];
- page.render_markdown(permalinks, tera, config, insert_anchor)
- })
- .collect::<Result<()>>()?;
-
- self.library
- .sections_mut()
- .values_mut()
- .collect::<Vec<_>>()
- .par_iter_mut()
- .map(|section| section.render_markdown(permalinks, tera, config))
- .collect::<Result<()>>()?;
-
- Ok(())
- }
-
- /// Adds global fns that are to be available to shortcodes while rendering markdown
- pub fn register_early_global_fns(&mut self) {
- self.tera.register_function(
- "get_url",
- global_fns::make_get_url(self.permalinks.clone(), self.config.clone()),
- );
- self.tera.register_function(
- "resize_image",
- global_fns::make_resize_image(self.imageproc.clone()),
- );
- }
-
- pub fn register_tera_global_fns(&mut self) {
- self.tera.register_function("trans", global_fns::make_trans(self.config.clone()));
- self.tera.register_function("get_page", global_fns::make_get_page(&self.library));
- self.tera.register_function("get_section", global_fns::make_get_section(&self.library));
- self.tera.register_function(
- "get_taxonomy",
- global_fns::make_get_taxonomy(&self.taxonomies, &self.library),
- );
- self.tera.register_function(
- "get_taxonomy_url",
- global_fns::make_get_taxonomy_url(&self.taxonomies),
- );
- self.tera.register_function(
- "load_data",
- global_fns::make_load_data(self.content_path.clone(), self.base_path.clone()),
- );
- }
-
- /// Add a page to the site
- /// The `render` parameter is used in the serve command, when rebuilding a page.
- /// If `true`, it will also render the markdown for that page
- /// Returns the previous page struct if there was one at the same path
- pub fn add_page(&mut self, mut page: Page, render: bool) -> Result<Option<Page>> {
- self.permalinks.insert(page.file.relative.clone(), page.permalink.clone());
- if render {
- let insert_anchor = self.find_parent_section_insert_anchor(&page.file.parent);
- page.render_markdown(&self.permalinks, &self.tera, &self.config, insert_anchor)?;
- }
- let prev = self.library.remove_page(&page.file.path);
- self.library.insert_page(page);
-
- Ok(prev)
- }
-
- /// Add a section to the site
- /// The `render` parameter is used in the serve command, when rebuilding a page.
- /// If `true`, it will also render the markdown for that page
- /// Returns the previous section struct if there was one at the same path
- pub fn add_section(&mut self, mut section: Section, render: bool) -> Result<Option<Section>> {
- self.permalinks.insert(section.file.relative.clone(), section.permalink.clone());
- if render {
- section.render_markdown(&self.permalinks, &self.tera, &self.config)?;
- }
- let prev = self.library.remove_section(§ion.file.path);
- self.library.insert_section(section);
-
- Ok(prev)
- }
-
- /// Finds the insert_anchor for the parent section of the directory at `path`.
- /// Defaults to `AnchorInsert::None` if no parent section found
- pub fn find_parent_section_insert_anchor(&self, parent_path: &PathBuf) -> InsertAnchor {
- match self.library.get_section(&parent_path.join("_index.md")) {
- Some(s) => s.meta.insert_anchor_links,
- None => InsertAnchor::None,
- }
- }
-
- /// Find out the direct subsections of each subsection if there are some
- /// as well as the pages for each section
- pub fn populate_sections(&mut self) {
- self.library.populate_sections();
- }
-
- /// Find all the tags and categories if it's asked in the config
- pub fn populate_taxonomies(&mut self) -> Result<()> {
- if self.config.taxonomies.is_empty() {
- return Ok(());
- }
-
- self.taxonomies = find_taxonomies(&self.config, &self.library)?;
-
- Ok(())
- }
-
- /// Inject live reload script tag if in live reload mode
- fn inject_livereload(&self, html: String) -> String {
- if let Some(port) = self.live_reload {
- return html.replace(
- "</body>",
- &format!(
- r#"<script src="/livereload.js?port={}&mindelay=10"></script></body>"#,
- port
- ),
- );
- }
-
- html
- }
-
- /// Copy the main `static` folder and the theme `static` folder if a theme is used
- pub fn copy_static_directories(&self) -> Result<()> {
- // The user files will overwrite the theme files
- if let Some(ref theme) = self.config.theme {
- copy_directory(
- &self.base_path.join("themes").join(theme).join("static"),
- &self.output_path,
- )?;
- }
- // We're fine with missing static folders
- if self.static_path.exists() {
- copy_directory(&self.static_path, &self.output_path)?;
- }
-
- Ok(())
- }
-
- pub fn num_img_ops(&self) -> usize {
- let imageproc = self.imageproc.lock().unwrap();
- imageproc.num_img_ops()
- }
-
- pub fn process_images(&self) -> Result<()> {
- let mut imageproc = self.imageproc.lock().unwrap();
- imageproc.prune()?;
- imageproc.do_process()
- }
-
- /// Deletes the `public` directory if it exists
- pub fn clean(&self) -> Result<()> {
- if self.output_path.exists() {
- // Delete current `public` directory so we can start fresh
- remove_dir_all(&self.output_path).chain_err(|| "Couldn't delete output directory")?;
- }
-
- Ok(())
- }
-
- /// Renders a single content page
- pub fn render_page(&self, page: &Page) -> Result<()> {
- ensure_directory_exists(&self.output_path)?;
-
- // Copy the nesting of the content directory if we have sections for that page
- let mut current_path = self.output_path.to_path_buf();
-
- for component in page.path.split('/') {
- current_path.push(component);
-
- if !current_path.exists() {
- create_directory(¤t_path)?;
- }
- }
-
- // Make sure the folder exists
- create_directory(¤t_path)?;
-
- // Finally, create a index.html file there with the page rendered
- let output = page.render_html(&self.tera, &self.config, &self.library)?;
- create_file(¤t_path.join("index.html"), &self.inject_livereload(output))?;
-
- // Copy any asset we found previously into the same directory as the index.html
- for asset in &page.assets {
- let asset_path = asset.as_path();
- copy(&asset_path, ¤t_path.join(asset_path.file_name().unwrap()))?;
- }
-
- Ok(())
- }
-
- /// Deletes the `public` directory and builds the site
- pub fn build(&self) -> Result<()> {
- self.clean()?;
- // Render aliases first to allow overwriting
- self.render_aliases()?;
- self.render_sections()?;
- self.render_orphan_pages()?;
- self.render_sitemap()?;
- if self.config.generate_rss {
- self.render_rss_feed(self.library.pages_values(), None)?;
- }
- self.render_404()?;
- self.render_robots()?;
- self.render_taxonomies()?;
-
- if let Some(ref theme) = self.config.theme {
- let theme_path = self.base_path.join("themes").join(theme);
- if theme_path.join("sass").exists() {
- self.compile_sass(&theme_path)?;
- }
- }
-
- if self.config.compile_sass {
- self.compile_sass(&self.base_path)?;
- }
-
- self.process_images()?;
- self.copy_static_directories()?;
-
- if self.config.build_search_index {
- self.build_search_index()?;
- }
-
- Ok(())
- }
-
- pub fn build_search_index(&self) -> Result<()> {
- // index first
- create_file(
- &self.output_path.join(&format!("search_index.{}.js", self.config.default_language)),
- &format!(
- "window.searchIndex = {};",
- search::build_index(&self.config.default_language, &self.library)?
- ),
- )?;
-
- // then elasticlunr.min.js
- create_file(&self.output_path.join("elasticlunr.min.js"), search::ELASTICLUNR_JS)?;
-
- Ok(())
- }
-
- pub fn compile_sass(&self, base_path: &Path) -> Result<()> {
- ensure_directory_exists(&self.output_path)?;
-
- let sass_path = {
- let mut sass_path = PathBuf::from(base_path);
- sass_path.push("sass");
- sass_path
- };
-
- let mut options = SassOptions::default();
- options.output_style = OutputStyle::Compressed;
- let mut compiled_paths = self.compile_sass_glob(&sass_path, "scss", &options.clone())?;
-
- options.indented_syntax = true;
- compiled_paths.extend(self.compile_sass_glob(&sass_path, "sass", &options)?);
-
- compiled_paths.sort();
- for window in compiled_paths.windows(2) {
- if window[0].1 == window[1].1 {
- bail!(
- "SASS path conflict: \"{}\" and \"{}\" both compile to \"{}\"",
- window[0].0.display(),
- window[1].0.display(),
- window[0].1.display(),
- );
- }
- }
-
- Ok(())
- }
-
- fn compile_sass_glob(
- &self,
- sass_path: &Path,
- extension: &str,
- options: &SassOptions,
- ) -> Result<Vec<(PathBuf, PathBuf)>> {
- let glob_string = format!("{}/**/*.{}", sass_path.display(), extension);
- let files = glob(&glob_string)
- .unwrap()
- .filter_map(|e| e.ok())
- .filter(|entry| {
- !entry.as_path().file_name().unwrap().to_string_lossy().starts_with('_')
- })
- .collect::<Vec<_>>();
-
- let mut compiled_paths = Vec::new();
- for file in files {
- let css = compile_file(&file, options.clone())?;
-
- let path_inside_sass = file.strip_prefix(&sass_path).unwrap();
- let parent_inside_sass = path_inside_sass.parent();
- let css_output_path = self.output_path.join(path_inside_sass).with_extension("css");
-
- if parent_inside_sass.is_some() {
- create_dir_all(&css_output_path.parent().unwrap())?;
- }
-
- create_file(&css_output_path, &css)?;
- compiled_paths.push((path_inside_sass.to_owned(), css_output_path));
- }
-
- Ok(compiled_paths)
- }
-
- pub fn render_aliases(&self) -> Result<()> {
- ensure_directory_exists(&self.output_path)?;
- for (_, page) in self.library.pages() {
- for alias in &page.meta.aliases {
- let mut output_path = self.output_path.to_path_buf();
- let mut split = alias.split('/').collect::<Vec<_>>();
-
- // If the alias ends with an html file name, use that instead of mapping
- // as a path containing an `index.html`
- let page_name = match split.pop() {
- Some(part) if part.ends_with(".html") => part,
- Some(part) => {
- split.push(part);
- "index.html"
- }
- None => "index.html",
- };
-
- for component in split {
- output_path.push(&component);
-
- if !output_path.exists() {
- create_directory(&output_path)?;
- }
- }
- create_file(
- &output_path.join(page_name),
- &render_redirect_template(&page.permalink, &self.tera)?,
- )?;
- }
- }
- Ok(())
- }
-
- /// Renders 404.html
- pub fn render_404(&self) -> Result<()> {
- ensure_directory_exists(&self.output_path)?;
- let mut context = Context::new();
- context.insert("config", &self.config);
- create_file(
- &self.output_path.join("404.html"),
- &render_template("404.html", &self.tera, &context, &self.config.theme)?,
- )
- }
-
- /// Renders robots.txt
- pub fn render_robots(&self) -> Result<()> {
- ensure_directory_exists(&self.output_path)?;
- let mut context = Context::new();
- context.insert("config", &self.config);
- create_file(
- &self.output_path.join("robots.txt"),
- &render_template("robots.txt", &self.tera, &context, &self.config.theme)?,
- )
- }
-
- /// Renders all taxonomies with at least one non-draft post
- pub fn render_taxonomies(&self) -> Result<()> {
- for taxonomy in &self.taxonomies {
- self.render_taxonomy(taxonomy)?;
- }
-
- Ok(())
- }
-
- fn render_taxonomy(&self, taxonomy: &Taxonomy) -> Result<()> {
- if taxonomy.items.is_empty() {
- return Ok(());
- }
-
- ensure_directory_exists(&self.output_path)?;
- let output_path = self.output_path.join(&taxonomy.kind.name);
- let list_output = taxonomy.render_all_terms(&self.tera, &self.config, &self.library)?;
- create_directory(&output_path)?;
- create_file(&output_path.join("index.html"), &self.inject_livereload(list_output))?;
-
- taxonomy
- .items
- .par_iter()
- .map(|item| {
- let path = output_path.join(&item.slug);
- if taxonomy.kind.is_paginated() {
- self.render_paginated(
- &path,
- &Paginator::from_taxonomy(&taxonomy, item, &self.library),
- )?;
- } else {
- let single_output =
- taxonomy.render_term(item, &self.tera, &self.config, &self.library)?;
- create_directory(&path)?;
- create_file(&path.join("index.html"), &self.inject_livereload(single_output))?;
- }
-
- if taxonomy.kind.rss {
- self.render_rss_feed(
- item.pages.iter().map(|p| self.library.get_page_by_key(*p)).collect(),
- Some(&PathBuf::from(format!("{}/{}", taxonomy.kind.name, item.slug))),
- )
- } else {
- Ok(())
- }
- })
- .collect::<Result<()>>()
- }
-
- /// What it says on the tin
- pub fn render_sitemap(&self) -> Result<()> {
- ensure_directory_exists(&self.output_path)?;
-
- let mut context = Context::new();
-
- let mut pages = self
- .library
- .pages_values()
- .iter()
- .filter(|p| !p.is_draft())
- .map(|p| {
- let date = match p.meta.date {
- Some(ref d) => Some(d.to_string()),
- None => None,
- };
- SitemapEntry::new(p.permalink.clone(), date)
- })
- .collect::<Vec<_>>();
- pages.sort_by(|a, b| a.permalink.cmp(&b.permalink));
- context.insert("pages", &pages);
-
- let mut sections = self
- .library
- .sections_values()
- .iter()
- .map(|s| SitemapEntry::new(s.permalink.clone(), None))
- .collect::<Vec<_>>();
- for section in self.library.sections_values().iter().filter(|s| s.meta.paginate_by.is_some()) {
- let number_pagers = (section.pages.len() as f64 / section.meta.paginate_by.unwrap() as f64).ceil() as isize;
- for i in 1..number_pagers+1 {
- let permalink = format!("{}{}/{}/", section.permalink, section.meta.paginate_path, i);
- sections.push(SitemapEntry::new(permalink, None))
- }
- }
- sections.sort_by(|a, b| a.permalink.cmp(&b.permalink));
- context.insert("sections", §ions);
-
- let mut taxonomies = vec![];
- for taxonomy in &self.taxonomies {
- let name = &taxonomy.kind.name;
- let mut terms = vec![];
- terms.push(SitemapEntry::new(self.config.make_permalink(name), None));
- for item in &taxonomy.items {
- terms.push(SitemapEntry::new(
- self.config.make_permalink(&format!("{}/{}", &name, item.slug)),
- None,
- ));
-
- if taxonomy.kind.is_paginated() {
- let number_pagers = (item.pages.len() as f64 / taxonomy.kind.paginate_by.unwrap() as f64).ceil() as isize;
- for i in 1..number_pagers+1 {
- let permalink = self.config.make_permalink(&format!("{}/{}/{}/{}", name, item.slug, taxonomy.kind.paginate_path(), i));
- terms.push(SitemapEntry::new(permalink, None))
- }
- }
- }
-
- terms.sort_by(|a, b| a.permalink.cmp(&b.permalink));
- taxonomies.push(terms);
- }
-
- context.insert("taxonomies", &taxonomies);
- context.insert("config", &self.config);
-
- let sitemap = &render_template("sitemap.xml", &self.tera, &context, &self.config.theme)?;
-
- create_file(&self.output_path.join("sitemap.xml"), sitemap)?;
-
- Ok(())
- }
-
- /// Renders a RSS feed for the given path and at the given path
- /// If both arguments are `None`, it will render only the RSS feed for the whole
- /// site at the root folder.
- pub fn render_rss_feed(
- &self,
- all_pages: Vec<&Page>,
- base_path: Option<&PathBuf>,
- ) -> Result<()> {
- ensure_directory_exists(&self.output_path)?;
-
- let mut context = Context::new();
- let mut pages = all_pages
- .into_iter()
- .filter(|p| p.meta.date.is_some() && !p.is_draft())
- .collect::<Vec<_>>();
-
- // Don't generate a RSS feed if none of the pages has a date
- if pages.is_empty() {
- return Ok(());
- }
-
- pages.par_sort_unstable_by(sort_actual_pages_by_date);
-
- context.insert("last_build_date", &pages[0].meta.date.clone());
- // limit to the last n elements if the limit is set; otherwise use all.
- let num_entries = self.config.rss_limit.unwrap_or(pages.len());
- let p = pages
- .iter()
- .take(num_entries)
- .map(|x| x.to_serialized_basic(&self.library))
- .collect::<Vec<_>>();
-
- context.insert("pages", &p);
- context.insert("config", &self.config);
-
- let rss_feed_url = if let Some(ref base) = base_path {
- self.config.make_permalink(&base.join("rss.xml").to_string_lossy().replace('\\', "/"))
- } else {
- self.config.make_permalink("rss.xml")
- };
-
- context.insert("feed_url", &rss_feed_url);
-
- let feed = &render_template("rss.xml", &self.tera, &context, &self.config.theme)?;
-
- if let Some(ref base) = base_path {
- let mut output_path = self.output_path.clone();
- for component in base.components() {
- output_path.push(component);
- if !output_path.exists() {
- create_directory(&output_path)?;
- }
- }
- create_file(&output_path.join("rss.xml"), feed)?;
- } else {
- create_file(&self.output_path.join("rss.xml"), feed)?;
- }
- Ok(())
- }
-
- /// Renders a single section
- pub fn render_section(&self, section: &Section, render_pages: bool) -> Result<()> {
- ensure_directory_exists(&self.output_path)?;
- let mut output_path = self.output_path.clone();
- for component in §ion.file.components {
- output_path.push(component);
-
- if !output_path.exists() {
- create_directory(&output_path)?;
- }
- }
-
- // Copy any asset we found previously into the same directory as the index.html
- for asset in §ion.assets {
- let asset_path = asset.as_path();
- copy(&asset_path, &output_path.join(asset_path.file_name().unwrap()))?;
- }
-
- if render_pages {
- section
- .pages
- .par_iter()
- .map(|k| self.render_page(self.library.get_page_by_key(*k)))
- .collect::<Result<()>>()?;
- }
-
- if !section.meta.render {
- return Ok(());
- }
-
- if let Some(ref redirect_to) = section.meta.redirect_to {
- let permalink = self.config.make_permalink(redirect_to);
- create_file(
- &output_path.join("index.html"),
- &render_redirect_template(&permalink, &self.tera)?,
- )?;
- return Ok(());
- }
-
- if section.meta.is_paginated() {
- self.render_paginated(&output_path, &Paginator::from_section(§ion, &self.library))?;
- } else {
- let output = section.render_html(&self.tera, &self.config, &self.library)?;
- create_file(&output_path.join("index.html"), &self.inject_livereload(output))?;
- }
-
- Ok(())
- }
-
- /// Used only on reload
- pub fn render_index(&self) -> Result<()> {
- self.render_section(
- &self.library.get_section(&self.content_path.join("_index.md")).unwrap(),
- false,
- )
- }
-
- /// Renders all sections
- pub fn render_sections(&self) -> Result<()> {
- self.library
- .sections_values()
- .into_par_iter()
- .map(|s| self.render_section(s, true))
- .collect::<Result<()>>()
- }
-
- /// Renders all pages that do not belong to any sections
- pub fn render_orphan_pages(&self) -> Result<()> {
- ensure_directory_exists(&self.output_path)?;
-
- for page in self.get_all_orphan_pages() {
- self.render_page(page)?;
- }
-
- Ok(())
- }
-
- /// Renders a list of pages when the section/index is wanting pagination.
- pub fn render_paginated(&self, output_path: &Path, paginator: &Paginator) -> Result<()> {
- ensure_directory_exists(&self.output_path)?;
-
- let folder_path = output_path.join(&paginator.paginate_path);
- create_directory(&folder_path)?;
-
- paginator
- .pagers
- .par_iter()
- .map(|pager| {
- let page_path = folder_path.join(&format!("{}", pager.index));
- create_directory(&page_path)?;
- let output =
- paginator.render_pager(pager, &self.config, &self.tera, &self.library)?;
- if pager.index > 1 {
- create_file(&page_path.join("index.html"), &self.inject_livereload(output))?;
- } else {
- create_file(&output_path.join("index.html"), &self.inject_livereload(output))?;
- create_file(
- &page_path.join("index.html"),
- &render_redirect_template(&paginator.permalink, &self.tera)?,
- )?;
- }
- Ok(())
- })
- .collect::<Result<()>>()
- }
- }
|