package org.apache.maven.scm.provider.hg.command.branch;

 /*

  * 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.util.ArrayList;

 import java.util.Iterator;

 import java.util.List;

 import org.apache.maven.scm.ScmBranchParameters;

 import org.apache.maven.scm.ScmException;

 import org.apache.maven.scm.ScmFile;

 import org.apache.maven.scm.ScmFileSet;

 import org.apache.maven.scm.ScmFileStatus;

 import org.apache.maven.scm.ScmResult;

 import org.apache.maven.scm.command.Command;

 import org.apache.maven.scm.command.branch.AbstractBranchCommand;

 import org.apache.maven.scm.command.branch.BranchScmResult;

 import org.apache.maven.scm.provider.ScmProviderRepository;

 import org.apache.maven.scm.provider.hg.HgUtils;

 import org.apache.maven.scm.provider.hg.command.HgCommandConstants;

 import org.apache.maven.scm.provider.hg.command.HgConsumer;

 import org.apache.maven.scm.provider.hg.command.inventory.HgListConsumer;

 import org.apache.maven.scm.provider.hg.repository.HgScmProviderRepository;

 import org.codehaus.plexus.util.StringUtils;

 /**

  * Branch. Mercurial has weird branches. After a branch is created, it must be committed to the server, otherwise

  * the branch does not exist (yet) in the repository.

  *

  * @author Henning Schmiedehausen

  * @version $Id: HgBranchCommand.java 1061024 2011-01-19 21:54:37Z olamy $

  */

 public class HgBranchCommand

     extends AbstractBranchCommand

     implements Command

 {

     protected ScmResult executeBranchCommand( ScmProviderRepository scmProviderRepository, ScmFileSet fileSet, String branch,

                                            String message )

         throws ScmException

     {

         return executeBranchCommand( scmProviderRepository, fileSet, branch, new ScmBranchParameters( message ) );

     }

     /**

      * {@inheritDoc}

      */

     protected ScmResult executeBranchCommand( ScmProviderRepository scmProviderRepository, ScmFileSet fileSet, String branch,

                                            ScmBranchParameters scmBranchParameters )

         throws ScmException

     {

         if ( StringUtils.isBlank( branch ) )

         {

             throw new ScmException( "branch must be specified" );

         }

         if ( !fileSet.getFileList().isEmpty() )

         {

             throw new ScmException( "This provider doesn't support branchging subsets of a directory" );

         }

         File workingDir = fileSet.getBasedir();

         // build the command

         String[] branchCmd =

             new String[]{ HgCommandConstants.BRANCH_CMD, branch };

         // keep the command about in string form for reporting

         ;

         HgConsumer branchConsumer = new HgConsumer( getLogger() ) {

             public void doConsume( ScmFileStatus status, String trimmedLine )

             {

             }

         };

         ScmResult result = HgUtils.execute( branchConsumer, getLogger(), workingDir, branchCmd );

         HgScmProviderRepository repository = (HgScmProviderRepository) scmProviderRepository;

         if ( !result.isSuccess() )

         {

             throw new ScmException( "Error while executing command " + joinCmd( branchCmd ) );

         }

         // First commit.

         String[] commitCmd = new String[]{ HgCommandConstants.COMMIT_CMD, HgCommandConstants.MESSAGE_OPTION, scmBranchParameters.getMessage() };

         result = HgUtils.execute( new HgConsumer( getLogger() ), getLogger(), workingDir, commitCmd );

         if ( !result.isSuccess() )

         {

             throw new ScmException( "Error while executing command " + joinCmd( commitCmd ) );

         }

         // now push, if we should.

         if ( repository.isPushChanges() )

         {

             if ( !repository.getURI().equals( fileSet.getBasedir().getAbsolutePath() ) )

             {

                 String[] pushCmd = new String[] {

                     HgCommandConstants.PUSH_CMD,

                     HgCommandConstants.NEW_BRANCH_OPTION,

                     repository.getURI()

                 };

                 result = HgUtils.execute( new HgConsumer( getLogger() ), getLogger(), fileSet.getBasedir(), pushCmd );

                 if ( !result.isSuccess() )

                 {

                     throw new ScmException( "Error while executing command " + joinCmd( pushCmd ) );

                 }

             }

         }

         // do an inventory to return the files branched (all of them)

         String[] listCmd = new String[]{ HgCommandConstants.INVENTORY_CMD };

         HgListConsumer listconsumer = new HgListConsumer( getLogger() );

         result = HgUtils.execute( listconsumer, getLogger(), fileSet.getBasedir(), listCmd );

         if ( !result.isSuccess() )

         {

             throw new ScmException( "Error while executing command " + joinCmd(listCmd) );

         }

         List<ScmFile> files = listconsumer.getFiles();

         List<ScmFile> fileList = new ArrayList<ScmFile>();

         for ( Iterator<ScmFile> i = files.iterator(); i.hasNext(); )

         {

             ScmFile f = i.next();

             fileList.add( new ScmFile( f.getPath(), ScmFileStatus.TAGGED ) );

         }

         return new BranchScmResult( fileList, result );

     }

     private String joinCmd( String[] cmd )

     {

         StringBuilder result = new StringBuilder();

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

         {

             String s = cmd[i];

             result.append( s );

             if ( i < cmd.length - 1 )

             {

                 result.append( " " );

             }

         }

         return result.toString();

     }

 }