1 files changed, 27 insertions, 22 deletions
diff --git a/Swift/QtUI/UserSearch/QtContactListWidget.h b/Swift/QtUI/UserSearch/QtContactListWidget.h
index f360a91..f2483c3 100644
--- a/Swift/QtUI/UserSearch/QtContactListWidget.h
+++ b/Swift/QtUI/UserSearch/QtContactListWidget.h
@@ -4,18 +4,24 @@
  * See Documentation/Licenses/BSD-simplified.txt for more information.
  */
 
+/*
+ * Copyright (c) 2014-2016 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
+ */
+
 #pragma once
 
 #include <vector>
 
+#include <QDragEnterEvent>
+#include <QDragMoveEvent>
+#include <QDropEvent>
 #include <QTreeView>
 
-#include <Swift/Controllers/Contact.h>
 #include <Swiften/Base/Log.h>
 
-#include <QDragEnterEvent>
-#include <QDragMoveEvent>
-#include <QDropEvent>
+#include <Swift/Controllers/Contact.h>
 
 namespace Swift {
 
@@ -25,34 +31,33 @@ class SettingsProvider;
 class QtRemovableItemDelegate;
 
 class QtContactListWidget : public QTreeView {
-	Q_OBJECT
+    Q_OBJECT
 public:
-	QtContactListWidget(QWidget* parent, SettingsProvider* settings);
-	virtual ~QtContactListWidget();
+    QtContactListWidget(QWidget* parent, SettingsProvider* settings);
+    virtual ~QtContactListWidget();
 
-	void setList(const std::vector<Contact>& list);
-	std::vector<Contact> getList() const;
-	void setMaximumNoOfContactsToOne(bool limited);
+    void setList(const std::vector<Contact::ref>& list);
+    std::vector<Contact::ref> getList() const;
+    Contact::ref getContact(const size_t i);
+    void setMaximumNoOfContactsToOne(bool limited);
+    bool isFull() const;
 
 public slots:
-	void updateContacts(const std::vector<Contact>& contactUpdates);
+    void updateContacts(const std::vector<Contact::ref>& contactUpdates);
 
 signals:
-	void onListChanged(std::vector<Contact> list);
-	void onJIDsAdded(const std::vector<JID>& jids);
-
-private slots:
-	void handleListChanged(std::vector<Contact> list);
+    void onListChanged(std::vector<Contact::ref> list);
+    void onJIDsAdded(const std::vector<JID>& jids);
 
 private:
-	void handleSettingsChanged(const std::string&);
+    void handleSettingsChanged(const std::string&);
 
 private:
-	SettingsProvider* settings_;
-	ContactListModel* contactListModel_;
-	ContactListDelegate* contactListDelegate_;
-	QtRemovableItemDelegate* removableItemDelegate_;
-	bool limited_;
+    SettingsProvider* settings_;
+    ContactListModel* contactListModel_;
+    ContactListDelegate* contactListDelegate_;
+    QtRemovableItemDelegate* removableItemDelegate_;
+    bool limited_;
 };
 
 }