1 files changed, 70 insertions, 0 deletions
diff --git a/Swiften/Roster/GroupRosterItem.h b/Swiften/Roster/GroupRosterItem.h
new file mode 100644
index 0000000..f96a868
--- /dev/null
+++ b/Swiften/Roster/GroupRosterItem.h
@@ -0,0 +1,70 @@
+#ifndef SWIFTEN_GroupRosterItem_H
+#define SWIFTEN_GroupRosterItem_H
+
+#include "Swiften/Roster/RosterItem.h"
+#include "Swiften/Base/String.h"
+#include "Swiften/Roster/TreeWidget.h"
+#include "Swiften/Roster/TreeWidgetFactory.h"
+#include "Swiften/Roster/TreeWidgetItem.h"
+#include "Swiften/Roster/ContactRosterItem.h"
+
+#include <list>
+
+namespace Swift {
+
+class GroupRosterItem : public RosterItem {
+	public:
+		GroupRosterItem(const String& name, TreeWidget* tree, TreeWidgetFactory* factory) : name_(name) {
+			widget_ = factory->createTreeWidgetItem(tree);
+			widget_->setExpanded(true);
+			widget_->setText(name);
+      widget_->setTextColor(0xFFFFFF);
+      widget_->setBackgroundColor(0x969696);
+		}
+
+		~GroupRosterItem() {
+			delete widget_;
+		}
+
+		const String& getName() const {
+			return name_;
+		}
+
+		TreeWidgetItem* getWidget() const {
+			return widget_;
+		}
+
+		const std::list<RosterItem*>& getChildren() const {
+			return children_;
+		}
+
+		void addChild(RosterItem* item) {
+			children_.push_back(item);
+		}
+
+		void removeChild(const JID& jid) {
+			std::list<RosterItem*>::iterator it = children_.begin();
+			while (it != children_.end()) {
+				ContactRosterItem* contact = dynamic_cast<ContactRosterItem*>(*it);
+				if (contact && contact->getJID() == jid) {
+					delete contact;
+					it = children_.erase(it);
+					continue;
+				} 
+				GroupRosterItem* group = dynamic_cast<GroupRosterItem*>(*it);
+				if (group) {
+					group->removeChild(jid);
+				}
+				it++;
+			}
+		}
+
+	private:
+		String name_;
+		TreeWidgetItem* widget_;
+		std::list<RosterItem*> children_;
+};
+
+}
+#endif
+