Implemented Import/Export All Open MyMod Actions.

Author: Frodo45127

3rdparty/builds/qt_rpfm_extensions.lib

@@ -35,6 +35,8 @@ extern "C" void shortcut_collection_init(QWidget* parent, QList<QObject*>* short
     new_action(mymod_menu_actions, "delete_mymod", "Delete Open MyMod", Qt::ShortcutContext::ApplicationShortcut, QKeySequence::listFromString(""), "draw-eraser-delete-objects");
     new_action(mymod_menu_actions, "import_mymod", "Import MyMod", Qt::ShortcutContext::ApplicationShortcut, QKeySequence::listFromString("Ctrl+Alt+I"), "document-import");
     new_action(mymod_menu_actions, "export_mymod", "Export MyMod", Qt::ShortcutContext::ApplicationShortcut, QKeySequence::listFromString("Ctrl+Alt+E"), "document-export");
+    new_action(mymod_menu_actions, "import_all_mymod", "Import All Open MyMods", Qt::ShortcutContext::ApplicationShortcut, QKeySequence::listFromString("Ctrl+Alt+Shift+I"), "document-import");
+    new_action(mymod_menu_actions, "export_all_mymod", "Export All Open MyMods", Qt::ShortcutContext::ApplicationShortcut, QKeySequence::listFromString("Ctrl+Alt+Shift+E"), "document-export");
     mymod_menu_actions->readSettings();
 
     // View Menu actions.

3rdparty/src/qt_rpfm_extensions/src/kshortcuts_dialog.cpp

@@ -103,6 +103,8 @@ mymod_new = New MyMod
 mymod_delete_selected = Delete Selected MyMod
 mymod_import = Import
 mymod_export = Export
+mymod_import_all = Import All Open MyMods
+mymod_export_all = Export All Open MyMods
 
 mymod_name = Name of the Mod:
 mymod_name_default = For example: one_ring_for_me
@@ -224,6 +226,8 @@ tt_mymod_delete_selected = Delete the currently selected MyMod.
 
 tt_mymod_import = Move all of the contents of the MyMod folder into the .pack file. If any files have been removed in the MyMod folder, they will be deleted in the .pack file.
 tt_mymod_export = Move all of the contents from the .pack file into the MyMod folder. If any files have been removed from the .pack, they will be deleted in the MyMod folder.
+tt_mymod_import_all = Run <i>Import</i> on every open MyMod pack.
+tt_mymod_export_all = Run <i>Export</i> on every open MyMod pack.
 
 ## GameSelected menu tips
 

locale/English_en.ftl

@@ -52,6 +52,8 @@ pub unsafe fn set_connections(app_ui: &Rc<AppUI>, slots: &AppUISlots) {
     app_ui.menu_bar_mymod.about_to_show().connect(&slots.mymod_open_menu);
     app_ui.mymod_open_mymod_folder.triggered().connect(&slots.mymod_open_mymod_folder);
     app_ui.mymod_new.triggered().connect(&slots.mymod_new);
+    app_ui.mymod_import_all.triggered().connect(&slots.mymod_import_all);
+    app_ui.mymod_export_all.triggered().connect(&slots.mymod_export_all);
 
     //-----------------------------------------------//
     // `View` menu connections.

rpfm_ui/src/app_ui/connections.rs

@@ -201,6 +201,8 @@ pub struct AppUI {
     //-------------------------------------------------------------------------------//
     mymod_open_mymod_folder: QPtr<QAction>,
     mymod_new: QPtr<QAction>,
+    mymod_import_all: QPtr<QAction>,
+    mymod_export_all: QPtr<QAction>,
 
     mymod_open_pharaoh_dynasties: QPtr<QMenu>,
     mymod_open_pharaoh: QPtr<QMenu>,
@@ -466,6 +468,8 @@ impl AppUI {
         //-----------------------------------------------//
         let mymod_open_mymod_folder = add_action_to_menu(&menu_bar_mymod, shortcuts.as_ref(), "mymod_menu", "open_mymod_folder", "mymod_open_mymod_folder", Some(main_window.static_upcast::<qt_widgets::QWidget>()));
         let mymod_new = add_action_to_menu(&menu_bar_mymod, shortcuts.as_ref(), "mymod_menu", "new_mymod", "mymod_new", Some(main_window.static_upcast::<qt_widgets::QWidget>()));
+        let mymod_import_all = add_action_to_menu(&menu_bar_mymod, shortcuts.as_ref(), "mymod_menu", "import_all_mymod", "mymod_import_all", Some(main_window.static_upcast::<qt_widgets::QWidget>()));
+        let mymod_export_all = add_action_to_menu(&menu_bar_mymod, shortcuts.as_ref(), "mymod_menu", "export_all_mymod", "mymod_export_all", Some(main_window.static_upcast::<qt_widgets::QWidget>()));
 
         menu_bar_mymod.add_separator();
 
@@ -486,6 +490,8 @@ impl AppUI {
         menu_bar_mymod.insert_separator(&mymod_new);
 
         mymod_new.set_enabled(false);
+        mymod_import_all.set_enabled(false);
+        mymod_export_all.set_enabled(false);
 
         mymod_open_pharaoh_dynasties.menu_action().set_visible(false);
         mymod_open_pharaoh.menu_action().set_visible(false);
@@ -686,6 +692,8 @@ impl AppUI {
             //-------------------------------------------------------------------------------//
             mymod_open_mymod_folder,
             mymod_new,
+            mymod_import_all,
+            mymod_export_all,
 
             mymod_open_pharaoh_dynasties,
             mymod_open_pharaoh,
@@ -1338,6 +1346,8 @@ impl AppUI {
 
             // This one too, though we had to deal with it specially later on.
             app_ui.mymod_new.set_enabled(false);
+            app_ui.mymod_import_all.set_enabled(false);
+            app_ui.mymod_export_all.set_enabled(false);
         }
 
         // Otherwise...
@@ -1353,8 +1363,10 @@ impl AppUI {
 
             // If there is a "MyMod" path set in the settings...
             let path = settings_path_buf(MYMOD_BASE_PATH);
-            if path.is_dir() { app_ui.mymod_new.set_enabled(true); }
-            else { app_ui.mymod_new.set_enabled(false); }
+            let mymod_path_ok = path.is_dir();
+            app_ui.mymod_new.set_enabled(mymod_path_ok);
+            app_ui.mymod_import_all.set_enabled(enable && mymod_path_ok);
+            app_ui.mymod_export_all.set_enabled(enable && mymod_path_ok);
         }
 
         // If we are disabling...
@@ -3840,8 +3852,7 @@ impl AppUI {
                         paths_packedfile.push(ContainerPath::File(filtered_path.to_string_lossy().to_string()));
                     }
 
-                    let pack_key = pack_file_contents_ui.pack_key_from_selection_or_first().unwrap_or_default();
-                    let settings = send_ipc_command(Command::GetPackSettings(pack_key), response_extractor!(Response::PackSettings));
+                    let settings = send_ipc_command(Command::GetPackSettings(pack_key.to_string()), response_extractor!(Response::PackSettings));
 
                     let files_to_ignore = settings.setting_text("import_files_to_ignore").map(|files_to_ignore| {
                         if files_to_ignore.is_empty() { vec![] } else {
@@ -3852,7 +3863,7 @@ impl AppUI {
                         }
                     });
 
-                    PackFileContentsUI::add_files(app_ui, pack_file_contents_ui, &paths, &paths_packedfile, files_to_ignore);
+                    PackFileContentsUI::add_files(app_ui, pack_file_contents_ui, &paths, &paths_packedfile, files_to_ignore, Some(pack_key));
                 }
 
                 // If there is no MyMod path configured, report it.
@@ -3868,9 +3879,70 @@ impl AppUI {
     pub unsafe fn export_mymod(
         app_ui: &Rc<Self>,
         pack_file_contents_ui: &Rc<PackFileContentsUI>,
-        paths_to_extract: Option<Vec<ContainerPath>>
+        paths_to_extract: Option<Vec<ContainerPath>>,
+        target_pack_key: Option<&str>,
+    ) {
+        PackFileContentsUI::extract_packed_files(app_ui, pack_file_contents_ui, paths_to_extract, true, target_pack_key)
+    }
+
+    /// This function runs MyMod imports for every open MyMod pack.
+    pub unsafe fn import_all_mymod(
+        app_ui: &Rc<Self>,
+        pack_file_contents_ui: &Rc<PackFileContentsUI>,
+    ) {
+        let mymod_keys = AppUI::open_mymod_pack_keys(pack_file_contents_ui);
+        if mymod_keys.is_empty() {
+            return show_dialog(&app_ui.main_window, "There are no open MyMod packs to import.", false);
+        }
+
+        for pack_key in &mymod_keys {
+            AppUI::import_mymod(app_ui, pack_file_contents_ui, pack_key);
+        }
+    }
+
+    /// This function runs MyMod exports for every open MyMod pack.
+    pub unsafe fn export_all_mymod(
+        app_ui: &Rc<Self>,
+        pack_file_contents_ui: &Rc<PackFileContentsUI>,
     ) {
-        PackFileContentsUI::extract_packed_files(app_ui, pack_file_contents_ui, paths_to_extract, true)
+        let mymod_keys = AppUI::open_mymod_pack_keys(pack_file_contents_ui);
+        if mymod_keys.is_empty() {
+            return show_dialog(&app_ui.main_window, "There are no open MyMod packs to export.", false);
+        }
+
+        for pack_key in &mymod_keys {
+            AppUI::export_mymod(app_ui, pack_file_contents_ui, Some(vec![ContainerPath::Folder("".to_owned())]), Some(pack_key));
+        }
+    }
+
+    /// Collect the pack keys of every open pack that is in MyMod operational mode.
+    pub unsafe fn open_mymod_pack_keys(pack_file_contents_ui: &Rc<PackFileContentsUI>) -> Vec<String> {
+        let model = pack_file_contents_ui.packfile_contents_tree_model();
+        let mut keys = Vec::new();
+        for row in 0..model.row_count_0a() {
+            let item = model.item_1a(row);
+            let root_type = item.data_1a(rpfm_ui_common::ROOT_NODE_TYPE).to_int_0a();
+            if root_type != rpfm_ui_common::ROOT_NODE_TYPE_EDITABLE_PACKFILE && root_type != rpfm_ui_common::ROOT_NODE_TYPE_MYMOD_PACKFILE {
+                continue;
+            }
+
+            let variant = item.data_1a(rpfm_ui_common::ITEM_PACK_KEY);
+            if !variant.is_valid() || variant.is_null() {
+                continue;
+            }
+
+            let key = variant.to_string().to_std_string();
+            if key.is_empty() {
+                continue;
+            }
+
+            let mode = send_ipc_command(Command::GetPackOperationalMode(key.clone()), response_extractor!(Response::OperationalMode));
+            if matches!(mode, OperationalMode::MyMod(..)) {
+                keys.push(key);
+            }
+        }
+
+        keys
     }
 
     /// This function is used to build a snowman.

rpfm_ui/src/app_ui/mod.rs

@@ -120,6 +120,8 @@ pub struct AppUISlots {
     pub mymod_open_menu: QBox<SlotNoArgs>,
     pub mymod_open_mymod_folder: QBox<SlotOfBool>,
     pub mymod_new: QBox<SlotOfBool>,
+    pub mymod_import_all: QBox<SlotOfBool>,
+    pub mymod_export_all: QBox<SlotOfBool>,
 
     //-----------------------------------------------//
     // `View` menu slots.
@@ -789,6 +791,24 @@ impl AppUISlots {
             }
         ));
 
+        // This slot is used for the "Import All Open MyMods" action.
+        let mymod_import_all = SlotOfBool::new(&app_ui.main_window, clone!(
+            app_ui,
+            pack_file_contents_ui => move |_| {
+                info!("Triggering `Import All MyMods` By Slot");
+                AppUI::import_all_mymod(&app_ui, &pack_file_contents_ui);
+            }
+        ));
+
+        // This slot is used for the "Export All Open MyMods" action.
+        let mymod_export_all = SlotOfBool::new(&app_ui.main_window, clone!(
+            app_ui,
+            pack_file_contents_ui => move |_| {
+                info!("Triggering `Export All MyMods` By Slot");
+                AppUI::export_all_mymod(&app_ui, &pack_file_contents_ui);
+            }
+        ));
+
         //-----------------------------------------------//
         // `View` menu logic.
         //-----------------------------------------------//
@@ -1708,6 +1728,8 @@ impl AppUISlots {
             mymod_open_menu,
             mymod_open_mymod_folder,
             mymod_new,
+            mymod_import_all,
+            mymod_export_all,
 
             //-----------------------------------------------//
             // `View` menu slots.

rpfm_ui/src/app_ui/slots.rs

@@ -31,6 +31,8 @@ pub unsafe fn set_tips(app_ui: &Rc<AppUI>) {
     // `MyMod` menu tips.
     //-----------------------------------------------//
     app_ui.mymod_new.set_status_tip(&qtr("tt_mymod_new"));
+    app_ui.mymod_import_all.set_status_tip(&qtr("tt_mymod_import_all"));
+    app_ui.mymod_export_all.set_status_tip(&qtr("tt_mymod_export_all"));
 
     //-----------------------------------------------//
     // `Game Selected` menu tips.

rpfm_ui/src/app_ui/tips.rs

@@ -516,13 +516,17 @@ impl PackFileContentsUI {
         paths: &[PathBuf],
         paths_in_container: &[ContainerPath],
         paths_to_ignore: Option<Vec<PathBuf>>,
+        target_pack_key: Option<&str>,
     ) {
         let window_was_disabled = !app_ui.main_window().is_enabled();
         if !window_was_disabled {
             app_ui.toggle_main_window(false);
         }
 
-        let pack_key = pack_file_contents_ui.pack_key_from_selection_or_first().unwrap_or_default();
+        let pack_key = match target_pack_key {
+            Some(key) => key.to_owned(),
+            None => pack_file_contents_ui.pack_key_from_selection_or_first().unwrap_or_default(),
+        };
         match send_ipc_command_result(Command::AddPackedFiles(pack_key.clone(), paths.to_vec(), paths_in_container.to_vec(), paths_to_ignore), response_extractor!(Response::VecContainerPathOptionString, paths, error)) {
             Ok((paths, error)) => {
                 if !paths.is_empty() {
@@ -662,6 +666,7 @@ impl PackFileContentsUI {
         pack_file_contents_ui: &Rc<Self>,
         paths_to_extract: Option<Vec<ContainerPath>>,
         extract_tables_as_tsv: bool,
+        target_pack_key: Option<&str>,
     ) {
 
         // Get the currently selected paths (and visible) paths, or the ones received from the function.
@@ -670,9 +675,12 @@ impl PackFileContentsUI {
             None => <QPtr<QTreeView> as PackTree>::get_item_types_from_main_treeview_selection(pack_file_contents_ui),
         };
 
-        // Query the selected pack's operational mode for extraction path logic.
-        let selected_pack_key = pack_file_contents_ui.pack_key_from_selection_or_first().unwrap_or_default();
-        let pack_mode = send_ipc_command(Command::GetPackOperationalMode(selected_pack_key), response_extractor!(Response::OperationalMode));
+        // Resolve the pack key to operate on: explicit target, or the selected/first editable pack.
+        let selected_pack_key = match target_pack_key {
+            Some(key) => key.to_owned(),
+            None => pack_file_contents_ui.pack_key_from_selection_or_first().unwrap_or_default(),
+        };
+        let pack_mode = send_ipc_command(Command::GetPackOperationalMode(selected_pack_key.clone()), response_extractor!(Response::OperationalMode));
 
         let extraction_path = match pack_mode {
 
@@ -724,9 +732,8 @@ impl PackFileContentsUI {
         else {
             let mut paths_by_source = BTreeMap::new();
             paths_by_source.insert(DataSource::PackFile, items_to_extract);
-            let pack_key = pack_file_contents_ui.pack_key_from_selection_or_first().unwrap_or_default();
             app_ui.toggle_main_window(false);
-            match send_ipc_command_result_async(Command::ExtractPackedFiles(pack_key, paths_by_source, extraction_path, extract_tables_as_tsv), response_extractor!(Response::StringVecPathBuf, result, _paths)) {
+            match send_ipc_command_result_async(Command::ExtractPackedFiles(selected_pack_key, paths_by_source, extraction_path, extract_tables_as_tsv), response_extractor!(Response::StringVecPathBuf, result, _paths)) {
                 Ok((result, _)) => show_message_info(app_ui.message_widget(), result),
                 Err(error) => show_dialog(app_ui.main_window(), error, false),
             }

rpfm_ui/src/packfile_contents_ui/mod.rs

@@ -784,7 +784,7 @@ impl PackFileContentsSlots {
                                 };
 
                                 app_ui.toggle_main_window(false);
-                                PackFileContentsUI::add_files(&app_ui, &pack_file_contents_ui, &paths, &paths_packedfile, None);
+                                PackFileContentsUI::add_files(&app_ui, &pack_file_contents_ui, &paths, &paths_packedfile, None, None);
                                 app_ui.toggle_main_window(true);
                             }
                         }
@@ -815,7 +815,7 @@ impl PackFileContentsSlots {
                             }
 
                             app_ui.toggle_main_window(false);
-                            PackFileContentsUI::add_files(&app_ui, &pack_file_contents_ui, &paths, &paths_packedfile, None);
+                            PackFileContentsUI::add_files(&app_ui, &pack_file_contents_ui, &paths, &paths_packedfile, None, None);
                             app_ui.toggle_main_window(true);
                         }
                     }
@@ -918,15 +918,15 @@ impl PackFileContentsSlots {
                                             let filtered_path = path.strip_prefix(&assets_folder).unwrap();
                                             paths_packedfile.push(ContainerPath::File(filtered_path.to_string_lossy().to_string()));
                                         }
-                                        PackFileContentsUI::add_files(&app_ui, &pack_file_contents_ui, &paths, &paths_packedfile, None);
+                                        PackFileContentsUI::add_files(&app_ui, &pack_file_contents_ui, &paths, &paths_packedfile, None, None);
                                     }
 
                                     // Otherwise, they are added like normal files.
                                     else if let Some(selection) = pack_file_contents_ui.packfile_contents_tree_view.get_path_from_selection().first() {
                                         let destination_paths = (0..folder_paths.len()).map(|_| ContainerPath::Folder(selection.to_string())).collect::<Vec<_>>();
 
                                         app_ui.toggle_main_window(false);
-                                        PackFileContentsUI::add_files(&app_ui, &pack_file_contents_ui, &folder_paths, &destination_paths, None);
+                                        PackFileContentsUI::add_files(&app_ui, &pack_file_contents_ui, &folder_paths, &destination_paths, None, None);
                                         app_ui.toggle_main_window(true);
                                     }
                                 }
@@ -973,7 +973,7 @@ impl PackFileContentsSlots {
                                 let destination_paths = (0..folder_paths.len()).map(|_| ContainerPath::Folder(selection.to_string())).collect::<Vec<_>>();
 
                                 app_ui.toggle_main_window(false);
-                                PackFileContentsUI::add_files(&app_ui, &pack_file_contents_ui, &folder_paths, &destination_paths, None);
+                                PackFileContentsUI::add_files(&app_ui, &pack_file_contents_ui, &folder_paths, &destination_paths, None, None);
                                 app_ui.toggle_main_window(true);
                             }
                         }
@@ -1115,7 +1115,7 @@ impl PackFileContentsSlots {
             app_ui,
             pack_file_contents_ui => move |_| {
                 info!("Triggering `Extract` By Slot");
-                PackFileContentsUI::extract_packed_files(&app_ui, &pack_file_contents_ui, None, true);
+                PackFileContentsUI::extract_packed_files(&app_ui, &pack_file_contents_ui, None, true, None);
             }
         ));
 
@@ -2109,7 +2109,7 @@ impl PackFileContentsSlots {
         let context_menu_mymod_export = SlotOfBool::new(&pack_file_contents_ui.packfile_contents_dock_widget, clone!(
             app_ui,
             pack_file_contents_ui => move |_| {
-                AppUI::export_mymod(&app_ui, &pack_file_contents_ui, Some(vec![ContainerPath::Folder("".to_owned())]));
+                AppUI::export_mymod(&app_ui, &pack_file_contents_ui, Some(vec![ContainerPath::Folder("".to_owned())]), None);
             }
         ));