From 142b463faaad94f9fe8e33fe753ae5845b975ac6 Mon Sep 17 00:00:00 2001
From: 2xsaiko
Date: Fri, 12 Jun 2020 23:08:09 +0200
Subject: Fix "Mark as Deobfuscated" menu entry not working (#265)

---
 enigma-swing/src/main/java/cuchaz/enigma/gui/Gui.java                | 3 +--
 enigma-swing/src/main/java/cuchaz/enigma/gui/panels/EditorPanel.java | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

(limited to 'enigma-swing')

diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/Gui.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/Gui.java
index 0b2de460..72ec9656 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/Gui.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/Gui.java
@@ -646,9 +646,8 @@ public class Gui {
 		if (cursorReference == null) return;
 
 		Entry<?> obfEntry = cursorReference.entry;
-		Entry<?> deobfEntry = controller.project.getMapper().deobfuscate(obfEntry);
 
-		if (!Objects.equals(obfEntry, deobfEntry)) {
+		if (controller.project.getMapper().hasDeobfMapping(obfEntry)) {
 			if (!validateImmediateAction(vc -> this.controller.removeMapping(vc, cursorReference))) return;
 			this.controller.sendPacket(new RemoveMappingC2SPacket(cursorReference.getNameableEntry()));
 		} else {
diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/panels/EditorPanel.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/panels/EditorPanel.java
index f689a211..135a5291 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/panels/EditorPanel.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/panels/EditorPanel.java
@@ -475,7 +475,7 @@ public class EditorPanel {
 		this.popupMenu.openNextMenu.setEnabled(this.controller.hasNextReference());
 		this.popupMenu.toggleMappingMenu.setEnabled(isRenamable);
 
-		if (referenceEntry != null && referenceEntry.equals(this.controller.project.getMapper().deobfuscate(referenceEntry))) {
+		if (referenceEntry != null && this.controller.project.getMapper().hasDeobfMapping(referenceEntry)) {
 			this.popupMenu.toggleMappingMenu.setText(I18n.translate("popup_menu.reset_obfuscated"));
 		} else {
 			this.popupMenu.toggleMappingMenu.setText(I18n.translate("popup_menu.mark_deobfuscated"));
-- 
cgit v1.2.3