/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.apache.axis2.wsdl.codegen.jaxws;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import org.apache.axiom.om.util.LogOutputStream;
import org.apache.axis2.util.CommandLineOption;
import org.apache.axis2.util.CommandLineOptionConstants;
import org.apache.axis2.util.CommandLineOptionParser;
import org.apache.axis2.wsdl.codegen.CodeGenConfiguration;
import org.apache.axis2.wsdl.codegen.CodeGenerationException;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import com.sun.tools.ws.wscompile.WsimportTool;
/**
* The Class JAXWSCodeGenerationEngine.
*/
public class JAXWSCodeGenerationEngine {
private static final Log log = LogFactory
.getLog(JAXWSCodeGenerationEngine.class);
private CodeGenConfiguration configuration;
private CommandLineOptionParser commandLineOptionParser;
private String[] originalArgs;
/**
* Instantiates a new jAXWS code generation engine.
*
* @param configuration
* the configuration
* @param originalArgs
* the original args
* @throws CodeGenerationException
* the code generation exception
*/
public JAXWSCodeGenerationEngine(CodeGenConfiguration configuration,
String[] originalArgs) throws CodeGenerationException {
this.configuration = configuration;
this.originalArgs = originalArgs;
// loadExtensions();
}
/**
* Instantiates a new jAXWS code generation engine.
*
* @param commandLineOptionParser
* the command line option parser
* @param originalArgs
* the original args
*/
public JAXWSCodeGenerationEngine(
CommandLineOptionParser commandLineOptionParser,
String[] originalArgs) {
this.commandLineOptionParser = commandLineOptionParser;
this.originalArgs = originalArgs;
// loadExtensions();
}
/**
* Generate.
*
* @throws CodeGenerationException
* the code generation exception
*/
public void generate() throws CodeGenerationException {
LogOutputStream logOutputStream = new LogOutputStream(log,
Integer.MAX_VALUE);
WsimportTool importTool = new WsimportTool(logOutputStream);
ArrayList<String> args = new ArrayList<String>();
configurImportToolOptions(args);
mergeOriginalArgs(args);
boolean success = importTool.run(args.toArray(new String[args.size()]));
if (success) {
log.info("Code generation completed");
}
}
/**
* Merge original args.
*
* @param args
* the args
*/
private void mergeOriginalArgs(ArrayList<String> args) {
Map<String, CommandLineOption> allOptions = commandLineOptionParser
.getAllOptions();
List<String> axisOptionList = new ArrayList<String>();
List<String> originalArgsOps = new ArrayList<String>(
Arrays.asList(originalArgs));
originalArgsOps
.remove("-"
.concat(CommandLineOptionConstants.WSDL2JavaConstants.JAX_WS_SERVICE_OPTION));
originalArgsOps
.remove("-"
.concat(CommandLineOptionConstants.WSDL2JavaConstants.JAX_WS_SERVICE_OPTION_LONG));
Field[] allFields = CommandLineOptionConstants.WSDL2JavaConstants.class
.getFields();
Iterator<String> mapItr = allOptions.keySet().iterator();
for (Field field : allFields) {
if (String.class.equals(field.getType())) {
try {
axisOptionList
.add((String) field
.get(CommandLineOptionConstants.WSDL2JavaConstants.class));
} catch (IllegalArgumentException e) {
e.printStackTrace();
} catch (IllegalAccessException e) {
e.printStackTrace();
}
}
}
while (mapItr.hasNext()) {
CommandLineOption op = allOptions.get(mapItr.next());
if (axisOptionList.contains(op.getOptionType())) {
if (op.getOptionType() != null) {
originalArgsOps.remove("-".concat(op.getOptionType()));
}
if (op.getOptionValue() != null) {
originalArgsOps.remove(op.getOptionValue());
}
}
}
args.addAll(originalArgsOps);
log.info("Original WSImport options" + Arrays.asList(originalArgsOps));
log.info("All WSImport options" + Arrays.asList(args));
}
/**
* Configur import tool options.
*
* @param args
* the args
*/
private void configurImportToolOptions(ArrayList<String> args) {
Map allOptions = commandLineOptionParser.getAllOptions();
// Set some default options
args.add(WS_IMPORT_EXTENSION);
args.add(WS_IMPORT_NO_COMPILE);
args.add(WS_IMPORT_KEEP_FILE);
// Set some properties based on AXIS2 WSDL2JAVA options
String uri = getOptionValue(
allOptions,
CommandLineOptionConstants.WSDL2JavaConstants.WSDL_LOCATION_URI_OPTION);
if (uri != null) {
args.add(uri);
}
String location = getOptionValue(
allOptions,
CommandLineOptionConstants.WSDL2JavaConstants.OUTPUT_LOCATION_OPTION);
if (location == null) {
location = getOptionValue(
allOptions,
CommandLineOptionConstants.WSDL2JavaConstants.OUTPUT_LOCATION_OPTION_LONG);
}
if (location != null) {
args.add(WS_IMPORT_FILE_OUTPUT_DIR);
args.add(location);
}
String pkg = getOptionValue(allOptions,
CommandLineOptionConstants.WSDL2JavaConstants.PACKAGE_OPTION);
if (pkg == null) {
pkg = getOptionValue(
allOptions,
CommandLineOptionConstants.WSDL2JavaConstants.PACKAGE_OPTION_LONG);
}
if (pkg != null) {
args.add(WS_IMPORT_PKG);
args.add(pkg);
}
String proxcyHost = getOptionValue(
allOptions,
CommandLineOptionConstants.WSDL2JavaConstants.HTTP_PROXY_HOST_OPTION_LONG);
String proxcyPort = getOptionValue(
allOptions,
CommandLineOptionConstants.WSDL2JavaConstants.HTTP_PROXY_PORT_OPTION_LONG);
if (pkg != null) {
args.add(WS_IMPORT_PROXY);
args.add(proxcyHost + ":" + proxcyPort);
}
}
/**
* Gets the option value.
*
* @param allOptions
* the all options
* @param optionName
* the option name
* @return the option value
*/
private static String getOptionValue(Map allOptions, String optionName) {
CommandLineOption option = (CommandLineOption) allOptions
.get(optionName);
if (option != null) {
return option.getOptionValue().toString();
}
return null;
}
public static final String WS_IMPORT_EXTENSION = "-extension";
public static final String WS_IMPORT_NO_COMPILE = "-Xnocompile";
public static final String WS_IMPORT_FILE_OUTPUT_DIR = "-d";
public static final String WS_IMPORT_SOURCE_OUTPUT_DIR = "-s";
public static final String WS_IMPORT_KEEP_FILE = "-keep";
public static final String WS_IMPORT_PKG = "-p";
public static final String WS_IMPORT_PROXY = "-httpproxy";
}