Prepping encryption

5 files changed, 118 insertions, 51 deletions

diff --git a/components/kube/contents/ui/AddresseeListEditor.qml b/components/kube/contents/ui/AddresseeListEditor.qml
index adb2f785..91431b1e 100644
--- a/components/kube/contents/ui/AddresseeListEditor.qml
+++ b/components/kube/contents/ui/AddresseeListEditor.qml
@@ -52,21 +52,37 @@ FocusScope {
                 height: Kube.Units.gridUnit + Kube.Units.smallSpacing * 2 //smallSpacing for padding
                 width: parent.width
                 color: Kube.Colors.buttonColor
-                Kube.Label {
+                Row {
                     anchors {
                         top: parent.top
+                        bottom: parent.bottom
                         left: parent.left
-                        right: button.left
+                        right: removeButton.left
                         margins: Kube.Units.smallSpacing
                     }
-                    text: display
-                    elide: Text.ElideRight
+                    spacing: Kube.Units.smallSpacing
+                    Kube.Label {
+                        id: label
+                        anchors {
+                            top: parent.top
+                        }
+                        text: model.addresseeName
+                        elide: Text.ElideRight
+                    }
+                    Kube.Icon {
+                        anchors {
+                            top: parent.top
+                        }
+                        height: Kube.Units.gridUnit
+                        width: height
+                        iconName: model.keyFound ? Kube.Icons.secure: Kube.Icons.insecure
+                    }
                 }
                 Kube.IconButton {
-                    id: button
+                    id: removeButton
                     anchors {
-                        verticalCenter: parent.verticalCenter
                         right: parent.right
+                        verticalCenter: parent.verticalCenter
                         margins: Kube.Units.smallSpacing
                     }
                     height: Kube.Units.gridUnit
diff --git a/components/kube/contents/ui/ComposerView.qml b/components/kube/contents/ui/ComposerView.qml
index a4796932..29141f28 100644
--- a/components/kube/contents/ui/ComposerView.qml
+++ b/components/kube/contents/ui/ComposerView.qml
@@ -42,6 +42,8 @@ Kube.View {
         Kube.ComposerController {
             id: composerController
             htmlBody: html.checked
+            sign: signCheckbox.checked
+            encrypt: encryptCheckbox.checked
             onDone: Kube.Fabric.postMessage(Kube.Messages.componentDone, {})
         }
     ]
@@ -443,18 +445,20 @@ Kube.View {
             }
 
             RowLayout {
-                //FIXME: hide until it does something
-                visible: false
-                Kube.CheckBox {}
+                Kube.CheckBox {
+                    id: encryptCheckbox
+                    checked: composerController.encrypt
+                }
                 Kube.Label {
                     text: qsTr("encrypt")
                 }
             }
 
             RowLayout {
-                //FIXME: hide until it does something
-                visible: false
-                Kube.CheckBox {}
+                Kube.CheckBox {
+                    id: signCheckbox
+                    checked: composerController.sign
+                }
                 Kube.Label {
                     text: qsTr("sign")
                 }
diff --git a/framework/qml/Icons.qml b/framework/qml/Icons.qml
index 8cb28e86..6ecc069c 100644
--- a/framework/qml/Icons.qml
+++ b/framework/qml/Icons.qml
@@ -58,6 +58,8 @@ Item {
     property string checkbox: "checkbox"
     property string password_show: "password-show-on"
     property string password_hide: "password-show-off"
+    property string secure: "document-encrypt"
+    property string insecure: "document-decrypt"
 
     property string addNew: "list-add"
     property string remove: "kube-list-remove-inverted"
diff --git a/framework/src/domain/composercontroller.cpp b/framework/src/domain/composercontroller.cpp
index 258c25d5..2a82f0e1 100644
--- a/framework/src/domain/composercontroller.cpp
+++ b/framework/src/domain/composercontroller.cpp
@@ -87,6 +87,56 @@ public:
     }
 };
 
+class AddresseeModel : public QStandardItemModel
+{
+public:
+    AddresseeModel()
+        :QStandardItemModel()
+    {
+        setItemRoleNames({{ComposerController::AddresseeNameRole, "addresseeName"},
+                          {ComposerController::KeyFoundRole, "keyFound"},
+                          {ComposerController::KeyRole, "key"}});
+    }
+
+    void add(const QString &addressee)
+    {
+        auto item = new QStandardItem;
+        item->setData(addressee, ComposerController::AddresseeNameRole);
+        item->setData(false, ComposerController::KeyFoundRole);
+        appendRow(QList<QStandardItem*>() << item);
+    }
+
+    void remove(const QString &addressee)
+    {
+        auto root = invisibleRootItem();
+        for (int row = 0; row < root->rowCount(); row++) {
+            if (root->child(row, 0)->data(ComposerController::AddresseeNameRole).toString() == addressee) {
+                root->removeRow(row);
+                return;
+            }
+        }
+    }
+
+    void setStringList(const QStringList &list)
+    {
+        clear();
+        for (const auto &s : list) {
+            add(s);
+        }
+    }
+
+    QStringList stringList() const
+    {
+        QStringList list;
+        auto root = invisibleRootItem();
+        for (int row = 0; row < root->rowCount(); row++) {
+            list << root->child(row, 0)->data(ComposerController::AddresseeNameRole).toString();
+        }
+        return list;
+    }
+
+};
+
 
 ComposerController::ComposerController()
     : Kube::Controller(),
@@ -94,9 +144,9 @@ ComposerController::ComposerController()
     action_saveAsDraft{new Kube::ControllerAction{this, &ComposerController::saveAsDraft}},
     mRecipientCompleter{new RecipientCompleter},
     mIdentitySelector{new IdentitySelector{*this}},
-    mToModel{new QStringListModel},
-    mCcModel{new QStringListModel},
-    mBccModel{new QStringListModel},
+    mToModel{new AddresseeModel},
+    mCcModel{new AddresseeModel},
+    mBccModel{new AddresseeModel},
     mAttachmentModel{new QStandardItemModel}
 {
     mAttachmentModel->setItemRoleNames({{NameRole, "name"},
@@ -128,9 +178,9 @@ void ComposerController::clear()
     Controller::clear();
     //Reapply account and identity from selection
     mIdentitySelector->reapplyCurrentIndex();
-    mToModel->setStringList({});
-    mCcModel->setStringList({});
-    mBccModel->setStringList({});
+    mToModel->clear();
+    mCcModel->clear();
+    mBccModel->clear();
 }
 
 QAbstractItemModel *ComposerController::toModel() const
@@ -140,17 +190,13 @@ QAbstractItemModel *ComposerController::toModel() const
 
 void ComposerController::addTo(const QString &s)
 {
-    auto list = mToModel->stringList();
-    list.append(s);
-    mToModel->setStringList(list);
+    mToModel->add(s);
     updateSendAction();
 }
 
 void ComposerController::removeTo(const QString &s)
 {
-    auto list = mToModel->stringList();
-    list.removeAll(s);
-    mToModel->setStringList(list);
+    mToModel->remove(s);
     updateSendAction();
 }
 
@@ -161,17 +207,13 @@ QAbstractItemModel *ComposerController::ccModel() const
 
 void ComposerController::addCc(const QString &s)
 {
-    auto list = mCcModel->stringList();
-    list.append(s);
-    mCcModel->setStringList(list);
+    mCcModel->add(s);
     updateSendAction();
 }
 
 void ComposerController::removeCc(const QString &s)
 {
-    auto list = mCcModel->stringList();
-    list.removeAll(s);
-    mCcModel->setStringList(list);
+    mCcModel->remove(s);
     updateSendAction();
 }
 
@@ -182,17 +224,13 @@ QAbstractItemModel *ComposerController::bccModel() const
 
 void ComposerController::addBcc(const QString &s)
 {
-    auto list = mBccModel->stringList();
-    list.append(s);
-    mBccModel->setStringList(list);
+    mBccModel->add(s);
     updateSendAction();
 }
 
 void ComposerController::removeBcc(const QString &s)
 {
-    auto list = mBccModel->stringList();
-    list.removeAll(s);
-    mBccModel->setStringList(list);
+    mBccModel->remove(s);
     updateSendAction();
 }
 
@@ -239,7 +277,6 @@ void ComposerController::removeAttachment(const QUrl &url)
     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;
@@ -418,17 +455,15 @@ KMime::Message::Ptr ComposerController::assembleMessage()
 
     QList<Attachment> attachments;
     auto root = mAttachmentModel->invisibleRootItem();
-    if (root->hasChildren()) {
-        for (int row = 0; row < root->rowCount(); row++) {
-            auto item = root->child(row, 0);
-            attachments << Attachment{
-                item->data(NameRole).toString(),
-                item->data(FilenameRole).toString(),
-                item->data(MimeTypeRole).toByteArray(),
-                item->data(InlineRole).toBool(),
-                item->data(ContentRole).toByteArray()
-            };
-        }
+    for (int row = 0; row < root->rowCount(); row++) {
+        auto item = root->child(row, 0);
+        attachments << Attachment{
+            item->data(NameRole).toString(),
+            item->data(FilenameRole).toString(),
+            item->data(MimeTypeRole).toByteArray(),
+            item->data(InlineRole).toBool(),
+            item->data(ContentRole).toByteArray()
+        };
     }
     return MailTemplates::createMessage(mExistingMessage, mToModel->stringList(), mCcModel->stringList(), mBccModel->stringList(), getIdentity(), getSubject(), getBody(), getHtmlBody(), attachments);
 }
diff --git a/framework/src/domain/composercontroller.h b/framework/src/domain/composercontroller.h
index 784591eb..1a355b9a 100644
--- a/framework/src/domain/composercontroller.h
+++ b/framework/src/domain/composercontroller.h
@@ -44,6 +44,8 @@ namespace KMime {
 class Message;
 }
 
+class AddresseeModel;
+
 class ComposerController : public Kube::Controller
 {
     Q_OBJECT
@@ -52,6 +54,8 @@ class ComposerController : public Kube::Controller
     KUBE_CONTROLLER_PROPERTY(QString, Subject, subject)
     KUBE_CONTROLLER_PROPERTY(QString, Body, body)
     KUBE_CONTROLLER_PROPERTY(bool, HtmlBody, htmlBody)
+    KUBE_CONTROLLER_PROPERTY(bool, Encrypt, encrypt)
+    KUBE_CONTROLLER_PROPERTY(bool, Sign, sign)
 
     //Set by identitySelector
     KUBE_CONTROLLER_PROPERTY(KMime::Types::Mailbox, Identity, identity)
@@ -74,6 +78,12 @@ class ComposerController : public Kube::Controller
     KUBE_CONTROLLER_ACTION(saveAsDraft)
 
 public:
+    enum AddresseeRoles {
+        KeyFoundRole = Qt::UserRole + 1,
+        KeyRole,
+        AddresseeNameRole
+    };
+
     explicit ComposerController();
 
     Completer *recipientCompleter() const;
@@ -121,8 +131,8 @@ private:
 
     QScopedPointer<Completer> mRecipientCompleter;
     QScopedPointer<Selector> mIdentitySelector;
-    QScopedPointer<QStringListModel> mToModel;
-    QScopedPointer<QStringListModel> mCcModel;
-    QScopedPointer<QStringListModel> mBccModel;
+    QSharedPointer<AddresseeModel> mToModel;
+    QSharedPointer<AddresseeModel> mCcModel;
+    QSharedPointer<AddresseeModel> mBccModel;
     QScopedPointer<QStandardItemModel> mAttachmentModel;
 };