Merge pull request #1657 from pavelbraginskiy/drag-drop

Fix cursor when drag-dropping in MML
MegaMek · Nov 10, 2024 · 3cdebe9 · 3cdebe9
2 parents 04fbb9f + b99df8a
commit 3cdebe9
Show file tree

Hide file tree

Showing 4 changed files with 66 additions and 74 deletions.
diff --git a/megameklab/src/megameklab/ui/MegaMekLabMainUI.java b/megameklab/src/megameklab/ui/MegaMekLabMainUI.java
@@ -24,7 +24,7 @@
 import megameklab.ui.util.ExitOnWindowClosingListener;
 import megameklab.ui.util.RefreshListener;
 import megameklab.util.CConfig;
-import megameklab.util.MMLFileDropTarget;
+import megameklab.util.MMLFileDropTransferHandler;
 
 import javax.swing.*;
 import java.awt.*;
@@ -47,7 +47,7 @@ protected void finishSetup() {
         setJMenuBar(mmlMenuBar);
         reloadTabs();
         refreshAll();
-        this.setDropTarget(new MMLFileDropTarget(this));
+        this.setTransferHandler(new MMLFileDropTransferHandler(this));
     }
 
     protected void setSizeAndLocation() {

diff --git a/megameklab/src/megameklab/ui/StartupGUI.java b/megameklab/src/megameklab/ui/StartupGUI.java
@@ -29,7 +29,7 @@
 import megameklab.ui.dialog.UiLoader;
 import megameklab.ui.util.ExitOnWindowClosingListener;
 import megameklab.util.CConfig;
-import megameklab.util.MMLFileDropTarget;
+import megameklab.util.MMLFileDropTransferHandler;
 import megameklab.util.UnitUtil;
 
 import javax.swing.*;
@@ -237,7 +237,7 @@ private void initComponents() {
         frame.pack();
         frame.setLocationRelativeTo(null);
         frame.setVisible(true);
-        this.setDropTarget(new MMLFileDropTarget(this));
+        this.setTransferHandler(new MMLFileDropTransferHandler(this));
     }
 
     /**

diff --git a/megameklab/src/megameklab/util/MMLFileDropTarget.java b/megameklab/src/megameklab/util/MMLFileDropTarget.java
diff --git a/megameklab/src/megameklab/util/MMLFileDropTransferHandler.java b/megameklab/src/megameklab/util/MMLFileDropTransferHandler.java
@@ -0,0 +1,62 @@
+package megameklab.util;
+
+import megamek.logging.MMLogger;
+import megameklab.ui.MenuBarOwner;
+
+import javax.swing.*;
+import java.awt.datatransfer.DataFlavor;
+import java.awt.datatransfer.Transferable;
+import java.io.File;
+import java.util.List;
+
+public class MMLFileDropTransferHandler extends TransferHandler {
+    private static final MMLogger logger = MMLogger.create(MMLFileDropTransferHandler.class);
+    private final MenuBarOwner owner;
+
+    public MMLFileDropTransferHandler(MenuBarOwner owner) {
+        this.owner = owner;
+    }
+
+    @Override
+    public boolean canImport(TransferSupport support) {
+        if (!support.isDrop()) {
+            return false;
+        }
+        return support.isDataFlavorSupported(DataFlavor.javaFileListFlavor);
+    }
+
+
+    @Override
+    public boolean importData(TransferSupport support) {
+        if (!canImport(support)) {
+            return false;
+        }
+
+        try {
+            var files = (List<File>) support.getTransferable().getTransferData(DataFlavor.javaFileListFlavor);
+            if (files.size() != 1) {
+                logger.error("Cannot open multiple files at a time!", "Import error");
+                return false;
+            }
+
+            var file = files.get(0);
+            var name = file.getName();
+
+            if (name.toLowerCase().endsWith(".mtf") || name.toLowerCase().endsWith(".blk")) {
+                owner.getMMLMenuBar().loadFile(file);
+                return true;
+            } else if (name.toLowerCase().endsWith(".mul")) {
+                UnitPrintManager.printMUL(owner.getFrame(), CConfig.getBooleanParam(CConfig.MISC_MUL_OPEN_BEHAVIOUR), file);
+                return true;
+            } else {
+                logger.error("Can only open files of type .mtf, .blk, and .mul", "Import error");
+                return false;
+            }
+
+
+        } catch (Exception e) {
+            logger.error(e);
+            return false;
+        }
+    }
+}