#pragma once

#include "geometry/rect2d.hpp"

#include "search/result.hpp"
#include "search/search_params.hpp"

#include "std/condition_variable.hpp"
#include "std/mutex.hpp"
#include "std/string.hpp"
#include "std/vector.hpp"

#include "base/timer.hpp"

namespace search
{
namespace tests_support
{
class TestSearchEngine;

// This class wraps a search query to a search engine. Note that the
// last token will be handled as a complete token iff it will be
// followed by a space in a query.
class TestSearchRequest
{
public:
  TestSearchRequest(TestSearchEngine & engine, string const & query, string const & locale,
                    Mode mode, m2::RectD const & viewport);
  TestSearchRequest(TestSearchEngine & engine, SearchParams params, m2::RectD const & viewport);

  // Initiates the search and waits for it to finish.
  void Run();

  // Call these functions only after call to Wait().
  steady_clock::duration ResponseTime() const;
  vector<search::Result> const & Results() const;

protected:
  TestSearchRequest(TestSearchEngine & engine, string const & query, string const & locale,
                    Mode mode, m2::RectD const & viewport, SearchParams::TOnStarted onStarted,
                    SearchParams::TOnResults onResults);

  // Initiates the search.
  void Start();

  // Waits for the search to finish.
  void Wait();

  void SetUpCallbacks();

  void OnStarted();
  void OnResults(search::Results const & results);

  // Overrides the default onResults callback.
  void SetCustomOnResults(SearchParams::TOnResults const & onResults);

  condition_variable m_cv;
  mutable mutex m_mu;

  vector<search::Result> m_results;
  bool m_done = false;

  my::Timer m_timer;
  steady_clock::duration m_startTime;
  steady_clock::duration m_endTime;

  TestSearchEngine & m_engine;
  SearchParams m_params;
  m2::RectD m_viewport;
};
}  // namespace tests_support
}  // namespace search