From 80a188e6a375ec5d03e2a094ba973b02de0af941 Mon Sep 17 00:00:00 2001
From: Daniil Polyakov <arjentix@gmail.com>
Date: Tue, 9 Dec 2025 00:07:10 +0300
Subject: [PATCH] feat: remove print spam when running wallet account
 sync-private

---
 wallet/Cargo.toml           | 1 +
 wallet/src/chain_storage.rs | 3 ++-
 wallet/src/lib.rs           | 7 ++++++-
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/wallet/Cargo.toml b/wallet/Cargo.toml
index 6f97c63..c93b357 100644
--- a/wallet/Cargo.toml
+++ b/wallet/Cargo.toml
@@ -23,6 +23,7 @@ itertools.workspace = true
 sha2.workspace = true
 futures.workspace = true
 async-stream = "0.3.6"
+indicatif = { version = "0.18.3", features = ["improved_unicode"] }
 
 [dependencies.key_protocol]
 path = "../key_protocol"
diff --git a/wallet/src/chain_storage.rs b/wallet/src/chain_storage.rs
index 0625fce..68d685f 100644
--- a/wallet/src/chain_storage.rs
+++ b/wallet/src/chain_storage.rs
@@ -8,6 +8,7 @@ use key_protocol::{
     },
     key_protocol_core::NSSAUserData,
 };
+use log::debug;
 use nssa::program::Program;
 
 use crate::config::{InitialAccountData, PersistentAccountData, WalletConfig};
@@ -127,7 +128,7 @@ impl WalletChainStore {
         account_id: nssa::AccountId,
         account: nssa_core::account::Account,
     ) {
-        println!("inserting at address {account_id}, this account {account:?}");
+        debug!("inserting at address {account_id}, this account {account:?}");
 
         let entry = self
             .user_data
diff --git a/wallet/src/lib.rs b/wallet/src/lib.rs
index 91a0e4b..b27115c 100644
--- a/wallet/src/lib.rs
+++ b/wallet/src/lib.rs
@@ -306,11 +306,14 @@ impl WalletCore {
         }
 
         let before_polling = std::time::Instant::now();
+        let num_of_blocks = block_id - self.last_synced_block;
+        println!("Syncing to block {block_id}. Blocks to sync: {num_of_blocks}");
 
         let poller = self.poller.clone();
         let mut blocks =
             std::pin::pin!(poller.poll_block_range(self.last_synced_block + 1..=block_id));
 
+        let bar = indicatif::ProgressBar::new(num_of_blocks);
         while let Some(block) = blocks.try_next().await? {
             for tx in block.transactions {
                 let nssa_tx = NSSATransaction::try_from(&tx)?;
@@ -319,7 +322,9 @@ impl WalletCore {
 
             self.last_synced_block = block.block_id;
             self.store_persistent_data().await?;
+            bar.inc(1);
         }
+        bar.finish();
 
         println!(
             "Synced to block {block_id} in {:?}",
@@ -379,7 +384,7 @@ impl WalletCore {
             .collect::<Vec<_>>();
 
         for (affected_account_id, new_acc) in affected_accounts {
-            println!(
+            info!(
                 "Received new account for account_id {affected_account_id:#?} with account object {new_acc:#?}"
             );
             self.storage