Plugin rework along with API rework: Enigma split from EnigmaProject; plugins now provide services configurable via a profile

1 files changed, 38 insertions, 0 deletions

diff --git a/src/main/java/cuchaz/enigma/NoRetryMetadataSystem.java b/src/main/java/cuchaz/enigma/NoRetryMetadataSystem.java
new file mode 100644
index 0000000..269d31e
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/NoRetryMetadataSystem.java
@@ -0,0 +1,38 @@
+package cuchaz.enigma;
+
+import com.strobel.assembler.metadata.ITypeLoader;
+import com.strobel.assembler.metadata.MetadataSystem;
+import com.strobel.assembler.metadata.TypeDefinition;
+import com.strobel.assembler.metadata.TypeReference;
+
+import java.util.Collections;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+
+public final class NoRetryMetadataSystem extends MetadataSystem {
+        private final Set<String> failedTypes = Collections.newSetFromMap(new ConcurrentHashMap<>());
+
+        public NoRetryMetadataSystem(final ITypeLoader typeLoader) {
+                super(typeLoader);
+        }
+
+        @Override
+        protected synchronized TypeDefinition resolveType(final String descriptor, final boolean mightBePrimitive) {
+                if (failedTypes.contains(descriptor)) {
+                        return null;
+                }
+
+                final TypeDefinition result = super.resolveType(descriptor, mightBePrimitive);
+
+                if (result == null) {
+                        failedTypes.add(descriptor);
+                }
+
+                return result;
+        }
+
+        @Override
+        public synchronized TypeDefinition resolve(final TypeReference type) {
+                return super.resolve(type);
+        }
+}