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 java.util.*;
duke@1: import com.sun.mirror.apt.*;
duke@1: import com.sun.mirror.declaration.*;
duke@1: import com.sun.mirror.util.*;
duke@1: 
duke@1: 
duke@1: /**
duke@1:  * A factory for generating the TestProcessor annotation processor, which
duke@1:  * processes the @Test annotation.
duke@1:  *
duke@1:  * @author Scott Seligman
duke@1:  */
duke@1: public class TestProcessorFactory implements AnnotationProcessorFactory {
duke@1: 
duke@1:     public Collection<String> supportedOptions() {
duke@1:         return new ArrayList<String>();
duke@1:     }
duke@1: 
duke@1:     public Collection<String> supportedAnnotationTypes() {
duke@1:         ArrayList<String> res = new ArrayList<String>();
duke@1:         res.add("Test");
duke@1:         res.add("Ignore");
duke@1:         return res;
duke@1:     }
duke@1: 
duke@1:     public AnnotationProcessor getProcessorFor(
duke@1:                                         Set<AnnotationTypeDeclaration> as,
duke@1:                                         AnnotationProcessorEnvironment env) {
duke@1:         // The tester that's running.
duke@1:         Tester tester = Tester.activeTester;
duke@1: 
duke@1:         try {
duke@1:             // Find the tester's class declaration.
duke@1:             ClassDeclaration testerDecl = null;
duke@1:             for (TypeDeclaration decl : env.getSpecifiedTypeDeclarations()) {
duke@1:                 if (decl.getQualifiedName().equals(
duke@1:                                                tester.getClass().getName())) {
duke@1:                     testerDecl = (ClassDeclaration) decl;
duke@1:                     break;
duke@1:                 }
duke@1:             }
duke@1: 
duke@1:             // Give the tester access to its own declaration and to the env.
duke@1:             tester.thisClassDecl = testerDecl;
duke@1:             tester.env = env;
duke@1: 
duke@1:             // Initializer the tester.
duke@1:             tester.init();
duke@1: 
duke@1:             return new TestProcessor(env, tester);
duke@1: 
duke@1:         } catch (Exception e) {
duke@1:             throw new Error("Couldn't create test annotation processor", e);
duke@1:         }
duke@1:     }
duke@1: }