Adds imap response string to custom command operation

3 files changed, 14 insertions, 5 deletions

diff --git a/src/async/imap/MCIMAPAsyncConnection.cpp b/src/async/imap/MCIMAPAsyncConnection.cpp
index c03422ff..0dc29ee0 100755
--- a/src/async/imap/MCIMAPAsyncConnection.cpp
+++ b/src/async/imap/MCIMAPAsyncConnection.cpp
@@ -38,6 +38,7 @@
 #include "MCIMAPAsyncSession.h"
 #include "MCConnectionLogger.h"
 #include "MCIMAPMessageRenderingOperation.h"
+#include "MCIMAPCustomCommandOperation.h"
 #include "MCIMAPIdentity.h"
 
 using namespace mailcore;
diff --git a/src/async/imap/MCIMAPCustomCommandOperation.cpp b/src/async/imap/MCIMAPCustomCommandOperation.cpp
index 692c8f7d..61e0e655 100644
--- a/src/async/imap/MCIMAPCustomCommandOperation.cpp
+++ b/src/async/imap/MCIMAPCustomCommandOperation.cpp
@@ -16,21 +16,27 @@ using namespace mailcore;
 IMAPCustomCommandOperation::IMAPCustomCommandOperation()
 {
     mCustomCommand = NULL;
+    mResponse = NULL;
 }
 
 IMAPCustomCommandOperation::~IMAPCustomCommandOperation()
 {
     MC_SAFE_RELEASE(mCustomCommand);
+    MC_SAFE_RELEASE(mResponse);
 }
 
-void IMAPCustomCommandOperation::setCustomCommand(mailcore::String * command)
+void IMAPCustomCommandOperation::setCustomCommand(String * command)
 {
-    mCustomCommand = command;
+    MC_SAFE_REPLACE_COPY(String, mCustomCommand, command);
 }
 
-void IMAPCustomCommandOperation::main()
+String * IMAPCustomCommandOperation::response()
 {
+    return mResponse;
+}
 
-    session()->session()->sendCustomCommand(mCustomCommand);
-
+void IMAPCustomCommandOperation::main()
+{
+    mResponse = session()->session()->sendCustomCommand(mCustomCommand);
+    MC_SAFE_RETAIN(mResponse);
 }
diff --git a/src/async/imap/MCIMAPCustomCommandOperation.h b/src/async/imap/MCIMAPCustomCommandOperation.h
index 83e6c11f..a2ce20fe 100644
--- a/src/async/imap/MCIMAPCustomCommandOperation.h
+++ b/src/async/imap/MCIMAPCustomCommandOperation.h
@@ -22,12 +22,14 @@ namespace mailcore {
         virtual ~IMAPCustomCommandOperation();
 
         virtual void setCustomCommand(String *command);
+        virtual String * response();
 
     public: // subclass behavior
         virtual void main();
 
     private:
         String * mCustomCommand;
+        String * mResponse;
     };
 
 }