Fixes: Start a git branch dialog.

1 files changed, 136 insertions, 0 deletions

diff --git a/src/plugins/git/branchmodel.cpp b/src/plugins/git/branchmodel.cpp
new file mode 100644
index 0000000000..0881aea654
--- /dev/null
+++ b/src/plugins/git/branchmodel.cpp
@@ -0,0 +1,136 @@
+#include "branchmodel.h"
+#include "gitclient.h"
+
+#include <QtCore/QDebug>
+
+enum { debug = 0 };
+
+namespace Git {
+    namespace Internal {
+
+// Parse a branch line: " *name sha description".  Return  true if it is
+// the current one
+bool BranchModel::Branch::parse(const QString &lineIn, bool *isCurrent)
+{
+    if (debug)
+        qDebug() << Q_FUNC_INFO << lineIn;
+
+    *isCurrent = lineIn.startsWith(QLatin1String("* "));
+    if (lineIn.size() < 3)
+        return false;
+
+    const QStringList tokens =lineIn.mid(2).split(QLatin1Char(' '), QString::SkipEmptyParts);
+    if (tokens.size() < 2)
+        return false;
+    name = tokens.at(0);
+    currentSHA= tokens.at(1);
+    toolTip.clear();
+    return true;
+}
+
+static inline Qt::ItemFlags typeToModelFlags(BranchModel::Type t)
+{
+    Qt::ItemFlags rc = Qt::ItemIsSelectable|Qt::ItemIsEnabled;
+    if (t == BranchModel::BranchModel::LocalBranches)
+        rc |= Qt::ItemIsUserCheckable;
+    return rc;
+}
+
+// --- BranchModel
+BranchModel::BranchModel(GitClient *client, Type type, QObject *parent) :
+    QAbstractListModel(parent),
+    m_type(type),
+    m_flags(typeToModelFlags(type)),
+    m_client(client),
+    m_currentBranch(-1)
+{
+}
+
+int BranchModel::currentBranch() const
+{
+    return m_currentBranch;
+}
+
+QString BranchModel::branchName(int row) const
+{
+    return m_branches.at(row).name;
+}
+
+int BranchModel::rowCount(const QModelIndex & /* parent */) const
+{
+    return m_branches.size();
+}
+
+QVariant BranchModel::data(const QModelIndex &index, int role) const
+{
+    const int row = index.row();
+    switch (role) {
+        case Qt::DisplayRole:
+        return branchName(row);
+        case Qt::ToolTipRole:
+        if (m_branches.at(row).toolTip.isEmpty())
+            m_branches.at(row).toolTip = toolTip(m_branches.at(row).currentSHA);
+        return m_branches.at(row).toolTip;
+        break;
+        case Qt::CheckStateRole:
+        if (m_type == RemoteBranches)
+            return QVariant();
+        return row == m_currentBranch ? Qt::Checked : Qt::Unchecked;
+        default:
+        break;
+    }
+    return QVariant();
+}
+
+Qt::ItemFlags BranchModel::flags(const QModelIndex & /*index */) const
+{
+    return m_flags;
+}
+
+bool BranchModel::refresh(const QString &workingDirectory, QString *errorMessage)
+{
+    // Run branch command with verbose.
+    QStringList branchArgs(QLatin1String("-v"));
+    QString output;
+    if (m_type == RemoteBranches)
+        branchArgs.push_back(QLatin1String("-r"));
+    if (!m_client->synchronousBranchCmd(workingDirectory, branchArgs, &output, errorMessage))
+        return false;
+    if (debug)
+        qDebug() << Q_FUNC_INFO << workingDirectory << output;
+    // Parse output
+    m_workingDirectory = workingDirectory;
+    m_branches.clear();
+    m_currentBranch = -1;
+    const QStringList branches = output.split(QLatin1Char('\n'));
+    const int branchCount = branches.size();
+    bool isCurrent;
+    for (int b = 0; b < branchCount; b++) {
+        Branch newBranch;
+        if (newBranch.parse(branches.at(b), &isCurrent)) {
+            m_branches.push_back(newBranch);
+            if (isCurrent)
+                m_currentBranch = b;
+        }
+    }
+    reset();
+    return true;
+}
+
+QString BranchModel::toolTip(const QString &sha) const
+{
+    // Show the sha description excluding diff as toolTip
+    QString output;
+    QString errorMessage;
+    if (!m_client->synchronousShow(m_workingDirectory, sha, &output, &errorMessage))
+        return errorMessage;
+    // Remove 'diff' output
+    const int diffPos = output.indexOf(QLatin1String("\ndiff --"));
+    if (diffPos != -1)
+        output.remove(diffPos, output.size() - diffPos);
+    return output;
+}
+
+}
+}
+