1 files changed, 10 insertions, 4 deletions
diff --git a/src/xbel/xbel.h b/src/xbel/xbel.h
index 7f80eeb..6ffafb1 100644
--- a/src/xbel/xbel.h
+++ b/src/xbel/xbel.h
@@ -22,6 +22,7 @@
 #define XBELREADER_H
 
 #include <QXmlStreamReader>
+#include <QIcon>
 
 class QTreeWidget;
 class QTreeWidgetItem;
@@ -32,17 +33,22 @@ public:
     bool read(const QString &xbel);
     bool write(const QString &xbel);
 
+    QTreeWidgetItem *addFolder(QTreeWidgetItem *parentItem);
+    QTreeWidgetItem *addBookmark(QTreeWidgetItem *parentItem);
+    void addSeparator(QTreeWidgetItem *parentItem);
+
 private:
     void readChildElements(QTreeWidgetItem *parentItem);
     void readTitle(QTreeWidgetItem *item);
-    void readFolder(QTreeWidgetItem *parentItem);
-    void readBookmark(QTreeWidgetItem *parentItem);
-    void readSeparator(QTreeWidgetItem *parentItem);
 
-    QTreeWidgetItem *createChildItem(QTreeWidgetItem *item);
+    QTreeWidgetItem *parentFolder(QTreeWidgetItem *item);
+    QTreeWidgetItem *createChildItem(QTreeWidgetItem *item, const QString &type);
 
     void writeItem(QTreeWidgetItem *item);
 
+    QIcon folderIcon;
+    QIcon bookmarkIcon;
+
     QTreeWidget *treeWidget;
     QXmlStreamReader xmlReader;
     QXmlStreamWriter xmlWriter;