//  Copyright (c) 2014-2017 John Biddiscombe
//
//  SPDX-License-Identifier: BSL-1.0
//  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)

#pragma once

#include <hpx/config.hpp>
#include <hpx/modules/preprocessor.hpp>
#include <hpx/modules/threading.hpp>
//
#include <boost/preprocessor.hpp>
#if defined(__linux) || defined(__linux__)
#include <sched.h>
#else
#define sched_getcpu() '#'
#endif

#include <chrono>
#include <iomanip>
#include <iostream>
#include <sstream>
#include <string>
#include <thread>

// ------------------------------------------------------------------
// Set flags to help simplify the log defines
// ------------------------------------------------------------------

// full logging support, we want everything
#if defined(HPX_PARCELPORT_LIBFABRIC_HAVE_LOGGING)
#define HPX_PARCELPORT_LOGGING_INCLUDE_FILES
#define HPX_PARCELPORT_LOGGING_HAVE_TRACE_LOG
#define HPX_PARCELPORT_LOGGING_HAVE_DEBUG_LOG
#define HPX_PARCELPORT_LOGGING_HAVE_TIMED_LOG
#define HPX_PARCELPORT_LOGGING_HAVE_DEVEL_LOG

// just a subset of logging for dev mode enabled
#elif defined(HPX_PARCELPORT_LIBFABRIC_HAVE_DEV_MODE)
#define HPX_PARCELPORT_LOGGING_INCLUDE_FILES
#define HPX_PARCELPORT_LOGGING_HAVE_TIMED_LOG
#define HPX_PARCELPORT_LOGGING_HAVE_DEVEL_LOG
#endif

// ------------------------------------------------------------------
// useful macros for formatting log messages
// ------------------------------------------------------------------
#define nhex(n)                                                                \
    "0x" << std::setfill('0') << std::setw(n) << std::noshowbase << std::hex
#define hexpointer(p) nhex(16) << (uintptr_t)(p) << " "
#define hexuint64(p) nhex(16) << (uintptr_t)(p) << " "
#define hexuint32(p) nhex(8) << (uint32_t)(p) << " "
#define hexlength(p) nhex(6) << (uintptr_t)(p) << " "
#define hexnumber(p) nhex(4) << (uintptr_t)(p) << " "
#define hexbyte(p) nhex(2) << static_cast<int>(p) << " "
#define decimal(n)                                                             \
    std::setfill('0') << std::setw(n) << std::noshowbase << std::dec
#define decnumber(p) std::dec << p << " "
#define dec4(p) decimal(4) << p << " "
#define ipaddress(p)                                                           \
    std::dec << (int) (reinterpret_cast<const uint8_t*>(&p))[0] << "."         \
             << (int) (reinterpret_cast<const uint8_t*>(&p))[1] << "."         \
             << (int) (reinterpret_cast<const uint8_t*>(&p))[2] << "."         \
             << (int) (reinterpret_cast<const uint8_t*>(&p))[3] << " "
#define sockaddress(p) ipaddress(((struct sockaddr_in*) (p))->sin_addr.s_addr)

// ------------------------------------------------------------------
// include files needed for boost::log
// ------------------------------------------------------------------
#ifdef HPX_PARCELPORT_LOGGING_INCLUDE_FILES
#include <hpx/modules/threading.hpp>
#include <boost/log/expressions.hpp>
#include <boost/log/expressions/formatter.hpp>
#include <boost/log/expressions/formatters.hpp>
#include <boost/log/expressions/formatters/stream.hpp>
#include <boost/log/sources/record_ostream.hpp>
#include <boost/log/sources/severity_logger.hpp>
#include <boost/log/trivial.hpp>
#include <boost/log/utility/formatting_ostream.hpp>
#include <boost/log/utility/manipulators/to_log.hpp>
#include <boost/log/utility/setup/common_attributes.hpp>
#include <boost/log/utility/setup/console.hpp>
#endif

#include <boost/crc.hpp>

// ------------------------------------------------------------------
// helper classes/functions used in logging
// ------------------------------------------------------------------
namespace hpx { namespace parcelset { namespace policies { namespace libfabric {
    namespace detail {

        // ------------------------------------------------------------------
        // helper class for printing thread ID
        // ------------------------------------------------------------------
        struct rdma_thread_print_helper
        {
        };

        inline std::ostream& operator<<(
            std::ostream& os, const rdma_thread_print_helper&)
        {
            if (hpx::threads::get_self_id() == hpx::threads::invalid_thread_id)
            {
                os << "------------------ ";
            }
            else
            {
                hpx::threads::thread_data* dummy =
                    hpx::threads::get_self_id_data();
                os << hexpointer(dummy);
            }
            os << nhex(12) << std::this_thread::get_id() << " cpu "
               << decnumber(sched_getcpu());
            return os;
        }

        // ------------------------------------------------------------------
        // helper function for printing CRC32
        // ------------------------------------------------------------------
        inline uint32_t crc32(const void* address, size_t length)
        {
            boost::crc_32_type result;
            result.process_bytes(address, length);
            return result.checksum();
        }

        // ------------------------------------------------------------------
        // helper function for printing CRC32 and short memory dump
        // ------------------------------------------------------------------
        inline std::string mem_crc32(
            const void* address, size_t length, const char* txt)
        {
            const uint64_t* uintBuf = static_cast<const uint64_t*>(address);
            std::stringstream temp;
            temp << "Memory: ";
            temp << "address " << hexpointer(address) << "length "
                 << hexuint32(length)
                 << "CRC32: " << hexuint32(crc32(address, length));
            for (size_t i = 0; i < (std::min)(length / 8, size_t(128)); i++)
            {
                temp << hexuint64(*uintBuf++);
            }
            temp << ": " << txt;
            return temp.str();
        }

}}}}}    // namespace hpx::parcelset::policies::libfabric::detail

#define THREAD_ID                                                              \
    "" << hpx::parcelset::policies::libfabric::detail::                        \
            rdma_thread_print_helper()

#define CRC32(buf, len)                                                        \
    "" << hpx::parcelset::policies::libfabric::detail::crc32(buf, len)

#define CRC32_MEM(buf, len, txt)                                               \
    "" << hpx::parcelset::policies::libfabric::detail::mem_crc32(buf, len, txt)

// ------------------------------------------------------------------
// Trace messages are enabled for full debug
// ------------------------------------------------------------------
#ifdef HPX_PARCELPORT_LOGGING_HAVE_TRACE_LOG
#define LOG_TRACE_MSG(x) BOOST_LOG_TRIVIAL(trace) << THREAD_ID << " " << x;
#else
#define LOG_TRACE_MSG(x)
#endif

// ------------------------------------------------------------------
// if enabled : define all main logging macros
// ------------------------------------------------------------------
#ifdef HPX_PARCELPORT_LOGGING_HAVE_DEBUG_LOG

#define LOG_DEBUG_MSG(x) BOOST_LOG_TRIVIAL(debug) << THREAD_ID << " " << x;
#define LOG_INFO_MSG(x) BOOST_LOG_TRIVIAL(info) << THREAD_ID << " " << x;
#define LOG_WARN_MSG(x) BOOST_LOG_TRIVIAL(warning) << THREAD_ID << " " << x;
#define LOG_ERROR_MSG(x) BOOST_LOG_TRIVIAL(error) << THREAD_ID << " " << x;
#define LOG_FATAL_MSG(x) BOOST_LOG_TRIVIAL(fatal) << THREAD_ID << " " << x;
//
#define LOG_EXCLUSIVE(x) x
//
#define FUNC_START_DEBUG_MSG LOG_TRACE_MSG("*** Enter " << __func__);
#define FUNC_END_DEBUG_MSG LOG_TRACE_MSG("### Exit  " << __func__);
//
#define LOG_FORMAT_MSG(x)                                                      \
    (dynamic_cast<std::ostringstream&>(                                        \
         std::ostringstream().seekp(0, std::ios_base::cur)                     \
         << x << __FILE__ << " " << std::dec << __LINE__))                     \
        .str()

#else
#define LOG_DEBUG_MSG(x)
#define LOG_INFO_MSG(x)
#define LOG_WARN_MSG(x)
#define LOG_ERROR_MSG(x)                                                       \
    std::cout << "00: <ERROR> " << THREAD_ID << " " << x << " " << __FILE__    \
              << " " << std::dec << __LINE__ << std::endl;
#define LOG_FATAL_MSG(x) LOG_ERROR_MSG(x)
//
#define LOG_EXCLUSIVE(x)
//
#define FUNC_START_DEBUG_MSG
#define FUNC_END_DEBUG_MSG
//
#define LOG_FORMAT_MSG(x) ""

#endif

// ------------------------------------------------------------------
// dev logging: just enable the LOG_DEVEL macro to bypass most log output
// but still show some that have been specially marked
// ------------------------------------------------------------------
#ifdef HPX_PARCELPORT_LOGGING_HAVE_DEVEL_LOG
#define LOG_DEVEL_MSG(x)                                                       \
    BOOST_LOG_TRIVIAL(debug) << "" << THREAD_ID << " " << x;
#else
#define LOG_DEVEL_MSG(x)
#endif

// ------------------------------------------------------------------
// Timed log macros : used during long loops to avoid excessive output
// only prints the log message every N seconds
// ------------------------------------------------------------------
#ifdef HPX_PARCELPORT_LOGGING_HAVE_TIMED_LOG

#define LOG_TIMED_INIT(name)                                                   \
    using namespace std::chrono;                                               \
    static time_point<system_clock> log_timed_start_##name =                   \
        system_clock::now();

#define LOG_TIMED_MSG(name, level, delay, x)                                   \
    time_point<system_clock> log_timed_now_##name = system_clock::now();       \
    duration<double> log_timed_elapsed_##name =                                \
        log_timed_now_##name - log_timed_start_##name;                         \
    if (log_timed_elapsed_##name.count() > delay)                              \
    {                                                                          \
        LOG_DEVEL_MSG(x);                                                      \
        log_timed_start_##name = log_timed_now_##name;                         \
    }

#define LOG_TIMED_BLOCK(name, level, delay, x)                                 \
    time_point<system_clock> log_timed_now_##name = system_clock::now();       \
    duration<double> log_timed_elapsed_##name =                                \
        log_timed_now_##name - log_timed_start_##name;                         \
    if (log_timed_elapsed_##name.count() > delay)                              \
    {                                                                          \
        log_timed_start_##name = log_timed_now_##name;                         \
        x;                                                                     \
    }

#else
#define LOG_TIMED_INIT(name)
#define LOG_TIMED_MSG(name, level, delay, x)
#define LOG_TIMED_BLOCK(name, level, delay, x)
#endif

// ------------------------------------------------------------------
// Utility to allow automatic printing of enum names in log messages
//
// example of usage
// DEFINE_ENUM_WITH_STRING_CONVERSIONS(test_type, (test1)(test2)(test3))
// ------------------------------------------------------------------

#define X_DEFINE_ENUM_WITH_STRING_CONVERSIONS_TOSTRING_CASE(r, data, elem)     \
    case elem:                                                                 \
        return HPX_PP_STRINGIZE(elem);                                         \
        /**/

#define DEFINE_ENUM_WITH_STRING_CONVERSIONS(name, enumerators)                 \
    enum name                                                                  \
    {                                                                          \
        BOOST_PP_SEQ_ENUM(enumerators)                                         \
    };                                                                         \
                                                                               \
    static const char* ToString(name v)                                        \
    {                                                                          \
        switch (v)                                                             \
        {                                                                      \
            BOOST_PP_SEQ_FOR_EACH(                                             \
                X_DEFINE_ENUM_WITH_STRING_CONVERSIONS_TOSTRING_CASE, name,     \
                enumerators)                                                   \
        default:                                                               \
            return "[Unknown " HPX_PP_STRINGIZE(name) "]";                     \
        }                                                                      \
    }                                                                          \
    /**/
