001    package org.apache.maven.scm.provider.starteam.command.checkout;
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.ScmFile;
023    import org.apache.maven.scm.ScmFileStatus;
024    import org.apache.maven.scm.log.ScmLogger;
025    import org.apache.maven.scm.provider.starteam.command.StarteamCommandLineUtils;
026    import org.codehaus.plexus.util.cli.StreamConsumer;
027    
028    import java.io.File;
029    import java.util.ArrayList;
030    import java.util.List;
031    
032    /**
033     * @author <a href="mailto:dantran@gmail.com">Dan T. Tran</a>
034     * @author Olivier Lamy
035     * @version $Id: StarteamCheckOutConsumer.java 1054738 2011-01-03 20:40:39Z olamy $
036     */
037    public class StarteamCheckOutConsumer
038        implements StreamConsumer
039    {
040        private ScmLogger logger;
041    
042        private String workingDirectory;
043    
044        private String currentDir = "";
045    
046        private List<ScmFile> files = new ArrayList<ScmFile>();
047    
048        /**
049         * Marks current directory data
050         */
051        private static final String DIR_MARKER = "(working dir: ";
052    
053        /**
054         * Marks current file data
055         */
056        private static final String CHECKOUT_MARKER = ": checked out";
057    
058        /**
059         * Marks skipped file during update
060         */
061        private static final String SKIPPED_MARKER = ": skipped";
062    
063        public StarteamCheckOutConsumer( ScmLogger logger, File workingDirectory )
064        {
065            this.logger = logger;
066    
067            this.workingDirectory = workingDirectory.getPath().replace( '\\', '/' );
068        }
069    
070        /** {@inheritDoc} */
071        public void consumeLine( String line )
072        {
073            if ( logger.isDebugEnabled() )
074            {
075                logger.debug( line );
076            }
077    
078            int pos = 0;
079    
080            if ( ( pos = line.indexOf( CHECKOUT_MARKER ) ) != -1 )
081            {
082                processCheckedOutFile( line, pos );
083            }
084            else if ( ( pos = line.indexOf( DIR_MARKER ) ) != -1 )
085            {
086                processDirectory( line, pos );
087            }
088            else if ( ( pos = line.indexOf( CHECKOUT_MARKER ) ) != -1 )
089            {
090                processCheckedOutFile( line, pos );
091            }
092            else if ( ( pos = line.indexOf( SKIPPED_MARKER ) ) != -1 )
093            {
094                processSkippedFile( line, pos );
095            }
096            else
097            {
098                if ( logger.isWarnEnabled() )
099                {
100                    logger.warn( "Unknown checkout ouput: " + line );
101                }
102            }
103        }
104    
105        public List<ScmFile> getCheckedOutFiles()
106        {
107            return files;
108        }
109    
110        private void processDirectory( String line, int pos )
111        {
112            String dirPath = line.substring( pos + DIR_MARKER.length(), line.length() - 1 ).replace( '\\', '/' );
113    
114            try
115            {
116                this.currentDir = StarteamCommandLineUtils.getRelativeChildDirectory( this.workingDirectory, dirPath );
117            }
118            catch ( IllegalStateException e )
119            {
120                String error = "Working and checkout directories are not on the same tree";
121    
122                if ( logger.isErrorEnabled() )
123                {
124                    logger.error( error );
125                    logger.error( "Working directory: " + workingDirectory );
126                    logger.error( "Checked out directory: " + dirPath );
127                }
128    
129                throw new IllegalStateException( error );
130            }
131    
132        }
133    
134        private void processCheckedOutFile( String line, int pos )
135        {
136            String checkedOutFilePath = this.currentDir + "/" + line.substring( 0, pos );
137    
138            this.files.add( new ScmFile( checkedOutFilePath, ScmFileStatus.CHECKED_OUT ) );
139    
140            if ( logger.isInfoEnabled() )
141            {
142                logger.info( "Checked out: " + checkedOutFilePath );
143            }
144        }
145    
146        private void processSkippedFile( String line, int pos )
147        {
148            String skippedFilePath = this.currentDir + "/" + line.substring( 0, pos );
149    
150            if ( logger.isDebugEnabled() )
151            {
152                logger.debug( "Skipped: " + skippedFilePath );
153            }
154        }
155    
156    }