File indexing completed on 2024-05-12 04:58:24

 /* ============================================================
 * Falkon - Qt web browser
 * Copyright (C) 2018 David Rosca <nowrep@gmail.com>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 * ============================================================ */
 #include "tabmodel.h"
 #include "webtab.h"
 #include "tabwidget.h"
 #include "browserwindow.h"
 
 // TabModelMimeData
 TabModelMimeData::TabModelMimeData()
     : QMimeData()
 {
 }
 
 WebTab *TabModelMimeData::tab() const
 {
     return m_tab;
 }
 
 void TabModelMimeData::setTab(WebTab *tab)
 {
     m_tab = tab;
 }
 
 bool TabModelMimeData::hasFormat(const QString &format) const
 {
     return mimeType() == format;
 }
 
 QStringList TabModelMimeData::formats() const
 {
     return {mimeType()};
 }
 
 // static
 QString TabModelMimeData::mimeType()
 {
     return QSL("application/falkon.tabmodel.tab");
 }
 
 // TabModel
 TabModel::TabModel(BrowserWindow *window, QObject *parent)
     : QAbstractListModel(parent)
     , m_window(window)
 {
     init();
 }
 
 QModelIndex TabModel::tabIndex(WebTab *tab) const
 {
     const int idx = m_tabs.indexOf(tab);
     if (idx < 0) {
         return {};
     }
     return index(idx);
 }
 
 WebTab *TabModel::tab(const QModelIndex &index) const
 {
     return m_tabs.value(index.row());
 }
 
 int TabModel::rowCount(const QModelIndex &parent) const
 {
     if (parent.isValid()) {
         return 0;
     }
     return m_tabs.count();
 }
 
 Qt::ItemFlags TabModel::flags(const QModelIndex &index) const
 {
     if (!index.isValid()) {
         return Qt::ItemIsDropEnabled;
     }
     return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled;
 }
 
 QVariant TabModel::data(const QModelIndex &index, int role) const
 {
     if (index.row() < 0 || index.row() > m_tabs.count()) {
         return {};
     }
 
     WebTab *t = tab(index);
     if (!t) {
         return {};
     }
 
     switch (role) {
     case WebTabRole:
         return QVariant::fromValue(t);
 
     case TitleRole:
     case Qt::DisplayRole:
         return t->title();
 
     case IconRole:
     case Qt::DecorationRole:
         return t->icon();
 
     case PinnedRole:
         return t->isPinned();
 
     case RestoredRole:
         return t->isRestored();
 
     case CurrentTabRole:
         return t->isCurrentTab();
 
     case LoadingRole:
         return t->isLoading();
 
     case AudioPlayingRole:
         return t->isPlaying();
 
     case AudioMutedRole:
         return t->isMuted();
 
     case BackgroundActivityRole:
         return t->backgroundActivity();
 
     default:
         return {};
     }
 }
 
 Qt::DropActions TabModel::supportedDropActions() const
 {
     return Qt::MoveAction;
 }
 
 QStringList TabModel::mimeTypes() const
 {
     return {TabModelMimeData::mimeType()};
 }
 
 QMimeData *TabModel::mimeData(const QModelIndexList &indexes) const
 {
     if (indexes.isEmpty()) {
         return nullptr;
     }
     auto *tab = indexes.at(0).data(WebTabRole).value<WebTab*>();
     if (!tab) {
         return nullptr;
     }
     auto *mimeData = new TabModelMimeData;
     mimeData->setTab(tab);
     return mimeData;
 }
 
 bool TabModel::canDropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) const
 {
     Q_UNUSED(row)
     if (action != Qt::MoveAction || parent.isValid() || column > 0 || !m_window) {
         return false;
     }
     const auto *mimeData = qobject_cast<const TabModelMimeData*>(data);
     if (!mimeData) {
         return false;
     }
     return mimeData->tab();
 }
 
 bool TabModel::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent)
 {
     if (!canDropMimeData(data, action, row, column, parent)) {
         return false;
     }
 
     const auto *mimeData = static_cast<const TabModelMimeData*>(data);
     WebTab *tab = mimeData->tab();
 
     if (tab->browserWindow() == m_window) {
         if (row < 0) {
             row = tab->isPinned() ? m_window->tabWidget()->pinnedTabsCount() : m_window->tabWidget()->count();
         }
         tab->moveTab(row > mimeData->tab()->tabIndex() ? row - 1 : row);
     } else {
         if (row < 0) {
             row = m_window->tabCount();
         }
         if (tab->browserWindow()) {
             tab->browserWindow()->tabWidget()->detachTab(tab);
         }
         tab->setPinned(row < m_window->tabWidget()->pinnedTabsCount());
         m_window->tabWidget()->insertView(row, tab, Qz::NT_SelectedTab);
     }
 
     return true;
 }
 
 void TabModel::init()
 {
     for (int i = 0; i < m_window->tabCount(); ++i) {
         tabInserted(i);
     }
 
     connect(m_window->tabWidget(), &TabWidget::tabInserted, this, &TabModel::tabInserted);
     connect(m_window->tabWidget(), &TabWidget::tabRemoved, this, &TabModel::tabRemoved);
     connect(m_window->tabWidget(), &TabWidget::tabMoved, this, &TabModel::tabMoved);
 
     connect(m_window, &QObject::destroyed, this, [this]() {
         beginResetModel();
         m_window = nullptr;
         m_tabs.clear();
         endResetModel();
     });
 }
 
 void TabModel::tabInserted(int index)
 {
     WebTab *tab = m_window->tabWidget()->webTab(index);
 
     beginInsertRows(QModelIndex(), index, index);
     m_tabs.insert(index, tab);
     endInsertRows();
 
     auto emitDataChanged = [this](WebTab *tab, int role) {
         const QModelIndex idx = tabIndex(tab);
         Q_EMIT dataChanged(idx, idx, {role});
     };
 
     connect(tab, &WebTab::titleChanged, this, std::bind(emitDataChanged, tab, Qt::DisplayRole));
     connect(tab, &WebTab::titleChanged, this, std::bind(emitDataChanged, tab, TitleRole));
     connect(tab, &WebTab::iconChanged, this, std::bind(emitDataChanged, tab, Qt::DecorationRole));
     connect(tab, &WebTab::iconChanged, this, std::bind(emitDataChanged, tab, IconRole));
     connect(tab, &WebTab::pinnedChanged, this, std::bind(emitDataChanged, tab, PinnedRole));
     connect(tab, &WebTab::restoredChanged, this, std::bind(emitDataChanged, tab, RestoredRole));
     connect(tab, &WebTab::currentTabChanged, this, std::bind(emitDataChanged, tab, CurrentTabRole));
     connect(tab, &WebTab::loadingChanged, this, std::bind(emitDataChanged, tab, LoadingRole));
     connect(tab, &WebTab::playingChanged, this, std::bind(emitDataChanged, tab, AudioPlayingRole));
     connect(tab, &WebTab::mutedChanged, this, std::bind(emitDataChanged, tab, AudioMutedRole));
     connect(tab, &WebTab::backgroundActivityChanged, this, std::bind(emitDataChanged, tab, BackgroundActivityRole));
 }
 
 void TabModel::tabRemoved(int index)
 {
     beginRemoveRows(QModelIndex(), index, index);
     m_tabs.remove(index);
     endRemoveRows();
 }
 
 void TabModel::tabMoved(int from, int to)
 {
     if (!beginMoveRows(QModelIndex(), from, from, QModelIndex(), to > from ? to + 1 : to)) {
         qWarning() << "Invalid beginMoveRows" << from << (to > from ? to + 1 : to);
         return;
     }
     m_tabs.insert(to, m_tabs.takeAt(from));
     endMoveRows();
 }