package org.apache.maven.scm.provider.tfs.command;

 /*

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

 import org.apache.maven.scm.command.checkout.AbstractCheckOutCommand;

 import org.apache.maven.scm.command.checkout.CheckOutScmResult;

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

 import org.apache.maven.scm.provider.tfs.TfsScmProviderRepository;

 import org.apache.maven.scm.provider.tfs.command.consumer.ErrorStreamConsumer;

 import org.apache.maven.scm.provider.tfs.command.consumer.FileListConsumer;

 // Usage: mvn scm:checkout -DcheckoutDirectory=<dir>

 public class TfsCheckOutCommand

     extends AbstractCheckOutCommand

 {

     protected CheckOutScmResult executeCheckOutCommand( ScmProviderRepository r, ScmFileSet f, ScmVersion v,

                                                         boolean recursive )

         throws ScmException

     {

         TfsScmProviderRepository tfsRepo = (TfsScmProviderRepository) r;

         String url = tfsRepo.getServerPath();

         String tfsUrl = tfsRepo.getTfsUrl();

         String workspace = tfsRepo.getWorkspace();

         // Try creating workspace

         boolean workspaceProvided = workspace != null && !workspace.trim().equals( "" );

         if ( workspaceProvided )

         {

             createWorkspace( r, f, workspace, tfsUrl );

         }

         TfsCommand command;

         int status;

         if ( workspaceProvided )

         {

             status = executeUnmapCommand( r, f );

         }

         ErrorStreamConsumer out = new ErrorStreamConsumer();

         ErrorStreamConsumer err = new ErrorStreamConsumer();

         if ( workspaceProvided )

         {

             command = new TfsCommand( "workfold", r, null, getLogger() );

             command.addArgument( "-workspace:" + workspace );

             command.addArgument( "-map" );

             command.addArgument( url );

             command.addArgument( f.getBasedir().getAbsolutePath() );

             status = command.execute( out, err );

             if ( status != 0 || err.hasBeenFed() )

             {

                 return new CheckOutScmResult( command.getCommandString(),

                                               "Error code for TFS checkout (workfold map) command - " + status,

                                               err.getOutput(), false );

             }

         }

         FileListConsumer fileConsumer = new FileListConsumer();

         err = new ErrorStreamConsumer();

         command = createGetCommand( r, f, v, recursive );

         status = command.execute( fileConsumer, err );

         if ( status != 0 || err.hasBeenFed() )

         {

             return new CheckOutScmResult( command.getCommandString(), "Error code for TFS checkout (get) command - "

                 + status, err.getOutput(), false );

         }

         return new CheckOutScmResult( command.getCommandString(), fileConsumer.getFiles() );

     }

     public TfsCommand createGetCommand( ScmProviderRepository r, ScmFileSet f, ScmVersion v, boolean recursive )

     {

         TfsCommand command = new TfsCommand( "get", r, f, getLogger() );

         if ( recursive )

         {

             command.addArgument( "-recursive" );

         }

         command.addArgument( "-force" );

         if ( v != null && !v.equals( "" ) )

         {

             String vType = "";

             if ( v.getType().equals( "Tag" ) )

             {

                 vType = "L";

             }

             if ( v.getType().equals( "Revision" ) )

             {

                 vType = "C";

             }

             command.addArgument( "-version:" + vType + v.getName() );

         }

         command.addArgument( f.getBasedir().getAbsolutePath() );

         return command;

     }

     public int executeUnmapCommand( ScmProviderRepository r, ScmFileSet f )

         throws ScmException

     {

         TfsScmProviderRepository tfsRepo = (TfsScmProviderRepository) r;

         String url = tfsRepo.getServerPath();

         String workspace = tfsRepo.getWorkspace();

         ErrorStreamConsumer out = new ErrorStreamConsumer();

         ErrorStreamConsumer err = new ErrorStreamConsumer();

         TfsCommand command = new TfsCommand( "workfold", r, null, getLogger() );

         command.addArgument( "-workspace:" + workspace );

         command.addArgument( "-unmap" );

         command.addArgument( url );

         return command.execute( out, err );

     }

     private void createWorkspace( ScmProviderRepository r, ScmFileSet f, String workspace, String url )

         throws ScmException

     {

         ErrorStreamConsumer out = new ErrorStreamConsumer();

         ErrorStreamConsumer err = new ErrorStreamConsumer();

         // Checkout dir may not exist yet

         TfsCommand command = new TfsCommand( "workspace", r, null, getLogger() );

         command.addArgument( "-new" );

         command.addArgument( "-comment:Creating workspace for maven command" );

         command.addArgument( "-server:" + url );

         command.addArgument( workspace );

         command.execute( out, err );

     }

 }