package org.apache.maven.shared.jarsigner;

 /*

  * 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.logging.AbstractLogEnabled;

 import org.codehaus.plexus.util.Os;

 import org.codehaus.plexus.util.StringUtils;

 import org.codehaus.plexus.util.cli.CommandLineException;

 import org.codehaus.plexus.util.cli.CommandLineUtils;

 import org.codehaus.plexus.util.cli.Commandline;

 import org.codehaus.plexus.util.cli.StreamConsumer;

 import java.io.File;

 import java.io.IOException;

 import java.io.InputStream;

 import java.util.Map;

 /**

  * Default implementation of component {@link JarSigner}.

  *

  * @author tchemit <chemit@codelutin.com>

  * @version $Id: DefaultJarSigner.java 1195937 2011-11-01 11:38:37Z olamy $

  * @plexus.component role="org.apache.maven.shared.jarsigner.JarSigner" role-hint="default"

  * @since 1.0

  */

 public class DefaultJarSigner

     extends AbstractLogEnabled

     implements JarSigner

 {

     /**

      * The location of the jarSigner executable file.

      */

     protected String jarSignerFile;

     public JarSignerResult execute( JarSignerRequest request )

         throws JarSignerException

     {

         if ( jarSignerFile == null )

         {

             // find the jar singer to use

             try

             {

                 jarSignerFile = findJarSignerExecutable();

             }

             catch ( IOException e )

             {

                 throw new JarSignerException( "Error finding jar signer executable. Reason: " + e.getMessage(), e );

             }

         }

         // creates the command line

         Commandline cli = createCommandLine( request );

         // execute it

         return executeCommandLine( cli, request );

     }

     protected Commandline createCommandLine( JarSignerRequest request )

         throws JarSignerException

     {

         JarSignerCommandLineBuilder cliBuilder = new JarSignerCommandLineBuilder();

         cliBuilder.setLogger( getLogger() );

         cliBuilder.setJarSignerFile( jarSignerFile );

         try

         {

             return cliBuilder.build( request );

         }

         catch ( CommandLineConfigurationException e )

         {

             throw new JarSignerException( "Error configuring command-line. Reason: " + e.getMessage(), e );

         }

     }

     protected JarSignerResult executeCommandLine( Commandline cli, JarSignerRequest request )

     {

         if ( getLogger().isDebugEnabled() )

         {

             getLogger().debug( "Executing: " + cli );

         }

         final boolean verbose = request.isVerbose();

         InputStream systemIn = new InputStream()

         {

             public int read()

             {

                 return -1;

             }

         };

         StreamConsumer systemOut = request.getSystemOutStreamConsumer();

         if ( systemOut == null )

         {

             systemOut = new StreamConsumer()

             {

                 public void consumeLine( final String line )

                 {

                     if ( verbose )

                     {

                         getLogger().info( line );

                     }

                     else

                     {

                         getLogger().debug( line );

                     }

                 }

             };

         }

         StreamConsumer systemErr = request.getSystemErrorStreamConsumer();

         if ( systemErr == null )

         {

             systemErr = new StreamConsumer()

             {

                 public void consumeLine( final String line )

                 {

                     getLogger().warn( line );

                 }

             };

         }

         DefaultJarSignerResult result = new DefaultJarSignerResult();

         result.setCommandline( cli );

         if ( verbose )

         {

             getLogger().info( cli.toString() );

         }

         else

         {

             getLogger().debug( cli.toString() );

         }

         try

         {

             int resultCode = CommandLineUtils.executeCommandLine( cli, systemIn, systemOut, systemErr );

             result.setExitCode( resultCode );

         }

         catch ( CommandLineException e )

         {

             result.setExecutionException( e );

         }

         return result;

     }

     protected String findJarSignerExecutable()

         throws IOException

     {

         String command = "jarsigner" + ( Os.isFamily( Os.FAMILY_WINDOWS ) ? ".exe" : "" );

         String executable =

             findExecutable( command, System.getProperty( "java.home" ), new String[]{ "../bin", "bin", "../sh" } );

         if ( executable == null )

         {

             Map<String, String> env = System.getenv();

             String[] variables = { "JDK_HOME", "JAVA_HOME" };

             for ( int i = 0; i < variables.length && executable == null; i++ )

             {

                 executable = findExecutable( command, env.get( variables[i] ), new String[]{ "bin", "sh" } );

             }

         }

         if ( executable == null )

         {

             executable = command;

         }

         return executable;

     }

     /**

      * Finds the specified command in any of the given sub directories of the specified JDK/JRE home directory.

      *

      * @param command The command to find, must not be <code>null</code>.

      * @param homeDir The home directory to search in, may be <code>null</code>.

      * @param subDirs The sub directories of the home directory to search in, must not be <code>null</code>.

      * @return The (absolute) path to the command if found, <code>null</code> otherwise.

      */

     protected String findExecutable( String command, String homeDir, String[] subDirs )

     {

         if ( StringUtils.isNotEmpty( homeDir ) )

         {

             for ( int i = 0; i < subDirs.length; i++ )

             {

                 File file = new File( new File( homeDir, subDirs[i] ), command );

                 if ( file.isFile() )

                 {

                     return file.getAbsolutePath();

                 }

             }

         }

         return null;

     }

 }