zola/src/main.rs

108 lines
3.6 KiB
Rust
Raw Normal View History

use std::time::Instant;
2017-03-25 06:52:51 +00:00
use utils::net::{get_available_port, port_is_available};
2018-10-31 07:18:57 +00:00
mod cli;
2016-12-06 05:51:33 +00:00
mod cmd;
2017-03-25 06:52:51 +00:00
mod console;
2017-07-27 09:24:43 +00:00
mod prompt;
2016-12-06 06:55:17 +00:00
2016-12-06 05:51:33 +00:00
fn main() {
2017-07-15 04:24:31 +00:00
let matches = cli::build_cli().get_matches();
2016-12-06 05:51:33 +00:00
2018-03-16 18:11:08 +00:00
let config_file = matches.value_of("config").unwrap();
2016-12-06 05:51:33 +00:00
match matches.subcommand() {
2016-12-19 07:58:03 +00:00
("init", Some(matches)) => {
2016-12-06 05:51:33 +00:00
match cmd::create_new_project(matches.value_of("name").unwrap()) {
2017-07-27 09:24:43 +00:00
Ok(()) => (),
Err(e) => {
console::unravel_errors("Failed to create the project", &e);
::std::process::exit(1);
2018-10-31 07:18:57 +00:00
}
2016-12-06 08:27:03 +00:00
};
2018-10-31 07:18:57 +00:00
}
("build", Some(matches)) => {
2017-03-25 06:52:51 +00:00
console::info("Building site...");
let start = Instant::now();
let output_dir = matches.value_of("output_dir").unwrap();
2019-08-24 20:23:08 +00:00
match cmd::build(
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);
2018-10-31 07:18:57 +00:00
}
2016-12-06 05:51:33 +00:00
};
2018-10-31 07:18:57 +00:00
}
("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");
2019-08-24 20:23:08 +00:00
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);
}
2019-07-19 09:10:28 +00:00
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 = matches.value_of("output_dir").unwrap();
let base_url = matches.value_of("base_url").unwrap();
2017-03-25 06:52:51 +00:00
console::info("Building site...");
2019-08-24 20:23:08 +00:00
match cmd::serve(
interface,
port,
output_dir,
base_url,
config_file,
watch_only,
open,
include_drafts,
) {
2017-03-06 10:35:56 +00:00
Ok(()) => (),
Err(e) => {
2017-05-20 15:00:41 +00:00
console::unravel_errors("", &e);
::std::process::exit(1);
2018-10-31 07:18:57 +00:00
}
};
2018-10-31 07:18:57 +00:00
}
("check", Some(matches)) => {
console::info("Checking site...");
let start = Instant::now();
match cmd::check(
config_file,
matches.value_of("base_path"),
matches.value_of("base_url"),
2019-08-24 20:23:08 +00:00
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);
}
};
}
2016-12-06 05:51:33 +00:00
_ => unreachable!(),
}
}