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

 /* ============================================================
 * Falkon - Qt web browser
 * Copyright (C) 2015-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 "recoveryjsobject.h"
 #include "mainapplication.h"
 #include "webpage.h"
 #include "tabbedwebview.h"
 #include "browserwindow.h"
 #include "qztools.h"
 #include "iconprovider.h"
 
 #include <QJsonObject>
 
 RecoveryJsObject::RecoveryJsObject(RestoreManager *manager)
     : QObject()
     , m_manager(manager)
     , m_page(nullptr)
 {
 }
 
 void RecoveryJsObject::setPage(WebPage *page)
 {
     Q_ASSERT(page);
 
     m_page = page;
 }
 
 QJsonArray RecoveryJsObject::restoreData() const
 {
     QJsonArray out;
 
     int i = 0;
     const auto windows = m_manager->restoreData().windows;
     for (const BrowserWindow::SavedWindow &w : windows) {
         int j = 0;
         QJsonArray tabs;
         const auto windowTabs = w.tabs;
         for (const WebTab::SavedTab &t : windowTabs) {
             const QIcon icon = t.icon.isNull() ? IconProvider::emptyWebIcon() : t.icon;
             QJsonObject tab;
             tab[QSL("tab")] = j;
             tab[QSL("icon")] = QzTools::pixmapToDataUrl(icon.pixmap(16)).toString();
             tab[QSL("title")] = t.title;
             tab[QSL("url")] = t.url.toString();
             tab[QSL("pinned")] = t.isPinned;
             tab[QSL("current")] = w.currentTab == j;
             tabs.append(tab);
             j++;
         }
 
         QJsonObject window;
         window[QSL("window")] = i++;
         window[QSL("tabs")] = tabs;
         out.append(window);
     }
 
     return out;
 }
 
 void RecoveryJsObject::startNewSession()
 {
     auto *view = qobject_cast<TabbedWebView*>(m_page->view());
     if (!view) {
         return;
     }
 
     if (view->browserWindow()->tabCount() > 1) {
         view->closeView();
     } else {
         auto *oldWindow = view->browserWindow();
         mApp->createWindow(Qz::BW_NewWindow);
         oldWindow->close();
     }
 
     mApp->restoreManager()->clearRestoreData();
     mApp->destroyRestoreManager();
 }
 
 void RecoveryJsObject::restoreSession(const QStringList &excludeWin, const QStringList &excludeTab)
 {
     Q_ASSERT(excludeWin.size() == excludeTab.size());
 
     // This assumes that excludeWin and excludeTab are sorted in descending order
 
     RestoreData data = m_manager->restoreData();
 
     for (int i = 0; i < excludeWin.size(); ++i) {
         int win = excludeWin.at(i).toInt();
         int tab = excludeTab.at(i).toInt();
 
         if (!QzTools::containsIndex(data.windows, win) || !QzTools::containsIndex(data.windows.at(win).tabs, tab))
             continue;
 
         BrowserWindow::SavedWindow &wd = data.windows[win];
 
         wd.tabs.remove(tab);
         if (wd.currentTab >= tab)
             --wd.currentTab;
 
         if (wd.tabs.isEmpty()) {
             data.windows.remove(win);
             continue;
         }
 
         if (wd.currentTab < 0)
             wd.currentTab = wd.tabs.size() - 1;
     }
 
     if (mApp->restoreSession(nullptr, data)) {
         closeTab();
     } else {
         startNewSession();
     }
 }
 
 void RecoveryJsObject::closeTab()
 {
     auto *view = qobject_cast<TabbedWebView*>(m_page->view());
     if (!view) {
         return;
     }
 
     if (view->browserWindow()->tabCount() > 1) {
         view->closeView();
     } else {
         view->browserWindow()->close();
     }
 }