#include "map/displayed_categories_modifiers.hpp"

#include "base/macros.hpp"

#include <algorithm>
#include <vector>

namespace
{
std::unordered_set<std::string> const kLuggageHeroesSupportedCities{"London", "New York",
                                                                    "Copenhagen"};

std::unordered_set<std::string> const kFc2018SupportedCities{
    "Moscow", "Saint Petersburg", "Kazan",         "Yekaterinburg", "Saransk",        "Samara",
    "Sochi",  "Volgograd",        "Rostov-on-Don", "Kaliningrad",   "Nizhny Novgorod"};
  
std::string const kLuggageHeroCategory = "luggagehero";
std::string const kFc2018Category = "fc2018";

std::vector<std::string> const kSponsoredCategories = {kLuggageHeroCategory, kFc2018Category};

search::DisplayedCategories::Keys::const_iterator FindInsertionPlace(
    search::DisplayedCategories::Keys & keys, uint32_t position)
{
  for (auto it = keys.cbegin(); it != keys.cend(); ++it)
  {
    if (position == 0)
      return it;

    // Do not count sponsored categories.
    if (std::find(kSponsoredCategories.cbegin(), kSponsoredCategories.cend(), *it) ==
        kSponsoredCategories.cend())
    {
      position--;
    }
  }
  return keys.cend();
}
}  // namespace

SponsoredCategoryModifier::SponsoredCategoryModifier(std::string const & currentCity,
                                                     SupportedCities const & supportedCities,
                                                     std::string const & categoryName,
                                                     uint32_t position)
  : m_currentCity(currentCity)
  , m_supportedCities(supportedCities)
  , m_categoryName(categoryName)
  , m_position(position)
{}

void SponsoredCategoryModifier::Modify(search::DisplayedCategories::Keys & keys)
{
  auto const supported = m_supportedCities.find(m_currentCity) != m_supportedCities.cend();
  auto const contains = std::find(keys.cbegin(), keys.cend(), m_categoryName) != keys.cend();

  ASSERT_LESS(m_position, keys.size(), ());

  if (supported && !contains)
    keys.insert(FindInsertionPlace(keys, m_position), m_categoryName);
  else if (!supported && contains)
    keys.erase(std::remove(keys.begin(), keys.end(), m_categoryName), keys.end());
}

LuggageHeroModifier::LuggageHeroModifier(std::string const & currentCity)
  : SponsoredCategoryModifier(currentCity, kLuggageHeroesSupportedCities, kLuggageHeroCategory,
                              4 /* position */)
{}

Fc2018Modifier::Fc2018Modifier(std::string const & currentCity)
  : SponsoredCategoryModifier(currentCity, kFc2018SupportedCities, kFc2018Category, 3 /* position */)
{}