package nl.beesting.beangenerator.locators;

 import java.util.List;

 import java.lang.reflect.Method;

 import nl.beesting.beangenerator.config.BeanGeneratorConfig;

 import nl.beesting.beangenerator.config.BeanGeneratorConfigParser;

 import nl.beesting.beangenerator.config.ClassDescriptor;

 import nl.beesting.beangenerator.config.InstanceGeneratorDescriptor;

 import org.aopalliance.intercept.MethodInvocation;

 import org.apache.commons.lang.StringUtils;

 public class ConfigurationClassLocator implements ClassLocator {

    private BeanGeneratorConfig configuration;

    public void setConfigParser(BeanGeneratorConfigParser configParser) {

       this.configuration = configParser.getConfiguration();

    }

    public Class<?> findClass(MethodInvocation invocation) {

       Class<?> result = null;

       Class<?> returnType = invocation.getMethod().getReturnType();

       List<ClassDescriptor> definitions = configuration.getClassDefinitions();

        for (ClassDescriptor classDescriptor : definitions) {

            Class<?> clazz = classDescriptor.getClazz();

            if (clazz.equals(returnType)) {

                return clazz;

            }

        }

       InstanceGeneratorDescriptor defaultGenerator = configuration.getDefaultGenerator(returnType);

       if (defaultGenerator != null) {

           result = returnType;

       }

       ClassDescriptor classDescriptor = null;

       String name = null;

       if (result == null) {

          Method method = invocation.getMethod();

          name = getMethodName(method);

          classDescriptor = configuration.getClassDefinitionByMethod(name);

          if (classDescriptor == null) {

                    name = method.getName(); 

             classDescriptor = configuration.getClassDefinitionByMethod(name);

          }

       }

       if (classDescriptor != null) {

          result = classDescriptor.getFieldDefinitions().get(name).getClazz();

       }

       return result;

    }

    private String getMethodName(Method method) {

       String name = method.getName();

       if (name.startsWith("get") || name.startsWith("set")) {

          name = StringUtils.uncapitalize(name.substring(3));

       }

       return name;

    }

 }