/**
* Copyright 2006-2016 the original author or authors.
*
* 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 org.mybatis.generator.api;
import static org.mybatis.generator.internal.util.messages.Messages.getString;
import java.io.File;
import java.io.IOException;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.StringTokenizer;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.exception.InvalidConfigurationException;
import org.mybatis.generator.exception.XMLParserException;
import org.mybatis.generator.internal.DefaultShellCallback;
import org.mybatis.generator.logging.LogFactory;
/**
* This class allows the code generator to be run from the command line.
*
* @author Jeff Butler
*/
public class ShellRunner {
private static final String CONFIG_FILE = "-configfile"; //$NON-NLS-1$
private static final String OVERWRITE = "-overwrite"; //$NON-NLS-1$
private static final String CONTEXT_IDS = "-contextids"; //$NON-NLS-1$
private static final String TABLES = "-tables"; //$NON-NLS-1$
private static final String VERBOSE = "-verbose"; //$NON-NLS-1$
private static final String FORCE_JAVA_LOGGING = "-forceJavaLogging"; //$NON-NLS-1$
private static final String HELP_1 = "-?"; //$NON-NLS-1$
private static final String HELP_2 = "-h"; //$NON-NLS-1$
public static void main(String[] args) {
if (args.length == 0) {
usage();
System.exit(0);
return; // only to satisfy compiler, never returns
}
Map<String, String> arguments = parseCommandLine(args);
if (arguments.containsKey(HELP_1)) {
usage();
System.exit(0);
return; // only to satisfy compiler, never returns
}
if (!arguments.containsKey(CONFIG_FILE)) {
writeLine(getString("RuntimeError.0")); //$NON-NLS-1$
return;
}
List<String> warnings = new ArrayList<String>();
String configfile = arguments.get(CONFIG_FILE);
File configurationFile = new File(configfile);
if (!configurationFile.exists()) {
writeLine(getString("RuntimeError.1", configfile)); //$NON-NLS-1$
return;
}
Set<String> fullyqualifiedTables = new HashSet<String>();
if (arguments.containsKey(TABLES)) {
StringTokenizer st = new StringTokenizer(arguments.get(TABLES), ","); //$NON-NLS-1$
while (st.hasMoreTokens()) {
String s = st.nextToken().trim();
if (s.length() > 0) {
fullyqualifiedTables.add(s);
}
}
}
Set<String> contexts = new HashSet<String>();
if (arguments.containsKey(CONTEXT_IDS)) {
StringTokenizer st = new StringTokenizer(
arguments.get(CONTEXT_IDS), ","); //$NON-NLS-1$
while (st.hasMoreTokens()) {
String s = st.nextToken().trim();
if (s.length() > 0) {
contexts.add(s);
}
}
}
try {
ConfigurationParser cp = new ConfigurationParser(warnings);
Configuration config = cp.parseConfiguration(configurationFile);
DefaultShellCallback shellCallback = new DefaultShellCallback(
arguments.containsKey(OVERWRITE));
MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, shellCallback, warnings);
ProgressCallback progressCallback = arguments.containsKey(VERBOSE) ? new VerboseProgressCallback()
: null;
myBatisGenerator.generate(progressCallback, contexts, fullyqualifiedTables);
} catch (XMLParserException e) {
writeLine(getString("Progress.3")); //$NON-NLS-1$
writeLine();
for (String error : e.getErrors()) {
writeLine(error);
}
return;
} catch (SQLException e) {
e.printStackTrace();
return;
} catch (IOException e) {
e.printStackTrace();
return;
} catch (InvalidConfigurationException e) {
writeLine(getString("Progress.16")); //$NON-NLS-1$
for (String error : e.getErrors()) {
writeLine(error);
}
return;
} catch (InterruptedException e) {
// ignore (will never happen with the DefaultShellCallback)
}
for (String warning : warnings) {
writeLine(warning);
}
if (warnings.size() == 0) {
writeLine(getString("Progress.4")); //$NON-NLS-1$
} else {
writeLine();
writeLine(getString("Progress.5")); //$NON-NLS-1$
}
}
private static void usage() {
String lines = getString("Usage.Lines"); //$NON-NLS-1$
int iLines = Integer.parseInt(lines);
for (int i = 0; i < iLines; i++) {
String key = "Usage." + i; //$NON-NLS-1$
writeLine(getString(key));
}
}
private static void writeLine(String message) {
System.out.println(message);
}
private static void writeLine() {
System.out.println();
}
private static Map<String, String> parseCommandLine(String[] args) {
List<String> errors = new ArrayList<String>();
Map<String, String> arguments = new HashMap<String, String>();
for (int i = 0; i < args.length; i++) {
if (CONFIG_FILE.equalsIgnoreCase(args[i])) {
if ((i + 1) < args.length) {
arguments.put(CONFIG_FILE, args[i + 1]);
} else {
errors.add(getString(
"RuntimeError.19", CONFIG_FILE)); //$NON-NLS-1$
}
i++;
} else if (OVERWRITE.equalsIgnoreCase(args[i])) {
arguments.put(OVERWRITE, "Y"); //$NON-NLS-1$
} else if (VERBOSE.equalsIgnoreCase(args[i])) {
arguments.put(VERBOSE, "Y"); //$NON-NLS-1$
} else if (HELP_1.equalsIgnoreCase(args[i])) {
arguments.put(HELP_1, "Y"); //$NON-NLS-1$
} else if (HELP_2.equalsIgnoreCase(args[i])) {
// put HELP_1 in the map here too - so we only
// have to check for one entry in the mainline
arguments.put(HELP_1, "Y"); //$NON-NLS-1$
} else if (FORCE_JAVA_LOGGING.equalsIgnoreCase(args[i])) {
LogFactory.forceJavaLogging();
} else if (CONTEXT_IDS.equalsIgnoreCase(args[i])) {
if ((i + 1) < args.length) {
arguments.put(CONTEXT_IDS, args[i + 1]);
} else {
errors.add(getString(
"RuntimeError.19", CONTEXT_IDS)); //$NON-NLS-1$
}
i++;
} else if (TABLES.equalsIgnoreCase(args[i])) {
if ((i + 1) < args.length) {
arguments.put(TABLES, args[i + 1]);
} else {
errors.add(getString("RuntimeError.19", TABLES)); //$NON-NLS-1$
}
i++;
} else {
errors.add(getString("RuntimeError.20", args[i])); //$NON-NLS-1$
}
}
if (!errors.isEmpty()) {
for (String error : errors) {
writeLine(error);
}
System.exit(-1);
}
return arguments;
}
}