Update Pool Box

core/src/box_kind.rs

@@ -2,8 +2,10 @@ mod ballot_box;
 mod oracle_box;
 mod pool_box;
 mod refresh_box;
+mod update_box;
 
 pub use ballot_box::*;
 pub use oracle_box::*;
 pub use pool_box::*;
 pub use refresh_box::*;
+pub use update_box::*;

core/src/box_kind/ballot_box.rs

@@ -131,7 +131,7 @@ impl TryFrom<ErgoBox> for BallotBoxWrapper {
         if ergo_box
             .get_register(NonMandatoryRegisterId::R8.into())
             .ok_or(BallotBoxError::NoRewardTokenQuantityInR8)?
-            .try_extract_into::<i32>()
+            .try_extract_into::<i64>()
             .is_err()
         {
             return Err(BallotBoxError::NoRewardTokenQuantityInR8);
@@ -153,6 +153,7 @@ pub fn make_local_ballot_box_candidate(
     value: BoxValue,
     creation_height: u32,
 ) -> Result<ErgoBoxCandidate, ErgoBoxCandidateBuilderError> {
+    dbg!(&reward_tokens);
     let mut builder = ErgoBoxCandidateBuilder::new(value, contract.ergo_tree(), creation_height);
     builder.set_register_value(
         NonMandatoryRegisterId::R4,
@@ -169,7 +170,7 @@ pub fn make_local_ballot_box_candidate(
     );
     builder.set_register_value(
         NonMandatoryRegisterId::R8,
-        (*reward_tokens.amount.as_u64() as i32).into(),
+        (*reward_tokens.amount.as_u64() as i64).into(),
     );
     builder.add_token(ballot_token);
     builder.build()

core/src/box_kind/update_box.rs

@@ -0,0 +1,74 @@
+use std::convert::TryFrom;
+
+use ergo_lib::chain::ergo_box::box_builder::ErgoBoxCandidateBuilder;
+use ergo_lib::chain::ergo_box::box_builder::ErgoBoxCandidateBuilderError;
+use ergo_lib::ergo_chain_types::EcPoint;
+use ergo_lib::ergotree_ir::chain::ergo_box::box_value::BoxValue;
+use ergo_lib::ergotree_ir::chain::ergo_box::ErgoBox;
+use ergo_lib::ergotree_ir::chain::ergo_box::ErgoBoxCandidate;
+use ergo_lib::ergotree_ir::chain::ergo_box::NonMandatoryRegisterId;
+use ergo_lib::ergotree_ir::chain::token::Token;
+use ergo_lib::ergotree_ir::chain::token::TokenId;
+use ergo_lib::ergotree_ir::ergo_tree::ErgoTree;
+use ergo_lib::ergotree_ir::mir::constant::TryExtractInto;
+use ergo_lib::ergotree_ir::sigma_protocol::sigma_boolean::ProveDlog;
+use thiserror::Error;
+
+use crate::contracts::update::UpdateContract;
+use crate::contracts::update::UpdateContractError;
+
+#[derive(Debug, Error)]
+pub enum UpdateBoxError {
+    #[error("oracle box: no tokens found")]
+    NoTokens,
+    #[error("update contract: {0:?}")]
+    UpdateContractError(#[from] UpdateContractError),
+}
+
+#[derive(Clone)]
+pub struct UpdateBoxWrapper(ErgoBox, UpdateContract);
+
+impl UpdateBoxWrapper {
+    pub fn new(b: ErgoBox) -> Result<Self, UpdateBoxError> {
+        let _update_token_id = b
+            .tokens
+            .as_ref()
+            .ok_or(UpdateBoxError::NoTokens)?
+            .get(0)
+            .ok_or(UpdateBoxError::NoTokens)?
+            .token_id
+            .clone();
+        let contract = UpdateContract::from_ergo_tree(b.ergo_tree.clone())?;
+
+        Ok(Self(b, contract))
+    }
+    pub fn ergo_tree(&self) -> ErgoTree {
+        self.1.ergo_tree()
+    }
+    pub fn update_nft(&self) -> Token {
+        self.0.tokens.as_ref().unwrap().get(0).unwrap().clone()
+    }
+    pub fn ballot_token_id(&self) -> TokenId {
+        self.1.ballot_token_id().clone()
+    }
+    pub fn get_box(&self) -> &ErgoBox {
+        &self.0
+    }
+    pub fn min_votes(&self) -> u32 {
+        self.1.min_votes() as u32
+    }
+}
+
+impl TryFrom<ErgoBox> for UpdateBoxWrapper {
+    type Error = UpdateBoxError;
+
+    fn try_from(value: ErgoBox) -> Result<Self, Self::Error> {
+        UpdateBoxWrapper::new(value)
+    }
+}
+
+impl From<UpdateBoxWrapper> for ErgoBox {
+    fn from(w: UpdateBoxWrapper) -> Self {
+        w.0.clone()
+    }
+}

core/src/cli_commands.rs

@@ -4,6 +4,7 @@ pub mod bootstrap;
 pub mod extract_reward_tokens;
 pub mod print_reward_tokens;
 pub mod transfer_oracle_token;
+pub mod update_pool;
 pub mod vote_update_pool;
 
 pub(crate) fn ergo_explorer_transaction_link(tx_id_str: String, prefix: NetworkPrefix) -> String {