polysite 0.0.2

Highly customizable, polymorphic static site generator library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use crate::*;
use log::info;
use std::fs::remove_dir_all;
use tokio::task::JoinSet;
use tracing_error::SpanTrace;

/// A site builder to use build one site
pub struct Builder {
    ctx: Context,
    steps: Vec<Vec<Rule>>,
}

impl Builder {
    /// Create new builder with [`Config`]
    pub fn new(config: Config) -> Self {
        Self {
            ctx: Context::new(config),
            steps: Vec::new(),
        }
    }

    /// Add a new build step with multiple rules that are built concurrently.
    pub fn add_step(mut self, step: impl IntoIterator<Item = Rule>) -> Self {
        self.steps.push(step.into_iter().collect());
        self
    }

    /// Run all registered build steps
    #[tracing::instrument(skip(self))]
    pub async fn build(self) -> Result<(), Error> {
        let conf = self.ctx.config();
        let target_dir = conf.target_dir();
        if conf.target_clean() && target_dir.is_dir() {
            remove_dir_all(&target_dir).map_err(|io_error| Error::FileIo {
                trace: SpanTrace::capture(),
                io_error,
            })?;
            info!("Target directory ({}) cleaned", target_dir.display());
        }
        for step in self.steps.into_iter() {
            let mut set = JoinSet::new();
            for rule in step.into_iter() {
                let ctx = self.ctx.clone();
                set.spawn(rule.compile(ctx));
            }
            while let Some(res) = set.join_next().await {
                let _ctx = res.unwrap()?;
                //self.ctx.metadata_mut().merge(ctx.metadata().clone());
            }
        }
        Ok(())
    }
}