diff options
| author | Kevin Smith <git@kismith.co.uk> | 2012-02-06 12:12:00 (GMT) | 
|---|---|---|
| committer | Kevin Smith <git@kismith.co.uk> | 2012-02-06 13:39:43 (GMT) | 
| commit | f3d68b13e120f471326449f68aa4140587d444fc (patch) | |
| tree | a13586525ee07d8aff8ec9ec459caf37fd5a941c | |
| parent | 42a1bbc568aab7e5b29c7fe16c13659291d01cae (diff) | |
| download | swift-contrib-f3d68b13e120f471326449f68aa4140587d444fc.zip swift-contrib-f3d68b13e120f471326449f68aa4140587d444fc.tar.bz2  | |
Parse mediated invite stuff
4 files changed, 93 insertions, 5 deletions
diff --git a/Swiften/Elements/MUCUserPayload.h b/Swiften/Elements/MUCUserPayload.h index c9ea62c..fcb585e 100644 --- a/Swiften/Elements/MUCUserPayload.h +++ b/Swiften/Elements/MUCUserPayload.h @@ -27,17 +27,26 @@ namespace Swift {  				int code;  			}; -			// struct Password { +			//struct Password { -			// } +			//}  			// struct History {  			// } -			// struct Invite { - -			// } +			/** +			 * reason is optional. +			 * from and to are mutually exclusive. +			 * From is used for MUC sending to invited client. To is used sending to MUC from inviting client. +			 * from is the JID the MUC claims the invite is from. +			 * to is the JID to send the invite to. +			 */ +			struct Invite { +				std::string reason; +				JID from; +				JID to; +			};  			MUCUserPayload() {  			} @@ -58,9 +67,29 @@ namespace Swift {  				payload_ = p;  			} +			const boost::optional<std::string>& getPassword() const { +				return password_; +			} + +			void setPassword(const std::string& password) { +				password_ = password; +			} + + +			const boost::optional<Invite>& getInvite() const { +				return invite_; +			} + +			void setInvite(const Invite& invite) { +				invite_ = invite; +			} + +  		private:  			std::vector<MUCItem> items_;  			std::vector<StatusCode> statusCodes_;  			boost::shared_ptr<Payload> payload_; +			boost::optional<std::string> password_; +			boost::optional<Invite> invite_;  	};  } diff --git a/Swiften/Parser/PayloadParsers/MUCUserPayloadParser.cpp b/Swiften/Parser/PayloadParsers/MUCUserPayloadParser.cpp index 2da8b35..d206010 100644 --- a/Swiften/Parser/PayloadParsers/MUCUserPayloadParser.cpp +++ b/Swiften/Parser/PayloadParsers/MUCUserPayloadParser.cpp @@ -22,6 +22,25 @@ void MUCUserPayloadParser::handleTree(ParserElement::ref root) {  			MUCItem item = MUCItemParser::itemFromTree(child);  			getPayloadInternal()->addItem(item);  		} +		else if (child->getName() == "password" && child->getNamespace() == root->getNamespace()) { +			getPayloadInternal()->setPassword(child->getText()); +		} +		else if (child->getName() == "invite" && child->getNamespace() == root->getNamespace()) { +			MUCUserPayload::Invite invite; +			std::string to = child->getAttributes().getAttribute("to"); +			if (!to.empty()) { +				invite.to = to; +			} +			std::string from = child->getAttributes().getAttribute("from"); +			if (!from.empty()) { +				invite.from = from; +			} +			ParserElement::ref reason = child->getChild("reason", root->getNamespace()); +			if (reason) { +				invite.reason = reason->getText(); +			} +			getPayloadInternal()->setInvite(invite); +		}  		else if (child->getName() == "status" && child->getNamespace() == root->getNamespace()) {  			MUCUserPayload::StatusCode status;  			try { diff --git a/Swiften/Parser/PayloadParsers/UnitTest/MUCUserPayloadParserTest.cpp b/Swiften/Parser/PayloadParsers/UnitTest/MUCUserPayloadParserTest.cpp index 45862e2..40d7358 100644 --- a/Swiften/Parser/PayloadParsers/UnitTest/MUCUserPayloadParserTest.cpp +++ b/Swiften/Parser/PayloadParsers/UnitTest/MUCUserPayloadParserTest.cpp @@ -20,6 +20,7 @@ class MUCUserPayloadParserTest : public CppUnit::TestFixture  		CPPUNIT_TEST(testParseEmpty);  		CPPUNIT_TEST(testParse);  		CPPUNIT_TEST(testParseDestroy); +		CPPUNIT_TEST(testParseInvite);  		CPPUNIT_TEST_SUITE_END();  	public: @@ -71,6 +72,23 @@ class MUCUserPayloadParserTest : public CppUnit::TestFixture  			CPPUNIT_ASSERT_EQUAL(std::string("bert"), destroy->getReason());  			CPPUNIT_ASSERT_EQUAL(JID("alice@wonderland.lit"), destroy->getNewVenue());  		} + +		void testParseInvite() { +			PayloadsParserTester parser; + +			CPPUNIT_ASSERT(parser.parse("<x xmlns=\"http://jabber.org/protocol/muc#user\"><invite from='crone1@shakespeare.lit/desktop' to='alice@wonderland.lit/xxx'>      <reason>Hey Hecate, this is the place for all good witches!</reason>    </invite>    <password>cauldronburn</password></x>")); + +			MUCUserPayload::ref payload = boost::dynamic_pointer_cast<MUCUserPayload>(parser.getPayload()); +			CPPUNIT_ASSERT(payload); +			CPPUNIT_ASSERT(payload->getInvite()); +			CPPUNIT_ASSERT(payload->getPassword()); +			CPPUNIT_ASSERT_EQUAL(std::string("cauldronburn"), *payload->getPassword()); +			MUCUserPayload::Invite invite = *payload->getInvite(); +			CPPUNIT_ASSERT_EQUAL(std::string("Hey Hecate, this is the place for all good witches!"), invite.reason); +			CPPUNIT_ASSERT_EQUAL(JID("crone1@shakespeare.lit/desktop"), invite.from); +			CPPUNIT_ASSERT_EQUAL(JID("alice@wonderland.lit/xxx"), invite.to); +		} +  };  CPPUNIT_TEST_SUITE_REGISTRATION(MUCUserPayloadParserTest); diff --git a/Swiften/Serializer/PayloadSerializers/MUCUserPayloadSerializer.cpp b/Swiften/Serializer/PayloadSerializers/MUCUserPayloadSerializer.cpp index 96fba90..66ca5d0 100644 --- a/Swiften/Serializer/PayloadSerializers/MUCUserPayloadSerializer.cpp +++ b/Swiften/Serializer/PayloadSerializers/MUCUserPayloadSerializer.cpp @@ -34,6 +34,28 @@ std::string MUCUserPayloadSerializer::serializePayload(boost::shared_ptr<MUCUser  	foreach (const MUCItem& item, payload->getItems()) {  		mucElement.addNode(MUCItemSerializer::itemToElement(item));  	} + +	if (payload->getPassword()) { +		boost::shared_ptr<XMLElement> passwordElement = boost::make_shared<XMLElement>("password"); +		passwordElement->addNode(boost::make_shared<XMLTextNode>(*payload->getPassword())); +	} + +	if (payload->getInvite()) { +		MUCUserPayload::Invite invite = *payload->getInvite(); +		boost::shared_ptr<XMLElement> inviteElement = boost::make_shared<XMLElement>("invite"); +		if (invite.to.isValid()) { +			inviteElement->setAttribute("to", invite.to.toString()); +		} +		if (invite.from.isValid()) { +			inviteElement->setAttribute("from", invite.from.toString()); +		} +		if (!invite.reason.empty()) { +			boost::shared_ptr<XMLElement> reasonElement = boost::make_shared<XMLElement>("reason"); +			reasonElement->addNode(boost::make_shared<XMLTextNode>(invite.reason)); +		} +		mucElement.addNode(inviteElement); +	} +  	boost::shared_ptr<Payload> childPayload = payload->getPayload();  	if (childPayload) {  		PayloadSerializer* serializer = serializers->getPayloadSerializer(childPayload);  | 
 Swift