Handle loading monocle jar from file url

hollingsworthd · Apr 8, 2015 · d46ef72 · d46ef72
1 parent 13b0413
commit d46ef72
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/browser/src/com/machinepublishers/jbrowserdriver/JavaFx.java b/browser/src/com/machinepublishers/jbrowserdriver/JavaFx.java
@@ -30,6 +30,7 @@
 import java.net.JarURLConnection;
 import java.net.URL;
 import java.net.URLClassLoader;
+import java.net.URLConnection;
 import java.nio.file.FileAlreadyExistsException;
 import java.nio.file.Files;
 import java.util.ArrayList;
@@ -244,10 +245,13 @@ private static URL[] urls() {
         }
         try {
           File monocleTmp = new File(tmpDir, "monocle.jar");
-          Files.copy(((JarURLConnection) NativePlatformFactory.class.
-              getProtectionDomain().getCodeSource().getLocation().openConnection())
-              .getJarFileURL().openStream(),
-              monocleTmp.toPath());
+          URLConnection conn = NativePlatformFactory.class.
+              getProtectionDomain().getCodeSource().getLocation().openConnection();
+          if (conn instanceof JarURLConnection) {
+            Files.copy(((JarURLConnection) conn).getJarFileURL().openStream(), monocleTmp.toPath());
+          } else {
+            Files.copy(conn.getInputStream(), monocleTmp.toPath());
+          }
           urlList.add(monocleTmp.toURI().toURL());
         } catch (Throwable t) {
           Logs.exception(t);