#include "search/categories_cache.hpp"

#include "search/mwm_context.hpp"
#include "search/query_params.hpp"
#include "search/retrieval.hpp"

#include "indexer/classificator.hpp"
#include "indexer/ftypes_matcher.hpp"
#include "indexer/search_string_utils.hpp"

#include "base/assert.hpp"
#include "base/levenshtein_dfa.hpp"

using namespace std;

namespace search
{
// CategoriesCache ---------------------------------------------------------------------------------
CBV CategoriesCache::Get(MwmContext const & context)
{
  CHECK(context.m_handle.IsAlive(), ());
  ASSERT(context.m_value.HasSearchIndex(), ());

  auto const id = context.m_handle.GetId();
  auto const it = m_cache.find(id);
  if (it != m_cache.cend())
    return it->second;

  auto const cbv = Load(context);
  m_cache[id] = cbv;
  return cbv;
}

CBV CategoriesCache::Load(MwmContext const & context) const
{
  ASSERT(context.m_handle.IsAlive(), ());
  ASSERT(context.m_value.HasSearchIndex(), ());

  auto const & c = classif();

  // Any DFA will do, since we only use requests's m_categories,
  // but the interface of Retrieval forces us to make a choice.
  SearchTrieRequest<strings::UniStringDFA> request;

  // m_categories usually has truncated types; add them together with their subtrees.
  m_categories.ForEach([&request, &c](uint32_t const type) {
    c.ForEachInSubtree(
        [&](uint32_t descendantType) {
          request.m_categories.emplace_back(FeatureTypeToString(c.GetIndexForType(descendantType)));
        },
        type);
  });

  Retrieval retrieval(context, m_cancellable);
  return CBV(retrieval.RetrieveAddressFeatures(request));
}

// StreetsCache ------------------------------------------------------------------------------------
StreetsCache::StreetsCache(base::Cancellable const & cancellable)
  : CategoriesCache(ftypes::IsStreetOrSuburbChecker::Instance(), cancellable)
{
}

// VillagesCache -----------------------------------------------------------------------------------
VillagesCache::VillagesCache(base::Cancellable const & cancellable)
  : CategoriesCache(ftypes::IsVillageChecker::Instance(), cancellable)
{
}

// LocalitiesCache ----------------------------------------------------------------------------------
LocalitiesCache::LocalitiesCache(base::Cancellable const & cancellable)
  : CategoriesCache(ftypes::IsLocalityChecker::Instance(), cancellable)
{
}

// HotelsCache -------------------------------------------------------------------------------------
HotelsCache::HotelsCache(base::Cancellable const & cancellable)
  : CategoriesCache(ftypes::IsHotelChecker::Instance(), cancellable)
{
}

// FoodCache ---------------------------------------------------------------------------------------
FoodCache::FoodCache(base::Cancellable const & cancellable)
  : CategoriesCache(ftypes::IsEatChecker::Instance(), cancellable)
{
}
}  // namespace search