/*
 * Copyright 2004-2007 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.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.Map;
import java.util.Arrays;
import java.util.Collections;

public class OptionChecker implements AnnotationProcessorFactory {
    static class OptionCheck implements AnnotationProcessor {
	AnnotationProcessorEnvironment ape;
	OptionCheck(AnnotationProcessorEnvironment ape) {
	    this.ape = ape;
	}
    
	public void process() {
	    Map<String, String> options = ape.getOptions();
	    if (options.containsKey("-Afoo") && 
		options.containsKey("-Abar") && 
		options.containsKey("-classpath") ) {
		System.out.println("Expected options found.");
		return;  // All is well
	    } else {
		System.err.println("Unexpected options values: " + options);
		throw new RuntimeException();
	    }
	}
    }

    static class HelloWorld implements AnnotationProcessor {
	AnnotationProcessorEnvironment ape;
	HelloWorld(AnnotationProcessorEnvironment ape) {
	    this.ape = ape;
	}
    
	public void process() {
	    java.io.PrintWriter pw;
	    try {
		pw = ape.getFiler().createSourceFile("HelloWorld");
	    } catch (Exception e) {
		throw new RuntimeException(e);
	    }

	    pw.println("public class HelloWorld {");
	    pw.println("  public static void main (String argv[]) {");
	    pw.println("    System.out.println(\"Hello apt world.\");");
	    pw.println("  }");
	    pw.println("}");
	}
    }
	

    static Collection<String> supportedTypes;
    static {
	String types[] = {"*"};
	supportedTypes = Collections.unmodifiableCollection(Arrays.asList(types));
    }

    static Collection<String> supportedOptions;
    static {
	String options[] = {"-Afoo", "-Abar"};
	supportedOptions = Collections.unmodifiableCollection(Arrays.asList(options));
    }

    public Collection<String> supportedOptions() {
	return supportedOptions;
    }

    public Collection<String> supportedAnnotationTypes() {
	return supportedTypes;
    }

    /*
     * Return the same processor independent of what annotations are
     * present, if any.
     */
    public AnnotationProcessor getProcessorFor(Set<AnnotationTypeDeclaration> atds,
					       AnnotationProcessorEnvironment ape) {

	if (atds.contains(ape.getTypeDeclaration("Marker"))) {
	    System.out.println("Returning composite processor.");
	    return AnnotationProcessors.getCompositeAnnotationProcessor(new OptionCheck(ape),
									new HelloWorld(ape));
	}
	else { 
	    System.out.println("Returning single processor.");
	    return new OptionCheck(ape);
	}
    }
}