From e27d5967029f4f3da8889dd673ba516dcd9f3ac8 Mon Sep 17 00:00:00 2001
From: gegy1000
Date: Sun, 16 Jun 2019 23:49:25 +0200
Subject: Plugin rework along with API rework: Enigma split from EnigmaProject;
 plugins now provide services configurable via a profile

---
 .../java/cuchaz/enigma/NoRetryMetadataSystem.java  | 38 ++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 src/main/java/cuchaz/enigma/NoRetryMetadataSystem.java

(limited to 'src/main/java/cuchaz/enigma/NoRetryMetadataSystem.java')

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);
+	}
+}
-- 
cgit v1.2.3