1 files changed, 92 insertions, 0 deletions
diff --git a/Swift/Controllers/Chat/MUCController.cpp b/Swift/Controllers/Chat/MUCController.cpp
new file mode 100644
index 0000000..7736aec
--- /dev/null
+++ b/Swift/Controllers/Chat/MUCController.cpp
@@ -0,0 +1,92 @@
+#include "Swift/Controllers/Chat/MUCController.h"
+
+#include <boost/bind.hpp>
+
+#include "Swiften/Base/foreach.h"
+#include "Swift/Controllers/UIInterfaces/ChatWindow.h"
+#include "Swift/Controllers/UIInterfaces/ChatWindowFactory.h"
+#include "Swiften/Avatars/AvatarManager.h"
+#include "Swiften/MUC/MUC.h"
+#include "Swiften/Client/StanzaChannel.h"
+#include "Swiften/Roster/Roster.h"
+#include "Swiften/Roster/SetAvatar.h"
+#include "Swiften/Roster/SetPresence.h"
+#include "Swiften/Roster/TreeWidgetFactory.h"
+
+namespace Swift {
+	
+/**
+ * The controller does not gain ownership of the stanzaChannel, nor the factory.
+ */
+MUCController::MUCController (
+		const JID& self,
+		const JID &muc, 
+		const String &nick, 
+		StanzaChannel* stanzaChannel, 
+		PresenceSender* presenceSender,
+		IQRouter* iqRouter, 
+		ChatWindowFactory* chatWindowFactory, 
+		TreeWidgetFactory *treeWidgetFactory,
+		PresenceOracle* presenceOracle,
+		AvatarManager* avatarManager) : 
+			ChatControllerBase(self, stanzaChannel, iqRouter, chatWindowFactory, muc, presenceOracle, avatarManager),
+			muc_(new MUC(stanzaChannel, presenceSender, muc)), 
+			nick_(nick), 
+			treeWidgetFactory_(treeWidgetFactory) { 
+	roster_ = new Roster(chatWindow_->getTreeWidget(), treeWidgetFactory_);
+	chatWindow_->onClosed.connect(boost::bind(&MUCController::handleWindowClosed, this));
+	muc_->joinAs(nick);
+	muc_->onOccupantJoined.connect(boost::bind(&MUCController::handleOccupantJoined, this, _1));
+	muc_->onOccupantPresenceChange.connect(boost::bind(&MUCController::handleOccupantPresenceChange, this, _1));
+	muc_->onOccupantLeft.connect(boost::bind(&MUCController::handleOccupantLeft, this, _1, _2, _3));
+	chatWindow_->convertToMUC();
+	chatWindow_->show();
+	if (avatarManager_ != NULL) {
+		avatarManager_->onAvatarChanged.connect(boost::bind(&MUCController::handleAvatarChanged, this, _1, _2));
+	} 
+}
+
+MUCController::~MUCController() {
+	delete muc_;
+	delete roster_;
+}
+
+void MUCController::handleAvatarChanged(const JID& jid, const String&) {
+	String path = avatarManager_->getAvatarPath(jid).string();
+	roster_->applyOnItems(SetAvatar(jid, path, JID::WithResource));
+}
+
+void MUCController::handleWindowClosed() {
+	muc_->part();
+}
+
+void MUCController::handleOccupantJoined(const MUCOccupant& occupant) {
+	JID jid(JID(toJID_.getNode(), toJID_.getDomain(), occupant.getNick()));
+	roster_->addContact(jid, occupant.getNick(), "Occupants");
+	if (avatarManager_ != NULL) {
+		handleAvatarChanged(jid, "dummy");
+	}
+}
+
+void MUCController::handleOccupantLeft(const MUCOccupant& occupant, MUC::LeavingType, const String& /*reason*/) {
+	roster_->removeContact(JID(toJID_.getNode(), toJID_.getDomain(), occupant.getNick()));
+}
+
+void MUCController::handleOccupantPresenceChange(boost::shared_ptr<Presence> presence) {
+	roster_->applyOnItems(SetPresence(presence, JID::WithResource));
+}
+
+bool MUCController::isIncomingMessageFromMe(boost::shared_ptr<Message> message) {
+	JID from = message->getFrom();
+	return nick_ == from.getResource();
+}
+
+String MUCController::senderDisplayNameFromMessage(const JID& from) {
+	return from.getResource();
+}
+
+void MUCController::preSendMessageRequest(boost::shared_ptr<Message> message) {
+	message->setType(Swift::Message::Groupchat);
+}
+
+}