1 files changed, 11 insertions, 6 deletions
diff --git a/src/managment/mail.hpp b/src/managment/mail.hpp
index 5e14831..644a477 100644
--- a/src/managment/mail.hpp
+++ b/src/managment/mail.hpp
@@ -1,23 +1,28 @@
 #pragma once
 
-#include <list>
+#include <set>
 #include <string>
 
 #include "file/mail.hpp"
 
 namespace smtp::manage
 {
+
     class Mail
     {
     public:
-        explicit Mail( file::Mail const& file_reader);
+        explicit Mail( file::Mail const& file_reader );
         ~Mail() = default;
 
-        std::list<std::string> GetMailToSend() const;
-        bool AddMailToSend();
-        bool DeleteMailToSend( std::string const& mail_to_send );
+        MailsSet GetMailToSend() const;
+        bool AddMailsToSend( MailsSet const& mails );
+        bool DeleteMailToSend(MailsSet const&mail_to_send );
     private:
-        std::list<std::string> mMails;
+        MailsSet GetMergedMails( MailsSet const& mails_to_add );
+        MailsSet GetItersectionMails( MailsSet const& mail_to_send ) const;
+        MailsSet GetSetAfterDelete( MailsSet const& result_set_after_delete);
+        MailsSet mMails;
+        file::Mail mFileReader;
     };
 
 }