use std::{fs, num::NonZeroUsize, path::Path}; use chrono::{DateTime, Local}; use clap::{App, AppSettings, Arg, ArgMatches}; use flexi_logger::LevelFilter as LogLevel; use itertools::Itertools; type Error = crate::errors::CliError; const DEFAULT_MAX_CONN: usize = 8; #[derive(derive_builder::Builder)] pub struct AppConfig { /// Urls for store in epub pub urls: Vec, pub max_conn: usize, /// Path to file of multiple articles into a single epub pub merged: Option, pub output_directory: Option, pub log_level: LogLevel, pub can_disable_progress_bar: bool, pub start_time: DateTime, pub is_logging_to_file: bool, pub inline_toc: bool, } impl AppConfig { pub fn init_with_cli() -> Result { let app = App::new("paperoni") .settings(&[ AppSettings::ArgRequiredElseHelp, AppSettings::UnifiedHelpMessage, ]) .version(clap::crate_version!()) .about( "Paperoni is a CLI tool made in Rust for downloading web articles as EPUBs", ) .arg( Arg::with_name("urls") .help("Urls of web articles") .multiple(true), ) .arg( Arg::with_name("file") .short("f") .long("file") .help("Input file containing links") .takes_value(true), ) .arg( Arg::with_name("output_directory") .long("output-dir") .short("o") .help("Directory to store output epub documents") .conflicts_with("output_name") .takes_value(true), ) .arg( Arg::with_name("output_name") .long("merge") .help("Merge multiple articles into a single epub") .long_help("Merge multiple articles into a single epub that will be given the name provided") .conflicts_with("output_directory") .takes_value(true), ).arg( Arg::with_name("max-conn") .long("max_conn") .help("The maximum number of concurrent HTTP connections when downloading articles. Default is 8") .long_help("The maximum number of concurrent HTTP connections when downloading articles. Default is 8.\nNOTE: It is advised to use as few connections as needed i.e between 1 and 50. Using more connections can end up overloading your network card with too many concurrent requests.") .takes_value(true)) .arg( Arg::with_name("verbosity") .short("v") .multiple(true) .help("Enables logging of events and set the verbosity level. Use --help to read on its usage") .long_help( "This takes upto 4 levels of verbosity in the following order. - Error (-v) - Warn (-vv) - Info (-vvv) - Debug (-vvvv) When this flag is passed, it disables the progress bars and logs to stderr. If you would like to send the logs to a file (and enable progress bars), pass the log-to-file flag." ) .takes_value(false)) .arg( Arg::with_name("log-to-file") .long("log-to-file") .help("Enables logging of events to a file located in .paperoni/logs with a default log level of debug. Use -v to specify the logging level") .takes_value(false)) .arg( Arg::with_name("inline-toc") .long("inline-toc") .requires("output_name") .help("Add an inlined Table of Contents page at the start of the merged article.") .long_help("Add an inlined Table of Contents page at the start of the merged article. This does not affect the Table of Contents navigation") ); Self::try_from(app.get_matches()) } fn init_merge_file(self) -> Result { self.merged .as_deref() .map(fs::File::create) .transpose() .err() .map(|err| Err(Error::InvalidOutputPath(err.to_string()))) .unwrap_or(Ok(self)) } fn init_logger(self) -> Result { use crate::logs; logs::init_logger(self.log_level, &self.start_time, self.is_logging_to_file) .map(|_| self) .map_err(Error::LogError) } } use std::convert::TryFrom; impl<'a> TryFrom> for AppConfig { type Error = Error; fn try_from(arg_matches: ArgMatches<'a>) -> Result { AppConfigBuilder::default() .urls({ let url_filter = |url: &str| { let url = url.trim(); if !url.is_empty() { Some(url.to_owned()) } else { None } }; let direct_urls = arg_matches .values_of("urls") .and_then(|urls| urls.map(url_filter).collect::>>()) .unwrap_or(Vec::new()); let file_urls = arg_matches .value_of("file") .map(fs::read_to_string) .transpose()? .and_then(|content| content.lines().map(url_filter).collect::>>()) .unwrap_or(Vec::new()); let urls = [direct_urls, file_urls] .concat() .into_iter() .unique() .collect_vec(); if !urls.is_empty() { Ok(urls) } else { Err(Error::NoUrls) } }?) .max_conn(match arg_matches.value_of("max-conn") { Some(max_conn) => max_conn.parse::()?.get(), None => DEFAULT_MAX_CONN, }) .merged(arg_matches.value_of("output_name").map(ToOwned::to_owned)) .can_disable_progress_bar( arg_matches.is_present("verbosity") && !arg_matches.is_present("log-to-file"), ) .log_level(match arg_matches.occurrences_of("verbosity") { 0 => { if !arg_matches.is_present("log-to-file") { LogLevel::Off } else { LogLevel::Debug } } 1 => LogLevel::Error, 2 => LogLevel::Warn, 3 => LogLevel::Info, 4..=u64::MAX => LogLevel::Debug, }) .is_logging_to_file(arg_matches.is_present("log-to-file")) .inline_toc(arg_matches.is_present("inline-toc")) .output_directory( arg_matches .value_of("output_directory") .map(|output_directory| { let path = Path::new(output_directory); if !path.exists() { Err(Error::OutputDirectoryNotExists) } else if !path.is_dir() { Err(Error::WrongOutputDirectory) } else { Ok(output_directory.to_owned()) } }) .transpose()?, ) .start_time(Local::now()) .try_init() } } impl AppConfigBuilder { pub fn try_init(&self) -> Result { self.build() .map_err(Error::AppBuildError)? .init_logger()? .init_merge_file() } }