com.nitorcreations.willow.protocols.jaronclasspathcontaining.Handler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of willow-utils Show documentation

There is a newer version: 2.0.2

package com.nitorcreations.willow.protocols.jaronclasspathcontaining;

import java.io.IOException;
import java.net.URL;
import java.net.URLConnection;

import com.nitorcreations.willow.protocols.ProxylessURLStreamHandler;

public class Handler extends ProxylessURLStreamHandler {
  private final ClassLoader classLoader;

  public Handler() {
    this.classLoader = getClass().getClassLoader();
  }
  public Handler(ClassLoader classLoader) {
    this.classLoader = classLoader;
  }
  @Override
  protected URLConnection openConnection(URL u) throws IOException {
    final URL resourceUrl = classLoader.getResource(u.getPath());
    if (resourceUrl != null && resourceUrl.toString().startsWith("jar:")) {
      String resource = resourceUrl.toString();
      int jarEnd = resource.lastIndexOf("!");
      URL fileUrl = new URL(resource.substring(4, jarEnd));
      return fileUrl.openConnection();
    } else if (resourceUrl != null) {
      throw new IOException("Resource " + u.getPath() + " not found in jar, unable to resolve jar");
    } else {
      throw new IOException("Resource " + u.getPath() + " not found on classpath, unable to resolve jar");
    }
  }
}