package org.apache.maven.plugins.jlink;
   
   /*
    * 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.IOException;
  import java.lang.reflect.Method;
  import java.util.Collection;
  import java.util.List;
  import java.util.Map;
  import java.util.Properties;
  
  import org.apache.commons.lang3.SystemUtils;
  import org.apache.maven.execution.MavenSession;
  import org.apache.maven.plugin.AbstractMojo;
  import org.apache.maven.plugin.MojoExecutionException;
  import org.apache.maven.plugins.annotations.Component;
  import org.apache.maven.plugins.annotations.Parameter;
  import org.apache.maven.project.MavenProject;
  import org.apache.maven.toolchain.Toolchain;
  import org.apache.maven.toolchain.ToolchainManager;
  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;
  
  /**
   * @author Karl Heinz Marbaise <a href="mailto:khmarbaise@apache.org">khmarbaise@apache.org</a>
   */
  public abstract class AbstractJLinkMojo
      extends AbstractMojo
  {
      /**
       * <p>
       * Specify the requirements for this jdk toolchain. This overrules the toolchain selected by the
       * maven-toolchain-plugin.
       * </p>
       * <strong>note:</strong> requires at least Maven 3.3.1
       */
      @Parameter
      private Map<String, String> jdkToolchain;
  
      @Parameter( defaultValue = "${project}", readonly = true, required = true )
      private MavenProject project;
  
      @Parameter( defaultValue = "${session}", readonly = true, required = true )
      private MavenSession session;
  
      @Component
      private ToolchainManager toolchainManager;
  
      protected String getJLinkExecutable()
          throws IOException
      {
          Toolchain tc = getToolchain();
  
          String jLinkExecutable = null;
          if ( tc != null )
          {
              jLinkExecutable = tc.findTool( "jlink" );
          }
  
          // TODO: Check if there exist a more elegant way?
          String jLinkCommand = "jlink" + ( SystemUtils.IS_OS_WINDOWS ? ".exe" : "" );
  
          File jLinkExe;
  
          if ( StringUtils.isNotEmpty( jLinkExecutable ) )
          {
              jLinkExe = new File( jLinkExecutable );
  
              if ( jLinkExe.isDirectory() )
              {
                  jLinkExe = new File( jLinkExe, jLinkCommand );
              }
  
              if ( SystemUtils.IS_OS_WINDOWS && jLinkExe.getName().indexOf( '.' ) < 0 )
              {
                  jLinkExe = new File( jLinkExe.getPath() + ".exe" );
              }
  
              if ( !jLinkExe.isFile() )
             {
                 throw new IOException( "The jlink executable '" + jLinkExe + "' doesn't exist or is not a file." );
             }
             return jLinkExe.getAbsolutePath();
         }
 
         // ----------------------------------------------------------------------
         // Try to find jlink from System.getProperty( "java.home" )
         // By default, System.getProperty( "java.home" ) = JRE_HOME and JRE_HOME
         // should be in the JDK_HOME
         // ----------------------------------------------------------------------
         jLinkExe = new File( SystemUtils.getJavaHome() + File.separator + ".." + File.separator + "bin", jLinkCommand );
 
         // ----------------------------------------------------------------------
         // Try to find javadocExe from JAVA_HOME environment variable
         // ----------------------------------------------------------------------
         if ( !jLinkExe.exists() || !jLinkExe.isFile() )
         {
             Properties env = CommandLineUtils.getSystemEnvVars();
             String javaHome = env.getProperty( "JAVA_HOME" );
             if ( StringUtils.isEmpty( javaHome ) )
             {
                 throw new IOException( "The environment variable JAVA_HOME is not correctly set." );
             }
             if ( !new File( javaHome ).getCanonicalFile().exists()
                 || new File( javaHome ).getCanonicalFile().isFile() )
             {
                 throw new IOException( "The environment variable JAVA_HOME=" + javaHome
                     + " doesn't exist or is not a valid directory." );
             }
 
             jLinkExe = new File( javaHome + File.separator + "bin", jLinkCommand );
         }
 
         if ( !jLinkExe.getCanonicalFile().exists() || !jLinkExe.getCanonicalFile().isFile() )
         {
             throw new IOException( "The jlink executable '" + jLinkExe
                 + "' doesn't exist or is not a file. Verify the JAVA_HOME environment variable." );
         }
 
         return jLinkExe.getAbsolutePath();
     }
 
     protected void executeCommand( Commandline cmd, File outputDirectory )
         throws MojoExecutionException
     {
         if ( getLog().isDebugEnabled() )
         {
             // no quoted arguments ???
             getLog().debug( CommandLineUtils.toString( cmd.getCommandline() ).replaceAll( "'", "" ) );
         }
 
         CommandLineUtils.StringStreamConsumer err = new CommandLineUtils.StringStreamConsumer();
         CommandLineUtils.StringStreamConsumer out = new CommandLineUtils.StringStreamConsumer();
         try
         {
             int exitCode = CommandLineUtils.executeCommandLine( cmd, out, err );
 
             String output = ( StringUtils.isEmpty( out.getOutput() ) ? null : '\n' + out.getOutput().trim() );
 
             if ( exitCode != 0 )
             {
 
                 if ( StringUtils.isNotEmpty( output ) )
                 {
                     // Reconsider to use WARN / ERROR ?
                    //  getLog().error( output );
                     for ( String outputLine : output.split( "\n" ) )
                     {
                         getLog().error( outputLine );
                     }
                 }
 
                 StringBuilder msg = new StringBuilder( "\nExit code: " );
                 msg.append( exitCode );
                 if ( StringUtils.isNotEmpty( err.getOutput() ) )
                 {
                     msg.append( " - " ).append( err.getOutput() );
                 }
                 msg.append( '\n' );
                 msg.append( "Command line was: " ).append( cmd ).append( '\n' ).append( '\n' );
 
                 throw new MojoExecutionException( msg.toString() );
             }
 
             if ( StringUtils.isNotEmpty( output ) )
             {
                 //getLog().info( output );
                 for ( String outputLine : output.split( "\n" ) )
                 {
                     getLog().info( outputLine );
                 }
             }
         }
         catch ( CommandLineException e )
         {
             throw new MojoExecutionException( "Unable to execute jlink command: " + e.getMessage(), e );
         }
 
     }
 
     protected Toolchain getToolchain()
     {
         Toolchain tc = null;
 
         if ( jdkToolchain != null )
         {
             // Maven 3.3.1 has plugin execution scoped Toolchain Support
             try
             {
                 Method getToolchainsMethod = toolchainManager.getClass().getMethod( "getToolchains", MavenSession.class,
                                                                                     String.class, Map.class );
 
                 @SuppressWarnings( "unchecked" )
                 List<Toolchain> tcs =
                     (List<Toolchain>) getToolchainsMethod.invoke( toolchainManager, session, "jdk", jdkToolchain );
 
                 if ( tcs != null && tcs.size() > 0 )
                 {
                     tc = tcs.get( 0 );
                 }
             }
             catch ( ReflectiveOperationException e )
             {
                 // ignore
             }
             catch ( SecurityException e )
             {
                 // ignore
             }
             catch ( IllegalArgumentException e )
             {
                 // ignore
             }
         }
 
         if ( tc == null )
         {
             // TODO: Check if we should make the type configurable?
             tc = toolchainManager.getToolchainFromBuildContext( "jdk", session );
         }
 
         return tc;
     }
 
     protected MavenProject getProject()
     {
         return project;
     }
 
     protected MavenSession getSession()
     {
         return session;
     }
 
     /**
      * Returns the archive file to generate, based on an optional classifier.
      *
      * @param basedir the output directory
      * @param finalName the name of the ear file
      * @param classifier an optional classifier
      * @param archiveExt The extension of the file.
      * @return the file to generate
      */
     protected File getArchiveFile( File basedir, String finalName, String classifier, String archiveExt )
     {
         if ( basedir == null )
         {
             throw new IllegalArgumentException( "basedir is not allowed to be null" );
         }
         if ( finalName == null )
         {
             throw new IllegalArgumentException( "finalName is not allowed to be null" );
         }
         if ( archiveExt == null )
         {
             throw new IllegalArgumentException( "archiveExt is not allowed to be null" );
         }
 
         if ( finalName.isEmpty() )
         {
             throw new IllegalArgumentException( "finalName is not allowed to be empty." );
         }
         if ( archiveExt.isEmpty() )
         {
             throw new IllegalArgumentException( "archiveExt is not allowed to be empty." );
         }
 
         StringBuilder fileName = new StringBuilder( finalName );
 
         if ( hasClassifier( classifier ) )
         {
             fileName.append( "-" ).append( classifier );
         }
 
         fileName.append( '.' );
         fileName.append( archiveExt );
 
         return new File( basedir, fileName.toString() );
     }
 
     protected boolean hasClassifier( String classifier )
     {
         boolean result = false;
         if ( classifier != null && classifier.trim().length() > 0 )
         {
             result = true;
         }
 
         return result;
     }
 
     /**
      * This will convert a module path separated by either {@code :} or {@code ;} into a string which uses the platform
      * depend path separator uniformly.
      * 
      * @param pluginModulePath The module path.
      * @return The platform separated module path.
      */
     protected StringBuilder convertSeparatedModulePathToPlatformSeparatedModulePath( String pluginModulePath )
     {
         StringBuilder sb = new StringBuilder();
         // Split the module path by either ":" or ";" linux/windows path separator and
         // convert uniformly to the platform used separator.
         String[] splitModule = pluginModulePath.split( "[;:]" );
         for ( String module : splitModule )
         {
             if ( sb.length() > 0 )
             {
                 sb.append( File.pathSeparatorChar );
             }
             sb.append( module );
         }
         return sb;
     }
 
     /**
      * Convert a list into a string which is separated by platform depend path separator.
      * 
      * @param modulePaths The list of elements.
      * @return The string which contains the elements separated by {@link File#pathSeparatorChar}.
      */
     protected String getPlatformDependSeparateList( Collection<String> modulePaths )
     {
         StringBuilder sb = new StringBuilder();
         for ( String module : modulePaths )
         {
             if ( sb.length() > 0 )
             {
                 sb.append( File.pathSeparatorChar );
             }
             sb.append( module );
         }
         return sb.toString();
     }
 
     /**
      * Convert a list into a 
      * @param modules The list of modules.
      * @return The string with the module list which is separated by {@code ,}.
      */
     protected String getCommaSeparatedList( Collection<String> modules )
     {
         StringBuilder sb = new StringBuilder();
         for ( String module : modules )
         {
             if ( sb.length() > 0 )
             {
                 sb.append( ',' );
             }
             sb.append( module );
         }
         return sb.toString();
     }
 
 }