package com.btc.tutorial.todo;

import java.lang.annotation.*;

@Retention(RetentionPolicy.CLASS)
@Target ({ElementType.ANNOTATION_TYPE})
public @interface Développeur {
    String prénom();
    String nomDeFamille();
}

package com.btc.tutorial.todo;

public @interface Todo {
    public enum Severité{ CRITIQUE, IMPORTANT, TRIVIALE, DOCUMENTATION };
    Severité severité() default Severité.IMPORTANT;
    String quoi();
    Développeur qui();
}

package com.btc.tutorial.todo;

@Todo (
        severité=Todo.Severité.CRITIQUE,
        quoi="Ajouter la liste des employés",
        qui = @Développeur (prénom = "Thibault", nomDeFamille="Cuvillier")
)
public class Entreprise {
    private String nom;

    public String getNom() {
        return nom;
    }

    public void setNom(String nom) {
        this.nom = nom;
    }

    @Todo (
            quoi="Corriger ce code incompréhensible!",
            qui = @Développeur (prénom = "Thibault", nomDeFamille="Cuvillier")
    )
    public void horreur() {
        int i = 0;
        i = ++i+i++;
    }
}

package com.btc.tutorial.todo;

import com.sun.mirror.declaration.*;
import com.sun.mirror.util.*;
import com.sun.mirror.apt.*;
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.*;
import static com.sun.mirror.util.DeclarationVisitors.*;

import java.util.*;


// Cette classe permet de créer des processeurs d'annotation.
public class TodoFabrique implements AnnotationProcessorFactory {

    // Taiter les annotations du package com.btc.tutorial.todo
    private static final Collection supportedAnnotations
            = unmodifiableCollection(Arrays.asList("com.btc.tutorial.todo.*"));

    // Options (voir la doc pour + d'info)
    private static final Collection supportedOptions = emptySet();

    public Collection supportedAnnotationTypes() {
        return supportedAnnotations;
    }

    public Collection supportedOptions() {
        return supportedOptions;
    }

    // Récupérer le processeur d'annotation
    public AnnotationProcessor getProcessorFor(
           Set atds,
           AnnotationProcessorEnvironment env) {

        return new TodoProcesseur(env);
    }

    // La classe traitant les annotations
    private static class TodoProcesseur implements AnnotationProcessor {
        private final AnnotationProcessorEnvironment env;

        TodoProcesseur(AnnotationProcessorEnvironment env) {
            this.env = env;
        }

        // Traiter les annotations d'une classe
        public void process() {
            for (TypeDeclaration typeDecl : env.getSpecifiedTypeDeclarations()) {
                System.out.println("process called: accept");

                // Lancer un visiteur
                typeDecl.accept(getDeclarationScanner(new TodoVisiteur(),NO_OP));
            }

        }

        // Le visiteur qui se promène sur le code Java
        private static class TodoVisiteur extends SimpleDeclarationVisitor {

            // Déclenchée lors de la déclaration d'une classe ayant une annotation Todo
            public void visitClassDeclaration(ClassDeclaration d) {
                Todo todo = d.getAnnotation(Todo.class);
                if( todo == null ) return;

                System.out.println(todo.qui().prénom() + " " +
                        todo.qui().nomDeFamille() + " doit " +
                        todo.quoi() + " dans la classe " +
                        d.getQualifiedName());
            }

            // Déclenchée lors de la déclaration d'une méthode ayant une annotation Todo
            public void visitMethodDeclaration(MethodDeclaration methodDeclaration) {
                Todo todo = methodDeclaration.getAnnotation(Todo.class);
                if( todo == null ) return;

                System.out.println(todo.qui().prénom() + " " +
                        todo.qui().nomDeFamille() + " doit " +
                        todo.quoi() + " dans la méthode " +
                        methodDeclaration.getSimpleName());
            }
        }
    }
}