From 9053ac1dd25713e3e8a8755105e67e5b89514a78 Mon Sep 17 00:00:00 2001
From: Srdjan Obucina <obucinac@gmail.com>
Date: Sun, 22 Mar 2020 09:04:59 +0100
Subject: [PATCH] When updated, import son wallet bitcoin address to bitcoin
 wallet

---
 .../sidechain_net_handler_bitcoin.hpp         |  2 +
 .../sidechain_net_handler_bitcoin.cpp         | 56 ++++++++++++++++++-
 2 files changed, 56 insertions(+), 2 deletions(-)

diff --git a/libraries/plugins/peerplays_sidechain/include/graphene/peerplays_sidechain/sidechain_net_handler_bitcoin.hpp b/libraries/plugins/peerplays_sidechain/include/graphene/peerplays_sidechain/sidechain_net_handler_bitcoin.hpp
index 289808ac..2d3b5d43 100644
--- a/libraries/plugins/peerplays_sidechain/include/graphene/peerplays_sidechain/sidechain_net_handler_bitcoin.hpp
+++ b/libraries/plugins/peerplays_sidechain/include/graphene/peerplays_sidechain/sidechain_net_handler_bitcoin.hpp
@@ -30,6 +30,7 @@ public:
    std::string encryptwallet(const std::string &passphrase);
    uint64_t estimatesmartfee();
    std::string finalizepsbt(std::string const &tx_psbt);
+   std::string getaddressinfo(const std::string &address);
    std::string getblock(const std::string &block_hash, int32_t verbosity = 2);
    void importaddress(const std::string &address_or_script);
    std::vector<btc_txout> listunspent();
@@ -113,6 +114,7 @@ private:
 
    void handle_event(const std::string &event_data);
    std::vector<info_for_vin> extract_info_from_block(const std::string &_block);
+   void on_changed_objects(const vector<object_id_type> &ids, const flat_set<account_id_type> &accounts);
 };
 
 }} // namespace graphene::peerplays_sidechain
diff --git a/libraries/plugins/peerplays_sidechain/sidechain_net_handler_bitcoin.cpp b/libraries/plugins/peerplays_sidechain/sidechain_net_handler_bitcoin.cpp
index b80b4472..b0c9d7d4 100644
--- a/libraries/plugins/peerplays_sidechain/sidechain_net_handler_bitcoin.cpp
+++ b/libraries/plugins/peerplays_sidechain/sidechain_net_handler_bitcoin.cpp
@@ -331,6 +331,34 @@ std::string bitcoin_rpc_client::finalizepsbt(std::string const &tx_psbt) {
    return "";
 }
 
+std::string bitcoin_rpc_client::getaddressinfo(const std::string &address) {
+   std::string body = std::string("{\"jsonrpc\": \"1.0\", \"id\":\"getaddressinfo\", \"method\": "
+                                  "\"getaddressinfo\", \"params\": [\"" +
+                                  address + "\"] }");
+
+   const auto reply = send_post_request(body);
+
+   if (reply.body.empty()) {
+      wlog("Bitcoin RPC call ${function} failed", ("function", __FUNCTION__));
+      return "";
+   }
+
+   std::stringstream ss(std::string(reply.body.begin(), reply.body.end()));
+   boost::property_tree::ptree json;
+   boost::property_tree::read_json(ss, json);
+
+   if (reply.status == 200) {
+      std::stringstream ss;
+      boost::property_tree::json_parser::write_json(ss, json.get_child("result"));
+      return ss.str();
+   }
+
+   if (json.count("error") && !json.get_child("error").empty()) {
+      wlog("Bitcoin RPC call ${function} with body ${body} failed with reply '${msg}'", ("function", __FUNCTION__)("body", body)("msg", ss.str()));
+   }
+   return "";
+}
+
 std::string bitcoin_rpc_client::getblock(const std::string &block_hash, int32_t verbosity) {
    std::string body = std::string("{\"jsonrpc\": \"1.0\", \"id\":\"getblock\", \"method\": "
                                   "\"getblock\", \"params\": [\"" +
@@ -364,7 +392,7 @@ void bitcoin_rpc_client::importaddress(const std::string &address_or_script) {
                                  "\"method\": \"importaddress\", \"params\": [") +
                      std::string("\"") + address_or_script + std::string("\"") + std::string("] }");
 
-   const auto reply = send_post_request(body);
+   const auto reply = send_post_request(body, true);
 
    if (reply.body.empty()) {
       wlog("Bitcoin RPC call ${function} failed", ("function", __FUNCTION__));
@@ -376,7 +404,6 @@ void bitcoin_rpc_client::importaddress(const std::string &address_or_script) {
    boost::property_tree::read_json(ss, json);
 
    if (reply.status == 200) {
-      idump((address_or_script)(ss.str()));
       return;
    } else if (json.count("error") && !json.get_child("error").empty()) {
       wlog("Bitcoin RPC call ${function} with body ${body} failed with reply '${msg}'", ("function", __FUNCTION__)("body", body)("msg", ss.str()));
@@ -761,6 +788,10 @@ sidechain_net_handler_bitcoin::sidechain_net_handler_bitcoin(peerplays_sidechain
    listener->event_received.connect([this](const std::string &event_data) {
       std::thread(&sidechain_net_handler_bitcoin::handle_event, this, event_data).detach();
    });
+
+   database.changed_objects.connect([this](const vector<object_id_type> &ids, const flat_set<account_id_type> &accounts) {
+      on_changed_objects(ids, accounts);
+   });
 }
 
 sidechain_net_handler_bitcoin::~sidechain_net_handler_bitcoin() {
@@ -1318,6 +1349,27 @@ std::vector<info_for_vin> sidechain_net_handler_bitcoin::extract_info_from_block
    return result;
 }
 
+void sidechain_net_handler_bitcoin::on_changed_objects(const vector<object_id_type> &ids, const flat_set<account_id_type> &accounts) {
+   for (auto id : ids) {
+      if (id.is<son_wallet_object>()) {
+         const auto &swi = database.get_index_type<son_wallet_index>().indices().get<by_id>();
+         auto swo = swi.find(id);
+         if (swo != swi.end()) {
+            std::stringstream pw_ss(swo->addresses.at(sidechain));
+            boost::property_tree::ptree pw_pt;
+            boost::property_tree::read_json(pw_ss, pw_pt);
+
+            std::string pw_redeem_script = "";
+
+            if (pw_pt.count("redeemScript")) {
+               pw_redeem_script = pw_pt.get<std::string>("redeemScript");
+               bitcoin_client->importaddress(pw_redeem_script);
+            }
+         }
+      }
+   }
+}
+
 // =============================================================================
 
 }} // namespace graphene::peerplays_sidechain