WorkingTreeItemView.cpp

/*
 * MacGitver
 * Copyright (C) 2012 Sascha Cunz <sascha@babbelbox.org>
 *
 * This program is free software; you can redistribute it and/or modify it under the terms of the
 * GNU General Public License (Version 2) as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
 * even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with this program; if
 * not, see <http://www.gnu.org/licenses/>.
 *
 */

#include "WorkingTreeItemView.h"

#include "libMacGitverCore/Config/Config.h"
#include "libMacGitverCore/Widgets/HeaderView.h"

#include "libGitWrap/Index.hpp"
#include "libGitWrap/Result.hpp"

#include "WorkingTreeAbstractItem.h"
#include "WorkingTreeModel.h"

#include <QAbstractProxyModel>
#include <QMessageBox>


WorkingTreeItemView::WorkingTreeItemView(QWidget *parent)
    : TreeViewCtxMenu( parent )
    , mHeader( new HeaderView( Qt::Horizontal ) )
    , mModel( 0 )
{
#ifdef Q_OS_MACX
    setAttribute( Qt::WA_MacShowFocusRect, false );
#endif

    setHeader( mHeader );
    mHeader->setConfigName( QLatin1String( "Worktree/Columns" ) );

    WorkingTreeCtxMenu::setupActions( this );

    connect( this, SIGNAL(contextMenu(QModelIndex, QPoint)),
             this, SLOT(contextMenu(QModelIndex, QPoint)) );
}

void WorkingTreeItemView::setModel(QAbstractItemModel *model)
{
    TreeViewCtxMenu::setModel( model );

    QAbstractProxyModel *apm = qobject_cast< QAbstractProxyModel* >( model );
    while ( apm )
    {
        model = apm->sourceModel();
        Q_ASSERT( model );
        apm = qobject_cast< QAbstractProxyModel* >( model );
    }

    mModel = qobject_cast< WorkingTreeModel * >( model );
}

void WorkingTreeItemView::onWtCtxStage()
{
    Heaven::Action* action = qobject_cast< Heaven::Action* >( sender() );
    if ( !action )
        return;

    QModelIndex srcIndex = deeplyMapToSource( currentIndex() );
    if ( !srcIndex.isValid() )
        return;

    WorkingTreeAbstractItem* item = mModel->indexToItem( srcIndex );
    Q_ASSERT( item );

    Git::Repository repo = mModel->repository();

    Git::Result r;
    Git::Index i = repo.index( r );

    i.read( r );
    i.addFile( r, item->path() );
    i.write( r );

    // TODO: workaround to update the model
    mModel->setRepository(repo);

    if ( !r )
    {
        QMessageBox::warning( this, trUtf8("Error while adding file to Git index"),
                              trUtf8("File not staged. Git message:\n%1").arg(r.errorText()) );
    }
}

void WorkingTreeItemView::onWtCtxDiscard()
{
    int button = QMessageBox::question( this, trUtf8("Discard unstaged changes"),
                                        trUtf8("Discard unstaged changes?\n\n"
                                               "You cannot undo this action."),
                                        QMessageBox::Ok | QMessageBox::Abort );
    if ( button != QMessageBox::Ok )
        return;

    Heaven::Action* action = qobject_cast< Heaven::Action* >( sender() );
    if ( !action )
        return;

    QModelIndex srcIndex = deeplyMapToSource( currentIndex() );
    if ( !srcIndex.isValid() )
        return;

    WorkingTreeAbstractItem* item = mModel->indexToItem( srcIndex );
    Q_ASSERT( item );

    Git::Repository repo = mModel->repository();

    Git::Result r;
    Git::Index i = repo.index( r );

    // index stays unchanged; no need to read/write
    i.checkoutFiles( r, QStringList( item->path() ) );

    // TODO: workaround to update the model
    mModel->setRepository(repo);

    if ( !r )
    {
        QMessageBox::warning( this, trUtf8("Error while discarding file changes"),
                              trUtf8("File remains unchanged. Git message:\n%1").arg(r.errorText()) );
    }
}

void WorkingTreeItemView::contextMenu( const QModelIndex& index, const QPoint& globalPos )
{
    Q_ASSERT( mModel );

    QModelIndex srcIndex = deeplyMapToSource( index );
    if ( !srcIndex.isValid() )
        return;

    WorkingTreeAbstractItem* item = mModel->indexToItem( srcIndex );

    Heaven::Menu* menu = 0;
    if ( item && !item->isDirectory() )
    {
        menu = menuCtxMenuWtFile;
        //menu->setActivationContext( item );
    }

    if ( menu )
        menu->showPopup( globalPos );
}

QModelIndex WorkingTreeItemView::deeplyMapToSource( QModelIndex current ) const
{
    while( current.isValid() )
    {
        const QAbstractProxyModel* apm = qobject_cast< const QAbstractProxyModel* >( current.model() );
        if( !apm )
            return current;

        current = apm->mapToSource( current );
    }

    return QModelIndex();
}