1 files changed, 21 insertions, 14 deletions
diff --git a/Swift/QtUI/MUCSearch/MUCSearchServiceItem.h b/Swift/QtUI/MUCSearch/MUCSearchServiceItem.h
index cfde071..9f5e000 100644
--- a/Swift/QtUI/MUCSearch/MUCSearchServiceItem.h
+++ b/Swift/QtUI/MUCSearch/MUCSearchServiceItem.h
@@ -6,27 +6,34 @@
 
 #pragma once
 
-#include <QList>
+#include <memory>
+
+#include <boost/optional.hpp>
+
 #include <QString>
+#include <QVector>
 
 #include <Swift/QtUI/MUCSearch/MUCSearchRoomItem.h>
 
 namespace Swift {
-    class MUCSearchServiceItem : public MUCSearchItem {
+    class MUCSearchServiceItem : public MUCSearchItem, public std::enable_shared_from_this<MUCSearchServiceItem> {
         public:
-            MUCSearchServiceItem(const QString& jidString) : jidString_(jidString) {}
-            void addRoom(MUCSearchItem* room) {rooms_.push_back(room);}
-            int rowCount() {return rooms_.count();}
-            MUCSearchItem* getItem(int i) {return rooms_[i];}
-            QVariant data(int role) {
-                switch (role) {
-                    case Qt::DisplayRole: return QVariant(jidString_);
-                    default: return QVariant();
-                }
-            }
-            QString getHost() const {return jidString_;}
+            MUCSearchServiceItem(const QString& jidString);
+
+            void addRoom(std::shared_ptr<MUCSearchItem> room);
+            void addRooms(const std::vector<std::shared_ptr<MUCSearchItem>>& rooms);
+            int rowCount();
+            MUCSearchItem* getItem(int i);
+            QVariant data(int role);
+            QString getHost() const;
+            void setSorting(Qt::SortOrder sortOrder);
+
+        private:
+            void sort();
+
         private:
-            QList<MUCSearchItem*> rooms_;
+            QVector<std::shared_ptr<MUCSearchItem>> rooms_;
             QString jidString_;
+            boost::optional<Qt::SortOrder> sortOrder_;
     };
 }