File indexing completed on 2024-06-23 05:18:29

 /*
   SPDX-FileCopyrightText: 2021 Sandro Knauß <sknauss@kde.org>
 
   SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "draftstatus.h"
 
 #include <MessageCore/AutocryptUtils>
 
 using namespace MessageComposer;
 
 void MessageComposer::removeDraftCryptoHeaders(const KMime::Message::Ptr &msg)
 {
     DraftEncryptionState(msg).removeState();
     DraftSignatureState(msg).removeState();
     DraftCryptoMessageFormatState(msg).removeState();
 }
 
 DraftEncryptionState::DraftEncryptionState(const KMime::Message::Ptr &msg)
     : mMsg(msg)
 {
 }
 
 void DraftEncryptionState::setState(bool encrypt)
 {
     auto hdr = new KMime::Headers::Generic("X-KMail-EncryptActionEnabled");
     hdr->fromUnicodeString(encrypt ? QStringLiteral("true") : QStringLiteral("false"), "utf-8");
     mMsg->setHeader(hdr);
 }
 
 void DraftEncryptionState::removeState()
 {
     mMsg->removeHeader("X-KMail-EncryptActionEnabled");
 }
 
 bool DraftEncryptionState::encryptionState() const
 {
     if (isDefined()) {
         auto hdr = mMsg->headerByType("X-KMail-EncryptActionEnabled");
         return hdr->as7BitString(false).contains("true");
     }
 
     return false;
 }
 
 bool DraftEncryptionState::isDefined() const
 {
     return mMsg->hasHeader("X-KMail-EncryptActionEnabled");
 }
 
 DraftSignatureState::DraftSignatureState(const KMime::Message::Ptr &msg)
     : mMsg(msg)
 {
 }
 
 void DraftSignatureState::setState(bool sign)
 {
     auto hdr = new KMime::Headers::Generic("X-KMail-SignatureActionEnabled");
     hdr->fromUnicodeString(sign ? QStringLiteral("true") : QStringLiteral("false"), "utf-8");
     mMsg->setHeader(hdr);
 }
 
 void DraftSignatureState::removeState()
 {
     mMsg->removeHeader("X-KMail-SignatureActionEnabled");
 }
 
 bool DraftSignatureState::signState() const
 {
     if (isDefined()) {
         auto hdr = mMsg->headerByType("X-KMail-SignatureActionEnabled");
         return hdr->as7BitString(false).contains("true");
     }
 
     return false;
 }
 
 bool DraftSignatureState::isDefined() const
 {
     return mMsg->hasHeader("X-KMail-SignatureActionEnabled");
 }
 
 DraftCryptoMessageFormatState::DraftCryptoMessageFormatState(const KMime::Message::Ptr &msg)
     : mMsg(msg)
 {
 }
 
 void DraftCryptoMessageFormatState::setState(Kleo::CryptoMessageFormat cryptoMessageFormat)
 {
     auto hdr = new KMime::Headers::Generic("X-KMail-CryptoMessageFormat");
     hdr->fromUnicodeString(QString::number(cryptoMessageFormat), "utf-8");
     mMsg->setHeader(hdr);
 }
 
 void DraftCryptoMessageFormatState::removeState()
 {
     mMsg->removeHeader("X-KMail-CryptoMessageFormat");
 }
 
 Kleo::CryptoMessageFormat DraftCryptoMessageFormatState::cryptoMessageFormatState() const
 {
     if (isDefined()) {
         auto hdr = mMsg->headerByType("X-KMail-CryptoMessageFormat");
         return static_cast<Kleo::CryptoMessageFormat>(hdr->asUnicodeString().toInt());
     }
 
     return Kleo::AutoFormat;
 }
 
 bool DraftCryptoMessageFormatState::isDefined() const
 {
     return mMsg->hasHeader("X-KMail-CryptoMessageFormat");
 }