//=========================================================================
//  Copyright (c) Kitware, Inc.
//  All rights reserved.
//  See LICENSE.txt for details.
//
//  This software is distributed WITHOUT ANY WARRANTY; without even
//  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
//  PURPOSE.  See the above copyright notice for more information.
//=========================================================================
#ifndef smtk_resource_filter_Action_h
#define smtk_resource_filter_Action_h

#include "smtk/resource/filter/Property.h"
#include "smtk/resource/filter/Rules.h"

#include "smtk/Regex.h"

SMTK_THIRDPARTY_PRE_INCLUDE
#include "tao/pegtl.hpp"
SMTK_THIRDPARTY_POST_INCLUDE

#include <memory>

namespace smtk
{
namespace resource
{
namespace filter
{

using namespace tao::pegtl;

/// A base class template for processing PEGTL rules. A specialization of this
/// class template must exist for each PEGTL rule to be processed.
template<typename Rule>
struct Action : nothing<Rule>
{
};

/// PEGTL requires that a template class be declared with a specialization for
/// each PEGTL rule to be processed. SMTK's Property-based grammar uses a common
/// syntax for each property type, making it practical for us to describe generic
/// Actions templated over the  property type. We then link these actions
/// together with their respective rules using inheritance (as is PEGTL's wont).

/// Construct a new filter rule specific to a given type.
template<typename Type, template<typename T> class RuleClass>
struct TypeNameAction
{
  template<typename Input>
  static void apply(const Input&, Rules& rules)
  {
    rules.emplace_back(new RuleClass<Type>());
  }
};

/// Append the filter rule with a means of discriminating property keys to match
/// the rule input.
template<typename Type, template<typename T> class RuleClass>
struct NameAction
{
  template<typename Input>
  static void apply(const Input& input, Rules& rules)
  {
    std::unique_ptr<Rule>& rule = rules.data().back();
    std::string name = input.string();
    static_cast<RuleClass<Type>*>(rule.get())->acceptableKeys =
      [name](const PersistentObject& object) -> std::vector<std::string> {
      std::vector<std::string> returnValue;
      if (object.properties().contains<Type>(name))
      {
        returnValue.push_back(name);
      }
      return returnValue;
    };
  }
};

/// Append the filter rule with a means of discriminating property keys to match
/// the rule input.
template<typename Type, template<typename T> class RuleClass>
struct RegexAction
{
  template<typename Input>
  static void apply(const Input& input, Rules& rules)
  {
    std::unique_ptr<Rule>& rule = rules.data().back();
    smtk::regex regex(input.string());
    static_cast<RuleClass<Type>*>(rule.get())->acceptableKeys =
      [regex](const PersistentObject& object) -> std::vector<std::string> {
      std::vector<std::string> returnValue;
      for (const auto& key : object.properties().get<Type>().keys())
      {
        if (smtk::regex_match(key, regex))
        {
          returnValue.push_back(key);
        }
      }
      return returnValue;
    };
  }
};

/// Append the filter rule with a means of discriminating property values to
/// match the rule input.
template<typename Type, template<typename T> class RuleClass>
struct ValueAction
{
  template<typename Input>
  static void apply(const Input& input, Rules& rules)
  {
    std::unique_ptr<Rule>& rule = rules.data().back();
    Type value = Property<Type>::convert(input.string());
    static_cast<RuleClass<Type>*>(rule.get())->acceptableValue = [value](const Type& val) -> bool {
      return val == value;
    };
  }
};

/// Append the filter rule with a means of discriminating property values to
/// match the rule input.
template<typename Type, template<typename T> class RuleClass>
struct ValueRegexAction
{
  template<typename Input>
  static void apply(const Input& input, Rules& rules)
  {
    std::unique_ptr<Rule>& rule = rules.data().back();
    smtk::regex regex(input.string());
    static_cast<RuleClass<Type>*>(rule.get())->acceptableValue = [regex](const Type& val) -> bool {
      return smtk::regex_match(val, regex);
    };
  }
};
} // namespace filter
} // namespace resource
} // namespace smtk

#endif
