Merge branch 'bugfix/write_file'

1 files changed, 52 insertions, 5 deletions

diff --git a/src/managment/mail.cpp b/src/managment/mail.cpp
index e43a68c..70c5375 100644
--- a/src/managment/mail.cpp
+++ b/src/managment/mail.cpp
@@ -1,25 +1,72 @@
+#include <algorithm>
+
 #include "mail.hpp"
 
 namespace smtp::manage
 {
-    Mail::Mail( file::Mail const& file_reader)
+    Mail::Mail( file::Mail const& file_reader )
+        : mFileReader( file_reader )
     {
-        mMails = file_reader.Read();
+        mMails = mFileReader.Read();
+        mMails.sort();
     }
 
-    std::list<std::string> Mail::GetMailToSend() const
+    MailsSet Mail::GetMailToSend() const
     {
         return mMails;
     }
 
-    bool Mail::AddMailToSend()
+    bool Mail::AddMailsToSend( MailsSet const& mails_to_add )
+    {
+        auto merged_mails = GetMergedMails( mails_to_add );
+
+        auto result = mFileReader.Write( merged_mails );
+        if( result )
+        {
+            mMails = merged_mails;
+        }
+        return result;
+    }
+
+    bool Mail::DeleteMailToSend( MailsSet const& mail_to_send )
     {
+        auto mails_to_delete = GetItersectionMails( mail_to_send );
+        auto result_set_after_delete = GetSetAfterDelete( mails_to_delete );
+        auto result_of_delete = mFileReader.Write( result_set_after_delete );
+        if(result_of_delete)
+        {
+            mMails = result_set_after_delete;
+        }
+        return result_of_delete;
+    }
 
+    MailsSet Mail::GetMergedMails( MailsSet const& mails_to_add )
+    {
+        auto result = mails_to_add;
+        result.merge( mMails );
+        return result;
     }
 
-    bool Mail::DeleteMailToSend( std::string const& mail_to_send)
+    MailsSet Mail::GetItersectionMails( MailsSet const& mail_to_send ) const
     {
+        MailsSet result;
 
+        auto sorted_mail_to_send = mail_to_send;
+        sorted_mail_to_send.sort();
+        std::set_intersection( mMails.cbegin(), mMails.cend(),
+                               sorted_mail_to_send.cbegin(), sorted_mail_to_send.cend(),
+                               std::back_inserter( result ));
+        return result;
+    }
+
+    MailsSet Mail::GetSetAfterDelete( MailsSet const& result_set_after_delete)
+    {
+        auto result = mMails;
+        for(const auto& element_to_delete : result_set_after_delete)
+        {
+            result.remove( element_to_delete );
+        }
+        return result;
     }
 
 }