CodeMirror/czkawka
1
0
Fork 0
mirror of synced 2024-05-07 05:53:03 +12:00
Code Issues Projects Releases Wiki Activity

Add saving to empty folders finder, remove exclude directories from it and put warnings to another function

Browse source
This commit is contained in:
Rafał Mikrut 2020-09-18 10:23:49 +02:00
parent cf127d5c36
commit 2b9c0d61df
3 changed files with 149 additions and 219 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
czkawka_cli/src/main.rs
View file

@ -139,11 +139,15 @@ fn main() {
df.print_duplicated_entries(); df.print_duplicated_entries();
#[allow(clippy::collapsible_if)]
if ArgumentsPair::has_command(&arguments, "-f") { if ArgumentsPair::has_command(&arguments, "-f") {
df.save_results_to_file(&ArgumentsPair::get_argument(&arguments, "-f", false)); if !df.save_results_to_file(&ArgumentsPair::get_argument(&arguments, "-f", false)) {
df.get_text_messages().print_messages();
process::exit(1);
}
} }
df.get_messages().print_messages(); df.get_text_messages().print_messages();
//print_information(df.get_information()); //print_information(df.get_information());
} }
"--h" | "--help" => { "--h" | "--help" => {
@ -158,8 +162,12 @@ fn main() {
println!("FATAL ERROR: Parameter -i with set of included files is required."); println!("FATAL ERROR: Parameter -i with set of included files is required.");
process::exit(1); process::exit(1);
} }
if ArgumentsPair::has_command(&arguments, "-e") { #[allow(clippy::collapsible_if)]
ef.set_exclude_directory(ArgumentsPair::get_argument(&arguments, "-e", false)); 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);
}
} }
if ArgumentsPair::has_command(&arguments, "-delete") { if ArgumentsPair::has_command(&arguments, "-delete") {
@ -206,9 +214,7 @@ Usage of Czkawka:
--e <-i directory_to_search> [-e exclude_directories = ""] [-o] [-f file_to_save] [-delete] - option to find and delete empty folders --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 -i directory_to_search - list of directories which should will be searched like /home/rafal
-o - this options prevents from recursive check of folders
-e exclude_directories - list of directories which will be excluded from search. -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)
-f file_to_save - saves results to file -f file_to_save - saves results to file
-delete - delete found empty folders -delete - delete found empty folders
czkawka --e -i "/home/rafal/rr, /home/gateway" -e "/home/rafal/rr/2" -delete czkawka --e -i "/home/rafal/rr, /home/gateway" -e "/home/rafal/rr/2" -delete

141
czkawka_core/src/duplicate.rs
View file

@ -76,7 +76,7 @@ impl Default for Info {
/// Struct with required information's to work /// Struct with required information's to work
pub struct DuplicateFinder { pub struct DuplicateFinder {
messages: Messages, text_messages: Messages,
information: Info, information: Info,
files_with_identical_size: BTreeMap<u64, Vec<FileEntry>>, files_with_identical_size: BTreeMap<u64, Vec<FileEntry>>,
files_with_identical_hashes: BTreeMap<u64, Vec<Vec<FileEntry>>>, files_with_identical_hashes: BTreeMap<u64, Vec<Vec<FileEntry>>>,
@ -93,7 +93,7 @@ pub struct DuplicateFinder {
impl DuplicateFinder { impl DuplicateFinder {
pub fn new() -> DuplicateFinder { pub fn new() -> DuplicateFinder {
DuplicateFinder { DuplicateFinder {
messages: Default::default(), text_messages: Default::default(),
information: Info::new(), information: Info::new(),
files_with_identical_size: Default::default(), files_with_identical_size: Default::default(),
files_with_identical_hashes: Default::default(), files_with_identical_hashes: Default::default(),
@ -108,8 +108,8 @@ impl DuplicateFinder {
} }
} }
pub fn get_messages(&self) -> &Messages { pub fn get_text_messages(&self) -> &Messages {
&self.messages &self.text_messages
} }
pub fn get_information(&self) -> &Info { pub fn get_information(&self) -> &Info {
&self.information &self.information
@ -163,7 +163,7 @@ impl DuplicateFinder {
continue; continue;
} }
if !expression.contains('*') { if !expression.contains('*') {
self.messages.warnings.push("Excluded Items Warning: Wildcard * is required in expression, ignoring ".to_string() + expression.as_str()); self.text_messages.warnings.push("Excluded Items Warning: Wildcard * is required in expression, ignoring ".to_string() + expression.as_str());
continue; continue;
} }
@ -175,7 +175,7 @@ impl DuplicateFinder {
pub fn set_allowed_extensions(&mut self, mut allowed_extensions: String) { pub fn set_allowed_extensions(&mut self, mut allowed_extensions: String) {
let start_time: SystemTime = SystemTime::now(); let start_time: SystemTime = SystemTime::now();
if allowed_extensions.is_empty() { if allowed_extensions.is_empty() {
self.messages.messages.push("No allowed extension was provided, so all are allowed".to_string()); self.text_messages.messages.push("No allowed extension was provided, so all are allowed".to_string());
return; return;
} }
allowed_extensions = allowed_extensions.replace("IMAGE", "jpg,kra,gif,png,bmp,tiff,webp,hdr,svg"); allowed_extensions = allowed_extensions.replace("IMAGE", "jpg,kra,gif,png,bmp,tiff,webp,hdr,svg");
@ -194,7 +194,7 @@ impl DuplicateFinder {
} }
if extension[1..].contains('.') { if extension[1..].contains('.') {
self.messages.warnings.push(".".to_string() + extension.as_str() + " is not valid extension(valid extension doesn't have dot inside)"); self.text_messages.warnings.push(".".to_string() + extension.as_str() + " is not valid extension(valid extension doesn't have dot inside)");
continue; continue;
} }
@ -204,7 +204,7 @@ impl DuplicateFinder {
} }
if self.allowed_extensions.is_empty() { if self.allowed_extensions.is_empty() {
self.messages.messages.push("No valid extensions were provided, so allowing all extensions by default.".to_string()); self.text_messages.messages.push("No valid extensions were provided, so allowing all extensions by default.".to_string());
} }
Common::print_time(start_time, SystemTime::now(), "set_allowed_extensions".to_string()); Common::print_time(start_time, SystemTime::now(), "set_allowed_extensions".to_string());
} }
@ -212,7 +212,7 @@ impl DuplicateFinder {
let start_time: SystemTime = SystemTime::now(); let start_time: SystemTime = SystemTime::now();
if include_directory.is_empty() { if include_directory.is_empty() {
self.messages.errors.push("At least one directory must be provided".to_string()); self.text_messages.errors.push("At least one directory must be provided".to_string());
return false; return false;
} }
@ -227,19 +227,19 @@ impl DuplicateFinder {
continue; continue;
} }
if directory.contains('*') { if directory.contains('*') {
self.messages.warnings.push("Include Directory Warning: Wildcards in path are not supported, ignoring ".to_string() + directory.as_str()); self.text_messages.warnings.push("Include Directory Warning: Wildcards in path are not supported, ignoring ".to_string() + directory.as_str());
continue; continue;
} }
if !directory.starts_with('/') { if !directory.starts_with('/') {
self.messages.warnings.push("Include Directory Warning: Relative path are not supported, ignoring ".to_string() + directory.as_str()); self.text_messages.warnings.push("Include Directory Warning: Relative path are not supported, ignoring ".to_string() + directory.as_str());
continue; continue;
} }
if !Path::new(&directory).exists() { if !Path::new(&directory).exists() {
self.messages.warnings.push("Include Directory Warning: Provided folder path must exits, ignoring ".to_string() + directory.as_str()); self.text_messages.warnings.push("Include Directory Warning: Provided folder path must exits, ignoring ".to_string() + directory.as_str());
continue; continue;
} }
if !Path::new(&directory).is_dir() { if !Path::new(&directory).is_dir() {
self.messages.warnings.push("Include Directory Warning: Provided path must point at the directory, ignoring ".to_string() + directory.as_str()); self.text_messages.warnings.push("Include Directory Warning: Provided path must point at the directory, ignoring ".to_string() + directory.as_str());
continue; continue;
} }
@ -252,7 +252,7 @@ impl DuplicateFinder {
} }
if checked_directories.is_empty() { if checked_directories.is_empty() {
self.messages.errors.push("Include Directory ERROR: Not found even one correct path to include which is required.".to_string()); self.text_messages.errors.push("Include Directory ERROR: Not found even one correct path to include which is required.".to_string());
return false; return false;
} }
@ -279,23 +279,23 @@ impl DuplicateFinder {
continue; continue;
} }
if directory == "/" { if directory == "/" {
self.messages.errors.push("Exclude Directory ERROR: Excluding / is pointless, because it means that no files will be scanned.".to_string()); self.text_messages.errors.push("Exclude Directory ERROR: Excluding / is pointless, because it means that no files will be scanned.".to_string());
break; break;
} }
if directory.contains('*') { if directory.contains('*') {
self.messages.warnings.push("Exclude Directory Warning: Wildcards in path are not supported, ignoring ".to_string() + directory.as_str()); self.text_messages.warnings.push("Exclude Directory Warning: Wildcards in path are not supported, ignoring ".to_string() + directory.as_str());
continue; continue;
} }
if !directory.starts_with('/') { if !directory.starts_with('/') {
self.messages.warnings.push("Exclude Directory Warning: Relative path are not supported, ignoring ".to_string() + directory.as_str()); self.text_messages.warnings.push("Exclude Directory Warning: Relative path are not supported, ignoring ".to_string() + directory.as_str());
continue; continue;
} }
if !Path::new(&directory).exists() { if !Path::new(&directory).exists() {
self.messages.warnings.push("Exclude Directory Warning: Provided folder path must exits, ignoring ".to_string() + directory.as_str()); self.text_messages.warnings.push("Exclude Directory Warning: Provided folder path must exits, ignoring ".to_string() + directory.as_str());
continue; continue;
} }
if !Path::new(&directory).is_dir() { if !Path::new(&directory).is_dir() {
self.messages.warnings.push("Exclude Directory Warning: Provided path must point at the directory, ignoring ".to_string() + directory.as_str()); self.text_messages.warnings.push("Exclude Directory Warning: Provided path must point at the directory, ignoring ".to_string() + directory.as_str());
continue; continue;
} }
@ -330,7 +330,7 @@ impl DuplicateFinder {
let read_dir = match fs::read_dir(&current_folder) { let read_dir = match fs::read_dir(&current_folder) {
Ok(t) => t, Ok(t) => t,
Err(_) => { Err(_) => {
self.messages.warnings.push("Cannot open dir ".to_string() + current_folder.as_str()); self.text_messages.warnings.push("Cannot open dir ".to_string() + current_folder.as_str());
continue; continue;
} // Permissions denied } // Permissions denied
}; };
@ -338,21 +338,21 @@ impl DuplicateFinder {
let entry_data = match entry { let entry_data = match entry {
Ok(t) => t, Ok(t) => t,
Err(_) => { Err(_) => {
self.messages.warnings.push("Cannot read entry in dir ".to_string() + current_folder.as_str()); self.text_messages.warnings.push("Cannot read entry in dir ".to_string() + current_folder.as_str());
continue; continue;
} //Permissions denied } //Permissions denied
}; };
let metadata: Metadata = match entry_data.metadata() { let metadata: Metadata = match entry_data.metadata() {
Ok(t) => t, Ok(t) => t,
Err(_) => { Err(_) => {
self.messages.warnings.push("Cannot read metadata in dir ".to_string() + current_folder.as_str()); self.text_messages.warnings.push("Cannot read metadata in dir ".to_string() + current_folder.as_str());
continue; continue;
} //Permissions denied } //Permissions denied
}; };
if metadata.is_dir() { if metadata.is_dir() {
self.information.number_of_checked_folders += 1; self.information.number_of_checked_folders += 1;
// if entry_data.file_name().into_string().is_err() { // Probably this can be removed, if crash still will be happens, then uncomment this line // if entry_data.file_name().into_string().is_err() { // Probably this can be removed, if crash still will be happens, then uncomment this line
// self.messages.warnings.push("Cannot read folder name in dir ".to_string() + current_folder.as_str()); // self.text_messages.warnings.push("Cannot read folder name in dir ".to_string() + current_folder.as_str());
// continue; // Permissions denied // continue; // Permissions denied
// } // }
@ -422,14 +422,14 @@ impl DuplicateFinder {
created_date: match metadata.created() { created_date: match metadata.created() {
Ok(t) => t, Ok(t) => t,
Err(_) => { Err(_) => {
self.messages.warnings.push("Unable to get creation date from file ".to_string() + current_file_name.as_str()); self.text_messages.warnings.push("Unable to get creation date from file ".to_string() + current_file_name.as_str());
SystemTime::now() SystemTime::now()
} // Permissions Denied } // Permissions Denied
}, },
modified_date: match metadata.modified() { modified_date: match metadata.modified() {
Ok(t) => t, Ok(t) => t,
Err(_) => { Err(_) => {
self.messages.warnings.push("Unable to get modification date from file ".to_string() + current_file_name.as_str()); self.text_messages.warnings.push("Unable to get modification date from file ".to_string() + current_file_name.as_str());
SystemTime::now() SystemTime::now()
} // Permissions Denied } // Permissions Denied
}, },
@ -466,7 +466,7 @@ impl DuplicateFinder {
Common::print_time(start_time, SystemTime::now(), "check_files_size".to_string()); Common::print_time(start_time, SystemTime::now(), "check_files_size".to_string());
} }
pub fn save_results_to_file(&mut self, file_name: &str) { pub fn save_results_to_file(&mut self, file_name: &str) -> bool {
let start_time: SystemTime = SystemTime::now(); let start_time: SystemTime = SystemTime::now();
let file_name: String = match file_name { let file_name: String = match file_name {
"" => "results.txt".to_string(), "" => "results.txt".to_string(),
@ -476,16 +476,16 @@ impl DuplicateFinder {
let mut file = match File::create(&file_name) { let mut file = match File::create(&file_name) {
Ok(t) => t, Ok(t) => t,
Err(_) => { Err(_) => {
self.messages.errors.push("Failed to create file ".to_string() + file_name.as_str()); self.text_messages.errors.push("Failed to create file ".to_string() + file_name.as_str());
return; return false;
} }
}; };
match file.write_all(b"Results of searching\n\n") { match file.write_all(b"Results of searching\n\n") {
Ok(_) => (), Ok(_) => (),
Err(_) => { Err(_) => {
self.messages.errors.push("Failed to save results to file ".to_string() + file_name.as_str()); self.text_messages.errors.push("Failed to save results to file ".to_string() + file_name.as_str());
return; return false;
} }
} }
@ -512,35 +512,38 @@ impl DuplicateFinder {
file.write_all((file_entry.path.clone() + "\n").as_bytes()).unwrap(); file.write_all((file_entry.path.clone() + "\n").as_bytes()).unwrap();
} }
} }
}
if !self.files_with_identical_hashes.is_empty() { if !self.files_with_identical_hashes.is_empty() {
file.write_all(b"-------------------------------------------------Files with same hashes-------------------------------------------------\n").unwrap(); file.write_all(b"-------------------------------------------------Files with same hashes-------------------------------------------------\n").unwrap();
file.write_all( file.write_all(
("Found ".to_string() ("Found ".to_string()
+ self.information.number_of_duplicated_files_by_hash.to_string().as_str() + self.information.number_of_duplicated_files_by_hash.to_string().as_str()
+ " duplicated files which in " + " duplicated files which in "
+ self.information.number_of_groups_by_hash.to_string().as_str() + self.information.number_of_groups_by_hash.to_string().as_str()
+ " groups which takes " + " groups which takes "
+ self.information.lost_space_by_hash.file_size(options::BINARY).unwrap().as_str() + self.information.lost_space_by_hash.file_size(options::BINARY).unwrap().as_str()
+ ".\n") + ".\n")
.as_bytes(), .as_bytes(),
) )
.unwrap(); .unwrap();
for (size, files) in self.files_with_identical_hashes.iter().rev() { for (size, files) in self.files_with_identical_hashes.iter().rev() {
for vector in files { for vector in files {
file.write_all(b"\n---- Size ").unwrap(); file.write_all(b"\n---- Size ").unwrap();
file.write_all(size.file_size(options::BINARY).unwrap().as_bytes()).unwrap(); file.write_all(size.file_size(options::BINARY).unwrap().as_bytes()).unwrap();
file.write_all((" (".to_string() + size.to_string().as_str() + ")").as_bytes()).unwrap(); file.write_all((" (".to_string() + size.to_string().as_str() + ")").as_bytes()).unwrap();
file.write_all((" - ".to_string() + vector.len().to_string().as_str() + " files").as_bytes()).unwrap(); file.write_all((" - ".to_string() + vector.len().to_string().as_str() + " files").as_bytes()).unwrap();
file.write_all(b"\n").unwrap(); file.write_all(b"\n").unwrap();
for file_entry in vector { for file_entry in vector {
file.write_all((file_entry.path.clone() + "\n").as_bytes()).unwrap(); file.write_all((file_entry.path.clone() + "\n").as_bytes()).unwrap();
}
} }
} }
} }
} else {
file.write_all(b"Not found any empty folders.").unwrap();
} }
Common::print_time(start_time, SystemTime::now(), "save_results_to_file".to_string()); Common::print_time(start_time, SystemTime::now(), "save_results_to_file".to_string());
true
} }
/// Should be slower than checking in different ways, but still needs to be checked /// Should be slower than checking in different ways, but still needs to be checked
@ -556,7 +559,7 @@ impl DuplicateFinder {
file_handler = match File::open(&file_entry.1.path) { file_handler = match File::open(&file_entry.1.path) {
Ok(t) => t, Ok(t) => t,
Err(_) => { Err(_) => {
self.messages.warnings.push("Unable to check hash of file ".to_string() + file_entry.1.path.as_str()); self.text_messages.warnings.push("Unable to check hash of file ".to_string() + file_entry.1.path.as_str());
continue; continue;
} }
}; };
@ -569,7 +572,7 @@ impl DuplicateFinder {
let n = match file_handler.read(&mut buffer) { let n = match file_handler.read(&mut buffer) {
Ok(t) => t, Ok(t) => t,
Err(_) => { Err(_) => {
self.messages.warnings.push("Error happened when checking hash of file ".to_string() + file_entry.1.path.as_str()); self.text_messages.warnings.push("Error happened when checking hash of file ".to_string() + file_entry.1.path.as_str());
error_reading_file = true; error_reading_file = true;
break; break;
} }
@ -615,18 +618,28 @@ impl DuplicateFinder {
println!("---------------DEBUG PRINT---------------"); println!("---------------DEBUG PRINT---------------");
println!("### Information's"); println!("### Information's");
println!("Errors size - {}", self.messages.errors.len()); println!("Errors size - {}", self.text_messages.errors.len());
println!("Warnings size - {}", self.messages.warnings.len()); println!("Warnings size - {}", self.text_messages.warnings.len());
println!("Messages size - {}", self.messages.messages.len()); println!("Messages size - {}", self.text_messages.messages.len());
println!("Number of checked files - {}", self.information.number_of_checked_files); println!("Number of checked files - {}", self.information.number_of_checked_files);
println!("Number of checked folders - {}", self.information.number_of_checked_folders); println!("Number of checked folders - {}", self.information.number_of_checked_folders);
println!("Number of ignored files - {}", self.information.number_of_ignored_files); println!("Number of ignored files - {}", self.information.number_of_ignored_files);
println!("Number of ignored things(like symbolic links) - {}", self.information.number_of_ignored_things); println!("Number of ignored things(like symbolic links) - {}", self.information.number_of_ignored_things);
println!("Number of duplicated files by size(in groups) - {} ({})", self.information.number_of_duplicated_files_by_size, self.information.number_of_groups_by_size); println!(
println!("Number of duplicated files by hash(in groups) - {} ({})", self.information.number_of_duplicated_files_by_hash, self.information.number_of_groups_by_hash); "Number of duplicated files by size(in groups) - {} ({})",
self.information.number_of_duplicated_files_by_size, self.information.number_of_groups_by_size
);
println!(
"Number of duplicated files by hash(in groups) - {} ({})",
self.information.number_of_duplicated_files_by_hash, self.information.number_of_groups_by_hash
);
println!("Lost space by size - {} ({} bytes)", self.information.lost_space_by_size.file_size(options::BINARY).unwrap(), self.information.lost_space_by_size); println!("Lost space by size - {} ({} bytes)", self.information.lost_space_by_size.file_size(options::BINARY).unwrap(), self.information.lost_space_by_size);
println!("Lost space by hash - {} ({} bytes)", self.information.lost_space_by_hash.file_size(options::BINARY).unwrap(), self.information.lost_space_by_hash); println!("Lost space by hash - {} ({} bytes)", self.information.lost_space_by_hash.file_size(options::BINARY).unwrap(), self.information.lost_space_by_hash);
println!("Gained space by removing duplicated entries - {} ({} bytes)", self.information.gained_space.file_size(options::BINARY).unwrap(), self.information.gained_space); println!(
"Gained space by removing duplicated entries - {} ({} bytes)",
self.information.gained_space.file_size(options::BINARY).unwrap(),
self.information.gained_space
);
println!("Number of removed files - {}", self.information.number_of_removed_files); println!("Number of removed files - {}", self.information.number_of_removed_files);
println!("Number of failed to remove files - {}", self.information.number_of_failed_to_remove_files); println!("Number of failed to remove files - {}", self.information.number_of_failed_to_remove_files);
@ -811,7 +824,7 @@ impl DuplicateFinder {
// optimized_excluded = Vec::<String>::new(); // optimized_excluded = Vec::<String>::new();
if self.included_directories.is_empty() { if self.included_directories.is_empty() {
self.messages.errors.push("Optimize Directories ERROR: Excluded directories overlaps all included directories.".to_string()); self.text_messages.errors.push("Optimize Directories ERROR: Excluded directories overlaps all included directories.".to_string());
return false; return false;
} }
@ -829,7 +842,7 @@ impl DuplicateFinder {
CheckingMethod::HASH => { CheckingMethod::HASH => {
for entry in &self.files_with_identical_hashes { for entry in &self.files_with_identical_hashes {
for vector in entry.1 { for vector in entry.1 {
let tuple: (u64, usize, usize) = delete_files(&vector, &self.delete_method, &mut self.messages.warnings); let tuple: (u64, usize, usize) = delete_files(&vector, &self.delete_method, &mut self.text_messages.warnings);
self.information.gained_space += tuple.0; self.information.gained_space += tuple.0;
self.information.number_of_removed_files += tuple.1; self.information.number_of_removed_files += tuple.1;
self.information.number_of_failed_to_remove_files += tuple.2; self.information.number_of_failed_to_remove_files += tuple.2;
@ -838,7 +851,7 @@ impl DuplicateFinder {
} }
CheckingMethod::SIZE => { CheckingMethod::SIZE => {
for entry in &self.files_with_identical_size { for entry in &self.files_with_identical_size {
let tuple: (u64, usize, usize) = delete_files(&entry.1, &self.delete_method, &mut self.messages.warnings); let tuple: (u64, usize, usize) = delete_files(&entry.1, &self.delete_method, &mut self.text_messages.warnings);
self.information.gained_space += tuple.0; self.information.gained_space += tuple.0;
self.information.number_of_removed_files += tuple.1; self.information.number_of_removed_files += tuple.1;
self.information.number_of_failed_to_remove_files += tuple.2; self.information.number_of_failed_to_remove_files += tuple.2;

209
czkawka_core/src/empty_folder.rs
View file

@ -1,6 +1,7 @@
use crate::common::{Common, Messages}; use crate::common::{Common, Messages};
use std::collections::HashMap; use std::collections::HashMap;
use std::fs::Metadata; use std::fs::{File, Metadata};
use std::io::Write;
use std::path::Path; use std::path::Path;
use std::time::SystemTime; use std::time::SystemTime;
use std::{fs, process}; use std::{fs, process};
@ -22,22 +23,23 @@ struct FolderEntry {
/// Struct to store most basics info about all folder /// Struct to store most basics info about all folder
pub struct EmptyFolder { pub struct EmptyFolder {
informations: Info, information: Info,
delete_folders: bool, delete_folders: bool,
messages: Messages, text_messages: Messages,
empty_folder_list: HashMap<String, FolderEntry>, empty_folder_list: HashMap<String, FolderEntry>, // Path, FolderEntry
excluded_directories: Vec<String>,
included_directories: Vec<String>, included_directories: Vec<String>,
} }
/// Info struck with helpful information's about results /// Info struck with helpful information's about results
pub struct Info { pub struct Info {
#[cfg(debug_assertions)]
number_of_checked_folders: usize, number_of_checked_folders: usize,
number_of_empty_folders: usize, number_of_empty_folders: usize,
} }
impl Info { impl Info {
pub fn new() -> Info { pub fn new() -> Info {
Info { Info {
#[cfg(debug_assertions)]
number_of_checked_folders: 0, number_of_checked_folders: 0,
number_of_empty_folders: 0, number_of_empty_folders: 0,
} }
@ -54,17 +56,16 @@ impl EmptyFolder {
/// New function providing basics values /// New function providing basics values
pub fn new() -> EmptyFolder { pub fn new() -> EmptyFolder {
EmptyFolder { EmptyFolder {
informations: Default::default(), information: Default::default(),
delete_folders: false, delete_folders: false,
messages: Default::default(), text_messages: Default::default(),
empty_folder_list: Default::default(), empty_folder_list: Default::default(),
excluded_directories: vec![],
included_directories: vec![], included_directories: vec![],
} }
} }
pub fn get_messages(&self) -> &Messages { pub fn get_text_messages(&self) -> &Messages {
&self.messages &self.text_messages
} }
/// Public function used by CLI to search for empty folders /// Public function used by CLI to search for empty folders
@ -83,6 +84,43 @@ impl EmptyFolder {
self.delete_folders = delete_folder; self.delete_folders = delete_folder;
} }
pub fn save_results_to_file(&mut self, file_name: &str) -> bool {
let start_time: SystemTime = SystemTime::now();
let file_name: String = match file_name {
"" => "results.txt".to_string(),
k => k.to_string(),
};
let mut file = match File::create(&file_name) {
Ok(t) => t,
Err(_) => {
self.text_messages.errors.push("Failed to create file ".to_string() + file_name.as_str());
return false;
}
};
match file.write_all(b"Results of searching\n") {
Ok(_) => (),
Err(_) => {
self.text_messages.errors.push("Failed to save results to file ".to_string() + file_name.as_str());
return false;
}
}
if !self.empty_folder_list.is_empty() {
file.write_all(b"-------------------------------------------------Empty folder list-------------------------------------------------\n").unwrap();
file.write_all(("Found ".to_string() + self.information.number_of_empty_folders.to_string().as_str() + " empty folders which in " + ".\n").as_bytes())
.unwrap();
for name in self.empty_folder_list.keys() {
file.write_all((name.clone() + "\n").as_bytes()).unwrap();
}
} else {
file.write_all(b"Not found any empty folders.").unwrap();
}
Common::print_time(start_time, SystemTime::now(), "save_results_to_file".to_string());
true
}
/// Clean directory tree /// Clean directory tree
/// If directory contains only 2 empty folders, then this directory should be removed instead two empty folders inside because it will produce another empty folder. /// If directory contains only 2 empty folders, then this directory should be removed instead two empty folders inside because it will produce another empty folder.
fn optimize_folders(&mut self) { fn optimize_folders(&mut self) {
@ -160,25 +198,16 @@ impl EmptyFolder {
}; };
// If child is dir, still folder may be considered as empty if all children are only directories. // If child is dir, still folder may be considered as empty if all children are only directories.
if metadata.is_dir() { if metadata.is_dir() {
let mut is_excluded_dir = false;
next_folder = "".to_owned() + &current_folder + &entry_data.file_name().into_string().unwrap() + "/"; next_folder = "".to_owned() + &current_folder + &entry_data.file_name().into_string().unwrap() + "/";
for ed in &self.excluded_directories { folders_to_check.push(next_folder.clone());
if next_folder == *ed {
is_excluded_dir = true;
break;
}
}
if !is_excluded_dir {
folders_to_check.push(next_folder.clone());
folders_checked.insert( folders_checked.insert(
next_folder.clone(), next_folder.clone(),
FolderEntry { FolderEntry {
parent_path: Option::from(current_folder.clone()), parent_path: Option::from(current_folder.clone()),
is_empty: FolderEmptiness::Maybe, is_empty: FolderEmptiness::Maybe,
}, },
); );
}
} else { } else {
// Not folder so it may be a file or symbolic link so it isn't empty // Not folder so it may be a file or symbolic link so it isn't empty
folders_checked.get_mut(&current_folder).unwrap().is_empty = FolderEmptiness::No; folders_checked.get_mut(&current_folder).unwrap().is_empty = FolderEmptiness::No;
@ -254,9 +283,8 @@ impl EmptyFolder {
#[cfg(debug_assertions)] #[cfg(debug_assertions)]
{ {
println!("---------------DEBUG PRINT---------------"); println!("---------------DEBUG PRINT---------------");
println!("Number of all checked folders - {}", self.informations.number_of_checked_folders); println!("Number of all checked folders - {}", self.information.number_of_checked_folders);
println!("Number of empty folders - {}", self.informations.number_of_empty_folders); println!("Number of empty folders - {}", self.information.number_of_empty_folders);
println!("Excluded directories - {:?}", self.excluded_directories);
println!("Included directories - {:?}", self.included_directories); println!("Included directories - {:?}", self.included_directories);
println!("-----------------------------------------"); println!("-----------------------------------------");
} }
@ -268,34 +296,15 @@ impl EmptyFolder {
let start_time: SystemTime = SystemTime::now(); let start_time: SystemTime = SystemTime::now();
let mut optimized_included: Vec<String> = Vec::<String>::new(); let mut optimized_included: Vec<String> = Vec::<String>::new();
let mut optimized_excluded: Vec<String> = Vec::<String>::new();
// Remove duplicated entries like: "/", "/" // Remove duplicated entries like: "/", "/"
self.excluded_directories.sort();
self.included_directories.sort(); self.included_directories.sort();
self.excluded_directories.dedup();
self.included_directories.dedup(); self.included_directories.dedup();
// Optimize for duplicated included directories - "/", "/home". "/home/Pulpit" to "/" // Optimize for duplicated included directories - "/", "/home". "/home/Pulpit" to "/"
let mut is_inside: bool;
for ed_checked in &self.excluded_directories {
is_inside = false;
for ed_help in &self.excluded_directories {
if ed_checked == ed_help {
// We checking same element
continue;
}
if ed_checked.starts_with(ed_help) {
is_inside = true;
break;
}
}
if !is_inside {
optimized_excluded.push(ed_checked.to_string());
}
}
let mut is_inside: bool;
for id_checked in &self.included_directories { for id_checked in &self.included_directories {
is_inside = false; is_inside = false;
for id_help in &self.included_directories { for id_help in &self.included_directories {
@ -313,24 +322,6 @@ impl EmptyFolder {
} }
} }
self.included_directories = optimized_included;
optimized_included = Vec::<String>::new();
self.excluded_directories = optimized_excluded;
optimized_excluded = Vec::<String>::new();
// Remove include directories which are inside any exclude directory
for id in &self.included_directories {
let mut is_inside: bool = false;
for ed in &self.excluded_directories {
if id.starts_with(ed) {
is_inside = true;
break;
}
}
if !is_inside {
optimized_included.push(id.to_string());
}
}
self.included_directories = optimized_included; self.included_directories = optimized_included;
optimized_included = Vec::<String>::new(); optimized_included = Vec::<String>::new();
@ -342,34 +333,8 @@ impl EmptyFolder {
} }
} }
for ed in &self.excluded_directories {
let path = Path::new(ed);
if path.exists() {
optimized_excluded.push(ed.to_string());
}
}
self.included_directories = optimized_included; self.included_directories = optimized_included;
// optimized_included = Vec::<String>::new(); //optimized_included = Vec::<String>::new();
self.excluded_directories = optimized_excluded;
optimized_excluded = Vec::<String>::new();
// Excluded paths must are inside include path, because
for ed in &self.excluded_directories {
let mut is_inside: bool = false;
for id in &self.included_directories {
if ed.starts_with(id) {
is_inside = true;
break;
}
}
if is_inside {
optimized_excluded.push(ed.to_string());
}
}
self.excluded_directories = optimized_excluded;
// optimized_excluded = Vec::<String>::new();
if self.included_directories.is_empty() { if self.included_directories.is_empty() {
println!("Optimize Directories ERROR: Excluded directories overlaps all included directories."); println!("Optimize Directories ERROR: Excluded directories overlaps all included directories.");
@ -377,7 +342,6 @@ impl EmptyFolder {
} }
// Not needed, but better is to have sorted everything // Not needed, but better is to have sorted everything
self.excluded_directories.sort();
self.included_directories.sort(); self.included_directories.sort();
Common::print_time(start_time, SystemTime::now(), "optimize_directories".to_string()); Common::print_time(start_time, SystemTime::now(), "optimize_directories".to_string());
} }
@ -441,59 +405,6 @@ impl EmptyFolder {
Common::print_time(start_time, SystemTime::now(), "set_include_directory".to_string()); Common::print_time(start_time, SystemTime::now(), "set_include_directory".to_string());
} }
pub fn set_exclude_directory(&mut self, mut exclude_directory: String) {
let start_time: SystemTime = SystemTime::now();
if exclude_directory.is_empty() {
return;
}
exclude_directory = exclude_directory.replace("\"", "");
let directories: Vec<String> = exclude_directory.split(',').map(String::from).collect();
let mut checked_directories: Vec<String> = Vec::new();
for directory in directories {
let directory: String = directory.trim().to_string();
if directory == "" {
continue;
}
if directory == "/" {
println!("Exclude Directory ERROR: Excluding / is pointless, because it means that no files will be scanned.");
break;
}
if directory.contains('*') {
println!("Exclude Directory ERROR: Wildcards are not supported, ignoring path {}.", directory);
continue;
}
if directory.starts_with('~') {
println!("Exclude Directory ERROR: ~ in path isn't supported, ignoring path {}.", directory);
continue;
}
if !directory.starts_with('/') {
println!("Exclude Directory ERROR: Relative path are not supported, ignoring path {}.", directory);
continue;
}
if !Path::new(&directory).exists() {
println!("Exclude Directory ERROR: Path {} doesn't exists.", directory);
continue;
}
if !Path::new(&directory).is_dir() {
println!("Exclude Directory ERROR: {} isn't folder.", directory);
continue;
}
// directory must end with /, due to possiblity of incorrect assumption, that e.g. /home/rafal is top folder to /home/rafalinho
if !directory.ends_with('/') {
checked_directories.push(directory.trim().to_string() + "/");
} else {
checked_directories.push(directory.trim().to_string());
}
}
self.excluded_directories = checked_directories;
Common::print_time(start_time, SystemTime::now(), "set_exclude_directory".to_string());
}
} }
impl Default for EmptyFolder { impl Default for EmptyFolder {
fn default() -> Self { fn default() -> Self {