package org.apache.maven.surefire.booter;
   
   /*
    * 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
   *
   *     http://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.
   */
  
  import org.apache.maven.surefire.util.NestedRuntimeException;
  
  import java.lang.reflect.InvocationTargetException;
  import java.lang.reflect.Method;
  import java.net.MalformedURLException;
  import java.net.URL;
  import java.util.Iterator;
  import java.util.List;
  
  /**
   * Represents the classpaths for the BooterConfiguration.
   * <p/>
   * 
   * @author Jason van Zyl
   * @author Emmanuel Venisse
   * @author Kristian Rosenvold
   * @version $Id$
   */
  public class ClasspathConfiguration
  {
      private static final String CHILD_DELEGATION = "childDelegation";
  
      private static final String ENABLE_ASSERTIONS = "enableAssertions";
  
      private static final String CLASSPATH = "classPathUrl.";
  
      private static final String SUREFIRE_CLASSPATH = "surefireClassPathUrl.";
  
      private final Classpath classpathUrls;
  
      private final Classpath surefireClasspathUrls;
  
      /**
       * Whether to enable assertions or not (can be affected by the fork arguments, and the ability to do so based on the
       * JVM).
       */
      private final boolean enableAssertions;
  
      // todo: @deprecated because the IsolatedClassLoader is really isolated - no parent.
      private final boolean childDelegation;
  
      public ClasspathConfiguration( boolean enableAssertions, boolean childDelegation )
      {
          this( new Classpath(), new Classpath(), enableAssertions, childDelegation );
      }
  
  
      ClasspathConfiguration( PropertiesWrapper properties )
      {
          this( properties.getClasspath( CLASSPATH ),
                properties.getClasspath( SUREFIRE_CLASSPATH ),
                properties.getBooleanProperty( ENABLE_ASSERTIONS ), properties.getBooleanProperty( CHILD_DELEGATION ) );
      }
  
      public ClasspathConfiguration( Classpath testClasspath, Classpath surefireClassPathUrls, boolean enableAssertions,
                                      boolean childDelegation )
      {
          this.enableAssertions = enableAssertions;
          this.childDelegation = childDelegation;
          this.classpathUrls = testClasspath;
          this.surefireClasspathUrls = surefireClassPathUrls;
      }
  
      public void setForkProperties( PropertiesWrapper properties )
      {
          properties.setClasspath( CLASSPATH, classpathUrls );
          properties.setClasspath( SUREFIRE_CLASSPATH, surefireClasspathUrls );
          properties.setProperty( ENABLE_ASSERTIONS, String.valueOf( enableAssertions ) );
          properties.setProperty( CHILD_DELEGATION, String.valueOf( childDelegation ) );
      }
  
      private static Method assertionStatusMethod;
  
      static
      {
          try
          {
              assertionStatusMethod =
                 ClassLoader.class.getMethod( "setDefaultAssertionStatus", new Class[]{ boolean.class } );
         }
         catch ( NoSuchMethodException e )
         {
             assertionStatusMethod = null;
         }
     }
 
     public ClassLoader createTestClassLoaderConditionallySystem( boolean useSystemClassLoader )
         throws SurefireExecutionException
     {
         return useSystemClassLoader
             ? ClassLoader.getSystemClassLoader()
             : createTestClassLoader( this.childDelegation );
     }
 
     public ClassLoader createTestClassLoader( boolean childDelegation )
         throws SurefireExecutionException
     {
         return createClassLoaderSEE( classpathUrls, null, childDelegation );
     }
 
     public ClassLoader createTestClassLoader()
         throws SurefireExecutionException
     {
         return createClassLoaderSEE( classpathUrls, null, this.childDelegation );
     }
 
     public ClassLoader createSurefireClassLoader( ClassLoader parent )
         throws SurefireExecutionException
     {
         return createClassLoaderSEE( surefireClasspathUrls, parent, false );
     }
 
     private ClassLoader createClassLoaderSEE( Classpath classPathUrls, ClassLoader parent, boolean childDelegation )
         throws SurefireExecutionException
     {
         try
         {
             return createClassLoader( classPathUrls, parent, childDelegation );
         }
         catch ( MalformedURLException e )
         {
             throw new SurefireExecutionException( "When creating classloader", e );
         }
 
     }
 
     private ClassLoader createClassLoader( Classpath classPathUrls, ClassLoader parent, boolean childDelegation )
         throws MalformedURLException
     {
         List urls = classPathUrls.getAsUrlList();
         IsolatedClassLoader classLoader = new IsolatedClassLoader( parent, childDelegation );
         if ( assertionStatusMethod != null )
         {
             try
             {
                 Object[] args = new Object[]{ enableAssertions ? Boolean.TRUE : Boolean.FALSE };
                 if ( parent != null )
                 {
                     assertionStatusMethod.invoke( parent, args );
                 }
                 assertionStatusMethod.invoke( classLoader, args );
             }
             catch ( IllegalAccessException e )
             {
                 throw new NestedRuntimeException( "Unable to access the assertion enablement method", e );
             }
             catch ( InvocationTargetException e )
             {
                 throw new NestedRuntimeException( "Unable to invoke the assertion enablement method", e );
             }
         }
         for ( Iterator iter = urls.iterator(); iter.hasNext(); )
         {
             URL url = (URL) iter.next();
             classLoader.addURL( url );
         }
         return classLoader;
     }
 
     public Classpath getTestClasspath()
     {
         return classpathUrls;
     }
 
     public void addClasspathUrl( String path )
     {
         classpathUrls.addClassPathElementUrl( path );
     }
 
     public void addSurefireClasspathUrl( String path )
     {
         surefireClasspathUrls.addClassPathElementUrl( path );
     }
 }