package org.apache.maven.wagon.tck.http.util;
   
   /*
    * 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 java.io.File;
  import java.io.FileOutputStream;
  import java.io.IOException;
  import java.io.InputStream;
  import java.io.OutputStream;
  import java.net.URISyntaxException;
  import java.net.URL;
  import java.util.Enumeration;
  import java.util.HashMap;
  import java.util.Map;
  import java.util.jar.JarEntry;
  import java.util.jar.JarFile;
  
  import org.apache.log4j.Logger;
  import org.codehaus.plexus.util.IOUtil;
  
  /**
   * 
   */
  public final class TestUtil
  {
      private static final Logger LOGGER = Logger.getLogger( TestUtil.class );
  
      private static final Map<String, File> BASES = new HashMap<String, File>();
  
      private TestUtil()
      {
      }
  
      public static File getResource( final String path )
          throws URISyntaxException, IOException
      {
          URL resource = Thread.currentThread().getContextClassLoader().getResource( path );
          if ( resource == null )
          {
              throw new IllegalStateException( "Cannot find classpath resource: " + path );
          }
  
          if ( resource.getProtocol().startsWith( "jar" ) )
          {
              // File f = new File( path );
              // f = File.createTempFile( f.getName() + ".", ".tmp" );
  
              String url = resource.toExternalForm();
              int startIdx = url.lastIndexOf( ':' ) + 1;
              int endIdx = url.indexOf( "!" );
              url = url.substring( startIdx, endIdx );
  
              File base = BASES.get( url );
              if ( base == null )
              {
                  File urlFile = new File( url );
  
                  base = new File( "target/tck-resources/" + urlFile.getName() );
                  base.getParentFile().mkdirs();
  
                  LOGGER.info( "unpacking test resources in jar: " + url );
                  JarFile jf = null;
                  try
                  {
                      jf = new JarFile( urlFile );
  
                      InputStream in = null;
                      OutputStream out = null;
  
                      for ( Enumeration<JarEntry> en = jf.entries(); en.hasMoreElements(); )
                      {
                          JarEntry je = en.nextElement();
                          File target = new File( base, je.getName() ).getAbsoluteFile();
                          if ( je.isDirectory() )
                          {
                              target.mkdirs();
                          }
                          else
                          {
                              target.getParentFile().mkdirs();
  
                              try
                             {
                                 in = jf.getInputStream( je );
                                 out = new FileOutputStream( target );
 
                                 IOUtil.copy( in, out );
                             }
                             finally
                             {
                                 IOUtil.close( in );
                                 IOUtil.close( out );
                             }
                         }
                     }
 
                     BASES.put( url, base );
                 }
                 finally
                 {
                     if ( jf != null )
                     {
                         try
                         {
                             jf.close();
                         }
                         catch ( Exception e )
                         {
                             // ignore
                         }
                     }
                 }
             }
 
             return new File( base, path );
         }
         else
         {
             return new File( resource.toURI().normalize() );
         }
     }
 
 }