// Copyright (c) 2016-2020 Dr. Colin Hirsch and Daniel Frey
// Please see LICENSE for license or visit https://github.com/taocpp/json/

#ifndef TAO_JSON_EVENTS_TEE_HPP
#define TAO_JSON_EVENTS_TEE_HPP

#include <cstdint>
#include <string>
#include <tuple>
#include <type_traits>
#include <utility>
#include <vector>

#include "discard.hpp"

namespace tao::json
{
   namespace internal
   {
      template< typename T >
      struct strip_reference_wrapper
      {
         using type = T;
      };

      template< typename T >
      struct strip_reference_wrapper< std::reference_wrapper< T > >
      {
         using type = T&;
      };

      template< typename T >
      using decay_and_strip_t = typename strip_reference_wrapper< std::decay_t< T > >::type;

      template< typename >
      struct events_apply;

      template<>
      struct events_apply< std::index_sequence<> >
      {
         template< typename... Ts >
         static void null( std::tuple< Ts... >& /*unused*/ )
         {}

         template< typename... Ts >
         static void boolean( std::tuple< Ts... >& /*unused*/, const bool /*unused*/ )
         {}

         template< typename... Ts >
         static void number( std::tuple< Ts... >& /*unused*/, const std::int64_t /*unused*/ )
         {}

         template< typename... Ts >
         static void number( std::tuple< Ts... >& /*unused*/, const std::uint64_t /*unused*/ )
         {}

         template< typename... Ts >
         static void number( std::tuple< Ts... >& /*unused*/, const double /*unused*/ )
         {}

         template< typename... Ts >
         static void string( std::tuple< Ts... >& /*unused*/, const std::string_view /*unused*/ )
         {}

         template< typename... Ts >
         static void binary( std::tuple< Ts... >& /*unused*/, const tao::binary_view /*unused*/ )
         {}

         template< typename... Ts >
         static void begin_array( std::tuple< Ts... >& /*unused*/ )
         {}

         template< typename... Ts >
         static void begin_array( std::tuple< Ts... >& /*unused*/, const std::size_t /*unused*/ )
         {}

         template< typename... Ts >
         static void element( std::tuple< Ts... >& /*unused*/ )
         {}

         template< typename... Ts >
         static void end_array( std::tuple< Ts... >& /*unused*/ )
         {}

         template< typename... Ts >
         static void end_array( std::tuple< Ts... >& /*unused*/, const std::size_t /*unused*/ )
         {}

         template< typename... Ts >
         static void begin_object( std::tuple< Ts... >& /*unused*/ )
         {}

         template< typename... Ts >
         static void begin_object( std::tuple< Ts... >& /*unused*/, const std::size_t /*unused*/ )
         {}

         template< typename... Ts >
         static void key( std::tuple< Ts... >& /*unused*/, const std::string_view /*unused*/ )
         {}

         template< typename... Ts >
         static void member( std::tuple< Ts... >& /*unused*/ )
         {}

         template< typename... Ts >
         static void end_object( std::tuple< Ts... >& /*unused*/ )
         {}

         template< typename... Ts >
         static void end_object( std::tuple< Ts... >& /*unused*/, const std::size_t /*unused*/ )
         {}
      };

      template< std::size_t... Is >
      struct events_apply< std::index_sequence< Is... > >
      {
         template< typename... Ts >
         static void null( std::tuple< Ts... >& t )
         {
            ( std::get< Is >( t ).null(), ... );
         }

         template< typename... Ts >
         static void boolean( std::tuple< Ts... >& t, const bool v )
         {
            ( std::get< Is >( t ).boolean( v ), ... );
         }

         template< typename... Ts >
         static void number( std::tuple< Ts... >& t, const std::int64_t v )
         {
            ( std::get< Is >( t ).number( v ), ... );
         }

         template< typename... Ts >
         static void number( std::tuple< Ts... >& t, const std::uint64_t v )
         {
            ( std::get< Is >( t ).number( v ), ... );
         }

         template< typename... Ts >
         static void number( std::tuple< Ts... >& t, const double v )
         {
            ( std::get< Is >( t ).number( v ), ... );
         }

         template< typename... Ts >
         static void string( std::tuple< Ts... >& t, const std::string_view v )
         {
            ( std::get< Is >( t ).string( v ), ... );
         }

         template< typename... Ts >
         static void binary( std::tuple< Ts... >& t, const tao::binary_view v )
         {
            ( std::get< Is >( t ).binary( v ), ... );
         }

         template< typename... Ts >
         static void begin_array( std::tuple< Ts... >& t )
         {
            ( std::get< Is >( t ).begin_array(), ... );
         }

         template< typename... Ts >
         static void begin_array( std::tuple< Ts... >& t, const std::size_t size )
         {
            ( std::get< Is >( t ).begin_array( size ), ... );
         }

         template< typename... Ts >
         static void element( std::tuple< Ts... >& t )
         {
            ( std::get< Is >( t ).element(), ... );
         }

         template< typename... Ts >
         static void end_array( std::tuple< Ts... >& t )
         {
            ( std::get< Is >( t ).end_array(), ... );
         }

         template< typename... Ts >
         static void end_array( std::tuple< Ts... >& t, const std::size_t size )
         {
            ( std::get< Is >( t ).end_array( size ), ... );
         }

         template< typename... Ts >
         static void begin_object( std::tuple< Ts... >& t )
         {
            ( std::get< Is >( t ).begin_object(), ... );
         }

         template< typename... Ts >
         static void begin_object( std::tuple< Ts... >& t, const std::size_t size )
         {
            ( std::get< Is >( t ).begin_object( size ), ... );
         }

         template< typename... Ts >
         static void key( std::tuple< Ts... >& t, const std::string_view v )
         {
            ( std::get< Is >( t ).key( v ), ... );
         }

         template< typename... Ts >
         static void member( std::tuple< Ts... >& t )
         {
            ( std::get< Is >( t ).member(), ... );
         }

         template< typename... Ts >
         static void end_object( std::tuple< Ts... >& t )
         {
            ( std::get< Is >( t ).end_object(), ... );
         }

         template< typename... Ts >
         static void end_object( std::tuple< Ts... >& t, const std::size_t size )
         {
            ( std::get< Is >( t ).end_object( size ), ... );
         }
      };

   }  // namespace internal

   namespace events
   {
      // Events consumer that forwards to multiple nested consumers.

      template< typename... Ts >
      class tee
      {
      private:
         static inline constexpr std::size_t S = sizeof...( Ts );  // NOLINT(readability-identifier-naming)

         using I = std::make_index_sequence< S >;
         using H = std::make_index_sequence< S - 1 >;

         std::tuple< Ts... > ts;

      public:
         template< typename... Us >
         explicit tee( Us&&... us )
            : ts( std::forward< Us >( us )... )
         {}

         void null()
         {
            internal::events_apply< I >::null( ts );
         }

         void boolean( const bool v )
         {
            internal::events_apply< I >::boolean( ts, v );
         }

         void number( const std::int64_t v )
         {
            internal::events_apply< I >::number( ts, v );
         }

         void number( const std::uint64_t v )
         {
            internal::events_apply< I >::number( ts, v );
         }

         void number( const double v )
         {
            internal::events_apply< I >::number( ts, v );
         }

         void string( const std::string_view v )
         {
            internal::events_apply< I >::string( ts, v );
         }

         void string( const char* v )
         {
            internal::events_apply< I >::string( ts, v );
         }

         void string( std::string&& v )
         {
            internal::events_apply< H >::string( ts, v );
            std::get< S - 1 >( ts ).string( std::move( v ) );
         }

         void binary( const tao::binary_view v )
         {
            internal::events_apply< I >::binary( ts, v );
         }

         void binary( std::vector< std::byte >&& v )
         {
            internal::events_apply< H >::binary( ts, v );
            std::get< S - 1 >( ts ).binary( std::move( v ) );
         }

         void begin_array()
         {
            internal::events_apply< I >::begin_array( ts );
         }

         void begin_array( const std::size_t size )
         {
            internal::events_apply< I >::begin_array( ts, size );
         }

         void element()
         {
            internal::events_apply< I >::element( ts );
         }

         void end_array()
         {
            internal::events_apply< I >::end_array( ts );
         }

         void end_array( const std::size_t size )
         {
            internal::events_apply< I >::end_array( ts, size );
         }

         void begin_object()
         {
            internal::events_apply< I >::begin_object( ts );
         }

         void begin_object( const std::size_t size )
         {
            internal::events_apply< I >::begin_object( ts, size );
         }

         void key( const std::string_view v )
         {
            internal::events_apply< I >::key( ts, v );
         }

         void key( const char* v )
         {
            internal::events_apply< I >::key( ts, v );
         }

         void key( std::string&& v )
         {
            internal::events_apply< H >::key( ts, v );
            std::get< S - 1 >( ts ).key( std::move( v ) );
         }

         void member()
         {
            internal::events_apply< I >::member( ts );
         }

         void end_object()
         {
            internal::events_apply< I >::end_object( ts );
         }

         void end_object( const std::size_t size )
         {
            internal::events_apply< I >::end_object( ts, size );
         }
      };

      template<>
      class tee<>
         : public discard
      {};

      template< typename... Ts >
      tee( Ts&&... ) -> tee< internal::decay_and_strip_t< Ts >... >;

      template< typename... T >
      [[nodiscard]] tee< T&... > tie( T&... t )
      {
         return tee< T&... >( t... );
      }

   }  // namespace events

}  // namespace tao::json

#endif