Attachment removal

We'd rather do something via the model index instead of requering an
additional role, but this works for the time being.

1 files changed, 14 insertions, 1 deletions

diff --git a/framework/src/domain/composercontroller.cpp b/framework/src/domain/composercontroller.cpp
index 31cb896d..9e006af4 100644
--- a/framework/src/domain/composercontroller.cpp
+++ b/framework/src/domain/composercontroller.cpp
@@ -97,6 +97,7 @@ ComposerController::ComposerController()
                                         {MimeTypeRole, "mimetype"},
                                         {DescriptionRole, "description"},
                                         {IconNameRole, "iconName"},
+                                        {UrlRole, "url"},
                                         {InlineRole, "inline"}});
     updateSaveAsDraftAction();
     // mSendAction->monitorProperty<To>();
@@ -218,13 +219,25 @@ void ComposerController::addAttachment(const QUrl &url)
         item->setData(fileInfo.fileName(), FilenameRole);
         item->setData(false, InlineRole);
         item->setData(mimeType.iconName(), IconNameRole);
+        item->setData(url, UrlRole);
         item->setData(data, ContentRole);
         mAttachmentModel->appendRow(item);
     }
 }
 
-void ComposerController::removeAttachment(const QString &s)
+void ComposerController::removeAttachment(const QUrl &url)
 {
+    auto root = mAttachmentModel->invisibleRootItem();
+    if (root->hasChildren()) {
+        for (int row = 0; row < root->rowCount(); row++) {
+            auto item = root->child(row, 0);
+            const auto url = item->data(UrlRole).toUrl();
+            if (url == item->data(UrlRole).toUrl()) {
+                root->removeRow(row);
+                return;
+            }
+        }
+    }
 }
 
 Completer *ComposerController::recipientCompleter() const