jnk/Send.hpp

   1 #ifndef SEND_DOT_HPP
   2 #define SEND_DOT_HPP
   3
   4 #include "DNS.hpp"
   5 #include "Mailbox.hpp"
   6 #include "Sock.hpp"
   7 #include "fs.hpp"
   8
   9 namespace SMTP {
  10 struct Connection {
  11   Connection(int fd_in, int fd_out, std::function<void(void)> read_hook)
  12     : sock(fd_in,
  13            fd_out,
  14            read_hook,
  15            std::chrono::minutes(2),
  16            std::chrono::minutes(2))
  17   {
  18   }
  19
  20   Sock sock;
  21
  22   std::string server_id;
  23
  24   Mailbox              mail_from;
  25   std::vector<Mailbox> rcpt_to;
  26
  27   std::string                                               ehlo_keyword;
  28   std::vector<std::string>                                  ehlo_param;
  29   std::unordered_map<std::string, std::vector<std::string>> ehlo_params;
  30
  31   std::string reply_code;
  32
  33   bool greeting_ok{false};
  34   bool ehlo_ok{false};
  35   bool active{false}; // At least one rcp_to for this MX
  36
  37   bool has_extension(char const* name) const
  38   {
  39     return ehlo_params.find(name) != ehlo_params.end();
  40   }
  41 };
  42 } // namespace SMTP
  43
  44 class Send {
  45 public:
  46   Send(fs::path config_path, char const* service);
  47
  48   void set_sender(Domain sender) { sender_ = sender; }
  49
  50   bool mail_from(Mailbox const& from);
  51   bool rcpt_to(DNS::Resolver& res, Mailbox const& to, std::string& error_msg);
  52
  53   bool send(std::string_view msg);
  54
  55   void rset();
  56   void quit();
  57
  58 private:
  59   fs::path config_path_;
  60   std::string service_;
  61
  62   Domain sender_;
  63
  64   Mailbox              mail_from_;
  65   std::vector<Mailbox> rcpt_to_;
  66
  67   // MX hostname or address => connection
  68   std::unordered_map<Domain, std::unique_ptr<SMTP::Connection>> exchangers_;
  69
  70   // @domain => MX hostname or address
  71   std::unordered_map<Domain, Domain> receivers_;
  72 };
  73
  74 #endif // SEND_DOT_HPP