1 files changed, 7 insertions, 7 deletions
diff --git a/src/sessionsiconholder.h b/src/sessionsiconholder.h
index 336ce40..d960654 100644
--- a/src/sessionsiconholder.h
+++ b/src/sessionsiconholder.h
@@ -19,20 +19,20 @@
 #include "sessionsiconholder.h"
 #include "FileDownloader.h"
 
+class SessionTreeModel;
 
 class SessionsIconHolder : QObject {
 	Q_OBJECT
 
 private:
-	QHash<QString, QIcon> iconsResource;
-	QHash<QString, QIcon> iconsFile;
-	QHash<QString, QIcon> iconsURL;
+	QHash<QString, QIcon> icons;
+	SessionTreeModel& treeModel;
+
 public:
-	SessionsIconHolder();
+	SessionsIconHolder(SessionTreeModel& sessionTreeModel);
+	QIcon getIcon(const QString& name);
+	QIcon getIcon(const QUrl& url);
 
-	QIcon getIconFromResource(const QString& name);
-	QIcon getIconFromFile(const QString& filename);
-	QIcon getIconFromURL(const QUrl& url);
 public slots:
 	void afterDownload(QString& iconName, QByteArray downloadedData);
 };