branchmodel.h 3.15 KB
Newer Older
1
2
3
4
5
#ifndef BRANCHMODEL_H
#define BRANCHMODEL_H

#include <QtCore/QAbstractListModel>
#include <QtCore/QList>
6
#include <QtCore/QVariant>
7
8
9
10
11
12

namespace Git {
    namespace Internal {

class GitClient;

13
14
15
16
/* A read-only model to list git remote branches in a simple list of branch names.
 * The tooltip describes the latest commit (delayed creation).
 * Provides virtuals to be able to derive a local branch model with the notion
 * of a "current branch". */
17

18
19
class RemoteBranchModel : public QAbstractListModel {
    Q_OBJECT
20
public:
21
    explicit RemoteBranchModel(GitClient *client, QObject *parent = 0);
22

23
    virtual bool refresh(const QString &workingDirectory, QString *errorMessage);
24
25
26
27
28
29
30
31

    QString branchName(int row) const;

    // QAbstractListModel
    virtual int rowCount(const QModelIndex &parent = QModelIndex()) const;
    virtual QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const;
    virtual Qt::ItemFlags flags(const QModelIndex &index) const;

32
    int branchCount() const;
33

34
35
36
37
    QString workingDirectory() const;
    int findBranchByName(const QString &name) const;

protected:
38
39
40
41
42
43
44
45
46
    struct Branch {
        bool parse(const QString &line, bool *isCurrent);

        QString name;
        QString currentSHA;
        mutable QString toolTip;
    };
    typedef QList<Branch> BranchList;

47
48
49
50
51
52
53
54
    /* Parse git output and populate m_branches. */
    bool refreshBranches(const QString &workingDirectory, bool remoteBranches,
                         int *currentBranch, QString *errorMessage);
    bool runGitBranchCommand(const QString &workingDirectory, const QStringList &additionalArgs, QString *output, QString *errorMessage);

private:
    QString toolTip(const QString &sha) const;

55
56
    const Qt::ItemFlags m_flags;

57
    GitClient *m_client;
58
59
    QString m_workingDirectory;
    BranchList m_branches;
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
};

/* LocalBranchModel: Extends RemoteBranchModel by a read-only
 * checkable column indicating the current branch. Provides an
 * editable "Type here" new-branch-row at the bottom to create
 * a new branch. */

class LocalBranchModel : public RemoteBranchModel {
    Q_OBJECT
public:

    explicit LocalBranchModel(GitClient *client,
                         QObject *parent = 0);

    virtual bool refresh(const QString &workingDirectory, QString *errorMessage);

    // is this the "type here" row?
    bool isNewBranchRow(int row) const;
    bool isNewBranchRow(const QModelIndex & index) const { return isNewBranchRow(index.row()); }

    int currentBranch() const;

    // QAbstractListModel
    virtual int rowCount(const QModelIndex &parent = QModelIndex()) const;
    virtual QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const;
    virtual bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole);
    virtual Qt::ItemFlags flags(const QModelIndex &index) const;

signals:
    void newBranchCreated(const QString &);
90
    void newBranchEntered(const QString &);
91
92
93
94
95
96
97
98
99
100

private slots:
    void slotNewBranchDelayedRefresh();

private:
    bool checkNewBranchName(const QString &name) const;

    const QVariant m_typeHere;
    const QVariant m_typeHereToolTip;

101
    int m_currentBranch;
102
    QString m_newBranch;
103
104
105
106
107
108
};

}
}

#endif // BRANCHMODEL_H