JAXWS2WSDLCodegenEngine.java

/*
 * 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.ws.java2wsdl.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.description.java2wsdl.Java2WSDLConstants;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.ws.java2wsdl.utils.Java2WSDLCommandLineOption;

import com.sun.tools.ws.wscompile.WsgenTool;

/**
 * The Class JAXWS2WSDLCodegenEngine.
 */
public class JAXWS2WSDLCodegenEngine {

    /** The Constant log. */
    private static final Log log = LogFactory
            .getLog(JAXWS2WSDLCodegenEngine.class);

    /** The options map. */
    private Map<String, Java2WSDLCommandLineOption> optionsMap;

    /** The original args. */
    private String[] originalArgs;

    /**
     * Instantiates a new jAXW s2 wsdl codegen engine.
     * 
     * @param optionsMap
     *            the options map
     * @param originalArgs
     *            the original args
     */
    public JAXWS2WSDLCodegenEngine(
            Map<String, Java2WSDLCommandLineOption> optionsMap,
            String[] originalArgs) {
        this.optionsMap = optionsMap;
        this.originalArgs = originalArgs;
    }

    /**
     * Generate.
     * 
     * @throws Exception
     *             the exception
     */
    public void generate() throws Exception {
        LogOutputStream logOutputStream = new LogOutputStream(log,
                Integer.MAX_VALUE);
        WsgenTool genTool = new WsgenTool(logOutputStream);

        List<String> args = new ArrayList<String>();
        configurImportToolOptions(args);
        mergeOriginalArgs(args);
        boolean success = genTool.run(args.toArray(new String[args.size()]));
        if (success) {
            log.info("Code generation completed");
        }

    }

    /**
     * Configur import tool options.
     * 
     * @param args
     *            the args
     */
    private void configurImportToolOptions(List<String> args) {

        // Set some default options
        args.add(WS_GEN_EXTENSION);
        args.add(WS_GEN_NO_COMPILE);
        args.add(WS_GEN_KEEP_FILE);
        args.add(WS_GEN_WSDL);

        // Set some properties based on AXIS2 JAVA2WSDL options
        String location = getOptionValue(optionsMap,
                Java2WSDLConstants.OUTPUT_LOCATION_OPTION);
        if (location == null) {
            location = getOptionValue(optionsMap,
                    Java2WSDLConstants.OUTPUT_LOCATION_OPTION_LONG);
        }
        if (location != null) {
            args.add(WS_GEN_FILE_OUTPUT_DIR);
            args.add(location);
        }

        String classPath = getOptionValue(optionsMap,
                Java2WSDLConstants.CLASSPATH_OPTION);
        if (classPath == null) {
            classPath = getOptionValue(optionsMap,
                    Java2WSDLConstants.CLASSPATH_OPTION_LONG);
        }
        if (classPath != null) {
            args.add(WS_GEN_CLASSPATH);
            args.add(classPath);
        }

        String className = getOptionValue(optionsMap,
                Java2WSDLConstants.CLASSNAME_OPTION);
        if (className == null) {
            className = getOptionValue(optionsMap,
                    Java2WSDLConstants.CLASSNAME_OPTION_LONG);
        }
        if (className != null) {
            args.add(className);
        }

    }

    /**
     * Merge original args.
     * 
     * @param args
     *            the args
     */
    private void mergeOriginalArgs(List<String> args) {

        List<String> axisOptionList = new ArrayList<String>();
        List<String> originalArgsOps = new ArrayList<String>(
                Arrays.asList(originalArgs));
        originalArgsOps.remove("-"
                .concat(Java2WSDLConstants.JAX_WS_SERVICE_OPTION));
        originalArgsOps.remove("-"
                .concat(Java2WSDLConstants.JAX_WS_SERVICE_OPTION_LONG));
        Field[] allFields = Java2WSDLConstants.class.getFields();
        Iterator<String> mapItr = optionsMap.keySet().iterator();
        for (Field field : allFields) {
            if (String.class.equals(field.getType())) {
                try {
                    axisOptionList.add((String) field
                            .get(Java2WSDLConstants.class));
                } catch (IllegalArgumentException e) {
                    e.printStackTrace();
                } catch (IllegalAccessException e) {
                    e.printStackTrace();
                }
            }
        }

        while (mapItr.hasNext()) {
            Java2WSDLCommandLineOption op = optionsMap.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.debug("Original WSImport options" + Arrays.asList(originalArgsOps));
        log.debug("All WSImport options" + Arrays.asList(args));

    }

    /**
     * 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) {
        Java2WSDLCommandLineOption option = (Java2WSDLCommandLineOption) allOptions
                .get(optionName);
        if (option != null) {
            return option.getOptionValue().toString();
        }
        return null;
    }

    public static final String WS_GEN_FILE_OUTPUT_DIR = "-d";
    public static final String WS_GEN_CLASSPATH = "-cp";
    public static final String WS_GEN_KEEP_FILE = "-keep";
    public static final String WS_GEN_WSDL = "-wsdl";
    public static final String WS_GEN_EXTENSION = "-extension";
    public static final String WS_GEN_NO_COMPILE = "-Xnocompile";

}