// Copyright (C) 2023 Joaquin M Lopez Munoz
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

// temporary #define till all transitive includes comply with
// https://github.com/boostorg/core/commit/5f6fe65

#define BOOST_ALLOW_DEPRECATED_HEADERS

#include "../helpers/unordered.hpp"

#include "../objects/test.hpp"
#include "../helpers/random_values.hpp"

#include <algorithm>
#include <boost/archive/text_oarchive.hpp>
#include <boost/archive/text_iarchive.hpp>
#include <boost/archive/xml_oarchive.hpp>
#include <boost/archive/xml_iarchive.hpp>
#include <boost/config.hpp>
#include <boost/config/workaround.hpp>
#include <boost/serialization/nvp.hpp>
#include <boost/serialization/string.hpp>
#include <boost/serialization/utility.hpp>
#include <boost/serialization/vector.hpp>
#include <cstddef>
#include <cstdio>
#include <fstream>

#if defined(BOOST_NO_CXX11_HDR_RANDOM) || BOOST_WORKAROUND(BOOST_MSVC, < 1700)
#define BOOST_UNORDERED_TEST_USE_STD_RANDOM_SHUFFLE
#endif

#ifndef BOOST_UNORDERED_TEST_USE_STD_RANDOM_SHUFFLE
#include <random>
#endif

namespace {

  template <class Container, typename ArchivePair>
  void serialization_tests(
    Container*, ArchivePair*, test::random_generator generator)
  {
    typedef typename Container::iterator      iterator;
    typedef std::vector<iterator>             iterator_vector;
    typedef typename ArchivePair::first_type  output_archive;
    typedef typename ArchivePair::second_type input_archive;

    BOOST_LIGHTWEIGHT_TEST_OSTREAM << "serialization_tests1\n";
    {
      Container c;
      iterator it = c.end();
  
      std::ostringstream oss;
      {
        output_archive oa(oss);
        oa << boost::serialization::make_nvp("container", c);
        oa << boost::serialization::make_nvp("iterator", it);
      }

      test::random_values<Container> values(100, generator);
      Container c2(values.begin(), values.end());
      iterator it2 = c2.begin();
      std::istringstream iss(oss.str());
      input_archive ia(iss);
      ia >> boost::serialization::make_nvp("container", c2);
      ia >> boost::serialization::make_nvp("iterator", it2);
      BOOST_TEST(c2.empty());
      BOOST_TEST(it2 == c2.end());
    }

    BOOST_LIGHTWEIGHT_TEST_OSTREAM << "serialization_tests2\n";
    {
      test::random_values<Container> values(100, generator);
      Container c(values.begin(), values.end());
  
      iterator_vector v;
      for (iterator first = c.begin(), last=c.end(); ; ) {
        v.push_back(first);
        if(first == last) break;
        ++first;
      }

#ifdef BOOST_UNORDERED_TEST_USE_STD_RANDOM_SHUFFLE
      std::random_shuffle(v.begin(), v.end());
#else
      std::shuffle(v.begin(), v.end(), std::mt19937(4213));
#endif

      std::ostringstream oss;
      {
        output_archive oa(oss);
        oa << boost::serialization::make_nvp("container", c);
        oa << boost::serialization::make_nvp("iterators", v);
      }

      Container c2;
      iterator_vector v2;
      std::istringstream iss(oss.str());
      input_archive ia(iss);
      ia >> boost::serialization::make_nvp("container", c2);
      ia >> boost::serialization::make_nvp("iterators", v2);
      BOOST_TEST(c == c2);
      BOOST_TEST_EQ(v.size(), v2.size());

      for (std::size_t i=0; i < v.size(); ++i) {
        iterator it = v[i];
        iterator it2 = v2[i];
        if (it == c.end()) {
          BOOST_TEST(it2 == c2.end());
        }
        else {
          BOOST_TEST(it2 != c2.end());
          BOOST_TEST(*it == *it2);
        }
      }
    }
  }

  // used by legacy_serialization_test, passed as argv[1]
  const char* test_dir=".";

  using test::default_generator;

  std::pair<
    boost::archive::text_oarchive, boost::archive::text_iarchive>*
    text_archive;
  std::pair<
    boost::archive::xml_oarchive, boost::archive::xml_iarchive>*
    xml_archive;

#ifdef BOOST_UNORDERED_FOA_TESTS
  boost::unordered_flat_map<
    test::object, test::object, test::hash, test::equal_to>* test_flat_map;
  boost::unordered_node_map<
    test::object, test::object, test::hash, test::equal_to>* test_node_map;
  boost::unordered_flat_set<
    test::object, test::hash, test::equal_to>* test_flat_set;
  boost::unordered_node_set<
    test::object, test::hash, test::equal_to>* test_node_set;

  UNORDERED_TEST(serialization_tests,
    ((test_flat_map)(test_node_map)(test_flat_set)(test_node_set))
    ((text_archive)(xml_archive))
    ((default_generator)))
#else
  boost::unordered_map<
    test::object, test::object, test::hash, test::equal_to>* test_map;
  boost::unordered_multimap<
    test::object, test::object, test::hash, test::equal_to>* test_multimap;
  boost::unordered_set<
    test::object, test::hash, test::equal_to>* test_set;
  boost::unordered_multiset<
    test::object, test::hash, test::equal_to>* test_multiset;

  UNORDERED_TEST(serialization_tests,
    ((test_map)(test_multimap)(test_set)(test_multiset))
    ((text_archive)(xml_archive))
    ((default_generator)))

  template<typename T>
  struct non_const
  {
    typedef T type;
  };

  template<typename T>
  struct non_const<const T>
  {
    typedef typename non_const<T>::type type;
  };

  template<typename T, typename Q>
  struct non_const<std::pair<T, Q> >
  {
    typedef std::pair<
      typename non_const<T>::type,
      typename non_const<Q>::type> type;
  };

  template<typename T>
  struct labeled
  {
    labeled(const char* label_): label(label_) {}

    const char* label;
  };

  template <class Container, typename Archive>
  void legacy_serialization_test(labeled<Container> lc, labeled<Archive> la)
  {
    typedef typename Container::value_type                    value_type;
    typedef std::vector<typename non_const<value_type>::type> value_vector;

    static const std::size_t sizes[] = {0, 10, 100};

    BOOST_LIGHTWEIGHT_TEST_OSTREAM << "legacy_serialization_test\n";

    for(int i = 0; i < sizeof(sizes)/sizeof(sizes[0]); ++i) {
      char filename[1024];
      std::sprintf(
        filename, "%s/legacy_archives/%s_%d.%s",
        test_dir, lc.label, (int)sizes[i], la.label);
      std::ifstream ifs(filename);
      Archive ia(ifs);
      Container c;
      value_vector v;
      ia >> boost::serialization::make_nvp("container", c);
      ia >> boost::serialization::make_nvp("values", v);
      BOOST_TEST(v.size() >= sizes[i]); // values generated with repetition
      BOOST_TEST((c==Container(v.begin(),v.end())));
    }
  }

  labeled<boost::unordered_map<int, int> >
    labeled_map_int("map_int");
  labeled<boost::unordered_map<std::string, std::string> >
    labeled_map_string("map_string");
  labeled<boost::unordered_multimap<int, int> >
    labeled_multimap_int("multimap_int");
  labeled<boost::unordered_multimap<std::string, std::string> >
    labeled_multimap_string("multimap_string");
  labeled<boost::unordered_set<int> >
    labeled_set_int("set_int");
  labeled<boost::unordered_set<std::string> >
    labeled_set_string("set_string");
  labeled<boost::unordered_multiset<int> >
    labeled_multiset_int("multiset_int");
  labeled<boost::unordered_multiset<std::string> >
    labeled_multiset_string("multiset_string");

  labeled<boost::archive::text_iarchive> labeled_text_iarchive("txt");
  labeled<boost::archive::xml_iarchive> labeled_xml_iarchive("xml");

  UNORDERED_TEST(legacy_serialization_test,
    ((labeled_map_int)(labeled_map_string)
     (labeled_multimap_int)(labeled_multimap_string)
     (labeled_set_int)(labeled_set_string)
     (labeled_multiset_int)(labeled_multiset_string))
    ((labeled_text_iarchive)(labeled_xml_iarchive)))
#endif
}

int main(int argc, char* argv[])
{
  if (argc > 1) test_dir = argv[1];

  BOOST_UNORDERED_TEST_COMPILER_INFO()
  ::test::get_state().run_tests();
  return boost::report_errors();
}