Domain.hpp

   1 #ifndef DOMAIN_DOT_HPP
   2 #define DOMAIN_DOT_HPP
   3
   4 #include "IP.hpp"
   5
   6 #include <iostream>
   7 #include <string>
   8 #include <string_view>
   9
  10 #include "iequal.hpp"
  11
  12 class Domain {
  13 public:
  14   Domain() = default;
  15
  16   explicit Domain(std::string_view dom) { set(dom); }
  17
  18   Domain& operator=(std::string_view dom)
  19   {
  20     set(dom);
  21     return *this;
  22   }
  23
  24   void set(std::string_view dom);
  25
  26   inline void clear();
  27
  28   bool empty() const { return ascii_.empty(); }
  29
  30   inline static std::string_view remove_trailing_dot(std::string_view a);
  31   inline static bool             match(std::string_view a, std::string_view b);
  32
  33   bool operator==(std::string_view rhs) const { return match(ascii_, rhs); }
  34   bool operator!=(std::string_view rhs) const { return !(*this == rhs); }
  35
  36   bool operator==(Domain const& rhs) const { return match(ascii_, rhs.ascii_); }
  37   bool operator!=(Domain const& rhs) const { return !(*this == rhs); }
  38
  39   bool is_address_literal() const { return is_address_literal_; }
  40   bool is_unicode() const { return utf8() != ascii(); }
  41
  42   std::string const& ascii() const { return ascii_; }
  43   std::string const& utf8() const { return utf8_; }
  44
  45   static bool validate(std::string_view dom);
  46
  47 private:
  48   std::string ascii_;
  49   std::string utf8_;
  50
  51   bool is_address_literal_{false};
  52 };
  53
  54 inline void Domain::clear()
  55 {
  56   ascii_.clear();
  57   utf8_.clear();
  58   is_address_literal_ = false;
  59 }
  60
  61 inline std::string_view Domain::remove_trailing_dot(std::string_view a)
  62 {
  63   if (a.length() && (a.back() == '.')) {
  64     a.remove_suffix(1);
  65   }
  66   return a;
  67 }
  68
  69 inline bool Domain::match(std::string_view a, std::string_view b)
  70 {
  71   return iequal(remove_trailing_dot(a), remove_trailing_dot(b));
  72 }
  73
  74 inline std::ostream& operator<<(std::ostream& os, Domain const& dom)
  75 {
  76   if (dom.is_unicode())
  77     return os << '{' << dom.ascii() << ',' << dom.utf8() << '}';
  78   return os << dom.ascii();
  79 }
  80
  81 namespace std {
  82 template <>
  83 struct hash<Domain> {
  84   std::size_t operator()(Domain const& k) const
  85   {
  86     return hash<std::string>()(k.ascii());
  87   }
  88 };
  89 } // namespace std
  90
  91 #endif // DOMAIN_DOT_HPP