package org.apache.maven.it;

 /*

  * 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.FileWriter;

 import java.io.IOException;

 import java.io.Writer;

 import java.util.Collections;

 import java.util.Iterator;

 import java.util.Map;

 import org.apache.maven.it.util.cli.CommandLineException;

 import org.apache.maven.it.util.cli.CommandLineUtils;

 import org.apache.maven.it.util.cli.Commandline;

 import org.apache.maven.it.util.cli.StreamConsumer;

 import org.apache.maven.it.util.cli.WriterStreamConsumer;

 /**

  * @author Benjamin Bentmann

  */

 class ForkedLauncher

     implements MavenLauncher

 {

     private final String mavenHome;

     private final String executable;

     public ForkedLauncher( String mavenHome )

     {

         this( mavenHome, false );

     }

     public ForkedLauncher( String mavenHome, boolean debugJvm )

     {

         this.mavenHome = mavenHome;

         String script = debugJvm ? "mvnDebug" : "mvn";

         if ( mavenHome != null )

         {

             executable = new File( mavenHome, "bin/" + script ).getPath();

         }

         else

         {

             executable = script;

         }

     }

     public int run( String[] cliArgs, Map envVars, String workingDirectory, File logFile )

         throws IOException, LauncherException

     {

         Commandline cmd = new Commandline();

         cmd.setExecutable( executable );

         if ( mavenHome != null )

         {

             cmd.addEnvironment( "M2_HOME", mavenHome );

         }

         if ( envVars != null )

         {

             for ( Iterator i = envVars.keySet().iterator(); i.hasNext(); )

             {

                 String key = (String) i.next();

                 cmd.addEnvironment( key, (String) envVars.get( key ) );

             }

         }

         if ( envVars == null || envVars.get( "JAVA_HOME" ) == null )

         {

             cmd.addEnvironment( "JAVA_HOME", System.getProperty( "java.home" ) );

         }

         cmd.addEnvironment( "MAVEN_TERMINATE_CMD", "on" );

         cmd.setWorkingDirectory( workingDirectory );

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

         {

             cmd.createArgument().setValue( cliArgs[i] );

         }

         Writer logWriter = new FileWriter( logFile );

         StreamConsumer out = new WriterStreamConsumer( logWriter );

         StreamConsumer err = new WriterStreamConsumer( logWriter );

         try

         {

             return CommandLineUtils.executeCommandLine( cmd, out, err );

         }

         catch ( CommandLineException e )

         {

             throw new LauncherException( "Failed to run Maven: " + e.getMessage() + "\n" + cmd, e );

         }

         finally

         {

             logWriter.close();

         }

     }

     public int run( String[] cliArgs, String workingDirectory, File logFile )

         throws IOException, LauncherException

     {

         return run( cliArgs, Collections.EMPTY_MAP, workingDirectory, logFile );

     }

 }