alog 0.7.2

use clap::{builder::ValueParser, Arg, ArgAction, Command};
use std::ffi::OsString;
use std::path::Path;

#[allow(clippy::too_many_lines)]
fn main() {
    let default_config = alog::Config::default();
    let mut config = alog::Config::default();
    let mut ioconfig = alog::IOConfig::default();

    let cli_arguments = Command::new(env!("CARGO_PKG_NAME"))
        .version(env!("CARGO_PKG_VERSION"))
        .about("Mangle common / combined logs")
        .override_help(format!(
            "{} {}
Mangle common / combined logs

USAGE:
    alog [FLAGS] [OPTIONS] [INPUT]...

FLAGS:
    -a, --authuser        Clear authuser
    -f, --flush-line      Flush output on every line
        --no-optimize     Don't try to reduce performance hit with `--authuser`
    -n, --notrim          Don't remove Space and Tab from the start of every line
    -s, --skip-invalid    Skip invalid lines

    -h, --help            Print this message
    -V, --version         Print version information

OPTIONS:
        --host-replacement <host-replacement>    Sets host replacement string [default: localhost]
    -4, --ipv4-replacement <ipv4-replacement>    Sets IPv4 replacement string [default: 127.0.0.1]
    -6, --ipv6-replacement <ipv6-replacement>    Sets IPv6 replacement string [default: ::1]
    -o, --output <FILE>                          Sets output file

ARGS:
    <INPUT>...    The input file(s) to use",
            env!("CARGO_PKG_NAME"),
            env!("CARGO_PKG_VERSION")
        ))
        .arg(
            Arg::new("ipv4-replacement")
                .short('4')
                .long("ipv4-replacement")
                .value_name("ipv4-replacement")
                .default_value(default_config.get_ipv4_value())
                .help("Sets IPv4 replacement string")
                .num_args(1),
        )
        .arg(
            Arg::new("ipv6-replacement")
                .short('6')
                .long("ipv6-replacement")
                .value_name("ipv6-replacement")
                .default_value(default_config.get_ipv6_value())
                .help("Sets IPv6 replacement string")
                .num_args(1),
        )
        .arg(
            Arg::new("host-replacement")
                .long("host-replacement")
                .value_name("host-replacement")
                .default_value(default_config.get_host_value())
                .help("Sets host replacement string")
                .num_args(1),
        )
        .arg(
            Arg::new("skip-invalid")
                .short('s')
                .long("skip-invalid")
                .help("Skip invalid lines")
                .action(clap::ArgAction::SetTrue),
        )
        .arg(
            Arg::new("authuser")
                .short('a')
                .long("authuser")
                .help("Clear authuser")
                .action(clap::ArgAction::SetTrue),
        )
        .arg(
            Arg::new("notrim")
                .short('n')
                .long("notrim")
                .help("Don't remove Space and Tab from the start of every line")
                .action(clap::ArgAction::SetTrue),
        )
        .arg(
            Arg::new("nooptimize")
                .long("no-optimize")
                .requires("authuser")
                .help("Don't try to reduce performance hit with `--authuser`")
                .action(clap::ArgAction::SetTrue),
        )
        .arg(
            Arg::new("flush-line")
                .short('f')
                .long("flush-line")
                .help("Flush output on every line")
                .action(clap::ArgAction::SetTrue),
        )
        .arg(
            Arg::new("output")
                .short('o')
                .long("output")
                .value_name("FILE")
                .help("Sets output file")
                .value_parser(ValueParser::os_string())
                .num_args(1),
        )
        .arg(
            Arg::new("input")
                .value_name("INPUT")
                .help("The input file(s) to use")
                .index(1)
                .value_parser(ValueParser::os_string())
                .action(ArgAction::Append),
        )
        .get_matches();

    if let Some(ipv4) = cli_arguments
        .get_one::<String>("ipv4-replacement")
        .map(std::string::String::as_str)
    {
        config.set_ipv4_value(ipv4);
    }

    if let Some(ipv6) = cli_arguments
        .get_one::<String>("ipv6-replacement")
        .map(std::string::String::as_str)
    {
        config.set_ipv6_value(ipv6);
    }

    if let Some(host) = cli_arguments
        .get_one::<String>("host-replacement")
        .map(std::string::String::as_str)
    {
        config.set_host_value(host);
    }

    if let Some(ipv4) = cli_arguments
        .get_one::<String>("ipv4-replacement")
        .map(std::string::String::as_str)
    {
        config.set_ipv4_value(ipv4);
    }

    if cli_arguments.get_flag("flush-line") {
        config.set_flush(true);
    }

    if cli_arguments.get_flag("authuser") {
        config.set_authuser(true);
    }

    if cli_arguments.get_flag("notrim") {
        config.set_trim(false);
    }

    if cli_arguments.get_flag("nooptimize") {
        config.set_optimize(false);
    }

    if cli_arguments.get_flag("skip-invalid") {
        config.set_skip(true);
    }

    if let Some(output) = cli_arguments
        .get_one::<OsString>("output")
        .map(std::ffi::OsString::as_os_str)
    {
        ioconfig.set_output(Path::new(output));
    }

    if let Some(input) = cli_arguments.get_many::<OsString>("input") {
        for file in input.collect::<Vec<_>>() {
            ioconfig.push_input(file);
        }
    }

    if let Err(e) = alog::run(&config, &ioconfig) {
        eprintln!("Error: {e}");
        std::process::exit(1);
    };
}