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 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 )

                         {

                         }

                     }

                 }

             }

             return new File( base, path );

         }

         else

         {

             return new File( resource.toURI().normalize() );

         }

     }

 }