File indexing completed on 2025-03-09 04:54:39

 /*
    SPDX-FileCopyrightText: 2014-2024 Laurent Montel <montel@kde.org>
 
    SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "mimeparttreeview.h"
 #include "mimetreemodel.h"
 #include "settings/messageviewersettings.h"
 
 #include <KMime/Content>
 
 #include <KConfigGroup>
 #include <QHeaderView>
 
 using namespace MessageViewer;
 namespace
 {
 static const char myMimePartTreeViewConfigGroupName[] = "MimePartTree";
 }
 MimePartTreeView::MimePartTreeView(QWidget *parent)
     : QTreeView(parent)
     , mMimePartModel(new MimeTreeModel(this))
 {
     setObjectName(QLatin1StringView("mMimePartTree"));
 
     setModel(mMimePartModel);
     setSelectionMode(QAbstractItemView::ExtendedSelection);
     setSelectionBehavior(QAbstractItemView::SelectRows);
     setContextMenuPolicy(Qt::CustomContextMenu);
     header()->setSectionResizeMode(QHeaderView::ResizeToContents);
     connect(mMimePartModel, &MimeTreeModel::modelReset, this, &MimePartTreeView::expandAll);
     restoreMimePartTreeConfig();
     setDragEnabled(true);
 }
 
 MimePartTreeView::~MimePartTreeView()
 {
     // root is either null or a modified tree that we need to clean up
     clearModel();
     saveMimePartTreeConfig();
 }
 
 MimeTreeModel *MimePartTreeView::mimePartModel() const
 {
     return mMimePartModel;
 }
 
 void MimePartTreeView::restoreMimePartTreeConfig()
 {
     KConfigGroup grp(KSharedConfig::openStateConfig(), QLatin1StringView(myMimePartTreeViewConfigGroupName));
     header()->restoreState(grp.readEntry("State", QByteArray()));
 }
 
 void MimePartTreeView::saveMimePartTreeConfig()
 {
     KConfigGroup grp(KSharedConfig::openStateConfig(), QLatin1StringView(myMimePartTreeViewConfigGroupName));
     grp.writeEntry("State", header()->saveState());
 }
 
 void MimePartTreeView::clearModel()
 {
     delete mMimePartModel->root();
     mMimePartModel->setRoot(nullptr);
 }
 
 void MimePartTreeView::setRoot(KMime::Content *root)
 {
     delete mMimePartModel->root();
     mMimePartModel->setRoot(root);
 }
 
 KMime::Content::List MimePartTreeView::selectedContents() const
 {
     KMime::Content::List contents;
     QItemSelectionModel *selectModel = selectionModel();
     const QModelIndexList selectedRows = selectModel->selectedRows();
     contents.reserve(selectedRows.count());
     for (const QModelIndex &index : selectedRows) {
         auto content = static_cast<KMime::Content *>(index.internalPointer());
         if (content) {
             contents.append(content);
         }
     }
     return contents;
 }
 
 #include "moc_mimeparttreeview.cpp"