Java tutorial
/* * Copyright 2010-2013, CloudBees Inc. * * 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.cloudbees.sdk; import com.cloudbees.api.BeesClientException; import com.cloudbees.sdk.cli.ACommand; import com.cloudbees.sdk.cli.CommandScope; import com.cloudbees.sdk.cli.CommandService; import com.cloudbees.sdk.cli.DirectoryStructure; import com.cloudbees.sdk.extensibility.AnnotationLiteral; import com.cloudbees.sdk.maven.MavenRepositorySystemSessionDecorator; import com.cloudbees.sdk.maven.RemoteRepositoryDecorator; import com.cloudbees.sdk.maven.RepositorySystemModule; import com.cloudbees.sdk.utils.Helper; import com.google.inject.AbstractModule; import com.google.inject.Guice; import com.google.inject.Injector; import com.staxnet.appserver.utils.XmlHelper; import com.staxnet.repository.LocalRepository; import hudson.util.VersionNumber; import org.apache.commons.cli.UnrecognizedOptionException; import org.apache.commons.io.IOUtils; import org.w3c.dom.*; import javax.inject.Inject; import java.io.File; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.util.*; /** * @author Fabian Donze * @author Kohsuke Kawaguchi */ public class Bees { public static String BOOTSTRAP_VERSION = "1.3.2"; /** * Version of the bees CLI. */ public static VersionNumber version = loadVersion(); private final static String app_template_xml_url = "http://cloudbees-downloads.s3.amazonaws.com/"; private final static String app_template_xml_name = "sdk/cloudbees-sdk-config-5.xml"; private final static String app_template_xml_desc = "CloudBees SDK configuration"; private static final long CHECK_INTERVAL = 1000 * 60 * 60 * 12; // 12 hours public final static String CHECK_FILE = "sdk/check-5.dat"; public static final String SDK_PLUGIN_INSTALL = "plugin:install"; @Inject private CommandService commandService; @Inject private PluginsToInstallList pluginsToInstallList; /** * Entry point to all the components. */ private final Injector injector; private final ClassLoader extLoader; public Bees() throws Exception { extLoader = getClass().getClassLoader(); // container that includes all the things that make a bees CLI. Injector injector = Guice.createInjector(new AbstractModule() { @Override protected void configure() { bind(CommandService.class).to(CommandServiceImpl.class); bind(ClassLoader.class).annotatedWith(AnnotationLiteral.of(ExtensionClassLoader.class)) .toInstance(extLoader); bindScope(CommandScope.class, new CommandScopeImpl()); bind(RemoteRepositoryDecorator.class).to(RemoteRepositoryDecoratorImpl.class); bind(MavenRepositorySystemSessionDecorator.class) .to(CliMavenRepositorySystemSessionDecorator.class); } }, new RepositorySystemModule()); this.injector = injector; this.injector.injectMembers(this); CommandServiceImpl service = (CommandServiceImpl) commandService; service.loadCommandProperties(); if (service.getCount() == 0) { throw new RuntimeException("Cannot find bees commands"); } } public int run(String[] args) throws Exception { reportTime("Bees.run"); // Load command definitions if (args.length == 0) args = new String[] { "help" }; Object context = CommandScopeImpl.begin(); try { // If it is the init command, execute before anything else if (args[0].equalsIgnoreCase("init")) { // Setup the configuration file ACommand setupCommand = commandService.getCommand(args[0]); if (setupCommand == null) throw new Error("Panic: init error"); setupCommand.run(Arrays.asList(args)); // Initialize the SDK initialize(false); // Install plugins installPlugins(args); return 0; } else { // Initialize the SDK initialize(false); // Install plugins installPlugins(args); ACommand command = commandService.getCommand(args[0]); if (command == null) { // no such command. print help System.err.println("No such command: " + args[0]); command = commandService.getCommand("help"); if (command == null) throw new Error("Panic: command " + args[0] + " was not found, and even the help command was not found"); } int r = command.run(Arrays.asList(args)); if (r == 99) { initialize(true); } return r; } } finally { CommandScopeImpl.end(context); } } private String getHome() { return System.getProperty("bees.home"); } private void initialize(boolean force) throws Exception { LocalRepository localRepository = new LocalRepository(); String beesRepoPath = localRepository.getRepositoryPath(); File lastCheckFile = new File(beesRepoPath, CHECK_FILE); boolean checkVersion = true; Properties p = new Properties(); if (!force && Helper.loadProperties(lastCheckFile, p)) { String str = p.getProperty("last"); if (str != null) { long interval = System.currentTimeMillis() - Long.parseLong(str); if (interval < CHECK_INTERVAL) checkVersion = false; } } if (checkVersion) { // Check SDK version File sdkConfig = getURLAsFile(localRepository, app_template_xml_url + app_template_xml_name, app_template_xml_name, app_template_xml_desc); Document doc = XmlHelper.readXMLFromFile(sdkConfig.getCanonicalPath()); Element e = doc.getDocumentElement(); String availVersion = e.getAttribute("version"); String minVersion = e.getAttribute("min-version"); VersionNumber currentVersion = new VersionNumber(BOOTSTRAP_VERSION); VersionNumber availableVersion = new VersionNumber(availVersion); VersionNumber minimumVersion = new VersionNumber(minVersion); if (currentVersion.compareTo(availableVersion) < 0) { System.out.println(); if (currentVersion.compareTo(minimumVersion) < 0) { throw new AbortException("ERROR - This version of the CloudBees SDK is no longer supported," + "" + " please install the latest version (" + availVersion + ")."); } else if (currentVersion.compareTo(availableVersion) < 0) { System.out.println( "WARNING - A new version of the CloudBees SDK is available, please install the latest version (" + availVersion + ")."); } String hRef = e.getAttribute("href"); String homeRef = "www.cloudbees.com"; NodeList nodeList = e.getElementsByTagName("link"); for (int i = 0; i < nodeList.getLength(); i++) { Node node = nodeList.item(i); NamedNodeMap nodeMap = node.getAttributes(); Node rel = nodeMap.getNamedItem("rel"); Node href = nodeMap.getNamedItem("href"); if (rel != null && rel.getTextContent().trim().equalsIgnoreCase("alternate") && href != null) { homeRef = href.getTextContent(); } } NodeList libsNL = e.getElementsByTagName("libraries"); Node libsNode = null; if (libsNL.getLength() > 0) { libsNode = libsNL.item(0); } if (libsNode != null) { NodeList libNL = e.getElementsByTagName("library"); for (int i = 0; i < libNL.getLength(); i++) { Node node = libNL.item(i); NamedNodeMap nodeMap = node.getAttributes(); Node nameNode = nodeMap.getNamedItem("name"); Node refNode = nodeMap.getNamedItem("href"); if (nameNode != null && refNode != null) { String libName = nameNode.getTextContent(); String libUrlString = refNode.getTextContent().trim(); int idx = libUrlString.lastIndexOf('/'); String libFileName = libUrlString.substring(idx); localRepository.getURLAsFile(libUrlString, "lib1" + libFileName, libName); } } } System.out.println(" SDK home: " + homeRef); System.out.println(" SDK download: " + hRef); System.out.println(); } // Check plugins version NodeList pluginsNL = e.getElementsByTagName("plugins"); Node pluginsNode = null; if (pluginsNL.getLength() > 0) { pluginsNode = pluginsNL.item(0); } if (pluginsNode != null) { NodeList pluginNL = e.getElementsByTagName("plugin"); CommandServiceImpl service = (CommandServiceImpl) commandService; for (int i = 0; i < pluginNL.getLength(); i++) { Node node = pluginNL.item(i); NamedNodeMap nodeMap = node.getAttributes(); Node nameNode = nodeMap.getNamedItem("artifact"); if (nameNode != null) { Node n = nodeMap.getNamedItem("required"); boolean forceInstall = (n != null && Boolean.parseBoolean(n.getTextContent())); String pluginArtifact = nameNode.getTextContent(); GAV gav = new GAV(pluginArtifact); VersionNumber pluginVersion = new VersionNumber(gav.version); Plugin plugin = service.getPlugin(gav.artifactId); if (plugin != null) { forceInstall = false; GAV pgav = new GAV(plugin.getArtifact()); VersionNumber currentPluginVersion = new VersionNumber(pgav.version); if (currentPluginVersion.compareTo(pluginVersion) < 0) { Node nf = nodeMap.getNamedItem("force-upgrade"); boolean forced = (nf != null && Boolean.parseBoolean(nf.getTextContent())); if (forced) { forceInstall = true; } else { System.out.println(); System.out.println("WARNING - A newer version of the [" + gav.artifactId + "] plugin is available, please update with:"); System.out.println(" > bees plugin:update " + gav.artifactId); System.out.println(); } } } if (forceInstall) pluginsToInstallList.put(gav.artifactId, gav); } } } // Update last check p.setProperty("last", "" + System.currentTimeMillis()); lastCheckFile.getParentFile().mkdirs(); FileOutputStream fos = new FileOutputStream(lastCheckFile); p.store(fos, "CloudBees SDK check"); fos.close(); } } private void installPlugins(String[] args) throws Exception { Set<Map.Entry<String, GAV>> set = pluginsToInstallList.entrySet(); if (set.size() > 0) { ACommand installPluginCmd = commandService.getCommand(SDK_PLUGIN_INSTALL); Iterator<Map.Entry<String, GAV>> it = set.iterator(); while (it.hasNext()) { Map.Entry<String, GAV> entry = it.next(); System.out.println("Installing plugin: " + entry.getValue()); List<String> piArgs; if (isVerbose(args)) piArgs = Arrays.asList(SDK_PLUGIN_INSTALL, entry.getValue().toString(), "-f", "-v"); else piArgs = Arrays.asList(SDK_PLUGIN_INSTALL, entry.getValue().toString(), "-f"); installPluginCmd.run(piArgs); pluginsToInstallList.remove(entry.getKey()); } // Reload the plugins commands CommandServiceImpl service = (CommandServiceImpl) commandService; service.loadCommandProperties(); } } private File getURLAsFile(LocalRepository localRepository, String urlStr, String localCachePath, String description) throws IOException { try { DirectoryStructure ds = new DirectoryStructure(); Properties properties = new Properties(); if (Helper.loadProperties(new File(ds.localRepository, "bees.config"), properties)) { Helper.setJVMProxySettings(properties); } return localRepository.getURLAsFile(urlStr, localCachePath, description); } catch (Exception e) { throw (IOException) new IOException("Failed to retrieve " + urlStr).initCause(e); } } public static void main(String[] args) { reportTime("Bees"); boolean verbose = isVerbose(args); if (verbose || isHelp(args) || isPluginCmd(args)) { System.out.println("# CloudBees SDK version: " + BOOTSTRAP_VERSION); System.out.println("# CloudBees Driver version: " + version); if (verbose) System.out.println(System.getProperties()); } try { new Bees().run(args); } catch (BeesClientException e) { System.err.println(); String errCode = e.getError().getErrorCode(); if (errCode != null && errCode.equals("AuthFailure")) { if (e.getError().getMessage() != null) System.err.println("ERROR: " + e.getError().getMessage()); else System.err.println("ERROR: Authentication failure, please check credentials!"); } else System.err.println("ERROR: " + e.getMessage()); // e.printStackTrace(); System.exit(2); } catch (UnrecognizedOptionException e) { System.err.println(); System.err.println("ERROR: " + e.getMessage()); System.exit(2); } catch (IllegalArgumentException e) { System.err.println(); System.err.println("ERROR: " + e.getMessage()); System.exit(2); } catch (BeesSecurityException e) { System.err.println(); System.err.println("ERROR: " + e.getMessage()); System.exit(2); } catch (Throwable e) { System.err.println(); System.err.println("ERROR: " + e.getMessage()); if (isVerbose(args)) e.printStackTrace(); System.exit(2); } } /** * Parses the version number of SDK from the resource file that Maven produces. * <p/> * To support running this from IDE and elsewhere, work gracefully if the version * is not available or not filtered. */ private static VersionNumber loadVersion() { Properties props = new Properties(); InputStream in = Bees.class.getResourceAsStream("version.properties"); if (in != null) { try { props.load(in); } catch (IOException e) { throw new Error(e); } finally { IOUtils.closeQuietly(in); } } Object v = props.get("version"); if (v != null) try { return new VersionNumber(v.toString()); } catch (Exception e) { // fall through } return new VersionNumber("0"); } private static boolean isVerbose(String[] args) { for (String arg : args) { if (arg.equalsIgnoreCase("-v") || arg.equalsIgnoreCase("--verbose")) return true; } return false; } private static boolean isHelp(String[] args) { if (args.length == 0) return true; for (String arg : args) { if (arg.equalsIgnoreCase("help")) return true; } return false; } private static boolean isPluginCmd(String[] args) { if (args.length == 0) return false; for (String arg : args) { if (arg.startsWith("plugin:")) return true; } return false; } private static void reportTime(String caption) { String profile = System.getProperty("profile"); if (profile != null) { System.out.println(caption + ": " + (System.nanoTime() - Long.valueOf(profile)) / 1000000L + "ms"); } } }