diff options
| author | Remko Tronçon <git@el-tramo.be> | 2011-05-13 21:57:48 (GMT) | 
|---|---|---|
| committer | Remko Tronçon <git@el-tramo.be> | 2011-05-13 21:57:48 (GMT) | 
| commit | 27d7cb1c99a55bfb45658b4f20f901926e526cae (patch) | |
| tree | b650af0f9aa85fa2ec61637e0ac36b8a01d64e99 | |
| parent | 7a4a9e1ed5733a6517124edf3c4bd55e87188dde (diff) | |
| download | swift-contrib-27d7cb1c99a55bfb45658b4f20f901926e526cae.zip swift-contrib-27d7cb1c99a55bfb45658b4f20f901926e526cae.tar.bz2 | |
Refactored OutgoingAdHocCommandSession a bit.
| -rw-r--r-- | Swiften/AdHoc/OutgoingAdHocCommandSession.cpp | 39 | ||||
| -rw-r--r-- | Swiften/AdHoc/OutgoingAdHocCommandSession.h | 7 | ||||
| -rw-r--r-- | Swiften/Base/Algorithm.h | 11 | 
3 files changed, 32 insertions, 25 deletions
| diff --git a/Swiften/AdHoc/OutgoingAdHocCommandSession.cpp b/Swiften/AdHoc/OutgoingAdHocCommandSession.cpp index 758b5e7..0632bbf 100644 --- a/Swiften/AdHoc/OutgoingAdHocCommandSession.cpp +++ b/Swiften/AdHoc/OutgoingAdHocCommandSession.cpp @@ -7,8 +7,10 @@  #include <Swiften/AdHoc/OutgoingAdHocCommandSession.h>  #include <boost/bind.hpp> +#include <boost/smart_ptr/make_shared.hpp>  #include <Swiften/Queries/GenericRequest.h> +#include <Swiften/Base/Algorithm.h>  namespace Swift {  OutgoingAdHocCommandSession::OutgoingAdHocCommandSession(const DiscoItems::Item& command, IQRouter* iqRouter) : command_(command), iqRouter_(iqRouter), isMultiStage_(false) { @@ -19,7 +21,7 @@ void OutgoingAdHocCommandSession::handleResponse(boost::shared_ptr<Command> payl  	if (error) {  		onError(error);  	} else { -		const std::vector<Command::Action> actions = payload->getAvailableActions(); +		const std::vector<Command::Action>& actions = payload->getAvailableActions();  		actionStates_.clear();  		if (payload->getStatus() == Command::Executing ) {  			actionStates_[Command::Cancel] = EnabledAndPresent; @@ -50,53 +52,44 @@ void OutgoingAdHocCommandSession::handleResponse(boost::shared_ptr<Command> payl  	}  } -bool OutgoingAdHocCommandSession::getIsMultiStage() { +bool OutgoingAdHocCommandSession::getIsMultiStage() const {  	return isMultiStage_;  }  void OutgoingAdHocCommandSession::start() { -	boost::shared_ptr<Payload> commandPayload(new Command(command_.getNode())); -	boost::shared_ptr<GenericRequest<Command> > commandRequest(new GenericRequest<Command>(IQ::Set, command_.getJID(), commandPayload, iqRouter_)); +	boost::shared_ptr<GenericRequest<Command> > commandRequest = boost::make_shared< GenericRequest<Command> >(IQ::Set, command_.getJID(), boost::make_shared<Command>(command_.getNode()), iqRouter_);  	commandRequest->onResponse.connect(boost::bind(&OutgoingAdHocCommandSession::handleResponse, this, _1, _2));  	commandRequest->send();  }  void OutgoingAdHocCommandSession::cancel() {  	if (!sessionID_.empty()) { -		boost::shared_ptr<Payload> commandPayload(new Command(command_.getNode(), sessionID_, Command::Cancel)); -		boost::shared_ptr<GenericRequest<Command> > commandRequest(new GenericRequest<Command>(IQ::Set, command_.getJID(), commandPayload, iqRouter_)); -		commandRequest->onResponse.connect(boost::bind(&OutgoingAdHocCommandSession::handleResponse, this, _1, _2)); -		commandRequest->send(); +		submitForm(Form::ref(), Command::Cancel);  	}  }  void OutgoingAdHocCommandSession::goBack() { -	boost::shared_ptr<Payload> commandPayload(new Command(command_.getNode(), sessionID_, Command::Prev)); -	boost::shared_ptr<GenericRequest<Command> > commandRequest(new GenericRequest<Command>(IQ::Set, command_.getJID(), commandPayload, iqRouter_)); -	commandRequest->onResponse.connect(boost::bind(&OutgoingAdHocCommandSession::handleResponse, this, _1, _2)); -	commandRequest->send(); +	submitForm(Form::ref(), Command::Prev);  }  void OutgoingAdHocCommandSession::complete(Form::ref form) { -	Command* command = new Command(command_.getNode(), sessionID_, Command::Complete); -	boost::shared_ptr<Payload> commandPayload(command); -	command->setForm(form); -	boost::shared_ptr<GenericRequest<Command> > commandRequest(new GenericRequest<Command>(IQ::Set, command_.getJID(), commandPayload, iqRouter_)); -	commandRequest->onResponse.connect(boost::bind(&OutgoingAdHocCommandSession::handleResponse, this, _1, _2)); -	commandRequest->send(); +	submitForm(form, Command::Complete);  }  void OutgoingAdHocCommandSession::goNext(Form::ref form) { -	Command* command = new Command(command_.getNode(), sessionID_, Command::Next); -	boost::shared_ptr<Payload> commandPayload(command); +	submitForm(form, Command::Next); +} + +void OutgoingAdHocCommandSession::submitForm(Form::ref form, Command::Action action) { +	boost::shared_ptr<Command> command(boost::make_shared<Command>(command_.getNode(), sessionID_, action));  	command->setForm(form); -	boost::shared_ptr<GenericRequest<Command> > commandRequest(new GenericRequest<Command>(IQ::Set, command_.getJID(), commandPayload, iqRouter_)); +	boost::shared_ptr<GenericRequest<Command> > commandRequest = boost::make_shared< GenericRequest<Command> >(IQ::Set, command_.getJID(), command, iqRouter_);  	commandRequest->onResponse.connect(boost::bind(&OutgoingAdHocCommandSession::handleResponse, this, _1, _2));  	commandRequest->send();  } -OutgoingAdHocCommandSession::ActionState OutgoingAdHocCommandSession::getActionState(Command::Action action) { -	return actionStates_[action]; +OutgoingAdHocCommandSession::ActionState OutgoingAdHocCommandSession::getActionState(Command::Action action) const { +	return get(actionStates_, action, Absent);  }  } diff --git a/Swiften/AdHoc/OutgoingAdHocCommandSession.h b/Swiften/AdHoc/OutgoingAdHocCommandSession.h index 1e0a42d..6b80300 100644 --- a/Swiften/AdHoc/OutgoingAdHocCommandSession.h +++ b/Swiften/AdHoc/OutgoingAdHocCommandSession.h @@ -56,7 +56,7 @@ namespace Swift {  			/**  			 * Is the form multi-stage?  			 */ -			bool getIsMultiStage(); +			bool getIsMultiStage() const;  			/**  			 * Emitted when the form for the next stage is available. @@ -75,9 +75,12 @@ namespace Swift {  			 * Use for Next, Prev, Cancel and Complete only.  			 * If no actions are available, the command has completed.  			 */ -			ActionState getActionState(Command::Action action); +			ActionState getActionState(Command::Action action) const; +  		private:  			void handleResponse(boost::shared_ptr<Command> payload, ErrorPayload::ref error); +			void submitForm(Form::ref, Command::Action action); +  		private:  			DiscoItems::Item command_;  			IQRouter* iqRouter_; diff --git a/Swiften/Base/Algorithm.h b/Swiften/Base/Algorithm.h index ea9530b..4d7f1de 100644 --- a/Swiften/Base/Algorithm.h +++ b/Swiften/Base/Algorithm.h @@ -93,6 +93,17 @@ namespace Swift {  		target.insert(target.end(), source.begin(), source.end());  	} +	template<typename A, typename B, typename C, typename D>  +	B get(const std::map<A, B, C, D>& map, const A& key, const B& defaultValue) { +		typename std::map<A, B, C, D>::const_iterator i = map.find(key); +		if (i != map.end()) { +			return i->second; +		} +		else { +			return defaultValue; +		} +	} +  	/*  	 * Functors  	 */ | 
 Swift
 Swift