package org.rzo.yajsw.boot;
import java.io.File;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.jar.Attributes;
import java.util.jar.JarFile;
import java.util.jar.Manifest;
import org.rzo.yajsw.Constants;
import org.rzo.yajsw.app.WrapperJVMMain;
public class WrapperLoader
{
private static boolean checkPath(String path)
{
int ix = path.indexOf("!");
if (ix == -1)
{
System.out.println("<yajsw>/wrapper.jar not found, please check classpath. aborting wrapper !");
Runtime.getRuntime().halt(999);// -> groovy eclipse plugin crashes
return false;
}
return true;
}
/**
* Gets the wrapper jar.
*
* @return the wrapper jar
*/
public static String getWrapperJar()
{
String cn = Constants.class.getCanonicalName();
String rn = cn.replace('.', '/') + ".class";
String path = ".";
try
{
path = Constants.class.getClassLoader().getResource(rn).getPath();
if (!checkPath(path))
return null;
path = path.substring(0, path.indexOf("!"));
path = new URI(path).getPath();
path.replaceAll("%20", " ");
//System.out.println("wrapper jar "+path);
// if (path.startsWith("/"))
// path = path.substring(1);
return path;
}
catch (Exception e1)
{
e1.printStackTrace();
}
return null;
}
public static String getWrapperAppJar()
{
String cn = WrapperJVMMain.class.getCanonicalName();
String rn = cn.replace('.', '/') + ".class";
String path = ".";
try
{
path = WrapperJVMMain.class.getClassLoader().getResource(rn).getPath();
if (!checkPath(path))
return null;
path = path.substring(0, path.indexOf("!"));
path = new URI(path).getPath();
path.replaceAll("%20", " ");
//System.out.println("wrapper jar "+path);
// if (path.startsWith("/"))
// path = path.substring(1);
return path;
}
catch (Exception e1)
{
e1.printStackTrace();
}
return null;
}
public static ArrayList getGroovyClasspath()
{
ArrayList result = new ArrayList();
String wrapperHome = getWrapperHome();
File groovyLib = new File(wrapperHome, "lib");
if (! groovyLib.exists())
{
System.out.println("<yajsw>/lib folder not found. Please check that relative the lib folder is in the same folder as <yajsw>/wrapper.jar");
return result;
}
File[] groovyLibs = groovyLib.listFiles();
for (File file : groovyLibs)
{
if (file.isDirectory())
result.addAll(getFiles(file));
else
try
{
result.add(file.toURI().toURL());
}
catch (MalformedURLException e)
{
System.out.println("Error in getGroovyClasspath: "+e.getMessage());
}
}
return result;
}
private static Collection getFiles(File parent)
{
ArrayList result = new ArrayList();
File[] files = parent.listFiles();
for (File file : files)
{
if (file.isDirectory())
result.addAll(getFiles(file));
else
try
{
result.add(file.toURI().toURL());
}
catch (MalformedURLException e)
{
System.out.println("Error in getGroovyClasspath: "+e.getMessage());
}
}
return result;
}
public static URL[] getWrapperClasspath(String type, boolean logErrors)
{
String wrapperJar;
if ("App".equals(type))
wrapperJar = getWrapperAppJar();
else
wrapperJar = getWrapperJar();
Manifest manifest;
try
{
manifest = new JarFile(wrapperJar).getManifest();
}
catch (IOException e1)
{
// TODO Auto-generated catch block
e1.printStackTrace();
return null;
}
Attributes attr = manifest.getMainAttributes();
String cl = attr.getValue("Class-Path-" + type);
if (cl == null)
return null;
ArrayList classpath = new ArrayList();
classpath.add(new File(wrapperJar));
String[] clArr = cl.split(" ");
File parent = new File(wrapperJar).getParentFile();
for (int i = 0; i < clArr.length; i++)
{
String file = clArr[i];
File myFile;
try
{
myFile = new File(parent, file);
if (!myFile.exists() && logErrors)
System.out.println("WARNING: lib not found: " + myFile.getCanonicalPath());
classpath.add(myFile);
}
catch (Exception e)
{
e.printStackTrace();
}
}
URL[] urlsArr = new URL[classpath.size()];
int i = 0;
for (Iterator it = classpath.iterator(); it.hasNext(); i++)
try
{
urlsArr[i] = ((File) it.next()).toURI().toURL();
// System.out.println("classpath: "+urlsArr[i]);
}
catch (Exception e)
{
// log.throwing(WrapperMain.class.getName(), "main", e);
e.printStackTrace();
}
return urlsArr;
}
public static URLClassLoader getWrapperClassLoader()
{
URL[] core = getWrapperClasspath("Wrapper-Core", true);
URL[] extended = getWrapperClasspath("Wrapper-Extended", false);
URL[] urls = new URL[core.length+ extended.length];
System.arraycopy(core, 0, urls, 0, core.length);
System.arraycopy(extended, 0, urls, core.length, extended.length);
return new WrapperClassLoader(urls, Thread.currentThread().getContextClassLoader());
}
public static String getWrapperHome()
{
return new File(getWrapperJar()).getParent();
}
}