001package org.apache.maven.scm.provider.git.jgit.command.status;
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
022import org.apache.maven.scm.ScmException;
023import org.apache.maven.scm.ScmFile;
024import org.apache.maven.scm.ScmFileSet;
025import org.apache.maven.scm.ScmFileStatus;
026import org.apache.maven.scm.command.status.AbstractStatusCommand;
027import org.apache.maven.scm.command.status.StatusScmResult;
028import org.apache.maven.scm.provider.ScmProviderRepository;
029import org.apache.maven.scm.provider.git.command.GitCommand;
030import org.apache.maven.scm.provider.git.jgit.command.JGitUtils;
031import org.eclipse.jgit.api.Git;
032import org.eclipse.jgit.api.Status;
033
034import java.util.ArrayList;
035import java.util.Collection;
036import java.util.List;
037
038/**
039 * @author <a href="mailto:struberg@yahoo.de">Mark Struberg</a>
040 * @author Dominik Bartholdi (imod)
041 * @since 1.9
042 */
043public class JGitStatusCommand
044    extends AbstractStatusCommand
045    implements GitCommand
046{
047    /**
048     * {@inheritDoc}
049     */
050    protected StatusScmResult executeStatusCommand( ScmProviderRepository repo, ScmFileSet fileSet )
051        throws ScmException
052    {
053        Git git = null;
054        try
055        {
056            git = JGitUtils.openRepo( fileSet.getBasedir() );
057            Status status = git.status().call();
058            List<ScmFile> changedFiles = getFileStati( status );
059
060            return new StatusScmResult( "JGit status", changedFiles );
061        }
062        catch ( Exception e )
063        {
064            throw new ScmException( "JGit status failure!", e );
065        }
066        finally
067        {
068            JGitUtils.closeRepo( git );
069        }
070    }
071
072    private List<ScmFile> getFileStati( Status status )
073    {
074        List<ScmFile> all = new ArrayList<ScmFile>();
075        addAsScmFiles( all, status.getAdded(), ScmFileStatus.ADDED );
076        addAsScmFiles( all, status.getChanged(), ScmFileStatus.UPDATED );
077        addAsScmFiles( all, status.getConflicting(), ScmFileStatus.CONFLICT );
078        addAsScmFiles( all, status.getModified(), ScmFileStatus.MODIFIED );
079        addAsScmFiles( all, status.getRemoved(), ScmFileStatus.DELETED );
080        return all;
081    }
082
083    private void addAsScmFiles( Collection<ScmFile> all, Collection<String> files, ScmFileStatus status )
084    {
085        for ( String f : files )
086        {
087            all.add( new ScmFile( f, status ) );
088        }
089    }
090}