File indexing completed on 2024-11-24 04:16:55

 /*
   SPDX-FileCopyrightText: 2023-2024 Laurent Montel <montel.org>
 
   SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "translatormodel.h"
 #include "libbergamot_debug.h"
 
 #include <KLocalizedString>
 #include <QColor>
 
 TranslatorModel::TranslatorModel(QObject *parent)
     : QAbstractListModel{parent}
 {
     updateInstalledLanguage();
 }
 
 TranslatorModel::~TranslatorModel() = default;
 
 void TranslatorModel::insertTranslators(const QVector<Translator> &translators)
 {
     if (rowCount() != 0) {
         beginResetModel();
         mTranslators.clear();
         endResetModel();
     }
     if (!translators.isEmpty()) {
         beginInsertRows(QModelIndex(), 0, translators.count() - 1);
         mTranslators = translators;
         endInsertRows();
     }
     updateInstalledLanguage();
 }
 
 void TranslatorModel::updateInstalledLanguage()
 {
     mLanguageInstalled = BergamotEngineUtils::languageLocallyStored();
     qCDebug(TRANSLATOR_LIBBERGAMOT_LOG) << "mLanguageInstalled " << mLanguageInstalled;
 }
 
 QVariant TranslatorModel::headerData(int section, Qt::Orientation orientation, int role) const
 {
     if (role == Qt::DisplayRole && orientation == Qt::Horizontal) {
         switch (static_cast<TranslatorRoles>(section)) {
         case TranslatorRoles::Source:
             return i18n("Source");
         case TranslatorModel::Target:
             return i18n("Target");
         case TranslatorModel::TypeTranslator:
             return i18n("Type");
         case TranslatorModel::Repository:
             return i18n("Repository");
         case TranslatorModel::InstalledVersion:
             return i18n("Installed Version");
         case TranslatorModel::AvailableVersion:
             return i18n("Available Version");
         case TranslatorModel::Installed:
             return i18n("Installed");
         case TranslatorModel::Identifier:
         case TranslatorModel::CheckSum:
         case TranslatorModel::NeedToUpdateLanguage:
         case TranslatorModel::Url:
             return {};
         }
     }
     return {};
 }
 
 int TranslatorModel::rowCount(const QModelIndex &parent) const
 {
     Q_UNUSED(parent)
     return mTranslators.count();
 }
 
 bool TranslatorModel::needToUpdateLanguageModel(const Translator &translator) const
 {
     const QString shortName{translator.shortName()};
     if (isInstalled(shortName)) {
         if (versionInstalled(shortName) < translator.version()) {
             return true;
         }
     }
     return false;
 }
 
 QVariant TranslatorModel::data(const QModelIndex &index, int role) const
 {
     if (index.row() < 0 || index.row() >= mTranslators.count()) {
         return {};
     }
     const auto translator = mTranslators.at(index.row());
     const int col = index.column();
     if (role == Qt::BackgroundRole) {
         if (needToUpdateLanguageModel(translator)) {
             if (static_cast<TranslatorRoles>(col) == TranslatorModel::InstalledVersion) {
                 return QColor(Qt::red);
             }
         }
         return {};
     } else if (role == Qt::DisplayRole) {
         switch (static_cast<TranslatorRoles>(col)) {
         case TranslatorModel::Source: {
             return translator.source();
         }
         case TranslatorModel::Target: {
             return translator.target();
         }
         case TranslatorModel::TypeTranslator: {
             return translator.type();
         }
         case TranslatorModel::Repository: {
             return translator.repository();
         }
         case TranslatorModel::InstalledVersion: {
             const QString shortName{translator.shortName()};
             if (isInstalled(shortName)) {
                 return versionInstalled(shortName);
             }
             return {};
         }
         case TranslatorModel::AvailableVersion: {
             return translator.version();
         }
         case TranslatorModel::Installed: {
             if (isInstalled(translator.shortName())) {
                 return i18n("Installed");
             }
             return {};
         }
         case TranslatorModel::Identifier: {
             return translator.shortName();
         }
         case TranslatorModel::Url: {
             return translator.url();
         }
         case TranslatorModel::CheckSum: {
             return translator.checkSum();
         }
         case TranslatorModel::NeedToUpdateLanguage: {
             return needToUpdateLanguageModel(translator);
         }
         }
     }
     return {};
 }
 
 bool TranslatorModel::isInstalled(const QString &shortName) const
 {
     for (const auto &lang : std::as_const(mLanguageInstalled)) {
         if (lang.shortName == shortName) {
             return true;
         }
     }
     return false;
 }
 
 int TranslatorModel::versionInstalled(const QString &shortName) const
 {
     for (const auto &lang : std::as_const(mLanguageInstalled)) {
         if (lang.shortName == shortName) {
             return lang.version;
         }
     }
     return false;
 }
 
 void TranslatorModel::clear()
 {
     if (rowCount() != 0) {
         beginResetModel();
         mTranslators.clear();
         endResetModel();
     }
     updateInstalledLanguage();
 }
 
 int TranslatorModel::columnCount(const QModelIndex &parent) const
 {
     Q_UNUSED(parent)
     return static_cast<int>(TranslatorRoles::LastColumn) + 1;
 }
 
 void TranslatorModel::removeLanguage(const QString &identifier)
 {
     auto index = std::find_if(mLanguageInstalled.begin(), mLanguageInstalled.end(), [identifier](const BergamotEngineUtils::LanguageInstalled &installed) {
         return (identifier == installed.shortName);
     });
     if (index != mLanguageInstalled.end()) {
         const QString absoluteLanguageModelPath = (*index).absoluteLanguageModelPath;
         if (!QDir(absoluteLanguageModelPath).removeRecursively()) {
             qCDebug(TRANSLATOR_LIBBERGAMOT_LOG) << "Impossible to delete " << absoluteLanguageModelPath;
             return;
         }
         mLanguageInstalled.removeAll(*index);
         beginResetModel();
         endResetModel();
     }
     updateInstalledLanguage();
 }
 
 #include "moc_translatormodel.cpp"