001    package org.apache.maven.scm.provider.tfs.command;
002    
003    /*
004     * Licensed to the Apache Software Foundation (ASF) under one
005     * or more contributor license agreements.  See the NOTICE file
006     * distributed with this work for additional information
007     * regarding copyright ownership.  The ASF licenses this file
008     * to you under the Apache License, Version 2.0 (the
009     * "License"); you may not use this file except in compliance
010     * with the License.  You may obtain a copy of the License at
011     *
012     * http://www.apache.org/licenses/LICENSE-2.0
013     *
014     * Unless required by applicable law or agreed to in writing,
015     * software distributed under the License is distributed on an
016     * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
017     * KIND, either express or implied.  See the License for the
018     * specific language governing permissions and limitations
019     * under the License.
020     */
021    
022    import org.apache.maven.scm.ScmException;
023    import org.apache.maven.scm.ScmFileSet;
024    import org.apache.maven.scm.ScmResult;
025    import org.apache.maven.scm.command.edit.AbstractEditCommand;
026    import org.apache.maven.scm.command.edit.EditScmResult;
027    import org.apache.maven.scm.provider.ScmProviderRepository;
028    import org.apache.maven.scm.provider.tfs.command.consumer.ErrorStreamConsumer;
029    import org.apache.maven.scm.provider.tfs.command.consumer.FileListConsumer;
030    
031    //Usage: mvn scm:edit -DworkingDirectory=<dir> -Dincludes=*
032    public class TfsEditCommand
033        extends AbstractEditCommand
034    {
035    
036        protected ScmResult executeEditCommand( ScmProviderRepository r, ScmFileSet f )
037            throws ScmException
038        {
039            FileListConsumer out = new FileListConsumer();
040            ErrorStreamConsumer err = new ErrorStreamConsumer();
041            
042            TfsCommand command = createCommand( r, f );
043            int status = command.execute( out, err );
044            
045            if ( status != 0 || err.hasBeenFed() )
046            {
047                return new EditScmResult( command.getCommandString(), "Error code for TFS edit command - " + status,
048                                          err.getOutput(), false );
049            }
050            
051            return new EditScmResult( command.getCommandString(), out.getFiles() );
052        }
053    
054        protected TfsCommand createCommand( ScmProviderRepository r, ScmFileSet f )
055        {
056            TfsCommand command = new TfsCommand( "checkout", r, f, getLogger() );
057            command.addArgument( f );
058            return command;
059        }
060    }