1 files changed, 102 insertions, 0 deletions
diff --git a/Swiften/AdHoc/OutgoingAdHocCommandSession.cpp b/Swiften/AdHoc/OutgoingAdHocCommandSession.cpp
new file mode 100644
index 0000000..da7f042
--- /dev/null
+++ b/Swiften/AdHoc/OutgoingAdHocCommandSession.cpp
@@ -0,0 +1,102 @@
+/*
+ * Copyright (c) 2010-2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include <Swiften/AdHoc/OutgoingAdHocCommandSession.h>
+
+#include <boost/bind.hpp>
+
+#include <Swiften/Queries/GenericRequest.h>
+
+namespace Swift {
+OutgoingAdHocCommandSession::OutgoingAdHocCommandSession(const DiscoItems::Item& command, AdHocCommandWindowFactory* /*factory*/, IQRouter* iqRouter) : command_(command), iqRouter_(iqRouter), isMultiStage_(false) {
+
+}
+
+void OutgoingAdHocCommandSession::handleResponse(boost::shared_ptr<Command> payload, ErrorPayload::ref error) {
+	if (error) {
+		onError(error);
+	} else {
+		const std::vector<Command::Action> actions = payload->getAvailableActions();
+		actionStates_.clear();
+		if (payload->getStatus() == Command::Executing ) {
+			actionStates_[Command::Cancel] = EnabledAndPresent;
+			actionStates_[Command::Complete] = Present;
+			if (std::find(actions.begin(), actions.end(), Command::Complete) != actions.end()) {
+				actionStates_[Command::Complete] = EnabledAndPresent;
+			}
+
+			if (getIsMultiStage()) {
+				actionStates_[Command::Next] = Present;
+				actionStates_[Command::Prev] = Present;
+			}
+
+		if (std::find(actions.begin(), actions.end(), Command::Next) != actions.end()) {
+				actionStates_[Command::Next] = EnabledAndPresent;
+			}
+		if (std::find(actions.begin(), actions.end(), Command::Prev) != actions.end()) {
+				actionStates_[Command::Prev] = EnabledAndPresent;
+			}
+		}
+
+		sessionID_ = payload->getSessionID();
+		if (std::find(actions.begin(), actions.end(), Command::Next) != actions.end()
+				|| std::find(actions.begin(), actions.end(), Command::Prev) != actions.end()) {
+			isMultiStage_ = true;
+		}
+		onNextStageReceived(payload);
+	}
+}
+
+bool OutgoingAdHocCommandSession::getIsMultiStage() {
+	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_));
+	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();
+	}
+}
+
+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();
+}
+
+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();
+}
+
+void OutgoingAdHocCommandSession::goNext(Form::ref form) {
+	Command* command = new Command(command_.getNode(), sessionID_, Command::Next);
+	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();
+}
+
+OutgoingAdHocCommandSession::ActionState OutgoingAdHocCommandSession::getActionState(Command::Action action) {
+	return actionStates_[action];
+}
+
+}