using System; using System.Collections.Generic; using System.Diagnostics; using System.Drawing; using System.IO; using System.Linq; using System.Media; using System.Threading.Tasks; using System.Windows.Forms; using PKHeX.Core; using PKHeX.Drawing.PokeSprite; namespace PKHeX.WinForms.Controls { /// /// Orchestrates the movement of slots within the GUI. /// public sealed class SlotChangeManager : IDisposable { public readonly SAVEditor SE; public readonly SlotTrackerImage LastSlot = new(); public readonly DragManager Drag = new(); public SaveDataEditor Env { get; set; } = null!; public readonly List Boxes = new(); public readonly SlotHoverHandler Hover = new(); public SlotChangeManager(SAVEditor se) => SE = se; public void Reset() { Drag.Initialize(); LastSlot.Reset(); } public void MouseEnter(object? sender, EventArgs e) { if (sender is not PictureBox pb) return; bool dataPresent = pb.Image is not null; if (dataPresent) Hover.Start(pb, LastSlot); pb.Cursor = dataPresent ? Cursors.Hand : Cursors.Default; } public void MouseLeave(object? sender, EventArgs e) { Hover.Stop(); } public void MouseClick(object? sender, MouseEventArgs e) { if (sender == null) return; if (!Drag.Info.DragDropInProgress) SE.ClickSlot(sender, e); } public void MouseUp(object? sender, MouseEventArgs e) { if (sender == null) return; if (e.Button == MouseButtons.Left) Drag.Info.LeftMouseIsDown = false; Drag.Info.Source = null; } public void MouseDown(object? sender, MouseEventArgs e) { if (sender == null) return; if (e.Button == MouseButtons.Left) { Drag.Info.LeftMouseIsDown = true; Drag.MouseDownPosition = Cursor.Position; } } public void QueryContinueDrag(object? sender, QueryContinueDragEventArgs e) { if (sender == null) return; if (e.Action != DragAction.Cancel && e.Action != DragAction.Drop) return; Drag.Info.LeftMouseIsDown = false; Drag.Info.DragDropInProgress = false; } public void DragEnter(object? sender, DragEventArgs e) { if (sender == null) return; if ((e.AllowedEffect & DragDropEffects.Copy) != 0) // external file e.Effect = DragDropEffects.Copy; else if (e.Data != null) // within e.Effect = DragDropEffects.Move; if (Drag.Info.DragDropInProgress) Drag.SetCursor(((Control)sender).FindForm(), Drag.Info.Cursor); } private static SlotViewInfo GetSlotInfo(T pb) where T : Control { var view = WinFormsUtil.FindFirstControlOfType>(pb); if (view == null) throw new InvalidCastException("Unable to find View Parent"); var src = view.GetSlotData(pb); return new SlotViewInfo(src, view); } public void MouseMove(object? sender, MouseEventArgs e) { if (sender == null) return; if (!Drag.CanStartDrag) return; // Abort if there is no Pokemon in the given slot. PictureBox pb = (PictureBox)sender; if (pb.Image == null) return; bool encrypt = Control.ModifierKeys == Keys.Control; HandleMovePKM(pb, encrypt); } public void DragDrop(object? sender, DragEventArgs e) { if (sender == null) return; PictureBox pb = (PictureBox)sender; var info = GetSlotInfo(pb); if (!info.CanWriteTo() || Drag.Info.Source?.CanWriteTo() == false) { SystemSounds.Asterisk.Play(); e.Effect = DragDropEffects.Copy; Drag.Reset(); return; } var mod = SlotUtil.GetDropModifier(); Drag.Info.Destination = info; HandleDropPKM(pb, e, mod); } private void HandleMovePKM(PictureBox pb, bool encrypt) { // Create a temporary PKM file to perform a drag drop operation. // Set flag to prevent re-entering. Drag.Info.DragDropInProgress = true; // Prepare Data Drag.Info.Source = GetSlotInfo(pb); Drag.Info.Destination = null; // Make a new file name based off the PID string newfile = CreateDragDropPKM(pb, encrypt, out bool external); // drop finished, clean up Drag.Info.Source = null; Drag.Reset(); Drag.ResetCursor(pb.FindForm()); // Browser apps need time to load data since the file isn't moved to a location on the user's local storage. // Tested 10ms -> too quick, 100ms was fine. 500ms should be safe? // Keep it to 20 seconds; Discord upload only stores the file path until you click Upload. int delay = external ? 20_000 : 0; DeleteAsync(newfile, delay); if (Drag.Info.DragIsParty) SE.SetParty(); } private async void DeleteAsync(string path, int delay) { await Task.Delay(delay).ConfigureAwait(true); if (!File.Exists(path) || Drag.Info.CurrentPath == path) return; try { File.Delete(path); } catch (Exception ex) { Debug.WriteLine(ex.Message); } } private string CreateDragDropPKM(PictureBox pb, bool encrypt, out bool external) { // Make File var pk = Drag.Info.Source!.ReadCurrent(); string newfile = FileUtil.GetPKMTempFileName(pk, encrypt); try { var data = encrypt ? pk.EncryptedPartyData : pk.DecryptedPartyData; external = TryMakeDragDropPKM(pb, data, newfile); } // Tons of things can happen with drag & drop; don't try to handle things, just indicate failure. catch (Exception x) { WinFormsUtil.Error("Drag && Drop Error", x); external = false; } return newfile; } private bool TryMakeDragDropPKM(PictureBox pb, byte[] data, string newfile) { File.WriteAllBytes(newfile, data); var img = (Bitmap)pb.Image; Drag.SetCursor(pb.FindForm(), new Cursor(img.GetHicon())); Hover.Stop(); pb.Image = null; pb.BackgroundImage = SpriteUtil.Spriter.Drag; // Thread Blocks on DoDragDrop Drag.Info.CurrentPath = newfile; var result = pb.DoDragDrop(new DataObject(DataFormats.FileDrop, new[] { newfile }), DragDropEffects.Copy); var external = Drag.Info.Destination == null || result != DragDropEffects.Link; if (external || Drag.Info.SameLocation) // not dropped to another box slot, restore img { pb.Image = img; pb.BackgroundImage = LastSlot.OriginalBackground; Drag.ResetCursor(pb.FindForm()); return external; } if (result == DragDropEffects.Copy) // viewed in tabs or cloned { if (Drag.Info.Destination == null) // apply 'view' highlight Env.Slots.Get(Drag.Info.Source!.Slot); return false; } return true; } private void HandleDropPKM(PictureBox pb, DragEventArgs? e, DropModifier mod) { if (e?.Data?.GetData(DataFormats.FileDrop) is not string[] {Length: not 0} files) { Drag.Reset(); return; } if (Directory.Exists(files[0])) // folder { SE.LoadBoxes(out string _, files[0]); Drag.Reset(); return; } e.Effect = mod == DropModifier.Clone ? DragDropEffects.Copy : DragDropEffects.Link; // file if (Drag.Info.SameLocation) { e.Effect = DragDropEffects.Link; return; } var dest = Drag.Info.Destination; if (Drag.Info.Source == null) // external source { bool badDest = !dest!.CanWriteTo(); if (!TryLoadFiles(files, e, badDest)) WinFormsUtil.Alert(MessageStrings.MsgSaveSlotBadData); } else if (!TrySetPKMDestination(pb, mod)) { WinFormsUtil.Alert(MessageStrings.MsgSaveSlotEmpty); } Drag.Reset(); } /// /// Tries to load the input /// /// Files to load /// Args /// Destination slot disallows eggs/blanks /// True if loaded private bool TryLoadFiles(IReadOnlyList files, DragEventArgs e, bool badDest) { if (files.Count == 0) return false; var sav = Drag.Info.Destination!.View.SAV; var path = files[0]; var temp = FileUtil.GetSingleFromPath(path, sav); if (temp == null) { Drag.RequestDD(this, e); // pass thru return true; // treat as handled } var pk = EntityConverter.ConvertToType(temp, sav.PKMType, out var result); if (pk == null) { var c = result.GetDisplayString(temp, sav.PKMType); WinFormsUtil.Error(c); Debug.WriteLine(c); return false; } if (badDest && (pk.Species == 0 || pk.IsEgg)) return false; if (sav is ILangDeviantSave il && EntityConverter.IsIncompatibleGB(temp, il.Japanese, pk.Japanese)) { var str = EntityConverterResult.IncompatibleLanguageGB.GetIncompatibleGBMessage(pk, il.Japanese); WinFormsUtil.Error(str); Debug.WriteLine(str); return false; } var errata = sav.EvaluateCompatibility(pk); if (errata.Count > 0) { string concat = string.Join(Environment.NewLine, errata); if (DialogResult.Yes != WinFormsUtil.Prompt(MessageBoxButtons.YesNo, concat, MessageStrings.MsgContinue)) { Debug.WriteLine(result.GetDisplayString(temp, sav.PKMType)); Debug.WriteLine(concat); return false; } } Env.Slots.Set(Drag.Info.Destination!.Slot, pk); Debug.WriteLine(result.GetDisplayString(temp, sav.PKMType)); return true; } private bool TrySetPKMDestination(PictureBox pb, DropModifier mod) { var info = Drag.Info; var pk = info.Source!.ReadCurrent(); var msg = Drag.Info.Destination!.CanWriteTo(pk); if (msg != WriteBlockedMessage.None) return false; if (Drag.Info.Source != null) TrySetPKMSource(pb, mod); // Copy from temp to destination slot. var type = info.DragIsSwap ? SlotTouchType.Swap : SlotTouchType.Set; Env.Slots.Set(info.Destination!.Slot, pk, type); Drag.ResetCursor(pb.FindForm()); return true; } private bool TrySetPKMSource(PictureBox sender, DropModifier mod) { var info = Drag.Info; if (info.Destination == null || mod == DropModifier.Clone) return false; if (sender.Image == null || mod == DropModifier.Overwrite) { Env.Slots.Delete(info.Source!.Slot); return true; } var type = info.DragIsSwap ? SlotTouchType.Swap : SlotTouchType.Set; var pk = info.Destination.ReadCurrent(); Env.Slots.Set(Drag.Info.Source!.Slot, pk, type); return true; } // Utility public void SwapBoxes(int index, int other, SaveFile SAV) { if (index == other) return; SAV.SwapBox(index, other); UpdateBoxViewAtBoxIndexes(index, other); } public void Dispose() { Hover.Dispose(); SE.Dispose(); LastSlot.OriginalBackground?.Dispose(); LastSlot.CurrentBackground?.Dispose(); } private void UpdateBoxViewAtBoxIndexes(params int[] boxIndexes) { foreach (var box in Boxes) { var current = box.CurrentBox; if (!boxIndexes.Contains(current)) continue; box.ResetSlots(); box.ResetBoxNames(current); } } } }