File indexing completed on 2024-05-12 16:25:52

 /*
    SPDX-FileCopyrightText: 2017-2024 Laurent Montel <montel@kde.org>
 
    SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "statusmodel.h"
 #include "customusers/customuserstatus.h"
 #include "utils.h"
 #include <KLocalizedString>
 
 StatusModel::StatusModel(QObject *parent)
     : QAbstractListModel(parent)
 {
     fillModel();
 }
 
 StatusModel::~StatusModel() = default;
 
 User::PresenceStatus StatusModel::currentUserStatus() const
 {
     return mStatusList.at(mCurrentStatus).status;
 }
 
 StatusModel::StatusInfo StatusModel::currentStatusInfo() const
 {
     const DisplayStatusInfo info = mStatusList.at(mCurrentStatus);
     StatusInfo currentStatusInfo;
     currentStatusInfo.status = info.status;
     currentStatusInfo.statusStr = info.statusStr;
     if (!mCustomText.isEmpty()) {
         currentStatusInfo.statusStr = mCustomText;
     }
     return currentStatusInfo;
 }
 
 User::PresenceStatus StatusModel::status(int index) const
 {
     return mStatusList.at(index).status;
 }
 
 void StatusModel::setCurrentPresenceStatus(User::PresenceStatus status)
 {
     int newStatusIndex = 0;
     for (int i = 0, total = mStatusList.count(); i < total; ++i) {
         if (mStatusList.at(i).status == status) {
             newStatusIndex = i;
             break;
         }
     }
     if (mCurrentStatus != newStatusIndex) {
         mCurrentStatus = newStatusIndex;
         Q_EMIT currentStatusChanged();
     }
 }
 
 QString StatusModel::textFromPresenceStatus(User::PresenceStatus status) const
 {
     switch (status) {
     case User::PresenceStatus::PresenceOnline:
         return i18n("Online");
     case User::PresenceStatus::PresenceBusy:
         return i18n("Busy");
     case User::PresenceStatus::PresenceAway:
         return i18n("Away");
     case User::PresenceStatus::PresenceOffline:
         return i18n("Offline");
     case User::PresenceStatus::Unknown:
         return i18n("Modify Status...");
     }
     return {};
 }
 
 QIcon StatusModel::iconFromPresenceStatus(User::PresenceStatus status) const
 {
     switch (status) {
     case User::PresenceStatus::PresenceOnline:
         return QIcon::fromTheme(QStringLiteral("user-online"));
     case User::PresenceStatus::PresenceBusy:
         return QIcon::fromTheme(QStringLiteral("user-busy"));
     case User::PresenceStatus::PresenceAway:
         return QIcon::fromTheme(QStringLiteral("user-away"));
     case User::PresenceStatus::PresenceOffline:
         return QIcon::fromTheme(QStringLiteral("user-offline"));
     case User::PresenceStatus::Unknown:
         return {};
     }
     return {};
 }
 
 StatusModel::DisplayStatusInfo StatusModel::createStatusInfo(User::PresenceStatus status, int order)
 {
     DisplayStatusInfo statusInfo;
     statusInfo.status = status;
     statusInfo.displayText = textFromPresenceStatus(statusInfo.status);
     statusInfo.icon = iconFromPresenceStatus(statusInfo.status);
     statusInfo.order = order;
     statusInfo.statusStr = Utils::presenceStatusToString(status);
     return statusInfo;
 }
 
 void StatusModel::fillModel()
 {
     mStatusList.clear();
     {
         const DisplayStatusInfo statusInfo = createStatusInfo(User::PresenceStatus::PresenceOnline, 20);
         mStatusList.append(std::move(statusInfo));
     }
     {
         const DisplayStatusInfo statusInfo = createStatusInfo(User::PresenceStatus::PresenceBusy, 19);
         mStatusList.append(std::move(statusInfo));
     }
     {
         const DisplayStatusInfo statusInfo = createStatusInfo(User::PresenceStatus::PresenceAway, 18);
         mStatusList.append(std::move(statusInfo));
     }
     {
         const DisplayStatusInfo statusInfo = createStatusInfo(User::PresenceStatus::PresenceOffline, 17);
         mStatusList.append(std::move(statusInfo));
     }
     {
         const DisplayStatusInfo statusInfo = createStatusInfo(User::PresenceStatus::Unknown, -1);
         mStatusList.append(std::move(statusInfo));
     }
 }
 
 QString StatusModel::customText() const
 {
     return mCustomText;
 }
 
 void StatusModel::setCustomText(const QString &customText)
 {
     mCustomText = customText;
 }
 
 void StatusModel::updateCustomStatus(const QVector<CustomUserStatus> &customUserStatuses)
 {
     beginResetModel();
     fillModel();
     for (const CustomUserStatus &status : customUserStatuses) {
         DisplayStatusInfo statusInfo;
         statusInfo.status = status.statusType();
         statusInfo.displayText = status.name();
         statusInfo.icon = iconFromPresenceStatus(statusInfo.status);
         statusInfo.order = 5;
         statusInfo.statusStr = status.name();
         mStatusList.append(statusInfo);
     }
     endResetModel();
 }
 
 int StatusModel::currentStatus() const
 {
     return mCurrentStatus;
 }
 
 int StatusModel::rowCount(const QModelIndex &parent) const
 {
     Q_UNUSED(parent)
     return mStatusList.count();
 }
 
 QVariant StatusModel::data(const QModelIndex &index, int role) const
 {
     if (index.row() < 0 || index.row() >= mStatusList.count()) {
         return {};
     }
     const DisplayStatusInfo statusInfo = mStatusList.at(index.row());
     switch (role) {
     case Qt::DisplayRole:
     case StatusI18n:
         return statusInfo.displayText;
     case Status:
         return QVariant::fromValue(statusInfo.status);
     case Order:
         return statusInfo.order;
     case Qt::DecorationRole:
     case Icon:
         return statusInfo.icon;
     }
 
     return {};
 }
 
 #include "moc_statusmodel.cpp"