package org.graphwalker.java.annotation;

/*
 * #%L
 * GraphWalker Java
 * %%
 * Copyright (C) 2011 - 2014 GraphWalker
 * %%
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 * #L%
 */

import org.reflections.Reflections;

import java.lang.annotation.Annotation;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.HashSet;
import java.util.Set;

/**
 * @author Nils Olsson
 */
public final class AnnotationUtils {

    private AnnotationUtils() {}

    public static Set<Class<?>> findTests() {
        return findTests("");
    }

    public static Set<Class<?>> findTests(String packageName) {
        Reflections reflections = new Reflections(packageName);
        return reflections.getTypesAnnotatedWith(GraphWalker.class);
    }

    public static Set<Annotation> getAnnotations(final Class<?> clazz, final Class<? extends Annotation> annotation) {
        Set<Annotation> annotations = new HashSet<>();
        for (Class<?> interfaceClass: clazz.getInterfaces()) {
            if (interfaceClass.isAnnotationPresent(annotation)) {
                annotations.add(interfaceClass.getAnnotation(annotation));
            }
        }
        return annotations;
    }

    public static void execute(Class<? extends Annotation> annotation, Object implementation) {
        for (Method method: implementation.getClass().getMethods()) {
            if (method.isAnnotationPresent(annotation)) {
                try {
                    method.invoke(implementation);
                } catch (IllegalAccessException | InvocationTargetException e) {
                    e.printStackTrace();
                }
            }
        }
    }


}

@@ -1975,24 +1975,102 @@
 ashSet%3C%3E();%0A
+        Class%3C?%3E queryClass = clazz;%0A        while (null != queryClass) %7B%0A    
         for 
@@ -2095,21 +2095,26 @@
 eClass: 
-clazz
+queryClass
 .getInte
@@ -2121,24 +2121,28 @@
 rfaces()) %7B%0A
+    
             
@@ -2211,16 +2211,20 @@
         
+  
+  
 annotati
@@ -2278,33 +2278,104 @@
 ));%0A            
-%7D
+    %7D%0A            %7D%0A            queryClass = queryClass.getSuperclass();
 %0A        %7D%0A     

Add support for inheriting a class with @Model definitions