package org.apache.maven.shared.dependency.analyzer;
   
   /*
    * 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.codehaus.plexus.util.DirectoryScanner;
  
  import java.io.File;
  import java.io.FileInputStream;
  import java.io.IOException;
  import java.io.InputStream;
  import java.net.URI;
  import java.net.URISyntaxException;
  import java.net.URL;
  import java.util.jar.JarEntry;
  import java.util.jar.JarInputStream;
  
  
  /**
   * Utility to visit classes in a library given either as a jar file or an exploded directory.
   * 
   * @author <a href="mailto:markhobson@gmail.com">Mark Hobson</a>
   * @version $Id: ClassFileVisitorUtils.java 1589261 2014-04-22 19:34:38Z mfriedenhagen $
   */
  public final class ClassFileVisitorUtils
  {
      // constants --------------------------------------------------------------
  
      private static final String[] CLASS_INCLUDES = { "**/*.class" };
  
      // constructors -----------------------------------------------------------
  
      private ClassFileVisitorUtils()
      {
          // private constructor for utility class
      }
  
      // public methods ---------------------------------------------------------
  
      public static void accept( URL url, ClassFileVisitor visitor )
          throws IOException
      {
          if ( url.getPath().endsWith( ".jar" ) )
          {
              acceptJar( url, visitor );
          }
          else if ( url.getProtocol().equalsIgnoreCase( "file" ) )
          {
              try
              {
                  File file = new File( new URI( url.toString() ) );
  
                  if ( file.isDirectory() )
                  {
                      acceptDirectory( file, visitor );
                  }
                  else if ( file.exists() )
                  {
                      throw new IllegalArgumentException( "Cannot accept visitor on URL: " + url );
                  }
              }
              catch ( URISyntaxException exception )
              {
                  IllegalArgumentException e = new IllegalArgumentException( "Cannot accept visitor on URL: " + url );
                  e.initCause( exception );
                  throw e;
              }
          }
          else
          {
              throw new IllegalArgumentException( "Cannot accept visitor on URL: " + url );
          }
      }
  
      // private methods --------------------------------------------------------
  
      private static void acceptJar( URL url, ClassFileVisitor visitor )
          throws IOException
      {
          JarInputStream in = new JarInputStream( url.openStream() );
          try
          {
              JarEntry entry = null;
 
             while ( ( entry = in.getNextJarEntry() ) != null )
             {
                 String name = entry.getName();
 
                 if ( name.endsWith( ".class" ) )
                 {
                     visitClass( name, in, visitor );
                 }
             }
         }
         finally
         {
             in.close();
         }
     }
 
     private static void acceptDirectory( File directory, ClassFileVisitor visitor )
         throws IOException
     {
         if ( !directory.isDirectory() )
         {
             throw new IllegalArgumentException( "File is not a directory" );
         }
 
         DirectoryScanner scanner = new DirectoryScanner();
 
         scanner.setBasedir( directory );
         scanner.setIncludes( CLASS_INCLUDES );
 
         scanner.scan();
 
         String[] paths = scanner.getIncludedFiles();
 
         for ( String path : paths )
         {
             path = path.replace( File.separatorChar, '/' );
 
             File file = new File( directory, path );
             FileInputStream in = new FileInputStream( file );
 
             try
             {
                 visitClass( path, in, visitor );
             }
             finally
             {
                 in.close();
             }
         }
     }
 
     private static void visitClass( String path, InputStream in, ClassFileVisitor visitor )
     {
         if ( !path.endsWith( ".class" ) )
         {
             throw new IllegalArgumentException( "Path is not a class" );
         }
 
         String className = path.substring( 0, path.length() - 6 );
 
         className = className.replace( '/', '.' );
 
         visitor.visitClass( className, in );
     }
 }