package nl.beesting.beangenerator.locators;

 import java.util.regex.Matcher;

 import java.util.regex.Pattern;

 import nl.beesting.beangenerator.locators.methodlocators.AbstractClassPathResolver;

 import org.aopalliance.intercept.MethodInvocation;

 import org.apache.commons.lang.StringUtils;

 /**

  * This class locator tries to find a classes by method name. It will parse the name of a given

  * method and try to find a class that that matches it. For example is the method is named

  * findCatsByIds it will try to load a class named Cat. Finding the classes is the work of the

  * {@link AbstractClassPathResolver}

  */

 public class MethodClassLocator implements ClassLocator {

     private AbstractClassPathResolver classResolverDelegate;

     public Class<?> findClass(MethodInvocation invocation) {

         Class<?> result = null;

         String methodName = invocation.getMethod().getName();

         Pattern pattern = Pattern.compile("[A-Z][a-z]*");

         Matcher matcher = pattern.matcher(methodName);

         while (matcher.find() && result == null) {

             String possibleClassName = matcher.group();

             for (int i = 0; i < 3 && result == null; i++) {

                 String name = possibleClassName.substring(0, possibleClassName.length() - i);

                 if (StringUtils.isNotBlank(name)) {

                     result = classResolverDelegate.findClass(name);

                 }

             }

         }

         return result;

     }

     /**

      * @param classResolverDelegate the classResolverDelegate to set

      */

     public void setClassResolverDelegate(AbstractClassPathResolver classResolverDelegate) {

         this.classResolverDelegate = classResolverDelegate;

     }

 }