viewemail.cpp

/*
 * SPDX-FileCopyrightText: (C) 2015-2022 Daniel Nicoletti <dantti12@gmail.com>
 * SPDX-License-Identifier: BSD-3-Clause
 */
#include "viewemail_p.h"

#include <Cutelyst/application.h>
#include <Cutelyst/context.h>
#include <Cutelyst/response.h>
#include <SimpleMail/emailaddress.h>
#include <SimpleMail/mimemessage.h>
#include <SimpleMail/mimetext.h>
#include <SimpleMail/serverreply.h>

#include <QtCore/QLoggingCategory>

Q_LOGGING_CATEGORY(CUTELYST_VIEW_EMAIL, "cutelyst.view.email", QtWarningMsg)

using namespace Cutelyst;
using namespace SimpleMail;
using namespace Qt::StringLiterals;

ViewEmail::ViewEmail(QObject *parent, const QString &name)
    : View(new ViewEmailPrivate, parent, name)
{
    initSender();
}

QString ViewEmail::stashKey() const
{
    Q_D(const ViewEmail);
    return d->stashKey;
}

void ViewEmail::setStashKey(const QString &stashKey)
{
    Q_D(ViewEmail);
    d->stashKey = stashKey;
    Q_EMIT changed();
}

QByteArray ViewEmail::defaultContentType() const
{
    Q_D(const ViewEmail);
    return d->defaultContentType;
}

void ViewEmail::setDefaultContentType(const QByteArray &contentType)
{
    Q_D(ViewEmail);
    d->defaultContentType = contentType;
    Q_EMIT changed();
}

QByteArray ViewEmail::defaultCharset() const
{
    Q_D(const ViewEmail);
    return d->defaultCharset;
}

void ViewEmail::setDefaultCharset(const QByteArray &charset)
{
    Q_D(ViewEmail);
    d->defaultCharset = charset;
    Q_EMIT changed();
}

QByteArray ViewEmail::defaultEncoding() const
{
    Q_D(const ViewEmail);
    return d->defaultEncoding;
}

void ViewEmail::setDefaultEncoding(const QByteArray &encoding)
{
    Q_D(ViewEmail);
    d->defaultEncoding = encoding;
    Q_EMIT changed();
}

QString ViewEmail::senderHost() const
{
    Q_D(const ViewEmail);
    return d->server->host();
}

void ViewEmail::setSenderHost(const QString &host)
{
    Q_D(ViewEmail);
    d->server->setHost(host);
}

int ViewEmail::senderPort() const
{
    Q_D(const ViewEmail);
    return d->server->port();
}

void ViewEmail::setSenderPort(int port)
{
    Q_D(ViewEmail);
    d->server->setPort(quint16(port));
}

ViewEmail::ConnectionType ViewEmail::senderConnectionType() const
{
    Q_D(const ViewEmail);
    return static_cast<ViewEmail::ConnectionType>(d->server->connectionType());
}

void ViewEmail::setSenderConnectionType(ViewEmail::ConnectionType ct)
{
    Q_D(ViewEmail);
    d->server->setConnectionType(static_cast<Server::ConnectionType>(ct));
}

ViewEmail::AuthMethod ViewEmail::senderAuthMethod() const
{
    Q_D(const ViewEmail);
    return static_cast<ViewEmail::AuthMethod>(d->server->authMethod());
}

void ViewEmail::setSenderAuthMethod(ViewEmail::AuthMethod method)
{
    Q_D(ViewEmail);
    d->server->setAuthMethod(static_cast<Server::AuthMethod>(method));
}

QString ViewEmail::senderUser() const
{
    Q_D(const ViewEmail);
    return d->server->username();
}

void ViewEmail::setSenderUser(const QString &user)
{
    Q_D(ViewEmail);
    d->server->setUsername(user);
}

QString ViewEmail::senderPassword() const
{
    Q_D(const ViewEmail);
    return d->server->password();
}

void ViewEmail::setSenderPassword(const QString &password)
{
    Q_D(ViewEmail);
    d->server->setPassword(password);
}

QByteArray ViewEmail::render(Context *c) const
{
    Q_D(const ViewEmail);
    QVariantHash email = c->stash(d->stashKey).toHash();
    if (email.isEmpty()) {
        c->appendError(QStringLiteral(
            "Cannot render template, template name or template stash key not defined"));
        return {};
    }

    MimeMessage message;

    QVariant value;
    value = email.value(u"to"_s);
    if (value.typeId() == QMetaType::QString && !value.toString().isEmpty()) {
        message.addTo(SimpleMail::EmailAddress{value.toString()});
    } else if (value.typeId() == QMetaType::QStringList) {
        const auto rcpts = value.toStringList();
        for (const QString &rcpt : rcpts) {
            message.addTo(SimpleMail::EmailAddress{rcpt});
        }
    }

    value = email.value(u"cc"_s);
    if (value.typeId() == QMetaType::QString && !value.toString().isEmpty()) {
        message.addCc(SimpleMail::EmailAddress{value.toString()});
    } else if (value.typeId() == QMetaType::QStringList) {
        const auto rcpts = value.toStringList();
        for (const QString &rcpt : rcpts) {
            message.addCc(SimpleMail::EmailAddress{rcpt});
        }
    }

    value = email.value(u"bcc"_s);
    if (value.typeId() == QMetaType::QString && !value.toString().isEmpty()) {
        message.addBcc(SimpleMail::EmailAddress{value.toString()});
    } else if (value.typeId() == QMetaType::QStringList) {
        const auto rcpts = value.toStringList();
        for (const QString &rcpt : rcpts) {
            message.addBcc(SimpleMail::EmailAddress{rcpt});
        }
    }

    message.setSender(SimpleMail::EmailAddress{email.value(u"from"_s).toString()});
    message.setSubject(email.value(u"subject"_s).toString());

    QVariant body  = email.value(u"body"_s);
    QVariant parts = email.value(u"parts"_s);
    if (body.isNull() && parts.isNull()) {
        c->appendError(u"Can't send email without parts or body, check stash"_s);
        return {};
    }

    if (!parts.isNull()) {
        const QVariantList partsVariant = parts.toList();
        for (const QVariant &part : partsVariant) {
            auto mime = part.value<std::shared_ptr<MimePart>>();
            if (mime) {
                message.addPart(mime);
            } else {
                qCCritical(CUTELYST_VIEW_EMAIL) << "Failed to cast MimePart";
            }
        }

        auto contentTypeIt = email.constFind(u"content_type"_s);
        if (contentTypeIt != email.constEnd() && !contentTypeIt.value().isNull() &&
            !contentTypeIt.value().toString().isEmpty()) {
            const QByteArray contentType = contentTypeIt.value().toString().toLatin1();
            qCDebug(CUTELYST_VIEW_EMAIL) << "Using specified content_type" << contentType;
            message.getContent().setContentType(contentType);
        } else if (!d->defaultContentType.isEmpty()) {
            qCDebug(CUTELYST_VIEW_EMAIL) << "Using default content_type" << d->defaultContentType;
            message.getContent().setContentType(d->defaultContentType);
        }
    } else {
        auto part = std::make_shared<MimeText>(body.toString());
        d->setupAttributes(part, email);
        message.setContent(part);
    }

    ServerReply *reply = d->server->sendMail(message);
    connect(reply, &ServerReply::finished, reply, &ServerReply::deleteLater);

    return {};
}

ViewEmail::ViewEmail(ViewEmailPrivate *d, QObject *parent, const QString &name)
    : View(d, parent, name)
{
    initSender();
}

void ViewEmail::initSender()
{
    Q_D(ViewEmail);
    d->server = new Server(this);

    QVariantHash config;
    const auto app = qobject_cast<Application *>(parent());
    if (app) {
        config = app->config(u"VIEW_EMAIL"_s).toHash();
    }

    d->stashKey = config.value(u"stash_key"_s, u"email"_s).toString();

    if (!config.value(u"sender_host"_s).isNull()) {
        d->server->setHost(config.value(u"sender_host"_s).toString());
    }
    if (!config.value(u"sender_port"_s).isNull()) {
        d->server->setPort(quint16(config.value(u"sender_port"_s).toInt()));
    }
    if (!config.value(u"sender_username"_s).isNull()) {
        d->server->setUsername(config.value(u"sender_username"_s).toString());
    }
    if (!config.value(u"sender_password"_s).isNull()) {
        d->server->setPassword(config.value(u"sender_password"_s).toString());
    }
}

void ViewEmailPrivate::setupAttributes(std::shared_ptr<MimePart> part,
                                       const QVariantHash &attrs) const
{
    auto contentTypeIt = attrs.constFind(u"content_type"_s);
    if (contentTypeIt != attrs.constEnd() && !contentTypeIt.value().isNull() &&
        !contentTypeIt.value().toString().isEmpty()) {
        const QByteArray contentType = contentTypeIt.value().toString().toLatin1();
        qCDebug(CUTELYST_VIEW_EMAIL) << "Using specified content_type" << contentType;
        part->setContentType(contentType);
    } else if (!defaultContentType.isEmpty()) {
        qCDebug(CUTELYST_VIEW_EMAIL) << "Using default content_type" << defaultContentType;
        part->setContentType(defaultContentType);
    }

    auto charsetIt = attrs.constFind(u"charset"_s);
    if (charsetIt != attrs.constEnd() && !charsetIt.value().isNull() &&
        !charsetIt.value().toString().isEmpty()) {
        const QByteArray charset = charsetIt.value().toString().toLatin1();
        qCDebug(CUTELYST_VIEW_EMAIL) << "Using specified charset" << charset;
        part->setCharset(charset);
    } else if (!defaultCharset.isEmpty()) {
        qCDebug(CUTELYST_VIEW_EMAIL) << "Using default charset" << defaultCharset;
        part->setCharset(defaultCharset);
    }

    auto encodingIt = attrs.constFind(u"encoding"_s);
    if (encodingIt != attrs.constEnd() && !encodingIt.value().isNull() &&
        !encodingIt.value().toString().isEmpty()) {
        const QByteArray encoding = encodingIt.value().toString().toLatin1();
        qCDebug(CUTELYST_VIEW_EMAIL) << "Using specified encoding" << encoding;
        setupEncoding(part, encoding);
    } else if (!defaultEncoding.isEmpty()) {
        qCDebug(CUTELYST_VIEW_EMAIL) << "Using default charset" << defaultEncoding;
        setupEncoding(part, defaultEncoding);
    }
}

void ViewEmailPrivate::setupEncoding(std::shared_ptr<MimePart> part,
                                     const QByteArray &encoding) const
{
    if (encoding == "7bit") {
        part->setEncoding(MimePart::_7Bit);
    } else if (encoding == "8bit") {
        part->setEncoding(MimePart::_8Bit);
    } else if (encoding == "base64") {
        part->setEncoding(MimePart::Base64);
    } else if (encoding == "quoted-printable") {
        part->setEncoding(MimePart::QuotedPrintable);
    } else {
        qCCritical(CUTELYST_VIEW_EMAIL) << "Unknown encoding" << encoding;
    }
}

#include "moc_viewemail.cpp"