zola/src/main.rs

121 lines
4 KiB
Rust

use std::env;
use std::path::PathBuf;
use std::time::Instant;
use utils::net::{get_available_port, port_is_available};
mod cli;
mod cmd;
mod console;
mod prompt;
fn main() {
let matches = cli::build_cli().get_matches();
let root_dir = match matches.value_of("root").unwrap() {
"." => env::current_dir().unwrap(),
path => PathBuf::from(path),
};
let config_file = match matches.value_of("config") {
Some(path) => PathBuf::from(path),
None => root_dir.join("config.toml"),
};
match matches.subcommand() {
("init", Some(matches)) => {
let force = matches.is_present("force");
match cmd::create_new_project(matches.value_of("name").unwrap(), force) {
Ok(()) => (),
Err(e) => {
console::unravel_errors("Failed to create the project", &e);
::std::process::exit(1);
}
};
}
("build", Some(matches)) => {
console::info("Building site...");
let start = Instant::now();
let output_dir = PathBuf::from(matches.value_of("output_dir").unwrap());
match cmd::build(
&root_dir,
&config_file,
matches.value_of("base_url"),
&output_dir,
matches.is_present("drafts"),
) {
Ok(()) => console::report_elapsed_time(start),
Err(e) => {
console::unravel_errors("Failed to build the site", &e);
::std::process::exit(1);
}
};
}
("serve", Some(matches)) => {
let interface = matches.value_of("interface").unwrap_or("127.0.0.1");
let mut port: u16 = match matches.value_of("port").unwrap().parse() {
Ok(x) => x,
Err(_) => {
console::error("The request port needs to be an integer");
::std::process::exit(1);
}
};
let watch_only = matches.is_present("watch_only");
let open = matches.is_present("open");
let include_drafts = matches.is_present("drafts");
// Default one
if port != 1111 && !watch_only && !port_is_available(port) {
console::error("The requested port is not available");
::std::process::exit(1);
}
if !watch_only && !port_is_available(port) {
port = if let Some(p) = get_available_port(1111) {
p
} else {
console::error("No port available.");
::std::process::exit(1);
}
}
let output_dir = PathBuf::from(matches.value_of("output_dir").unwrap());
let base_url = matches.value_of("base_url").unwrap();
console::info("Building site...");
match cmd::serve(
&root_dir,
interface,
port,
&output_dir,
base_url,
&config_file,
watch_only,
open,
include_drafts,
) {
Ok(()) => (),
Err(e) => {
console::unravel_errors("", &e);
::std::process::exit(1);
}
};
}
("check", Some(matches)) => {
console::info("Checking site...");
let start = Instant::now();
match cmd::check(
&root_dir,
&config_file,
matches.value_of("base_path"),
matches.value_of("base_url"),
matches.is_present("drafts"),
) {
Ok(()) => console::report_elapsed_time(start),
Err(e) => {
console::unravel_errors("Failed to check the site", &e);
::std::process::exit(1);
}
};
}
_ => unreachable!(),
}
}