X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=org.argeo.slc.agent%2Fsrc%2Fmain%2Fjava%2Forg%2Fargeo%2Fslc%2Fcli%2FSlcMain.java;h=e46bc8434c07164252186f8015c7514ce151e2bc;hb=4a8845940afb3a147f23dedbf75dea5e26103352;hp=10636f3d4191815a8a97d102dce1e10d8598828e;hpb=8912b1cdb0b73547d70d2a2729ea075751f37f35;p=gpl%2Fargeo-slc.git diff --git a/org.argeo.slc.agent/src/main/java/org/argeo/slc/cli/SlcMain.java b/org.argeo.slc.agent/src/main/java/org/argeo/slc/cli/SlcMain.java index 10636f3d4..e46bc8434 100644 --- a/org.argeo.slc.agent/src/main/java/org/argeo/slc/cli/SlcMain.java +++ b/org.argeo.slc.agent/src/main/java/org/argeo/slc/cli/SlcMain.java @@ -1,5 +1,8 @@ package org.argeo.slc.cli; +import java.io.FileInputStream; +import java.io.FileNotFoundException; +import java.io.IOException; import java.util.Properties; import org.apache.commons.cli.CommandLine; @@ -10,8 +13,10 @@ import org.apache.commons.cli.Option; import org.apache.commons.cli.OptionBuilder; import org.apache.commons.cli.Options; import org.apache.commons.cli.ParseException; +import org.apache.commons.io.IOUtils; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; +import org.argeo.slc.ant.AntConstants; import org.argeo.slc.core.SlcException; import org.argeo.slc.logging.Log4jUtils; @@ -25,7 +30,7 @@ public class SlcMain { private final static String BOOTSTRAP_LOG4J_CONFIG = "org/argeo/slc/cli/bootstrapLog4j.properties"; private final static Option modeOpt = OptionBuilder.withLongOpt("mode") - .withArgName("mode").hasArg().isRequired().withDescription( + .withArgName("mode").hasArg().withDescription( "SLC execution mode, one of: " + listModeValues()).create( 'm'); @@ -34,6 +39,11 @@ public class SlcMain { .withValueSeparator(',').withDescription( "use value for given property").create('p'); + private final static Option propertiesOpt = OptionBuilder.withLongOpt( + "properties").withArgName("properties file").hasArgs() + .withValueSeparator(',').withDescription( + "load properties from file (-p has priority)").create('P'); + private final static Option scriptOpt = OptionBuilder.withLongOpt("script") .withArgName("script").hasArg().withDescription( "SLC script to execute").create('s'); @@ -57,6 +67,7 @@ public class SlcMain { options.addOption(scriptOpt); options.addOption(targetsOpt); options.addOption(propertyOpt); + options.addOption(propertiesOpt); options.addOption(runtimeOpt); } @@ -74,10 +85,15 @@ public class SlcMain { // Mode String modeStr = cl.getOptionValue(modeOpt.getOpt()); - try { - mode = Mode.valueOf(modeStr); - } catch (IllegalArgumentException e) { - throw new SlcException("Unrecognized mode '" + modeStr + "'", e); + if (modeStr == null) { + mode = Mode.single; + } else { + try { + mode = Mode.valueOf(modeStr); + } catch (IllegalArgumentException e) { + throw new SlcException("Unrecognized mode '" + modeStr + + "'", e); + } } // Script @@ -94,6 +110,12 @@ public class SlcMain { } // Properties + if (cl.hasOption(propertiesOpt.getOpt())) { + for (String propertyFile : cl.getOptionValues(propertiesOpt + .getOpt())) { + loadPropertyFile(properties, propertyFile); + } + } if (cl.hasOption(propertyOpt.getOpt())) { for (String property : cl.getOptionValues(propertyOpt.getOpt())) { addProperty(properties, property); @@ -103,10 +125,7 @@ public class SlcMain { // Runtime if (cl.hasOption(runtimeOpt.getOpt())) { runtimeStr = cl.getOptionValue(runtimeOpt.getOpt()); - } else { - runtimeStr = "default"; } - } catch (ParseException e) { System.err.println("Problem with command line arguments. " + e.getMessage()); @@ -124,7 +143,8 @@ public class SlcMain { initLogging(properties); if (log.isDebugEnabled()) { log.debug("Mode: " + mode); - log.debug("Runtime: " + runtimeStr); + if (runtimeStr != null) + log.debug("Runtime: " + runtimeStr); log.debug("User properties: " + properties); if (script != null) log.debug("Script: " + script); @@ -154,7 +174,7 @@ public class SlcMain { return str.substring(0, str.length() - 2); } - private static void addProperty(Properties properties, String property) { + protected static void addProperty(Properties properties, String property) { int eqIndex = property.indexOf('='); if (eqIndex == 0) throw new SlcException("Badly formatted property " + property); @@ -167,14 +187,28 @@ public class SlcMain { } else { properties.setProperty(property, "true"); } + } + protected static void loadPropertyFile(Properties properties, + String propertyFile) { + FileInputStream in = null; + try { + in = new FileInputStream(propertyFile); + properties.load(in); + } catch (Exception e) { + throw new SlcException("Could not load proeprty file " + + propertyFile); + } finally { + IOUtils.closeQuietly(in); + } } private static void initLogging(Properties userProperties) { System.setProperty("log4j.defaultInitOverride", "true"); // Add log4j user properties to System properties - for (String key : userProperties.stringPropertyNames()) { + for (Object obj : userProperties.keySet()) { + String key = obj.toString(); if (key.startsWith("log4j.")) { System.setProperty(key, userProperties.getProperty(key)); }