package com.atlassian.plugin.osgi.util;
   
   import com.atlassian.plugin.util.resource.AlternativeResourceLoader;
   import com.atlassian.plugin.util.resource.NoOpAlternativeResourceLoader;
   import org.apache.commons.collections.iterators.IteratorEnumeration;
   import org.osgi.framework.Bundle;
   
   import java.io.IOException;
   import java.net.URL;
  import java.util.Arrays;
  import java.util.Collections;
  import java.util.Enumeration;
  
  import static com.atlassian.fugue.Option.option;
  import static com.google.common.base.Preconditions.checkNotNull;
  
  /**
   * Utility methods for accessing a bundle as if it was a classloader.
   *
   * @since 2.3.0
   */
  public class BundleClassLoaderAccessor {
      /**
       * Creates a classloader that delegates to the bundle
       *
       * @param bundle                    The bundle to delegate to
       * @param alternativeResourceLoader An alternative resource loader to bypass bundle, can be null
       * @return A new classloader instance
       */
      public static ClassLoader getClassLoader(final Bundle bundle, final AlternativeResourceLoader alternativeResourceLoader) {
          return new BundleClassLoader(bundle, alternativeResourceLoader);
      }
  
      /**
       * Loads a class from the bundle
       *
       * @param bundle The bundle
       * @param name   The name of the class to load
       * @param <T>    The type of the class
       * @return The class instance
       * @throws ClassNotFoundException If the class cannot be found in the bundle
       */
      public static <T> Class<T> loadClass(final Bundle bundle, final String name) throws ClassNotFoundException {
          @SuppressWarnings({"unchecked", "UnnecessaryLocalVariable"})
          final Class<T> loadedClass = (Class<T>) checkNotNull(bundle, "The bundle is required").loadClass(name);
          return loadedClass;
      }
  
      ///CLOVER:OFF
  
      /**
       * Fake classloader that delegates to a bundle
       */
      private static class BundleClassLoader extends ClassLoader {
          private final Bundle bundle;
          private final AlternativeResourceLoader altResourceLoader;
  
          public BundleClassLoader(final Bundle bundle, AlternativeResourceLoader altResourceLoader) {
              super(null);
              this.bundle = checkNotNull(bundle, "The bundle must not be null");
              this.altResourceLoader = option(altResourceLoader).getOrElse(new NoOpAlternativeResourceLoader());
          }
  
          @Override
          public Class<?> findClass(final String name) throws ClassNotFoundException {
              return bundle.loadClass(name);
          }
  
          @SuppressWarnings("unchecked")
          @Override
          public Enumeration<URL> findResources(final String name) throws IOException {
              Enumeration<URL> e = bundle.getResources(name);
  
              if (e == null) {
                  e = new IteratorEnumeration(Collections.emptyList().iterator());
              } else {
                  // For some reason, getResources() sometimes returns nothing, yet getResource() will return one.  This code
                  // handles that strange case
                  if (!e.hasMoreElements()) {
                      final URL resource = findResource(name);
                      if (resource != null) {
                          e = new IteratorEnumeration(Arrays.asList(resource).iterator());
                      }
                  }
              }
              return e;
          }
  
          @Override
          public URL findResource(final String name) {
              URL url = altResourceLoader.getResource(name);
              if (url == null) {
                  url = bundle.getResource(name);
              }
              return url;
          }
  
          @Override
          public String toString() {
             final String sym = bundle.getSymbolicName();
             return "BundleClassLoader@" + Integer.toHexString(System.identityHashCode(this)) + "[bundle=" +
                     ((sym != null) ? sym : "") + " [" + bundle.getBundleId() + "]]";
         }
     }
 
 }