1 files changed, 10 insertions, 8 deletions
diff --git a/src/file/mail/file_decorator.hpp b/src/file/mail/file_decorator.hpp
index 21dab10..a1767e6 100644
--- a/src/file/mail/file_decorator.hpp
+++ b/src/file/mail/file_decorator.hpp
@@ -1,24 +1,26 @@
 #pragma once
 
-#include "management/general.hpp"
-#include "message/builder/types/idecorator.hpp"
+#include "general/struct.hpp"
+#include "general/idecorator.hpp"
 #include "types/ifile_manipulator.hpp"
+#include "file/errors/registrator.hpp"
 
 namespace smtp::file::mail
 {
-    class FileDecorator : public smtp::message::builder::types::IDecorator< types::IFileManipulator>
+    class FileDecorator : public general::IDecorator< types::IFileManipulator>
     {
     public:
-        explicit FileDecorator( std::string const& path_file );
+        explicit FileDecorator( std::string const& path_file, file::errors::Registrator const& registrator_file_errors );
         ~FileDecorator() override = default;
 
-        manage::MailsSet Read() const override;
-        bool Write( manage::MailsSet const& data ) const override;
+        general::MailsSet Read() const override;
+        bool Write( general::MailsSet const& data ) const override;
     private:
-        manage::MailsSet ReadFile( std::ifstream& mail_file ) const;
-        bool WriteFile( std::ofstream& mail_file, manage::MailsSet const& data ) const;
+        general::MailsSet ReadFile( std::ifstream& mail_file ) const;
+        bool WriteFile( std::ofstream& mail_file, general::MailsSet const& data ) const;
 
         std::string mPathFile;
+        file::errors::Registrator mRegistratorFileErrors;
     };
 
 }