mirror of
https://github.com/arcan1s/queued.git
synced 2025-08-11 10:59:56 +00:00
some improvements
* add plugin tree * add emailnotify plugin template
This commit is contained in:
249
sources/plugins/emailnotify/QueuedEmailNotifyHelper.cpp
Normal file
249
sources/plugins/emailnotify/QueuedEmailNotifyHelper.cpp
Normal file
@ -0,0 +1,249 @@
|
||||
/*
|
||||
* Copyright (c) 2017 Queued team
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
* of this software and associated documentation files (the "Software"), to deal
|
||||
*
|
||||
* in the Software without restriction, including without limitation the rights
|
||||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
* copies of the Software, and to permit persons to whom the Software is
|
||||
* furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in
|
||||
* all copies or substantial portions of the Software.
|
||||
*/
|
||||
|
||||
|
||||
#include "QueuedEmailNotifyHelper.h"
|
||||
|
||||
#include <queued/Queued.h>
|
||||
|
||||
extern "C" {
|
||||
#include <curl/curl.h>
|
||||
}
|
||||
|
||||
|
||||
QueuedEmailNotifyHelper::QueuedEmailNotifyHelper(QObject *_parent)
|
||||
: QObject(_parent)
|
||||
{
|
||||
qCDebug(LOG_PL) << __PRETTY_FUNCTION__;
|
||||
}
|
||||
|
||||
|
||||
QString QueuedEmailNotifyHelper::curlEmail(const QString &_from)
|
||||
{
|
||||
qCDebug(LOG_PL) << "Convert email to curl format from" << _from;
|
||||
|
||||
return QString("<%1>").arg(_from);
|
||||
}
|
||||
|
||||
|
||||
char *QueuedEmailNotifyHelper::curlString(const QString &_source)
|
||||
{
|
||||
return strdup(_source.toStdString().c_str());
|
||||
}
|
||||
|
||||
|
||||
size_t QueuedEmailNotifyHelper::curlReadCallback(char *buffer, size_t size,
|
||||
size_t nitems, void *instream)
|
||||
{
|
||||
// FIXME not really best choice to use here
|
||||
auto text = reinterpret_cast<MailBody *>(instream);
|
||||
|
||||
// check sizes and whatever
|
||||
if ((size == 0) || (nitems == 0) || ((size * nitems) < 1))
|
||||
return 0;
|
||||
if (text->text.count() <= text->currentLine)
|
||||
return 0;
|
||||
|
||||
// get char* string from related qstring
|
||||
auto data = curlString(text->text.at(text->currentLine));
|
||||
size_t len = strlen(data);
|
||||
memcpy(buffer, data, len);
|
||||
text->currentLine++;
|
||||
|
||||
return len;
|
||||
}
|
||||
|
||||
|
||||
QString QueuedEmailNotifyHelper::from() const
|
||||
{
|
||||
return m_from;
|
||||
}
|
||||
|
||||
|
||||
bool QueuedEmailNotifyHelper::isInsecureCurl() const
|
||||
{
|
||||
return m_insecure;
|
||||
}
|
||||
|
||||
|
||||
bool QueuedEmailNotifyHelper::isSslEnabled() const
|
||||
{
|
||||
return m_ssl;
|
||||
}
|
||||
|
||||
|
||||
QString QueuedEmailNotifyHelper::password() const
|
||||
{
|
||||
return m_password;
|
||||
}
|
||||
|
||||
|
||||
int QueuedEmailNotifyHelper::port() const
|
||||
{
|
||||
return m_port;
|
||||
}
|
||||
|
||||
|
||||
QString QueuedEmailNotifyHelper::server() const
|
||||
{
|
||||
return m_server;
|
||||
}
|
||||
|
||||
|
||||
QString QueuedEmailNotifyHelper::username() const
|
||||
{
|
||||
return m_username;
|
||||
}
|
||||
|
||||
|
||||
void QueuedEmailNotifyHelper::setFrom(const QString &_from)
|
||||
{
|
||||
qCDebug(LOG_PL) << "Set from" << _from;
|
||||
|
||||
m_from = _from;
|
||||
}
|
||||
|
||||
|
||||
void QueuedEmailNotifyHelper::setInsecureCurl(const bool _insecureCurl)
|
||||
{
|
||||
qCDebug(LOG_PL) << "Set insecure curl" << _insecureCurl;
|
||||
|
||||
m_insecure = _insecureCurl;
|
||||
}
|
||||
|
||||
|
||||
void QueuedEmailNotifyHelper::setPassword(const QString &_password)
|
||||
{
|
||||
qCDebug(LOG_PL) << "Set password";
|
||||
|
||||
m_password = _password;
|
||||
}
|
||||
|
||||
|
||||
void QueuedEmailNotifyHelper::setPort(const int &_port)
|
||||
{
|
||||
qCDebug(LOG_PL) << "Set port" << _port;
|
||||
|
||||
m_port = _port;
|
||||
}
|
||||
|
||||
|
||||
void QueuedEmailNotifyHelper::setServer(const QString &_server)
|
||||
{
|
||||
qCDebug(LOG_PL) << "Set server" << _server;
|
||||
|
||||
m_server = _server;
|
||||
}
|
||||
|
||||
|
||||
void QueuedEmailNotifyHelper::setSslEnabled(const bool _sslEnabled)
|
||||
{
|
||||
qCDebug(LOG_PL) << "Set ssl enabled" << _sslEnabled;
|
||||
|
||||
m_ssl = _sslEnabled;
|
||||
}
|
||||
|
||||
|
||||
void QueuedEmailNotifyHelper::setUsername(const QString &_username)
|
||||
{
|
||||
qCDebug(LOG_PL) << "Set username" << _username;
|
||||
|
||||
m_username = _username;
|
||||
}
|
||||
|
||||
|
||||
void QueuedEmailNotifyHelper::sendEmail(const long long _id)
|
||||
{
|
||||
auto rcpt = getEmail(_id);
|
||||
if (rcpt.isEmpty())
|
||||
return;
|
||||
MailBody text = {getEmailText(_id, rcpt), 0};
|
||||
|
||||
CURL *curl = curl_easy_init();
|
||||
QString url = QString("%1:%2").arg(server()).arg(port());
|
||||
curl_easy_setopt(curl, CURLOPT_URL, curlString(url));
|
||||
|
||||
// ssl options
|
||||
if (isSslEnabled())
|
||||
curl_easy_setopt(curl, CURLOPT_USE_SSL, CURLUSESSL_ALL);
|
||||
if (isInsecureCurl())
|
||||
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L);
|
||||
|
||||
// auth if applicable
|
||||
if ((!username().isEmpty()) && (!password().isEmpty())) {
|
||||
curl_easy_setopt(curl, CURLOPT_USERNAME, curlString(username()));
|
||||
curl_easy_setopt(curl, CURLOPT_PASSWORD, curlString(password()));
|
||||
}
|
||||
|
||||
// from & to
|
||||
curl_easy_setopt(curl, CURLOPT_MAIL_FROM, curlString(curlEmail(from())));
|
||||
curl_slist *recipients = nullptr;
|
||||
recipients = curl_slist_append(recipients, curlString(curlEmail(rcpt)));
|
||||
curl_easy_setopt(curl, CURLOPT_MAIL_RCPT, recipients);
|
||||
|
||||
// mail body
|
||||
curl_easy_setopt(curl, CURLOPT_READFUNCTION,
|
||||
&QueuedEmailNotifyHelper::curlReadCallback);
|
||||
curl_easy_setopt(curl, CURLOPT_READDATA, &text);
|
||||
curl_easy_setopt(curl, CURLOPT_UPLOAD, 1L);
|
||||
|
||||
// send
|
||||
auto res = curl_easy_perform(curl);
|
||||
if (res != CURLE_OK)
|
||||
qCWarning(LOG_PL) << "Could not perform curl action"
|
||||
<< curl_easy_strerror(res);
|
||||
|
||||
// cleanup
|
||||
curl_slist_free_all(recipients);
|
||||
curl_easy_cleanup(curl);
|
||||
}
|
||||
|
||||
|
||||
QString QueuedEmailNotifyHelper::getEmail(const long long _id) const
|
||||
{
|
||||
qCDebug(LOG_PL) << "Get email for task ID" << _id;
|
||||
|
||||
auto task = QueuedCoreAdaptor::getTask(_id, "user");
|
||||
if (task.type() != Result::Content::Value) {
|
||||
qCWarning(LOG_LIB) << "Could not get task information" << _id;
|
||||
return "";
|
||||
}
|
||||
|
||||
auto userId = task.get().toLongLong();
|
||||
auto user = QueuedCoreAdaptor::getUser(userId, "email");
|
||||
if (user.type() != Result::Content::Value) {
|
||||
qCWarning(LOG_LIB) << "Could not get user information" << userId;
|
||||
return "";
|
||||
}
|
||||
|
||||
return user.get().toString();
|
||||
}
|
||||
|
||||
|
||||
QStringList QueuedEmailNotifyHelper::getEmailText(const long long _id,
|
||||
const QString &_to) const
|
||||
{
|
||||
qCDebug(LOG_PL) << "Get email text for user" << _to << "for task" << _id;
|
||||
|
||||
auto now
|
||||
= QDateTime::currentDateTimeUtc().toString(Qt::DateFormat::RFC2822Date);
|
||||
|
||||
return {QString("Date: %1\r\n").arg(now),
|
||||
QString("To: %1\r\n").arg(curlEmail(_to)),
|
||||
QString("From: %1\r\n").arg(curlEmail(from())),
|
||||
// message-id?
|
||||
QString("Subject: %1\r\n").arg("Job %1 done").arg(_id), "\r\n",
|
||||
QString("Job %1 done\r\n").arg(_id)};
|
||||
}
|
Reference in New Issue
Block a user