rpc: add dumpbeaconcontracts

Author: div72

src/rpc/blockchain.cpp

@@ -201,6 +201,56 @@ UniValue blockToJSON(const CBlock& block, const CBlockIndex* blockindex, bool fP
     return result;
 }
 
+UniValue dumpbeaconcontracts(const UniValue& params, bool fHelp)
+{
+    if (fHelp || params.size() != 1)
+        throw runtime_error(
+                "dumpbeaconcontracts <file>\n"
+                "\n"
+                "<file> Output file.\n"
+                "\n"
+                "Dumps current beacon storage and all beacon contracts in the chain to a file.\n");
+
+    std::string path = params[0].get_str();
+    if (path.empty() || fs::exists(path))
+        throw runtime_error("Invalid path");
+
+    CBlock block;
+    std::vector<CTransaction> vtx;
+    std::vector<GRC::Cpid> active_beacons;
+    std::vector<CKeyID> pending_beacons;
+    CAutoFile file(fsbridge::fopen(path, "wb"), SER_DISK, PROTOCOL_VERSION);
+
+    LOCK(cs_main);
+
+    CBlockIndex* pblockindex = pindexGenesisBlock;
+    while (pblockindex != nullptr) {
+        block.ReadFromDisk(pblockindex);
+        for (auto& tx : block.vtx) {
+            for (const auto& contract : tx.GetContracts()) {
+                if (contract.m_type == GRC::ContractType::BEACON) {
+                    vtx.push_back(tx);
+                    break;
+                }
+            }
+        }
+        pblockindex = pblockindex->pnext;
+    }
+
+    for (const auto& x : GRC::GetBeaconRegistry().Beacons()) {
+        active_beacons.push_back(x.first);
+    }
+
+    for (const auto& x : GRC::GetBeaconRegistry().PendingBeacons()) {
+        pending_beacons.push_back(x.first);
+    }
+
+    file << vtx;
+    file << active_beacons;
+    file << pending_beacons;
+
+    return UniValue(UniValue::VOBJ);
+}
 
 UniValue showblock(const UniValue& params, bool fHelp)
 {

src/rpc/server.cpp

@@ -369,6 +369,7 @@ static const CRPCCommand vRPCCommands[] =
     { "currentcontractaverage",  &currentcontractaverage,  cat_developer     },
     { "debug",                   &debug,                   cat_developer     },
     { "debug10",                 &debug10,                 cat_developer     },
+    { "dumpbeaconcontracts",     &dumpbeaconcontracts,     cat_developer     },
     { "exportstats1",            &rpc_exportstats,         cat_developer     },
     { "getblockstats",           &rpc_getblockstats,       cat_developer     },
     { "getlistof",               &getlistof,               cat_developer     },

src/rpc/server.h

@@ -191,6 +191,7 @@ extern UniValue currentcontractaverage(const UniValue& params, bool fHelp);
 extern UniValue debug(const UniValue& params, bool fHelp);
 extern UniValue debug10(const UniValue& params, bool fHelp);
 extern UniValue debug2(const UniValue& params, bool fHelp);
+extern UniValue dumpbeaconcontracts(const UniValue& params, bool fHelp);
 extern UniValue rpc_getblockstats(const UniValue& params, bool fHelp);
 extern UniValue getlistof(const UniValue& params, bool fHelp);
 extern UniValue inspectaccrualsnapshot(const UniValue& params, bool fHelp);