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.unedit.AbstractUnEditCommand;
026    import org.apache.maven.scm.command.unedit.UnEditScmResult;
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:unedit -DworkingDirectory=<dir> -Dincludes=*
032    public class TfsUnEditCommand
033        extends AbstractUnEditCommand
034    {
035    
036        protected ScmResult executeUnEditCommand( ScmProviderRepository r, ScmFileSet f )
037            throws ScmException
038        {
039            FileListConsumer out = new FileListConsumer();
040            ErrorStreamConsumer err = new ErrorStreamConsumer();
041            TfsCommand command = createCommand( r, f );
042            int status = command.execute( out, err );
043            if ( status != 0 || err.hasBeenFed() )
044            {
045                return new UnEditScmResult( command.getCommandString(), "Error code for TFS unedit command - " + status,
046                                            err.getOutput(), false );
047            }
048            return new UnEditScmResult( command.getCommandString(), out.getFiles() );
049        }
050    
051        public TfsCommand createCommand( ScmProviderRepository r, ScmFileSet f )
052        {
053            TfsCommand command = new TfsCommand( "undo", r, f, getLogger() );
054            command.addArgument( f );
055            return command;
056        }
057    }