- 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 = matches.value_of("config").unwrap();
-
- match matches.subcommand() {
- ("init", Some(matches)) => {
- match cmd::create_new_project(matches.value_of("name").unwrap()) {
- 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 = 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 = 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);
- }
- };
- }
- ("index", Some(matches)) => {
- console::info("Building search index...");
- let start = Instant::now();
- let output_dir = matches.value_of("output_dir").unwrap();
- let show_drafts = matches.is_present("drafts");
-
- let indexing_result = {
- #[cfg(feature = "tantivy-indexing")]
- {
- let index_type = matches.value_of("index_type").unwrap();
- cmd::index( &root_dir, config_file, matches.value_of("base_url"), output_dir, show_drafts, index_type)
- }
-
- #[cfg(not(feature = "tantivy-indexing"))]
- cmd::index( &root_dir, config_file, matches.value_of("base_url"), output_dir, show_drafts)
- };
-
- match indexing_result {
- Ok(()) => console::report_elapsed_time(start),
-
- Err(e) => {
- console::unravel_errors("Failed to build search index", &e);
- ::std::process::exit(1);
- }
- };
- }
-
- _ => unreachable!(),
- }
- }
|