File indexing completed on 2024-04-28 05:49:04

 /*
     SPDX-FileCopyrightText: 2021 Waqar Ahmed <waqar.17a@gmail.com>
 
     SPDX-License-Identifier: LGPL-2.0-or-later
 */
 #include "branchesdialogmodel.h"
 
 #include <QFont>
 #include <QIcon>
 
 BranchesDialogModel::BranchesDialogModel(QObject *parent)
     : QAbstractTableModel(parent)
 {
 }
 
 int BranchesDialogModel::rowCount(const QModelIndex &parent) const
 {
     if (parent.isValid()) {
         return 0;
     }
     return m_modelEntries.size();
 }
 
 int BranchesDialogModel::columnCount(const QModelIndex &parent) const
 {
     Q_UNUSED(parent);
     return 1;
 }
 
 QVariant BranchesDialogModel::data(const QModelIndex &idx, int role) const
 {
     if (!idx.isValid()) {
         return {};
     }
 
     const Branch &branch = m_modelEntries.at(idx.row());
     if (role == Qt::DisplayRole) {
         return branch.name;
     } else if (role == Role::FuzzyScore) {
         return branch.score;
     } else if (role == Qt::DecorationRole) {
         if (branch.itemType == BranchItem) {
             static const auto branchIcon = QIcon::fromTheme(QStringLiteral("vcs-branch"));
             return branchIcon;
         }
     } else if (role == Qt::FontRole) {
         if (branch.itemType == CreateBranch || branch.itemType == CreateBranchFrom) {
             QFont font;
             font.setBold(true);
             return font;
         }
     } else if (role == Role::CheckoutName) {
         return branch.refType == GitUtils::RefType::Remote ? branch.name.mid(branch.remote.size() + 1) : branch.name;
     } else if (role == Role::RefType) {
         return branch.refType;
     } else if (role == Role::ItemTypeRole) {
         return branch.itemType;
     }
 
     return {};
 }
 
 void BranchesDialogModel::refresh(const QList<GitUtils::Branch> &branches, bool checkingOut)
 {
     QList<Branch> temp;
     if (checkingOut) {
         Branch create{branches.at(0).name, {}, {}, 0, ItemType::CreateBranch};
         Branch createFrom{branches.at(1).name, {}, {}, 0, ItemType::CreateBranchFrom};
         temp.push_back(create);
         temp.push_back(createFrom);
     }
 
     int i = checkingOut ? 2 : 0;
     for (; i < branches.size(); ++i) {
         temp.append({branches.at(i).name, branches.at(i).remote, branches.at(i).type, -1, ItemType::BranchItem});
     }
 
     beginResetModel();
     m_modelEntries = std::move(temp);
     endResetModel();
 }
 
 void BranchesDialogModel::clear()
 {
     beginResetModel();
     QList<Branch>().swap(m_modelEntries);
     endResetModel();
 }
 
 void BranchesDialogModel::clearBranchCreationItems()
 {
     beginRemoveRows(QModelIndex(), 0, 1);
     m_modelEntries.removeFirst();
     m_modelEntries.removeFirst();
     endRemoveRows();
 }
 
 #include "moc_branchesdialogmodel.cpp"