/**
* Copyright (C) 2013 the original author or authors.
* See the notice.md file distributed with this work for additional
* information regarding copyright ownership.
*
* 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 io.hawt.app;
import io.hawt.embedded.Main;
import java.awt.*;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URI;
import java.net.URL;
import java.util.HashSet;
import java.util.Set;
public class App {
private static final String WAR_FILENAME = "hawtio.war";
private static final int KB = 1024;
public static void main(String[] args) {
Object val = System.getProperty("hawtio.authenticationEnabled");
if (val == null) {
System.setProperty("hawtio.authenticationEnabled", "false");
}
Main main = new Main();
try {
String virtualMachineClass = "com.sun.tools.attach.VirtualMachine";
try {
Class<?> aClass = loadClass(virtualMachineClass, App.class.getClassLoader(), Thread.currentThread().getContextClassLoader());
//System.out.println("Found " + aClass + " on the classpath!");
} catch (Exception e) {
// lets try find the tools.jar instead
Set<String> paths = new HashSet<String>();
String javaHome = System.getProperty("java.home", ".");
addPath(paths, javaHome);
// now lets try the JAVA_HOME environment variable just in case
javaHome = System.getenv("JAVA_HOME");
if (javaHome != null && javaHome.length() > 0) {
addPath(paths, javaHome);
}
boolean found = false;
for (String path : paths) {
File file = new File(path, "lib/tools.jar");
if (file.exists()) {
found = true;
String url = file.toURI().toURL().toString();
main.setExtraClassPath(url);
break;
}
}
if (!found) {
System.out.println("Failed to load class " + virtualMachineClass
+ " and find tools.jar in directories " + paths + ". " + e);
}
}
ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
URL resource = classLoader.getResource(WAR_FILENAME);
if (resource == null) {
System.err.println("Could not find the " + WAR_FILENAME + " on classpath!");
System.exit(1);
}
File warFile = File.createTempFile("hawtio-", ".war");
//System.out.println("Extracting " + WAR_FILENAME + " to " + warFile + " ...");
writeStreamTo(resource.openStream(), new FileOutputStream(warFile), 64 * KB);
String warPath = warFile.getCanonicalPath();
main.setWar(warPath);
} catch (Exception e) {
System.out.println("Failed to create hawtio: " + e.getMessage());
e.printStackTrace();
return;
}
if (!main.parseArguments(args) || main.isHelp()) {
main.showOptions();
} else {
try {
main.run();
if (Desktop.isDesktopSupported()) {
int port = main.getPort();
String url = "http://localhost:" + port + main.getContextPath();
try {
Desktop.getDesktop().browse(new URI(url));
} catch (Exception e) {
System.out.println("Failed to open browser session, to access hawtio visit \"" + url + "\"");
}
}
} catch (Exception e) {
System.out.println("Failed running hawtio: " + e.getMessage());
e.printStackTrace();
}
}
}
protected static void addPath(Set<String> paths, String path) {
paths.add(path);
String jreSuffix = File.separator + "jre";
if (path.endsWith(jreSuffix)) {
path = path.substring(0, path.length() - jreSuffix.length());
paths.add(path);
}
}
private static Class<?> loadClass(String name, ClassLoader... classLoaders) throws ClassNotFoundException {
for (ClassLoader classLoader : classLoaders) {
try {
return classLoader.loadClass(name);
} catch (ClassNotFoundException e) {
// ignore
}
}
return Class.forName(name);
}
public static int writeStreamTo(final InputStream input, final OutputStream output, int bufferSize) throws IOException {
int available = Math.min(input.available(), 256 * KB);
byte[] buffer = new byte[Math.max(bufferSize, available)];
int answer = 0;
int count = input.read(buffer);
while (count >= 0) {
output.write(buffer, 0, count);
answer += count;
count = input.read(buffer);
}
return answer;
}
}