#include "Swiften/Parser/StanzaParser.h" #include <iostream> #include <cassert> #include "Swiften/Parser/PayloadParser.h" #include "Swiften/Parser/PayloadParserFactory.h" #include "Swiften/Parser/PayloadParserFactoryCollection.h" #include "Swiften/Parser/UnknownPayloadParser.h" namespace Swift { StanzaParser::StanzaParser(PayloadParserFactoryCollection* factories) : currentDepth_(0), factories_(factories) { } StanzaParser::~StanzaParser() { } void StanzaParser::handleStartElement(const String& element, const String& ns, const AttributeMap& attributes) { if (inStanza()) { if (!inPayload()) { assert(!currentPayloadParser_.get()); PayloadParserFactory* payloadParserFactory = factories_->getPayloadParserFactory(element, ns, attributes); if (payloadParserFactory) { currentPayloadParser_.reset(payloadParserFactory->createPayloadParser()); } else { currentPayloadParser_.reset(new UnknownPayloadParser()); } } assert(currentPayloadParser_.get()); currentPayloadParser_->handleStartElement(element, ns, attributes); } else { AttributeMap::const_iterator from = attributes.find("from"); if (from != attributes.end()) { getStanza()->setFrom(JID(from->second)); } AttributeMap::const_iterator to = attributes.find("to"); if (to != attributes.end()) { getStanza()->setTo(JID(to->second)); } AttributeMap::const_iterator id = attributes.find("id"); if (id != attributes.end()) { getStanza()->setID(id->second); } handleStanzaAttributes(attributes); } ++currentDepth_; } void StanzaParser::handleEndElement(const String& element, const String& ns) { assert(inStanza()); if (inPayload()) { assert(currentPayloadParser_.get()); currentPayloadParser_->handleEndElement(element, ns); --currentDepth_; if (!inPayload()) { boost::shared_ptr<Payload> payload(currentPayloadParser_->getPayload()); if (payload) { getStanza()->addPayload(payload); } currentPayloadParser_.reset(); } } else { --currentDepth_; } } void StanzaParser::handleCharacterData(const String& data) { if (currentPayloadParser_.get()) { currentPayloadParser_->handleCharacterData(data); } } }