org.apache.axis2.schema.XSD2Java.java Source code

Introduction

Here is the source code for org.apache.axis2.schema.XSD2Java.java

Source

/*
 * 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);
    }
}