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