File indexing completed on 2024-05-12 16:23:40

 // SPDX-FileCopyrightText: 2020 Jonah Brüchert <jbb@kaidan.im>
 // SPDX-FileCopyrightText: 2020 Rinigus <rinigus.git@gmail.com>
 //
 // SPDX-License-Identifier: GPL-2.0-or-later
 
 #include "bookmarkshistorymodel.h"
 #include "browsermanager.h"
 
 #include <QDateTime>
 #include <QDebug>
 #include <QSqlError>
 
 #include <QCoroFuture>
 #include <QCoroTask>
 
 constexpr int QUERY_LIMIT = 1000;
 
 BookmarksHistoryModel::BookmarksHistoryModel(QObject *parent)
     : QAbstractListModel(parent)
 {
     connect(BrowserManager::instance(), &BrowserManager::databaseTableChanged, this, &BookmarksHistoryModel::onDatabaseChanged);
 }
 
 QHash<int, QByteArray> BookmarksHistoryModel::roleNames() const {
     return {
         {Id, "id"},
         {Url, "url"},
         {Title, "title"},
         {Icon, "iconName"},
         {LastVisitedDelta, "lastVisitedDelte"}
     };
 }
 
 QVariant BookmarksHistoryModel::data(const QModelIndex &index, int role) const
 {
     auto &item = m_entries.at(index.row());
     switch (role) {
     case Role::Id:
         return item.id;
     case Role::Title:
         return item.title;
     case Role::Url:
         return item.url;
     case Role::Icon:
         return item.icon;
     case Role::LastVisitedDelta:
         return item.lastVisitedDelta;
     }
 
     Q_UNREACHABLE();
 }
 
 void BookmarksHistoryModel::setActive(bool a)
 {
     if (m_active == a)
         return;
     m_active = a;
     if (m_active)
         fetchData();
     else
         clear();
 
     Q_EMIT activeChanged();
 }
 
 void BookmarksHistoryModel::setBookmarks(bool b)
 {
     if (m_bookmarks == b)
         return;
     m_bookmarks = b;
     fetchData();
     Q_EMIT bookmarksChanged();
 }
 
 void BookmarksHistoryModel::setHistory(bool h)
 {
     if (m_history == h)
         return;
     m_history = h;
     fetchData();
     Q_EMIT historyChanged();
 }
 
 void BookmarksHistoryModel::setFilter(const QString &f)
 {
     if (m_filter == f)
         return;
     m_filter = f;
     fetchData();
     Q_EMIT filterChanged();
 }
 
 void BookmarksHistoryModel::onDatabaseChanged(const QString &table)
 {
     if ((table == QLatin1String("bookmarks") && m_bookmarks) || (table == QLatin1String("history") && m_history))
         fetchData();
 }
 
 void BookmarksHistoryModel::fetchData()
 {
     if (!m_active)
         return;
 
     auto future = [history = m_history, bookmarks = m_bookmarks, filter = m_filter]() mutable -> QCoro::Task<std::vector<BookmarksHistoryRecord>> {
         auto db = BrowserManager::instance()
                 ->databaseManager()
                 ->database();
 
         const qint64 currentTimeInUnix = QDateTime::currentSecsSinceEpoch();
 
         if (filter.isEmpty()) {
             // No clue why this works
             filter = QStringLiteral("");
         }
 
         if (bookmarks && history) {
             co_return co_await db->getResults<BookmarksHistoryRecord>(
                 QStringLiteral("SELECT rowid AS id, url, title, icon, ? - lastVisited AS lastVisitedDelta "
                                "FROM (SELECT * FROM bookmarks UNION SELECT * FROM history) "
                                "WHERE url LIKE '%' || ? || '%' OR title LIKE '%' || ? || '%' "
                                "LIMIT %1").arg(QUERY_LIMIT),
                     currentTimeInUnix, filter, filter);
         } else if (bookmarks) {
             co_return co_await db->getResults<BookmarksHistoryRecord>(
                 QStringLiteral("SELECT rowid AS id, url, title, icon, ? - lastVisited AS lastVisitedDelta "
                                "FROM bookmarks "
                                "WHERE url LIKE '%' || ? || '%' OR title LIKE '%' || ? || '%'"),
                     currentTimeInUnix, filter, filter);
         } else if (history) {
             co_return co_await db->getResults<BookmarksHistoryRecord>(
                 QStringLiteral("SELECT rowid AS id, url, title, icon, ? - lastVisited AS lastVisitedDelta "
                                "FROM history "
                                "WHERE url LIKE '%' || ? || '%' OR title LIKE '%' || ? || '%'"
                                "LIMIT  %1").arg(QUERY_LIMIT),
                     currentTimeInUnix, filter, filter);
         }
 
         co_return {};
     }();
 
     QCoro::connect(std::move(future), this, [this](auto result) {
         if (m_entries == result) {
             return;
         }
 
         beginResetModel();
         m_entries = std::move(result);
         endResetModel();
     });
 }
 
 void BookmarksHistoryModel::clear()
 {
     beginResetModel();
     m_entries.clear();
     endResetModel();
 }
 
 #include "moc_bookmarkshistorymodel.cpp"