404 lines
21 KiB
Rust
404 lines
21 KiB
Rust
use czkawka_core::*;
|
|
|
|
extern crate gtk;
|
|
use crate::gui_data::GuiData;
|
|
use crate::help_functions::*;
|
|
use crate::notebook_enums::*;
|
|
use czkawka_core::big_file::BigFile;
|
|
use czkawka_core::broken_files::BrokenFiles;
|
|
use czkawka_core::duplicate::DuplicateFinder;
|
|
use czkawka_core::empty_files::EmptyFiles;
|
|
use czkawka_core::empty_folder::EmptyFolder;
|
|
use czkawka_core::invalid_symlinks::InvalidSymlinks;
|
|
use czkawka_core::same_music::{MusicSimilarity, SameMusic};
|
|
use czkawka_core::similar_images::SimilarImages;
|
|
use czkawka_core::temporary::Temporary;
|
|
use czkawka_core::zeroed::ZeroedFiles;
|
|
use glib::Sender;
|
|
use gtk::prelude::*;
|
|
use gtk::{ResponseType, WindowPosition};
|
|
use std::sync::atomic::{AtomicBool, Ordering};
|
|
use std::sync::Arc;
|
|
use std::thread;
|
|
|
|
#[allow(clippy::too_many_arguments)]
|
|
pub fn connect_button_search(
|
|
gui_data: &GuiData,
|
|
glib_stop_sender: Sender<Message>,
|
|
futures_sender_duplicate_files: futures::channel::mpsc::Sender<duplicate::ProgressData>,
|
|
futures_sender_empty_files: futures::channel::mpsc::Sender<empty_files::ProgressData>,
|
|
futures_sender_empty_folder: futures::channel::mpsc::Sender<empty_folder::ProgressData>,
|
|
futures_sender_big_file: futures::channel::mpsc::Sender<big_file::ProgressData>,
|
|
futures_sender_same_music: futures::channel::mpsc::Sender<same_music::ProgressData>,
|
|
futures_sender_similar_images: futures::channel::mpsc::Sender<similar_images::ProgressData>,
|
|
futures_sender_temporary: futures::channel::mpsc::Sender<temporary::ProgressData>,
|
|
futures_sender_zeroed: futures::channel::mpsc::Sender<zeroed::ProgressData>,
|
|
futures_sender_invalid_symlinks: futures::channel::mpsc::Sender<invalid_symlinks::ProgressData>,
|
|
futures_sender_broken_files: futures::channel::mpsc::Sender<broken_files::ProgressData>,
|
|
) {
|
|
let stop_sender = gui_data.stop_sender.clone();
|
|
let entry_info = gui_data.entry_info.clone();
|
|
let notebook_main = gui_data.main_notebook.notebook_main.clone();
|
|
let tree_view_included_directories = gui_data.upper_notebook.tree_view_included_directories.clone();
|
|
let tree_view_excluded_directories = gui_data.upper_notebook.tree_view_excluded_directories.clone();
|
|
let buttons_search_clone = gui_data.bottom_buttons.buttons_search.clone();
|
|
let buttons_array = gui_data.bottom_buttons.buttons_array.clone();
|
|
let check_button_recursive = gui_data.upper_notebook.check_button_recursive.clone();
|
|
let entry_excluded_items = gui_data.upper_notebook.entry_excluded_items.clone();
|
|
let entry_same_music_minimal_size = gui_data.main_notebook.entry_same_music_minimal_size.clone();
|
|
let entry_allowed_extensions = gui_data.upper_notebook.entry_allowed_extensions.clone();
|
|
let buttons_names = gui_data.bottom_buttons.buttons_names.clone();
|
|
let radio_button_duplicates_name = gui_data.main_notebook.radio_button_duplicates_name.clone();
|
|
let radio_button_duplicates_size = gui_data.main_notebook.radio_button_duplicates_size.clone();
|
|
let radio_button_duplicates_hashmb = gui_data.main_notebook.radio_button_duplicates_hashmb.clone();
|
|
let radio_button_duplicates_hash = gui_data.main_notebook.radio_button_duplicates_hash.clone();
|
|
let radio_button_similar_images_minimal = gui_data.main_notebook.radio_button_similar_images_minimal.clone();
|
|
let radio_button_similar_images_very_small = gui_data.main_notebook.radio_button_similar_images_very_small.clone();
|
|
let radio_button_similar_images_small = gui_data.main_notebook.radio_button_similar_images_small.clone();
|
|
let radio_button_similar_images_medium = gui_data.main_notebook.radio_button_similar_images_medium.clone();
|
|
let radio_button_similar_images_high = gui_data.main_notebook.radio_button_similar_images_high.clone();
|
|
let radio_button_similar_images_very_high = gui_data.main_notebook.radio_button_similar_images_very_high.clone();
|
|
let entry_duplicate_minimal_size = gui_data.main_notebook.entry_duplicate_minimal_size.clone();
|
|
let stop_receiver = gui_data.stop_receiver.clone();
|
|
let entry_big_files_number = gui_data.main_notebook.entry_big_files_number.clone();
|
|
let entry_similar_images_minimal_size = gui_data.main_notebook.entry_similar_images_minimal_size.clone();
|
|
let check_button_music_title: gtk::CheckButton = gui_data.main_notebook.check_button_music_title.clone();
|
|
let check_button_music_artist: gtk::CheckButton = gui_data.main_notebook.check_button_music_artist.clone();
|
|
let check_button_music_album_title: gtk::CheckButton = gui_data.main_notebook.check_button_music_album_title.clone();
|
|
let check_button_music_album_artist: gtk::CheckButton = gui_data.main_notebook.check_button_music_album_artist.clone();
|
|
let check_button_music_year: gtk::CheckButton = gui_data.main_notebook.check_button_music_year.clone();
|
|
let shared_buttons = gui_data.shared_buttons.clone();
|
|
let tree_view_empty_folder_finder = gui_data.main_notebook.tree_view_empty_folder_finder.clone();
|
|
let tree_view_empty_files_finder = gui_data.main_notebook.tree_view_empty_files_finder.clone();
|
|
let tree_view_big_files_finder = gui_data.main_notebook.tree_view_big_files_finder.clone();
|
|
let tree_view_duplicate_finder = gui_data.main_notebook.tree_view_duplicate_finder.clone();
|
|
let tree_view_temporary_files_finder = gui_data.main_notebook.tree_view_temporary_files_finder.clone();
|
|
let tree_view_same_music_finder = gui_data.main_notebook.tree_view_same_music_finder.clone();
|
|
let tree_view_similar_images_finder = gui_data.main_notebook.tree_view_similar_images_finder.clone();
|
|
let tree_view_zeroed_files_finder = gui_data.main_notebook.tree_view_zeroed_files_finder.clone();
|
|
let tree_view_invalid_symlinks = gui_data.main_notebook.tree_view_invalid_symlinks.clone();
|
|
let tree_view_broken_files = gui_data.main_notebook.tree_view_broken_files.clone();
|
|
let text_view_errors = gui_data.text_view_errors.clone();
|
|
let dialog_progress = gui_data.progress_dialog.dialog_progress.clone();
|
|
let label_stage = gui_data.progress_dialog.label_stage.clone();
|
|
let grid_progress_stages = gui_data.progress_dialog.grid_progress_stages.clone();
|
|
let progress_bar_current_stage = gui_data.progress_dialog.progress_bar_current_stage.clone();
|
|
let progress_bar_all_stages = gui_data.progress_dialog.progress_bar_all_stages.clone();
|
|
let image_preview_similar_images = gui_data.main_notebook.image_preview_similar_images.clone();
|
|
|
|
buttons_search_clone.connect_clicked(move |_| {
|
|
let included_directories = get_path_buf_from_vector_of_strings(get_string_from_list_store(&tree_view_included_directories));
|
|
let excluded_directories = get_path_buf_from_vector_of_strings(get_string_from_list_store(&tree_view_excluded_directories));
|
|
let recursive_search = check_button_recursive.get_active();
|
|
let excluded_items = entry_excluded_items.get_text().as_str().to_string().split(',').map(|e| e.to_string()).collect::<Vec<String>>();
|
|
let allowed_extensions = entry_allowed_extensions.get_text().as_str().to_string();
|
|
|
|
let show_dialog = Arc::new(AtomicBool::new(true));
|
|
|
|
hide_all_buttons(&buttons_array);
|
|
|
|
// Disable main notebook from any iteration until search will end
|
|
notebook_main.set_sensitive(false);
|
|
|
|
entry_info.set_text("Searching data, it may take a while, please wait...");
|
|
|
|
// Set dialog to center to current screen(it is impossible to center it to main window)
|
|
dialog_progress.set_position(WindowPosition::CenterAlways);
|
|
|
|
// Resets progress bars
|
|
progress_bar_all_stages.set_fraction(0 as f64);
|
|
progress_bar_current_stage.set_fraction(0 as f64);
|
|
|
|
reset_text_view(&text_view_errors);
|
|
|
|
let glib_stop_sender = glib_stop_sender.clone();
|
|
let stop_receiver = stop_receiver.clone();
|
|
|
|
match to_notebook_main_enum(notebook_main.get_current_page().unwrap()) {
|
|
NotebookMainEnum::Duplicate => {
|
|
label_stage.show();
|
|
grid_progress_stages.show_all();
|
|
dialog_progress.resize(1, 1);
|
|
|
|
get_list_store(&tree_view_duplicate_finder).clear();
|
|
|
|
let check_method;
|
|
if radio_button_duplicates_name.get_active() {
|
|
check_method = duplicate::CheckingMethod::Name;
|
|
} else if radio_button_duplicates_size.get_active() {
|
|
check_method = duplicate::CheckingMethod::Size;
|
|
} else if radio_button_duplicates_hashmb.get_active() {
|
|
check_method = duplicate::CheckingMethod::HashMB;
|
|
} else if radio_button_duplicates_hash.get_active() {
|
|
check_method = duplicate::CheckingMethod::Hash;
|
|
} else {
|
|
panic!("No radio button is pressed");
|
|
}
|
|
let minimal_file_size = entry_duplicate_minimal_size.get_text().as_str().parse::<u64>().unwrap_or(1024);
|
|
|
|
let futures_sender_duplicate_files = futures_sender_duplicate_files.clone();
|
|
// Find duplicates
|
|
thread::spawn(move || {
|
|
let mut df = DuplicateFinder::new();
|
|
df.set_included_directory(included_directories);
|
|
df.set_excluded_directory(excluded_directories);
|
|
df.set_recursive_search(recursive_search);
|
|
df.set_excluded_items(excluded_items);
|
|
df.set_allowed_extensions(allowed_extensions);
|
|
df.set_minimal_file_size(minimal_file_size);
|
|
df.set_check_method(check_method);
|
|
df.find_duplicates(Some(&stop_receiver), Some(&futures_sender_duplicate_files));
|
|
let _ = glib_stop_sender.send(Message::Duplicates(df));
|
|
});
|
|
}
|
|
NotebookMainEnum::EmptyFiles => {
|
|
label_stage.show();
|
|
grid_progress_stages.hide();
|
|
dialog_progress.resize(1, 1);
|
|
|
|
get_list_store(&tree_view_empty_files_finder).clear();
|
|
|
|
let futures_sender_empty_files = futures_sender_empty_files.clone();
|
|
// Find empty files
|
|
thread::spawn(move || {
|
|
let mut vf = EmptyFiles::new();
|
|
|
|
vf.set_included_directory(included_directories);
|
|
vf.set_excluded_directory(excluded_directories);
|
|
vf.set_recursive_search(recursive_search);
|
|
vf.set_excluded_items(excluded_items);
|
|
vf.set_allowed_extensions(allowed_extensions);
|
|
vf.find_empty_files(Some(&stop_receiver), Some(&futures_sender_empty_files));
|
|
let _ = glib_stop_sender.send(Message::EmptyFiles(vf));
|
|
});
|
|
}
|
|
NotebookMainEnum::EmptyDirectories => {
|
|
label_stage.show();
|
|
grid_progress_stages.hide();
|
|
dialog_progress.resize(1, 1);
|
|
|
|
get_list_store(&tree_view_empty_folder_finder).clear();
|
|
|
|
let futures_sender_empty_folder = futures_sender_empty_folder.clone();
|
|
// Find empty folders
|
|
thread::spawn(move || {
|
|
let mut ef = EmptyFolder::new();
|
|
ef.set_included_directory(included_directories);
|
|
ef.set_excluded_directory(excluded_directories);
|
|
ef.set_excluded_items(excluded_items);
|
|
ef.find_empty_folders(Some(&stop_receiver), Some(&futures_sender_empty_folder));
|
|
let _ = glib_stop_sender.send(Message::EmptyFolders(ef));
|
|
});
|
|
}
|
|
NotebookMainEnum::BigFiles => {
|
|
label_stage.show();
|
|
grid_progress_stages.hide();
|
|
dialog_progress.resize(1, 1);
|
|
|
|
get_list_store(&tree_view_big_files_finder).clear();
|
|
|
|
let numbers_of_files_to_check = entry_big_files_number.get_text().as_str().parse::<usize>().unwrap_or(50);
|
|
|
|
let futures_sender_big_file = futures_sender_big_file.clone();
|
|
// Find big files
|
|
thread::spawn(move || {
|
|
let mut bf = BigFile::new();
|
|
|
|
bf.set_included_directory(included_directories);
|
|
bf.set_excluded_directory(excluded_directories);
|
|
bf.set_recursive_search(recursive_search);
|
|
bf.set_excluded_items(excluded_items);
|
|
bf.set_number_of_files_to_check(numbers_of_files_to_check);
|
|
bf.find_big_files(Some(&stop_receiver), Some(&futures_sender_big_file));
|
|
let _ = glib_stop_sender.send(Message::BigFiles(bf));
|
|
});
|
|
}
|
|
NotebookMainEnum::Temporary => {
|
|
label_stage.show();
|
|
grid_progress_stages.hide();
|
|
dialog_progress.resize(1, 1);
|
|
|
|
get_list_store(&tree_view_temporary_files_finder).clear();
|
|
|
|
let futures_sender_temporary = futures_sender_temporary.clone();
|
|
// Find temporary files
|
|
thread::spawn(move || {
|
|
let mut tf = Temporary::new();
|
|
|
|
tf.set_included_directory(included_directories);
|
|
tf.set_excluded_directory(excluded_directories);
|
|
tf.set_recursive_search(recursive_search);
|
|
tf.set_excluded_items(excluded_items);
|
|
tf.find_temporary_files(Some(&stop_receiver), Some(&futures_sender_temporary));
|
|
let _ = glib_stop_sender.send(Message::Temporary(tf));
|
|
});
|
|
}
|
|
NotebookMainEnum::SimilarImages => {
|
|
image_preview_similar_images.hide();
|
|
|
|
label_stage.show();
|
|
grid_progress_stages.show_all();
|
|
dialog_progress.resize(1, 1);
|
|
|
|
get_list_store(&tree_view_similar_images_finder).clear();
|
|
|
|
let minimal_file_size = entry_similar_images_minimal_size.get_text().as_str().parse::<u64>().unwrap_or(1024 * 16);
|
|
|
|
let similarity;
|
|
if radio_button_similar_images_minimal.get_active() {
|
|
similarity = similar_images::Similarity::Minimal;
|
|
} else if radio_button_similar_images_very_small.get_active() {
|
|
similarity = similar_images::Similarity::VerySmall;
|
|
} else if radio_button_similar_images_small.get_active() {
|
|
similarity = similar_images::Similarity::Small;
|
|
} else if radio_button_similar_images_medium.get_active() {
|
|
similarity = similar_images::Similarity::Medium;
|
|
} else if radio_button_similar_images_high.get_active() {
|
|
similarity = similar_images::Similarity::High;
|
|
} else if radio_button_similar_images_very_high.get_active() {
|
|
similarity = similar_images::Similarity::VeryHigh;
|
|
} else {
|
|
panic!("No radio button is pressed");
|
|
}
|
|
|
|
let futures_sender_similar_images = futures_sender_similar_images.clone();
|
|
// Find similar images
|
|
thread::spawn(move || {
|
|
let mut sf = SimilarImages::new();
|
|
|
|
sf.set_included_directory(included_directories);
|
|
sf.set_excluded_directory(excluded_directories);
|
|
sf.set_recursive_search(recursive_search);
|
|
sf.set_excluded_items(excluded_items);
|
|
sf.set_minimal_file_size(minimal_file_size);
|
|
sf.set_similarity(similarity);
|
|
sf.find_similar_images(Some(&stop_receiver), Some(&futures_sender_similar_images));
|
|
let _ = glib_stop_sender.send(Message::SimilarImages(sf));
|
|
});
|
|
}
|
|
NotebookMainEnum::Zeroed => {
|
|
label_stage.show();
|
|
grid_progress_stages.show_all();
|
|
dialog_progress.resize(1, 1);
|
|
|
|
get_list_store(&tree_view_zeroed_files_finder).clear();
|
|
|
|
let futures_sender_zeroed = futures_sender_zeroed.clone();
|
|
// Find zeroed files
|
|
thread::spawn(move || {
|
|
let mut zf = ZeroedFiles::new();
|
|
|
|
zf.set_included_directory(included_directories);
|
|
zf.set_excluded_directory(excluded_directories);
|
|
zf.set_recursive_search(recursive_search);
|
|
zf.set_excluded_items(excluded_items);
|
|
zf.set_allowed_extensions(allowed_extensions);
|
|
zf.find_zeroed_files(Some(&stop_receiver), Some(&futures_sender_zeroed));
|
|
let _ = glib_stop_sender.send(Message::ZeroedFiles(zf));
|
|
});
|
|
}
|
|
NotebookMainEnum::SameMusic => {
|
|
label_stage.show();
|
|
grid_progress_stages.show_all();
|
|
dialog_progress.resize(1, 1);
|
|
|
|
get_list_store(&tree_view_same_music_finder).clear();
|
|
|
|
let minimal_file_size = entry_same_music_minimal_size.get_text().as_str().parse::<u64>().unwrap_or(1024);
|
|
|
|
let mut music_similarity: MusicSimilarity = MusicSimilarity::NONE;
|
|
|
|
if check_button_music_title.get_active() {
|
|
music_similarity |= MusicSimilarity::TITLE;
|
|
}
|
|
if check_button_music_artist.get_active() {
|
|
music_similarity |= MusicSimilarity::ARTIST;
|
|
}
|
|
if check_button_music_album_title.get_active() {
|
|
music_similarity |= MusicSimilarity::ALBUM_TITLE;
|
|
}
|
|
if check_button_music_album_artist.get_active() {
|
|
music_similarity |= MusicSimilarity::ALBUM_ARTIST;
|
|
}
|
|
if check_button_music_year.get_active() {
|
|
music_similarity |= MusicSimilarity::YEAR;
|
|
}
|
|
|
|
if music_similarity != MusicSimilarity::NONE {
|
|
let futures_sender_same_music = futures_sender_same_music.clone();
|
|
// Find Similar music
|
|
thread::spawn(move || {
|
|
let mut mf = SameMusic::new();
|
|
|
|
mf.set_included_directory(included_directories);
|
|
mf.set_excluded_directory(excluded_directories);
|
|
mf.set_excluded_items(excluded_items);
|
|
mf.set_minimal_file_size(minimal_file_size);
|
|
mf.set_recursive_search(recursive_search);
|
|
mf.set_music_similarity(music_similarity);
|
|
mf.find_same_music(Some(&stop_receiver), Some(&futures_sender_same_music));
|
|
let _ = glib_stop_sender.send(Message::SameMusic(mf));
|
|
});
|
|
} else {
|
|
notebook_main.set_sensitive(true);
|
|
set_buttons(&mut *shared_buttons.borrow_mut().get_mut(&NotebookMainEnum::SameMusic).unwrap(), &buttons_array, &buttons_names);
|
|
entry_info.set_text("ERROR: You must select at least one checkbox with music searching types.");
|
|
show_dialog.store(false, Ordering::Relaxed);
|
|
}
|
|
}
|
|
NotebookMainEnum::Symlinks => {
|
|
label_stage.show();
|
|
grid_progress_stages.hide();
|
|
dialog_progress.resize(1, 1);
|
|
|
|
get_list_store(&tree_view_invalid_symlinks).clear();
|
|
|
|
let futures_sender_invalid_symlinks = futures_sender_invalid_symlinks.clone();
|
|
|
|
thread::spawn(move || {
|
|
let mut isf = InvalidSymlinks::new();
|
|
|
|
isf.set_included_directory(included_directories);
|
|
isf.set_excluded_directory(excluded_directories);
|
|
isf.set_recursive_search(recursive_search);
|
|
isf.set_excluded_items(excluded_items);
|
|
isf.find_invalid_links(Some(&stop_receiver), Some(&futures_sender_invalid_symlinks));
|
|
let _ = glib_stop_sender.send(Message::InvalidSymlinks(isf));
|
|
});
|
|
}
|
|
NotebookMainEnum::BrokenFiles => {
|
|
label_stage.show();
|
|
grid_progress_stages.show();
|
|
dialog_progress.resize(1, 1);
|
|
|
|
get_list_store(&tree_view_broken_files).clear();
|
|
|
|
let futures_sender_broken_files = futures_sender_broken_files.clone();
|
|
|
|
thread::spawn(move || {
|
|
let mut br = BrokenFiles::new();
|
|
|
|
br.set_included_directory(included_directories);
|
|
br.set_excluded_directory(excluded_directories);
|
|
br.set_recursive_search(recursive_search);
|
|
br.set_excluded_items(excluded_items);
|
|
br.find_broken_files(Some(&stop_receiver), Some(&futures_sender_broken_files));
|
|
let _ = glib_stop_sender.send(Message::BrokenFiles(br));
|
|
});
|
|
}
|
|
}
|
|
|
|
// Show progress dialog
|
|
if show_dialog.load(Ordering::Relaxed) {
|
|
dialog_progress.show();
|
|
|
|
let response = dialog_progress.run();
|
|
if response == ResponseType::DeleteEvent {
|
|
stop_sender.send(()).unwrap();
|
|
}
|
|
|
|
dialog_progress.hide();
|
|
}
|
|
});
|
|
}
|