0N/A/*
553N/A * Copyright (c) 2004, 2007, Oracle and/or its affiliates. All rights reserved.
0N/A * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
0N/A *
0N/A * This code is free software; you can redistribute it and/or modify it
0N/A * under the terms of the GNU General Public License version 2 only, as
0N/A * published by the Free Software Foundation.
0N/A *
0N/A * This code is distributed in the hope that it will be useful, but WITHOUT
0N/A * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
0N/A * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
0N/A * version 2 for more details (a copy is included in the LICENSE file that
0N/A * accompanied this code).
0N/A *
0N/A * You should have received a copy of the GNU General Public License version
0N/A * 2 along with this work; if not, write to the Free Software Foundation,
0N/A * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
0N/A *
553N/A * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
553N/A * or visit www.oracle.com if you need additional information or have any
553N/A * questions.
0N/A */
0N/A
0N/A
0N/Aimport com.sun.mirror.apt.*;
0N/Aimport com.sun.mirror.declaration.*;
0N/Aimport com.sun.mirror.type.*;
0N/Aimport com.sun.mirror.util.*;
0N/A
0N/Aimport java.util.Collection;
0N/Aimport java.util.Set;
0N/Aimport java.util.Map;
0N/Aimport java.util.Arrays;
0N/Aimport java.util.Collections;
0N/A
0N/Apublic class OptionChecker implements AnnotationProcessorFactory {
0N/A static class OptionCheck implements AnnotationProcessor {
0N/A AnnotationProcessorEnvironment ape;
0N/A OptionCheck(AnnotationProcessorEnvironment ape) {
0N/A this.ape = ape;
0N/A }
0N/A
0N/A public void process() {
0N/A Map<String, String> options = ape.getOptions();
0N/A if (options.containsKey("-Afoo") &&
0N/A options.containsKey("-Abar") &&
0N/A options.containsKey("-classpath") ) {
0N/A System.out.println("Expected options found.");
0N/A return; // All is well
0N/A } else {
0N/A System.err.println("Unexpected options values: " + options);
0N/A throw new RuntimeException();
0N/A }
0N/A }
0N/A }
0N/A
0N/A static class HelloWorld implements AnnotationProcessor {
0N/A AnnotationProcessorEnvironment ape;
0N/A HelloWorld(AnnotationProcessorEnvironment ape) {
0N/A this.ape = ape;
0N/A }
0N/A
0N/A public void process() {
0N/A java.io.PrintWriter pw;
0N/A try {
0N/A pw = ape.getFiler().createSourceFile("HelloWorld");
0N/A } catch (Exception e) {
0N/A throw new RuntimeException(e);
0N/A }
0N/A
0N/A pw.println("public class HelloWorld {");
0N/A pw.println(" public static void main (String argv[]) {");
0N/A pw.println(" System.out.println(\"Hello apt world.\");");
0N/A pw.println(" }");
0N/A pw.println("}");
0N/A }
0N/A }
0N/A
0N/A
0N/A static Collection<String> supportedTypes;
0N/A static {
0N/A String types[] = {"*"};
0N/A supportedTypes = Collections.unmodifiableCollection(Arrays.asList(types));
0N/A }
0N/A
0N/A static Collection<String> supportedOptions;
0N/A static {
0N/A String options[] = {"-Afoo", "-Abar"};
0N/A supportedOptions = Collections.unmodifiableCollection(Arrays.asList(options));
0N/A }
0N/A
0N/A public Collection<String> supportedOptions() {
0N/A return supportedOptions;
0N/A }
0N/A
0N/A public Collection<String> supportedAnnotationTypes() {
0N/A return supportedTypes;
0N/A }
0N/A
0N/A /*
0N/A * Return the same processor independent of what annotations are
0N/A * present, if any.
0N/A */
0N/A public AnnotationProcessor getProcessorFor(Set<AnnotationTypeDeclaration> atds,
0N/A AnnotationProcessorEnvironment ape) {
0N/A
0N/A if (atds.contains(ape.getTypeDeclaration("Marker"))) {
0N/A System.out.println("Returning composite processor.");
0N/A return AnnotationProcessors.getCompositeAnnotationProcessor(new OptionCheck(ape),
0N/A new HelloWorld(ape));
0N/A }
0N/A else {
0N/A System.out.println("Returning single processor.");
0N/A return new OptionCheck(ape);
0N/A }
0N/A }
0N/A}