1 files changed, 95 insertions, 0 deletions
diff --git a/Swift/QtUI/QtSetGroupsDialog.cpp b/Swift/QtUI/QtSetGroupsDialog.cpp
new file mode 100644
index 0000000..ad24122
--- /dev/null
+++ b/Swift/QtUI/QtSetGroupsDialog.cpp
@@ -0,0 +1,95 @@
+/*
+ * Copyright (c) 2010 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include "QtSetGroupsDialog.h"
+
+#include <algorithm>
+
+#include <QScrollArea>
+#include <QBoxLayout>
+#include <QLabel>
+#include <QDialogButtonBox>
+
+#include "Swift/QtUI/QtSwiftUtil.h"
+
+namespace Swift {
+
+QtSetGroupsDialog::QtSetGroupsDialog(ContactRosterItem* contact, const QList<QString>& allGroups) : contact_(contact) {
+	QBoxLayout* layout = new QBoxLayout(QBoxLayout::TopToBottom, this);
+	QScrollArea* scroll = new QScrollArea(this);
+	layout->addWidget(scroll);
+	QBoxLayout* scrollLayout = new QBoxLayout(QBoxLayout::TopToBottom, scroll);
+	QLabel* label = new QLabel(scroll);
+	label->setText("Choose new groups for " + P2QSTRING(contact->getDisplayName()));
+	scrollLayout->addWidget(label);
+	foreach (QString group, allGroups) {
+			QCheckBox* check = new QCheckBox(scroll);
+			check->setText(group);
+			check->setCheckState(Qt::Unchecked);
+			checkBoxes_[Q2PSTRING(group)] = check;
+			scrollLayout->addWidget(check);
+		}
+	foreach (String group, contact->getGroups()) {
+		checkBoxes_[group]->setCheckState(Qt::Checked);
+	}
+	QWidget* newGroupWidget = new QWidget(scroll);
+	QBoxLayout* newGroupLayout = new QBoxLayout(QBoxLayout::LeftToRight, newGroupWidget);
+	scrollLayout->addWidget(newGroupWidget);
+	newGroup_ = new QCheckBox(newGroupWidget);
+	newGroup_->setText("New Group:");
+	newGroup_->setCheckState(Qt::Unchecked);
+	newGroupLayout->addWidget(newGroup_);
+	newGroupName_ = new QLineEdit(newGroupWidget);
+	newGroupLayout->addWidget(newGroupName_);
+	QDialogButtonBox* buttons = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel, Qt::Horizontal, this);
+	layout->addWidget(buttons);
+	connect(buttons, SIGNAL(accepted()), this, SLOT(accept()));
+	connect(buttons, SIGNAL(rejected()), this, SLOT(reject()));
+}
+
+QtSetGroupsDialog::~QtSetGroupsDialog() {
+	// TODO Auto-generated destructor stub
+}
+
+typedef std::pair<String, QCheckBox*> CheckStringPair;
+
+boost::shared_ptr<RegroupRosterItemUIEvent> QtSetGroupsDialog::getRegroupEvent() {
+	std::vector<String> addedGroups;
+	std::vector<String> removedGroups;
+	std::vector<String> existingGroups = contact_->getGroups();
+	int tickedCount = 0;
+	bool wantsContacts = false;
+	foreach (CheckStringPair pair, checkBoxes_) {
+		bool existing = std::find(existingGroups.begin(), existingGroups.end(), pair.first) != existingGroups.end();
+		if (pair.second->checkState() == Qt::Checked) {
+			tickedCount++;
+			if (pair.first == "Contacts") {
+				wantsContacts = true;
+			}
+			if (!existing) {
+				addedGroups.push_back(pair.first);
+			}
+		} else {
+			if (existing) {
+				removedGroups.push_back(pair.first);
+			}
+		}
+	}
+	if (newGroup_->checkState() == Qt::Checked) {
+		tickedCount++;
+		String name = Q2PSTRING(newGroupName_->text());
+		if (std::find(existingGroups.begin(), existingGroups.end(), name) == existingGroups.end()) {
+			addedGroups.push_back(name);
+		}
+	}
+	if (tickedCount > 1 && wantsContacts) {
+		addedGroups.push_back("Contacts");
+	}
+	boost::shared_ptr<RegroupRosterItemUIEvent> result(new RegroupRosterItemUIEvent(contact_->getJID(), addedGroups, removedGroups));
+	return result;
+}
+
+}