package org.apache.maven.scm.plugin;

 /*

  * 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.apache.maven.plugin.MojoExecutionException;

 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.DefaultConsumer;

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

 import java.io.BufferedReader;

 import java.io.File;

 import java.io.InputStreamReader;

 import java.util.Iterator;

 import java.util.Properties;

 /**

  * Pull the project source from the configured scm and execute the configured goals.

  *

  * @author <a href="dantran@gmail.com">Dan T. Tran</a>

  * @version $Id: BootstrapMojo.java 681929 2008-08-02 02:48:19Z dantran $

  * @goal bootstrap

  * @description Boostrap a project

  * @requiresProject false

  */

 public class BootstrapMojo

     extends CheckoutMojo

 {

     /**

      * The goals to run on the clean checkout of a project for the bootstrap goal.

      * If none are specified, then the default goal for the project is executed.

      * Multiple goals should be comma separated.

      *

      * @parameter expression="${goals}"

      */

     private String goals;

     /**

      * A list of profiles to run with the goals.

      * Multiple profiles must be comma separated with no spaces.

      *

      * @parameter expression="${profiles}"

      */

     private String profiles;

     /**

      * The subdirectory (under the project directory) in which to run the goals.

      * The project directory is the same as the checkout directory in most cases,

      * but for some SCMs, it is a subdirectory of the checkout directory.

      *

      * @parameter expression="${goalsDirectory}" default-value=""

      */

     private String goalsDirectory;

     public void execute()

         throws MojoExecutionException

     {

         super.execute();

         if ( this.getCheckoutResult() != null )

         {

                 runGoals( this.getCheckoutResult().getRelativePathProjectDirectory() );

         }

     }

     /**

      * @param relativePathProjectDirectory the project directory's path relative to the checkout

      *                                     directory; or "" if they are the same

      */

     private void runGoals( String relativePathProjectDirectory )

         throws MojoExecutionException

     {

         Commandline cl = new Commandline();

         try

         {

             addSystemEnvironment( cl );

         }

         catch ( Exception e )

         {

             throw new MojoExecutionException( "Can't add system environment variables to mvn command line.", e );

         }

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

         cl.setExecutable( "mvn" );

         cl.setWorkingDirectory( determineWorkingDirectoryPath( this.getCheckoutDirectory(),

                                                                relativePathProjectDirectory, goalsDirectory ) );

         if ( this.goals != null )

         {

             String[] tokens = StringUtils.split( this.goals, ", " );

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

             {

                 cl.createArgument().setValue( tokens[i] );

             }

         }

         if ( ! StringUtils.isEmpty( this.profiles ) )

         {

             cl.createArgument().setValue( "-P" + this.profiles );

         }

         StreamConsumer consumer = new DefaultConsumer();

         try

         {

             int result = CommandLineUtils.executeCommandLine( cl, consumer, consumer );

             if ( result != 0 )

             {

                 throw new MojoExecutionException( "Result of mvn execution is: \'" + result + "\'. Release failed." );

             }

         }

         catch ( CommandLineException e )

         {

             throw new MojoExecutionException( "Can't run goal " + goals, e );

         }

     }

     /**

      * Determines the path of the working directory. By default, this is the checkout directory. For some SCMs, the project root directory

      * is not the checkout directory itself, but a SCM-specific subdirectory. The build can furthermore optionally be executed in a

      * subdirectory of this project directory, in case

      *

      * @param checkoutDirectory

      * @param relativePathProjectDirectory

      * @param goalsDirectory

      * @return

      */

     protected String determineWorkingDirectoryPath( File checkoutDirectory, String relativePathProjectDirectory,

                                                     String goalsDirectory )

     {

         File projectDirectory;

         if ( StringUtils.isNotEmpty( relativePathProjectDirectory ) )

         {

             projectDirectory = new File( checkoutDirectory, relativePathProjectDirectory );

         }

         else

         {

             projectDirectory = checkoutDirectory;

         }

         if ( StringUtils.isEmpty( goalsDirectory ) )

         {

             return projectDirectory.getPath();

         }

         else

         {

             return new File( projectDirectory, goalsDirectory ).getPath();

         }

     }

     /**

      * Add system environment variables

      * Moved to plexus-utils 1.0.5

      */

     private void addSystemEnvironment( Commandline cl )

         throws Exception

     {

         Properties envVars = getSystemEnvVars();

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

         {

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

             cl.addEnvironment( key, envVars.getProperty( key ) );

         }

     }

     private Properties getSystemEnvVars()

         throws Exception

     {

         Process p = null;

         Properties envVars = new Properties();

         Runtime r = Runtime.getRuntime();

         String os = System.getProperty( "os.name" ).toLowerCase();

         //If this is windows set the shell to command.com or cmd.exe with correct arguments.

         if ( os.indexOf( "windows" ) != -1 )

         {

             if ( os.indexOf( "95" ) != -1 || os.indexOf( "98" ) != -1 || os.indexOf( "Me" ) != -1 )

             {

                 p = r.exec( "command.com /c set" );

             }

             else

             {

                 p = r.exec( "cmd.exe /c set" );

             }

         }

         else

         {

             p = r.exec( "env" );

         }

         BufferedReader br = new BufferedReader( new InputStreamReader( p.getInputStream() ) );

         String line;

         while ( ( line = br.readLine() ) != null )

         {

             int idx = line.indexOf( '=' );

             String key = line.substring( 0, idx );

             String value = line.substring( idx + 1 );

             envVars.setProperty( key, value );

             // System.out.println( key + " = " + value );

         }

         return envVars;

     }

 }