From b1da23ca72fce0713c22630771eb0e917231446d Mon Sep 17 00:00:00 2001
From: nowrep <nowrep@gmail.com>
Date: Mon, 30 Dec 2013 00:35:01 +0100
Subject: [PATCH] [Bookmarks] Added option to export bookmarks to HTML file.

---
 src/lib/bookmarks/bookmarksmanager.cpp | 16 ++++-
 src/lib/bookmarks/bookmarksmanager.h   |  2 +
 src/lib/bookmarks/bookmarksmanager.ui  | 40 +++++--------
 src/lib/bookmarks/bookmarksmodel.cpp   | 81 ++++++++++++++++++++++++++
 src/lib/bookmarks/bookmarksmodel.h     |  2 +
 5 files changed, 116 insertions(+), 25 deletions(-)

diff --git a/src/lib/bookmarks/bookmarksmanager.cpp b/src/lib/bookmarks/bookmarksmanager.cpp
index 8da7d4b80..49ecd1670 100644
--- a/src/lib/bookmarks/bookmarksmanager.cpp
+++ b/src/lib/bookmarks/bookmarksmanager.cpp
@@ -37,6 +37,7 @@
 #include <QShortcut>
 #include <QMenu>
 #include <QSqlQuery>
+#include <QLabel>
 
 BookmarksManager::BookmarksManager(QupZilla* mainClass, QWidget* parent)
     : QWidget(parent)
@@ -70,7 +71,11 @@ BookmarksManager::BookmarksManager(QupZilla* mainClass, QWidget* parent)
     connect(m_bookmarksModel, SIGNAL(folderParentChanged(QString,bool)), this, SLOT(changeFolderParent(QString,bool)));
     connect(m_bookmarksModel, SIGNAL(bookmarkParentChanged(QString,QByteArray,int,QUrl,QString,QString)), this, SLOT(changeBookmarkParent(QString,QByteArray,int,QUrl,QString,QString)));
 
-    connect(ui->importBookmarks, SIGNAL(clicked(QPoint)), this, SLOT(importBookmarks()));
+    QMenu* menu = new QMenu;
+    menu->addAction(tr("Import Bookmarks..."), this, SLOT(importBookmarks()));
+    menu->addAction(tr("Export Bookmarks to HTML..."), this, SLOT(exportBookmarks()));
+
+    ui->importExport->setMenu(menu);
 
     QShortcut* deleteAction = new QShortcut(QKeySequence("Del"), ui->bookmarksTree);
     connect(deleteAction, SIGNAL(activated()), this, SLOT(deleteItem()));
@@ -85,6 +90,15 @@ void BookmarksManager::importBookmarks()
     b->show();
 }
 
+void BookmarksManager::exportBookmarks()
+{
+    QString file = QzTools::getSaveFileName("BookmarksManager-Export", this, tr("Export to HTML..."), QDir::homePath() + "/bookmarks.html");
+
+    if (!file.isEmpty()) {
+        m_bookmarksModel->exportToHtml(file);
+    }
+}
+
 void BookmarksManager::search(const QString &string)
 {
     ui->bookmarksTree->filterString(string);
diff --git a/src/lib/bookmarks/bookmarksmanager.h b/src/lib/bookmarks/bookmarksmanager.h
index 1e7d6cf4a..008123403 100644
--- a/src/lib/bookmarks/bookmarksmanager.h
+++ b/src/lib/bookmarks/bookmarksmanager.h
@@ -64,7 +64,9 @@ private slots:
     void moveBookmark();
     void renameBookmark();
     void changeIcon();
+
     void importBookmarks();
+    void exportBookmarks();
 
     void addFolder(const QString &name);
     void addSubfolder(const QString &name);
diff --git a/src/lib/bookmarks/bookmarksmanager.ui b/src/lib/bookmarks/bookmarksmanager.ui
index 98b1d2816..d0cfa93ce 100644
--- a/src/lib/bookmarks/bookmarksmanager.ui
+++ b/src/lib/bookmarks/bookmarksmanager.ui
@@ -14,16 +14,17 @@
    <string>Bookmarks</string>
   </property>
   <layout class="QGridLayout" name="gridLayout">
-   <item row="1" column="7">
-    <widget class="ClickableLabel" name="importBookmarks">
-     <property name="cursor">
-      <cursorShape>PointingHandCursor</cursorShape>
-     </property>
+   <item row="1" column="0" colspan="4">
+    <widget class="QPushButton" name="addFolder">
      <property name="text">
-      <string>Import Bookmarks</string>
+      <string>Add Folder</string>
      </property>
-     <property name="html-link-look" stdset="0">
-      <bool>true</bool>
+    </widget>
+   </item>
+   <item row="1" column="5">
+    <widget class="QPushButton" name="collapseAll">
+     <property name="text">
+      <string>Collapse All</string>
      </property>
     </widget>
    </item>
@@ -40,14 +41,7 @@
      </property>
     </spacer>
    </item>
-   <item row="1" column="0" colspan="4">
-    <widget class="QPushButton" name="addFolder">
-     <property name="text">
-      <string>Add Folder</string>
-     </property>
-    </widget>
-   </item>
-   <item row="0" column="0" colspan="9">
+   <item row="0" column="0" colspan="8">
     <widget class="BookmarksTree" name="bookmarksTree">
      <property name="contextMenuPolicy">
       <enum>Qt::CustomContextMenu</enum>
@@ -83,21 +77,19 @@
      </property>
     </widget>
    </item>
-   <item row="1" column="5">
-    <widget class="QPushButton" name="collapseAll">
+   <item row="1" column="7">
+    <widget class="QToolButton" name="importExport">
      <property name="text">
-      <string>Collapse All</string>
+      <string>Import and Export</string>
+     </property>
+     <property name="popupMode">
+      <enum>QToolButton::InstantPopup</enum>
      </property>
     </widget>
    </item>
   </layout>
  </widget>
  <customwidgets>
-  <customwidget>
-   <class>ClickableLabel</class>
-   <extends>QLabel</extends>
-   <header>clickablelabel.h</header>
-  </customwidget>
   <customwidget>
    <class>BookmarksTree</class>
    <extends>QTreeWidget</extends>
diff --git a/src/lib/bookmarks/bookmarksmodel.cpp b/src/lib/bookmarks/bookmarksmodel.cpp
index b147eb0f3..4134d76f9 100644
--- a/src/lib/bookmarks/bookmarksmodel.cpp
+++ b/src/lib/bookmarks/bookmarksmodel.cpp
@@ -22,7 +22,10 @@
 #include "mainapplication.h"
 #include "settings.h"
 
+#include <QTextStream>
 #include <QBuffer>
+#include <QDebug>
+#include <QFile>
 
 // SQLite DB -> table bookmarks + folders
 // Unique in bookmarks table is id
@@ -447,6 +450,84 @@ bool BookmarksModel::renameFolder(const QString &before, const QString &after)
     return true;
 }
 
+void BookmarksModel::exportToHtml(const QString &fileName)
+{
+    QFile file(fileName);
+
+    if (!file.open(QFile::WriteOnly | QFile::Truncate)) {
+        qWarning() << "BookmarksModel::exportHtml Cannot open file for writing!" << file.errorString();
+    }
+
+    QTextStream out(&file);
+
+    out << "<!DOCTYPE NETSCAPE-Bookmark-file-1>" << endl;
+    out << "<!-- This is an automatically generated file." << endl;
+    out << "     It will be read and overwritten." << endl;
+    out << "     DO NOT EDIT! -->" << endl;
+    out << "<META HTTP-EQUIV=\"Content-Type\" CONTENT=\"text/html; charset=UTF-8\">" << endl;
+    out << "<TITLE>Bookmarks</TITLE>" << endl;
+    out << "<H1>Bookmarks</H1>" << endl;
+
+    out << "<DL><p>" << endl;
+
+    QString indent = "    ";
+    QList<QPair<QString, bool> > allFolders;
+
+    QPair<QString, bool> menu;
+    menu.first = "bookmarksMenu";
+    menu.second = false;
+
+    QPair<QString, bool> toolbar;
+    toolbar.first = "bookmarksToolbar";
+    toolbar.second = false;
+
+    allFolders.append(menu);
+    allFolders.append(toolbar);
+
+    QSqlQuery query;
+    query.exec("SELECT name, subfolder FROM folders");
+
+    while (query.next()) {
+        QPair<QString, bool> pair;
+        pair.first = query.value(0).toString();
+        pair.second = query.value(1).toString() == QLatin1String("yes");
+
+        allFolders.append(pair);
+    }
+
+    for (int i = 0; i < allFolders.size(); ++i) {
+        QPair<QString, bool> pair = allFolders.at(i);
+
+        out << indent << "<DT><H3 TOOLBAR_SUBFOLDER=\"" << (pair.second ? "yes" : "no") << "\">" << pair.first << "</H3>" << endl;
+        out << indent << "<DL><p>" << endl;
+
+        QSqlQuery q;
+        q.prepare("SELECT title, url FROM bookmarks WHERE folder = ?");
+        q.addBindValue(pair.first);
+        q.exec();
+
+        while (q.next()) {
+            QString title = q.value(0).toString();
+            QString url = q.value(1).toString();
+
+            out << indent << indent << "<DT><A HREF=\"" << url << "\">" << title << "</A>" << endl;
+        }
+
+        out << indent << "</DL><p>" << endl;
+    }
+
+    query.exec("SELECT title, url FROM bookmarks WHERE folder='' OR folder='unsorted'");
+
+    while (query.next()) {
+        QString title = query.value(0).toString();
+        QString url = query.value(1).toString();
+
+        out << indent << "<DT><A HREF=\"" << url << "\">" << title << "</A>" << endl;
+    }
+
+    out << "</DL><p>" << endl;
+}
+
 QVector<Bookmark> BookmarksModel::folderBookmarks(const QString &name)
 {
     QVector<Bookmark> list;
diff --git a/src/lib/bookmarks/bookmarksmodel.h b/src/lib/bookmarks/bookmarksmodel.h
index 5b9ee9dd9..960e5906a 100644
--- a/src/lib/bookmarks/bookmarksmodel.h
+++ b/src/lib/bookmarks/bookmarksmodel.h
@@ -98,6 +98,8 @@ public:
 
     bool renameFolder(const QString &before, const QString &after);
 
+    void exportToHtml(const QString &fileName);
+
     static bool bookmarksEqual(const Bookmark &one, const Bookmark &two);
     static QString toTranslatedFolder(const QString &name);
     static QString fromTranslatedFolder(const QString &name);