2021-11-28 08:49:20 +13:00
|
|
|
use std::fs;
|
|
|
|
use std::path::PathBuf;
|
|
|
|
|
|
|
|
use gtk::prelude::*;
|
|
|
|
use gtk::{TextView, TreeIter, TreePath};
|
|
|
|
|
|
|
|
use czkawka_core::duplicate::make_hard_link;
|
|
|
|
|
2021-02-25 01:53:46 +13:00
|
|
|
use crate::gui_data::GuiData;
|
|
|
|
use crate::help_functions::*;
|
|
|
|
use crate::notebook_enums::*;
|
|
|
|
|
2021-11-25 20:36:49 +13:00
|
|
|
pub fn connect_button_hardlink_symlink(gui_data: &GuiData) {
|
2021-02-25 01:53:46 +13:00
|
|
|
let buttons_hardlink = gui_data.bottom_buttons.buttons_hardlink.clone();
|
|
|
|
|
2021-11-25 20:36:49 +13:00
|
|
|
let notebook_main = gui_data.main_notebook.notebook_main.clone();
|
|
|
|
let main_tree_views = gui_data.main_notebook.get_main_tree_views();
|
2021-02-25 01:53:46 +13:00
|
|
|
|
|
|
|
let image_preview_similar_images = gui_data.main_notebook.image_preview_similar_images.clone();
|
2021-11-25 20:36:49 +13:00
|
|
|
let image_preview_duplicates = gui_data.main_notebook.image_preview_duplicates.clone();
|
2021-02-25 01:53:46 +13:00
|
|
|
|
2021-11-25 20:36:49 +13:00
|
|
|
let text_view_errors = gui_data.text_view_errors.clone();
|
|
|
|
|
|
|
|
buttons_hardlink.connect_clicked(move |_| {
|
|
|
|
let nb_number = notebook_main.current_page().unwrap();
|
|
|
|
let tree_view = &main_tree_views[nb_number as usize];
|
|
|
|
let nb_object = &NOTEBOOKS_INFOS[nb_number as usize];
|
|
|
|
|
|
|
|
let column_color = nb_object.column_color.expect("Hardinkning can be only used for tree views with grouped results");
|
|
|
|
hardlink_symlink(tree_view, nb_object.column_name, nb_object.column_path, column_color, nb_object.column_selection, true, &text_view_errors);
|
|
|
|
|
|
|
|
match &nb_object.notebook_type {
|
|
|
|
NotebookMainEnum::SimilarImages => {
|
|
|
|
image_preview_similar_images.hide();
|
|
|
|
}
|
|
|
|
NotebookMainEnum::Duplicate => {
|
|
|
|
image_preview_duplicates.hide();
|
|
|
|
}
|
|
|
|
_ => {}
|
2021-02-25 01:53:46 +13:00
|
|
|
}
|
2021-11-25 20:36:49 +13:00
|
|
|
});
|
|
|
|
|
|
|
|
let buttons_symlink = gui_data.bottom_buttons.buttons_symlink.clone();
|
|
|
|
|
|
|
|
let notebook_main = gui_data.main_notebook.notebook_main.clone();
|
|
|
|
let main_tree_views = gui_data.main_notebook.get_main_tree_views();
|
|
|
|
|
|
|
|
let image_preview_similar_images = gui_data.main_notebook.image_preview_similar_images.clone();
|
|
|
|
let image_preview_duplicates = gui_data.main_notebook.image_preview_duplicates.clone();
|
|
|
|
|
|
|
|
let text_view_errors = gui_data.text_view_errors.clone();
|
|
|
|
|
|
|
|
buttons_symlink.connect_clicked(move |_| {
|
|
|
|
let nb_number = notebook_main.current_page().unwrap();
|
|
|
|
let tree_view = &main_tree_views[nb_number as usize];
|
|
|
|
let nb_object = &NOTEBOOKS_INFOS[nb_number as usize];
|
|
|
|
|
|
|
|
let column_color = nb_object.column_color.expect("Symlinking can be only used for tree views with grouped results");
|
|
|
|
hardlink_symlink(tree_view, nb_object.column_name, nb_object.column_path, column_color, nb_object.column_selection, false, &text_view_errors);
|
|
|
|
|
|
|
|
match &nb_object.notebook_type {
|
|
|
|
NotebookMainEnum::SimilarImages => {
|
|
|
|
image_preview_similar_images.hide();
|
|
|
|
}
|
|
|
|
NotebookMainEnum::Duplicate => {
|
|
|
|
image_preview_duplicates.hide();
|
|
|
|
}
|
|
|
|
_ => {}
|
2021-11-23 23:10:24 +13:00
|
|
|
}
|
2021-02-25 01:53:46 +13:00
|
|
|
});
|
|
|
|
}
|
2021-08-08 00:53:34 +12:00
|
|
|
|
2021-11-25 20:36:49 +13:00
|
|
|
pub fn hardlink_symlink(tree_view: >k::TreeView, column_file_name: i32, column_path: i32, column_color: i32, column_selection: i32, hardlinking: bool, text_view_errors: &TextView) {
|
|
|
|
reset_text_view(text_view_errors);
|
2021-02-25 01:53:46 +13:00
|
|
|
|
2021-11-25 20:36:49 +13:00
|
|
|
let model = get_list_store(tree_view);
|
2021-02-25 01:53:46 +13:00
|
|
|
|
2021-08-08 00:53:34 +12:00
|
|
|
#[derive(Debug)]
|
|
|
|
struct SymHardlinkData {
|
2021-02-25 01:53:46 +13:00
|
|
|
original_data: String,
|
2021-08-08 00:53:34 +12:00
|
|
|
files_to_symhardlink: Vec<String>,
|
2021-02-25 01:53:46 +13:00
|
|
|
}
|
|
|
|
let mut vec_tree_path_to_remove: Vec<TreePath> = Vec::new(); // List of hardlinked files without its root
|
2021-08-08 00:53:34 +12:00
|
|
|
let mut vec_symhardlink_data: Vec<SymHardlinkData> = Vec::new();
|
|
|
|
|
|
|
|
let current_iter: TreeIter = match model.iter_first() {
|
|
|
|
Some(t) => t,
|
|
|
|
None => return, // No records
|
|
|
|
};
|
2021-02-25 01:53:46 +13:00
|
|
|
|
2021-08-08 00:53:34 +12:00
|
|
|
let mut selected_rows = Vec::new();
|
|
|
|
if let Some(iter) = model.iter_first() {
|
|
|
|
loop {
|
2021-11-25 20:36:49 +13:00
|
|
|
if model.value(&iter, column_selection).get::<bool>().unwrap() {
|
|
|
|
if model.value(&iter, column_color).get::<String>().unwrap() == MAIN_ROW_COLOR {
|
|
|
|
selected_rows.push(model.path(&iter).unwrap());
|
|
|
|
} else {
|
|
|
|
panic!("Header row shouldn't be selected, please report bug.");
|
|
|
|
}
|
2021-08-08 00:53:34 +12:00
|
|
|
}
|
|
|
|
if !model.iter_next(&iter) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-11-25 20:36:49 +13:00
|
|
|
if selected_rows.is_empty() {
|
|
|
|
return; // No selected rows
|
|
|
|
}
|
|
|
|
|
2021-08-08 00:53:34 +12:00
|
|
|
let mut current_symhardlink_data: Option<SymHardlinkData> = None;
|
2021-02-25 01:53:46 +13:00
|
|
|
let mut current_selected_index = 0;
|
|
|
|
loop {
|
2021-08-08 00:53:34 +12:00
|
|
|
if model.value(¤t_iter, column_color).get::<String>().unwrap() == HEADER_ROW_COLOR {
|
|
|
|
if let Some(current_symhardlink_data) = current_symhardlink_data {
|
|
|
|
if !current_symhardlink_data.files_to_symhardlink.is_empty() {
|
|
|
|
vec_symhardlink_data.push(current_symhardlink_data);
|
2021-02-25 01:53:46 +13:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-08-08 00:53:34 +12:00
|
|
|
current_symhardlink_data = None;
|
|
|
|
if !model.iter_next(¤t_iter) {
|
2021-02-25 01:53:46 +13:00
|
|
|
panic!("HEADER, shouldn't be a last item.");
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2021-08-08 00:53:34 +12:00
|
|
|
if model.path(¤t_iter).unwrap() == selected_rows[current_selected_index] {
|
|
|
|
let file_name = model.value(¤t_iter, column_file_name).get::<String>().unwrap();
|
|
|
|
let path = model.value(¤t_iter, column_path).get::<String>().unwrap();
|
2021-02-25 01:53:46 +13:00
|
|
|
let full_file_path = format!("{}/{}", path, file_name);
|
|
|
|
|
2021-08-08 00:53:34 +12:00
|
|
|
if current_symhardlink_data.is_some() {
|
|
|
|
vec_tree_path_to_remove.push(model.path(¤t_iter).unwrap());
|
|
|
|
let mut temp_data = current_symhardlink_data.unwrap();
|
|
|
|
temp_data.files_to_symhardlink.push(full_file_path);
|
|
|
|
current_symhardlink_data = Some(temp_data);
|
2021-02-25 01:53:46 +13:00
|
|
|
} else {
|
2021-08-08 00:53:34 +12:00
|
|
|
current_symhardlink_data = Some(SymHardlinkData {
|
2021-02-25 01:53:46 +13:00
|
|
|
original_data: full_file_path,
|
2021-08-08 00:53:34 +12:00
|
|
|
files_to_symhardlink: vec![],
|
2021-02-25 01:53:46 +13:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2021-08-08 00:53:34 +12:00
|
|
|
if current_selected_index != selected_rows.len() - 1 {
|
2021-02-25 01:53:46 +13:00
|
|
|
current_selected_index += 1;
|
|
|
|
} else {
|
2021-08-08 00:53:34 +12:00
|
|
|
if let Some(current_symhardlink_data) = current_symhardlink_data {
|
|
|
|
if !current_symhardlink_data.files_to_symhardlink.is_empty() {
|
|
|
|
vec_symhardlink_data.push(current_symhardlink_data);
|
2021-02-25 01:53:46 +13:00
|
|
|
}
|
|
|
|
}
|
|
|
|
break; // There is no more selected items, so we just end checking
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-08-08 00:53:34 +12:00
|
|
|
if !model.iter_next(¤t_iter) {
|
|
|
|
if let Some(current_symhardlink_data) = current_symhardlink_data {
|
|
|
|
if !current_symhardlink_data.files_to_symhardlink.is_empty() {
|
|
|
|
vec_symhardlink_data.push(current_symhardlink_data);
|
2021-02-25 01:53:46 +13:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2021-08-08 00:53:34 +12:00
|
|
|
if hardlinking {
|
|
|
|
dbg!(&vec_symhardlink_data);
|
|
|
|
for symhardlink_data in vec_symhardlink_data {
|
|
|
|
dbg!(&symhardlink_data);
|
|
|
|
for file_to_hardlink in symhardlink_data.files_to_symhardlink {
|
2021-12-01 23:09:47 +13:00
|
|
|
if let Err(e) = make_hard_link(&PathBuf::from(&symhardlink_data.original_data), &PathBuf::from(&file_to_hardlink)) {
|
|
|
|
add_text_to_text_view(text_view_errors, format!("Failed to hardlink {}, reason {}", file_to_hardlink, e).as_str());
|
|
|
|
continue;
|
2021-08-08 00:53:34 +12:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
for symhardlink_data in vec_symhardlink_data {
|
|
|
|
for file_to_symlink in symhardlink_data.files_to_symhardlink {
|
2021-12-01 23:09:47 +13:00
|
|
|
if let Err(e) = fs::remove_file(&file_to_symlink) {
|
|
|
|
add_text_to_text_view(text_view_errors, format!("Failed to remove file {} when creating symlink, reason {}", file_to_symlink, e).as_str());
|
|
|
|
continue;
|
2021-08-08 00:53:34 +12:00
|
|
|
};
|
|
|
|
|
|
|
|
#[cfg(target_family = "unix")]
|
|
|
|
{
|
2021-12-01 23:09:47 +13:00
|
|
|
if let Err(e) = std::os::unix::fs::symlink(&symhardlink_data.original_data, &file_to_symlink) {
|
|
|
|
add_text_to_text_view(text_view_errors, format!("Failed to remove file {} when creating symlink, reason {}", file_to_symlink, e).as_str());
|
|
|
|
continue;
|
2021-08-08 00:53:34 +12:00
|
|
|
};
|
|
|
|
}
|
|
|
|
#[cfg(target_family = "windows")]
|
|
|
|
{
|
2021-12-01 23:09:47 +13:00
|
|
|
if let Err(e) = std::os::windows::fs::symlink_file(&symhardlink_data.original_data, &file_to_symlink) {
|
|
|
|
add_text_to_text_view(&text_view_errors, format!("Failed to remove file {} when creating symlink, reason {}", file_to_symlink, e).as_str());
|
|
|
|
continue;
|
2021-08-08 00:53:34 +12:00
|
|
|
};
|
2021-02-25 01:53:46 +13:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for tree_path in vec_tree_path_to_remove.iter().rev() {
|
2021-08-08 00:53:34 +12:00
|
|
|
model.remove(&model.iter(tree_path).unwrap());
|
2021-02-25 01:53:46 +13:00
|
|
|
}
|
|
|
|
|
2021-11-25 20:36:49 +13:00
|
|
|
clean_invalid_headers(&model, column_color);
|
2021-02-25 01:53:46 +13:00
|
|
|
}
|