/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package com.sun.tools.internal.ws.wscompile;

import com.sun.codemodel.internal.JCodeModel;

import com.sun.tools.internal.ws.processor.generator.GeneratorExtension;

import com.sun.tools.internal.ws.resources.ConfigurationMessages;

import com.sun.tools.internal.ws.resources.WscompileMessages;

import com.sun.tools.internal.ws.util.ForkEntityResolver;

import com.sun.tools.internal.ws.wsdl.document.jaxws.JAXWSBindingsConstants;

import com.sun.tools.internal.ws.wsdl.document.schema.SchemaConstants;

import com.sun.tools.internal.xjc.api.SchemaCompiler;

import com.sun.tools.internal.xjc.api.SpecVersion;

import com.sun.tools.internal.xjc.api.XJC;

import com.sun.tools.internal.xjc.reader.Util;

import com.sun.xml.internal.ws.api.streaming.XMLStreamReaderFactory;

import com.sun.xml.internal.ws.streaming.XMLStreamReaderUtil;

import com.sun.xml.internal.ws.util.ServiceFinder;

import com.sun.xml.internal.ws.util.JAXWSUtils;

import com.sun.xml.internal.ws.util.xml.XmlUtil;

import org.w3c.dom.Element;

import org.xml.sax.EntityResolver;

import org.xml.sax.InputSource;

import org.xml.sax.helpers.LocatorImpl;

import javax.xml.namespace.QName;

import javax.xml.stream.XMLStreamReader;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.IOException;

import java.io.InputStream;

import java.io.Reader;

import java.lang.reflect.Array;

import java.net.MalformedURLException;

import java.net.URL;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import java.util.HashMap;

import java.util.logging.Level;

import java.util.logging.Logger;

/**

 * @author Vivek Pandey

 */

public class WsimportOptions extends Options {

    /**

     * -wsdlLocation

     */

    public String wsdlLocation;

    /**

     * Actually stores {@link com.sun.org.apache.xml.internal.resolver.tools.CatalogResolver}, but the field

     * type is made to {@link org.xml.sax.EntityResolver} so that XJC can be

     * used even if resolver.jar is not available in the classpath.

     */

    public EntityResolver entityResolver = null;

    /**

     * The -p option that should control the default Java package that

     * will contain the generated code. Null if unspecified.

     */

    public String defaultPackage = null;

    /**

     * The -clientjar option to package client artifacts as jar

     */

    public String clientjar = null;

    /**

     * -XadditionalHeaders

     */

    public boolean additionalHeaders;

    /**

     * The option indicates the dir where the jwsImpl will be generated.

     */

    public File implDestDir = null;

    /**

     * optional, generated impl file only for the ordered serviceName

     * Note: It is a QName string, formatted as: "{" + Namespace URI + "}" + local part

     */

    public String implServiceName = null;

    /**

     * optional, generated impl file only for the ordered portName

     * Note: It is a QName string, formatted as: "{" + Namespace URI + "}" + local part

     */

    public String implPortName = null;

    /**

     * optional, if true JWS file is generated

     */

    public boolean isGenerateJWS = false;

    /**

     * Setting disableSSLHostVerification to true disables the SSL Hostname verification while fetching the wsdls.

     * -XdisableSSLHostVerification

     */

    public boolean disableSSLHostnameVerification;

    /**

     * Setting useBaseResourceAndURLToLoadWSDL to true causes generated Service classes to load the WSDL file from

     * a URL generated from the base resource.

     * -XuseBaseResourceAndURLToLoadWSDL

     */

    public boolean useBaseResourceAndURLToLoadWSDL = false;

    /**

     * JAXB's {@link SchemaCompiler} to be used for handling the schema portion.

     * This object is also configured through options.

     */

    private SchemaCompiler schemaCompiler = XJC.createSchemaCompiler();

    /**

     * Authentication file

     */

    public File authFile = null;

    //can user.home value be null?

    public static final String defaultAuthfile

            = System.getProperty("user.home") + System.getProperty("file.separator")

            + ".metro" + System.getProperty("file.separator") + "auth";

    /**

     * Setting disableAuthenticator to true disables the DefaultAuthenticator.

     * -XdisableAuthenticator

     */

    public boolean disableAuthenticator;

    public String proxyAuth = null;

    private String proxyHost = null;

    private String proxyPort = null;

    /**

     * Additional arguments

     */

    public HashMap<String, String> extensionOptions = new HashMap<String, String>();

    /**

     * All discovered {@link Plugin}s.

     * This is lazily parsed, so that we can take '-cp' option into account.

     *

     * @see #getAllPlugins()

     */

    private List<Plugin> allPlugins;

    /**

     * {@link Plugin}s that are enabled in this compilation.

     */

    public final List<Plugin> activePlugins = new ArrayList<Plugin>();

    public JCodeModel getCodeModel() {

        if(codeModel == null)

            codeModel = new JCodeModel();

        return codeModel;

    }

    public SchemaCompiler getSchemaCompiler() {

        schemaCompiler.setTargetVersion(SpecVersion.parse(target.getVersion()));

        if(entityResolver != null) {

            //set if its not null so as not to override catalog option specified via xjc args

            schemaCompiler.setEntityResolver(entityResolver);

        }

        return schemaCompiler;

    }

    public void setCodeModel(JCodeModel codeModel) {

        this.codeModel = codeModel;

    }

    private JCodeModel codeModel;

    /**

     * This captures jars passed on the commandline and passes them to XJC and puts them in the classpath for compilation

     */

    public List<String> cmdlineJars = new ArrayList<String>();

    /**

     * Gets all the {@link Plugin}s discovered so far.

     *

     * <p>

     * A plugins are enumerated when this method is called for the first time,

     * by taking {@link #classpath} into account. That means

     * "-cp plugin.jar" has to come before you specify options to enable it.

     */

    public List<Plugin> getAllPlugins() {

        if(allPlugins==null) {

            allPlugins = new ArrayList<Plugin>();

            allPlugins.addAll(Arrays.asList(findServices(Plugin.class, getClassLoader())));

        }

        return allPlugins;

    }

    /**

     * Parses arguments and fill fields of this object.

     *

     * @exception BadCommandLineException

     *      thrown when there's a problem in the command-line arguments

     */

    @Override

    public final void parseArguments( String[] args ) throws BadCommandLineException {

        for (int i = 0; i < args.length; i++) {

            if(args[i].length()==0)

                throw new BadCommandLineException();

            if (args[i].charAt(0) == '-') {

                int j = parseArguments(args,i);

                if(j==0)

                    throw new BadCommandLineException(WscompileMessages.WSCOMPILE_INVALID_OPTION(args[i]));

                i += (j-1);

            } else {

                if(args[i].endsWith(".jar")) {

                    try {

                cmdlineJars.add(args[i]);

                schemaCompiler.getOptions().scanEpisodeFile(new File(args[i]));

            } catch (com.sun.tools.internal.xjc.BadCommandLineException e) {

                //Driver.usage(jaxbOptions,false);

                throw new BadCommandLineException(e.getMessage(), e);

            }

                } else{

                    addFile(args[i]);

                }

            }

        }

        if (encoding != null && schemaCompiler.getOptions().encoding == null) {

            try {

                schemaCompiler.getOptions().parseArgument(

                        new String[] {"-encoding", encoding}, 0);

            } catch (com.sun.tools.internal.xjc.BadCommandLineException ex) {

                Logger.getLogger(WsimportOptions.class.getName()).log(Level.SEVERE, null, ex);

            }

        }

        if(destDir == null)

            destDir = new File(".");

        if(sourceDir == null)

            sourceDir = destDir;

    }

    /** -Xno-addressing-databinding option to disable addressing namespace data binding. This is

     * experimental switch and will be working as a temporary workaround till

     * jaxb can provide a better way to selelctively disable compiling of an

     * schema component.

     * **/

    public boolean noAddressingBbinding;

    @Override

    public int parseArguments(String[] args, int i) throws BadCommandLineException {

        int j = super.parseArguments(args ,i);

        if(j>0) return j;   // understood by the super class

        if (args[i].equals("-b")) {

            addBindings(requireArgument("-b", args, ++i));

            return 2;

        } else if (args[i].equals("-wsdllocation")) {

            wsdlLocation = requireArgument("-wsdllocation", args, ++i);

            return 2;

        } else if (args[i].equals("-XadditionalHeaders")) {

            additionalHeaders = true;

            return 1;

        } else if (args[i].equals("-XdisableSSLHostnameVerification")) {

            disableSSLHostnameVerification = true;

            return 1;

        } else if (args[i].equals("-p")) {

            defaultPackage = requireArgument("-p", args, ++i);

            return 2;

        } else if (args[i].equals("-catalog")) {

            String catalog = requireArgument("-catalog", args, ++i);

            try {

                if (entityResolver == null) {

                    if (catalog != null && catalog.length() > 0)

                        entityResolver = XmlUtil.createEntityResolver(JAXWSUtils.getFileOrURL(JAXWSUtils.absolutize(Util.escapeSpace(catalog))));

                } else if (catalog != null && catalog.length() > 0) {

                    EntityResolver er = XmlUtil.createEntityResolver(JAXWSUtils.getFileOrURL(JAXWSUtils.absolutize(Util.escapeSpace(catalog))));

                    entityResolver = new ForkEntityResolver(er, entityResolver);

                }

            } catch (IOException e) {

                throw new BadCommandLineException(WscompileMessages.WSIMPORT_FAILED_TO_PARSE(catalog, e.getMessage()));

            }

            return 2;

        } else if (args[i].startsWith("-httpproxy:")) {

            String value = args[i].substring(11);

            if (value.length() == 0) {

                throw new BadCommandLineException(WscompileMessages.WSCOMPILE_INVALID_OPTION(args[i]));

            }

            parseProxy(value);

            if (proxyHost != null || proxyPort != null) {

                System.setProperty("proxySet", "true");

            }

            if (proxyHost != null) {

                System.setProperty("proxyHost", proxyHost);

            }

            if (proxyPort != null) {

                System.setProperty("proxyPort", proxyPort);

            }

            return 1;

        } else if (args[i].equals("-Xno-addressing-databinding")) {

            noAddressingBbinding = true;

            return 1;

        } else if (args[i].startsWith("-B")) {

            // JAXB option pass through.

            String[] subCmd = new String[args.length-i];

            System.arraycopy(args,i,subCmd,0,subCmd.length);

            subCmd[0] = subCmd[0].substring(2); // trim off the first "-B"

            com.sun.tools.internal.xjc.Options jaxbOptions = schemaCompiler.getOptions();

            try {

                int r = jaxbOptions.parseArgument(subCmd, 0);

                if(r==0) {

                    //Driver.usage(jaxbOptions,false);

                    throw new BadCommandLineException(WscompileMessages.WSIMPORT_NO_SUCH_JAXB_OPTION(subCmd[0]));

                }

                return r;

            } catch (com.sun.tools.internal.xjc.BadCommandLineException e) {

                //Driver.usage(jaxbOptions,false);

                throw new BadCommandLineException(e.getMessage(),e);

            }

        } else if (args[i].equals("-Xauthfile")) {

            String authfile = requireArgument("-Xauthfile", args, ++i);

            authFile = new File(authfile);

            return 2;

        } else if (args[i].equals("-clientjar")) {

            clientjar = requireArgument("-clientjar", args, ++i);

            return 2;

        } else if (args[i].equals("-implDestDir")) {

                        implDestDir = new File(requireArgument("-implDestDir", args, ++i));

            if (!implDestDir.exists())

              throw new BadCommandLineException(WscompileMessages.WSCOMPILE_NO_SUCH_DIRECTORY(implDestDir.getPath()));

                        return 2;

        } else if (args[i].equals("-implServiceName")) {

                implServiceName = requireArgument("-implServiceName", args, ++i);

          return 2;

        } else if (args[i].equals("-implPortName")) {

                implPortName = requireArgument("-implPortName", args, ++i);

          return 2;

        } else if (args[i].equals("-generateJWS")) {

            isGenerateJWS = true;

            return 1;

        } else if (args[i].equals("-XuseBaseResourceAndURLToLoadWSDL")) {

                useBaseResourceAndURLToLoadWSDL = true;

            return 1;

        } else if (args[i].equals("-XdisableAuthenticator")) {

            disableAuthenticator = true;

            return 1;

        }

        // handle additional options

        for (GeneratorExtension f:ServiceFinder.find(GeneratorExtension.class)) {

            if (f.validateOption(args[i])) {

                extensionOptions.put(args[i], requireArgument(args[i], args, ++i));

                return 2;

            }

        }

        // see if this is one of the extensions

        for( Plugin plugin : getAllPlugins() ) {

            try {

                if(('-' + plugin.getOptionName()).equals(args[i])) {

                    activePlugins.add(plugin);

                    plugin.onActivated(this);

                    return 1;

                }

                int r = plugin.parseArgument(this, args, i);

                if (r != 0) {

                    return r;

                }

            } catch (IOException e) {

                throw new BadCommandLineException(e.getMessage(),e);

            }

        }

        return 0; // what's this option?

    }

    public void validate() throws BadCommandLineException {

        if (wsdls.isEmpty()) {

            throw new BadCommandLineException(WscompileMessages.WSIMPORT_MISSING_FILE());

        }

        if(wsdlLocation !=null && clientjar != null) {

           throw new BadCommandLineException(WscompileMessages.WSIMPORT_WSDLLOCATION_CLIENTJAR());

        }

        if(wsdlLocation == null){

            wsdlLocation = wsdls.get(0).getSystemId();

        }

    }

    @Override

    protected void addFile(String arg) throws BadCommandLineException {

        addFile(arg, wsdls, ".wsdl");

    }

    private final List<InputSource> wsdls = new ArrayList<InputSource>();

    private final List<InputSource> schemas = new ArrayList<InputSource>();

    private final List<InputSource> bindingFiles = new ArrayList<InputSource>();

    private final List<InputSource> jaxwsCustomBindings = new ArrayList<InputSource>();

    private final List<InputSource> jaxbCustomBindings = new ArrayList<InputSource>();

    private final List<Element> handlerConfigs = new ArrayList<Element>();

    /**

     * There is supposed to be one handler chain per generated SEI.

     * TODO: There is possible bug, how to associate a @HandlerChain

     * with each port on the generated SEI. For now lets preserve the JAXWS 2.0 FCS

     * behaviour and generate only one @HandlerChain on the SEI

     */

    public Element getHandlerChainConfiguration(){

        if(handlerConfigs.size() > 0)

            return handlerConfigs.get(0);

        return null;

    }

    public void addHandlerChainConfiguration(Element config){

        handlerConfigs.add(config);

    }

    public InputSource[] getWSDLs() {

        return wsdls.toArray(new InputSource[wsdls.size()]);

    }

    public InputSource[] getSchemas() {

        return schemas.toArray(new InputSource[schemas.size()]);

    }

    public InputSource[] getWSDLBindings() {

        return jaxwsCustomBindings.toArray(new InputSource[jaxwsCustomBindings.size()]);

    }

    public InputSource[] getSchemaBindings() {

        return jaxbCustomBindings.toArray(new InputSource[jaxbCustomBindings.size()]);

    }

    public void addWSDL(File source) {

        addWSDL(fileToInputSource(source));

    }

    public void addWSDL(InputSource is) {

        wsdls.add(absolutize(is));

    }

    public void addSchema(File source) {

        addSchema(fileToInputSource(source));

    }

    public void addSchema(InputSource is) {

        schemas.add(is);

    }

    private InputSource fileToInputSource(File source) {

        try {

            String url = source.toURL().toExternalForm();

            return new InputSource(Util.escapeSpace(url));

        } catch (MalformedURLException e) {

            return new InputSource(source.getPath());

        }

    }

    /**

     * Recursively scan directories and add all XSD files in it.

     */

    public void addGrammarRecursive(File dir) {

        addRecursive(dir, ".wsdl", wsdls);

        addRecursive(dir, ".xsd", schemas);

    }

    /**

     * Adds a new input schema.

     */

    public void addWSDLBindFile(InputSource is) {

        jaxwsCustomBindings.add(new RereadInputSource(absolutize(is)));

    }

    public void addSchemmaBindFile(InputSource is) {

        jaxbCustomBindings.add(new RereadInputSource(absolutize(is)));

    }

    private void addRecursive(File dir, String suffix, List<InputSource> result) {