czkawka/czkawka_cli/src/main.rs

use czkawka_core::*;
use std::{env, process};

fn main() {
    // Parse argument
    let all_arguments: Vec<String> = env::args().skip(1).collect(); // Not need to check program name
    let mut commands_arguments: Vec<String> = Vec::new();

    #[cfg(debug_assertions)]
    println!("{:?}", all_arguments);

    // No arguments, so we print help to allow user to learn more about program
    if all_arguments.is_empty() {
        print_help();
        process::exit(0);
    }

    // Assigning commands with arguments
    let mut arguments: Vec<ArgumentsPair> = Vec::new();

    for argument in all_arguments {
        if argument.starts_with("--") {
            commands_arguments.push(argument);
        } else if argument.starts_with('-') {
            let a: ArgumentsPair = ArgumentsPair { command: argument, argument: Option::None };
            arguments.push(a);
        } else {
            if arguments.is_empty() {
                println!("FATAL ERROR: Trying to use {} without any arguments(like -i -e -delete)", argument);
                process::exit(1);
            }
            if arguments[arguments.len() - 1].argument != Option::None {
                println!(
                    "FATAL ERROR: Trying set second parameter \"{}\" for \"{}\" which already have this parameter \"{}\" ",
                    argument,
                    arguments[arguments.len() - 1].command,
                    arguments[arguments.len() - 1].argument.as_ref().unwrap()
                ); // This may be changed in future to support 2 or more attributes with space
                process::exit(1);
            }
            let last_element = arguments.len() - 1;
            arguments[last_element].argument = Option::from(argument);
        }
    }

    #[cfg(debug_assertions)]
    for a in &arguments {
        println!(
            "Argument number {} - {}",
            a.command,
            match &a.argument {
                Some(t) => t.clone(),
                None => "NO_ARGUMENT".to_string(),
            }
        );
    }

    if commands_arguments.is_empty() {
        println! {"FATAL ERROR: Missing type of app which you want to run, please read help for more info."};
        process::exit(0);
    }
    match commands_arguments[0].as_ref() {
        "--d" => {
            let mut df = duplicate::DuplicateFinder::new();

            if ArgumentsPair::has_command(&arguments, "-i") {
                df.set_include_directory(ArgumentsPair::get_argument(&arguments, "-i", false));
            } else {
                println!("FATAL ERROR: Parameter -i with set of included files is required.");
                process::exit(1);
            }
            if ArgumentsPair::has_command(&arguments, "-e") {
                df.set_exclude_directory(ArgumentsPair::get_argument(&arguments, "-e", false));
            }

            if ArgumentsPair::has_command(&arguments, "-s") {
                let min_size = match ArgumentsPair::get_argument(&arguments, "-s", false).parse::<u64>() {
                    Ok(t) => {
                        if t == 0 {
                            println!("ERROR: Minimum file size must be at least 1 byte.");
                            1
                        } else {
                            t
                        }
                    }
                    Err(_) => {
                        println!("FATAL ERROR: \"{}\" is not valid file size(allowed range <1,u64::max>)", ArgumentsPair::get_argument(&arguments, "-s", false));
                        process::exit(1);
                    }
                };
                df.set_min_file_size(min_size);
            }

            if ArgumentsPair::has_command(&arguments, "-x") {
                df.set_allowed_extensions(ArgumentsPair::get_argument(&arguments, "-x", false));
            }
            if ArgumentsPair::has_command(&arguments, "-k") {
                df.set_excluded_items(ArgumentsPair::get_argument(&arguments, "-k", false));
            }

            if ArgumentsPair::has_command(&arguments, "-o") {
                df.set_recursive_search(false);
            }
            df.set_check_method(duplicate::CheckingMethod::Hash); // Default
            if ArgumentsPair::has_command(&arguments, "-l") {
                let argument_name = ArgumentsPair::get_argument(&arguments, "-l", false).to_lowercase();
                if argument_name == "size" {
                    df.set_check_method(duplicate::CheckingMethod::Size);
                } else if argument_name == "hash" {
                    df.set_check_method(duplicate::CheckingMethod::Hash);
                } else {
                    println!("-l can only have values hash or size");
                    process::exit(1);
                }
            }

            df.set_delete_method(duplicate::DeleteMethod::None);
            if ArgumentsPair::has_command(&arguments, "-delete") {
                let argument_name = ArgumentsPair::get_argument(&arguments, "-delete", true).to_lowercase();
                if argument_name == "aen" {
                    df.set_delete_method(duplicate::DeleteMethod::AllExceptNewest);
                } else if argument_name == "aeo" {
                    df.set_delete_method(duplicate::DeleteMethod::AllExceptOldest);
                } else if argument_name == "on" {
                    df.set_delete_method(duplicate::DeleteMethod::OneNewest);
                } else if argument_name == "oo" {
                    df.set_delete_method(duplicate::DeleteMethod::OneOldest);
                } else if argument_name == "" {
                    // Default
                    df.set_delete_method(duplicate::DeleteMethod::AllExceptOldest);
                } else {
                    println!(
                        "Invalid argument {} for command -delete, available arguments - aen(All except newest one), aeo(All except oldest one), on(Only one newest), oo(Only one oldest)",
                        argument_name
                    );
                    process::exit(1);
                }
            }

            df.find_duplicates();

            #[allow(clippy::collapsible_if)]
            if ArgumentsPair::has_command(&arguments, "-f") {
                if !df.save_results_to_file(&ArgumentsPair::get_argument(&arguments, "-f", false)) {
                    df.get_text_messages().print_messages();
                    process::exit(1);
                }
            }

            #[cfg(not(debug_assertions))] // This will show too much probably unnecessary data to debug, comment line only if needed
            df.print_duplicated_entries();

            df.get_text_messages().print_messages();
        }
        "--h" | "--help" => {
            print_help();
        }
        "--e" => {
            let mut ef = empty_folder::EmptyFolder::new();

            if ArgumentsPair::has_command(&arguments, "-i") {
                ef.set_include_directory(ArgumentsPair::get_argument(&arguments, "-i", false));
            } else {
                println!("FATAL ERROR: Parameter -i with set of included files is required.");
                process::exit(1);
            }

            if ArgumentsPair::has_command(&arguments, "-delete") {
                ef.set_delete_folder(true);
            }

            ef.find_empty_folders();

            #[allow(clippy::collapsible_if)]
            if ArgumentsPair::has_command(&arguments, "-f") {
                if !ef.save_results_to_file(&ArgumentsPair::get_argument(&arguments, "-f", false)) {
                    ef.get_text_messages().print_messages();
                    process::exit(1);
                }
            }

            #[cfg(not(debug_assertions))] // This will show too much probably unnecessary data to debug, comment line only if needed
            ef.print_empty_folders();
        }
        "--b" => {
            let mut bf = big_file::BigFile::new();

            if ArgumentsPair::has_command(&arguments, "-i") {
                bf.set_include_directory(ArgumentsPair::get_argument(&arguments, "-i", false));
            } else {
                println!("FATAL ERROR: Parameter -i with set of included files is required.");
                process::exit(1);
            }
            if ArgumentsPair::has_command(&arguments, "-e") {
                bf.set_exclude_directory(ArgumentsPair::get_argument(&arguments, "-e", false));
            }

            if ArgumentsPair::has_command(&arguments, "-s") {
                let number_of_files = match ArgumentsPair::get_argument(&arguments, "-s", false).parse::<usize>() {
                    Ok(t) => {
                        if t == 0 {
                            println!("ERROR: Minimum one biggest file must be showed..");
                            1
                        } else {
                            t
                        }
                    }
                    Err(_) => {
                        println!("FATAL ERROR: \"{}\" is not valid number of files to show(allowed range <1,usize::max>)", ArgumentsPair::get_argument(&arguments, "-s", false));
                        process::exit(1);
                    }
                };
                bf.set_number_of_files_to_check(number_of_files);
            }

            if ArgumentsPair::has_command(&arguments, "-x") {
                bf.set_allowed_extensions(ArgumentsPair::get_argument(&arguments, "-x", false));
            }
            if ArgumentsPair::has_command(&arguments, "-k") {
                bf.set_excluded_items(ArgumentsPair::get_argument(&arguments, "-k", false));
            }

            if ArgumentsPair::has_command(&arguments, "-o") {
                bf.set_recursive_search(false);
            }

            bf.find_big_files();

            #[allow(clippy::collapsible_if)]
            if ArgumentsPair::has_command(&arguments, "-f") {
                if !bf.save_results_to_file(&ArgumentsPair::get_argument(&arguments, "-f", false)) {
                    bf.get_text_messages().print_messages();
                    process::exit(1);
                }
            }

            #[cfg(not(debug_assertions))] // This will show too much probably unnecessary data to debug, comment line only if needed
            bf.print_duplicated_entries();

            bf.get_text_messages().print_messages();
        }
        "--version" | "v" => {
            println!("Czkawka CLI {}", CZKAWKA_VERSION);
            process::exit(0);
        }
        argum => {
            println!("FATAL ERROR: \"{}\" argument is not supported, check help for more info.", argum);
            process::exit(1);
        }
    };
}

fn print_help() {
    println!(
        r###"
Usage of Czkawka:

## Main arguments:
  --h / --help - prints help, also works without any arguments

  Usage example:
    czkawka --help
    czkawka


  --d <-i directory_to_search> [-e exclude_directories = ""] [-k excluded_items = ""] [-s min_size = 1024] [-x allowed_extension = ""] [-l type_of_search = "hash"] [-o] [-f file_to_save = "results.txt"] [-delete = "aeo"] - search for duplicates files
    -i directory_to_search - list of directories which should will be searched like /home/rafal
    -e exclude_directories - list of directories which will be excluded from search.
    -k excluded_items - list of excluded items which contains * wildcard(may be slow)
    -o - this options prevents from recursive check of folders
    -f file_to_save - saves results to file
    -s min_size - minimum size of checked files in bytes, assigning bigger value may speed up searching.
    -x allowed_extension - list of checked extension, e.g. "jpg,mp4" will allow to check "book.jpg" and "car.mp4" but not roman.png. There are also helpful macros which allow to easy use a typcal extension like IMAGE("jpg,kra,gif,png,bmp,tiff,webp,hdr,svg") or TEXT("txt,doc,docx,odt,rtf")
    -l type_of_search - allows to use fastest which takes into account only size, and more accurate which check if file contnet is same(hashes).
    -delete - delete found files, by default remove all except the most oldest one, it can take arguments: aen(All except newest one), aeo(All except oldest one), on(Only one newest), oo(Only one oldest)

  Usage example:
    czkawka --d -i "/home/rafal/,/home/szczekacz" -e "/home/rafal/Pulpit,/home/rafal/Obrazy" -s 25 -x "7z,rar,IMAGE" -l "size" -delete
    czkawka --d -i "/etc/,/mnt/Miecz" -s 1000 -x "VIDEO" -l "hash" -o
    czkawka --d -i "/var/" -k "/var/l*b/,/var/lo*,*tmp"
    czkawka --d -i "/etc/" -delete "aeo"


  --e <-i directory_to_search> [-e exclude_directories = ""] [-o] [-f file_to_save] [-delete] - option to find and delete empty folders
    -i directory_to_search - list of directories which should will be searched like /home/rafal
    -e exclude_directories - list of directories which will be excluded from search.
    -f file_to_save - saves results to file
    -delete - delete found empty folders

  Usage example:
    czkawka --e -i "/home/rafal/rr, /home/gateway" -e "/home/rafal/rr/2" -delete

  --b  <-i directory_to_search> [-e exclude_directories = ""] [-k excluded_items = ""] [-s number_of_files = 50] [-x allowed_extension = ""] [-o] [-f file_to_save = "results.txt"]
    -i directory_to_search - list of directories which should will be searched like /home/rafal
    -e exclude_directories - list of directories which will be excluded from search.
    -k excluded_items - list of excluded items which contains * wildcard(may be slow)
    -o - this options prevents from recursive check of folders
    -f file_to_save - saves results to file
    -s number_of_files - number of showed the biggest files.
    -x allowed_extension - list of checked extension, e.g. "jpg,mp4" will allow to check "book.jpg" and "car.mp4" but not roman.png. There are also helpful macros which allow to easy use a typcal extension like IMAGE("jpg,kra,gif,png,bmp,tiff,webp,hdr,svg") or TEXT("txt,doc,docx,odt,rtf")

  --version / --v - prints program name and version

    "###
    );
}

struct ArgumentsPair {
    command: String,
    argument: Option<String>,
}

impl ArgumentsPair {
    pub fn has_command(ar: &[ArgumentsPair], command: &str) -> bool {
        for a in ar {
            if a.command == command {
                return true;
            }
        }
        false
    }
    pub fn get_argument(ar: &[ArgumentsPair], command: &str, can_be_empty: bool) -> String {
        for a in ar {
            if a.command == command {
                if !can_be_empty && a.argument == Option::None {
                    println!("FATAL ERROR: {} commands should have argument passed", command);
                    process::exit(1);
                }
                return match &a.argument {
                    Some(t) => t.clone(),
                    None => "".to_string(),
                };
            }
        }
        panic!("INTERNAL ERROR: Get argument should always return value");
    }
}