Remove drafts after sending the message

2 files changed, 10 insertions, 1 deletions

diff --git a/framework/src/domain/composercontroller.cpp b/framework/src/domain/composercontroller.cpp
index de7ab4b1..60e955ea 100644
--- a/framework/src/domain/composercontroller.cpp
+++ b/framework/src/domain/composercontroller.cpp
@@ -342,9 +342,12 @@ void ComposerController::loadMessage(const QVariant &message, bool loadAsDraft)
     using namespace Sink;
     using namespace Sink::ApplicationDomain;
 
-    Query query(*message.value<Mail::Ptr>());
+    auto msg = message.value<Mail::Ptr>();
+    Q_ASSERT(msg);
+    Query query(*msg);
     query.request<Mail::MimeMessage>();
     Store::fetchOne<Mail>(query).then([this, loadAsDraft](const Mail &mail) {
+        mRemoveDraft = loadAsDraft;
         setExistingMail(mail);
 
         const auto mailData = KMime::CRLFtoLF(mail.getMimeMessage());
@@ -468,6 +471,11 @@ void ComposerController::send()
                     .then<void>([=] {
                         //Trigger a sync, but don't wait for it.
                         Store::synchronize(Sink::SyncScope{}.resourceFilter(resourceId)).exec();
+                        if (mRemoveDraft) {
+                            SinkLog() << "Removing draft message.";
+                            //Remove draft
+                            Store::remove(getExistingMail()).exec();
+                        }
                     });
             }
             SinkWarning() << "Failed to find a mailtransport resource";
diff --git a/framework/src/domain/composercontroller.h b/framework/src/domain/composercontroller.h
index df3c7b1b..0ace365b 100644
--- a/framework/src/domain/composercontroller.h
+++ b/framework/src/domain/composercontroller.h
@@ -103,4 +103,5 @@ private:
 
     QScopedPointer<Completer> mRecipientCompleter;
     QScopedPointer<Selector> mIdentitySelector;
+    bool mRemoveDraft = false;
 };