text: // MIT License
  
  // Copyright (c) 2017 Zhuhao Wang
  
  // Permission is hereby granted, free of charge, to any person obtaining a copy
  // of this software and associated documentation files (the "Software"), to deal
  // in the Software without restriction, including without limitation the rights
  // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  // copies of the Software, and to permit persons to whom the Software is
  // furnished to do so, subject to the following conditions:
  
  // The above copyright notice and this permission notice shall be included in
  // all copies or substantial portions of the Software.
  
  // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  // SOFTWARE.
  
  #pragma once
  
  #include <type_traits>
  
  #include <boost/assert.hpp>
  
  #include "../utils/trie.h"
  #include "rule_interface.h"
  
  namespace nekit {
  namespace rule {
  template <bool reverse>
  class DomainAffixRule : public RuleInterface {
   public:
    explicit DomainAffixRule(RuleHandler handler) : handler_{handler} {}
  
    template <typename = std::enable_if_t<!reverse>>
    void AddPrefix(const std::string& prefix) {
      trie_.AddPrefix(prefix);
    }
  
    template <typename = std::enable_if_t<reverse>>
    void AddSuffix(const std::string& suffix) {
      trie_.AddPrefix(suffix);
    }
  
    MatchResult Match(std::shared_ptr<utils::Session> session) override {
      BOOST_ASSERT(session->endpoint());
  
      if (session->endpoint()->type() == utils::Endpoint::Type::Address) {
        return MatchResult::NotMatch;
      }
  
      if (trie_.MatchPrefixWith(session->endpoint()->host())) {
        return MatchResult::Match;
      } else {
        return MatchResult::NotMatch;
      }
    }
  
    std::unique_ptr<data_flow::RemoteDataFlowInterface> GetDataFlow(
        std::shared_ptr<utils::Session> session) override {
      BOOST_ASSERT(session->endpoint());
  
      return handler_(session);
    }
  
   private:
    utils::DomainTrie<reverse> trie_;
  
    RuleHandler handler_;
  };
  
  using DomainPrefixRule = DomainAffixRule<false>;
  using DomainSuffixRule = DomainAffixRule<true>;
  }  // namespace rule
  }  // namespace nekit
  