mod functions;
mod internal;

use crate::functions::*;
use clap::{App, Arg, SubCommand};

fn main() {
    let app = App::new("jade")
        .version(env!("CARGO_PKG_VERSION"))
        .about(env!("CARGO_PKG_DESCRIPTION"))
        .author(env!("CARGO_PKG_AUTHORS"))
        .subcommand(
            SubCommand::with_name("partition")
                .about("Partition the install destination")
                .arg(
                    Arg::with_name("mode")
                        .help("If jade should automatically partition (mode = auto) or the user manually partitioned it (mode = manual)")
                        .possible_values(&["auto", "manual"])
                        .required(true),
                )
                .arg(
                    Arg::with_name("device")
                        .help("The device to partition")
                        .required_if("mode", "auto"),
                )
                .arg(
                    Arg::with_name("efi")
                        .help("If the install destination should be partitioned with EFI")
                        .long("efi")
                        .takes_value(false),
                ),
        )
        .subcommand(
            SubCommand::with_name("install-base")
                .about("Install base packages")
        )
        .subcommand(
            SubCommand::with_name("genfstab")
                .about("Generate fstab")
        )
        .subcommand(
            SubCommand::with_name("setup-timeshift")
                .about("Setup timeshift")
        )
        .subcommand(
            SubCommand::with_name("bootloader")
                .about("Install bootloader")
                .subcommand(
                    SubCommand::with_name("grub-efi")
                        .about("Install grub-efi")
                        .arg(
                            Arg::with_name("efidir")
                                .help("The directory to install the EFI bootloader to")
                                .required(true),
                        ),
                )
                .subcommand(
                    SubCommand::with_name("grub-legacy")
                        .about("Install grub-legacy")
                        .arg(
                            Arg::with_name("device")
                                .help("The device to install the bootloader to")
                                .required(true),
                        ),
                ),
        )
        .subcommand(
            SubCommand::with_name("locale")
                .about("Set locale stuff")
                .arg(
                    Arg::with_name("keyboard")
                        .help("The keyboard layout to use")
                        .required(true),
                )
                .arg(
                    Arg::with_name("timezone")
                        .help("The timezone to use")
                        .required(true),
                )
                .arg(
                    Arg::with_name("locales")
                        .help("The locales to set")
                        .multiple(true)
                        .index(3)
                        .required(true),
                ),
        )
        .subcommand(
            SubCommand::with_name("networking")
                .about("Set networking stuff")
                .arg(
                    Arg::with_name("hostname")
                        .help("The hostname to use")
                        .required(true),
                )
                .arg(
                    Arg::with_name("ipv6")
                        .help("Wether ipv6 should be enabled")
                        .short("i6")
                        .long("ipv6")
                        .takes_value(false),
                ),
        )
        .subcommand(
            SubCommand::with_name("users")
                .about("Configure users")
                .subcommand(
                    SubCommand::with_name("newUser")
                        .about("Create a new user")
                        .arg(
                            Arg::with_name("username")
                                .help("The username to create")
                                .required(true),
                        )
                        .arg(
                            Arg::with_name("hasroot")
                                .help("If the user should have root privileges")
                                .required(true),
                        )
                        .arg(
                            Arg::with_name("password")
                                .help("The password to set. NOTE: Takes hashed password, use `openssl passwd -1 <password>` to generate the hash.")
                                .required(true),
                        )
                    )
                .subcommand(
                    SubCommand::with_name("rootPass")
                        .about("Set the root password")
                        .arg(
                            Arg::with_name("rootPass")
                                .help("The password to set. NOTE: Takes hashed password, use `openssl passwd -1 <password>` to generate the hash._")
                                .required(true),
                        ),
                )
        )
        .subcommand(
            SubCommand::with_name("nix")
                .about("Install nix"),
        )
        .subcommand(
            SubCommand::with_name("config")
                .about("read a jade installation config")
                .arg(
                    Arg::with_name("config")
                        .help("The config to read")
                        .required(true),
                ),
            )
        .subcommand(
            SubCommand::with_name("desktops")
                .about("Graphical stuff (Desktop environment and Display Manager)")
                .arg(
                    Arg::with_name("desktopsetup")
                        .help("The desktop setup to use")
                        .required(true),
                ),
    ).get_matches();

    if let Some(app) = app.subcommand_matches("partition") {
        partition::partition(
            app.value_of("device").unwrap(),
            app.value_of("mode").unwrap(),
            app.is_present("efi"),
        );
    } else if let Some(app) = app.subcommand_matches("locale") {
        locale::set_locale(
            app.values_of("locales")
                .unwrap()
                .collect::<Vec<&str>>()
                .join(" "),
        );
        locale::set_keyboard(app.value_of("keyboard").unwrap());
        locale::set_timezone(app.value_of("timezone").unwrap());
    } else if let Some(app) = app.subcommand_matches("networking") {
        if app.is_present("ipv6") {
            network::create_hosts();
            network::enable_ipv6()
        } else {
            network::create_hosts()
        }
        network::set_hostname(app.value_of("hostname").unwrap())
    } else if let Some(app) = app.subcommand_matches("users") {
        if let Some(app) = app.subcommand_matches("newUser") {
            users::new_user(
                app.value_of("username").unwrap(),
                app.value_of("hasroot").unwrap().parse::<bool>().unwrap(),
                app.value_of("password").unwrap(),
            );
        } else if let Some(app) = app.subcommand_matches("rootPass") {
            users::root_pass(app.value_of("rootPass").unwrap());
        }
    } else if let Some(app) = app.subcommand_matches("desktops") {
        desktops::choose_pkgs(app.value_of("desktopsetup").unwrap());
    } else if let Some(app) = app.subcommand_matches("bootloader") {
        if let Some(app) = app.subcommand_matches("grub-efi") {
            base::install_bootloader_efi(app.value_of("efidir").unwrap());
        } else if let Some(app) = app.subcommand_matches("grub-legacy") {
            base::install_bootloader_legacy(app.value_of("device").unwrap());
        }
    } else if app.subcommand_matches("install-base").is_some() {
        base::install_base_packages();
    } else if app.subcommand_matches("genfstab").is_some() {
        base::genfstab();
    } else if app.subcommand_matches("setup-timeshift").is_some() {
        base::setup_timeshift();
    } else if app.subcommand_matches("nix").is_some() {
        base::install_homemgr();
    } else if let Some(app) = app.subcommand_matches("config") {
        crate::internal::config::read_config(app.value_of("config").unwrap());
    } else {
        println!("{}", app.usage());
    }
}