Add some finishing touches

* Changed save file dialog filter options * Added some comments * Added some cleanup if the plugin needs to be unloaded
arbrown · Jul 26, 2016 · 5a13a89 · 5a13a89
1 parent 4599b68
commit 5a13a89
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/KPUsage/KPUsagePlugin.cs b/KPUsage/KPUsagePlugin.cs
@@ -32,6 +32,16 @@ public override bool Initialize(IPluginHost host)
             return true;
         }
 
+        public override void Terminate()
+        {
+
+            // Clean up menu items
+            exportUsageMenuItem.Click -= ExportUsageMenuItem_Click;
+            var menu = this.host.MainWindow.ToolsMenu.DropDownItems;
+            menu.Remove(sep);
+            menu.Remove(exportUsageMenuItem);
+        }
+
         private void ExportUsageMenuItem_Click(object sender, EventArgs e)
         {
             if (!this.host.Database.IsOpen)
@@ -40,16 +50,22 @@ private void ExportUsageMenuItem_Click(object sender, EventArgs e)
                 return;
             }
 
-            var sfDialog = new SaveFileDialog();
+            var sfDialog = new SaveFileDialog()
+            {
+                AddExtension = true,
+                DefaultExt = "csv",
+                Filter = "Comma-Separated Values (*.csv)|*.csv",
+            };
 
             if (sfDialog.ShowDialog() == DialogResult.OK)
             {
                 var dic = new Dictionary<ulong, List<PwEntry>>();
 
+                // Traverse all entries
                 this.host.Database.RootGroup.TraverseTree(TraversalMethod.PreOrder, null,
                     (PwEntry en) => 
                     {
-
+                        // Add entry to multi-dictionary by usage count
                         if (!dic.ContainsKey(en.UsageCount))
                         {
                             var tempList = new List<PwEntry>();
@@ -61,6 +77,7 @@ private void ExportUsageMenuItem_Click(object sender, EventArgs e)
                         return true;
                     });
 
+                // Write out every entry, unsorted
                 using (var file = new StreamWriter(sfDialog.FileName, false))
                 {
                     file.WriteLine("Name, Usage Count");

diff --git a/KPUsage/Properties/AssemblyInfo.cs b/KPUsage/Properties/AssemblyInfo.cs
@@ -8,7 +8,7 @@
 [assembly: AssemblyTitle("KPUsage")]
 [assembly: AssemblyDescription("")]
 [assembly: AssemblyConfiguration("")]
-[assembly: AssemblyCompany("Microsoft")]
+[assembly: AssemblyCompany("")]
 [assembly: AssemblyProduct("KeePass Plugin")]
 [assembly: AssemblyCopyright("Copyright © Drew Brown 2016")]
 [assembly: AssemblyTrademark("")]