C&C Remastered Map Editor

Initial commit of C&C Remastered Map Editor code

CnCTDRAMapEditor/Utility/MRU.cs

@@ -0,0 +1,135 @@
+//
+// Copyright 2020 Electronic Arts Inc.
+//
+// The Command & Conquer Map Editor and corresponding source code is free 
+// software: you can redistribute it and/or modify it under the terms of 
+// the GNU General Public License as published by the Free Software Foundation, 
+// either version 3 of the License, or (at your option) any later version.
+
+// The Command & Conquer Map Editor and corresponding source code is distributed 
+// in the hope that it will be useful, but with permitted additional restrictions 
+// under Section 7 of the GPL. See the GNU General Public License in LICENSE.TXT 
+// distributed with this program. You should have received a copy of the 
+// GNU General Public License along with permitted additional restrictions 
+// with this program. If not, see https://github.com/electronicarts/CnC_Remastered_Collection
+using Microsoft.Win32;
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Windows.Forms;
+
+namespace MobiusEditor.Utility
+{
+    public class MRU
+    {
+        private readonly RegistryKey registryKey;
+        private readonly int maxFiles;
+
+        private readonly List<FileInfo> files = new List<FileInfo>();
+
+        private readonly ToolStripMenuItem menu;
+        private readonly ToolStripMenuItem[] fileItems;
+
+        public event EventHandler<FileInfo> FileSelected;
+
+        public MRU(string registryPath, int maxFiles, ToolStripMenuItem menu)
+        {
+            var subKey = Registry.CurrentUser;
+            foreach (var key in registryPath.Split('\\'))
+            {
+                subKey = subKey.CreateSubKey(key, true);
+            }
+            registryKey = subKey.CreateSubKey("MRU");
+
+            this.maxFiles = maxFiles;
+            this.menu = menu;
+            this.menu.DropDownItems.Clear();
+
+            fileItems = new ToolStripMenuItem[maxFiles];
+            for (var i = 0; i < fileItems.Length; ++i)
+            {
+                var fileItem = fileItems[i] = new ToolStripMenuItem();
+                fileItem.Visible = false;
+                menu.DropDownItems.Add(fileItem);
+            }
+
+            LoadMRU();
+            ShowMRU();
+        }
+
+        public void Add(FileInfo file)
+        {
+            files.RemoveAll(f => f.FullName == file.FullName);
+            files.Insert(0, file);
+
+            if (files.Count > maxFiles)
+            {
+                files.RemoveAt(files.Count - 1);
+            }
+
+            SaveMRU();
+            ShowMRU();
+        }
+
+        public void Remove(FileInfo file)
+        {
+            files.RemoveAll(f => f.FullName == file.FullName);
+
+            SaveMRU();
+            ShowMRU();
+        }
+
+        private void LoadMRU()
+        {
+            files.Clear();
+            for (var i = 0; i < maxFiles; ++i)
+            {
+                string fileName = registryKey.GetValue(i.ToString()) as string;
+                if (!string.IsNullOrEmpty(fileName))
+                {
+                    files.Add(new FileInfo(fileName));
+                }
+            }
+        }
+
+        private void SaveMRU()
+        {
+            for (var i = 0; i < files.Count; ++i)
+            {
+                registryKey.SetValue(i.ToString(), files[i].FullName);
+            }
+            for (var i = files.Count; i < maxFiles; ++i)
+            {
+                registryKey.DeleteValue(i.ToString(), false);
+            }
+        }
+
+        private void ShowMRU()
+        {
+            for (var i = 0; i < files.Count; ++i)
+            {
+                var file = files[i];
+                var fileItem = fileItems[i];
+
+                fileItem.Text = string.Format("&{0} {1}", i + 1, file.FullName);
+                fileItem.Tag = file;
+                fileItem.Click -= FileItem_Click;
+                fileItem.Click += FileItem_Click;
+                fileItem.Visible = true;
+            }
+            for (var i = files.Count; i < maxFiles; ++i)
+            {
+                var fileItem = fileItems[i];
+
+                fileItem.Visible = false;
+                fileItem.Click -= FileItem_Click;
+            }
+            menu.Enabled = files.Count > 0;
+        }
+
+        private void FileItem_Click(object sender, EventArgs e)
+        {
+            FileSelected?.Invoke(this, (sender as ToolStripMenuItem).Tag as FileInfo);
+        }
+    }
+}