gitmaster/qgitrepotreewidget.cpp at master · bokic/gitmaster · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include "qgitrepotreewidget.h"
#include "qgitrepotreeitemdelegate.h"


QGitRepoTreeWidget::QGitRepoTreeWidget(QWidget *parent)
    : QTreeWidget(parent)
    , m_thread(this)
    , m_git(new QGit())
{
    m_git->moveToThread(&m_thread);

    connect(this, &QGitRepoTreeWidget::repositoryStatus, m_git, &QGit::status);
    connect(m_git, &QGit::statusReply, this, &QGitRepoTreeWidget::repositoryStatusReply);

    m_thread.setObjectName("QGit(tree)");
    m_thread.start();
}

QGitRepoTreeWidget::~QGitRepoTreeWidget()
{
    m_thread.quit();
    m_thread.wait();

    delete m_git; m_git = nullptr;
}

void QGitRepoTreeWidget::refreshItems()
{
    m_refreshIndex = 0;

    refreshItem();
}

void QGitRepoTreeWidget::repositoryStatusReply(QMap<git_status_t,int> items, QGitError error)
{
    Q_UNUSED(error)

    for(int index = 0; index < topLevelItemCount(); index++)
    {
        QTreeWidgetItem *item = topLevelItem(index);

        QString item_path = item->data(0, QGitRepoTreeItemDelegate::QItemPath).toString();

        if (item_path == QDir::toNativeSeparators(m_git->path().absolutePath()))
        {
            item->setData(0, QGitRepoTreeItemDelegate::QItemModifiedFiles, 0);
            item->setData(0, QGitRepoTreeItemDelegate::QItemDeletedFiles, 0);
            item->setData(0, QGitRepoTreeItemDelegate::QItemNewFiles, 0);
            item->setData(0, QGitRepoTreeItemDelegate::QItemUnversionedFiles, 0);
            item->setData(0, QGitRepoTreeItemDelegate::QItemBranchName, QGit::getBranchNameFromPath(item_path));

            for(const auto &[key, value]: items.asKeyValueRange())
            {
                if (key & (GIT_STATUS_WT_MODIFIED | GIT_STATUS_INDEX_MODIFIED))
                {
                    item->setData(0, QGitRepoTreeItemDelegate::QItemModifiedFiles, item->data(0, QGitRepoTreeItemDelegate::QItemModifiedFiles).toInt() + value);
                }

                if (key & (GIT_STATUS_WT_DELETED | GIT_STATUS_INDEX_DELETED))
                {
                    item->setData(0, QGitRepoTreeItemDelegate::QItemDeletedFiles, item->data(0, QGitRepoTreeItemDelegate::QItemDeletedFiles).toInt() + value);
                }

                if (key & (GIT_STATUS_WT_NEW | GIT_STATUS_INDEX_NEW))
                {
                    item->setData(0, QGitRepoTreeItemDelegate::QItemNewFiles, item->data(0, QGitRepoTreeItemDelegate::QItemNewFiles).toInt() + value);
                }
            }

            update();
            break;
        }
    }

    refreshItem();
}

void QGitRepoTreeWidget::refreshItem()
{
    if (m_refreshIndex >= topLevelItemCount())
    {
        m_refreshIndex = 0;

        return;
    }

    QTreeWidgetItem *item = topLevelItem(m_refreshIndex++);

    QDir dir(item->data(0, QGitRepoTreeItemDelegate::QItemPath).toString());

    m_git->setPath(dir);

    emit repositoryStatus();
}