modules/adb-codegen/src/org/apache/axis2/schema/XSD2Java.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.axis2.schema;

import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.GnuParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.OptionBuilder;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.apache.ws.commons.schema.XmlSchema;
import org.apache.ws.commons.schema.XmlSchemaCollection;
import org.w3c.dom.Document;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import static org.apache.axis2.schema.i18n.SchemaCompilerMessages.getMessage;

public class XSD2Java {
    private static Options options;
    private static CommandLine line;

    /**
     * for now the arguments this main method accepts is the source schema and the output
     * location
     *
     * @param args
     */
    @SuppressWarnings("static-access")
    public static void main(String[] args) throws Exception {
        options = new Options();
        options.addOption(OptionBuilder.withArgName(getMessage("schema.ns2p.argname"))
                                       .hasArgs(2)
                                       .withValueSeparator()
                                       .withDescription(getMessage("schema.ns2p.description"))
                                       .create("ns2p"));
        options.addOption(OptionBuilder.withArgName(getMessage("schema.mp.argname"))
                                       .hasArg()
                                       .withDescription(getMessage("schema.mp.description"))
                                       .create("mp"));
        options.addOption(OptionBuilder.withArgName(getMessage("schema.dp.argname"))
                                       .hasArg()
                                       .withDescription(getMessage("schema.dp.description"))
                                       .create("dp"));
        options.addOption(OptionBuilder.withDescription(getMessage("schema.h.description"))
                                       .create("h"));
        options.addOption(OptionBuilder.withArgName(getMessage("schema.p.argname"))
                                       .hasArg()
                                       .withDescription(getMessage("schema.p.description"))
                                       .create("p"));
        CommandLineParser parser = new GnuParser();
        try {
            line = parser.parse(options, args);
        } catch (ParseException ex) {
            System.out.println(ex.getLocalizedMessage());
            System.out.println();
            printUsage();
            System.out.println(ex);
            System.exit(1);
        }
        args = line.getArgs();
        if (args.length < 2) {
            printUsage();
            System.exit(1);
        } else {
            File outputFolder = new File(args[args.length-1]);
            for (int i=0; i<args.length-1; i++) {
                File xsdFile = new File(args[i]);
                // Only output a message if the user has specified more than one schema file
                if (args.length > 2) {
                    System.out.println(getMessage("schema.compiling", xsdFile.getName()));
                }
                compile(xsdFile, outputFolder);
            }
        }
    }
    
    private static void printUsage() {
        HelpFormatter formatter = new HelpFormatter();
        formatter.printHelp(getMessage("schema.usage"), options);
    }

    /**
     * @param xsdName
     * @param outputLocation
     */
    private static void compile(File xsdFile, File outputFolder) throws Exception {
            //load the current Schema through a file
            //first read the file into a DOM
            DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();
            documentBuilderFactory.setNamespaceAware(true);

            DocumentBuilder builder = documentBuilderFactory.newDocumentBuilder();
            Document doc = builder.parse(xsdFile);

            //now read it to a schema
            XmlSchemaCollection schemaCol = new XmlSchemaCollection();
            XmlSchema currentSchema = schemaCol.read(doc, xsdFile.toURI().toString(), null);

            if (outputFolder.exists()) {
                if (outputFolder.isFile()) {
                    throw new IOException(getMessage("schema.locationNotFolder"));
                }
            } else {
                outputFolder.mkdirs();
            }

            CompilerOptions compilerOptions = new CompilerOptions();
            compilerOptions.setOutputLocation(outputFolder);
            compilerOptions.setGenerateAll(true);

            Map ns2p = new HashMap();
            if (line.hasOption("ns2p")) {
                ns2p.putAll(line.getOptionProperties("ns2p"));
            }
            if (line.hasOption("dp")) {
                ns2p.put("", line.getOptionValue("dp"));
            }
            if (!ns2p.isEmpty()) {
                compilerOptions.setNs2PackageMap(ns2p);
            }
            
            if (line.hasOption("mp")) {
                compilerOptions.setMapperClassPackage(line.getOptionValue("mp"));
            }

            compilerOptions.setHelperMode(line.hasOption("h"));
            
            if (line.hasOption("p")) {
                compilerOptions.setPackageName(line.getOptionValue("p"));
            }
            
            //todo - this should come from the users preferences
             compilerOptions.setWrapClasses(false);

            //there's no point in not writing the classes here.
             compilerOptions.setWriteOutput(true);

//             compilerOptions.setUseWrapperClasses(true);

            SchemaCompiler compiler = new SchemaCompiler(compilerOptions);
            compiler.compile(currentSchema);
    }
}