// Copyright (c) 2010 Martin Knafve / hMailServer.com.  
// http://www.hmailserver.com

#pragma once

namespace HM
{
   class Message;
   class MimeHeader;
   class Attachments;

   class MimeBody;

   class MessageData  
   {
   public:
      MessageData();
	   virtual ~MessageData();

      bool LoadFromMessage(const String &fileName, std::shared_ptr<Message> pMessage);
      bool LoadFromMessage(std::shared_ptr<const Account> account, std::shared_ptr<Message> pMessage);

      bool RefreshFromMessage();

      String GetHeader() const;
      
      String GetSubject() const;
      void SetSubject(const String &sSubject);

      String GetCharset() const;
      void SetCharset(const String &sCharset);

      void DeleteField(const AnsiString &sHeaderName);

      String GetSentTime() const;
      void SetSentTime(const String &sSentTime);
      
      String GetTo() const;
      void SetTo(const String &sTo);
      
      String GetReturnPath() const;
      void SetReturnPath(const String &sReturnPath);

      String GetCC() const;
      void SetCC(const String &sCC);
      
      String GetBCC() const;
      void SetBCC(const String &sBCC);

      String GetFrom() const;
      void SetFrom(const String &sFrom);

      String GetHTMLBody() const;
      void SetHTMLBody(const String &sHTMLBody);
      
      String GetBody() const;
      void SetBody(const String &sBody);

      void SetFieldValue(const String &sField, const String &sValue);
      String GetFieldValue(const String &sName) const;

      int GetRuleLoopCount();
      void SetRuleLoopCount(int iLoopCount);
      void IncreaseRuleLoopCount();
      
      std::shared_ptr<MimeBody> GetBodyTextPlainPart() const;
      std::shared_ptr<MimeBody> GetBodyTextHtmlPart() const;

      bool GetHasBodyType(const String &sBodyType);

      bool Write(const String &fileName);

      int GetSize() const;

      std::shared_ptr<Attachments> GetAttachments();

      std::shared_ptr<Message> GetMessage() {return message_; }
   
      bool GetEncodeFields() {return encode_fields_; }
      void SetEncodeFields(bool bNewVal) {encode_fields_ = bNewVal; }

      void GenerateMessageID();

      std::shared_ptr<MimeBody> CreatePart(const String &sContentType);

      std::shared_ptr<MimeBody> GetMimeMessage();

	  void SetAutoReplied();
	  void SetAutoGenerated();
	  bool IsAutoSubmitted();

   private:

      std::shared_ptr<MimeBody> GetViewBodyPart_(int recursion_level, std::shared_ptr<MimeBody> source, const String &requested_content_type) const;

      bool IsTextType(const String &sContentType);
      bool IsHTMLType(const String &sContentType);

      std::shared_ptr<MimeBody> FindPart(const String &sType) const;
      std::shared_ptr<MimeBody> FindPartNoRecurse(std::shared_ptr<MimeBody> parent, const AnsiString &sType) const;

      std::shared_ptr<Message> message_;
      std::shared_ptr<MimeBody> mime_mail_;
      std::shared_ptr<Attachments> attachments_;

      String message_file_name_;

      bool encode_fields_;
      bool unfold_with_space_;
   };

   class MessageDataTester
   {
   public:
      MessageDataTester() {};
      void Test();
   };

}