File indexing completed on 2024-05-12 15:59:58

 /*
  *    This file is part of the KDE project
  *    SPDX-FileCopyrightText: 2002 Patrick Julien <freak@codepimps.org>
  *    SPDX-FileCopyrightText: 2007 Jan Hambrecht <jaham@gmx.net>
  *    SPDX-FileCopyrightText: 2007 Sven Langkamp <sven.langkamp@gmail.com>
  *    SPDX-FileCopyrightText: 2011 Srikanth Tiyyagura <srikanth.tulasiram@gmail.com>
  *    SPDX-FileCopyrightText: 2011 José Luis Vergara <pentalis@gmail.com>
  *    SPDX-FileCopyrightText: 2013 Sascha Suelzer <s.suelzer@gmail.com>
  *    SPDX-FileCopyrightText: 2020 Agata Cacko <cacko.azh@gmail.com>
  *
  *    SPDX-License-Identifier: LGPL-2.0-or-later
  */
 
 #include "KisTagChooserWidget.h"
 
 #include <QDebug>
 #include <QToolButton>
 #include <QGridLayout>
 #include <QComboBox>
 #include <QMessageBox>
 
 #include <kconfig.h>
 #include <kconfiggroup.h>
 #include <ksharedconfig.h>
 #include <klocalizedstring.h>
 #include <KisSqueezedComboBox.h>
 
 #include <KoIcon.h>
 
 #include "KisResourceItemChooserContextMenu.h"
 #include "KisTagToolButton.h"
 #include "kis_debug.h"
 #include <KisTagResourceModel.h>
 
 class Q_DECL_HIDDEN KisTagChooserWidget::Private
 {
 public:
     QComboBox *comboBox;
     KisTagToolButton *tagToolButton;
     KisTagModel *model;
     KisTagSP cachedTag;
     QString resourceType;
     QScopedPointer<KisTagModel> allTagsModel;
 };
 
 KisTagChooserWidget::KisTagChooserWidget(KisTagModel *model, QString resourceType, QWidget* parent)
     : QWidget(parent)
     , d(new Private)
 {
     d->resourceType = resourceType;
 
     d->comboBox = new QComboBox(this);
 
     d->comboBox->setToolTip(i18n("Tag"));
     d->comboBox->setSizePolicy(QSizePolicy::MinimumExpanding , QSizePolicy::Fixed);
     d->comboBox->setInsertPolicy(QComboBox::InsertAlphabetically);
     model->sort(KisAllTagsModel::Name);
     d->comboBox->setModel(model);
 
     d->model = model;
     d->allTagsModel.reset(new KisTagModel(resourceType));
     d->allTagsModel->setTagFilter(KisTagModel::ShowAllTags);
 
     QGridLayout* comboLayout = new QGridLayout(this);
 
     comboLayout->addWidget(d->comboBox, 0, 0);
 
     d->tagToolButton = new KisTagToolButton(this);
     d->tagToolButton->setToolTip(i18n("Tag options"));
     comboLayout->addWidget(d->tagToolButton, 0, 1);
 
     comboLayout->setSpacing(0);
     comboLayout->setMargin(0);
     comboLayout->setColumnStretch(0, 3);
     this->setEnabled(true);
 
     connect(d->comboBox, SIGNAL(currentIndexChanged(int)),
             this, SLOT(tagChanged(int)));
 
     connect(d->tagToolButton, SIGNAL(popupMenuAboutToShow()),
             this, SLOT (tagToolContextMenuAboutToShow()));
 
     connect(d->tagToolButton, SIGNAL(newTagRequested(QString)),
             this, SLOT(addTag(QString)));
 
     connect(d->tagToolButton, SIGNAL(deletionOfCurrentTagRequested()),
             this, SLOT(tagToolDeleteCurrentTag()));
 
     connect(d->tagToolButton, SIGNAL(renamingOfCurrentTagRequested(const QString&)),
             this, SLOT(tagToolRenameCurrentTag(const QString&)));
 
     connect(d->tagToolButton, SIGNAL(undeletionOfTagRequested(KisTagSP)),
             this, SLOT(tagToolUndeleteLastTag(KisTagSP)));
 
 
     // Workaround for handling tag selection deselection when model resets.
     // Occurs when model changes under the user e.g. +/- a resource storage.
     connect(d->model, SIGNAL(modelAboutToBeReset()), this, SLOT(cacheSelectedTag()));
     connect(d->model, SIGNAL(modelReset()), this, SLOT(restoreTagFromCache()));
     connect(d->allTagsModel.data(), SIGNAL(dataChanged(const QModelIndex&, const QModelIndex&, const QVector<int>&)),
             this, SLOT(slotTagModelDataChanged(const QModelIndex&, const QModelIndex&, const QVector<int>&)));
 
 }
 
 KisTagChooserWidget::~KisTagChooserWidget()
 {
     delete d;
 }
 
 void KisTagChooserWidget::tagToolDeleteCurrentTag()
 {
     KisTagSP currentTag = currentlySelectedTag();
     if (!currentTag.isNull() && currentTag->id() >= 0) {
         d->model->setTagInactive(currentTag);
         setCurrentIndex(0);
         d->model->sort(KisAllTagsModel::Name);
     }
 }
 
 void KisTagChooserWidget::tagChanged(int tagIndex)
 {
     if (tagIndex >= 0) {
         KisTagSP tag = currentlySelectedTag();
         d->tagToolButton->setCurrentTag(tag);
         KConfigGroup group =  KSharedConfig::openConfig()->group("SelectedTags");
         group.writeEntry(d->resourceType, currentlySelectedTag()->url());
         d->model->sort(KisAllTagsModel::Name);
         emit sigTagChosen(tag);
     } else {
         setCurrentIndex(0);
     }
 }
 
 void KisTagChooserWidget::tagToolRenameCurrentTag(const QString& tagName)
 {
     KisTagSP tag = currentlySelectedTag();
     bool canRenameCurrentTag = !tag.isNull() && (tagName != tag->name());
 
     if (tagName == KisAllTagsModel::urlAll() || tagName == KisAllTagsModel::urlAllUntagged()) {
         QMessageBox::information(this, i18nc("Dialog title", "Can't rename the tag"), i18nc("Dialog message", "You can't use this name for your custom tags."), QMessageBox::Ok);
         return;
     }
 
     bool result = false;
 
     if (canRenameCurrentTag && !tagName.isEmpty()) {
          result = d->model->renameTag(tag, tagName, false);
 
         if (!result) {
             KisTagSP tagToRemove = d->model->tagForUrl(tagName);
 
             if (tagToRemove &&
                 QMessageBox::question(this, i18nc("Dialog title", "Remove existing tag with that name?"),
                 i18nc("Dialog message (the arguments are both somewhat user readable nouns or adjectives (names of the tags), can be treated as nouns since they represent the tags)",
                 "A tag with this unique name already exists. In order to continue renaming, the existing tag needs to be removed. Do you want to continue?\n"
                 "Tag to be removed: %1\n"
                 "Tag's unique name: %2", tagToRemove->name(), tagToRemove->url()), QMessageBox::Yes | QMessageBox::Cancel, QMessageBox::Cancel) != QMessageBox::Cancel) {
                 result = d->model->renameTag(tag, tagName, true);
                 KIS_SAFE_ASSERT_RECOVER_RETURN(result);
             }
         }
     }
 
     if (result) {
         KisTagSP renamedTag = d->model->tagForUrl(tagName);
         KIS_SAFE_ASSERT_RECOVER_RETURN(renamedTag);
         const QModelIndex idx = d->model->indexForTag(renamedTag);
         setCurrentIndex(idx.row());
     }
 }
 
 void KisTagChooserWidget::tagToolUndeleteLastTag(KisTagSP tag)
 {
     int previousIndex = d->comboBox->currentIndex();
 
     bool success = d->model->setTagActive(tag);
     setCurrentIndex(previousIndex);
     if (success) {
         setCurrentItem(tag->name());
         d->model->sort(KisAllTagsModel::Name);
     }
 }
 
 void KisTagChooserWidget::cacheSelectedTag()
 {
     d->cachedTag = currentlySelectedTag();
 }
 
 void KisTagChooserWidget::restoreTagFromCache()
 {
     if (d->cachedTag) {
         QModelIndex cachedIndex = d->model->indexForTag(d->cachedTag);
         setCurrentIndex(cachedIndex.row());
         d->cachedTag = nullptr;
     }
 }
 
 void KisTagChooserWidget::slotTagModelDataChanged(const QModelIndex &topLeft, const QModelIndex &bottomRight, const QVector<int> roles)
 {
     // we care only about the check status
     if (!roles.isEmpty() && !roles.contains(Qt::CheckStateRole)) {
         return;
     }
 
     const QModelIndex currIdx =
             d->allTagsModel->indexForTag(d->tagToolButton->undeletionCandidate());
 
     if (currIdx.isValid() &&
         currIdx.row() >= topLeft.row() && currIdx.row() <= bottomRight.row() &&
         currIdx.column() >= topLeft.column() && currIdx.column() <= bottomRight.column()) {
 
         const bool isNowActive = d->allTagsModel->data(currIdx, Qt::CheckStateRole).toBool();
 
         if (isNowActive) {
             d->tagToolButton->setUndeletionCandidate(KisTagSP());
         }
     }
 
     for (int row = topLeft.row(); row <= bottomRight.row(); row++) {
         for (int column = topLeft.column(); column <= bottomRight.column(); column++) {
             const QModelIndex idx = d->allTagsModel->index(row, column);
 
             const bool isActive = d->allTagsModel->data(idx, Qt::CheckStateRole).toBool();
 
             if (idx != currIdx && !isActive) {
                 d->tagToolButton->setUndeletionCandidate(d->allTagsModel->tagForIndex(idx));
                 break;
             }
         }
     }
 }
 
 void KisTagChooserWidget::setCurrentIndex(int index)
 {
     d->comboBox->setCurrentIndex(index);
 }
 
 int KisTagChooserWidget::currentIndex() const
 {
     return d->comboBox->currentIndex();
 }
 
 void KisTagChooserWidget::setCurrentItem(const QString &tag)
 {
     for (int i = 0; i < d->model->rowCount(); i++) {
         QModelIndex index = d->model->index(i, 0);
         QString currentRowTag = d->model->data(index, Qt::UserRole + KisAllTagsModel::Url).toString();
         if (currentRowTag == tag) {
             setCurrentIndex(i);
         }
     }
 }
 
 void KisTagChooserWidget::addTag(const QString &tag)
 {
     addTag(tag, 0);
 }
 
 KisTagChooserWidget::OverwriteDialogOptions KisTagChooserWidget::overwriteTagDialog(KisTagChooserWidget* parent, bool tagIsActive)
 {
     QString undeleteOption = !tagIsActive ? i18nc("Option in a dialog to undelete (reactivate) existing tag with its old assigned resources", "Restore previous tag")
                                       : i18nc("Option in a dialog to use existing tag with its old assigned resources", "Use existing tag");
     // if you use this simple cast, the order of buttons must match order of options in the enum
     return (KisTagChooserWidget::OverwriteDialogOptions)QMessageBox::question(parent, i18nc("Dialog title", "Overwrite tag?"), i18nc("Question to the user in a dialog about creating a tag",
                                                                                       "A tag with this unique name already exists. Do you want to replace it?"),
                                        i18nc("Option in a dialog to discard the previously existing tag and creating a new one in its place", "Replace (overwrite) tag"),
                                        undeleteOption, i18n("Cancel"));
 }
 
 void KisTagChooserWidget::addTag(const QString &tagName, KoResourceSP resource)
 {
     if (tagName == KisAllTagsModel::urlAll() || tagName == KisAllTagsModel::urlAllUntagged()) {
         QMessageBox::information(this, i18nc("Dialog title", "Can't create the tag"), i18nc("Dialog message", "You can't use this name for your custom tags."), QMessageBox::Ok);
         return;
     }
 
     if (tagName.isEmpty()) return;
 
     KisTagSP tagForUrl = d->model->tagForUrl(tagName);
     if (!tagForUrl.isNull()) {
         int response = overwriteTagDialog(this, tagForUrl->active());
         if (response == Undelete) { // Undelete
             d->model->setTagActive(tagForUrl);
             if (!resource.isNull()) {
                 KisTagResourceModel(d->resourceType).tagResources(tagForUrl, QVector<int>() << resource->resourceId());
             }
             d->model->sort(KisAllTagsModel::Name);
             return;
         } else if (response == Cancel) { // Cancel
             return;
         }
     }
     QVector<KoResourceSP> resources = (resource.isNull() ? QVector<KoResourceSP>() : (QVector<KoResourceSP>() << resource));
     d->model->addTag(tagName, true, resources); // this will overwrite the tag
     d->model->sort(KisAllTagsModel::Name);
 }
 
 void KisTagChooserWidget::addTag(KisTagSP tag, KoResourceSP resource)
 {
     if (tag->name() == KisAllTagsModel::urlAll() || tag->name() == KisAllTagsModel::urlAllUntagged()) {
         QMessageBox::information(this, i18nc("Dialog title", "Can't rename the tag"), i18nc("Dialog message", "You can't use this name for your custom tags."), QMessageBox::Ok);
         return;
     }
 
     KisTagSP tagForUrl = d->model->tagForUrl(tag->url());
     if (!tagForUrl.isNull()) {
         int response = overwriteTagDialog(this, tagForUrl->active());
         if (response == Undelete) { // Undelete
             d->model->setTagActive(tagForUrl);
             if (!resource.isNull()) {
                 KisTagResourceModel(d->resourceType).tagResources(tagForUrl, QVector<int>() << resource->resourceId());
             }
             d->model->sort(KisAllTagsModel::Name);
             return;
         } else if (response == Cancel) { // Cancel
             return;
         }
     }
     QVector<KoResourceSP> resources = (resource.isNull() ? QVector<KoResourceSP>() : (QVector<KoResourceSP>() << resource));
     d->model->addTag(tag, true, resources); // this will overwrite the tag
     d->model->sort(KisAllTagsModel::Name);
 }
 
 KisTagSP KisTagChooserWidget::currentlySelectedTag()
 {
     int row = d->comboBox->currentIndex();
     if (row < 0) {
         return nullptr;
     }
 
     QModelIndex index = d->model->index(row, 0);
     KisTagSP tag =  d->model->tagForIndex(index);
     return tag;
 }
 
 void KisTagChooserWidget::updateIcons()
 {
     d->tagToolButton->loadIcon();
 }
 
 void KisTagChooserWidget::tagToolContextMenuAboutToShow()
 {
     /* only enable the save button if the selected tag set is editable */
     if (currentlySelectedTag()) {
         d->tagToolButton->readOnlyMode(currentlySelectedTag()->id() < 0);
     }
     else {
         d->tagToolButton->readOnlyMode(true);
     }
 }