1 files changed, 23 insertions, 21 deletions
diff --git a/Swift/QtUI/MUCSearch/MUCSearchModel.h b/Swift/QtUI/MUCSearch/MUCSearchModel.h
index e18a1c0..f36d147 100644
--- a/Swift/QtUI/MUCSearch/MUCSearchModel.h
+++ b/Swift/QtUI/MUCSearch/MUCSearchModel.h
@@ -1,35 +1,37 @@
 /*
- * Copyright (c) 2010 Isode Limited.
+ * Copyright (c) 2010-2016 Isode Limited.
  * All rights reserved.
  * See the COPYING file for more information.
  */
 
 #pragma once
 
-#include <boost/shared_ptr.hpp>
+#include <memory>
+
+#include <boost/optional.hpp>
 
 #include <QAbstractItemModel>
-#include <QList>
+#include <QVector>
 
-#include "Swift/QtUI/MUCSearch/MUCSearchServiceItem.h"
+#include <Swift/QtUI/MUCSearch/MUCSearchServiceItem.h>
 
 namespace Swift {
-	class MUCSearchModel : public QAbstractItemModel {
-		Q_OBJECT
-		public:
-			MUCSearchModel();
-			void clear();
-			void addService(MUCSearchServiceItem* service);
-			int columnCount(const QModelIndex& parent = QModelIndex()) const;
-			QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const;
-			QModelIndex index(int row, int column, const QModelIndex & parent = QModelIndex()) const;
-			QModelIndex parent(const QModelIndex& index) const;
-			int rowCount(const QModelIndex& parent = QModelIndex()) const;
-//			ChatListItem* getItemForIndex(const QModelIndex& index) const;
-		private:
-//			ChatListGroupItem* mucBookmarks_;
-//			ChatListGroupItem* root_;
-			QList<MUCSearchServiceItem*> services_;
-	};
+    class MUCSearchModel : public QAbstractItemModel {
+        Q_OBJECT
+        public:
+            MUCSearchModel();
+            void clear();
+            void addService(std::shared_ptr<MUCSearchServiceItem> service);
+            int columnCount(const QModelIndex& parent = QModelIndex()) const;
+            QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const;
+            QModelIndex index(int row, int column, const QModelIndex & parent = QModelIndex()) const;
+            QModelIndex parent(const QModelIndex& index) const;
+            int rowCount(const QModelIndex& parent = QModelIndex()) const;
+            virtual void sort(int column, Qt::SortOrder order = Qt::AscendingOrder);
+
+        private:
+            QVector<std::shared_ptr<MUCSearchServiceItem>> services_;
+            boost::optional<Qt::SortOrder> sortOrder_;
+    };
 
 }