content: /*****************************************************************
   *   Licensed to the Apache Software Foundation (ASF) under one
   *  or more contributor license agreements.  See the NOTICE file
   *  distributed with this work for additional information
   *  regarding copyright ownership.  The ASF licenses this file
   *  to you under the Apache License, Version 2.0 (the
   *  "License"); you may not use this file except in compliance
   *  with the License.  You may obtain a copy of the License at
   *
   *    https://www.apache.org/licenses/LICENSE-2.0
   *
   *  Unless required by applicable law or agreed to in writing,
   *  software distributed under the License is distributed on an
   *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   *  KIND, either express or implied.  See the License for the
   *  specific language governing permissions and limitations
   *  under the License.
   ****************************************************************/
  package org.apache.cayenne.configuration.osgi;
  
  import java.util.Map;
  
  import org.apache.cayenne.di.ClassLoaderManager;
  import org.apache.cayenne.di.Injector;
  
  /**
   * @since 4.0
   */
  public class OsgiClassLoaderManager implements ClassLoaderManager {
  
      private static final String CAYENNE_PACKAGE = "org/apache/cayenne";
      private static final String CAYENNE_DI_PACKAGE_SUFFIX = "/di";
  
      private ClassLoader applicationClassLoader;
      private ClassLoader cayenneRuntimeClassLoader;
      private ClassLoader cayenneDiClassLoader;
      private Map<String, ClassLoader> perResourceClassLoaders;
  
      public OsgiClassLoaderManager(ClassLoader applicationClassLoader, Map<String, ClassLoader> perResourceClassLoaders) {
          this.applicationClassLoader = applicationClassLoader;
          this.cayenneDiClassLoader = Injector.class.getClassLoader();
          this.cayenneRuntimeClassLoader = OsgiClassLoaderManager.class.getClassLoader();
          this.perResourceClassLoaders = perResourceClassLoaders;
      }
  
      @Override
      public ClassLoader getClassLoader(String resourceName) {
          if (resourceName == null || resourceName.length() < 2) {
              return resourceClassLoader(resourceName);
          }
  
          String normalizedName = resourceName.charAt(0) == '/' ? resourceName.substring(1) : resourceName;
          if (normalizedName.startsWith(CAYENNE_PACKAGE)) {
  
              return (normalizedName.substring(CAYENNE_PACKAGE.length()).startsWith(CAYENNE_DI_PACKAGE_SUFFIX))
                      ? cayenneDiClassLoader()
                      : cayenneRuntimeClassLoader();
          }
  
          return resourceClassLoader(resourceName);
      }
  
      protected ClassLoader resourceClassLoader(String resourceName) {
          ClassLoader cl = perResourceClassLoaders.get(resourceName);
          return cl != null ? cl : applicationClassLoader;
      }
  
      protected ClassLoader cayenneDiClassLoader() {
          return cayenneDiClassLoader;
      }
  
      protected ClassLoader cayenneRuntimeClassLoader() {
          return cayenneRuntimeClassLoader;
      }
  }
  