[replay-2] Make progress during replay of a digests file nicer (#24515)

## Description 

Make the output when replaying a digests file much nicer. 

## Test plan 

Ran it locally. 

---

## Release notes

Check each box that your changes affect. If none of the boxes relate to
your changes, release notes aren't required.

For each box you select, include information after the relevant heading
that describes the impact of your changes that a user might notice and
any actions they must take to implement updates.

- [ ] Protocol: 
- [ ] Nodes (Validators and Full nodes): 
- [ ] gRPC:
- [ ] JSON-RPC: 
- [ ] GraphQL: 
- [ ] CLI: 
- [ ] Rust SDK:
- [ ] Indexing Framework:

Author: tzakian

Cargo.lock

@@ -45,6 +45,7 @@ tokio = { workspace = true, features = ["full"] }
 toml.workspace = true
 tracing.workspace = true
 zstd.workspace = true
+indicatif.workspace = true
 
 [features]
 tracing = [

crates/sui-replay-2/Cargo.toml

@@ -9,12 +9,14 @@ use crate::{
 };
 use anyhow::{Result, anyhow, bail};
 use clap::{Parser, ValueEnum};
+use indicatif::{MultiProgress, ProgressBar, ProgressStyle};
 use serde::Deserialize;
 use similar::{ChangeTag, TextDiff};
 use std::{
     fs,
     io::Write,
     path::{Path, PathBuf},
+    time::Duration,
 };
 use sui_config::sui_config_dir;
 use sui_data_store::{
@@ -24,7 +26,7 @@ use sui_data_store::{
 use sui_json_rpc_types::SuiTransactionBlockEffects;
 use sui_types::effects::TransactionEffects;
 // Disambiguate external tracing crate from local `crate::tracing` module using absolute path.
-use ::tracing::{Instrument, debug, error, info, info_span, warn};
+use ::tracing::{Instrument, debug, error, info_span, warn};
 
 pub mod artifacts;
 pub mod displays;
@@ -505,11 +507,20 @@ where
     let mut total_metrics = TotalMetrics::new();
     let mut executor_provider = ExecutorProvider::new(cache_executor);
 
+    let mp = MultiProgress::new();
+    let tx_spinner = mp.add(ProgressBar::new_spinner());
+    let progress_bar = mp.add(ProgressBar::new(digests.len() as u64));
+
+    tx_spinner.set_style(ProgressStyle::with_template("{spinner}: {msg}").unwrap());
+    tx_spinner.enable_steady_tick(Duration::from_millis(80));
+
     for tx_digest in digests {
         let tx_dir = output_root_dir.join(tx_digest);
         let artifact_manager = ArtifactManager::new(&tx_dir, overwrite_existing)?;
         let span = info_span!("replay", tx_digest = %tx_digest);
 
+        tx_spinner.set_message(format!("Executing transaction {}", tx_digest));
+
         let tx_start = Instant::now();
         let result = replay_transaction(
             &artifact_manager,
@@ -530,14 +541,17 @@ where
 
         // Print per-transaction result
         let status = if success { "OK" } else { "FAILED" };
-        if track_time {
-            println!(
-                "> Replayed txn {} ({}): exec_ms={}, total_ms={}",
-                tx_digest, status, exec_ms, tx_total_ms
-            );
+
+        let time_info = if track_time {
+            format!(
+                " ({}): exec_ms={}, total_ms={}",
+                status, exec_ms, tx_total_ms
+            )
         } else {
-            println!("> Replayed txn {} ({})", tx_digest, status);
-        }
+            "".to_owned()
+        };
+
+        tx_spinner.println(format!("Executed transaction {}{}", tx_digest, time_info));
 
         match result {
             Err(e) if terminate_early => {
@@ -547,12 +561,13 @@ where
             Err(e) => {
                 error!(tx_digest = %tx_digest, error = ?e, "Replay failed");
             }
-            Ok(_) => {
-                info!(tx_digest = %tx_digest, "Replay succeeded");
-            }
+            Ok(_) => {}
         }
+        progress_bar.inc(1);
     }
 
+    tx_spinner.finish_and_clear();
+
     if verbose {
         let mut out = std::io::stdout().lock();
         let _ = writeln!(out, "\nData store summary:");

crates/sui-replay-2/src/lib.rs

@@ -39,7 +39,7 @@ use sui_types::{
     gas::SuiGasStatusAPI,
     transaction::{InputObjectKind, ObjectReadResult, ObjectReadResultKind},
 };
-use tracing::{debug, error, info, info_span, trace};
+use tracing::{debug, error, info_span, trace, warn};
 
 pub type ObjectVersion = u64;
 pub type PackageVersion = u64;
@@ -148,7 +148,7 @@ pub(crate) async fn replay_transaction<S: ReadDataStore>(
     }
 
     // Save results
-    info!(
+    debug!(
         tx_digest = %tx_digest,
         result = ?result,
         output_dir = %artifact_manager.base_path.display(),
@@ -191,15 +191,15 @@ pub(crate) async fn replay_transaction<S: ReadDataStore>(
     if let sui_types::transaction::TransactionKind::ProgrammableTransaction(ptb) =
         context_and_effects.txn_data.kind()
     {
-        info!(tx_digest = %tx_digest, "Extracting move call info for {} commands", ptb.commands.len());
+        debug!(tx_digest = %tx_digest, "Extracting move call info for {} commands", ptb.commands.len());
         match MoveCallInfo::from_transaction(ptb, &context_and_effects.object_cache) {
             Ok(move_call_info) => {
                 let successful_extractions = move_call_info
                     .command_signatures
                     .iter()
                     .filter(|s| s.is_some())
                     .count();
-                info!(tx_digest = %tx_digest, "Successfully extracted {} function signatures out of {} commands",
+                debug!(tx_digest = %tx_digest, "Successfully extracted {} function signatures out of {} commands",
                     successful_extractions, move_call_info.command_signatures.len());
 
                 artifact_manager
@@ -209,7 +209,7 @@ pub(crate) async fn replay_transaction<S: ReadDataStore>(
                     .unwrap();
             }
             Err(e) => {
-                info!(tx_digest = %tx_digest, "Failed to extract move call info: {}", e);
+                warn!(tx_digest = %tx_digest, "Failed to extract move call info: {}", e);
             }
         }
     }