jaxws/src/jdk.xml.bind/share/classes/com/sun/tools/internal/jxc/SchemaGenerator.java
changeset 25871 b80b84e87032
parent 16791 fe5141eabb0e
child 27837 86d4f46e622a
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jaxws/src/jdk.xml.bind/share/classes/com/sun/tools/internal/jxc/SchemaGenerator.java	Sun Aug 17 15:52:15 2014 +0100
@@ -0,0 +1,335 @@
+/*
+ * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
+ * 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.jxc;
+
+import com.sun.tools.internal.jxc.ap.Options;
+import com.sun.tools.internal.xjc.BadCommandLineException;
+import com.sun.xml.internal.bind.util.Which;
+
+import javax.lang.model.SourceVersion;
+import javax.tools.DiagnosticCollector;
+import javax.tools.JavaCompiler;
+import javax.tools.JavaFileObject;
+import javax.tools.OptionChecker;
+import javax.tools.StandardJavaFileManager;
+import javax.tools.ToolProvider;
+import javax.xml.bind.JAXBContext;
+import java.io.File;
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
+import java.net.MalformedURLException;
+import java.net.URISyntaxException;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+/**
+ * CLI entry-point to the schema generator.
+ *
+ * @author Bhakti Mehta
+ */
+public class SchemaGenerator {
+    /**
+     * Runs the schema generator.
+     */
+    public static void main(String[] args) throws Exception {
+        System.exit(run(args));
+    }
+
+    public static int run(String[] args) throws Exception {
+        try {
+            ClassLoader cl = SecureLoader.getClassClassLoader(SchemaGenerator.class);
+            if (cl==null) {
+                cl = SecureLoader.getSystemClassLoader();
+            }
+            return run(args, cl);
+        } catch(Exception e) {
+            System.err.println(e.getMessage());
+            return -1;
+        }
+    }
+
+    /**
+     * Runs the schema generator.
+     *
+     * @param classLoader
+     *      the schema generator will run in this classLoader.
+     *      It needs to be able to load annotation processing and JAXB RI classes. Note that
+     *      JAXB RI classes refer to annotation processing classes. Must not be null.
+     *
+     * @return
+     *      exit code. 0 if success.
+     *
+     */
+    public static int run(String[] args, ClassLoader classLoader) throws ClassNotFoundException, NoSuchMethodException, IllegalAccessException, InvocationTargetException {
+        final Options options = new Options();
+        if (args.length ==0) {
+            usage();
+            return -1;
+        }
+        for (String arg : args) {
+            if (arg.equals("-help")) {
+                usage();
+                return -1;
+            }
+
+            if (arg.equals("-version")) {
+                System.out.println(Messages.VERSION.format());
+                return -1;
+            }
+
+            if (arg.equals("-fullversion")) {
+                System.out.println(Messages.FULLVERSION.format());
+                return -1;
+            }
+
+        }
+
+        try {
+            options.parseArguments(args);
+        } catch (BadCommandLineException e) {
+            // there was an error in the command line.
+            // print usage and abort.
+            System.out.println(e.getMessage());
+            System.out.println();
+            usage();
+            return -1;
+        }
+
+        Class schemagenRunner = classLoader.loadClass(Runner.class.getName());
+        Method compileMethod = schemagenRunner.getDeclaredMethod("compile",String[].class,File.class);
+
+        List<String> aptargs = new ArrayList<String>();
+
+        if (options.encoding != null) {
+            aptargs.add("-encoding");
+            aptargs.add(options.encoding);
+        }
+
+        aptargs.add("-cp");
+        aptargs.add(setClasspath(options.classpath)); // set original classpath + jaxb-api to be visible to annotation processor
+
+        if(options.targetDir!=null) {
+            aptargs.add("-d");
+            aptargs.add(options.targetDir.getPath());
+        }
+
+        aptargs.addAll(options.arguments);
+
+        String[] argsarray = aptargs.toArray(new String[aptargs.size()]);
+        return ((Boolean) compileMethod.invoke(null, argsarray, options.episodeFile)) ? 0 : 1;
+    }
+
+    private static String setClasspath(String givenClasspath) {
+        StringBuilder cp = new StringBuilder();
+        appendPath(cp, givenClasspath);
+        ClassLoader cl = Thread.currentThread().getContextClassLoader();
+        while (cl != null) {
+            if (cl instanceof URLClassLoader) {
+                for (URL url : ((URLClassLoader) cl).getURLs()) {
+                    appendPath(cp, url.getPath());
+                }
+            }
+            cl = cl.getParent();
+        }
+
+        appendPath(cp, findJaxbApiJar());
+        return cp.toString();
+    }
+
+    private static void appendPath(StringBuilder cp, String url) {
+        if (url == null || url.trim().isEmpty())
+            return;
+        if (cp.length() != 0)
+            cp.append(File.pathSeparatorChar);
+        cp.append(url);
+    }
+
+    /**
+     * Computes the file system path of <tt>jaxb-api.jar</tt> so that
+     * Annotation Processing will see them in the <tt>-cp</tt> option.
+     *
+     * <p>
+     * In Java, you can't do this reliably (for that matter there's no guarantee
+     * that such a jar file exists, such as in Glassfish), so we do the best we can.
+     *
+     * @return
+     *      null if failed to locate it.
+     */
+    private static String findJaxbApiJar() {
+        String url = Which.which(JAXBContext.class);
+        if(url==null)       return null;    // impossible, but hey, let's be defensive
+
+        if(!url.startsWith("jar:") || url.lastIndexOf('!')==-1)
+            // no jar file
+            return null;
+
+        String jarFileUrl = url.substring(4,url.lastIndexOf('!'));
+        if(!jarFileUrl.startsWith("file:"))
+            return null;    // not from file system
+
+        try {
+            File f = new File(new URL(jarFileUrl).toURI());
+            if (f.exists() && f.getName().endsWith(".jar")) { // see 6510966
+                return f.getPath();
+            }
+            f = new File(new URL(jarFileUrl).getFile());
+            if (f.exists() && f.getName().endsWith(".jar")) { // this is here for potential backw. compatibility issues
+                return f.getPath();
+            }
+        } catch (URISyntaxException ex) {
+            Logger.getLogger(SchemaGenerator.class.getName()).log(Level.SEVERE, null, ex);
+        } catch (MalformedURLException ex) {
+            Logger.getLogger(SchemaGenerator.class.getName()).log(Level.SEVERE, null, ex);
+        }
+        return null;
+    }
+
+    private static void usage( ) {
+        System.out.println(Messages.USAGE.format());
+    }
+
+    public static final class Runner {
+        public static boolean compile(String[] args, File episode) throws Exception {
+
+            JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
+            DiagnosticCollector<JavaFileObject> diagnostics = new DiagnosticCollector<JavaFileObject>();
+            StandardJavaFileManager fileManager = compiler.getStandardFileManager(diagnostics, null, null);
+            JavacOptions options = JavacOptions.parse(compiler, fileManager, args);
+            List<String> unrecognizedOptions = options.getUnrecognizedOptions();
+            if (!unrecognizedOptions.isEmpty())
+                Logger.getLogger(SchemaGenerator.class.getName()).log(Level.WARNING, "Unrecognized options found: {0}", unrecognizedOptions);
+            Iterable<? extends JavaFileObject> compilationUnits = fileManager.getJavaFileObjectsFromFiles(options.getFiles());
+            JavaCompiler.CompilationTask task = compiler.getTask(
+                    null,
+                    fileManager,
+                    diagnostics,
+                    options.getRecognizedOptions(),
+                    options.getClassNames(),
+                    compilationUnits);
+            com.sun.tools.internal.jxc.ap.SchemaGenerator r = new com.sun.tools.internal.jxc.ap.SchemaGenerator();
+            if (episode != null)
+                r.setEpisodeFile(episode);
+            task.setProcessors(Collections.singleton(r));
+            return task.call();
+        }
+    }
+
+    /**
+          *  @author Peter von der Ahe
+          */
+    private static final class JavacOptions {
+        private final List<String> recognizedOptions;
+        private final List<String> classNames;
+        private final List<File> files;
+        private final List<String> unrecognizedOptions;
+
+        private JavacOptions(List<String> recognizedOptions, List<String> classNames, List<File> files,
+                             List<String> unrecognizedOptions) {
+            this.recognizedOptions = recognizedOptions;
+            this.classNames = classNames;
+            this.files = files;
+            this.unrecognizedOptions = unrecognizedOptions;
+        }
+
+        public static JavacOptions parse(OptionChecker primary, OptionChecker secondary, String... arguments) {
+            List<String> recognizedOptions = new ArrayList<String>();
+            List<String> unrecognizedOptions = new ArrayList<String>();
+            List<String> classNames = new ArrayList<String>();
+            List<File> files = new ArrayList<File>();
+            for (int i = 0; i < arguments.length; i++) {
+                String argument = arguments[i];
+                int optionCount = primary.isSupportedOption(argument);
+                if (optionCount < 0) {
+                    optionCount = secondary.isSupportedOption(argument);
+                }
+                if (optionCount < 0) {
+                    File file = new File(argument);
+                    if (file.exists())
+                        files.add(file);
+                    else if (SourceVersion.isName(argument))
+                        classNames.add(argument);
+                    else
+                        unrecognizedOptions.add(argument);
+                } else {
+                    for (int j = 0; j < optionCount + 1; j++) {
+                        int index = i + j;
+                        if (index == arguments.length) throw new IllegalArgumentException(argument);
+                        recognizedOptions.add(arguments[index]);
+                    }
+                    i += optionCount;
+                }
+            }
+            return new JavacOptions(recognizedOptions, classNames, files, unrecognizedOptions);
+        }
+
+        /**
+                     * Returns the list of recognized options and their arguments.
+                     *
+                     * @return a list of options
+                     */
+        public List<String> getRecognizedOptions() {
+            return Collections.unmodifiableList(recognizedOptions);
+        }
+
+        /**
+                     * Returns the list of file names.
+                     *
+                     * @return a list of file names
+                     */
+        public List<File> getFiles() {
+            return Collections.unmodifiableList(files);
+        }
+
+        /**
+                     * Returns the list of class names.
+                     *
+                     * @return a list of class names
+                     */
+        public List<String> getClassNames() {
+            return Collections.unmodifiableList(classNames);
+        }
+
+        /**
+                     * Returns the list of unrecognized options.
+                     *
+                     * @return a list of unrecognized options
+                     */
+        public List<String> getUnrecognizedOptions() {
+            return Collections.unmodifiableList(unrecognizedOptions);
+        }
+
+        @Override
+        public String toString() {
+            return String.format("recognizedOptions = %s; classNames = %s; " + "files = %s; unrecognizedOptions = %s", recognizedOptions, classNames, files, unrecognizedOptions);
+        }
+    }
+}