Add right click menu options to remove all loaded or unloaded archives

clayne · Sep 17, 2022 · 26efc16 · 26efc16
1 parent 44f5036
commit 26efc16
Show file tree

Hide file tree

Showing 3 changed files with 85 additions and 23 deletions.
diff --git a/BSA Browser/BSABrowser.Designer.cs b/BSA Browser/BSABrowser.Designer.cs
diff --git a/BSA Browser/BSABrowser.cs b/BSA Browser/BSABrowser.cs
@@ -1,4 +1,14 @@
-using System;
+using BSA_Browser.Classes;
+using BSA_Browser.Dialogs;
+using BSA_Browser.Enums;
+using BSA_Browser.Extensions;
+using BSA_Browser.Properties;
+using BSA_Browser.Sorting;
+using BSA_Browser.Tools;
+using SharpBSABA2;
+using SharpBSABA2.BA2Util;
+using SharpBSABA2.Enums;
+using System;
 using System.Collections.Generic;
 using System.Collections.Specialized;
 using System.ComponentModel;
@@ -12,16 +22,6 @@
 using System.Threading;
 using System.Threading.Tasks;
 using System.Windows.Forms;
-using BSA_Browser.Classes;
-using BSA_Browser.Dialogs;
-using BSA_Browser.Enums;
-using BSA_Browser.Extensions;
-using BSA_Browser.Properties;
-using BSA_Browser.Sorting;
-using BSA_Browser.Tools;
-using SharpBSABA2;
-using SharpBSABA2.BA2Util;
-using SharpBSABA2.Enums;
 
 namespace BSA_Browser
 {
@@ -69,6 +69,7 @@ public BSABrowser()
             };
             archiveNode.Loaded = true;
             tvFolders.Nodes.Add(archiveNode);
+            tvFolders.ContextMenu = foldersContextMenu;
 
             lvFiles.ContextMenu = contextMenu1;
 
@@ -1321,6 +1322,29 @@ private void unloadedRemoveMenuItem_Click(object sender, EventArgs e)
 
         #endregion
 
+        #region foldersContextMenu
+
+        private void removeLoadedMenuItem_Click(object sender, EventArgs e)
+        {
+            this.CloseArchives(true);
+        }
+
+        private void removeUnloadedMenuItem_Click(object sender, EventArgs e)
+        {
+            for (int i = tvFolders.Nodes.Count - 1; i > 0; i--)
+            {
+                if (!(tvFolders.Nodes[i] as ArchiveNode).Loaded)
+                    this.CloseArchive((ArchiveNode)tvFolders.Nodes[i]);
+            }
+
+            if (tvFolders.Nodes.Count == 1)
+                this.ClearList();
+            else
+                this.DoSearch();
+        }
+
+        #endregion
+
         /// <summary>
         /// Opens the given archive, adding it to the <see cref="tvFolders"/> and making it browsable, then returns containing <see cref="ArchiveNode"/>.
         /// </summary>
@@ -1588,15 +1612,19 @@ private void CloseArchive(ArchiveNode archiveNode)
         /// <summary>
         /// Closes all open archives, clearing the <see cref="TreeView"/>.
         /// </summary>
-        private void CloseArchives()
+        private void CloseArchives(bool loadedOnly = false)
         {
-            this.ClearList();
+            if (!loadedOnly || this.SelectedArchiveNode.Loaded)
+                this.ClearList();
 
             _pauseFiltering = true;
             for (int i = tvFolders.Nodes.Count - 1; i > 0; i--)
             {
                 var node = (ArchiveNode)tvFolders.Nodes[i];
 
+                if (loadedOnly && !node.Loaded)
+                    continue;
+
                 if (node.Loaded)
                 {
                     node.Archive.Close();
@@ -1610,7 +1638,11 @@ private void CloseArchives()
             GC.Collect();
 
             // Disable buttons
-            btnExtractAllFolders.Enabled = btnExtractAll.Enabled = false;
+            if (tvFolders.GetNodeCount(false) == 1)
+            {
+                btnExtractAllFolders.Enabled = false;
+                btnExtractAll.Enabled = false;
+            }
         }
 
         /// <summary>

diff --git a/BSA Browser/BSABrowser.resx b/BSA Browser/BSABrowser.resx
@@ -168,12 +168,18 @@
   <metadata name="archiveContextMenu.TrayLocation" type="System.Drawing.Point, System.Drawing, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a">
     <value>421, 17</value>
   </metadata>
+  <metadata name="toolTip1.TrayLocation" type="System.Drawing.Point, System.Drawing, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a">
+    <value>585, 17</value>
+  </metadata>
   <metadata name="label2.Locked" type="System.Boolean, mscorlib, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089">
     <value>True</value>
   </metadata>
   <metadata name="unloadedArchiveContextMenu.TrayLocation" type="System.Drawing.Point, System.Drawing, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a">
     <value>948, 17</value>
   </metadata>
+  <metadata name="foldersContextMenu.TrayLocation" type="System.Drawing.Point, System.Drawing, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a">
+    <value>1165, 17</value>
+  </metadata>
   <metadata name="$this.Locked" type="System.Boolean, mscorlib, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089">
     <value>True</value>
   </metadata>