Mailbox.hpp

   1 #ifndef MAILBOX_DOT_HPP
   2 #define MAILBOX_DOT_HPP
   3
   4 #include "Domain.hpp"
   5
   6 #include <cstdint>
   7 #include <optional>
   8 #include <ostream>
   9 #include <string>
  10 #include <utility>
  11
  12 class Mailbox {
  13 public:
  14   Mailbox() = default;
  15   Mailbox(std::string_view mailbox);
  16   Mailbox(std::string_view local_part, std::string_view domain)
  17   {
  18     set_local(local_part);
  19     set_domain(domain);
  20   }
  21   Mailbox(std::string_view local_part, Domain domain)
  22   {
  23     set_local(local_part);
  24     set_domain(domain);
  25   }
  26
  27   void set_local(std::string_view local_part) { local_part_ = local_part; }
  28   void set_domain(std::string_view d) { domain_.set(d); }
  29   void set_domain(Domain d) { domain_ = d; }
  30   void clear()
  31   {
  32     local_part_.clear();
  33     domain_.clear();
  34   }
  35   std::string const& local_part() const { return local_part_; }
  36   Domain const&      domain() const { return domain_; }
  37
  38   enum class domain_encoding : bool { ascii, utf8 };
  39
  40   size_t length(domain_encoding enc = domain_encoding::utf8) const;
  41
  42   bool empty() const { return length() == 0; }
  43
  44   std::string as_string(domain_encoding enc = domain_encoding::utf8) const;
  45
  46   operator std::string() const { return as_string(domain_encoding::utf8); }
  47
  48   bool operator==(Mailbox const& rhs) const
  49   {
  50     return (local_part_ == rhs.local_part_) && (domain_ == rhs.domain_);
  51   }
  52   bool operator!=(Mailbox const& rhs) const { return !(*this == rhs); }
  53
  54   enum class local_types : uint8_t {
  55     unknown,
  56     dot_string,
  57     quoted_string,
  58   };
  59
  60   enum class domain_types : uint8_t {
  61     unknown,
  62     domain,
  63     address_literal,
  64   };
  65
  66   struct parse_results {
  67     std::string_view local;
  68     std::string_view domain;
  69     local_types      local_type  = local_types::unknown;
  70     domain_types     domain_type = domain_types::unknown;
  71   };
  72
  73   static std::optional<parse_results> parse(std::string_view mailbox);
  74
  75   static bool validate(std::string_view mailbox)
  76   {
  77     try {
  78       Mailbox x(mailbox);
  79     }
  80     catch (...) {
  81       return false;
  82     }
  83     return true;
  84   }
  85
  86   private:
  87     std::string local_part_;
  88     Domain      domain_;
  89 };
  90
  91 inline std::ostream& operator<<(std::ostream& s, Mailbox const& mb)
  92 {
  93   return s << static_cast<std::string>(mb);
  94 }
  95
  96 #endif // MAILBOX_DOT_HPP