87 lines
2.4 KiB
Rust
87 lines
2.4 KiB
Rust
use std::path::PathBuf;
|
|
use std::time::Instant;
|
|
|
|
use argh::FromArgs;
|
|
use shtola::log::*;
|
|
use shtola::plugins;
|
|
use shtola::Shtola;
|
|
|
|
const VERSION: &'static str = "0.5.6";
|
|
|
|
#[derive(FromArgs)]
|
|
/// A static site generator.
|
|
struct Args {
|
|
/// output the current version
|
|
#[argh(switch)]
|
|
version: bool,
|
|
|
|
/// the source directory
|
|
#[argh(option, short = 's', default = "PathBuf::from(\"./\")")]
|
|
source: PathBuf,
|
|
|
|
/// the destination directory
|
|
#[argh(option, short = 'd', default = "PathBuf::from(\"./dest\")")]
|
|
destination: PathBuf,
|
|
|
|
/// whether to wipe the destination directory before building
|
|
#[argh(switch, short = 'c')]
|
|
clean: bool,
|
|
|
|
/// what plugins to not run
|
|
#[argh(option, short = 'i')]
|
|
ignored_plugins: Vec<String>,
|
|
|
|
/// whether to skip parsing frontmatter (will probably break plugins)
|
|
#[argh(switch)]
|
|
no_frontmatter: bool,
|
|
|
|
/// whether to output debug stuff
|
|
#[argh(switch, short = 'v')]
|
|
verbose: bool,
|
|
}
|
|
|
|
fn main() {
|
|
let args: Args = argh::from_env();
|
|
|
|
if args.version {
|
|
println!("{}", VERSION);
|
|
std::process::exit(0);
|
|
}
|
|
|
|
if args.verbose {
|
|
minifemme::start(minifemme::LevelFilter::Trace, minifemme::LogMode::Pretty);
|
|
} else {
|
|
minifemme::start(minifemme::LevelFilter::Info, minifemme::LogMode::Pretty);
|
|
}
|
|
|
|
info!("Starting build");
|
|
let start = Instant::now();
|
|
let mut sh = Shtola::new();
|
|
sh.source(&args.source);
|
|
sh.destination(args.destination);
|
|
sh.clean(args.clean);
|
|
sh.frontmatter(!args.no_frontmatter);
|
|
let potential_gitignore = args.source.join(".gitignore");
|
|
if potential_gitignore.exists() {
|
|
sh.source_ignores(&potential_gitignore).unwrap();
|
|
}
|
|
let potential_krileignore = args.source.join(".krileignore");
|
|
if potential_krileignore.exists() {
|
|
sh.source_ignores(&potential_krileignore).unwrap();
|
|
}
|
|
if !args.ignored_plugins.contains(&"markdown".to_string()) {
|
|
sh.register(plugins::markdown::plugin());
|
|
}
|
|
if !args.ignored_plugins.contains(&"tera_layouts".to_string()) {
|
|
sh.register(plugins::tera_layouts::plugin());
|
|
}
|
|
if !args.ignored_plugins.contains(&"pretty_links".to_string()) {
|
|
sh.register(plugins::pretty_links::plugin());
|
|
}
|
|
|
|
sh.build().unwrap();
|
|
info!(
|
|
"Finished build in {:?}",
|
|
Instant::now().duration_since(start)
|
|
);
|
|
}
|