#include <peerstats/types.hpp>

#include <util/str.hpp>
#include <lokimq/bt_serialize.h>
#include <stdexcept>

namespace llarp
{
  constexpr auto RouterIdKey = "routerId";
  constexpr auto NumConnectionAttemptsKey = "numConnectionAttempts";
  constexpr auto NumConnectionSuccessesKey = "numConnectionSuccesses";
  constexpr auto NumConnectionRejectionsKey = "numConnectionRejections";
  constexpr auto NumConnectionTimeoutsKey = "numConnectionTimeouts";
  constexpr auto NumPathBuildsKey = "numPathBuilds";
  constexpr auto NumPacketsAttemptedKey = "numPacketsAttempted";
  constexpr auto NumPacketsSentKey = "numPacketsSent";
  constexpr auto NumPacketsDroppedKey = "numPacketsDropped";
  constexpr auto NumPacketsResentKey = "numPacketsResent";
  constexpr auto NumDistinctRCsReceivedKey = "numDistinctRCsReceived";
  constexpr auto NumLateRCsKey = "numLateRCs";
  constexpr auto PeakBandwidthBytesPerSecKey = "peakBandwidthBytesPerSec";
  constexpr auto LongestRCReceiveIntervalKey = "longestRCReceiveInterval";
  constexpr auto LeastRCRemainingLifetimeKey = "leastRCRemainingLifetime";
  constexpr auto LastRCUpdatedKey = "lastRCUpdated";

  PeerStats::PeerStats() = default;

  PeerStats::PeerStats(const RouterID& routerId_) : routerId(routerId_)
  {}

  PeerStats&
  PeerStats::operator+=(const PeerStats& other)
  {
    numConnectionAttempts += other.numConnectionAttempts;
    numConnectionSuccesses += other.numConnectionSuccesses;
    numConnectionRejections += other.numConnectionRejections;
    numConnectionTimeouts += other.numConnectionTimeouts;

    numPathBuilds += other.numPathBuilds;
    numPacketsAttempted += other.numPacketsAttempted;
    numPacketsSent += other.numPacketsSent;
    numPacketsDropped += other.numPacketsDropped;
    numPacketsResent += other.numPacketsResent;

    numDistinctRCsReceived += other.numDistinctRCsReceived;
    numLateRCs += other.numLateRCs;

    peakBandwidthBytesPerSec = std::max(peakBandwidthBytesPerSec, other.peakBandwidthBytesPerSec);
    longestRCReceiveInterval = std::max(longestRCReceiveInterval, other.longestRCReceiveInterval);
    leastRCRemainingLifetime = std::max(leastRCRemainingLifetime, other.leastRCRemainingLifetime);
    lastRCUpdated = std::max(lastRCUpdated, other.lastRCUpdated);

    return *this;
  }

  bool
  PeerStats::operator==(const PeerStats& other) const
  {
    return routerId == other.routerId and numConnectionAttempts == other.numConnectionAttempts
        and numConnectionSuccesses == other.numConnectionSuccesses
        and numConnectionRejections == other.numConnectionRejections
        and numConnectionTimeouts == other.numConnectionTimeouts

        and numPathBuilds == other.numPathBuilds
        and numPacketsAttempted == other.numPacketsAttempted
        and numPacketsSent == other.numPacketsSent and numPacketsDropped == other.numPacketsDropped
        and numPacketsResent == other.numPacketsResent

        and numDistinctRCsReceived == other.numDistinctRCsReceived
        and numLateRCs == other.numLateRCs

        and peakBandwidthBytesPerSec == other.peakBandwidthBytesPerSec
        and longestRCReceiveInterval == other.longestRCReceiveInterval
        and leastRCRemainingLifetime == other.leastRCRemainingLifetime
        and lastRCUpdated == other.lastRCUpdated;
  }

  util::StatusObject
  PeerStats::toJson() const
  {
    return {
        {RouterIdKey, routerId.ToString()},
        {NumConnectionAttemptsKey, numConnectionAttempts},
        {NumConnectionSuccessesKey, numConnectionSuccesses},
        {NumConnectionRejectionsKey, numConnectionRejections},
        {NumConnectionTimeoutsKey, numConnectionTimeouts},
        {NumPathBuildsKey, numPathBuilds},
        {NumPacketsAttemptedKey, numPacketsAttempted},
        {NumPacketsSentKey, numPacketsSent},
        {NumPacketsDroppedKey, numPacketsDropped},
        {NumPacketsResentKey, numPacketsResent},
        {NumDistinctRCsReceivedKey, numDistinctRCsReceived},
        {NumLateRCsKey, numLateRCs},
        {PeakBandwidthBytesPerSecKey, peakBandwidthBytesPerSec},
        {LongestRCReceiveIntervalKey, longestRCReceiveInterval.count()},
        {LeastRCRemainingLifetimeKey, leastRCRemainingLifetime.count()},
        {LastRCUpdatedKey, lastRCUpdated.count()},
    };
  }

  void
  PeerStats::BEncode(llarp_buffer_t* buf) const
  {
    if (not buf)
      throw std::runtime_error("PeerStats: Can't use null buf");
    const lokimq::bt_dict data = {
        {NumConnectionAttemptsKey, numConnectionAttempts},
        {NumConnectionSuccessesKey, numConnectionSuccesses},
        {NumConnectionRejectionsKey, numConnectionRejections},
        {NumConnectionTimeoutsKey, numConnectionTimeouts},
        {NumPathBuildsKey, numPathBuilds},
        {NumPacketsAttemptedKey, numPacketsAttempted},
        {NumPacketsSentKey, numPacketsSent},
        {NumPacketsDroppedKey, numPacketsDropped},
        {NumPacketsResentKey, numPacketsResent},
        {NumDistinctRCsReceivedKey, numDistinctRCsReceived},
        {NumLateRCsKey, numLateRCs},
        {PeakBandwidthBytesPerSecKey, (uint64_t)peakBandwidthBytesPerSec},
        {LongestRCReceiveIntervalKey, longestRCReceiveInterval.count()},
        {LeastRCRemainingLifetimeKey, leastRCRemainingLifetime.count()},
        {LastRCUpdatedKey, lastRCUpdated.count()},
    };
    const auto serialized = lokimq::bt_serialize(data);
    if (not buf->write(serialized.begin(), serialized.end()))
      throw std::runtime_error("PeerStats: buffer too small");
  }

  void
  PeerStats::BEncodeList(const std::vector<PeerStats>& statsList, llarp_buffer_t* buf)
  {
    if (not buf)
      throw std::runtime_error("PeerStats: Can't use null buf");

    if (not bencode_start_list(buf))
      throw std::runtime_error("PeerStats: Could not create bencode dict");

    for (const auto& stats : statsList)
    {
      stats.BEncode(buf);
    }

    if (not bencode_end(buf))
      throw std::runtime_error("PeerStats: Could not end bencode dict");
  }

};  // namespace llarp