From e98dae3df050f180b1dfef8576d32df4c3c64663 Mon Sep 17 00:00:00 2001
From: Thiakil
Date: Mon, 23 Jul 2018 13:20:47 +0800
Subject: add classloader for adding deps

---
 src/main/java/oml/ExtraClasspathTypeLoader.java | 59 +++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 src/main/java/oml/ExtraClasspathTypeLoader.java

(limited to 'src/main/java/oml')

diff --git a/src/main/java/oml/ExtraClasspathTypeLoader.java b/src/main/java/oml/ExtraClasspathTypeLoader.java
new file mode 100644
index 00000000..f8ec2e0c
--- /dev/null
+++ b/src/main/java/oml/ExtraClasspathTypeLoader.java
@@ -0,0 +1,59 @@
+package oml;
+
+import com.strobel.assembler.metadata.Buffer;
+import com.strobel.assembler.metadata.ITypeLoader;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.util.Arrays;
+
+/**
+ * Copy of ClasspathTypeLoader supporting a classpath constructor.
+ */
+public class ExtraClasspathTypeLoader implements ITypeLoader {
+	private final ClassLoader _loader;
+
+	public ExtraClasspathTypeLoader(String extraClasspath){
+		_loader = new URLClassLoader(Arrays.stream(extraClasspath.split(File.pathSeparator)).map(path-> {
+			try {
+				return new File(path).toURI().toURL();
+			} catch (MalformedURLException e) {
+				throw new RuntimeException(e);
+			}
+		}).toArray(URL[]::new));
+	}
+
+	@Override
+	public boolean tryLoadType(final String internalName, final Buffer buffer) {
+
+		final String path = internalName.concat(".class");
+		final URL resource = _loader.getResource(path);
+
+		if (resource == null) {
+			return false;
+		}
+
+		try (final InputStream stream = _loader.getResourceAsStream(path)) {
+			final byte[] temp = new byte[4096];
+
+			int bytesRead;
+
+			while ((bytesRead = stream.read(temp, 0, temp.length)) > 0) {
+				//buffer.ensureWriteableBytes(bytesRead);
+				buffer.putByteArray(temp, 0, bytesRead);
+			}
+
+			buffer.flip();
+
+
+			return true;
+		}
+		catch (final IOException ignored) {
+			return false;
+		}
+	}
+}
-- 
cgit v1.2.3