Add remap crash handler

4 files changed, 16 insertions, 4 deletions

diff --git a/enigma/src/main/java/cuchaz/enigma/classhandle/ClassHandleError.java b/enigma/src/main/java/cuchaz/enigma/classhandle/ClassHandleError.java
index a11b9dc..20f847a 100644
--- a/enigma/src/main/java/cuchaz/enigma/classhandle/ClassHandleError.java
+++ b/enigma/src/main/java/cuchaz/enigma/classhandle/ClassHandleError.java
@@ -28,8 +28,13 @@ public final class ClassHandleError {
                 return new ClassHandleError(Type.DECOMPILE, cause);
         }
 
+        public static ClassHandleError remap(Throwable cause) {
+                return new ClassHandleError(Type.REMAP, cause);
+        }
+
         public enum Type {
                 DECOMPILE,
+                REMAP,
         }
 
 }
\ No newline at end of file
diff --git a/enigma/src/main/java/cuchaz/enigma/classhandle/ClassHandleProvider.java b/enigma/src/main/java/cuchaz/enigma/classhandle/ClassHandleProvider.java
index 2d9b52d..f9e4eff 100644
--- a/enigma/src/main/java/cuchaz/enigma/classhandle/ClassHandleProvider.java
+++ b/enigma/src/main/java/cuchaz/enigma/classhandle/ClassHandleProvider.java
@@ -11,6 +11,8 @@ import java.util.concurrent.locks.ReentrantReadWriteLock;
 
 import javax.annotation.Nullable;
 
+import org.objectweb.asm.tree.ClassNode;
+
 import cuchaz.enigma.Enigma;
 import cuchaz.enigma.EnigmaProject;
 import cuchaz.enigma.bytecode.translators.SourceFixVisitor;
@@ -19,7 +21,6 @@ import cuchaz.enigma.events.ClassHandleListener.InvalidationType;
 import cuchaz.enigma.source.*;
 import cuchaz.enigma.translation.representation.entry.ClassEntry;
 import cuchaz.enigma.utils.Result;
-import org.objectweb.asm.tree.ClassNode;
 
 import static cuchaz.enigma.utils.Utils.withLock;
 
@@ -285,9 +286,13 @@ public final class ClassHandleProvider {
                         int v = mappedVersion.incrementAndGet();
                         f.thenAcceptAsync(res -> {
                                 if (res == null || mappedVersion.get() != v) return;
-                                res = res.map(source -> {
-                                        source.remapSource(p.project, p.project.getMapper().getDeobfuscator());
-                                        return source;
+                                res = res.andThen(source -> {
+                                        try {
+                                                source.remapSource(p.project, p.project.getMapper().getDeobfuscator());
+                                                return Result.ok(source);
+                                        } catch (Throwable e) {
+                                                return Result.err(ClassHandleError.remap(e));
+                                        }
                                 });
                                 Entry.this.source = res;
                                 Entry.this.waitingSources.forEach(s -> s.complete(source));
diff --git a/enigma/src/main/resources/lang/de_de.json b/enigma/src/main/resources/lang/de_de.json
index ef41da1..65685cf 100644
--- a/enigma/src/main/resources/lang/de_de.json
+++ b/enigma/src/main/resources/lang/de_de.json
@@ -11,6 +11,7 @@
 
         "editor.decompiling": "Dekompiliere...",
         "editor.decompile_error": "Ein Fehler ist während des Dekompilierens aufgetreten.",
+        "editor.remap_error": "Ein Fehler ist während des Remappens aufgetreten.",
 
         "validation.message.empty_field": "Dieses Feld muss ausgefüllt werden.",
         "validation.message.not_int": "Wert muss eine ganze Zahl sein.",
diff --git a/enigma/src/main/resources/lang/en_us.json b/enigma/src/main/resources/lang/en_us.json
index 9b12f42..c2a739c 100644
--- a/enigma/src/main/resources/lang/en_us.json
+++ b/enigma/src/main/resources/lang/en_us.json
@@ -79,6 +79,7 @@
 
         "editor.decompiling": "Decompiling...",
         "editor.decompile_error": "An error was encountered while decompiling.",
+        "editor.remap_error": "An error was encountered while remapping.",
 
         "info_panel.classes.obfuscated": "Obfuscated Classes",
         "info_panel.classes.deobfuscated": "De-obfuscated Classes",