package org.apache.maven.scm.provider.perforce.command.tag;

 /*

  * 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.scm.ScmException;

 import org.apache.maven.scm.ScmFileSet;

 import org.apache.maven.scm.ScmResult;

 import org.apache.maven.scm.ScmTagParameters;

 import org.apache.maven.scm.command.tag.AbstractTagCommand;

 import org.apache.maven.scm.command.tag.TagScmResult;

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

 import org.apache.maven.scm.provider.perforce.PerforceScmProvider;

 import org.apache.maven.scm.provider.perforce.command.PerforceCommand;

 import org.apache.maven.scm.provider.perforce.command.PerforceInfoCommand;

 import org.apache.maven.scm.provider.perforce.repository.PerforceScmProviderRepository;

 import org.codehaus.plexus.util.IOUtil;

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

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

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

 import java.io.BufferedReader;

 import java.io.DataOutputStream;

 import java.io.File;

 import java.io.IOException;

 import java.io.InputStreamReader;

 import java.io.OutputStream;

 import java.util.List;

 /**

  * @author Mike Perham

  * @author Olivier Lamy

  * @version $Id: PerforceTagCommand.java 1306867 2012-03-29 13:45:10Z olamy $

  */

 public class PerforceTagCommand

     extends AbstractTagCommand

     implements PerforceCommand

 {

     private String actualRepoLocation = null;

     protected ScmResult executeTagCommand( ScmProviderRepository repo, ScmFileSet files, String tag, String message )

         throws ScmException

     {

         return executeTagCommand( repo, files, tag, new ScmTagParameters( message ) );

     }

     /**

      * {@inheritDoc}

      */

     protected ScmResult executeTagCommand( ScmProviderRepository repo, ScmFileSet files, String tag,

                                            ScmTagParameters scmTagParameters )

         throws ScmException

     {

         PerforceScmProviderRepository prepo = (PerforceScmProviderRepository) repo;

         actualRepoLocation = PerforceScmProvider.getRepoPath( getLogger(), prepo, files.getBasedir() );

         PerforceTagConsumer consumer = new PerforceTagConsumer();

         createLabel( repo, files, tag, consumer, false );

         if ( consumer.isSuccess() )

         {

             syncLabel( repo, files, tag, consumer );

         }

         if ( consumer.isSuccess() )

         {

             // Now update the label if we need to lock it

             if ( shouldLock() )

             {

                 consumer = new PerforceTagConsumer();

                 createLabel( repo, files, tag, consumer, true );

             }

         }

         if ( consumer.isSuccess() )

         {

             // Unclear what to pass as the first arg

             return new TagScmResult( "p4 label -i", consumer.getTagged() );

         }

         // Unclear what to pass as the first arg

         return new TagScmResult( "p4 label -i", "Tag failed", consumer.getOutput(), false );

     }

     private boolean shouldLock()

     {

         return Boolean.valueOf( System.getProperty( "maven.scm.locktag", "true" ) ).booleanValue();

     }

     private void syncLabel( ScmProviderRepository repo, ScmFileSet files, String tag, PerforceTagConsumer consumer )

     {

         Commandline cl =

             createLabelsyncCommandLine( (PerforceScmProviderRepository) repo, files.getBasedir(), files, tag );

         try

         {

             if ( getLogger().isDebugEnabled() )

             {

                 getLogger().debug( PerforceScmProvider.clean( "Executing: " + cl.toString() ) );

             }

             CommandLineUtils.StringStreamConsumer err = new CommandLineUtils.StringStreamConsumer();

             int exitCode = CommandLineUtils.executeCommandLine( cl, consumer, err );

             if ( exitCode != 0 )

             {

                 String cmdLine = CommandLineUtils.toString( cl.getCommandline() );

                 StringBuilder msg = new StringBuilder( "Exit code: " + exitCode + " - " + err.getOutput() );

                 msg.append( '\n' );

                 msg.append( "Command line was:" + cmdLine );

                 throw new CommandLineException( msg.toString() );

             }

         }

         catch ( CommandLineException e )

         {

             if ( getLogger().isErrorEnabled() )

             {

                 getLogger().error( "CommandLineException " + e.getMessage(), e );

             }

         }

     }

     private void createLabel( ScmProviderRepository repo, ScmFileSet files, String tag, PerforceTagConsumer consumer,

                               boolean lock )

     {

         Commandline cl = createLabelCommandLine( (PerforceScmProviderRepository) repo, files.getBasedir() );

         DataOutputStream dos = null;

         InputStreamReader isReader = null;

         InputStreamReader isReaderErr = null;

         try

         {

             if ( getLogger().isDebugEnabled() )

             {

                 getLogger().debug( PerforceScmProvider.clean( "Executing: " + cl.toString() ) );

             }

             Process proc = cl.execute();

             OutputStream out = proc.getOutputStream();

             dos = new DataOutputStream( out );

             String label = createLabelSpecification( (PerforceScmProviderRepository) repo, tag, lock );

             if ( getLogger().isDebugEnabled() )

             {

                 getLogger().debug( "LabelSpec: " + NEWLINE + label );

             }

             dos.write( label.getBytes() );

             dos.close();

             out.close();

             // TODO find & use a less naive InputStream multiplexer

             isReader = new InputStreamReader( proc.getInputStream() );

             isReaderErr = new InputStreamReader( proc.getErrorStream() );

             BufferedReader stdout = new BufferedReader( isReader );

             BufferedReader stderr = new BufferedReader( isReaderErr );

             String line;

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

             {

                 if ( getLogger().isDebugEnabled() )

                 {

                     getLogger().debug( "Consuming stdout: " + line );

                 }

                 consumer.consumeLine( line );

             }

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

             {

                 if ( getLogger().isDebugEnabled() )

                 {

                     getLogger().debug( "Consuming stderr: " + line );

                 }

                 consumer.consumeLine( line );

             }

             stderr.close();

             stdout.close();

         }

         catch ( CommandLineException e )

         {

             if ( getLogger().isErrorEnabled() )

             {

                 getLogger().error( "CommandLineException " + e.getMessage(), e );

             }

         }

         catch ( IOException e )

         {

             if ( getLogger().isErrorEnabled() )

             {

                 getLogger().error( "IOException " + e.getMessage(), e );

             }

         }

         finally

         {

             IOUtil.close( dos );

             IOUtil.close( isReader );

             IOUtil.close( isReaderErr );

         }

     }

     public static Commandline createLabelCommandLine( PerforceScmProviderRepository repo, File workingDirectory )

     {

         Commandline command = PerforceScmProvider.createP4Command( repo, workingDirectory );

         command.createArg().setValue( "label" );

         command.createArg().setValue( "-i" );

         return command;

     }

     public static Commandline createLabelsyncCommandLine( PerforceScmProviderRepository repo, File workingDirectory,

                                                           ScmFileSet files, String tag )

     {

         Commandline command = PerforceScmProvider.createP4Command( repo, workingDirectory );

         command.createArg().setValue( "labelsync" );

         command.createArg().setValue( "-l" );

         command.createArg().setValue( tag );

         List<File> fs = files.getFileList();

         for ( File file : fs )

         {

             command.createArg().setValue( file.getPath() );

         }

         return command;

     }

     private static final String NEWLINE = "\r\n";

     /*

      * Label: foo-label

      * View: //depot/path/to/repos/...

      * Owner: mperham

      */

     public String createLabelSpecification( PerforceScmProviderRepository repo, String tag, boolean lock )

     {

         StringBuilder buf = new StringBuilder();

         buf.append( "Label: " ).append( tag ).append( NEWLINE );

         buf.append( "View: " ).append( PerforceScmProvider.getCanonicalRepoPath( actualRepoLocation ) ).append(

             NEWLINE );

         String username = repo.getUser();

         if ( username == null )

         {

             // I have no idea why but Perforce doesn't default the owner to the current user.

             // Since the user is not explicitly set, we use 'p4 info' to query for the current user.

             username = PerforceInfoCommand.getInfo( getLogger(), repo ).getEntry( "User name" );

         }

         buf.append( "Owner: " ).append( username ).append( NEWLINE );

         buf.append( "Options: " ).append( lock ? "" : "un" ).append( "locked" ).append( NEWLINE );

         return buf.toString();

     }

 }