#include #include "sender.hpp" #include "builder/date.hpp" #include "builder/mail_to.hpp" #include "builder/cc.hpp" #include "builder/subject.hpp" #include "builder/text.hpp" #include "builder/from.hpp" #include "logger/logger_set.hpp" namespace smtp::message { static std::string mText; // // Constructors/Destructors // Sender::Sender( manage::Settings const& settings_storage, manage::Mail& mail_to, thread::Queue& message_queue ) : mSettingsStorage( settings_storage ) , mMailTo( mail_to ) , mMessageQueue( message_queue ) { } // //Public methods // void Sender::Send() { static const std::string METHOD_NAME = "Send message"; CURLcode result = CURLE_OK; curl_slist* recipients = nullptr; WriteThis upload_ctx{}; auto curl = curl_easy_init(); //TODO сделать инициализацию через регистратор инициализаторов while(1) { auto message = mMessageQueue.WaitAndPop(); if( !InitCurl( curl, upload_ctx, mSettingsStorage.GetUserName() )) { logger::LoggerSet::GetInstance()->LogError( METHOD_NAME, "Error to initializate message service" ); } //TODO раздать mail_to другим методам mMailTo.WaitWhileEmptyMails(); auto mail_to = mMailTo.GetMailToSend(); FillRecipients( curl, recipients ); UpdateMailText( mSettingsStorage.GetUserName(), message->subject, message->text ); result = curl_easy_perform( curl ); if( result != CURLE_OK ) { std::string message = "Error to send messge: " + std::string( curl_easy_strerror( result )); logger::LoggerSet::GetInstance()->LogError( METHOD_NAME, message ); } upload_ctx.counter = 0; } //TODO сделать деинициализацию через регистратор деинициализаторов curl_slist_free_all( recipients ); curl_easy_cleanup( curl ); } // //Private methods // bool smtp::message::Sender::InitCurl( CURL* curl, WriteThis const& upload_ctx, std::string const& mail_from ) { if( !curl ) { return false; } if( mSettingsStorage.IsNeedAuth()) { curl_easy_setopt( curl, CURLOPT_USERNAME, mSettingsStorage.GetUserName().c_str()); curl_easy_setopt( curl, CURLOPT_PASSWORD, mSettingsStorage.GetPassword().c_str()); } curl_easy_setopt( curl, CURLOPT_URL, GetHostPortData().c_str()); mSettingsStorage.IsNeedSsl() ? curl_easy_setopt( curl, CURLOPT_USE_SSL, ( long ) CURLUSESSL_ALL ) : curl_easy_setopt( curl, CURLOPT_USE_SSL, ( long ) CURLUSESSL_NONE ); curl_easy_setopt( curl, CURLOPT_MAIL_FROM, mail_from.c_str()); curl_easy_setopt( curl, CURLOPT_READFUNCTION, ReadCallBack ); curl_easy_setopt( curl, CURLOPT_READDATA, &upload_ctx ); curl_easy_setopt( curl, CURLOPT_UPLOAD, 1L ); return true; } void Sender::UpdateMailText( std::string const& mail_from, std::string const& subject, std::string const& text ) const { auto mail_to = mMailTo.GetMailToSend(); auto text_decorator = std::make_shared < builder::Text >( text ); text_decorator->Apply( std::make_shared < builder::Subject >( subject )) .Apply( std::make_shared < builder::Cc >( mail_to )) .Apply( std::make_shared < builder::From >( mail_from )) .Apply( std::make_shared < builder::MailTo >( mail_to )) .Apply( std::make_shared < builder::Date >()); mText = text_decorator->Get(); } void Sender::FillRecipients( CURL* curl, curl_slist* recipients ) { auto mail_to = mMailTo.GetMailToSend(); for( const auto& recipient: mail_to ) { recipients = curl_slist_append( recipients, recipient.c_str()); } curl_easy_setopt( curl, CURLOPT_MAIL_RCPT, recipients ); } std::string Sender::GetHostPortData() const { auto result = "smtp://" + mSettingsStorage.GetHost(); if( !mSettingsStorage.GetPort().empty()) { result += ":" + mSettingsStorage.GetPort(); } return result; } //TODO Надо убрать этот ужас. Без статики!!! size_t Sender::ReadCallBack( void* ptr, size_t size, size_t nmemb, void* userp ) { auto pooh = reinterpret_cast( userp ); if( size * nmemb < 1 || pooh->counter++ > 0 ) { return 0; } memcpy( ptr, mText.c_str(), mText.size()); return mText.size(); } }