Swiften/Queries/Responder.h

/*
 * Copyright (c) 2010 Remko Tronçon
 * Licensed under the GNU General Public License v3.
 * See Documentation/Licenses/GPLv3.txt for more information.
 */

#pragma once

#include <Swiften/Queries/IQHandler.h>
#include <Swiften/Queries/IQRouter.h>
#include <Swiften/Elements/ErrorPayload.h>

namespace Swift {
  template<typename PAYLOAD_TYPE>
  class Responder : public IQHandler {
    public:
      Responder(IQRouter* router) : router_(router) {
      }

      ~Responder() {
      }

      void start() {
        router_->addHandler(this);
      }

      void stop() {
        router_->removeHandler(this);
      }

    protected:
      virtual bool handleGetRequest(const JID& from, const JID& to, const std::string& id, boost::shared_ptr<PAYLOAD_TYPE> payload) = 0;

      virtual bool handleSetRequest(const JID& from, const JID& to, const std::string& id, boost::shared_ptr<PAYLOAD_TYPE> payload) = 0;

      void sendResponse(const JID& to, const std::string& id, boost::shared_ptr<PAYLOAD_TYPE> payload) {
        router_->sendIQ(IQ::createResult(to, id, payload));
      }

      void sendResponse(const JID& to, const JID& from, const std::string& id, boost::shared_ptr<PAYLOAD_TYPE> payload) {
        router_->sendIQ(IQ::createResult(to, from, id, payload));
      }

      void sendError(const JID& to, const std::string& id, ErrorPayload::Condition condition, ErrorPayload::Type type, Payload::ref payload = Payload::ref()) {
        router_->sendIQ(IQ::createError(to, id, condition, type, payload));
      }

      void sendError(const JID& to, const JID& from, const std::string& id, ErrorPayload::Condition condition, ErrorPayload::Type type, Payload::ref payload = Payload::ref()) {
        router_->sendIQ(IQ::createError(to, from, id, condition, type, payload));
      }

      IQRouter* getIQRouter() const {
        return router_;
      }

    private:
      virtual bool handleIQ(boost::shared_ptr<IQ> iq) {
        if (iq->getType() == IQ::Set || iq->getType() == IQ::Get) {
          boost::shared_ptr<PAYLOAD_TYPE> payload(iq->getPayload<PAYLOAD_TYPE>());
          if (payload) {
            bool result;
            if (iq->getType() == IQ::Set) {
              result = handleSetRequest(iq->getFrom(), iq->getTo(), iq->getID(), payload);
            }
            else {
              result = handleGetRequest(iq->getFrom(), iq->getTo(), iq->getID(), payload);
            }
            if (!result) {
              router_->sendIQ(IQ::createError(iq->getFrom(), iq->getID(), ErrorPayload::NotAllowed, ErrorPayload::Cancel));
            }
            return true;
          }
        }
        return false;
      }
    
    private:
      IQRouter* router_;
  };
}