add process_disclosure API

Author: zoedberg

rpc/src/client.rs

@@ -12,7 +12,7 @@ use std::collections::BTreeSet;
 use std::thread::sleep;
 use std::time::Duration;
 
-use bitcoin::OutPoint;
+use bitcoin::{OutPoint, Txid};
 use internet2::addr::{NodeAddr, ServiceAddr};
 use internet2::ZmqSocketType;
 use lnpbp::chain::Chain;
@@ -260,6 +260,21 @@ impl Client {
             }
         }
     }
+
+    pub fn process_disclosure(
+        &mut self,
+        txid: Txid,
+        progress: impl Fn(String),
+    ) -> Result<bool, Error> {
+        self.request(RpcMsg::ProcessDisclosure(txid))?;
+        loop {
+            match self.response()?.failure_to_error()? {
+                RpcMsg::Success(_) => return Ok(true),
+                RpcMsg::Progress(info) => progress(info),
+                _ => return Err(Error::UnexpectedServerResponse),
+            }
+        }
+    }
 }
 
 pub struct Handler {

rpc/src/messages.rs

@@ -68,6 +68,9 @@ pub enum RpcMsg {
     #[display("accept_transfer(...)")]
     ConsumeTransfer(AcceptReq<TransferConsignment>),
 
+    #[display("process_disclosure({0})")]
+    ProcessDisclosure(Txid),
+
     #[display(inner)]
     Transfer(TransferReq),
 

src/bucketd/processor.rs

@@ -78,6 +78,12 @@ pub enum StashError {
     /// stash data storage.
     BundleAbsent(ContractId, Txid),
 
+    /// disclosure for txid {0} is absent
+    ///
+    /// It may happen due to RGB Node bug, or indicate internal stash inconsistency and compromised
+    /// stash data storage.
+    DisclosureAbsent(Txid),
+
     /// the anchor is not related to the contract
     ///
     /// It may happen due to RGB Node bug, or indicate internal stash inconsistency and compromised
@@ -283,6 +289,72 @@ impl Runtime {
         Ok(status)
     }
 
+    pub(super) fn process_disclosure(
+        &mut self,
+        txid: Txid,
+    ) -> Result<(), DaemonError> {
+        let disclosure: Disclosure = self.store.retrieve_sten(db::DISCLOSURES, txid)?
+            .ok_or(StashError::DisclosureAbsent(txid))?;
+
+        for (_anchor_id, (anchor, bundle_map)) in disclosure.anchored_bundles() {
+            for (contract_id, bundle) in bundle_map {
+                let mut state: ContractState =
+                    self.store.retrieve_sten(db::CONTRACTS, *contract_id)?.ok_or(StashError::StateAbsent(*contract_id))?;
+                trace!("Starting with contract state {:?}", state);
+
+                let bundle_id = bundle.bundle_id();
+                let witness_txid = anchor.txid;
+                debug!("Processing anchored bundle {} for txid {}", bundle_id, witness_txid);
+                trace!("Anchor: {:?}", anchor);
+                trace!("Bundle: {:?}", bundle);
+                self.store.store_sten(db::ANCHORS, anchor.txid, anchor)?;
+                let concealed: BTreeMap<NodeId, BTreeSet<u16>> =
+                    bundle.concealed_iter().map(|(id, set)| (*id, set.clone())).collect();
+                let mut revealed: BTreeMap<Transition, BTreeSet<u16>> = bmap!();
+                for (transition, inputs) in bundle.revealed_iter() {
+                    let node_id = transition.node_id();
+                    let transition_type = transition.transition_type();
+                    debug!("Processing state transition {}", node_id);
+                    trace!("State transition: {:?}", transition);
+
+                    state.add_transition(witness_txid, transition);
+                    trace!("Contract state now is {:?}", state);
+
+                    trace!("Storing state transition data");
+                    revealed.insert(transition.clone(), inputs.clone());
+                    self.store.store_merge(db::TRANSITIONS, node_id, transition.clone())?;
+                    self.store.store_sten(db::TRANSITION_WITNESS, node_id, &witness_txid)?;
+
+                    trace!("Indexing transition");
+                    let index_id = ChunkId::with_fixed_fragments(*contract_id, transition_type);
+                    self.store.insert_into_set(
+                        db::CONTRACT_TRANSITIONS,
+                        index_id,
+                        node_id.into_array(),
+                    )?;
+
+                    self.store.store_sten(db::NODE_CONTRACTS, node_id, contract_id)?;
+
+                    for seal in transition.filter_revealed_seals() {
+                        let index_id = ChunkId::with_fixed_fragments(
+                            seal.txid.expect("seal should contain revealed txid"),
+                            seal.vout,
+                        );
+                        self.store.insert_into_set(db::OUTPOINTS, index_id, node_id.into_array())?;
+                    }
+                }
+                let data = TransitionBundle::with(revealed, concealed)
+                    .expect("enough data should be available to create bundle");
+                let chunk_id = ChunkId::with_fixed_fragments(*contract_id, witness_txid);
+                self.store.store_sten(db::BUNDLES, chunk_id, &data)?;
+
+                self.store.store_sten(db::CONTRACTS, *contract_id, &state)?;
+            }
+        }
+
+        Ok(())
+    }
+
     pub(super) fn compose_consignment<T: ConsignmentType>(
         &mut self,
         contract_id: ContractId,

src/bucketd/service.rs

@@ -15,7 +15,7 @@ use std::time::Duration;
 
 use amplify::num::u24;
 use bitcoin::secp256k1::rand::random;
-use bitcoin::OutPoint;
+use bitcoin::{OutPoint, Txid};
 use commit_verify::ConsensusCommit;
 use electrum_client::Client as ElectrumClient;
 use internet2::addr::NodeAddr;
@@ -39,7 +39,7 @@ use strict_encoding::{MediumVec, StrictEncode};
 
 use crate::bus::{
     BusMsg, ConsignReq, CtlMsg, DaemonId, Endpoints, FinalizeTransferReq, OutpointStateReq,
-    ProcessReq, Responder, ServiceBus, ServiceId, ValidityResp,
+    ProcessDisclosureReq, ProcessReq, Responder, ServiceBus, ServiceId, ValidityResp,
 };
 use crate::{Config, DaemonError, LaunchError};
 
@@ -178,6 +178,12 @@ impl Runtime {
             }) => {
                 self.handle_consignment(endpoints, client_id, consignment, force)?;
             }
+            CtlMsg::ProcessDisclosure(ProcessDisclosureReq {
+                client_id,
+                txid,
+            }) => {
+                self.handle_disclosure(endpoints, client_id, txid)?;
+            }
             CtlMsg::ProcessTransferContainer(container_id) => {
                 self.handle_container(endpoints, container_id)?;
             }
@@ -310,6 +316,26 @@ impl Runtime {
         Ok(())
     }
 
+    fn handle_disclosure(
+        &mut self,
+        endpoints: &mut Endpoints,
+        client_id: ClientId,
+        txid: Txid,
+    ) -> Result<(), DaemonError> {
+        match self.process_disclosure(txid) {
+            Err(err) => {
+                let _ = self.send_rpc(endpoints, client_id, err);
+                self.send_ctl(endpoints, ServiceId::rgbd(), CtlMsg::ProcessingFailed)?
+            }
+            Ok(_) => {
+                let _ =
+                    self.send_rpc(endpoints, client_id, RpcMsg::success());
+                self.send_ctl(endpoints, ServiceId::rgbd(), CtlMsg::ProcessingComplete)?
+            }
+        }
+        Ok(())
+    }
+
     fn handle_consign_contract(
         &mut self,
         endpoints: &mut Endpoints,

src/bus/ctl.rs

@@ -10,7 +10,7 @@
 
 use std::collections::BTreeSet;
 
-use bitcoin::OutPoint;
+use bitcoin::{OutPoint, Txid};
 use internet2::addr::NodeAddr;
 use microservices::esb::ClientId;
 use psbt::Psbt;
@@ -36,6 +36,9 @@ pub enum CtlMsg {
     #[display("process_transfer({0})")]
     ProcessTransfer(ProcessReq<TransferConsignment>),
 
+    #[display("process_disclosure({0})")]
+    ProcessDisclosure(ProcessDisclosureReq),
+
     #[display("process_transfer_container({0})")]
     ProcessTransferContainer(ContainerId),
 
@@ -70,6 +73,13 @@ pub struct ProcessReq<T: ConsignmentType> {
     pub force: bool,
 }
 
+#[derive(Clone, Debug, Display, StrictEncode, StrictDecode)]
+#[display("{client_id}, txid = {txid}, ...")]
+pub struct ProcessDisclosureReq {
+    pub client_id: ClientId,
+    pub txid: Txid,
+}
+
 #[derive(Clone, Debug, Display, StrictEncode, StrictDecode)]
 #[display("{client_id}, {contract_id}, ...")]
 pub struct ConsignReq<T: ConsignmentType> {

src/bus/mod.rs

@@ -16,7 +16,8 @@ use rgb_rpc::RpcMsg;
 use storm_ext::ExtMsg as StormMsg;
 
 pub use self::ctl::{
-    ConsignReq, CtlMsg, FinalizeTransferReq, OutpointStateReq, ProcessReq, ValidityResp,
+    ConsignReq, CtlMsg, FinalizeTransferReq, OutpointStateReq, ProcessDisclosureReq,
+    ProcessReq, ValidityResp,
 };
 pub use self::services::{DaemonId, ServiceId};
 pub(crate) use self::services::{Endpoints, Responder, ServiceBus};

src/rgbd/service.rs

@@ -12,7 +12,7 @@ use std::collections::{BTreeSet, VecDeque};
 
 use amplify::Wrapper;
 use bitcoin::hashes::Hash;
-use bitcoin::OutPoint;
+use bitcoin::{OutPoint, Txid};
 use internet2::addr::NodeAddr;
 use internet2::ZmqSocketType;
 use lnpbp::chain::Chain;
@@ -34,7 +34,7 @@ use storm_rpc::AddressedMsg;
 use crate::bucketd::StashError;
 use crate::bus::{
     BusMsg, ConsignReq, CtlMsg, DaemonId, Endpoints, FinalizeTransferReq, OutpointStateReq,
-    ProcessReq, Responder, ServiceBus, ServiceId,
+    ProcessDisclosureReq, ProcessReq, Responder, ServiceBus, ServiceId,
 };
 use crate::db::ChunkHolder;
 use crate::rgbd::daemons::Daemon;
@@ -254,6 +254,9 @@ impl Runtime {
             }) => {
                 self.accept_transfer(endpoints, client_id, transfer, force)?;
             }
+            RpcMsg::ProcessDisclosure(txid) => {
+                self.process_disclosure(endpoints, client_id, txid)?;
+            }
 
             RpcMsg::Transfer(TransferReq {
                 consignment,
@@ -507,6 +510,19 @@ impl Runtime {
         self.pick_or_start(endpoints, client_id)
     }
 
+    fn process_disclosure(
+        &mut self,
+        endpoints: &mut Endpoints,
+        client_id: ClientId,
+        txid: Txid,
+    ) -> Result<(), DaemonError> {
+        self.ctl_queue.push_back(CtlMsg::ProcessDisclosure(ProcessDisclosureReq {
+            client_id,
+            txid,
+        }));
+        self.pick_or_start(endpoints, client_id)
+    }
+
     fn process_transfer(
         &mut self,
         endpoints: &mut Endpoints,