Separated colors for encryption and signature.

Typically we have a good encryption and potentialyl a problem with the
signature. It's weird if there is a problem with the encryption indicated,
while there isn't actually one.

2 files changed, 33 insertions, 0 deletions

diff --git a/framework/src/domain/mime/partmodel.cpp b/framework/src/domain/mime/partmodel.cpp
index ca8856fd..02ef2f6d 100644
--- a/framework/src/domain/mime/partmodel.cpp
+++ b/framework/src/domain/mime/partmodel.cpp
@@ -85,6 +85,8 @@ QHash<int, QByteArray> PartModel::roleNames() const
     roles[IsEncryptedRole] = "encrypted";
     roles[IsSignedRole] = "signed";
     roles[SecurityLevelRole] = "securityLevel";
+    roles[EncryptionSecurityLevelRole] = "encryptionSecurityLevel";
+    roles[SignatureSecurityLevelRole] = "signatureSecurityLevel";
     roles[ErrorType] = "errorType";
     roles[ErrorString] = "errorString";
     roles[IsErrorRole] = "error";
@@ -281,6 +283,35 @@ QVariant PartModel::data(const QModelIndex &index, int role) const
                 //No info
                 return "unknown";
             }
+            case EncryptionSecurityLevelRole: {
+                auto encryption = messagePart->encryptionState();
+                bool messageIsEncrypted = encryption == MimeTreeParser::KMMsgPartiallyEncrypted ||
+                                          encryption == MimeTreeParser::KMMsgFullyEncrypted;
+                //All good
+                if (messageIsEncrypted) {
+                    return "good";
+                }
+                //No info
+                return "unknown";
+            }
+            case SignatureSecurityLevelRole: {
+                auto signature = messagePart->signatureState();
+                bool messageIsSigned = signature == MimeTreeParser::KMMsgPartiallySigned ||
+                                       signature == MimeTreeParser::KMMsgFullySigned;
+                if (messageIsSigned) {
+                    auto sigInfo = std::unique_ptr<SignatureInfo>{signatureInfo(messagePart)};
+                    if (!sigInfo->signatureIsGood) {
+                        if (sigInfo->keyMissing || sigInfo->keyExpired) {
+                            return "notsogood";
+                        }
+                        return "bad";
+                    }
+                    return "good";
+                }
+                //No info
+                return "unknown";
+
+            }
             case SignatureDetails:
                 return QVariant::fromValue(signatureInfo(messagePart));
             case EncryptionDetails:
diff --git a/framework/src/domain/mime/partmodel.h b/framework/src/domain/mime/partmodel.h
index 0c8a133d..945fda12 100644
--- a/framework/src/domain/mime/partmodel.h
+++ b/framework/src/domain/mime/partmodel.h
@@ -45,6 +45,8 @@ public:
         IsSignedRole,
         IsErrorRole,
         SecurityLevelRole,
+        EncryptionSecurityLevelRole,
+        SignatureSecurityLevelRole,
         SignatureDetails,
         EncryptionDetails,
         ErrorType,