358 lines
16 KiB
Rust
358 lines
16 KiB
Rust
use crossbeam_channel::Receiver;
|
|
use std::cell::RefCell;
|
|
use std::collections::HashMap;
|
|
use std::rc::Rc;
|
|
|
|
use std::time::Duration;
|
|
|
|
use glib::MainContext;
|
|
use gtk4::prelude::*;
|
|
use gtk4::ProgressBar;
|
|
|
|
use common_dir_traversal::CheckingMethod;
|
|
use czkawka_core::common_dir_traversal;
|
|
use czkawka_core::common_dir_traversal::{ProgressData, ToolType};
|
|
|
|
use crate::flg;
|
|
use crate::gui_structs::gui_data::GuiData;
|
|
use crate::localizer_core::generate_translation_hashmap;
|
|
use crate::taskbar_progress::tbp_flags::TBPF_INDETERMINATE;
|
|
use crate::taskbar_progress::TaskbarProgress;
|
|
|
|
#[allow(clippy::too_many_arguments)]
|
|
pub fn connect_progress_window(gui_data: &GuiData, progress_receiver: Receiver<ProgressData>) {
|
|
let main_context = MainContext::default();
|
|
let _guard = main_context.acquire().unwrap();
|
|
|
|
let gui_data = gui_data.clone();
|
|
|
|
let future = async move {
|
|
loop {
|
|
loop {
|
|
let item = progress_receiver.try_recv();
|
|
if let Ok(item) = item {
|
|
match item.tool_type {
|
|
ToolType::Duplicate => process_bar_duplicates(&gui_data, &item),
|
|
ToolType::EmptyFiles => process_bar_empty_files(&gui_data, &item),
|
|
ToolType::EmptyFolders => process_bar_empty_folder(&gui_data, &item),
|
|
ToolType::BigFile => process_bar_big_files(&gui_data, &item),
|
|
ToolType::SameMusic => process_bar_same_music(&gui_data, &item),
|
|
ToolType::SimilarImages => process_bar_similar_images(&gui_data, &item),
|
|
ToolType::SimilarVideos => process_bar_similar_videos(&gui_data, &item),
|
|
ToolType::TemporaryFiles => process_bar_temporary(&gui_data, &item),
|
|
ToolType::InvalidSymlinks => process_bar_invalid_symlinks(&gui_data, &item),
|
|
ToolType::BrokenFiles => process_bar_broken_files(&gui_data, &item),
|
|
ToolType::BadExtensions => process_bar_bad_extensions(&gui_data, &item),
|
|
ToolType::None => panic!(),
|
|
}
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
glib::timeout_future(Duration::from_millis(300)).await;
|
|
}
|
|
};
|
|
|
|
main_context.spawn_local(future);
|
|
}
|
|
|
|
fn process_bar_empty_files(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
label_stage.set_text(&flg!("progress_scanning_general_file", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
|
|
fn process_bar_empty_folder(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
label_stage.set_text(&flg!(
|
|
"progress_scanning_empty_folders",
|
|
generate_translation_hashmap(vec![("folder_number", item.entries_checked.to_string())])
|
|
));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
|
|
fn process_bar_big_files(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
label_stage.set_text(&flg!("progress_scanning_general_file", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
|
|
fn process_bar_same_music(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let progress_bar_current_stage = gui_data.progress_window.progress_bar_current_stage.clone();
|
|
let progress_bar_all_stages = gui_data.progress_window.progress_bar_all_stages.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
match item.current_stage {
|
|
0 => {
|
|
progress_bar_current_stage.hide();
|
|
label_stage.set_text(&flg!("progress_scanning_general_file", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
// Loading cache
|
|
1 => {
|
|
progress_bar_current_stage.hide();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
label_stage.set_text(&flg!("progress_cache_loading"));
|
|
}
|
|
2 => {
|
|
progress_bar_current_stage.show();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
|
|
match item.checking_method {
|
|
CheckingMethod::AudioTags => label_stage.set_text(&flg!("progress_scanning_music_tags", progress_ratio_tm(item))),
|
|
CheckingMethod::AudioContent => label_stage.set_text(&flg!("progress_scanning_music_content", progress_ratio_tm(item))),
|
|
_ => panic!(),
|
|
}
|
|
}
|
|
// Saving cache
|
|
3 => {
|
|
progress_bar_current_stage.hide();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
label_stage.set_text(&flg!("progress_cache_saving"));
|
|
}
|
|
4 => {
|
|
progress_bar_current_stage.show();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
|
|
match item.checking_method {
|
|
CheckingMethod::AudioTags => label_stage.set_text(&flg!("progress_scanning_music_tags_end", progress_ratio_tm(item))),
|
|
CheckingMethod::AudioContent => label_stage.set_text(&flg!("progress_scanning_music_content_end", progress_ratio_tm(item))),
|
|
_ => panic!(),
|
|
}
|
|
}
|
|
5 => {
|
|
progress_bar_current_stage.show();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
|
|
if item.checking_method == CheckingMethod::AudioContent {
|
|
label_stage.set_text(&flg!("progress_scanning_music_tags", progress_ratio_tm(item)));
|
|
} else {
|
|
panic!();
|
|
}
|
|
}
|
|
_ => panic!(),
|
|
}
|
|
}
|
|
|
|
fn process_bar_similar_images(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let progress_bar_current_stage = gui_data.progress_window.progress_bar_current_stage.clone();
|
|
let progress_bar_all_stages = gui_data.progress_window.progress_bar_all_stages.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
match item.current_stage {
|
|
0 => {
|
|
progress_bar_current_stage.hide();
|
|
label_stage.set_text(&flg!("progress_scanning_general_file", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
1 => {
|
|
progress_bar_current_stage.show();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
label_stage.set_text(&flg!("progress_scanning_image", progress_ratio_tm(item)));
|
|
}
|
|
2 => {
|
|
progress_bar_current_stage.show();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
label_stage.set_text(&flg!("progress_comparing_image_hashes", progress_ratio_tm(item)));
|
|
}
|
|
_ => panic!(),
|
|
}
|
|
}
|
|
|
|
fn process_bar_similar_videos(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let progress_bar_current_stage = gui_data.progress_window.progress_bar_current_stage.clone();
|
|
let progress_bar_all_stages = gui_data.progress_window.progress_bar_all_stages.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
match item.current_stage {
|
|
0 => {
|
|
progress_bar_current_stage.hide();
|
|
label_stage.set_text(&flg!("progress_scanning_general_file", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
1 => {
|
|
progress_bar_current_stage.show();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
label_stage.set_text(&flg!("progress_scanning_video", progress_ratio_tm(item)));
|
|
}
|
|
_ => panic!(),
|
|
}
|
|
}
|
|
|
|
fn process_bar_temporary(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
label_stage.set_text(&flg!("progress_scanning_general_file", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
|
|
fn process_bar_invalid_symlinks(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
label_stage.set_text(&flg!("progress_scanning_general_file", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
|
|
fn process_bar_broken_files(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let progress_bar_current_stage = gui_data.progress_window.progress_bar_current_stage.clone();
|
|
let progress_bar_all_stages = gui_data.progress_window.progress_bar_all_stages.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
match item.current_stage {
|
|
0 => {
|
|
progress_bar_current_stage.hide();
|
|
label_stage.set_text(&flg!("progress_scanning_general_file", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
1 => {
|
|
progress_bar_current_stage.show();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
label_stage.set_text(&flg!("progress_scanning_broken_files", progress_ratio_tm(item)));
|
|
}
|
|
_ => panic!(),
|
|
}
|
|
}
|
|
|
|
fn process_bar_bad_extensions(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let progress_bar_current_stage = gui_data.progress_window.progress_bar_current_stage.clone();
|
|
let progress_bar_all_stages = gui_data.progress_window.progress_bar_all_stages.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
match item.current_stage {
|
|
0 => {
|
|
progress_bar_current_stage.hide();
|
|
label_stage.set_text(&flg!("progress_scanning_general_file", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
1 => {
|
|
progress_bar_current_stage.show();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
label_stage.set_text(&flg!("progress_scanning_extension_of_files", progress_ratio_tm(item)));
|
|
}
|
|
_ => panic!(),
|
|
}
|
|
}
|
|
|
|
fn process_bar_duplicates(gui_data: &GuiData, item: &ProgressData) {
|
|
let label_stage = gui_data.progress_window.label_stage.clone();
|
|
let progress_bar_current_stage = gui_data.progress_window.progress_bar_current_stage.clone();
|
|
let progress_bar_all_stages = gui_data.progress_window.progress_bar_all_stages.clone();
|
|
let grid_progress_stages = gui_data.progress_window.grid_progress_stages.clone();
|
|
let taskbar_state = gui_data.taskbar_state.clone();
|
|
|
|
match item.checking_method {
|
|
CheckingMethod::Hash => {
|
|
label_stage.show();
|
|
match item.current_stage {
|
|
// Checking Size
|
|
0 => {
|
|
progress_bar_current_stage.hide();
|
|
progress_bar_all_stages.set_fraction(0f64);
|
|
label_stage.set_text(&flg!("progress_scanning_size", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
// Loading cache
|
|
1 | 4 => {
|
|
progress_bar_current_stage.hide();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
|
|
if item.current_stage == 1 {
|
|
label_stage.set_text(&flg!("progress_prehash_cache_loading"));
|
|
} else {
|
|
label_stage.set_text(&flg!("progress_hash_cache_loading"));
|
|
}
|
|
}
|
|
// Saving cache
|
|
3 | 6 => {
|
|
progress_bar_current_stage.hide();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
|
|
if item.current_stage == 3 {
|
|
label_stage.set_text(&flg!("progress_prehash_cache_saving"));
|
|
} else {
|
|
label_stage.set_text(&flg!("progress_hash_cache_saving"));
|
|
}
|
|
}
|
|
// Hash - first 1KB file
|
|
2 => {
|
|
progress_bar_current_stage.show();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
|
|
label_stage.set_text(&flg!("progress_analyzed_partial_hash", progress_ratio_tm(item)));
|
|
}
|
|
// Hash - normal hash
|
|
5 => {
|
|
progress_bar_current_stage.show();
|
|
common_set_data(item, &progress_bar_all_stages, &progress_bar_current_stage, &taskbar_state);
|
|
label_stage.set_text(&flg!("progress_analyzed_full_hash", progress_ratio_tm(item)));
|
|
}
|
|
_ => {
|
|
panic!("Not available current_stage");
|
|
}
|
|
}
|
|
}
|
|
CheckingMethod::Name => {
|
|
label_stage.show();
|
|
grid_progress_stages.hide();
|
|
|
|
label_stage.set_text(&flg!("progress_scanning_name", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
CheckingMethod::SizeName => {
|
|
label_stage.show();
|
|
grid_progress_stages.hide();
|
|
|
|
label_stage.set_text(&flg!("progress_scanning_size_name", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
CheckingMethod::Size => {
|
|
label_stage.show();
|
|
grid_progress_stages.hide();
|
|
|
|
label_stage.set_text(&flg!("progress_scanning_size", file_number_tm(item)));
|
|
taskbar_state.borrow().set_progress_state(TBPF_INDETERMINATE);
|
|
}
|
|
_ => panic!(),
|
|
};
|
|
}
|
|
|
|
fn common_set_data(item: &ProgressData, progress_bar_all_stages: &ProgressBar, progress_bar_current_stage: &ProgressBar, taskbar_state: &Rc<RefCell<TaskbarProgress>>) {
|
|
if item.entries_to_check != 0 {
|
|
let all_stages = (item.current_stage as f64 + (item.entries_checked) as f64 / item.entries_to_check as f64) / (item.max_stage + 1) as f64;
|
|
let all_stages = if all_stages > 0.99 { 0.99 } else { all_stages };
|
|
progress_bar_all_stages.set_fraction(all_stages);
|
|
progress_bar_current_stage.set_fraction((item.entries_checked) as f64 / item.entries_to_check as f64);
|
|
taskbar_state.borrow().set_progress_value(
|
|
((item.current_stage as usize) * item.entries_to_check + item.entries_checked) as u64,
|
|
item.entries_to_check as u64 * (item.max_stage + 1) as u64,
|
|
);
|
|
} else {
|
|
let all_stages = (item.current_stage as f64) / (item.max_stage + 1) as f64;
|
|
let all_stages = if all_stages > 0.99 { 0.99 } else { all_stages };
|
|
progress_bar_all_stages.set_fraction(all_stages);
|
|
progress_bar_current_stage.set_fraction(0f64);
|
|
taskbar_state.borrow().set_progress_value(item.current_stage as u64, 1 + item.max_stage as u64);
|
|
}
|
|
}
|
|
|
|
fn file_number_tm(item: &ProgressData) -> HashMap<&'static str, String> {
|
|
generate_translation_hashmap(vec![("file_number", item.entries_checked.to_string())])
|
|
}
|
|
|
|
fn progress_ratio_tm(item: &ProgressData) -> HashMap<&'static str, String> {
|
|
generate_translation_hashmap(vec![("file_checked", item.entries_checked.to_string()), ("all_files", item.entries_to_check.to_string())])
|
|
}
|