package nl.beesting.beangenerator.locators.methodlocators;

 import java.lang.reflect.Method;

 import java.net.URL;

 /**

  * Extension of the AbstractClassPathResolver for use in tomcat will use reflection to add the add

  * the web applications classes.

  */

 public class TomcatClassPathResolver extends AbstractClassPathResolver {

     /**

      * {@inheritDoc}

      */

     public void addAddtionalPaths() {

         ClassLoader classLoader = getClass().getClassLoader();

         URL[] urls = null;

         try {

             Method getUrlsMethod = classLoader.getClass().getMethod("getURLs");

             urls = (URL[]) getUrlsMethod.invoke(classLoader);

         } catch (Exception e) {

             // ignore

         }

         if (urls != null) {

             for (URL url : urls) {

                 add(url.getFile());

             }

         }

     }

 }