From e11a94c95cde21a97bc9b2972f148f50a9249f18 Mon Sep 17 00:00:00 2001
From: Sean <darcys22@gmail.com>
Date: Wed, 13 Oct 2021 22:20:36 +1100
Subject: [PATCH] RPC call for summary get_status (#1742)

* RPC call for summary get_status

* lint

* update with review notes

* further review points

* uint64_t
---
 llarp/router/abstractrouter.hpp |  3 ++
 llarp/router/router.cpp         | 71 +++++++++++++++++++++++++++++++++
 llarp/router/router.hpp         |  3 ++
 llarp/rpc/rpc_server.cpp        |  7 ++++
 4 files changed, 84 insertions(+)

diff --git a/llarp/router/abstractrouter.hpp b/llarp/router/abstractrouter.hpp
index 1e43ac0c2..c249ef420 100644
--- a/llarp/router/abstractrouter.hpp
+++ b/llarp/router/abstractrouter.hpp
@@ -324,6 +324,9 @@ namespace llarp
     virtual util::StatusObject
     ExtractStatus() const = 0;
 
+    virtual util::StatusObject
+    ExtractSummaryStatus() const = 0;
+
     /// gossip an rc if required
     virtual void
     GossipRCIfNeeded(const RouterContact rc) = 0;
diff --git a/llarp/router/router.cpp b/llarp/router/router.cpp
index 5655a3092..e7eae19a1 100644
--- a/llarp/router/router.cpp
+++ b/llarp/router/router.cpp
@@ -96,6 +96,77 @@ namespace llarp
     }
   }
 
+  util::StatusObject
+  Router::ExtractSummaryStatus() const
+  {
+    if (!_running)
+      return util::StatusObject{{"running", false}};
+
+    auto services = _hiddenServiceContext.ExtractStatus();
+    auto link_types = _linkManager.ExtractStatus();
+
+    uint64_t tx_rate = 0;
+    uint64_t rx_rate = 0;
+    uint64_t peers = 0;
+    for (const auto& links : link_types)
+    {
+      for (const auto& link : links)
+      {
+        if (link.empty())
+          continue;
+        for (const auto& peer : link["sessions"]["established"])
+        {
+          tx_rate += peer["tx"].get<uint64_t>();
+          rx_rate += peer["rx"].get<uint64_t>();
+          peers++;
+        }
+      }
+    }
+
+    // Compute all stats on all path builders on the default endpoint
+    // Merge snodeSessions, remoteSessions and default into a single array
+    std::vector<nlohmann::json> builders;
+    auto snode_sessions = services["default"]["snodeSessions"];
+    for (const auto& session : snode_sessions)
+      builders.push_back(session["buildStats"]);
+
+    auto remote_sessions = services["default"]["remoteSessions"];
+    for (const auto& session : remote_sessions)
+      builders.push_back(session["buildStats"]);
+
+    builders.push_back(services["default"]["buildStats"]);
+
+    // Iterate over all items on this array to build the global pathStats
+    uint64_t paths = 0;
+    uint64_t success = 0;
+    uint64_t attempts = 0;
+    for (const auto& builder : builders)
+    {
+      if (builder.is_null())
+        continue;
+      if (builder["length"].is_number())
+        paths += builder["length"].get<uint64_t>();
+      if (builder["success"].is_number())
+        success += builder["success"].get<uint64_t>();
+      if (builder["attempts"].is_number())
+        attempts += builder["attempts"].get<uint64_t>();
+    }
+    double ratio = static_cast<double>(success) / (attempts + 1);
+
+    return util::StatusObject{
+        {"running", true},
+        {"authCodes", services["default"]["authCodes"]},
+        {"exitMap", services["default"]["exitMap"]},
+        {"lokiAddress", services["default"]["identity"]},
+        {"numPathsBuilt", paths},
+        {"numPeersConnected", peers},
+        {"numRoutersKnown", _nodedb->NumLoaded()},
+        {"ratio", ratio},
+        {"txRate", tx_rate},
+        {"rxRate", rx_rate},
+    };
+  }
+
   bool
   Router::HandleRecvLinkMessageBuffer(ILinkSession* session, const llarp_buffer_t& buf)
   {
diff --git a/llarp/router/router.hpp b/llarp/router/router.hpp
index 9131dbec9..946bc8594 100644
--- a/llarp/router/router.hpp
+++ b/llarp/router/router.hpp
@@ -103,6 +103,9 @@ namespace llarp
     util::StatusObject
     ExtractStatus() const override;
 
+    util::StatusObject
+    ExtractSummaryStatus() const override;
+
     const std::shared_ptr<NodeDB>&
     nodedb() const override
     {
diff --git a/llarp/rpc/rpc_server.cpp b/llarp/rpc/rpc_server.cpp
index 8e880a187..5bb14e84e 100644
--- a/llarp/rpc/rpc_server.cpp
+++ b/llarp/rpc/rpc_server.cpp
@@ -133,6 +133,13 @@ namespace llarp::rpc
                 defer.reply(data);
               });
             })
+        .add_request_command(
+            "get_status",
+            [&](oxenmq::Message& msg) {
+              m_Router->loop()->call([defer = msg.send_later(), r = m_Router]() {
+                defer.reply(CreateJSONResponse(r->ExtractSummaryStatus()));
+              });
+            })
         .add_request_command(
             "quic_connect",
             [&](oxenmq::Message& msg) {