/*
|
* Copyright 2011 gitblit.com.
|
*
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
*/
|
package com.gitblit;
|
|
import java.io.File;
|
import java.io.FileFilter;
|
import java.io.IOException;
|
import java.lang.reflect.Method;
|
import java.net.URL;
|
import java.net.URLClassLoader;
|
import java.security.ProtectionDomain;
|
import java.text.MessageFormat;
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.Collections;
|
import java.util.List;
|
|
/**
|
* Launch helper class that adds all jars found in the local "lib" & "ext"
|
* folders and then calls the application main. Using this technique we do not
|
* have to specify a classpath and we can dynamically add jars to the
|
* distribution.
|
*
|
* @author James Moger
|
*
|
*/
|
public class Launcher {
|
|
public static final boolean DEBUG = false;
|
|
/**
|
* Parameters of the method to add an URL to the System classes.
|
*/
|
private static final Class<?>[] PARAMETERS = new Class[] { URL.class };
|
|
public static void main(String[] args) {
|
if (DEBUG) {
|
System.out.println("jcp=" + System.getProperty("java.class.path"));
|
ProtectionDomain protectionDomain = Launcher.class.getProtectionDomain();
|
System.out.println("launcher="
|
+ protectionDomain.getCodeSource().getLocation().toExternalForm());
|
}
|
|
// Load the JARs in the lib and ext folder
|
String[] folders = new String[] { "lib", "ext" };
|
List<File> jars = new ArrayList<File>();
|
for (String folder : folders) {
|
if (folder == null) {
|
continue;
|
}
|
File libFolder = new File(folder);
|
if (!libFolder.exists()) {
|
continue;
|
}
|
List<File> found = findJars(libFolder.getAbsoluteFile());
|
jars.addAll(found);
|
}
|
// sort the jars by name and then reverse the order so the newer version
|
// of the library gets loaded in the event that this is an upgrade
|
Collections.sort(jars);
|
Collections.reverse(jars);
|
|
if (jars.size() == 0) {
|
for (String folder : folders) {
|
File libFolder = new File(folder);
|
// this is a test of adding a comment
|
// more really interesting things
|
System.err.println("Failed to find any JARs in " + libFolder.getPath());
|
}
|
System.exit(-1);
|
} else {
|
for (File jar : jars) {
|
try {
|
jar.canRead();
|
addJarFile(jar);
|
} catch (Throwable t) {
|
t.printStackTrace();
|
}
|
}
|
}
|
|
// Start Server
|
GitBlitServer.main(args);
|
}
|
|
public static List<File> findJars(File folder) {
|
List<File> jars = new ArrayList<File>();
|
if (folder.exists()) {
|
File[] libs = folder.listFiles(new FileFilter() {
|
@Override
|
public boolean accept(File file) {
|
return !file.isDirectory() && file.getName().toLowerCase().endsWith(".jar");
|
}
|
});
|
if (libs != null && libs.length > 0) {
|
jars.addAll(Arrays.asList(libs));
|
if (DEBUG) {
|
for (File jar : jars) {
|
System.out.println("found " + jar);
|
}
|
}
|
}
|
}
|
|
return jars;
|
}
|
|
/**
|
* Adds a file to the classpath
|
*
|
* @param f
|
* the file to be added
|
* @throws IOException
|
*/
|
public static void addJarFile(File f) throws IOException {
|
if (f.getName().indexOf("-sources") > -1 || f.getName().indexOf("-javadoc") > -1) {
|
// don't add source or javadoc jars to runtime classpath
|
return;
|
}
|
URL u = f.toURI().toURL();
|
if (DEBUG) {
|
System.out.println("load=" + u.toExternalForm());
|
}
|
URLClassLoader sysloader = (URLClassLoader) ClassLoader.getSystemClassLoader();
|
Class<?> sysclass = URLClassLoader.class;
|
try {
|
Method method = sysclass.getDeclaredMethod("addURL", PARAMETERS);
|
method.setAccessible(true);
|
method.invoke(sysloader, new Object[] { u });
|
} catch (Throwable t) {
|
throw new IOException(MessageFormat.format(
|
"Error, could not add {0} to system classloader", f.getPath()), t);
|
}
|
}
|
}
|
