using System; using System.Collections.Generic; using System.Drawing; using System.IO; using System.Linq; using System.Reflection; using System.ComponentModel; using System.Windows.Forms; using BizHawk.Client.ApiHawk; using BizHawk.Client.Common; using BizHawk.Client.EmuHawk.CoreExtensions; using BizHawk.Common; using BizHawk.Common.ReflectionExtensions; using BizHawk.Emulation.Common; namespace BizHawk.Client.EmuHawk { public class ToolManager { private readonly Form _owner; // TODO: merge ToolHelper code where logical // For instance, add an IToolForm property called UsesCheats, so that a UpdateCheatRelatedTools() method can update all tools of this type // Also a UsesRam, and similar method private readonly List _tools = new List(); /// /// Initializes a new instance of the class. /// /// Form that handle the ToolManager public ToolManager(Form owner) { _owner = owner; } /// /// Loads the tool dialog T (T must implements ) , if it does not exist it will be created, if it is already open, it will be focused /// This method should be used only if you can't use the generic one /// /// Type of tool you want to load /// Define if the tool form has to get the focus or not (Default is true) /// An instantiated /// Raised if can't cast into IToolForm internal IToolForm Load(Type toolType, bool focus = true) { if (!typeof(IToolForm).IsAssignableFrom(toolType)) { throw new ArgumentException($"Type {toolType.Name} does not implement {nameof(IToolForm)}."); } // The type[] in parameter is used to avoid an ambiguous name exception MethodInfo method = GetType().GetMethod("Load", new Type[] { typeof(bool) }).MakeGenericMethod(toolType); return (IToolForm)method.Invoke(this, new object[] { focus }); } /// /// Loads the tool dialog T (T must implement ) , if it does not exist it will be created, if it is already open, it will be focused /// /// Type of tool you want to load /// Define if the tool form has to get the focus or not (Default is true) /// An instantiated public T Load(bool focus = true) where T : class, IToolForm { return Load("", focus); } /// /// Loads the tool dialog T (T must implement ) , if it does not exist it will be created, if it is already open, it will be focused /// /// Type of tool you want to load /// Path to the .dll of the external tool /// Define if the tool form has to get the focus or not (Default is true) /// An instantiated public T Load(string toolPath, bool focus = true) where T : class, IToolForm { bool isExternal = typeof(T) == typeof(IExternalToolForm); if (!IsAvailable() && !isExternal) { return null; } T existingTool; if (isExternal) { existingTool = (T)_tools.FirstOrDefault(t => t is T && t.GetType().Assembly.Location == toolPath); } else { existingTool = (T)_tools.FirstOrDefault(t => t is T); } if (existingTool != null) { if (existingTool.IsDisposed) { _tools.Remove(existingTool); } else { if (focus) { existingTool.Show(); existingTool.Focus(); } return existingTool; } } IToolForm newTool = CreateInstance(toolPath); if (newTool == null) { return null; } if (newTool is Form form) { form.Owner = _owner; } if (isExternal) { ApiInjector.UpdateApis(GlobalWin.ApiProvider, newTool); } ServiceInjector.UpdateServices(Global.Emulator.ServiceProvider, newTool); string toolType = typeof(T).ToString(); // auto settings if (newTool is IToolFormAutoConfig tool) { if (!Global.Config.CommonToolSettings.TryGetValue(toolType, out var settings)) { settings = new ToolDialogSettings(); Global.Config.CommonToolSettings[toolType] = settings; } AttachSettingHooks(tool, settings); } // custom settings if (HasCustomConfig(newTool)) { if (!Global.Config.CustomToolSettings.TryGetValue(toolType, out var settings)) { settings = new Dictionary(); Global.Config.CustomToolSettings[toolType] = settings; } InstallCustomConfig(newTool, settings); } newTool.Restart(); if (OSTailoredCode.IsUnixHost && newTool is RamSearch) { // the mono winforms implementation is buggy, skip to the return statement and call Show in MainForm instead } else { newTool.Show(); } return (T)newTool; } public void AutoLoad() { var genericSettings = Global.Config.CommonToolSettings .Where(kvp => kvp.Value.AutoLoad) .Select(kvp => kvp.Key); var customSettings = Global.Config.CustomToolSettings .Where(list => list.Value.Any(kvp => kvp.Value is ToolDialogSettings settings && settings.AutoLoad)) .Select(kvp => kvp.Key); var typeNames = genericSettings.Concat(customSettings); foreach (var typename in typeNames) { // this type resolution might not be sufficient. more investigation is needed Type t = Type.GetType(typename); if (t == null) { Console.WriteLine("BENIGN: Couldn't find type {0}", typename); } else { Load(t, false); } } } private void RefreshSettings(Form form, ToolStripItemCollection menu, ToolDialogSettings settings, int idx) { ((ToolStripMenuItem)menu[idx + 0]).Checked = settings.SaveWindowPosition; ((ToolStripMenuItem)menu[idx + 1]).Checked = settings.TopMost; ((ToolStripMenuItem)menu[idx + 2]).Checked = settings.FloatingWindow; ((ToolStripMenuItem)menu[idx + 3]).Checked = settings.AutoLoad; form.TopMost = settings.TopMost; // do we need to do this OnShown() as well? form.Owner = settings.FloatingWindow ? null : _owner; } private void AttachSettingHooks(IToolFormAutoConfig tool, ToolDialogSettings settings) { var form = (Form)tool; ToolStripItemCollection dest = null; var oldSize = form.Size; // this should be the right time to grab this size foreach (Control c in form.Controls) { if (c is MenuStrip ms) { foreach (ToolStripMenuItem submenu in ms.Items) { if (submenu.Text.Contains("Settings")) { dest = submenu.DropDownItems; dest.Add(new ToolStripSeparator()); break; } } if (dest == null) { var submenu = new ToolStripMenuItem("&Settings"); ms.Items.Add(submenu); dest = submenu.DropDownItems; } break; } } if (dest == null) { throw new InvalidOperationException($"{nameof(IToolFormAutoConfig)} must have menu to bind to!"); } int idx = dest.Count; dest.Add("Save Window &Position"); dest.Add("Stay on &Top"); dest.Add("&Float from Parent"); dest.Add("&Autoload"); dest.Add("Restore &Defaults"); RefreshSettings(form, dest, settings, idx); if (settings.UseWindowPosition && IsOnScreen(settings.TopLeft)) { form.StartPosition = FormStartPosition.Manual; form.Location = settings.WindowPosition; } if (settings.UseWindowSize) { if (form.FormBorderStyle == FormBorderStyle.Sizable || form.FormBorderStyle == FormBorderStyle.SizableToolWindow) { form.Size = settings.WindowSize; } } form.FormClosing += (o, e) => { if (form.WindowState == FormWindowState.Normal) { settings.Wndx = form.Location.X; settings.Wndy = form.Location.Y; if (settings.Wndx < 0) { settings.Wndx = 0; } if (settings.Wndy < 0) { settings.Wndy = 0; } settings.Width = form.Right - form.Left; // why not form.Size.Width? settings.Height = form.Bottom - form.Top; } }; dest[idx + 0].Click += (o, e) => { bool val = !((ToolStripMenuItem)o).Checked; settings.SaveWindowPosition = val; ((ToolStripMenuItem)o).Checked = val; }; dest[idx + 1].Click += (o, e) => { bool val = !((ToolStripMenuItem)o).Checked; settings.TopMost = val; ((ToolStripMenuItem)o).Checked = val; form.TopMost = val; }; dest[idx + 2].Click += (o, e) => { bool val = !((ToolStripMenuItem)o).Checked; settings.FloatingWindow = val; ((ToolStripMenuItem)o).Checked = val; form.Owner = val ? null : _owner; }; dest[idx + 3].Click += (o, e) => { bool val = !((ToolStripMenuItem)o).Checked; settings.AutoLoad = val; ((ToolStripMenuItem)o).Checked = val; }; dest[idx + 4].Click += (o, e) => { settings.RestoreDefaults(); RefreshSettings(form, dest, settings, idx); form.Size = oldSize; form.GetType() .GetMethodsWithAttrib(typeof(RestoreDefaultsAttribute)) .FirstOrDefault() ?.Invoke(form, new object[0]); }; } private static bool HasCustomConfig(IToolForm tool) { return tool.GetType().GetPropertiesWithAttrib(typeof(ConfigPersistAttribute)).Any(); } private static void InstallCustomConfig(IToolForm tool, Dictionary data) { Type type = tool.GetType(); var props = type.GetPropertiesWithAttrib(typeof(ConfigPersistAttribute)).ToList(); if (props.Count == 0) { return; } foreach (var prop in props) { if (data.TryGetValue(prop.Name, out var val)) { if (val is string str && prop.PropertyType != typeof(string)) { // if a type has a TypeConverter, and that converter can convert to string, // that will be used in place of object markup by JSON.NET // but that doesn't work with $type metadata, and JSON.NET fails to fall // back on regular object serialization when needed. so try to undo a TypeConverter // operation here var converter = TypeDescriptor.GetConverter(prop.PropertyType); val = converter.ConvertFromString(null, System.Globalization.CultureInfo.InvariantCulture, str); } else if (!(val is bool) && prop.PropertyType.IsPrimitive) { // numeric constants are similarly hosed val = Convert.ChangeType(val, prop.PropertyType, System.Globalization.CultureInfo.InvariantCulture); } prop.SetValue(tool, val, null); } } ((Form)tool).FormClosing += (o, e) => SaveCustomConfig(tool, data, props); } private static void SaveCustomConfig(IToolForm tool, Dictionary data, List props) { data.Clear(); foreach (var prop in props) { data.Add(prop.Name, prop.GetValue(tool, BindingFlags.GetProperty, Type.DefaultBinder, null, System.Globalization.CultureInfo.InvariantCulture)); } } /// /// Determines whether a given IToolForm is already loaded /// /// Type of tool to check public bool IsLoaded() where T : IToolForm { var existingTool = _tools.FirstOrDefault(t => t is T); if (existingTool != null) { return !existingTool.IsDisposed; } return false; } public static bool IsOnScreen(Point topLeft) { return Screen.AllScreens.Any( screen => screen.WorkingArea.Contains(topLeft)); } /// /// Returns true if an instance of T exists /// /// Type of tool to check public bool Has() where T : IToolForm { return _tools.Any(t => t is T && !t.IsDisposed); } /// /// Gets the instance of T, or creates and returns a new instance /// /// Type of tool to get public IToolForm Get() where T : class, IToolForm { return Load(false); } public IEnumerable AvailableTools => Assembly .GetAssembly(typeof(IToolForm)) .GetTypes() .Where(t => typeof(IToolForm).IsAssignableFrom(t)) .Where(t => !t.IsInterface) .Where(IsAvailable); public void UpdateBefore() { var beforeList = _tools.Where(t => t.UpdateBefore); foreach (var tool in beforeList) { if (!tool.IsDisposed || (tool is RamWatch && Global.Config.DisplayRamWatch)) // RAM Watch hack, on screen display should run even if RAM Watch is closed { tool.UpdateValues(); } } foreach (var tool in _tools) { if (!tool.IsDisposed) { tool.NewUpdate(ToolFormUpdateType.PreFrame); } } } public void UpdateAfter() { var afterList = _tools.Where(t => !t.UpdateBefore); foreach (var tool in afterList) { if (!tool.IsDisposed || (tool is RamWatch && Global.Config.DisplayRamWatch)) // RAM Watch hack, on screen display should run even if RAM Watch is closed { tool.UpdateValues(); } } foreach (var tool in _tools) { if (!tool.IsDisposed) { tool.NewUpdate(ToolFormUpdateType.PostFrame); } } } /// /// Calls UpdateValues() on an instance of T, if it exists /// /// Type of tool to update public void UpdateValues() where T : IToolForm { var tool = _tools.FirstOrDefault(t => t is T); if (tool != null) { if (!tool.IsDisposed || (tool is RamWatch && Global.Config.DisplayRamWatch)) // RAM Watch hack, on screen display should run even if RAM Watch is closed { tool.UpdateValues(); } } } public void Restart() { // If Cheat tool is loaded, restarting will restart the list too anyway if (!Has()) { Global.CheatList.NewList(GenerateDefaultCheatFilename(), autosave: true); } var unavailable = new List(); foreach (var tool in _tools) { if (ServiceInjector.IsAvailable(Global.Emulator.ServiceProvider, tool.GetType())) { ServiceInjector.UpdateServices(Global.Emulator.ServiceProvider, tool); if ((tool.IsHandleCreated && !tool.IsDisposed) || tool is RamWatch) // Hack for RAM Watch - in display watches mode it wants to keep running even closed, it will handle disposed logic { if (tool is IExternalToolForm) ApiInjector.UpdateApis(GlobalWin.ApiProvider, tool); tool.Restart(); } } else { unavailable.Add(tool); ServiceInjector.ClearServices(tool); // the services of the old emulator core are no longer valid on the tool } } foreach (var tool in unavailable) { tool.Close(); _tools.Remove(tool); } } /// /// Calls Restart() on an instance of T, if it exists /// /// Type of tool to restart public void Restart() where T : IToolForm { var tool = _tools.FirstOrDefault(t => t is T); tool?.Restart(); } /// /// Runs AskSave on every tool dialog, false is returned if any tool returns false /// public bool AskSave() { if (Global.Config.SuppressAskSave) // User has elected to not be nagged { return true; } return _tools .Select(tool => tool.AskSaveChanges()) .All(result => result); } /// /// Calls AskSave() on an instance of T, if it exists, else returns true /// The caller should interpret false as cancel and will back out of the action that invokes this call /// /// Type of tool public bool AskSave() where T : IToolForm { if (Global.Config.SuppressAskSave) // User has elected to not be nagged { return true; } var tool = _tools.FirstOrDefault(t => t is T); return tool != null && tool.AskSaveChanges(); } /// /// If T exists, this call will close the tool, and remove it from memory /// /// Type of tool to close public void Close() where T : IToolForm { var tool = _tools.FirstOrDefault(t => t is T); if (tool != null) { tool.Close(); _tools.Remove(tool); } } public void Close(Type toolType) { var tool = _tools.FirstOrDefault(toolType.IsInstanceOfType); if (tool != null) { tool.Close(); _tools.Remove(tool); } } public void Close() { _tools.ForEach(t => t.Close()); _tools.Clear(); } /// /// Create a new instance of an IToolForm and return it /// /// Type of tool you want to create /// Path .dll for an external tool /// New instance of an IToolForm private IToolForm CreateInstance(string dllPath) where T : IToolForm { return CreateInstance(typeof(T), dllPath); } /// /// Create a new instance of an IToolForm and return it /// /// Type of tool you want to create /// Path dll for an external tool /// New instance of an IToolForm private IToolForm CreateInstance(Type toolType, string dllPath) { IToolForm tool; // Specific case for custom tools // TODO: Use AppDomain in order to be able to unload the assembly // Hard stuff as we need a proxy object that inherit from MarshalByRefObject. if (toolType == typeof(IExternalToolForm)) { if (MessageBox.Show( "Are you sure want to load this external tool?\r\nAccept ONLY if you trust the source and if you know what you're doing. In any other case, choose no.", "Confirm loading", MessageBoxButtons.YesNo, MessageBoxIcon.Question) == DialogResult.Yes) { try { tool = Activator.CreateInstanceFrom(dllPath, "BizHawk.Client.EmuHawk.CustomMainForm").Unwrap() as IExternalToolForm; if (tool == null) { MessageBox.Show($"It seems that the object CustomMainForm does not implement {nameof(IExternalToolForm)}. Please review the code.", "No, no, no. Wrong Way !", MessageBoxButtons.OK, MessageBoxIcon.Exclamation); return null; } } catch (MissingMethodException) { MessageBox.Show("It seems that the object CustomMainForm does not have a public default constructor. Please review the code.", "No, no, no. Wrong Way !", MessageBoxButtons.OK, MessageBoxIcon.Exclamation); return null; } catch (TypeLoadException) { MessageBox.Show("It seems that the object CustomMainForm does not exists. Please review the code.", "No, no, no. Wrong Way !", MessageBoxButtons.OK, MessageBoxIcon.Exclamation); return null; } } else { return null; } } else { tool = (IToolForm)Activator.CreateInstance(toolType); } // Add to our list of tools _tools.Add(tool); return tool; } public void UpdateToolsBefore(bool fromLua = false) { if (Has()) { if (!fromLua) { LuaConsole.LuaImp.StartLuaDrawing(); } } UpdateBefore(); } public void UpdateToolsAfter(bool fromLua = false) { if (!fromLua && Has()) { LuaConsole.ResumeScripts(true); } UpdateAfter(); if (Has()) { if (!fromLua) { LuaConsole.LuaImp.EndLuaDrawing(); } } } public void FastUpdateBefore() { var beforeList = _tools.Where(t => t.UpdateBefore); foreach (var tool in beforeList) { if (!tool.IsDisposed || (tool is RamWatch && Global.Config.DisplayRamWatch)) // RAM Watch hack, on screen display should run even if RAM Watch is closed { tool.FastUpdate(); } } } public void FastUpdateAfter(bool fromLua = false) { if (!fromLua && Global.Config.RunLuaDuringTurbo && Has()) { LuaConsole.ResumeScripts(true); } var afterList = _tools.Where(t => !t.UpdateBefore); foreach (var tool in afterList) { if (!tool.IsDisposed || (tool is RamWatch && Global.Config.DisplayRamWatch)) // RAM Watch hack, on screen display should run even if RAM Watch is closed { tool.FastUpdate(); } } if (Global.Config.RunLuaDuringTurbo && Has()) { LuaConsole.LuaImp.EndLuaDrawing(); } } private static readonly Lazy> LazyAsmTypes = new Lazy>(() => Assembly.GetAssembly(typeof(ToolManager)) // Confining the search to only EmuHawk, for now at least, we may want to broaden for ApiHawk one day .GetTypes() .Select(t => t.AssemblyQualifiedName) .ToList()); public bool IsAvailable(Type tool) { if (!ServiceInjector.IsAvailable(Global.Emulator.ServiceProvider, tool) || !LazyAsmTypes.Value.Contains(tool.AssemblyQualifiedName)) // not a tool { return false; } ToolAttribute attr = tool.GetCustomAttributes(false).OfType().SingleOrDefault(); if (attr == null) { return true; // no ToolAttribute on given type -> assumed all supported } var displayName = Global.Emulator.DisplayName(); var systemId = Global.Emulator.SystemId; return !attr.UnsupportedCores.Contains(displayName) // not unsupported && (!attr.SupportedSystems.Any() || attr.SupportedSystems.Contains(systemId)); // supported (no supported list -> assumed all supported) } public bool IsAvailable() => IsAvailable(typeof(T)); // Note: Referencing these properties creates an instance of the tool and persists it. They should be referenced by type if this is not desired #region Tools private T GetTool() where T : class, IToolForm, new() { T tool = _tools.OfType().FirstOrDefault(); if (tool != null) { if (!tool.IsDisposed) { return tool; } _tools.Remove(tool); } tool = new T(); _tools.Add(tool); return tool; } public RamWatch RamWatch => GetTool(); public RamSearch RamSearch => GetTool(); public Cheats Cheats => GetTool(); public HexEditor HexEditor => GetTool(); public VirtualpadTool VirtualPad => GetTool(); public SNESGraphicsDebugger SNESGraphicsDebugger => GetTool(); public LuaConsole LuaConsole => GetTool(); public TAStudio TAStudio => GetTool(); #endregion #region Specialized Tool Loading Logic public void LoadRamWatch(bool loadDialog) { if (IsLoaded()) { return; } Load(); if (IsAvailable()) // Just because we attempted to load it, doesn't mean it was, the current core may not have the correct dependencies { if (Global.Config.RecentWatches.AutoLoad && !Global.Config.RecentWatches.Empty) { RamWatch.LoadFileFromRecent(Global.Config.RecentWatches.MostRecent); } if (!loadDialog) { Get().Close(); } } } public void LoadGameGenieEc() { if (IsAvailable()) { Load(); } } #endregion public static string GenerateDefaultCheatFilename() { var pathEntry = Global.Config.PathEntries[Global.Game.System, "Cheats"] ?? Global.Config.PathEntries[Global.Game.System, "Base"]; var path = PathManager.MakeAbsolutePath(pathEntry.Path, Global.Game.System); var f = new FileInfo(path); if (f.Directory != null && f.Directory.Exists == false) { f.Directory.Create(); } return Path.Combine(path, $"{PathManager.FilesystemSafeName(Global.Game)}.cht"); } } }