File indexing completed on 2024-12-22 04:40:57

 /* ============================================================
 * Falkon - Qt web browser
 * Copyright (C) 2010-2016 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 "chromeimporter.h"
 #include "bookmarkitem.h"
 
 #include <QDir>
 #include <QFileDialog>
 #include <QJsonDocument>
 #include <QMetaType>
 
 ChromeImporter::ChromeImporter(QObject* parent)
     : BookmarksImporter(parent)
 {
 }
 
 QString ChromeImporter::description() const
 {
     return BookmarksImporter::tr("Google Chrome stores its bookmarks in <b>Bookmarks</b> text file. "
                                  "This file is usually located in");
 }
 
 QString ChromeImporter::standardPath() const
 {
 #if defined(Q_OS_WIN)
     return QString("%APPDATA%/Chrome/");
 #elif defined(Q_OS_OSX)
     return QDir::homePath() + QLatin1String("/Library/Application Support/Google/Chrome/");
 #else
     return QDir::homePath() + QLatin1String("/.config/chrome/");
 #endif
 }
 
 QString ChromeImporter::getPath(QWidget* parent)
 {
     m_path = QFileDialog::getOpenFileName(parent, BookmarksImporter::tr("Choose file..."), standardPath(), QSL("Bookmarks (Bookmarks)"));
     return m_path;
 }
 
 bool ChromeImporter::prepareImport()
 {
     m_file.setFileName(m_path);
 
     if (!m_file.open(QFile::ReadOnly)) {
         setError(BookmarksImporter::tr("Unable to open file."));
         return false;
     }
 
     return true;
 }
 
 BookmarkItem* ChromeImporter::importBookmarks()
 {
     const QByteArray data = m_file.readAll();
     m_file.close();
 
     QJsonParseError err;
     QJsonDocument json = QJsonDocument::fromJson(data, &err);
     const QVariant res = json.toVariant();
 
     if (err.error != QJsonParseError::NoError || res.typeId() != QMetaType::QVariantMap) {
         setError(BookmarksImporter::tr("Cannot parse JSON file!"));
         return nullptr;
     }
 
     QVariantMap rootMap = res.toMap().value(QSL("roots")).toMap();
 
     auto* root = new BookmarkItem(BookmarkItem::Folder);
     root->setTitle(QSL("Chrome Import"));
 
     auto* toolbar = new BookmarkItem(BookmarkItem::Folder, root);
     toolbar->setTitle(rootMap.value(QSL("bookmark_bar")).toMap().value(QSL("name")).toString());
     readBookmarks(rootMap.value(QSL("bookmark_bar")).toMap().value(QSL("children")).toList(), toolbar);
 
     auto* other = new BookmarkItem(BookmarkItem::Folder, root);
     other->setTitle(rootMap.value(QSL("other")).toMap().value(QSL("name")).toString());
     readBookmarks(rootMap.value(QSL("other")).toMap().value(QSL("children")).toList(), other);
 
     auto* synced = new BookmarkItem(BookmarkItem::Folder, root);
     synced->setTitle(rootMap.value(QSL("synced")).toMap().value(QSL("name")).toString());
     readBookmarks(rootMap.value(QSL("synced")).toMap().value(QSL("synced")).toList(), other);
 
     return root;
 }
 
 void ChromeImporter::readBookmarks(const QVariantList &list, BookmarkItem* parent)
 {
     Q_ASSERT(parent);
 
     for (const QVariant &entry : list) {
         const QVariantMap map = entry.toMap();
         const QString typeString = map.value(QSL("type")).toString();
         BookmarkItem::Type type;
 
         if (typeString == QLatin1String("url")) {
             type = BookmarkItem::Url;
         }
         else if (typeString == QLatin1String("folder")) {
             type = BookmarkItem::Folder;
         }
         else {
             continue;
         }
 
         auto* item = new BookmarkItem(type, parent);
         item->setTitle(map.value(QSL("name")).toString());
 
         if (item->isUrl()) {
             item->setUrl(QUrl::fromEncoded(map.value(QSL("url")).toByteArray()));
         }
 
         if (map.contains(QSL("children"))) {
             readBookmarks(map.value(QSL("children")).toList(), item);
         }
     }
 }