/*

 * Copyright (c) 2004, 2007, 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.

 *

 * 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.

 */

import com.sun.mirror.apt.*;

import com.sun.mirror.declaration.*;

import com.sun.mirror.type.*;

import com.sun.mirror.util.*;

import java.util.Collection;

import java.util.Set;

import java.util.Arrays;

import static java.util.Collections.*;

/*

 * This class is used to test the ability to store static state across

 * apt rounds.

 */

public class StaticApf implements AnnotationProcessorFactory {

    static int round = -1;

    // Process any set of annotations

    private static final Collection<String> supportedAnnotations

        = unmodifiableCollection(Arrays.asList("*"));

    // No supported options

    private static final Collection<String> supportedOptions = emptySet();

    public Collection<String> supportedAnnotationTypes() {

        return supportedAnnotations;

    }

    public Collection<String> supportedOptions() {

        return supportedOptions;

    }

    public AnnotationProcessor getProcessorFor(Set<AnnotationTypeDeclaration> atds,

                                               AnnotationProcessorEnvironment env) {

        return new StaticAp(env);

    }

    private static class StaticAp implements AnnotationProcessor {

        private final AnnotationProcessorEnvironment env;

        StaticAp(AnnotationProcessorEnvironment env) {

            this.env = env;

        }

        public void process() {

            int size = env.getSpecifiedTypeDeclarations().size();

            try {

                round++;

                switch (size) {

                case 0:

                    if (round == 0) {

                        env.getFiler().createSourceFile("Round1").print("class Round1 {}");

                    } else

                        throw new RuntimeException("Got " + size + " decl's in round " + round);

                    break;

                case 1:

                    if (round == 1) {

                        env.getFiler().createSourceFile("AhOne").print("class AhOne {}");

                        env.getFiler().createSourceFile("AndAhTwo").print("class AndAhTwo {}");

                        env.getFiler().createClassFile("Foo");

                    } else

                        throw new RuntimeException("Got " + size + " decl's in round " + round);

                    break;

                case 2:

                    if (round != 2) {

                        throw new RuntimeException("Got " + size + " decl's in round " + round);

                    }

                    break;

                }

            } catch (java.io.IOException ioe) {

                    throw new RuntimeException();

                }

            }

    }

}